blob: c705d6459334097f816aacf78822599ec30751bf [file] [log] [blame]
landley13bab2f2006-09-27 00:45:05 -04001/* Toybox infrastructure.
2 *
3 * Copyright 2006 Rob Landley <rob@landley.net>
landley13bab2f2006-09-27 00:45:05 -04004 */
5
landleyc5621502006-09-28 17:18:51 -04006#include "toys.h"
7
Rob Landleyf2311a42006-11-04 17:45:18 -05008// Populate toy_list[].
landleyc5621502006-09-28 17:18:51 -04009
Rob Landleya6d696b2007-01-31 13:31:19 -050010#undef NEWTOY
11#undef OLDTOY
12#define NEWTOY(name, opts, flags) {#name, name##_main, opts, flags},
13#define OLDTOY(name, oldname, opts, flags) {#name, oldname##_main, opts, flags},
14
landleyc5621502006-09-28 17:18:51 -040015struct toy_list toy_list[] = {
Rob Landley55928b12008-01-19 17:43:27 -060016#include "generated/newtoys.h"
landleyc5621502006-09-28 17:18:51 -040017};
18
19// global context for this applet.
20
21struct toy_context toys;
Rob Landleyb1aaba12008-01-20 17:25:44 -060022union global_union this;
Rob Landley055cfcb2007-01-14 20:20:06 -050023char toybuf[4096];
landleyc5621502006-09-28 17:18:51 -040024
landley4f344e32006-10-05 16:18:03 -040025struct toy_list *toy_find(char *name)
landley13bab2f2006-09-27 00:45:05 -040026{
Rob Landley7aa651a2012-11-13 17:14:08 -060027 int top, bottom, middle;
landley13bab2f2006-09-27 00:45:05 -040028
Rob Landley7aa651a2012-11-13 17:14:08 -060029 // If the name starts with "toybox" accept that as a match. Otherwise
30 // skip the first entry, which is out of order.
landley13bab2f2006-09-27 00:45:05 -040031
Rob Landley7aa651a2012-11-13 17:14:08 -060032 if (!strncmp(name,"toybox",6)) return toy_list;
33 bottom = 1;
landley13bab2f2006-09-27 00:45:05 -040034
Rob Landley7aa651a2012-11-13 17:14:08 -060035 // Binary search to find this applet.
landley13bab2f2006-09-27 00:45:05 -040036
Rob Landley7aa651a2012-11-13 17:14:08 -060037 top = ARRAY_LEN(toy_list)-1;
38 for (;;) {
39 int result;
Rob Landley2c226852007-11-15 18:30:30 -060040
Rob Landley7aa651a2012-11-13 17:14:08 -060041 middle = (top+bottom)/2;
42 if (middle<bottom || middle>top) return NULL;
43 result = strcmp(name,toy_list[middle].name);
44 if (!result) return toy_list+middle;
45 if (result<0) top=--middle;
46 else bottom = ++middle;
47 }
landley13bab2f2006-09-27 00:45:05 -040048}
49
Rob Landleyfc2224b2007-06-01 14:31:45 -040050// Figure out whether or not anything is using the option parsing logic,
51// because the compiler can't figure out whether or not to optimize it away
Rob Landley8f8c5042012-01-14 23:28:15 -060052// on its' own. NEED_OPTIONS becomes a constant allowing if() to optimize
53// stuff out via dead code elimination.
Rob Landleyfc2224b2007-06-01 14:31:45 -040054
55#undef NEWTOY
56#undef OLDTOY
57#define NEWTOY(name, opts, flags) opts ||
58#define OLDTOY(name, oldname, opts, flags) opts ||
Rob Landley4307a7b2007-06-07 15:20:26 -040059static const int NEED_OPTIONS =
Rob Landley55928b12008-01-19 17:43:27 -060060#include "generated/newtoys.h"
Rob Landleyfc2224b2007-06-01 14:31:45 -0400610; // Ends the opts || opts || opts...
62
Rob Landley8f8c5042012-01-14 23:28:15 -060063// Setup toybox global state for this command.
64
landleycd9dfc32006-10-18 18:38:16 -040065void toy_init(struct toy_list *which, char *argv[])
66{
Rob Landley7aa651a2012-11-13 17:14:08 -060067 // Drop permissions for non-suid commands.
Rob Landleye0377fb2010-01-05 12:17:05 -060068
Rob Landley7aa651a2012-11-13 17:14:08 -060069 if (CFG_TOYBOX_SUID) {
70 uid_t uid = getuid(), euid = geteuid();
Rob Landleye0377fb2010-01-05 12:17:05 -060071
Rob Landley7aa651a2012-11-13 17:14:08 -060072 if (!(which->flags & TOYFLAG_STAYROOT)) {
73 if (uid != euid) xsetuid(euid=uid);
74 } else if (CFG_TOYBOX_DEBUG && uid) error_exit("Not installed suid root");
Rob Landleye0377fb2010-01-05 12:17:05 -060075
Rob Landley7aa651a2012-11-13 17:14:08 -060076 if ((which->flags & TOYFLAG_NEEDROOT) && euid) error_exit("Not root");
77 }
Rob Landleye0377fb2010-01-05 12:17:05 -060078
Rob Landley7aa651a2012-11-13 17:14:08 -060079 // Free old toys contents (to be reentrant)
Rob Landleydb037ef2010-01-06 05:29:17 -060080
Rob Landley7aa651a2012-11-13 17:14:08 -060081 if (toys.optargs != toys.argv+1) free(toys.optargs);
82 memset(&toys, 0, sizeof(struct toy_context));
landleycd9dfc32006-10-18 18:38:16 -040083
Rob Landley7aa651a2012-11-13 17:14:08 -060084 toys.which = which;
85 toys.argv = argv;
86 if (NEED_OPTIONS && which->options) get_optflags();
87 else toys.optargs = argv+1;
88 toys.old_umask = umask(0);
89 if (!(which->flags & TOYFLAG_UMASK)) umask(toys.old_umask);
landleycd9dfc32006-10-18 18:38:16 -040090}
91
Rob Landley8f8c5042012-01-14 23:28:15 -060092// Like exec() but runs an internal toybox command instead of another file.
93// Only returns if it can't find the command, otherwise exit() when done.
landley4f344e32006-10-05 16:18:03 -040094void toy_exec(char *argv[])
95{
Rob Landley7aa651a2012-11-13 17:14:08 -060096 struct toy_list *which;
Rob Landleyb841cd22007-06-01 13:41:24 -040097
Rob Landley7aa651a2012-11-13 17:14:08 -060098 which = toy_find(argv[0]);
99 if (!which) return;
100 toy_init(which, argv);
101 toys.which->toy_main();
102 exit(toys.exitval);
landley4f344e32006-10-05 16:18:03 -0400103}
104
Rob Landley8f8c5042012-01-14 23:28:15 -0600105// Multiplexer command, first argument is command to run, rest are args to that.
106// If first argument starts with - output list of command install paths.
107
Rob Landleyefda21c2007-11-29 18:14:37 -0600108void toybox_main(void)
landley4f344e32006-10-05 16:18:03 -0400109{
Rob Landley7aa651a2012-11-13 17:14:08 -0600110 static char *toy_paths[]={"usr/","bin/","sbin/",0};
111 int i, len = 0;
landley4f344e32006-10-05 16:18:03 -0400112
Rob Landley7aa651a2012-11-13 17:14:08 -0600113 if (toys.argv[1]) {
114 if (toys.argv[1][0]!='-') {
115 toy_exec(toys.argv+1);
116 toys.which = toy_list;
117 error_exit("Unknown command %s",toys.argv[1]);
118 }
119 }
landley4f344e32006-10-05 16:18:03 -0400120
Rob Landley7aa651a2012-11-13 17:14:08 -0600121 // Output list of applets.
122 for (i=1; i<ARRAY_LEN(toy_list); i++) {
123 int fl = toy_list[i].flags;
124 if (fl & TOYMASK_LOCATION) {
125 if (toys.argv[1]) {
126 int j;
127 for (j=0; toy_paths[j]; j++)
128 if (fl & (1<<j)) len += printf("%s", toy_paths[j]);
129 }
130 len += printf("%s ",toy_list[i].name);
131 if (len>65) {
132 xputc('\n');
133 len=0;
134 }
135 }
136 }
137 xputc('\n');
landley4f344e32006-10-05 16:18:03 -0400138}
139
landley13bab2f2006-09-27 00:45:05 -0400140int main(int argc, char *argv[])
141{
Rob Landley7aa651a2012-11-13 17:14:08 -0600142 // Artificial scope to eat less stack for things we call
143 {
144 char *name;
landley13bab2f2006-09-27 00:45:05 -0400145
Rob Landley7aa651a2012-11-13 17:14:08 -0600146 // Trim path off of command name
147 name = strrchr(argv[0], '/');
148 if (!name) name=argv[0];
149 else name++;
150 argv[0] = name;
151 }
landley13bab2f2006-09-27 00:45:05 -0400152
Rob Landley7aa651a2012-11-13 17:14:08 -0600153 // Call the multiplexer, adjusting this argv[] to be its' argv[1].
154 // (It will adjust it back before calling toy_exec().)
155 toys.argv = argv-1;
156 toybox_main();
157 return 0;
landley13bab2f2006-09-27 00:45:05 -0400158}