Mercurial > hg > toybox
annotate lib/xwrap.c @ 1215:4eaac3e63fa7 draft
Cleanup freeramdisk: tabs to 2 spaces, square brackets for option name, do optional cleanup under if (CFG_TOYBOX_FREE) guard.
author | Rob Landley <rob@landley.net> |
---|---|
date | Sun, 09 Mar 2014 14:38:51 -0500 |
parents | 0752b2d58909 |
children | 468444e5c7c5 |
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); | |
97 if (ferror(stdout)) perror_exit("write"); | |
98 } | |
99 | |
100 void xputs(char *s) | |
101 { | |
102 if (EOF == puts(s) || fflush(stdout)) perror_exit("write"); | |
103 } | |
104 | |
105 void xputc(char c) | |
106 { | |
107 if (EOF == fputc(c, stdout) || fflush(stdout)) perror_exit("write"); | |
108 } | |
109 | |
110 void xflush(void) | |
111 { | |
112 if (fflush(stdout)) perror_exit("write");; | |
113 } | |
114 | |
952
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
115 // 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
|
116 // 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
|
117 void xexec_optargs(int skip) |
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
118 { |
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
119 char **s = toys.optargs; |
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
120 |
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
121 toys.optargs = 0; |
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
122 xexec(s+skip); |
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
123 } |
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
124 |
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
125 |
951 | 126 // Die unless we can exec argv[] (or run builtin command). Note that anything |
127 // with a path isn't a builtin, so /bin/sh won't match the builtin sh. | |
128 void xexec(char **argv) | |
129 { | |
1032
40d0c96a8e89
Add scripts/single.sh to build individual non-multiplexed standalone commands.
Rob Landley <rob@landley.net>
parents:
1028
diff
changeset
|
130 if (CFG_TOYBOX) toy_exec(argv); |
951 | 131 execvp(argv[0], argv); |
132 | |
133 perror_exit("exec %s", argv[0]); | |
134 } | |
135 | |
136 void xaccess(char *path, int flags) | |
137 { | |
138 if (access(path, flags)) perror_exit("Can't access '%s'", path); | |
139 } | |
140 | |
141 // Die unless we can delete a file. (File must exist to be deleted.) | |
142 void xunlink(char *path) | |
143 { | |
144 if (unlink(path)) perror_exit("unlink '%s'", path); | |
145 } | |
146 | |
147 // Die unless we can open/create a file, returning file descriptor. | |
148 int xcreate(char *path, int flags, int mode) | |
149 { | |
150 int fd = open(path, flags, mode); | |
151 if (fd == -1) perror_exit("%s", path); | |
152 return fd; | |
153 } | |
154 | |
155 // Die unless we can open a file, returning file descriptor. | |
156 int xopen(char *path, int flags) | |
157 { | |
158 return xcreate(path, flags, 0); | |
159 } | |
160 | |
161 void xclose(int fd) | |
162 { | |
163 if (close(fd)) perror_exit("xclose"); | |
164 } | |
165 | |
166 int xdup(int fd) | |
167 { | |
168 if (fd != -1) { | |
169 fd = dup(fd); | |
170 if (fd == -1) perror_exit("xdup"); | |
171 } | |
172 return fd; | |
173 } | |
174 | |
991
252caf3d2b88
Forgot to check in xfdopen(). My bad.
Rob Landley <rob@landley.net>
parents:
956
diff
changeset
|
175 FILE *xfdopen(int fd, char *mode) |
252caf3d2b88
Forgot to check in xfdopen(). My bad.
Rob Landley <rob@landley.net>
parents:
956
diff
changeset
|
176 { |
252caf3d2b88
Forgot to check in xfdopen(). My bad.
Rob Landley <rob@landley.net>
parents:
956
diff
changeset
|
177 FILE *f = fdopen(fd, mode); |
252caf3d2b88
Forgot to check in xfdopen(). My bad.
Rob Landley <rob@landley.net>
parents:
956
diff
changeset
|
178 |
252caf3d2b88
Forgot to check in xfdopen(). My bad.
Rob Landley <rob@landley.net>
parents:
956
diff
changeset
|
179 if (!f) perror_exit("xfdopen"); |
252caf3d2b88
Forgot to check in xfdopen(). My bad.
Rob Landley <rob@landley.net>
parents:
956
diff
changeset
|
180 |
252caf3d2b88
Forgot to check in xfdopen(). My bad.
Rob Landley <rob@landley.net>
parents:
956
diff
changeset
|
181 return f; |
252caf3d2b88
Forgot to check in xfdopen(). My bad.
Rob Landley <rob@landley.net>
parents:
956
diff
changeset
|
182 } |
252caf3d2b88
Forgot to check in xfdopen(). My bad.
Rob Landley <rob@landley.net>
parents:
956
diff
changeset
|
183 |
951 | 184 // Die unless we can open/create a file, returning FILE *. |
185 FILE *xfopen(char *path, char *mode) | |
186 { | |
187 FILE *f = fopen(path, mode); | |
188 if (!f) perror_exit("No file %s", path); | |
189 return f; | |
190 } | |
191 | |
192 // Die if there's an error other than EOF. | |
193 size_t xread(int fd, void *buf, size_t len) | |
194 { | |
195 ssize_t ret = read(fd, buf, len); | |
196 if (ret < 0) perror_exit("xread"); | |
197 | |
198 return ret; | |
199 } | |
200 | |
201 void xreadall(int fd, void *buf, size_t len) | |
202 { | |
203 if (len != readall(fd, buf, len)) perror_exit("xreadall"); | |
204 } | |
205 | |
206 // There's no xwriteall(), just xwrite(). When we read, there may or may not | |
207 // be more data waiting. When we write, there is data and it had better go | |
208 // somewhere. | |
209 | |
210 void xwrite(int fd, void *buf, size_t len) | |
211 { | |
212 if (len != writeall(fd, buf, len)) perror_exit("xwrite"); | |
213 } | |
214 | |
215 // Die if lseek fails, probably due to being called on a pipe. | |
216 | |
217 off_t xlseek(int fd, off_t offset, int whence) | |
218 { | |
219 offset = lseek(fd, offset, whence); | |
220 if (offset<0) perror_exit("lseek"); | |
221 | |
222 return offset; | |
223 } | |
224 | |
225 char *xgetcwd(void) | |
226 { | |
227 char *buf = getcwd(NULL, 0); | |
228 if (!buf) perror_exit("xgetcwd"); | |
229 | |
230 return buf; | |
231 } | |
232 | |
233 void xstat(char *path, struct stat *st) | |
234 { | |
235 if(stat(path, st)) perror_exit("Can't stat %s", path); | |
236 } | |
237 | |
238 // Cannonicalize path, even to file with one or more missing components at end. | |
239 // if exact, require last path component to exist | |
240 char *xabspath(char *path, int exact) | |
241 { | |
242 struct string_list *todo, *done = 0; | |
243 int try = 9999, dirfd = open("/", 0);; | |
244 char buf[4096], *ret; | |
245 | |
246 // If this isn't an absolute path, start with cwd. | |
247 if (*path != '/') { | |
248 char *temp = xgetcwd(); | |
249 | |
250 splitpath(path, splitpath(temp, &todo)); | |
251 free(temp); | |
252 } else splitpath(path, &todo); | |
253 | |
254 // Iterate through path components | |
255 while (todo) { | |
256 struct string_list *new = llist_pop(&todo), **tail; | |
257 ssize_t len; | |
258 | |
259 if (!try--) { | |
260 errno = ELOOP; | |
261 goto error; | |
262 } | |
263 | |
264 // Removable path componenents. | |
265 if (!strcmp(new->str, ".") || !strcmp(new->str, "..")) { | |
266 int x = new->str[1]; | |
267 | |
268 free(new); | |
269 if (x) { | |
270 if (done) free(llist_pop(&done)); | |
271 len = 0; | |
272 } else continue; | |
273 | |
274 // Is this a symlink? | |
275 } else len=readlinkat(dirfd, new->str, buf, 4096); | |
276 | |
277 if (len>4095) goto error; | |
278 if (len<1) { | |
279 int fd; | |
280 char *s = ".."; | |
281 | |
282 // For .. just move dirfd | |
283 if (len) { | |
284 // Not a symlink: add to linked list, move dirfd, fail if error | |
285 if ((exact || todo) && errno != EINVAL) goto error; | |
286 new->next = done; | |
287 done = new; | |
288 if (errno == EINVAL && !todo) break; | |
289 s = new->str; | |
290 } | |
291 fd = openat(dirfd, s, 0); | |
292 if (fd == -1 && (exact || todo || errno != ENOENT)) goto error; | |
293 close(dirfd); | |
294 dirfd = fd; | |
295 continue; | |
296 } | |
297 | |
298 // If this symlink is to an absolute path, discard existing resolved path | |
299 buf[len] = 0; | |
300 if (*buf == '/') { | |
301 llist_traverse(done, free); | |
302 done=0; | |
303 close(dirfd); | |
304 dirfd = open("/", 0); | |
305 } | |
306 free(new); | |
307 | |
308 // prepend components of new path. Note symlink to "/" will leave new NULL | |
309 tail = splitpath(buf, &new); | |
310 | |
311 // symlink to "/" will return null and leave tail alone | |
312 if (new) { | |
313 *tail = todo; | |
314 todo = new; | |
315 } | |
316 } | |
317 close(dirfd); | |
318 | |
319 // At this point done has the path, in reverse order. Reverse list while | |
320 // calculating buffer length. | |
321 | |
322 try = 2; | |
323 while (done) { | |
324 struct string_list *temp = llist_pop(&done);; | |
325 | |
326 if (todo) try++; | |
327 try += strlen(temp->str); | |
328 temp->next = todo; | |
329 todo = temp; | |
330 } | |
331 | |
332 // Assemble return buffer | |
333 | |
334 ret = xmalloc(try); | |
335 *ret = '/'; | |
336 ret [try = 1] = 0; | |
337 while (todo) { | |
338 if (try>1) ret[try++] = '/'; | |
339 try = stpcpy(ret+try, todo->str) - ret; | |
340 free(llist_pop(&todo)); | |
341 } | |
342 | |
343 return ret; | |
344 | |
345 error: | |
346 close(dirfd); | |
347 llist_traverse(todo, free); | |
348 llist_traverse(done, free); | |
349 | |
350 return NULL; | |
351 } | |
352 | |
353 // Resolve all symlinks, returning malloc() memory. | |
354 char *xrealpath(char *path) | |
355 { | |
356 char *new = realpath(path, NULL); | |
357 if (!new) perror_exit("realpath '%s'", path); | |
358 return new; | |
359 } | |
360 | |
361 void xchdir(char *path) | |
362 { | |
363 if (chdir(path)) error_exit("chdir '%s'", path); | |
364 } | |
365 | |
1156
faf7117c4489
Fix some issues raised (albeit indirectly) by Isaac Dunham.
Rob Landley <rob@landley.net>
parents:
1130
diff
changeset
|
366 void xchroot(char *path) |
faf7117c4489
Fix some issues raised (albeit indirectly) by Isaac Dunham.
Rob Landley <rob@landley.net>
parents:
1130
diff
changeset
|
367 { |
faf7117c4489
Fix some issues raised (albeit indirectly) by Isaac Dunham.
Rob Landley <rob@landley.net>
parents:
1130
diff
changeset
|
368 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
|
369 xchdir("/"); |
faf7117c4489
Fix some issues raised (albeit indirectly) by Isaac Dunham.
Rob Landley <rob@landley.net>
parents:
1130
diff
changeset
|
370 } |
faf7117c4489
Fix some issues raised (albeit indirectly) by Isaac Dunham.
Rob Landley <rob@landley.net>
parents:
1130
diff
changeset
|
371 |
951 | 372 // Ensure entire path exists. |
373 // If mode != -1 set permissions on newly created dirs. | |
374 // Requires that path string be writable (for temporary null terminators). | |
375 void xmkpath(char *path, int mode) | |
376 { | |
377 char *p, old; | |
378 mode_t mask; | |
379 int rc; | |
380 struct stat st; | |
381 | |
382 for (p = path; ; p++) { | |
383 if (!*p || *p == '/') { | |
384 old = *p; | |
385 *p = rc = 0; | |
386 if (stat(path, &st) || !S_ISDIR(st.st_mode)) { | |
387 if (mode != -1) { | |
388 mask=umask(0); | |
389 rc = mkdir(path, mode); | |
390 umask(mask); | |
391 } else rc = mkdir(path, 0777); | |
392 } | |
393 *p = old; | |
394 if(rc) perror_exit("mkpath '%s'", path); | |
395 } | |
396 if (!*p) break; | |
397 } | |
398 } | |
399 | |
1129
c644f85444d0
Move xgetpwuid() and xgetgrgid() into xwrap.c
Rob Landley <rob@landley.net>
parents:
1105
diff
changeset
|
400 struct passwd *xgetpwuid(uid_t uid) |
c644f85444d0
Move xgetpwuid() and xgetgrgid() into xwrap.c
Rob Landley <rob@landley.net>
parents:
1105
diff
changeset
|
401 { |
c644f85444d0
Move xgetpwuid() and xgetgrgid() into xwrap.c
Rob Landley <rob@landley.net>
parents:
1105
diff
changeset
|
402 struct passwd *pwd = getpwuid(uid); |
1130
6df194c6de88
Add xgetpwnam() to lib/xwrap.c.
Rob Landley <rob@landley.net>
parents:
1129
diff
changeset
|
403 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
|
404 return pwd; |
c644f85444d0
Move xgetpwuid() and xgetgrgid() into xwrap.c
Rob Landley <rob@landley.net>
parents:
1105
diff
changeset
|
405 } |
c644f85444d0
Move xgetpwuid() and xgetgrgid() into xwrap.c
Rob Landley <rob@landley.net>
parents:
1105
diff
changeset
|
406 |
c644f85444d0
Move xgetpwuid() and xgetgrgid() into xwrap.c
Rob Landley <rob@landley.net>
parents:
1105
diff
changeset
|
407 struct group *xgetgrgid(gid_t gid) |
c644f85444d0
Move xgetpwuid() and xgetgrgid() into xwrap.c
Rob Landley <rob@landley.net>
parents:
1105
diff
changeset
|
408 { |
c644f85444d0
Move xgetpwuid() and xgetgrgid() into xwrap.c
Rob Landley <rob@landley.net>
parents:
1105
diff
changeset
|
409 struct group *group = getgrgid(gid); |
1130
6df194c6de88
Add xgetpwnam() to lib/xwrap.c.
Rob Landley <rob@landley.net>
parents:
1129
diff
changeset
|
410 if (!group) error_exit("bad gid %ld", (long)gid); |
1129
c644f85444d0
Move xgetpwuid() and xgetgrgid() into xwrap.c
Rob Landley <rob@landley.net>
parents:
1105
diff
changeset
|
411 return group; |
c644f85444d0
Move xgetpwuid() and xgetgrgid() into xwrap.c
Rob Landley <rob@landley.net>
parents:
1105
diff
changeset
|
412 } |
c644f85444d0
Move xgetpwuid() and xgetgrgid() into xwrap.c
Rob Landley <rob@landley.net>
parents:
1105
diff
changeset
|
413 |
1130
6df194c6de88
Add xgetpwnam() to lib/xwrap.c.
Rob Landley <rob@landley.net>
parents:
1129
diff
changeset
|
414 struct passwd *xgetpwnam(char *name) |
6df194c6de88
Add xgetpwnam() to lib/xwrap.c.
Rob Landley <rob@landley.net>
parents:
1129
diff
changeset
|
415 { |
6df194c6de88
Add xgetpwnam() to lib/xwrap.c.
Rob Landley <rob@landley.net>
parents:
1129
diff
changeset
|
416 struct passwd *up = getpwnam(name); |
6df194c6de88
Add xgetpwnam() to lib/xwrap.c.
Rob Landley <rob@landley.net>
parents:
1129
diff
changeset
|
417 if (!up) error_exit("bad user '%s'", name); |
6df194c6de88
Add xgetpwnam() to lib/xwrap.c.
Rob Landley <rob@landley.net>
parents:
1129
diff
changeset
|
418 return up; |
6df194c6de88
Add xgetpwnam() to lib/xwrap.c.
Rob Landley <rob@landley.net>
parents:
1129
diff
changeset
|
419 } |
6df194c6de88
Add xgetpwnam() to lib/xwrap.c.
Rob Landley <rob@landley.net>
parents:
1129
diff
changeset
|
420 |
1156
faf7117c4489
Fix some issues raised (albeit indirectly) by Isaac Dunham.
Rob Landley <rob@landley.net>
parents:
1130
diff
changeset
|
421 // 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
|
422 // 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
|
423 |
faf7117c4489
Fix some issues raised (albeit indirectly) by Isaac Dunham.
Rob Landley <rob@landley.net>
parents:
1130
diff
changeset
|
424 void xsetuser(struct passwd *pwd) |
faf7117c4489
Fix some issues raised (albeit indirectly) by Isaac Dunham.
Rob Landley <rob@landley.net>
parents:
1130
diff
changeset
|
425 { |
faf7117c4489
Fix some issues raised (albeit indirectly) by Isaac Dunham.
Rob Landley <rob@landley.net>
parents:
1130
diff
changeset
|
426 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
|
427 || 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
|
428 } |
faf7117c4489
Fix some issues raised (albeit indirectly) by Isaac Dunham.
Rob Landley <rob@landley.net>
parents:
1130
diff
changeset
|
429 |
951 | 430 // This can return null (meaning file not found). It just won't return null |
431 // for memory allocation reasons. | |
432 char *xreadlink(char *name) | |
433 { | |
434 int len, size = 0; | |
435 char *buf = 0; | |
436 | |
437 // Grow by 64 byte chunks until it's big enough. | |
438 for(;;) { | |
439 size +=64; | |
440 buf = xrealloc(buf, size); | |
441 len = readlink(name, buf, size); | |
442 | |
443 if (len<0) { | |
444 free(buf); | |
445 return 0; | |
446 } | |
447 if (len<size) { | |
448 buf[len]=0; | |
449 return buf; | |
450 } | |
451 } | |
452 } | |
453 | |
1170
8afe1fde9314
Pass through all the readfile() arguments from xreadfile().
Rob Landley <rob@landley.net>
parents:
1156
diff
changeset
|
454 char *xreadfile(char *name, char *buf, off_t len) |
951 | 455 { |
1170
8afe1fde9314
Pass through all the readfile() arguments from xreadfile().
Rob Landley <rob@landley.net>
parents:
1156
diff
changeset
|
456 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
|
457 |
951 | 458 return buf; |
459 } | |
460 | |
461 int xioctl(int fd, int request, void *data) | |
462 { | |
463 int rc; | |
464 | |
465 errno = 0; | |
466 rc = ioctl(fd, request, data); | |
467 if (rc == -1 && errno) perror_exit("ioctl %x", request); | |
468 | |
469 return rc; | |
470 } | |
471 | |
472 // Open a /var/run/NAME.pid file, dying if we can't write it or if it currently | |
473 // exists and is this executable. | |
474 void xpidfile(char *name) | |
475 { | |
476 char pidfile[256], spid[32]; | |
477 int i, fd; | |
478 pid_t pid; | |
479 | |
480 sprintf(pidfile, "/var/run/%s.pid", name); | |
481 // Try three times to open the sucker. | |
482 for (i=0; i<3; i++) { | |
1028 | 483 fd = open(pidfile, O_CREAT|O_EXCL|O_WRONLY, 0644); |
951 | 484 if (fd != -1) break; |
485 | |
486 // If it already existed, read it. Loop for race condition. | |
487 fd = open(pidfile, O_RDONLY); | |
488 if (fd == -1) continue; | |
489 | |
490 // Is the old program still there? | |
491 spid[xread(fd, spid, sizeof(spid)-1)] = 0; | |
492 close(fd); | |
493 pid = atoi(spid); | |
1053
501fd74c028e
Fix for xpidfile spotted by Felix Janda.
Rob Landley <rob@landley.net>
parents:
1043
diff
changeset
|
494 if (pid < 1 || (kill(pid, 0) && errno == ESRCH)) unlink(pidfile); |
951 | 495 |
496 // An else with more sanity checking might be nice here. | |
497 } | |
498 | |
499 if (i == 3) error_exit("xpidfile %s", name); | |
500 | |
501 xwrite(fd, spid, sprintf(spid, "%ld\n", (long)getpid())); | |
502 close(fd); | |
503 } | |
504 | |
505 // Copy the rest of in to out and close both files. | |
506 | |
507 void xsendfile(int in, int out) | |
508 { | |
509 long len; | |
510 char buf[4096]; | |
511 | |
512 if (in<0) return; | |
513 for (;;) { | |
514 len = xread(in, buf, 4096); | |
515 if (len<1) break; | |
516 xwrite(out, buf, len); | |
517 } | |
518 } | |
952
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
519 |
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
520 // 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
|
521 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
|
522 { |
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
523 double d; |
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
524 long l; |
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
525 |
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
526 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
|
527 else l = strtoul(arg, &arg, 10); |
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
528 |
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
529 // Parse suffix |
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
530 if (*arg) { |
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
531 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
|
532 |
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
533 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
|
534 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
|
535 else l *= ismhd[i]; |
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
536 } |
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
537 |
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
538 if (CFG_TOYBOX_FLOAT) { |
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
539 l = (long)d; |
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
540 if (fraction) *fraction = units*(d-l); |
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
541 } else if (fraction) *fraction = 0; |
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
542 |
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
543 return l; |
ce0519f6457c
Add timeout, factoring out common code from sleep.
Rob Landley <rob@landley.net>
parents:
951
diff
changeset
|
544 } |