Mercurial > hg > toybox
annotate lib/xwrap.c @ 1528:ec07449e8e4a draft
Add TOYBOX_NORECURSE so xexec() won't make internal function calls.
| author | Rob Landley <rob@landley.net> |
|---|---|
| date | Mon, 20 Oct 2014 19:52:29 -0500 |
| parents | bff076394df5 |
| children | e127aa575ff2 |
| rev | line source |
|---|---|
| 951 | 1 /* xwrap.c - wrappers around existing library functions. |
| 2 * | |
| 3 * Functions with the x prefix are wrappers that either succeed or kill the | |
| 4 * program with an error message, but never return failure. They usually have | |
| 5 * the same arguments and return value as the function they wrap. | |
| 6 * | |
| 7 * Copyright 2006 Rob Landley <rob@landley.net> | |
| 8 */ | |
| 9 | |
| 10 #include "toys.h" | |
| 11 | |
| 12 // Strcpy with size checking: exit if there's not enough space for the string. | |
| 13 void xstrncpy(char *dest, char *src, size_t size) | |
| 14 { | |
|
1105
1bca28705a87
Give xstrncpy() a more informative error message.
Rob Landley <rob@landley.net>
parents:
1053
diff
changeset
|
15 if (strlen(src)+1 > size) error_exit("'%s' > %ld bytes", src, (long)size); |
| 951 | 16 strcpy(dest, src); |
| 17 } | |
| 18 | |
| 19 void xexit(void) | |
| 20 { | |
| 21 if (toys.rebound) longjmp(*toys.rebound, 1); | |
| 22 else exit(toys.exitval); | |
| 23 } | |
| 24 | |
| 25 // Die unless we can allocate memory. | |
| 26 void *xmalloc(size_t size) | |
| 27 { | |
| 28 void *ret = malloc(size); | |
| 29 if (!ret) error_exit("xmalloc"); | |
| 30 | |
| 31 return ret; | |
| 32 } | |
| 33 | |
| 34 // Die unless we can allocate prezeroed memory. | |
| 35 void *xzalloc(size_t size) | |
| 36 { | |
| 37 void *ret = xmalloc(size); | |
| 38 memset(ret, 0, size); | |
| 39 return ret; | |
| 40 } | |
| 41 | |
| 42 // Die unless we can change the size of an existing allocation, possibly | |
| 43 // moving it. (Notice different arguments from libc function.) | |
| 44 void *xrealloc(void *ptr, size_t size) | |
| 45 { | |
| 46 ptr = realloc(ptr, size); | |
| 47 if (!ptr) error_exit("xrealloc"); | |
| 48 | |
| 49 return ptr; | |
| 50 } | |
| 51 | |
| 52 // Die unless we can allocate a copy of this many bytes of string. | |
| 53 char *xstrndup(char *s, size_t n) | |
| 54 { | |
| 55 char *ret = xmalloc(++n); | |
| 56 strncpy(ret, s, n); | |
| 57 ret[--n]=0; | |
| 58 | |
| 59 return ret; | |
| 60 } | |
| 61 | |
| 62 // Die unless we can allocate a copy of this string. | |
| 63 char *xstrdup(char *s) | |
| 64 { | |
| 65 return xstrndup(s, strlen(s)); | |
| 66 } | |
| 67 | |
| 68 // Die unless we can allocate enough space to sprintf() into. | |
|
1183
0752b2d58909
Rename xmsprintf() to just xmprintf().
Rob Landley <rob@landley.net>
parents:
1170
diff
changeset
|
69 char *xmprintf(char *format, ...) |
| 951 | 70 { |
| 71 va_list va, va2; | |
| 72 int len; | |
| 73 char *ret; | |
| 74 | |
| 75 va_start(va, format); | |
| 76 va_copy(va2, va); | |
| 77 | |
| 78 // How long is it? | |
| 79 len = vsnprintf(0, 0, format, va); | |
| 80 len++; | |
| 81 va_end(va); | |
| 82 | |
| 83 // Allocate and do the sprintf() | |
| 84 ret = xmalloc(len); | |
| 85 vsnprintf(ret, len, format, va2); | |
| 86 va_end(va2); | |
| 87 | |
| 88 return ret; | |
| 89 } | |
| 90 | |
| 91 void xprintf(char *format, ...) | |
| 92 { | |
| 93 va_list va; | |
| 94 va_start(va, format); | |
| 95 | |
| 96 vprintf(format, va); | |
|
1318
955169e818d0
Isaac Dunham suggested xprintf() should call fflush() instead of ferror(), and posix-2008 doesn't say if fflush() covers ferror() (or can return success when the stream's error state is set), so call both.
Rob Landley <rob@landley.net>
parents:
1235
diff
changeset
|
97 if (fflush(stdout) || ferror(stdout)) perror_exit("write"); |
| 951 | 98 } |
| 99 | |
| 100 void xputs(char *s) | |
| 101 { | |
|
1318
955169e818d0
Isaac Dunham suggested xprintf() should call fflush() instead of ferror(), and posix-2008 doesn't say if fflush() covers ferror() (or can return success when the stream's error state is set), so call both.
Rob Landley <rob@landley.net>
parents:
1235
diff
changeset
|
102 if (EOF == puts(s) || fflush(stdout) || ferror(stdout)) perror_exit("write"); |
| 951 | 103 } |
| 104 | |
| 105 void xputc(char c) | |
| 106 { | |
|
1318
955169e818d0
Isaac Dunham suggested xprintf() should call fflush() instead of ferror(), and posix-2008 doesn't say if fflush() covers ferror() (or can return success when the stream's error state is set), so call both.
Rob Landley <rob@landley.net>
parents:
1235
diff
changeset
|
107 if (EOF == fputc(c, stdout) || fflush(stdout) || ferror(stdout)) |
|
955169e818d0
Isaac Dunham suggested xprintf() should call fflush() instead of ferror(), and posix-2008 doesn't say if fflush() covers ferror() (or can return success when the stream's error state is set), so call both.
Rob Landley <rob@landley.net>
parents:
1235
diff
changeset
|
108 perror_exit("write"); |
| 951 | 109 } |
| 110 | |
| 111 void xflush(void) | |
| 112 { | |
|
1318
955169e818d0
Isaac Dunham suggested xprintf() should call fflush() instead of ferror(), and posix-2008 doesn't say if fflush() covers ferror() (or can return success when the stream's error state is set), so call both.
Rob Landley <rob@landley.net>
parents:
1235
diff
changeset
|
113 if (fflush(stdout) || ferror(stdout)) perror_exit("write");; |
| 951 | 114 } |
| 115 | |
|
1327
85f297591693
Introduce xfork() and make commands use it, and make some WEXITSTATUS() use WIFEXITED() and WTERMSIG()+127.
Rob Landley <rob@landley.net>
parents:
1318
diff
changeset
|
116 pid_t xfork(void) |
|
85f297591693
Introduce xfork() and make commands use it, and make some WEXITSTATUS() use WIFEXITED() and WTERMSIG()+127.
Rob Landley <rob@landley.net>
parents:
1318
diff
changeset
|
117 { |
|
85f297591693
Introduce xfork() and make commands use it, and make some WEXITSTATUS() use WIFEXITED() and WTERMSIG()+127.
Rob Landley <rob@landley.net>
parents:
1318
diff
changeset
|
118 pid_t pid = fork(); |
|
85f297591693
Introduce xfork() and make commands use it, and make some WEXITSTATUS() use WIFEXITED() and WTERMSIG()+127.
Rob Landley <rob@landley.net>
parents:
1318
diff
changeset
|
119 |
|
85f297591693
Introduce xfork() and make commands use it, and make some WEXITSTATUS() use WIFEXITED() and WTERMSIG()+127.
Rob Landley <rob@landley.net>
parents:
1318
diff
changeset
|
120 if (pid < 0) perror_exit("fork"); |
|
85f297591693
Introduce xfork() and make commands use it, and make some WEXITSTATUS() use WIFEXITED() and WTERMSIG()+127.
Rob Landley <rob@landley.net>
parents:
1318
diff
changeset
|
121 |
|
85f297591693
Introduce xfork() and make commands use it, and make some WEXITSTATUS() use WIFEXITED() and WTERMSIG()+127.
Rob Landley <rob@landley.net>
parents:
1318
diff
changeset
|
122 return pid; |
|
85f297591693
Introduce xfork() and make commands use it, and make some WEXITSTATUS() use WIFEXITED() and WTERMSIG()+127.
Rob Landley <rob@landley.net>
parents:
1318
diff
changeset
|
123 } |
|
85f297591693
Introduce xfork() and make commands use it, and make some WEXITSTATUS() use WIFEXITED() and WTERMSIG()+127.
Rob Landley <rob@landley.net>
parents:
1318
diff
changeset
|
124 |
|
952
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
125 // Call xexec with a chunk of optargs, starting at skip. (You can't just |
|
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
126 // call xexec() directly because toy_init() frees optargs.) |
|
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
127 void xexec_optargs(int skip) |
|
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
128 { |
|
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
129 char **s = toys.optargs; |
|
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
130 |
|
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
131 toys.optargs = 0; |
|
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
132 xexec(s+skip); |
|
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
133 } |
|
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
134 |
|
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
135 |
| 951 | 136 // Die unless we can exec argv[] (or run builtin command). Note that anything |
| 137 // with a path isn't a builtin, so /bin/sh won't match the builtin sh. | |
| 138 void xexec(char **argv) | |
| 139 { | |
|
1528
ec07449e8e4a
Add TOYBOX_NORECURSE so xexec() won't make internal function calls.
Rob Landley <rob@landley.net>
parents:
1473
diff
changeset
|
140 if (CFG_TOYBOX && !CFG_TOYBOX_NORECURSE) toy_exec(argv); |
| 951 | 141 execvp(argv[0], argv); |
| 142 | |
| 143 perror_exit("exec %s", argv[0]); | |
| 144 } | |
| 145 | |
|
1334
9fd2bcedbeb5
mount: start on option parsing, implement loopback and bind mount autodetection.
Rob Landley <rob@landley.net>
parents:
1327
diff
changeset
|
146 // Spawn child process, capturing stdin/stdout. |
|
9fd2bcedbeb5
mount: start on option parsing, implement loopback and bind mount autodetection.
Rob Landley <rob@landley.net>
parents:
1327
diff
changeset
|
147 // argv[]: command to exec. If null, child returns to original program. |
|
1472
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
148 // pipes[2]: stdin, stdout of new process. If -1 will not have pipe allocated. |
|
1334
9fd2bcedbeb5
mount: start on option parsing, implement loopback and bind mount autodetection.
Rob Landley <rob@landley.net>
parents:
1327
diff
changeset
|
149 // return: pid of child process |
|
1472
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
150 pid_t xpopen_both(char **argv, int *pipes) |
|
1334
9fd2bcedbeb5
mount: start on option parsing, implement loopback and bind mount autodetection.
Rob Landley <rob@landley.net>
parents:
1327
diff
changeset
|
151 { |
|
9fd2bcedbeb5
mount: start on option parsing, implement loopback and bind mount autodetection.
Rob Landley <rob@landley.net>
parents:
1327
diff
changeset
|
152 int cestnepasun[4], pid; |
|
9fd2bcedbeb5
mount: start on option parsing, implement loopback and bind mount autodetection.
Rob Landley <rob@landley.net>
parents:
1327
diff
changeset
|
153 |
|
1472
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
154 // Make the pipes? Not this won't set either pipe to 0 because if fds are |
|
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
155 // allocated in order and if fd0 was free it would go to cestnepasun[0] |
|
1334
9fd2bcedbeb5
mount: start on option parsing, implement loopback and bind mount autodetection.
Rob Landley <rob@landley.net>
parents:
1327
diff
changeset
|
156 if (pipes) { |
|
1472
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
157 for (pid = 0; pid < 2; pid++) { |
|
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
158 if (pipes[pid] == -1) continue; |
|
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
159 if (pipe(cestnepasun+(2*pid))) perror_exit("pipe"); |
|
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
160 pipes[pid] = cestnepasun[pid+1]; |
|
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
161 } |
|
1334
9fd2bcedbeb5
mount: start on option parsing, implement loopback and bind mount autodetection.
Rob Landley <rob@landley.net>
parents:
1327
diff
changeset
|
162 } |
|
9fd2bcedbeb5
mount: start on option parsing, implement loopback and bind mount autodetection.
Rob Landley <rob@landley.net>
parents:
1327
diff
changeset
|
163 |
|
9fd2bcedbeb5
mount: start on option parsing, implement loopback and bind mount autodetection.
Rob Landley <rob@landley.net>
parents:
1327
diff
changeset
|
164 // Child process |
|
9fd2bcedbeb5
mount: start on option parsing, implement loopback and bind mount autodetection.
Rob Landley <rob@landley.net>
parents:
1327
diff
changeset
|
165 if (!(pid = xfork())) { |
|
9fd2bcedbeb5
mount: start on option parsing, implement loopback and bind mount autodetection.
Rob Landley <rob@landley.net>
parents:
1327
diff
changeset
|
166 // Dance of the stdin/stdout redirection. |
|
9fd2bcedbeb5
mount: start on option parsing, implement loopback and bind mount autodetection.
Rob Landley <rob@landley.net>
parents:
1327
diff
changeset
|
167 if (pipes) { |
|
1472
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
168 // if we had no stdin/out, pipe handles could overlap, so test for it |
|
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
169 // and free up potentially overlapping pipe handles before reuse |
|
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
170 if (pipes[1] != -1) close(cestnepasun[2]); |
|
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
171 if (pipes[0] != -1) { |
|
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
172 close(cestnepasun[1]); |
|
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
173 if (cestnepasun[0]) { |
|
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
174 dup2(cestnepasun[0], 0); |
|
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
175 close(cestnepasun[0]); |
|
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
176 } |
|
1334
9fd2bcedbeb5
mount: start on option parsing, implement loopback and bind mount autodetection.
Rob Landley <rob@landley.net>
parents:
1327
diff
changeset
|
177 } |
|
1472
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
178 if (pipes[1] != -1) { |
|
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
179 dup2(cestnepasun[3], 1); |
|
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
180 dup2(cestnepasun[3], 2); |
|
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
181 if (cestnepasun[3] > 2 || !cestnepasun[3]) close(cestnepasun[3]); |
|
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
182 } |
|
1334
9fd2bcedbeb5
mount: start on option parsing, implement loopback and bind mount autodetection.
Rob Landley <rob@landley.net>
parents:
1327
diff
changeset
|
183 } |
|
9fd2bcedbeb5
mount: start on option parsing, implement loopback and bind mount autodetection.
Rob Landley <rob@landley.net>
parents:
1327
diff
changeset
|
184 if (argv) { |
|
9fd2bcedbeb5
mount: start on option parsing, implement loopback and bind mount autodetection.
Rob Landley <rob@landley.net>
parents:
1327
diff
changeset
|
185 if (CFG_TOYBOX) toy_exec(argv); |
|
9fd2bcedbeb5
mount: start on option parsing, implement loopback and bind mount autodetection.
Rob Landley <rob@landley.net>
parents:
1327
diff
changeset
|
186 execvp(argv[0], argv); |
|
9fd2bcedbeb5
mount: start on option parsing, implement loopback and bind mount autodetection.
Rob Landley <rob@landley.net>
parents:
1327
diff
changeset
|
187 _exit(127); |
|
9fd2bcedbeb5
mount: start on option parsing, implement loopback and bind mount autodetection.
Rob Landley <rob@landley.net>
parents:
1327
diff
changeset
|
188 } |
|
9fd2bcedbeb5
mount: start on option parsing, implement loopback and bind mount autodetection.
Rob Landley <rob@landley.net>
parents:
1327
diff
changeset
|
189 return 0; |
|
9fd2bcedbeb5
mount: start on option parsing, implement loopback and bind mount autodetection.
Rob Landley <rob@landley.net>
parents:
1327
diff
changeset
|
190 |
|
1472
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
191 } |
|
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
192 |
|
1334
9fd2bcedbeb5
mount: start on option parsing, implement loopback and bind mount autodetection.
Rob Landley <rob@landley.net>
parents:
1327
diff
changeset
|
193 // Parent process |
|
1472
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
194 if (pipes) { |
|
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
195 if (pipes[0] != -1) close(cestnepasun[0]); |
|
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
196 if (pipes[1] != -1) close(cestnepasun[3]); |
|
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
197 } |
|
1334
9fd2bcedbeb5
mount: start on option parsing, implement loopback and bind mount autodetection.
Rob Landley <rob@landley.net>
parents:
1327
diff
changeset
|
198 |
|
1472
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
199 return pid; |
|
1334
9fd2bcedbeb5
mount: start on option parsing, implement loopback and bind mount autodetection.
Rob Landley <rob@landley.net>
parents:
1327
diff
changeset
|
200 } |
|
9fd2bcedbeb5
mount: start on option parsing, implement loopback and bind mount autodetection.
Rob Landley <rob@landley.net>
parents:
1327
diff
changeset
|
201 |
|
1472
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
202 int xpclose_both(pid_t pid, int *pipes) |
|
1334
9fd2bcedbeb5
mount: start on option parsing, implement loopback and bind mount autodetection.
Rob Landley <rob@landley.net>
parents:
1327
diff
changeset
|
203 { |
|
9fd2bcedbeb5
mount: start on option parsing, implement loopback and bind mount autodetection.
Rob Landley <rob@landley.net>
parents:
1327
diff
changeset
|
204 int rc = 127; |
|
9fd2bcedbeb5
mount: start on option parsing, implement loopback and bind mount autodetection.
Rob Landley <rob@landley.net>
parents:
1327
diff
changeset
|
205 |
|
9fd2bcedbeb5
mount: start on option parsing, implement loopback and bind mount autodetection.
Rob Landley <rob@landley.net>
parents:
1327
diff
changeset
|
206 if (pipes) { |
|
9fd2bcedbeb5
mount: start on option parsing, implement loopback and bind mount autodetection.
Rob Landley <rob@landley.net>
parents:
1327
diff
changeset
|
207 close(pipes[0]); |
|
9fd2bcedbeb5
mount: start on option parsing, implement loopback and bind mount autodetection.
Rob Landley <rob@landley.net>
parents:
1327
diff
changeset
|
208 close(pipes[1]); |
|
9fd2bcedbeb5
mount: start on option parsing, implement loopback and bind mount autodetection.
Rob Landley <rob@landley.net>
parents:
1327
diff
changeset
|
209 } |
|
9fd2bcedbeb5
mount: start on option parsing, implement loopback and bind mount autodetection.
Rob Landley <rob@landley.net>
parents:
1327
diff
changeset
|
210 waitpid(pid, &rc, 0); |
|
9fd2bcedbeb5
mount: start on option parsing, implement loopback and bind mount autodetection.
Rob Landley <rob@landley.net>
parents:
1327
diff
changeset
|
211 |
|
9fd2bcedbeb5
mount: start on option parsing, implement loopback and bind mount autodetection.
Rob Landley <rob@landley.net>
parents:
1327
diff
changeset
|
212 return WIFEXITED(rc) ? WEXITSTATUS(rc) : WTERMSIG(rc) + 127; |
|
9fd2bcedbeb5
mount: start on option parsing, implement loopback and bind mount autodetection.
Rob Landley <rob@landley.net>
parents:
1327
diff
changeset
|
213 } |
|
9fd2bcedbeb5
mount: start on option parsing, implement loopback and bind mount autodetection.
Rob Landley <rob@landley.net>
parents:
1327
diff
changeset
|
214 |
|
1472
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
215 // Wrapper to xpopen with a pipe for just one of stdin/stdout |
|
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
216 pid_t xpopen(char **argv, int *pipe, int stdout) |
|
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
217 { |
|
1473
bff076394df5
Second attempt at one way xpopen().
Rob Landley <rob@landley.net>
parents:
1472
diff
changeset
|
218 int pipes[2], pid; |
|
1472
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
219 |
|
1473
bff076394df5
Second attempt at one way xpopen().
Rob Landley <rob@landley.net>
parents:
1472
diff
changeset
|
220 pipes[!stdout] = -1; |
|
bff076394df5
Second attempt at one way xpopen().
Rob Landley <rob@landley.net>
parents:
1472
diff
changeset
|
221 pipes[!!stdout] = 0; |
|
bff076394df5
Second attempt at one way xpopen().
Rob Landley <rob@landley.net>
parents:
1472
diff
changeset
|
222 pid = xpopen_both(argv, pipes); |
|
bff076394df5
Second attempt at one way xpopen().
Rob Landley <rob@landley.net>
parents:
1472
diff
changeset
|
223 *pipe = pid ? pipes[!!stdout] : -1; |
|
1472
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
224 |
|
1473
bff076394df5
Second attempt at one way xpopen().
Rob Landley <rob@landley.net>
parents:
1472
diff
changeset
|
225 return pid; |
|
1472
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
226 } |
|
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
227 |
|
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
228 int xpclose(pid_t pid, int pipe) |
|
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
229 { |
|
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
230 close(pipe); |
|
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
231 |
|
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
232 return xpclose_both(pid, 0); |
|
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
233 } |
|
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
234 |
|
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
235 // Call xpopen and wait for it to finish, keeping existing stdin/stdout. |
|
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
236 int xrun(char **argv) |
|
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
237 { |
|
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
238 return xpclose_both(xpopen_both(argv, 0), 0); |
|
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
239 } |
|
2f9bc9495144
Split xpopen() into xpopen_both(), xpopen(), and xrun() depending on whether we want to redirect both, one, or neither of stdin/stdout.
Rob Landley <rob@landley.net>
parents:
1420
diff
changeset
|
240 |
| 951 | 241 void xaccess(char *path, int flags) |
| 242 { | |
| 243 if (access(path, flags)) perror_exit("Can't access '%s'", path); | |
| 244 } | |
| 245 | |
| 246 // Die unless we can delete a file. (File must exist to be deleted.) | |
| 247 void xunlink(char *path) | |
| 248 { | |
| 249 if (unlink(path)) perror_exit("unlink '%s'", path); | |
| 250 } | |
| 251 | |
| 252 // Die unless we can open/create a file, returning file descriptor. | |
| 253 int xcreate(char *path, int flags, int mode) | |
| 254 { | |
|
1403
411cf82cdf77
Default xcreate/xopen to O_CLOEXEC. (Pass O_CLOEXEC in the flags to switch it back off.)
Rob Landley <rob@landley.net>
parents:
1400
diff
changeset
|
255 int fd = open(path, flags^O_CLOEXEC, mode); |
| 951 | 256 if (fd == -1) perror_exit("%s", path); |
| 257 return fd; | |
| 258 } | |
| 259 | |
| 260 // Die unless we can open a file, returning file descriptor. | |
| 261 int xopen(char *path, int flags) | |
| 262 { | |
| 263 return xcreate(path, flags, 0); | |
| 264 } | |
| 265 | |
| 266 void xclose(int fd) | |
| 267 { | |
| 268 if (close(fd)) perror_exit("xclose"); | |
| 269 } | |
| 270 | |
| 271 int xdup(int fd) | |
| 272 { | |
| 273 if (fd != -1) { | |
| 274 fd = dup(fd); | |
| 275 if (fd == -1) perror_exit("xdup"); | |
| 276 } | |
| 277 return fd; | |
| 278 } | |
| 279 | |
|
991
252caf3d2b88
Forgot to check in xfdopen(). My bad.
Rob Landley <rob@landley.net>
parents:
956
diff
changeset
|
280 FILE *xfdopen(int fd, char *mode) |
|
252caf3d2b88
Forgot to check in xfdopen(). My bad.
Rob Landley <rob@landley.net>
parents:
956
diff
changeset
|
281 { |
|
252caf3d2b88
Forgot to check in xfdopen(). My bad.
Rob Landley <rob@landley.net>
parents:
956
diff
changeset
|
282 FILE *f = fdopen(fd, mode); |
|
252caf3d2b88
Forgot to check in xfdopen(). My bad.
Rob Landley <rob@landley.net>
parents:
956
diff
changeset
|
283 |
|
252caf3d2b88
Forgot to check in xfdopen(). My bad.
Rob Landley <rob@landley.net>
parents:
956
diff
changeset
|
284 if (!f) perror_exit("xfdopen"); |
|
252caf3d2b88
Forgot to check in xfdopen(). My bad.
Rob Landley <rob@landley.net>
parents:
956
diff
changeset
|
285 |
|
252caf3d2b88
Forgot to check in xfdopen(). My bad.
Rob Landley <rob@landley.net>
parents:
956
diff
changeset
|
286 return f; |
|
252caf3d2b88
Forgot to check in xfdopen(). My bad.
Rob Landley <rob@landley.net>
parents:
956
diff
changeset
|
287 } |
|
252caf3d2b88
Forgot to check in xfdopen(). My bad.
Rob Landley <rob@landley.net>
parents:
956
diff
changeset
|
288 |
| 951 | 289 // Die unless we can open/create a file, returning FILE *. |
| 290 FILE *xfopen(char *path, char *mode) | |
| 291 { | |
| 292 FILE *f = fopen(path, mode); | |
| 293 if (!f) perror_exit("No file %s", path); | |
| 294 return f; | |
| 295 } | |
| 296 | |
| 297 // Die if there's an error other than EOF. | |
| 298 size_t xread(int fd, void *buf, size_t len) | |
| 299 { | |
| 300 ssize_t ret = read(fd, buf, len); | |
| 301 if (ret < 0) perror_exit("xread"); | |
| 302 | |
| 303 return ret; | |
| 304 } | |
| 305 | |
| 306 void xreadall(int fd, void *buf, size_t len) | |
| 307 { | |
| 308 if (len != readall(fd, buf, len)) perror_exit("xreadall"); | |
| 309 } | |
| 310 | |
| 311 // There's no xwriteall(), just xwrite(). When we read, there may or may not | |
| 312 // be more data waiting. When we write, there is data and it had better go | |
| 313 // somewhere. | |
| 314 | |
| 315 void xwrite(int fd, void *buf, size_t len) | |
| 316 { | |
| 317 if (len != writeall(fd, buf, len)) perror_exit("xwrite"); | |
| 318 } | |
| 319 | |
| 320 // Die if lseek fails, probably due to being called on a pipe. | |
| 321 | |
| 322 off_t xlseek(int fd, off_t offset, int whence) | |
| 323 { | |
| 324 offset = lseek(fd, offset, whence); | |
| 325 if (offset<0) perror_exit("lseek"); | |
| 326 | |
| 327 return offset; | |
| 328 } | |
| 329 | |
| 330 char *xgetcwd(void) | |
| 331 { | |
| 332 char *buf = getcwd(NULL, 0); | |
| 333 if (!buf) perror_exit("xgetcwd"); | |
| 334 | |
| 335 return buf; | |
| 336 } | |
| 337 | |
| 338 void xstat(char *path, struct stat *st) | |
| 339 { | |
| 340 if(stat(path, st)) perror_exit("Can't stat %s", path); | |
| 341 } | |
| 342 | |
| 343 // Cannonicalize path, even to file with one or more missing components at end. | |
| 344 // if exact, require last path component to exist | |
| 345 char *xabspath(char *path, int exact) | |
| 346 { | |
| 347 struct string_list *todo, *done = 0; | |
| 348 int try = 9999, dirfd = open("/", 0);; | |
| 349 char buf[4096], *ret; | |
| 350 | |
| 351 // If this isn't an absolute path, start with cwd. | |
| 352 if (*path != '/') { | |
| 353 char *temp = xgetcwd(); | |
| 354 | |
| 355 splitpath(path, splitpath(temp, &todo)); | |
| 356 free(temp); | |
| 357 } else splitpath(path, &todo); | |
| 358 | |
| 359 // Iterate through path components | |
| 360 while (todo) { | |
| 361 struct string_list *new = llist_pop(&todo), **tail; | |
| 362 ssize_t len; | |
| 363 | |
| 364 if (!try--) { | |
| 365 errno = ELOOP; | |
| 366 goto error; | |
| 367 } | |
| 368 | |
| 369 // Removable path componenents. | |
| 370 if (!strcmp(new->str, ".") || !strcmp(new->str, "..")) { | |
| 371 int x = new->str[1]; | |
| 372 | |
| 373 free(new); | |
| 374 if (x) { | |
| 375 if (done) free(llist_pop(&done)); | |
| 376 len = 0; | |
| 377 } else continue; | |
| 378 | |
| 379 // Is this a symlink? | |
| 380 } else len=readlinkat(dirfd, new->str, buf, 4096); | |
| 381 | |
| 382 if (len>4095) goto error; | |
| 383 if (len<1) { | |
| 384 int fd; | |
| 385 char *s = ".."; | |
| 386 | |
| 387 // For .. just move dirfd | |
| 388 if (len) { | |
| 389 // Not a symlink: add to linked list, move dirfd, fail if error | |
| 390 if ((exact || todo) && errno != EINVAL) goto error; | |
| 391 new->next = done; | |
| 392 done = new; | |
| 393 if (errno == EINVAL && !todo) break; | |
| 394 s = new->str; | |
| 395 } | |
| 396 fd = openat(dirfd, s, 0); | |
| 397 if (fd == -1 && (exact || todo || errno != ENOENT)) goto error; | |
| 398 close(dirfd); | |
| 399 dirfd = fd; | |
| 400 continue; | |
| 401 } | |
| 402 | |
| 403 // If this symlink is to an absolute path, discard existing resolved path | |
| 404 buf[len] = 0; | |
| 405 if (*buf == '/') { | |
| 406 llist_traverse(done, free); | |
| 407 done=0; | |
| 408 close(dirfd); | |
| 409 dirfd = open("/", 0); | |
| 410 } | |
| 411 free(new); | |
| 412 | |
| 413 // prepend components of new path. Note symlink to "/" will leave new NULL | |
| 414 tail = splitpath(buf, &new); | |
| 415 | |
| 416 // symlink to "/" will return null and leave tail alone | |
| 417 if (new) { | |
| 418 *tail = todo; | |
| 419 todo = new; | |
| 420 } | |
| 421 } | |
| 422 close(dirfd); | |
| 423 | |
| 424 // At this point done has the path, in reverse order. Reverse list while | |
| 425 // calculating buffer length. | |
| 426 | |
| 427 try = 2; | |
| 428 while (done) { | |
| 429 struct string_list *temp = llist_pop(&done);; | |
| 430 | |
| 431 if (todo) try++; | |
| 432 try += strlen(temp->str); | |
| 433 temp->next = todo; | |
| 434 todo = temp; | |
| 435 } | |
| 436 | |
| 437 // Assemble return buffer | |
| 438 | |
| 439 ret = xmalloc(try); | |
| 440 *ret = '/'; | |
| 441 ret [try = 1] = 0; | |
| 442 while (todo) { | |
| 443 if (try>1) ret[try++] = '/'; | |
| 444 try = stpcpy(ret+try, todo->str) - ret; | |
| 445 free(llist_pop(&todo)); | |
| 446 } | |
| 447 | |
| 448 return ret; | |
| 449 | |
| 450 error: | |
| 451 close(dirfd); | |
| 452 llist_traverse(todo, free); | |
| 453 llist_traverse(done, free); | |
| 454 | |
| 455 return NULL; | |
| 456 } | |
| 457 | |
| 458 void xchdir(char *path) | |
| 459 { | |
| 460 if (chdir(path)) error_exit("chdir '%s'", path); | |
| 461 } | |
| 462 | |
|
1156
faf7117c4489
Fix some issues raised (albeit indirectly) by Isaac Dunham.
Rob Landley <rob@landley.net>
parents:
1130
diff
changeset
|
463 void xchroot(char *path) |
|
faf7117c4489
Fix some issues raised (albeit indirectly) by Isaac Dunham.
Rob Landley <rob@landley.net>
parents:
1130
diff
changeset
|
464 { |
|
faf7117c4489
Fix some issues raised (albeit indirectly) by Isaac Dunham.
Rob Landley <rob@landley.net>
parents:
1130
diff
changeset
|
465 if (chroot(path)) error_exit("chroot '%s'", path); |
|
faf7117c4489
Fix some issues raised (albeit indirectly) by Isaac Dunham.
Rob Landley <rob@landley.net>
parents:
1130
diff
changeset
|
466 xchdir("/"); |
|
faf7117c4489
Fix some issues raised (albeit indirectly) by Isaac Dunham.
Rob Landley <rob@landley.net>
parents:
1130
diff
changeset
|
467 } |
|
faf7117c4489
Fix some issues raised (albeit indirectly) by Isaac Dunham.
Rob Landley <rob@landley.net>
parents:
1130
diff
changeset
|
468 |
|
1129
c644f85444d0
Move xgetpwuid() and xgetgrgid() into xwrap.c
Rob Landley <rob@landley.net>
parents:
1105
diff
changeset
|
469 struct passwd *xgetpwuid(uid_t uid) |
|
c644f85444d0
Move xgetpwuid() and xgetgrgid() into xwrap.c
Rob Landley <rob@landley.net>
parents:
1105
diff
changeset
|
470 { |
|
c644f85444d0
Move xgetpwuid() and xgetgrgid() into xwrap.c
Rob Landley <rob@landley.net>
parents:
1105
diff
changeset
|
471 struct passwd *pwd = getpwuid(uid); |
|
1130
6df194c6de88
Add xgetpwnam() to lib/xwrap.c.
Rob Landley <rob@landley.net>
parents:
1129
diff
changeset
|
472 if (!pwd) error_exit("bad uid %ld", (long)uid); |
|
1129
c644f85444d0
Move xgetpwuid() and xgetgrgid() into xwrap.c
Rob Landley <rob@landley.net>
parents:
1105
diff
changeset
|
473 return pwd; |
|
c644f85444d0
Move xgetpwuid() and xgetgrgid() into xwrap.c
Rob Landley <rob@landley.net>
parents:
1105
diff
changeset
|
474 } |
|
c644f85444d0
Move xgetpwuid() and xgetgrgid() into xwrap.c
Rob Landley <rob@landley.net>
parents:
1105
diff
changeset
|
475 |
|
c644f85444d0
Move xgetpwuid() and xgetgrgid() into xwrap.c
Rob Landley <rob@landley.net>
parents:
1105
diff
changeset
|
476 struct group *xgetgrgid(gid_t gid) |
|
c644f85444d0
Move xgetpwuid() and xgetgrgid() into xwrap.c
Rob Landley <rob@landley.net>
parents:
1105
diff
changeset
|
477 { |
|
c644f85444d0
Move xgetpwuid() and xgetgrgid() into xwrap.c
Rob Landley <rob@landley.net>
parents:
1105
diff
changeset
|
478 struct group *group = getgrgid(gid); |
|
1400
31cb9ba1815c
Improve gid/uid error messages.
Rob Landley <rob@landley.net>
parents:
1334
diff
changeset
|
479 |
|
31cb9ba1815c
Improve gid/uid error messages.
Rob Landley <rob@landley.net>
parents:
1334
diff
changeset
|
480 if (!group) perror_exit("gid %ld", (long)gid); |
|
1129
c644f85444d0
Move xgetpwuid() and xgetgrgid() into xwrap.c
Rob Landley <rob@landley.net>
parents:
1105
diff
changeset
|
481 return group; |
|
c644f85444d0
Move xgetpwuid() and xgetgrgid() into xwrap.c
Rob Landley <rob@landley.net>
parents:
1105
diff
changeset
|
482 } |
|
c644f85444d0
Move xgetpwuid() and xgetgrgid() into xwrap.c
Rob Landley <rob@landley.net>
parents:
1105
diff
changeset
|
483 |
|
1130
6df194c6de88
Add xgetpwnam() to lib/xwrap.c.
Rob Landley <rob@landley.net>
parents:
1129
diff
changeset
|
484 struct passwd *xgetpwnam(char *name) |
|
6df194c6de88
Add xgetpwnam() to lib/xwrap.c.
Rob Landley <rob@landley.net>
parents:
1129
diff
changeset
|
485 { |
|
6df194c6de88
Add xgetpwnam() to lib/xwrap.c.
Rob Landley <rob@landley.net>
parents:
1129
diff
changeset
|
486 struct passwd *up = getpwnam(name); |
|
1400
31cb9ba1815c
Improve gid/uid error messages.
Rob Landley <rob@landley.net>
parents:
1334
diff
changeset
|
487 |
|
31cb9ba1815c
Improve gid/uid error messages.
Rob Landley <rob@landley.net>
parents:
1334
diff
changeset
|
488 if (!up) perror_exit("user '%s'", name); |
|
1130
6df194c6de88
Add xgetpwnam() to lib/xwrap.c.
Rob Landley <rob@landley.net>
parents:
1129
diff
changeset
|
489 return up; |
|
6df194c6de88
Add xgetpwnam() to lib/xwrap.c.
Rob Landley <rob@landley.net>
parents:
1129
diff
changeset
|
490 } |
|
6df194c6de88
Add xgetpwnam() to lib/xwrap.c.
Rob Landley <rob@landley.net>
parents:
1129
diff
changeset
|
491 |
|
1420
9d760c092c41
Implement exec -user, -group, and -newer. Enable find in defconfig.
Rob Landley <rob@landley.net>
parents:
1408
diff
changeset
|
492 struct group *xgetgrnam(char *name) |
|
9d760c092c41
Implement exec -user, -group, and -newer. Enable find in defconfig.
Rob Landley <rob@landley.net>
parents:
1408
diff
changeset
|
493 { |
|
9d760c092c41
Implement exec -user, -group, and -newer. Enable find in defconfig.
Rob Landley <rob@landley.net>
parents:
1408
diff
changeset
|
494 struct group *gr = getgrnam(name); |
|
9d760c092c41
Implement exec -user, -group, and -newer. Enable find in defconfig.
Rob Landley <rob@landley.net>
parents:
1408
diff
changeset
|
495 |
|
9d760c092c41
Implement exec -user, -group, and -newer. Enable find in defconfig.
Rob Landley <rob@landley.net>
parents:
1408
diff
changeset
|
496 if (!gr) perror_exit("group '%s'", name); |
|
9d760c092c41
Implement exec -user, -group, and -newer. Enable find in defconfig.
Rob Landley <rob@landley.net>
parents:
1408
diff
changeset
|
497 return gr; |
|
9d760c092c41
Implement exec -user, -group, and -newer. Enable find in defconfig.
Rob Landley <rob@landley.net>
parents:
1408
diff
changeset
|
498 } |
|
9d760c092c41
Implement exec -user, -group, and -newer. Enable find in defconfig.
Rob Landley <rob@landley.net>
parents:
1408
diff
changeset
|
499 |
|
1156
faf7117c4489
Fix some issues raised (albeit indirectly) by Isaac Dunham.
Rob Landley <rob@landley.net>
parents:
1130
diff
changeset
|
500 // setuid() can fail (for example, too many processes belonging to that user), |
|
faf7117c4489
Fix some issues raised (albeit indirectly) by Isaac Dunham.
Rob Landley <rob@landley.net>
parents:
1130
diff
changeset
|
501 // which opens a security hole if the process continues as the original user. |
|
faf7117c4489
Fix some issues raised (albeit indirectly) by Isaac Dunham.
Rob Landley <rob@landley.net>
parents:
1130
diff
changeset
|
502 |
|
faf7117c4489
Fix some issues raised (albeit indirectly) by Isaac Dunham.
Rob Landley <rob@landley.net>
parents:
1130
diff
changeset
|
503 void xsetuser(struct passwd *pwd) |
|
faf7117c4489
Fix some issues raised (albeit indirectly) by Isaac Dunham.
Rob Landley <rob@landley.net>
parents:
1130
diff
changeset
|
504 { |
|
faf7117c4489
Fix some issues raised (albeit indirectly) by Isaac Dunham.
Rob Landley <rob@landley.net>
parents:
1130
diff
changeset
|
505 if (initgroups(pwd->pw_name, pwd->pw_gid) || setgid(pwd->pw_uid) |
|
faf7117c4489
Fix some issues raised (albeit indirectly) by Isaac Dunham.
Rob Landley <rob@landley.net>
parents:
1130
diff
changeset
|
506 || setuid(pwd->pw_uid)) perror_exit("xsetuser '%s'", pwd->pw_name); |
|
faf7117c4489
Fix some issues raised (albeit indirectly) by Isaac Dunham.
Rob Landley <rob@landley.net>
parents:
1130
diff
changeset
|
507 } |
|
faf7117c4489
Fix some issues raised (albeit indirectly) by Isaac Dunham.
Rob Landley <rob@landley.net>
parents:
1130
diff
changeset
|
508 |
| 951 | 509 // This can return null (meaning file not found). It just won't return null |
| 510 // for memory allocation reasons. | |
| 511 char *xreadlink(char *name) | |
| 512 { | |
| 513 int len, size = 0; | |
| 514 char *buf = 0; | |
| 515 | |
| 516 // Grow by 64 byte chunks until it's big enough. | |
| 517 for(;;) { | |
| 518 size +=64; | |
| 519 buf = xrealloc(buf, size); | |
| 520 len = readlink(name, buf, size); | |
| 521 | |
| 522 if (len<0) { | |
| 523 free(buf); | |
| 524 return 0; | |
| 525 } | |
| 526 if (len<size) { | |
| 527 buf[len]=0; | |
| 528 return buf; | |
| 529 } | |
| 530 } | |
| 531 } | |
| 532 | |
|
1170
8afe1fde9314
Pass through all the readfile() arguments from xreadfile().
Rob Landley <rob@landley.net>
parents:
1156
diff
changeset
|
533 char *xreadfile(char *name, char *buf, off_t len) |
| 951 | 534 { |
|
1170
8afe1fde9314
Pass through all the readfile() arguments from xreadfile().
Rob Landley <rob@landley.net>
parents:
1156
diff
changeset
|
535 if (!(buf = readfile(name, buf, len))) perror_exit("Bad '%s'", name); |
|
8afe1fde9314
Pass through all the readfile() arguments from xreadfile().
Rob Landley <rob@landley.net>
parents:
1156
diff
changeset
|
536 |
| 951 | 537 return buf; |
| 538 } | |
| 539 | |
| 540 int xioctl(int fd, int request, void *data) | |
| 541 { | |
| 542 int rc; | |
| 543 | |
| 544 errno = 0; | |
| 545 rc = ioctl(fd, request, data); | |
| 546 if (rc == -1 && errno) perror_exit("ioctl %x", request); | |
| 547 | |
| 548 return rc; | |
| 549 } | |
| 550 | |
| 551 // Open a /var/run/NAME.pid file, dying if we can't write it or if it currently | |
| 552 // exists and is this executable. | |
| 553 void xpidfile(char *name) | |
| 554 { | |
| 555 char pidfile[256], spid[32]; | |
| 556 int i, fd; | |
| 557 pid_t pid; | |
| 558 | |
| 559 sprintf(pidfile, "/var/run/%s.pid", name); | |
| 560 // Try three times to open the sucker. | |
| 561 for (i=0; i<3; i++) { | |
| 1028 | 562 fd = open(pidfile, O_CREAT|O_EXCL|O_WRONLY, 0644); |
| 951 | 563 if (fd != -1) break; |
| 564 | |
| 565 // If it already existed, read it. Loop for race condition. | |
| 566 fd = open(pidfile, O_RDONLY); | |
| 567 if (fd == -1) continue; | |
| 568 | |
| 569 // Is the old program still there? | |
| 570 spid[xread(fd, spid, sizeof(spid)-1)] = 0; | |
| 571 close(fd); | |
| 572 pid = atoi(spid); | |
|
1053
501fd74c028e
Fix for xpidfile spotted by Felix Janda.
Rob Landley <rob@landley.net>
parents:
1043
diff
changeset
|
573 if (pid < 1 || (kill(pid, 0) && errno == ESRCH)) unlink(pidfile); |
| 951 | 574 |
| 575 // An else with more sanity checking might be nice here. | |
| 576 } | |
| 577 | |
| 578 if (i == 3) error_exit("xpidfile %s", name); | |
| 579 | |
| 580 xwrite(fd, spid, sprintf(spid, "%ld\n", (long)getpid())); | |
| 581 close(fd); | |
| 582 } | |
| 583 | |
| 584 // Copy the rest of in to out and close both files. | |
| 585 | |
| 586 void xsendfile(int in, int out) | |
| 587 { | |
| 588 long len; | |
| 589 char buf[4096]; | |
| 590 | |
| 591 if (in<0) return; | |
| 592 for (;;) { | |
| 593 len = xread(in, buf, 4096); | |
| 594 if (len<1) break; | |
| 595 xwrite(out, buf, len); | |
| 596 } | |
| 597 } | |
|
952
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
598 |
|
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
599 // parse fractional seconds with optional s/m/h/d suffix |
|
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
600 long xparsetime(char *arg, long units, long *fraction) |
|
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
601 { |
|
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
602 double d; |
|
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
603 long l; |
|
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
604 |
|
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
605 if (CFG_TOYBOX_FLOAT) d = strtod(arg, &arg); |
|
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
606 else l = strtoul(arg, &arg, 10); |
|
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
607 |
|
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
608 // Parse suffix |
|
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
609 if (*arg) { |
|
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
610 int ismhd[]={1,60,3600,86400}, i = stridx("smhd", *arg); |
|
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
611 |
|
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
612 if (i == -1) error_exit("Unknown suffix '%c'", *arg); |
|
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
613 if (CFG_TOYBOX_FLOAT) d *= ismhd[i]; |
|
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
614 else l *= ismhd[i]; |
|
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
615 } |
|
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
616 |
|
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
617 if (CFG_TOYBOX_FLOAT) { |
|
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
618 l = (long)d; |
|
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
619 if (fraction) *fraction = units*(d-l); |
|
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
620 } else if (fraction) *fraction = 0; |
|
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
621 |
|
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
622 return l; |
|
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
623 } |
|
1235
63db77909fc8
Group headers by standard (POSIX or LSB) or function (internationalization, networking). Move headers standards ignore (but which have been there >15 years) to lib/portability.h. Fold xregcomp into lib since it's posix.
Rob Landley <rob@landley.net>
parents:
1219
diff
changeset
|
624 |
|
63db77909fc8
Group headers by standard (POSIX or LSB) or function (internationalization, networking). Move headers standards ignore (but which have been there >15 years) to lib/portability.h. Fold xregcomp into lib since it's posix.
Rob Landley <rob@landley.net>
parents:
1219
diff
changeset
|
625 // Compile a regular expression into a regex_t |
|
63db77909fc8
Group headers by standard (POSIX or LSB) or function (internationalization, networking). Move headers standards ignore (but which have been there >15 years) to lib/portability.h. Fold xregcomp into lib since it's posix.
Rob Landley <rob@landley.net>
parents:
1219
diff
changeset
|
626 void xregcomp(regex_t *preg, char *regex, int cflags) |
|
63db77909fc8
Group headers by standard (POSIX or LSB) or function (internationalization, networking). Move headers standards ignore (but which have been there >15 years) to lib/portability.h. Fold xregcomp into lib since it's posix.
Rob Landley <rob@landley.net>
parents:
1219
diff
changeset
|
627 { |
|
63db77909fc8
Group headers by standard (POSIX or LSB) or function (internationalization, networking). Move headers standards ignore (but which have been there >15 years) to lib/portability.h. Fold xregcomp into lib since it's posix.
Rob Landley <rob@landley.net>
parents:
1219
diff
changeset
|
628 int rc = regcomp(preg, regex, cflags); |
|
63db77909fc8
Group headers by standard (POSIX or LSB) or function (internationalization, networking). Move headers standards ignore (but which have been there >15 years) to lib/portability.h. Fold xregcomp into lib since it's posix.
Rob Landley <rob@landley.net>
parents:
1219
diff
changeset
|
629 |
|
63db77909fc8
Group headers by standard (POSIX or LSB) or function (internationalization, networking). Move headers standards ignore (but which have been there >15 years) to lib/portability.h. Fold xregcomp into lib since it's posix.
Rob Landley <rob@landley.net>
parents:
1219
diff
changeset
|
630 if (rc) { |
|
63db77909fc8
Group headers by standard (POSIX or LSB) or function (internationalization, networking). Move headers standards ignore (but which have been there >15 years) to lib/portability.h. Fold xregcomp into lib since it's posix.
Rob Landley <rob@landley.net>
parents:
1219
diff
changeset
|
631 regerror(rc, preg, libbuf, sizeof(libbuf)); |
|
63db77909fc8
Group headers by standard (POSIX or LSB) or function (internationalization, networking). Move headers standards ignore (but which have been there >15 years) to lib/portability.h. Fold xregcomp into lib since it's posix.
Rob Landley <rob@landley.net>
parents:
1219
diff
changeset
|
632 error_exit("xregcomp: %s", libbuf); |
|
63db77909fc8
Group headers by standard (POSIX or LSB) or function (internationalization, networking). Move headers standards ignore (but which have been there >15 years) to lib/portability.h. Fold xregcomp into lib since it's posix.
Rob Landley <rob@landley.net>
parents:
1219
diff
changeset
|
633 } |
|
63db77909fc8
Group headers by standard (POSIX or LSB) or function (internationalization, networking). Move headers standards ignore (but which have been there >15 years) to lib/portability.h. Fold xregcomp into lib since it's posix.
Rob Landley <rob@landley.net>
parents:
1219
diff
changeset
|
634 } |