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 and strncat with size checking. Size is the total space in "dest",
13 // including null terminator. Exit if there's not enough space for the string
14 // (including space for the null terminator), because silently truncating is
15 // still broken behavior. (And leaving the string unterminated is INSANE.)
xstrncpy(char * dest,char * src,size_t size)16 void xstrncpy(char *dest, char *src, size_t size)
17 {
18 if (strlen(src)+1 > size) error_exit("'%s' > %ld bytes", src, (long)size);
19 strcpy(dest, src);
20 }
21
xstrncat(char * dest,char * src,size_t size)22 void xstrncat(char *dest, char *src, size_t size)
23 {
24 long len = strlen(dest);
25
26 if (len+strlen(src)+1 > size)
27 error_exit("'%s%s' > %ld bytes", dest, src, (long)size);
28 strcpy(dest+len, src);
29 }
30
31 // We replaced exit(), _exit(), and atexit() with xexit(), _xexit(), and
32 // sigatexit(). This gives _xexit() the option to siglongjmp(toys.rebound, 1)
33 // instead of exiting, lets xexit() report stdout flush failures to stderr
34 // and change the exit code to indicate error, lets our toys.exit function
35 // change happen for signal exit paths and lets us remove the functions
36 // after we've called them.
37
_xexit(void)38 void _xexit(void)
39 {
40 if (toys.rebound) siglongjmp(*toys.rebound, 1);
41
42 _exit(toys.exitval);
43 }
44
xexit(void)45 void xexit(void)
46 {
47 // Call toys.xexit functions in reverse order added.
48 while (toys.xexit) {
49 struct arg_list *al = llist_pop(&toys.xexit);
50
51 // typecast xexit->arg to a function pointer, then call it using invalid
52 // signal 0 to let signal handlers tell actual signal from regular exit.
53 ((void (*)(int))(al->arg))(0);
54
55 free(al);
56 }
57 xflush(1);
58 _xexit();
59 }
60
xmmap(void * addr,size_t length,int prot,int flags,int fd,off_t off)61 void *xmmap(void *addr, size_t length, int prot, int flags, int fd, off_t off)
62 {
63 void *ret = mmap(addr, length, prot, flags, fd, off);
64 if (ret == MAP_FAILED) perror_exit("mmap");
65 return ret;
66 }
67
68 // Die unless we can allocate memory.
xmalloc(size_t size)69 void *xmalloc(size_t size)
70 {
71 void *ret = malloc(size);
72 if (!ret) error_exit("xmalloc(%ld)", (long)size);
73
74 return ret;
75 }
76
77 // Die unless we can allocate prezeroed memory.
xzalloc(size_t size)78 void *xzalloc(size_t size)
79 {
80 void *ret = xmalloc(size);
81 memset(ret, 0, size);
82 return ret;
83 }
84
85 // Die unless we can change the size of an existing allocation, possibly
86 // moving it. (Notice different arguments from libc function.)
xrealloc(void * ptr,size_t size)87 void *xrealloc(void *ptr, size_t size)
88 {
89 ptr = realloc(ptr, size);
90 if (!ptr) error_exit("xrealloc");
91
92 return ptr;
93 }
94
95 // Die unless we can allocate a copy of this many bytes of string.
xstrndup(char * s,size_t n)96 char *xstrndup(char *s, size_t n)
97 {
98 char *ret = strndup(s, n);
99
100 if (!ret) error_exit("xstrndup");
101
102 return ret;
103 }
104
105 // Die unless we can allocate a copy of this string.
xstrdup(char * s)106 char *xstrdup(char *s)
107 {
108 return xstrndup(s, strlen(s));
109 }
110
xmemdup(void * s,long len)111 void *xmemdup(void *s, long len)
112 {
113 void *ret = xmalloc(len);
114 memcpy(ret, s, len);
115
116 return ret;
117 }
118
119 // Die unless we can allocate enough space to sprintf() into.
xmprintf(char * format,...)120 char *xmprintf(char *format, ...)
121 {
122 va_list va, va2;
123 int len;
124 char *ret;
125
126 va_start(va, format);
127 va_copy(va2, va);
128
129 // How long is it?
130 len = vsnprintf(0, 0, format, va);
131 len++;
132 va_end(va);
133
134 // Allocate and do the sprintf()
135 ret = xmalloc(len);
136 vsnprintf(ret, len, format, va2);
137 va_end(va2);
138
139 return ret;
140 }
141
142 // if !flush just check for error on stdout without flushing
xflush(int flush)143 void xflush(int flush)
144 {
145 if ((flush && fflush(0)) || ferror(stdout))
146 if (!toys.exitval) perror_msg("write");
147 }
148
xprintf(char * format,...)149 void xprintf(char *format, ...)
150 {
151 va_list va;
152 va_start(va, format);
153
154 vprintf(format, va);
155 va_end(va);
156 xflush(0);
157 }
158
159 // Put string with length (does not append newline)
xputsl(char * s,int len)160 void xputsl(char *s, int len)
161 {
162 int out;
163
164 while (len != (out = fwrite(s, 1, len, stdout))) {
165 if (out<1) perror_exit("write");
166 len -= out;
167 s += out;
168 }
169 xflush(0);
170 }
171
172 // xputs with no newline
xputsn(char * s)173 void xputsn(char *s)
174 {
175 xputsl(s, strlen(s));
176 }
177
178 // Write string to stdout with newline, flushing and checking for errors
xputs(char * s)179 void xputs(char *s)
180 {
181 puts(s);
182 xflush(0);
183 }
184
xputc(char c)185 void xputc(char c)
186 {
187 if (EOF == fputc(c, stdout)) perror_exit("write");
188 xflush(0);
189 }
190
191 // This is called through the XVFORK macro because parent/child of vfork
192 // share a stack, so child returning from a function would stomp the return
193 // address parent would need. Solution: make vfork() an argument so processes
194 // diverge before function gets called.
xvforkwrap(pid_t pid)195 pid_t __attribute__((returns_twice)) xvforkwrap(pid_t pid)
196 {
197 if (pid == -1) perror_exit("vfork");
198
199 // Signal to xexec() and friends that we vforked so can't recurse
200 if (!pid) toys.stacktop = 0;
201
202 return pid;
203 }
204
205 // Die unless we can exec argv[] (or run builtin command). Note that anything
206 // with a path isn't a builtin, so /bin/sh won't match the builtin sh.
xexec(char ** argv)207 void xexec(char **argv)
208 {
209 // Only recurse to builtin when we have multiplexer and !vfork context.
210 if (CFG_TOYBOX && !CFG_TOYBOX_NORECURSE && toys.stacktop && **argv != '/')
211 toy_exec(argv);
212 execvp(argv[0], argv);
213
214 toys.exitval = 126+(errno == ENOENT);
215 perror_msg("exec %s", argv[0]);
216 if (!toys.stacktop) _exit(toys.exitval);
217 xexit();
218 }
219
220 // Spawn child process, capturing stdin/stdout.
221 // argv[]: command to exec. If null, child re-runs original program with
222 // toys.stacktop zeroed.
223 // pipes[2]: Filehandle to move to stdin/stdout of new process.
224 // If -1, replace with pipe handle connected to stdin/stdout.
225 // NULL treated as {0, 1}, I.E. leave stdin/stdout as is
226 // return: pid of child process
xpopen_both(char ** argv,int * pipes)227 pid_t xpopen_both(char **argv, int *pipes)
228 {
229 int cestnepasun[4], pid;
230
231 // Make the pipes?
232 memset(cestnepasun, 0, sizeof(cestnepasun));
233 if (pipes) for (pid = 0; pid < 2; pid++) {
234 if (pipes[pid] != -1) continue;
235 if (pipe(cestnepasun+(2*pid))) perror_exit("pipe");
236 }
237
238 if (!(pid = CFG_TOYBOX_FORK ? xfork() : XVFORK())) {
239 // Child process: Dance of the stdin/stdout redirection.
240 // cestnepasun[1]->cestnepasun[0] and cestnepasun[3]->cestnepasun[2]
241 if (pipes) {
242 // if we had no stdin/out, pipe handles could overlap, so test for it
243 // and free up potentially overlapping pipe handles before reuse
244
245 // in child, close read end of output pipe, use write end as new stdout
246 if (cestnepasun[2]) {
247 close(cestnepasun[2]);
248 pipes[1] = cestnepasun[3];
249 }
250
251 // in child, close write end of input pipe, use read end as new stdin
252 if (cestnepasun[1]) {
253 close(cestnepasun[1]);
254 pipes[0] = cestnepasun[0];
255 }
256
257 // If swapping stdin/stdout, dup a filehandle that gets closed before use
258 if (!pipes[1]) pipes[1] = dup(0);
259
260 // Are we redirecting stdin?
261 if (pipes[0]) {
262 dup2(pipes[0], 0);
263 close(pipes[0]);
264 }
265
266 // Are we redirecting stdout?
267 if (pipes[1] != 1) {
268 dup2(pipes[1], 1);
269 close(pipes[1]);
270 }
271 }
272 if (argv) xexec(argv);
273
274 // In fork() case, force recursion because we know it's us.
275 if (CFG_TOYBOX_FORK) {
276 toy_init(toys.which, toys.argv);
277 toys.stacktop = 0;
278 toys.which->toy_main();
279 xexit();
280 // In vfork() case, exec /proc/self/exe with high bit of first letter set
281 // to tell main() we reentered.
282 } else {
283 char *s = "/proc/self/exe";
284
285 // We did a nommu-friendly vfork but must exec to continue.
286 // setting high bit of argv[0][0] to let new process know
287 **toys.argv |= 0x80;
288 execv(s, toys.argv);
289 perror_msg_raw(s);
290
291 _exit(127);
292 }
293 }
294
295 // Parent process: vfork had a shared environment, clean up.
296 if (!CFG_TOYBOX_FORK) **toys.argv &= 0x7f;
297
298 if (pipes) {
299 if (cestnepasun[1]) {
300 pipes[0] = cestnepasun[1];
301 close(cestnepasun[0]);
302 }
303 if (cestnepasun[2]) {
304 pipes[1] = cestnepasun[2];
305 close(cestnepasun[3]);
306 }
307 }
308
309 return pid;
310 }
311
312 // Wait for child process to exit, then return adjusted exit code.
xwaitpid(pid_t pid)313 int xwaitpid(pid_t pid)
314 {
315 int status;
316
317 while (-1 == waitpid(pid, &status, 0) && errno == EINTR);
318
319 return WIFEXITED(status) ? WEXITSTATUS(status) : WTERMSIG(status)+128;
320 }
321
xpclose_both(pid_t pid,int * pipes)322 int xpclose_both(pid_t pid, int *pipes)
323 {
324 if (pipes) {
325 close(pipes[0]);
326 close(pipes[1]);
327 }
328
329 return xwaitpid(pid);
330 }
331
332 // Wrapper to xpopen with a pipe for just one of stdin/stdout
xpopen(char ** argv,int * pipe,int isstdout)333 pid_t xpopen(char **argv, int *pipe, int isstdout)
334 {
335 int pipes[2], pid;
336
337 pipes[0] = isstdout ? 0 : -1;
338 pipes[1] = isstdout ? -1 : 1;
339 pid = xpopen_both(argv, pipes);
340 *pipe = pid ? pipes[!!isstdout] : -1;
341
342 return pid;
343 }
344
xpclose(pid_t pid,int pipe)345 int xpclose(pid_t pid, int pipe)
346 {
347 close(pipe);
348
349 return xpclose_both(pid, 0);
350 }
351
352 // Call xpopen and wait for it to finish, keeping existing stdin/stdout.
xrun(char ** argv)353 int xrun(char **argv)
354 {
355 return xpclose_both(xpopen_both(argv, 0), 0);
356 }
357
xaccess(char * path,int flags)358 void xaccess(char *path, int flags)
359 {
360 if (access(path, flags)) perror_exit("Can't access '%s'", path);
361 }
362
363 // Die unless we can delete a file. (File must exist to be deleted.)
xunlink(char * path)364 void xunlink(char *path)
365 {
366 if (unlink(path)) perror_exit("unlink '%s'", path);
367 }
368
369 // Die unless we can open/create a file, returning file descriptor.
370 // The meaning of O_CLOEXEC is reversed (it defaults on, pass it to disable)
371 // and WARN_ONLY tells us not to exit.
xcreate_stdio(char * path,int flags,int mode)372 int xcreate_stdio(char *path, int flags, int mode)
373 {
374 int fd = open(path, (flags^O_CLOEXEC)&~WARN_ONLY, mode);
375
376 if (fd == -1) ((mode&WARN_ONLY) ? perror_msg_raw : perror_exit_raw)(path);
377 return fd;
378 }
379
380 // Die unless we can open a file, returning file descriptor.
xopen_stdio(char * path,int flags)381 int xopen_stdio(char *path, int flags)
382 {
383 return xcreate_stdio(path, flags, 0);
384 }
385
xpipe(int * pp)386 void xpipe(int *pp)
387 {
388 if (pipe(pp)) perror_exit("xpipe");
389 }
390
xclose(int fd)391 void xclose(int fd)
392 {
393 if (close(fd)) perror_exit("xclose");
394 }
395
xdup(int fd)396 int xdup(int fd)
397 {
398 if (fd != -1) {
399 fd = dup(fd);
400 if (fd == -1) perror_exit("xdup");
401 }
402 return fd;
403 }
404
405 // Move file descriptor above stdin/stdout/stderr, using /dev/null to consume
406 // old one. (We should never be called with stdin/stdout/stderr closed, but...)
notstdio(int fd)407 int notstdio(int fd)
408 {
409 if (fd<0) return fd;
410
411 while (fd<3) {
412 int fd2 = xdup(fd);
413
414 close(fd);
415 xopen_stdio("/dev/null", O_RDWR);
416 fd = fd2;
417 }
418
419 return fd;
420 }
421
xrename(char * from,char * to)422 void xrename(char *from, char *to)
423 {
424 if (rename(from, to)) perror_exit("rename %s -> %s", from, to);
425 }
426
xtempfile(char * name,char ** tempname)427 int xtempfile(char *name, char **tempname)
428 {
429 int fd;
430
431 *tempname = xmprintf("%s%s", name, "XXXXXX");
432 if(-1 == (fd = mkstemp(*tempname))) error_exit("no temp file");
433
434 return fd;
435 }
436
437 // Create a file but don't return stdin/stdout/stderr
xcreate(char * path,int flags,int mode)438 int xcreate(char *path, int flags, int mode)
439 {
440 return notstdio(xcreate_stdio(path, flags, mode));
441 }
442
443 // Open a file descriptor NOT in stdin/stdout/stderr
xopen(char * path,int flags)444 int xopen(char *path, int flags)
445 {
446 return notstdio(xopen_stdio(path, flags));
447 }
448
449 // Open read only, treating "-" as a synonym for stdin, defaulting to warn only
openro(char * path,int flags)450 int openro(char *path, int flags)
451 {
452 if (!strcmp(path, "-")) return 0;
453
454 return xopen(path, flags^WARN_ONLY);
455 }
456
457 // Open read only, treating "-" as a synonym for stdin.
xopenro(char * path)458 int xopenro(char *path)
459 {
460 return openro(path, O_RDONLY|WARN_ONLY);
461 }
462
xfdopen(int fd,char * mode)463 FILE *xfdopen(int fd, char *mode)
464 {
465 FILE *f = fdopen(fd, mode);
466
467 if (!f) perror_exit("xfdopen");
468
469 return f;
470 }
471
472 // Die unless we can open/create a file, returning FILE *.
xfopen(char * path,char * mode)473 FILE *xfopen(char *path, char *mode)
474 {
475 FILE *f = fopen(path, mode);
476 if (!f) perror_exit("No file %s", path);
477 return f;
478 }
479
480 // Die if there's an error other than EOF.
xread(int fd,void * buf,size_t len)481 size_t xread(int fd, void *buf, size_t len)
482 {
483 ssize_t ret = read(fd, buf, len);
484 if (ret < 0) perror_exit("xread");
485
486 return ret;
487 }
488
xreadall(int fd,void * buf,size_t len)489 void xreadall(int fd, void *buf, size_t len)
490 {
491 if (len != readall(fd, buf, len)) perror_exit("xreadall");
492 }
493
494 // There's no xwriteall(), just xwrite(). When we read, there may or may not
495 // be more data waiting. When we write, there is data and it had better go
496 // somewhere.
497
xwrite(int fd,void * buf,size_t len)498 void xwrite(int fd, void *buf, size_t len)
499 {
500 if (len != writeall(fd, buf, len)) perror_exit("xwrite");
501 }
502
503 // Die if lseek fails, probably due to being called on a pipe.
504
xlseek(int fd,off_t offset,int whence)505 off_t xlseek(int fd, off_t offset, int whence)
506 {
507 offset = lseek(fd, offset, whence);
508 if (offset<0) perror_exit("lseek");
509
510 return offset;
511 }
512
xgetcwd(void)513 char *xgetcwd(void)
514 {
515 char *buf = getcwd(NULL, 0);
516 if (!buf) perror_exit("xgetcwd");
517
518 return buf;
519 }
520
xstat(char * path,struct stat * st)521 void xstat(char *path, struct stat *st)
522 {
523 if(stat(path, st)) perror_exit("Can't stat %s", path);
524 }
525
526 // Canonicalize path, even to file with one or more missing components at end.
527 // Returns allocated string for pathname or NULL if doesn't exist
528 // exact = 1 file must exist, 0 dir must exist, -1 show theoretical location
xabspath(char * path,int exact)529 char *xabspath(char *path, int exact)
530 {
531 struct string_list *todo, *done = 0;
532 int try = 9999, dirfd = open("/", O_PATH), missing = 0;
533 char *ret;
534
535 // If this isn't an absolute path, start with cwd.
536 if (*path != '/') {
537 char *temp = xgetcwd();
538
539 splitpath(path, splitpath(temp, &todo));
540 free(temp);
541 } else splitpath(path, &todo);
542
543 // Iterate through path components in todo, prepend processed ones to done.
544 while (todo) {
545 struct string_list *new = llist_pop(&todo), **tail;
546 ssize_t len;
547
548 // Eventually break out of endless loops
549 if (!try--) {
550 errno = ELOOP;
551 goto error;
552 }
553
554 // Removable path componenents.
555 if (!strcmp(new->str, ".") || !strcmp(new->str, "..")) {
556 int x = new->str[1];
557
558 free(new);
559 if (!x) continue;
560 if (done) free(llist_pop(&done));
561 len = 0;
562
563 if (missing) missing--;
564 else {
565 if (-1 == (x = openat(dirfd, "..", O_PATH))) goto error;
566 close(dirfd);
567 dirfd = x;
568 }
569 continue;
570 }
571
572 // Is this a symlink?
573 len = readlinkat(dirfd, new->str, libbuf, sizeof(libbuf));
574 if (len>4095) goto error;
575
576 // Not a symlink: add to linked list, move dirfd, fail if error
577 if (len<1) {
578 int fd;
579
580 new->next = done;
581 done = new;
582 if (errno == EINVAL && !todo) break;
583 if (errno == ENOENT && exact<0) {
584 missing++;
585 continue;
586 }
587 if (errno != EINVAL && (exact || todo)) goto error;
588
589 fd = openat(dirfd, new->str, O_PATH);
590 if (fd == -1 && (exact || todo || errno != ENOENT)) goto error;
591 close(dirfd);
592 dirfd = fd;
593 continue;
594 }
595
596 // If this symlink is to an absolute path, discard existing resolved path
597 libbuf[len] = 0;
598 if (*libbuf == '/') {
599 llist_traverse(done, free);
600 done=0;
601 close(dirfd);
602 dirfd = open("/", O_PATH);
603 }
604 free(new);
605
606 // prepend components of new path. Note symlink to "/" will leave new NULL
607 tail = splitpath(libbuf, &new);
608
609 // symlink to "/" will return null and leave tail alone
610 if (new) {
611 *tail = todo;
612 todo = new;
613 }
614 }
615 close(dirfd);
616
617 // At this point done has the path, in reverse order. Reverse list while
618 // calculating buffer length.
619
620 try = 2;
621 while (done) {
622 struct string_list *temp = llist_pop(&done);
623
624 if (todo) try++;
625 try += strlen(temp->str);
626 temp->next = todo;
627 todo = temp;
628 }
629
630 // Assemble return buffer
631
632 ret = xmalloc(try);
633 *ret = '/';
634 ret [try = 1] = 0;
635 while (todo) {
636 if (try>1) ret[try++] = '/';
637 try = stpcpy(ret+try, todo->str) - ret;
638 free(llist_pop(&todo));
639 }
640
641 return ret;
642
643 error:
644 close(dirfd);
645 llist_traverse(todo, free);
646 llist_traverse(done, free);
647
648 return 0;
649 }
650
xchdir(char * path)651 void xchdir(char *path)
652 {
653 if (chdir(path)) perror_exit("chdir '%s'", path);
654 }
655
xchroot(char * path)656 void xchroot(char *path)
657 {
658 if (chroot(path)) error_exit("chroot '%s'", path);
659 xchdir("/");
660 }
661
xgetpwuid(uid_t uid)662 struct passwd *xgetpwuid(uid_t uid)
663 {
664 struct passwd *pwd = getpwuid(uid);
665 if (!pwd) error_exit("bad uid %ld", (long)uid);
666 return pwd;
667 }
668
xgetgrgid(gid_t gid)669 struct group *xgetgrgid(gid_t gid)
670 {
671 struct group *group = getgrgid(gid);
672
673 if (!group) perror_exit("gid %ld", (long)gid);
674 return group;
675 }
676
xgetuid(char * name)677 unsigned xgetuid(char *name)
678 {
679 struct passwd *up = getpwnam(name);
680 char *s = 0;
681 long uid;
682
683 if (up) return up->pw_uid;
684
685 uid = estrtol(name, &s, 10);
686 if (!errno && s && !*s && uid>=0 && uid<=UINT_MAX) return uid;
687
688 error_exit("bad user '%s'", name);
689 }
690
xgetgid(char * name)691 unsigned xgetgid(char *name)
692 {
693 struct group *gr = getgrnam(name);
694 char *s = 0;
695 long gid;
696
697 if (gr) return gr->gr_gid;
698
699 gid = estrtol(name, &s, 10);
700 if (!errno && s && !*s && gid>=0 && gid<=UINT_MAX) return gid;
701
702 error_exit("bad group '%s'", name);
703 }
704
xgetpwnam(char * name)705 struct passwd *xgetpwnam(char *name)
706 {
707 struct passwd *up = getpwnam(name);
708
709 if (!up) perror_exit("user '%s'", name);
710 return up;
711 }
712
xgetgrnam(char * name)713 struct group *xgetgrnam(char *name)
714 {
715 struct group *gr = getgrnam(name);
716
717 if (!gr) perror_exit("group '%s'", name);
718 return gr;
719 }
720
721 // setuid() can fail (for example, too many processes belonging to that user),
722 // which opens a security hole if the process continues as the original user.
723
xsetuser(struct passwd * pwd)724 void xsetuser(struct passwd *pwd)
725 {
726 if (initgroups(pwd->pw_name, pwd->pw_gid) || setgid(pwd->pw_uid)
727 || setuid(pwd->pw_uid)) perror_exit("xsetuser '%s'", pwd->pw_name);
728 }
729
730 // This can return null (meaning file not found). It just won't return null
731 // for memory allocation reasons.
xreadlink(char * name)732 char *xreadlink(char *name)
733 {
734 int len, size = 0;
735 char *buf = 0;
736
737 // Grow by 64 byte chunks until it's big enough.
738 for(;;) {
739 size +=64;
740 buf = xrealloc(buf, size);
741 len = readlink(name, buf, size);
742
743 if (len<0) {
744 free(buf);
745 return 0;
746 }
747 if (len<size) {
748 buf[len]=0;
749 return buf;
750 }
751 }
752 }
753
xreadfile(char * name,char * buf,off_t len)754 char *xreadfile(char *name, char *buf, off_t len)
755 {
756 if (!(buf = readfile(name, buf, len))) perror_exit("Bad '%s'", name);
757
758 return buf;
759 }
760
761 // The data argument to ioctl() is actually long, but it's usually used as
762 // a pointer. If you need to feed in a number, do (void *)(long) typecast.
xioctl(int fd,int request,void * data)763 int xioctl(int fd, int request, void *data)
764 {
765 int rc;
766
767 errno = 0;
768 rc = ioctl(fd, request, data);
769 if (rc == -1 && errno) perror_exit("ioctl %x", request);
770
771 return rc;
772 }
773
774 // Open a /var/run/NAME.pid file, dying if we can't write it or if it currently
775 // exists and is this executable.
xpidfile(char * name)776 void xpidfile(char *name)
777 {
778 char pidfile[256], spid[32];
779 int i, fd;
780 pid_t pid;
781
782 sprintf(pidfile, "/var/run/%s.pid", name);
783 // Try three times to open the sucker.
784 for (i=0; i<3; i++) {
785 fd = open(pidfile, O_CREAT|O_EXCL|O_WRONLY, 0644);
786 if (fd != -1) break;
787
788 // If it already existed, read it. Loop for race condition.
789 fd = open(pidfile, O_RDONLY);
790 if (fd == -1) continue;
791
792 // Is the old program still there?
793 spid[xread(fd, spid, sizeof(spid)-1)] = 0;
794 close(fd);
795 pid = atoi(spid);
796 if (pid < 1 || (kill(pid, 0) && errno == ESRCH)) unlink(pidfile);
797
798 // An else with more sanity checking might be nice here.
799 }
800
801 if (i == 3) error_exit("xpidfile %s", name);
802
803 xwrite(fd, spid, sprintf(spid, "%ld\n", (long)getpid()));
804 close(fd);
805 }
806
807 // Return bytes copied from in to out. If bytes <0 copy all of in to out.
808 // If consuemd isn't null, amount read saved there (return is written or error)
sendfile_len(int in,int out,long long bytes,long long * consumed)809 long long sendfile_len(int in, int out, long long bytes, long long *consumed)
810 {
811 long long total = 0, len;
812
813 if (consumed) *consumed = 0;
814 if (in<0) return 0;
815 while (bytes != total) {
816 len = bytes-total;
817 if (bytes<0 || len>sizeof(libbuf)) len = sizeof(libbuf);
818
819 len = read(in, libbuf, len);
820 if (!len && errno==EAGAIN) continue;
821 if (len<1) break;
822 if (consumed) *consumed += len;
823 if (writeall(out, libbuf, len) != len) return -1;
824 total += len;
825 }
826
827 return total;
828 }
829
830 // error_exit if we couldn't copy all bytes
xsendfile_len(int in,int out,long long bytes)831 long long xsendfile_len(int in, int out, long long bytes)
832 {
833 long long len = sendfile_len(in, out, bytes, 0);
834
835 if (bytes != -1 && bytes != len) {
836 if (out == 1 && len<0) xexit();
837 error_exit("short %s", (len<0) ? "write" : "read");
838 }
839
840 return len;
841 }
842
843 // warn and pad with zeroes if we couldn't copy all bytes
xsendfile_pad(int in,int out,long long len)844 void xsendfile_pad(int in, int out, long long len)
845 {
846 len -= xsendfile_len(in, out, len);
847 if (len) {
848 perror_msg("short read");
849 memset(libbuf, 0, sizeof(libbuf));
850 while (len) {
851 int i = len>sizeof(libbuf) ? sizeof(libbuf) : len;
852
853 xwrite(out, libbuf, i);
854 len -= i;
855 }
856 }
857 }
858
859 // copy all of in to out
xsendfile(int in,int out)860 long long xsendfile(int in, int out)
861 {
862 return xsendfile_len(in, out, -1);
863 }
864
xstrtod(char * s)865 double xstrtod(char *s)
866 {
867 char *end;
868 double d;
869
870 errno = 0;
871 d = strtod(s, &end);
872 if (!errno && *end) errno = E2BIG;
873 if (errno) perror_exit("strtod %s", s);
874
875 return d;
876 }
877
878 // parse fractional seconds with optional s/m/h/d suffix
xparsetime(char * arg,long zeroes,long * fraction)879 long xparsetime(char *arg, long zeroes, long *fraction)
880 {
881 long l, fr = 0, mask = 1;
882 char *end;
883
884 if (*arg != '.' && !isdigit(*arg)) error_exit("Not a number '%s'", arg);
885 l = strtoul(arg, &end, 10);
886 if (*end == '.') {
887 end++;
888 while (zeroes--) {
889 fr *= 10;
890 mask *= 10;
891 if (isdigit(*end)) fr += *end++-'0';
892 }
893 while (isdigit(*end)) end++;
894 }
895
896 // Parse suffix
897 if (*end) {
898 int ismhd[]={1,60,3600,86400}, i = stridx("smhd", *end);
899
900 if (i == -1 || *(end+1)) error_exit("Unknown suffix '%s'", end);
901 l *= ismhd[i];
902 fr *= ismhd[i];
903 l += fr/mask;
904 fr %= mask;
905 }
906 if (fraction) *fraction = fr;
907
908 return l;
909 }
910
xparsemillitime(char * arg)911 long long xparsemillitime(char *arg)
912 {
913 long l, ll;
914
915 l = xparsetime(arg, 3, &ll);
916
917 return (l*1000LL)+ll;
918 }
919
920
921
922 // Compile a regular expression into a regex_t
xregcomp(regex_t * preg,char * regex,int cflags)923 void xregcomp(regex_t *preg, char *regex, int cflags)
924 {
925 int rc;
926
927 // BSD regex implementations don't support the empty regex (which isn't
928 // allowed in the POSIX grammar), but glibc does. Fake it for BSD.
929 if (!*regex) {
930 regex = "()";
931 cflags |= REG_EXTENDED;
932 }
933
934 if ((rc = regcomp(preg, regex, cflags))) {
935 regerror(rc, preg, libbuf, sizeof(libbuf));
936 error_exit("bad regex: %s", libbuf);
937 }
938 }
939
xtzset(char * new)940 char *xtzset(char *new)
941 {
942 char *old = getenv("TZ");
943
944 if (old) old = xstrdup(old);
945 if (new ? setenv("TZ", new, 1) : unsetenv("TZ")) perror_exit("setenv");
946 tzset();
947
948 return old;
949 }
950
951 // Set a signal handler
xsignal_flags(int signal,void * handler,int flags)952 void xsignal_flags(int signal, void *handler, int flags)
953 {
954 struct sigaction *sa = (void *)libbuf;
955
956 memset(sa, 0, sizeof(struct sigaction));
957 sa->sa_handler = handler;
958 sa->sa_flags = flags;
959
960 if (sigaction(signal, sa, 0)) perror_exit("xsignal %d", signal);
961 }
962
xsignal(int signal,void * handler)963 void xsignal(int signal, void *handler)
964 {
965 xsignal_flags(signal, handler, 0);
966 }
967
968
xvali_date(struct tm * tm,char * str)969 time_t xvali_date(struct tm *tm, char *str)
970 {
971 time_t t;
972
973 if (tm && (unsigned)tm->tm_sec<=60 && (unsigned)tm->tm_min<=59
974 && (unsigned)tm->tm_hour<=23 && tm->tm_mday && (unsigned)tm->tm_mday<=31
975 && (unsigned)tm->tm_mon<=11 && (t = mktime(tm)) != -1) return t;
976
977 error_exit("bad date %s", str);
978 }
979
980 // Parse date string (relative to current *t). Sets time_t and nanoseconds.
xparsedate(char * str,time_t * t,unsigned * nano,int endian)981 void xparsedate(char *str, time_t *t, unsigned *nano, int endian)
982 {
983 struct tm tm;
984 time_t now = *t;
985 int len = 0, i = 0;
986 // Formats with seconds come first. Posix can't agree on whether 12 digits
987 // has year before (touch -t) or year after (date), so support both.
988 char *s = str, *p, *oldtz = 0, *formats[] = {"%Y-%m-%d %T", "%Y-%m-%dT%T",
989 "%H:%M:%S", "%Y-%m-%d %H:%M", "%Y-%m-%d", "%H:%M", "%m%d%H%M",
990 endian ? "%m%d%H%M%y" : "%y%m%d%H%M",
991 endian ? "%m%d%H%M%C%y" : "%C%y%m%d%H%M"};
992
993 *nano = 0;
994
995 // Parse @UNIXTIME[.FRACTION]
996 if (*str == '@') {
997 long long ll;
998
999 // Collect seconds and nanoseconds.
1000 // &ll is not just t because we can't guarantee time_t is 64 bit (yet).
1001 sscanf(s, "@%lld%n", &ll, &len);
1002 if (s[len]=='.') {
1003 s += len+1;
1004 for (len = 0; len<9; len++) {
1005 *nano *= 10;
1006 if (isdigit(*s)) *nano += *s++-'0';
1007 }
1008 }
1009 *t = ll;
1010 if (!s[len]) return;
1011 xvali_date(0, str);
1012 }
1013
1014 // Trailing Z means UTC timezone, don't expect libc to know this.
1015 // (Trimming it off here means it won't show up in error messages.)
1016 if ((i = strlen(str)) && toupper(str[i-1])=='Z') {
1017 str[--i] = 0;
1018 oldtz = getenv("TZ");
1019 if (oldtz) oldtz = xstrdup(oldtz);
1020 setenv("TZ", "UTC0", 1);
1021 }
1022
1023 // Try each format
1024 for (i = 0; i<ARRAY_LEN(formats); i++) {
1025 localtime_r(&now, &tm);
1026 tm.tm_hour = tm.tm_min = tm.tm_sec = 0;
1027 tm.tm_isdst = -endian;
1028
1029 if ((p = strptime(s, formats[i], &tm))) {
1030 if (*p == '.') {
1031 p++;
1032 // If format didn't already specify seconds, grab seconds
1033 if (i>2) {
1034 len = 0;
1035 sscanf(p, "%2u%n", &tm.tm_sec, &len);
1036 p += len;
1037 }
1038 // nanoseconds
1039 for (len = 0; len<9; len++) {
1040 *nano *= 10;
1041 if (isdigit(*p)) *nano += *p++-'0';
1042 }
1043 }
1044
1045 if (!*p) break;
1046 }
1047 }
1048
1049 // Sanity check field ranges
1050 *t = xvali_date((i!=ARRAY_LEN(formats)) ? &tm : 0, str);
1051
1052 if (oldtz) setenv("TZ", oldtz, 1);
1053 free(oldtz);
1054 }
1055
xgetline(FILE * fp,int * len)1056 char *xgetline(FILE *fp, int *len)
1057 {
1058 char *new = 0;
1059 size_t linelen = 0;
1060
1061 errno = 0;
1062 if (1>(linelen = getline(&new, &linelen, fp))) {
1063 if (errno) perror_msg("getline");
1064 new = 0;
1065 } else if (new[linelen-1] == '\n') new[--linelen] = 0;
1066 if (len) *len = linelen;
1067
1068 return new;
1069 }
1070