Switch to more uses of wcstokenizer from wcstok()

Work towards cleaning up path_get_path
This commit is contained in:
ridiculousfish 2012-07-20 15:01:56 -07:00
parent 966bbd476f
commit b290fd33b9
3 changed files with 41 additions and 68 deletions

View file

@ -1056,12 +1056,8 @@ void completer_t::complete_cmd( const wcstring &str_cmd, bool use_function, bool
if (str_cmd.empty()) if (str_cmd.empty())
return; return;
wchar_t *path_cpy;
wchar_t *nxt_path;
wchar_t *state;
std::vector<completion_t> possible_comp; std::vector<completion_t> possible_comp;
env_var_t cdpath = env_get_string(L"CDPATH"); env_var_t cdpath = env_get_string(L"CDPATH");
if (cdpath.missing_or_empty()) if (cdpath.missing_or_empty())
cdpath = L"."; cdpath = L".";
@ -1090,14 +1086,10 @@ void completer_t::complete_cmd( const wcstring &str_cmd, bool use_function, bool
const env_var_t path = env_get_string(L"PATH"); const env_var_t path = env_get_string(L"PATH");
if( !path.missing() ) if( !path.missing() )
{ {
wcstring base_path;
path_cpy = wcsdup( path.c_str() ); wcstokenizer tokenizer(path, ARRAY_SEP_STR);
while (tokenizer.next(base_path))
for( nxt_path = wcstok( path_cpy, ARRAY_SEP_STR, &state );
nxt_path != 0;
nxt_path = wcstok( 0, ARRAY_SEP_STR, &state) )
{ {
wcstring base_path = nxt_path;
if (base_path.empty()) if (base_path.empty())
continue; continue;
@ -1125,7 +1117,6 @@ void completer_t::complete_cmd( const wcstring &str_cmd, bool use_function, bool
} }
} }
} }
free( path_cpy );
if (wants_description) if (wants_description)
this->complete_cmd_desc( str_cmd ); this->complete_cmd_desc( str_cmd );
} }

View file

@ -126,37 +126,39 @@ bool path_get_path_string(const wcstring &cmd_str, wcstring &output, const env_v
} }
wchar_t *path_get_path( const wchar_t *cmd ) bool path_get_path_string(const wcstring &cmd, wcstring &out_path)
{ {
int err = ENOENT; int err = ENOENT;
CHECK( cmd, 0 ); debug( 3, L"path_get_path( '%ls' )", cmd.c_str() );
debug( 3, L"path_get_path( '%ls' )", cmd ); /* If the command has a slash, it must be a full path */
if (cmd.find(L'/') != wcstring::npos)
if(wcschr( cmd, L'/' ) != 0 )
{ {
if( waccess( cmd, X_OK )==0 ) if( waccess( cmd, X_OK )==0 )
{ {
struct stat buff; struct stat buff;
if(wstat( cmd, &buff )) if(wstat( cmd, &buff ))
{ {
return 0; return false;
} }
if( S_ISREG(buff.st_mode) ) if( S_ISREG(buff.st_mode) )
return wcsdup( cmd ); {
out_path = cmd;
return true;
}
else else
{ {
errno = EACCES; errno = EACCES;
return 0; return false;
} }
} }
else else
{ {
struct stat buff; struct stat buff;
wstat( cmd, &buff ); wstat( cmd, &buff );
return 0; return false;
} }
} }
@ -175,38 +177,17 @@ wchar_t *path_get_path( const wchar_t *cmd )
} }
} }
/* wcstring nxt_path;
Allocate string long enough to hold the whole command wcstokenizer tokenizer(path, ARRAY_SEP_STR);
*/ while (tokenizer.next(nxt_path))
wchar_t *new_cmd = (wchar_t *)calloc(wcslen(cmd)+path.size()+2, sizeof(wchar_t) );
/*
We tokenize a copy of the path, since strtok modifies
its arguments
*/
wchar_t *path_cpy = wcsdup( path.c_str() );
wchar_t *state;
if( (new_cmd==0) || (path_cpy==0) )
{ {
DIE_MEM(); if (nxt_path.empty())
} continue;
append_path_component(nxt_path, cmd);
for( const wchar_t *nxt_path = wcstok( path_cpy, ARRAY_SEP_STR, &state ); if( waccess( nxt_path, X_OK )==0 )
nxt_path != 0;
nxt_path = wcstok( 0, ARRAY_SEP_STR, &state) )
{
int path_len = wcslen( nxt_path );
wcscpy( new_cmd, nxt_path );
if( new_cmd[path_len-1] != L'/' )
{
new_cmd[path_len++]=L'/';
}
wcscpy( &new_cmd[path_len], cmd );
if( waccess( new_cmd, X_OK )==0 )
{ {
struct stat buff; struct stat buff;
if( wstat( new_cmd, &buff )==-1 ) if( wstat( nxt_path, &buff )==-1 )
{ {
if( errno != EACCES ) if( errno != EACCES )
{ {
@ -216,8 +197,8 @@ wchar_t *path_get_path( const wchar_t *cmd )
} }
if( S_ISREG(buff.st_mode) ) if( S_ISREG(buff.st_mode) )
{ {
free( path_cpy ); out_path.swap(nxt_path);
return new_cmd; return true;
} }
err = EACCES; err = EACCES;
@ -235,33 +216,33 @@ wchar_t *path_get_path( const wchar_t *cmd )
{ {
debug( 1, debug( 1,
MISSING_COMMAND_ERR_MSG, MISSING_COMMAND_ERR_MSG,
new_cmd ); nxt_path.c_str() );
wperror( L"access" ); wperror( L"access" );
} }
} }
} }
} }
free( new_cmd );
free( path_cpy );
} }
errno = err; errno = err;
return 0; return false;
} }
bool path_get_path_string(const wcstring &cmd, wcstring &output) wchar_t *path_get_path(const wchar_t *cmd)
{ {
bool success = false; wcstring tmp;
wchar_t *tmp = path_get_path(cmd.c_str()); wchar_t *result = NULL;
if (tmp) { if (cmd != NULL && path_get_path_string(cmd, tmp))
output = tmp; {
free(tmp); result = wcsdup(tmp.c_str());
success = true;
} }
return success; return result;
} }
bool path_get_path( const wcstring &cmd, wcstring &output )
{
return path_get_path_string(cmd, output);
}
bool path_get_cdpath_string(const wcstring &dir_str, wcstring &result, const env_vars &vars) bool path_get_cdpath_string(const wcstring &dir_str, wcstring &result, const env_vars &vars)
{ {

1
path.h
View file

@ -30,6 +30,7 @@ bool path_get_config(wcstring &path);
\return 0 if the command can not be found, the path of the command otherwise. The result should be freed with free(). \return 0 if the command can not be found, the path of the command otherwise. The result should be freed with free().
*/ */
wchar_t *path_get_path( const wchar_t *cmd ); wchar_t *path_get_path( const wchar_t *cmd );
bool path_get_path( const wcstring &cmd, wcstring &output );
/** Returns whether the path can be used for an implicit cd command; if so, also returns the path by reference (if desired). This requires it to start with one of the allowed prefixes (., .., ~) and resolve to a directory. */ /** Returns whether the path can be used for an implicit cd command; if so, also returns the path by reference (if desired). This requires it to start with one of the allowed prefixes (., .., ~) and resolve to a directory. */
bool path_can_be_implicit_cd(const wcstring &path, wcstring *out_path = NULL, const wchar_t *wd = NULL); bool path_can_be_implicit_cd(const wcstring &path, wcstring *out_path = NULL, const wchar_t *wd = NULL);