5471db597b193b6f2052f659be3b91f6b4bb1a8c
[oweals/busybox.git] / applets / busybox.c
1 /* vi: set sw=4 ts=4: */
2 #include "internal.h"
3 #include <stdio.h>
4 #include <string.h>
5 #include <errno.h>
6
7 static int been_there_done_that = 0;
8
9 /* It has been alledged that doing such things can
10  * help reduce binary size when staticly linking,
11  * of course with glibc, this is unlikely as long
12  * as we use things like printf -- perhaps a printf
13  * replacement may be in order 
14  */
15 #if 0
16 void exit(int status) __attribute__ ((noreturn));
17 void exit(int status)
18 {
19         _exit(status);
20 };
21 void abort(void) __attribute__ ((__noreturn__));
22 void abort(void)
23 {
24         _exit(0);
25 };
26 int atexit(void (*__func) (void))
27 {
28         _exit(0);
29 };
30 void *__libc_stack_end;
31 #endif
32
33 const struct BB_applet applets[] = {
34
35 #ifdef BB_AR
36         {"ar", ar_main, _BB_DIR_USR_BIN},
37 #endif
38 #ifdef BB_BASENAME
39         {"basename", basename_main, _BB_DIR_USR_BIN},
40 #endif
41         {"busybox", busybox_main, _BB_DIR_BIN},
42 #ifdef BB_CAT
43         {"cat", cat_main, _BB_DIR_BIN},
44 #endif
45 #ifdef BB_CHMOD_CHOWN_CHGRP
46         {"chgrp", chmod_chown_chgrp_main, _BB_DIR_BIN},
47 #endif
48 #ifdef BB_CHMOD_CHOWN_CHGRP
49         {"chmod", chmod_chown_chgrp_main, _BB_DIR_BIN},
50 #endif
51 #ifdef BB_CHMOD_CHOWN_CHGRP
52         {"chown", chmod_chown_chgrp_main, _BB_DIR_BIN},
53 #endif
54 #ifdef BB_CHROOT
55         {"chroot", chroot_main, _BB_DIR_SBIN},
56 #endif
57 #ifdef BB_CLEAR
58         {"clear", clear_main, _BB_DIR_USR_BIN},
59 #endif
60 #ifdef BB_CHVT
61         {"chvt", chvt_main, _BB_DIR_USR_BIN},
62 #endif
63 #ifdef BB_CP_MV
64         {"cp", cp_mv_main, _BB_DIR_BIN},
65 #endif
66 #ifdef BB_CUT
67         {"cut", cut_main, _BB_DIR_USR_BIN},
68 #endif
69 #ifdef BB_DATE
70         {"date", date_main, _BB_DIR_BIN},
71 #endif
72 #ifdef BB_DC
73         {"dc", dc_main, _BB_DIR_USR_BIN},
74 #endif
75 #ifdef BB_DD
76         {"dd", dd_main, _BB_DIR_BIN},
77 #endif
78 #ifdef BB_DF
79         {"df", df_main, _BB_DIR_BIN},
80 #endif
81 #ifdef BB_DIRNAME
82         {"dirname", dirname_main, _BB_DIR_USR_BIN},
83 #endif
84 #ifdef BB_DMESG
85         {"dmesg", dmesg_main, _BB_DIR_BIN},
86 #endif
87 #ifdef BB_DU
88         {"du", du_main, _BB_DIR_BIN},
89 #endif
90 #ifdef BB_DUMPKMAP
91         {"dumpkmap", dumpkmap_main, _BB_DIR_BIN},
92 #endif
93 #ifdef BB_DUTMP
94         {"dutmp", dutmp_main, _BB_DIR_USR_SBIN},
95 #endif
96 #ifdef BB_ECHO
97         {"echo", echo_main, _BB_DIR_BIN},
98 #endif
99 #ifdef BB_TRUE_FALSE
100         {"false", false_main, _BB_DIR_BIN},
101 #endif
102 #ifdef BB_FBSET
103         {"fbset", fbset_main, _BB_DIR_USR_SBIN},
104 #endif
105 #ifdef BB_FDFLUSH
106         {"fdflush", fdflush_main, _BB_DIR_BIN},
107 #endif
108 #ifdef BB_FIND
109         {"find", find_main, _BB_DIR_USR_BIN},
110 #endif
111 #ifdef BB_FREE
112         {"free", free_main, _BB_DIR_USR_BIN},
113 #endif
114 #ifdef BB_FREERAMDISK
115         {"freeramdisk", freeramdisk_main, _BB_DIR_SBIN},
116 #endif
117 #ifdef BB_DEALLOCVT
118         {"deallocvt", deallocvt_main, _BB_DIR_USR_BIN},
119 #endif
120 #ifdef BB_FSCK_MINIX
121         {"fsck.minix", fsck_minix_main, _BB_DIR_SBIN},
122 #endif
123 #ifdef BB_GREP
124         {"grep", grep_main, _BB_DIR_BIN},
125 #endif
126 #ifdef BB_GUNZIP
127         {"gunzip", gunzip_main, _BB_DIR_BIN},
128 #endif
129 #ifdef BB_GZIP
130         {"gzip", gzip_main, _BB_DIR_BIN},
131 #endif
132 #ifdef BB_HALT
133         {"halt", halt_main, _BB_DIR_SBIN},
134 #endif
135 #ifdef BB_HEAD
136         {"head", head_main, _BB_DIR_BIN},
137 #endif
138 #ifdef BB_HOSTID
139         {"hostid", hostid_main, _BB_DIR_USR_BIN},
140 #endif
141 #ifdef BB_HOSTNAME
142         {"hostname", hostname_main, _BB_DIR_BIN},
143 #endif
144 #ifdef BB_ID
145         {"id", id_main, _BB_DIR_USR_BIN},
146 #endif
147 #ifdef BB_INIT
148         {"init", init_main, _BB_DIR_SBIN},
149 #endif
150 #ifdef BB_INSMOD
151         {"insmod", insmod_main, _BB_DIR_SBIN},
152 #endif
153 #ifdef BB_KILL
154         {"kill", kill_main, _BB_DIR_BIN},
155 #endif
156 #ifdef BB_KILLALL
157         {"killall", kill_main, _BB_DIR_USR_BIN},
158 #endif
159 #ifdef BB_LENGTH
160         {"length", length_main, _BB_DIR_USR_BIN},
161 #endif
162 #ifdef BB_LINUXRC
163         {"linuxrc", init_main, _BB_DIR_ROOT},
164 #endif
165 #ifdef BB_LN
166         {"ln", ln_main, _BB_DIR_BIN},
167 #endif
168 #ifdef BB_LOADACM
169         {"loadacm", loadacm_main, _BB_DIR_USR_BIN},
170 #endif
171 #ifdef BB_LOADFONT
172         {"loadfont", loadfont_main, _BB_DIR_USR_BIN},
173 #endif
174 #ifdef BB_LOADKMAP
175         {"loadkmap", loadkmap_main, _BB_DIR_SBIN},
176 #endif
177 #ifdef BB_LOGGER
178         {"logger", logger_main, _BB_DIR_USR_BIN},
179 #endif
180 #ifdef BB_LOGNAME
181         {"logname", logname_main, _BB_DIR_USR_BIN},
182 #endif
183 #ifdef BB_LS
184         {"ls", ls_main, _BB_DIR_BIN},
185 #endif
186 #ifdef BB_LSMOD
187         {"lsmod", lsmod_main, _BB_DIR_SBIN},
188 #endif
189 #ifdef BB_MAKEDEVS
190         {"makedevs", makedevs_main, _BB_DIR_SBIN},
191 #endif
192 #ifdef BB_MD5SUM
193         {"md5sum", md5sum_main, _BB_DIR_USR_BIN},
194 #endif
195 #ifdef BB_MKDIR
196         {"mkdir", mkdir_main, _BB_DIR_BIN},
197 #endif
198 #ifdef BB_MKFIFO
199         {"mkfifo", mkfifo_main, _BB_DIR_USR_BIN},
200 #endif
201 #ifdef BB_MKFS_MINIX
202         {"mkfs.minix", mkfs_minix_main, _BB_DIR_SBIN},
203 #endif
204 #ifdef BB_MKNOD
205         {"mknod", mknod_main, _BB_DIR_BIN},
206 #endif
207 #ifdef BB_MKSWAP
208         {"mkswap", mkswap_main, _BB_DIR_SBIN},
209 #endif
210 #ifdef BB_MKTEMP
211         {"mktemp", mktemp_main, _BB_DIR_BIN},
212 #endif
213 #ifdef BB_NC
214         {"nc", nc_main, _BB_DIR_USR_BIN},
215 #endif
216 #ifdef BB_MORE
217         {"more", more_main, _BB_DIR_BIN},
218 #endif
219 #ifdef BB_MOUNT
220         {"mount", mount_main, _BB_DIR_BIN},
221 #endif
222 #ifdef BB_MT
223         {"mt", mt_main, _BB_DIR_BIN},
224 #endif
225 #ifdef BB_CP_MV
226         {"mv", cp_mv_main, _BB_DIR_BIN},
227 #endif
228 #ifdef BB_NSLOOKUP
229         {"nslookup", nslookup_main, _BB_DIR_USR_BIN},
230 #endif
231 #ifdef BB_PING
232         {"ping", ping_main, _BB_DIR_BIN},
233 #endif
234 #ifdef BB_POWEROFF
235         {"poweroff", poweroff_main, _BB_DIR_SBIN},
236 #endif
237 #ifdef BB_PRINTF
238         {"printf", printf_main, _BB_DIR_USR_BIN},
239 #endif
240 #ifdef BB_PS
241         {"ps", ps_main, _BB_DIR_BIN},
242 #endif
243 #ifdef BB_PWD
244         {"pwd", pwd_main, _BB_DIR_BIN},
245 #endif
246 #ifdef BB_REBOOT
247         {"reboot", reboot_main, _BB_DIR_SBIN},
248 #endif
249 #ifdef BB_RM
250         {"rm", rm_main, _BB_DIR_BIN},
251 #endif
252 #ifdef BB_RMDIR
253         {"rmdir", rmdir_main, _BB_DIR_BIN},
254 #endif
255 #ifdef BB_RMMOD
256         {"rmmod", rmmod_main, _BB_DIR_SBIN},
257 #endif
258 #ifdef BB_SED
259         {"sed", sed_main, _BB_DIR_BIN},
260 #endif
261 #ifdef BB_SETKEYCODES
262         {"setkeycodes", setkeycodes_main, _BB_DIR_USR_BIN},
263 #endif
264 #ifdef BB_SH
265         {"sh", shell_main, _BB_DIR_BIN},
266 #endif
267 #ifdef BB_SLEEP
268         {"sleep", sleep_main, _BB_DIR_BIN},
269 #endif
270 #ifdef BB_SORT
271         {"sort", sort_main, _BB_DIR_BIN},
272 #endif
273 #ifdef BB_SYNC
274         {"sync", sync_main, _BB_DIR_BIN},
275 #endif
276 #ifdef BB_SYSLOGD
277         {"syslogd", syslogd_main, _BB_DIR_SBIN},
278 #endif
279 #ifdef BB_SWAPONOFF
280         {"swapon", swap_on_off_main, _BB_DIR_SBIN},
281 #endif
282 #ifdef BB_SWAPONOFF
283         {"swapoff", swap_on_off_main, _BB_DIR_SBIN},
284 #endif
285 #ifdef BB_TAIL
286         {"tail", tail_main, _BB_DIR_USR_BIN},
287 #endif
288 #ifdef BB_TAR
289         {"tar", tar_main, _BB_DIR_BIN},
290 #endif
291 #ifdef BB_TELNET
292         {"telnet", telnet_main, _BB_DIR_USR_BIN},
293 #endif
294 #ifdef BB_TEST
295         {"test", test_main, _BB_DIR_USR_BIN},
296 #endif
297 #ifdef BB_TEE
298         {"tee", tee_main, _BB_DIR_BIN},
299 #endif
300 #ifdef BB_TOUCH
301         {"touch", touch_main, _BB_DIR_USR_BIN},
302 #endif
303 #ifdef BB_TR
304         {"tr", tr_main, _BB_DIR_USR_BIN},
305 #endif
306 #ifdef BB_TRUE_FALSE
307         {"true", true_main, _BB_DIR_BIN},
308 #endif
309 #ifdef BB_TTY
310         {"tty", tty_main, _BB_DIR_USR_BIN},
311 #endif
312 #ifdef BB_UMOUNT
313         {"umount", umount_main, _BB_DIR_BIN},
314 #endif
315 #ifdef BB_UNAME
316         {"uname", uname_main, _BB_DIR_BIN},
317 #endif
318 #ifdef BB_UNIQ
319         {"uniq", uniq_main, _BB_DIR_USR_BIN},
320 #endif
321 #ifdef BB_UPDATE
322         {"update", update_main, _BB_DIR_SBIN},
323 #endif
324 #ifdef BB_UPTIME
325         {"uptime", uptime_main, _BB_DIR_USR_BIN},
326 #endif
327 #ifdef BB_UUENCODE
328         {"uuencode", uuencode_main, _BB_DIR_USR_BIN},
329 #endif
330 #ifdef BB_UUDECODE
331         {"uudecode", uudecode_main, _BB_DIR_USR_BIN},
332 #endif
333 #ifdef BB_USLEEP
334         {"usleep", usleep_main, _BB_DIR_BIN},
335 #endif
336 #ifdef BB_WC
337         {"wc", wc_main, _BB_DIR_USR_BIN},
338 #endif
339 #ifdef BB_WHICH
340         {"which", which_main, _BB_DIR_USR_BIN},
341 #endif
342 #ifdef BB_WHOAMI
343         {"whoami", whoami_main, _BB_DIR_USR_BIN},
344 #endif
345 #ifdef BB_YES
346         {"yes", yes_main, _BB_DIR_USR_BIN},
347 #endif
348 #ifdef BB_GUNZIP
349         {"zcat", gunzip_main, _BB_DIR_BIN},
350 #endif
351 #ifdef BB_TEST
352         {"[", test_main, _BB_DIR_USR_BIN},
353 #endif
354         {0}
355 };
356
357
358 #ifdef BB_FEATURE_INSTALLER
359 /* 
360  * directory table
361  *              this should be consistent w/ the enum, internal.h::Location,
362  *              or else...
363  */
364 static char* install_dir[] = {
365         "/",
366         "/bin",
367         "/sbin",
368         "/usr/bin",
369         "/usr/sbin",
370 };
371
372 /* abstract link() */
373 typedef int (*__link_f)(const char *, const char *);
374
375 /* 
376  * Where in the filesystem is this busybox?
377  * [return]
378  *              malloc'd string w/ full pathname of busybox's location
379  *              NULL on failure
380  */
381 static char *busybox_fullpath()
382 {
383         pid_t pid;
384         char path[256];
385         char proc[256];
386         int len;
387
388         pid = getpid();
389         sprintf(proc, "/proc/%d/exe", pid);
390         len = readlink(proc, path, 256);
391         if (len != -1) {
392                 path[len] = 0;
393         } else {
394                 fprintf(stderr, "busybox : %s : %s\n", proc, strerror(errno));
395                 return NULL;
396         }
397         return strdup(path);
398 }
399
400 /* create (sym)links for each applet */
401 static int install_links(const char *busybox, int use_symbolic_links)
402 {
403         __link_f Link = link;
404
405         char command[256];
406         int i;
407         int rc = 0;
408
409         if (use_symbolic_links) Link = symlink;
410
411         for (i = 0; applets[i].name != NULL; i++) {
412                 sprintf (
413                         command, 
414                         "%s/%s", 
415                         install_dir[applets[i].location], 
416                         applets[i].name
417                 );
418 #if 1
419                 rc |= Link(busybox, command);
420 #else
421                 puts(command);
422 #endif
423                 if (rc) {
424                         fprintf(stderr,"busybox : %s : %s\n", command, strerror(errno));
425                 }
426         }
427         return rc;
428 }
429
430 #if 0
431 int uninstall_links() ?
432 #endif
433 #endif /* BB_FEATURE_INSTALLER */
434
435
436 int main(int argc, char **argv)
437 {
438         char                            *s;
439         char                            *name;
440         const struct BB_applet  *a              = applets;
441
442 #ifdef BB_FEATURE_INSTALLER     
443         /* 
444          * This style of argument parsing doesn't scale well 
445          * in the event that busybox starts wanting more --options.
446          * If someone has a cleaner approach, by all means implement it.
447          */
448         if (argc > 1 && (strcmp(argv[1], "--install") == 0)) {
449                 int use_symbolic_links = 0;
450                 int rc = 0;
451                 char *busybox;
452
453                 /* to use symlinks, or not to use symlinks... */
454                 if (argc > 2) {
455                         if ((strcmp(argv[2], "-s") == 0)) { 
456                                 use_symbolic_links = 1; 
457                         }
458                 }
459
460                 /* link */
461                 busybox = busybox_fullpath();
462                 if (busybox) {
463                         install_links(busybox, use_symbolic_links);
464                         free(busybox);
465                 } else {
466                         rc = 1;
467                 }
468                 return rc;
469         }
470 #endif /* BB_FEATURE_INSTALLER */
471
472         for (s = name = argv[0]; *s != '\0';) {
473                 if (*s++ == '/')
474                         name = s;
475         }
476
477         *argv = name;
478
479 #ifdef BB_SH
480         /* Add in a special case hack -- whenever **argv == '-'
481          * (i.e. '-su' or '-sh') always invoke the shell */
482         if (**argv == '-')
483                 exit(((*(shell_main)) (argc, argv)));
484 #endif
485
486         while (a->name != 0) {
487                 if (strcmp(name, a->name) == 0) {
488                         exit(((*(a->main)) (argc, argv)));
489                 }
490                 a++;
491         }
492         return(busybox_main(argc, argv));
493 }
494
495
496 int busybox_main(int argc, char **argv)
497 {
498         int col = 0;
499
500         argc--;
501         argv++;
502
503         if (been_there_done_that == 1 || argc < 1) {
504                 const struct BB_applet *a = applets;
505
506                 fprintf(stderr, "BusyBox v%s (%s) multi-call binary -- GPL2\n\n"
507                                 "Usage: busybox [function] [arguments]...\n"
508                                 "   or: [function] [arguments]...\n\n"
509                                 "\tBusyBox is a multi-call binary that combines many common Unix\n"
510                                 "\tutilities into a single executable.  Most people will create a\n"
511                                 "\tlink to busybox for each function they wish to use, and BusyBox\n"
512                                 "\twill act like whatever it was invoked as.\n" 
513                                 "\nCurrently defined functions:\n", BB_VER, BB_BT);
514
515                 while (a->name != 0) {
516                         col +=
517                                 fprintf(stderr, "%s%s", ((col == 0) ? "\t" : ", "),
518                                                 (a++)->name);
519                         if (col > 60 && a->name != 0) {
520                                 fprintf(stderr, ",\n");
521                                 col = 0;
522                         }
523                 }
524                 fprintf(stderr, "\n\n");
525                 exit(-1);
526         }
527         /* If we've already been here once, exit now */
528         been_there_done_that = 1;
529         return (main(argc, argv));
530 }
531
532 /*
533 Local Variables:
534 c-file-style: "linux"
535 c-basic-offset: 4
536 tab-width: 4
537 End:
538 */