Mercurial > hg > toybox
annotate lib/xwrap.c @ 1183:0752b2d58909 draft
Rename xmsprintf() to just xmprintf().
Partly because there's no supplied target string ala sprintf, and partly
because I can never remember what order the m and s go in.
| author | Rob Landley <rob@landley.net> |
|---|---|
| date | Thu, 16 Jan 2014 09:26:50 -0600 |
| parents | 8afe1fde9314 |
| 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 } |