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