which: -a support (needed for bfin uclibc build script)

real support (with CONFIG_DESKTOP=y): 120+ bytes:
   text	   data	    bss	    dec	    hex	filename
 807958	    624	   7036	 815618	  c7202	busybox_old
 808085	    624	   7036	 815745	  c7281	busybox_unstripped

"fake" support (with CONFIG_DESKTOP unset): ~45 bytes:
   text	   data	    bss	    dec	    hex	filename
 797790	    611	   6996	 805397	  c4a15	busybox_old
 797834	    611	   6996	 805441	  c4a41	busybox_unstripped
This commit is contained in:
Denis Vlasenko 2008-06-05 13:33:59 +00:00
parent f0d6068086
commit f592aa36f3
3 changed files with 70 additions and 23 deletions

View File

@ -13,30 +13,69 @@
#include "libbb.h" #include "libbb.h"
int which_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; int which_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
int which_main(int argc, char **argv) int which_main(int argc ATTRIBUTE_UNUSED, char **argv)
{ {
USE_DESKTOP(int opt;)
int status = EXIT_SUCCESS; int status = EXIT_SUCCESS;
char *path;
char *p; char *p;
if (argc <= 1 || argv[1][0] == '-') { opt_complementary = "-1"; /* at least one argument */
bb_show_usage(); USE_DESKTOP(opt =) getopt32(argv, "a");
argv += optind;
/* This matches what is seen on e.g. ubuntu.
* "which" there is a shell script. */
path = getenv("PATH");
if (!path) {
path = (char*)bb_PATH_root_path;
putenv(path);
path += 5; /* skip "PATH=" */
} }
/* This matches what is seen on e.g. ubuntu do {
* "which" there is a shell script */ #if ENABLE_DESKTOP
if (!getenv("PATH")) { /* Much bloat just to support -a */
putenv((char*)bb_PATH_root_path); if (strchr(*argv, '/')) {
} if (execable_file(*argv)) {
puts(*argv);
continue;
}
status = EXIT_FAILURE;
} else {
char *path2 = xstrdup(path);
char *tmp = path2;
while (--argc > 0) { p = find_execable(*argv, &tmp);
argv++; if (!p)
status = EXIT_FAILURE;
else {
print:
puts(p);
free(p);
if (opt) {
/* -a: show matches in all PATH components */
if (tmp) {
p = find_execable(*argv, &tmp);
if (p)
goto print;
}
}
}
free(path2);
}
#else
/* Just ignoring -a */
if (strchr(*argv, '/')) { if (strchr(*argv, '/')) {
if (execable_file(*argv)) { if (execable_file(*argv)) {
puts(*argv); puts(*argv);
continue; continue;
} }
} else { } else {
p = find_execable(*argv); char *path2 = xstrdup(path);
char *tmp = path2;
p = find_execable(*argv, &tmp);
free(path2);
if (p) { if (p) {
puts(p); puts(p);
free(p); free(p);
@ -44,7 +83,8 @@ int which_main(int argc, char **argv)
} }
} }
status = EXIT_FAILURE; status = EXIT_FAILURE;
} #endif
} while (*(++argv) != NULL);
fflush_stdout_and_exit(status); fflush_stdout_and_exit(status);
} }

View File

@ -700,7 +700,7 @@ void die_if_bad_username(const char* name);
#endif #endif
int execable_file(const char *name); int execable_file(const char *name);
char *find_execable(const char *filename); char *find_execable(const char *filename, char **PATHp);
int exists_execable(const char *filename); int exists_execable(const char *filename);
/* BB_EXECxx always execs (it's not doing NOFORK/NOEXEC stuff), /* BB_EXECxx always execs (it's not doing NOFORK/NOEXEC stuff),

View File

@ -19,15 +19,20 @@ int execable_file(const char *name)
return (!access(name, X_OK) && !stat(name, &s) && S_ISREG(s.st_mode)); return (!access(name, X_OK) && !stat(name, &s) && S_ISREG(s.st_mode));
} }
/* search $PATH for an executable file; /* search (*PATHp) for an executable file;
* return allocated string containing full path if found; * return allocated string containing full path if found;
* return NULL otherwise; * PATHp points to the component after the one where it was found
* (or NULL),
* you may call find_execable again with this PATHp to continue
* (if it's not NULL).
* return NULL otherwise; (PATHp is undefined)
* in all cases (*PATHp) contents will be trashed (s/:/NUL/).
*/ */
char *find_execable(const char *filename) char *find_execable(const char *filename, char **PATHp)
{ {
char *path, *p, *n; char *p, *n;
p = path = xstrdup(getenv("PATH")); p = *PATHp;
while (p) { while (p) {
n = strchr(p, ':'); n = strchr(p, ':');
if (n) if (n)
@ -35,15 +40,14 @@ char *find_execable(const char *filename)
if (*p != '\0') { /* it's not a PATH="foo::bar" situation */ if (*p != '\0') { /* it's not a PATH="foo::bar" situation */
p = concat_path_file(p, filename); p = concat_path_file(p, filename);
if (execable_file(p)) { if (execable_file(p)) {
free(path); *PATHp = n;
return p; return p;
} }
free(p); free(p);
} }
p = n; p = n;
} } /* on loop exit p == NULL */
free(path); return p;
return NULL;
} }
/* search $PATH for an executable file; /* search $PATH for an executable file;
@ -52,7 +56,10 @@ char *find_execable(const char *filename)
*/ */
int exists_execable(const char *filename) int exists_execable(const char *filename)
{ {
char *ret = find_execable(filename); char *path = xstrdup(getenv("PATH"));
char *tmp = path;
char *ret = find_execable(filename, &tmp);
free(path);
if (ret) { if (ret) {
free(ret); free(ret);
return 1; return 1;