Rob Landley | d390493 | 2013-07-16 00:04:56 -0500 | [diff] [blame] | 1 | /* lib.c - various reusable stuff. |
landley | 09ea7ac | 2006-10-30 01:38:00 -0500 | [diff] [blame] | 2 | * |
| 3 | * Copyright 2006 Rob Landley <rob@landley.net> |
landley | 4f344e3 | 2006-10-05 16:18:03 -0400 | [diff] [blame] | 4 | */ |
| 5 | |
| 6 | #include "toys.h" |
| 7 | |
landley | 09ea7ac | 2006-10-30 01:38:00 -0500 | [diff] [blame] | 8 | void verror_msg(char *msg, int err, va_list va) |
| 9 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 10 | char *s = ": %s"; |
Rob Landley | 12138e4 | 2008-01-27 15:26:08 -0600 | [diff] [blame] | 11 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 12 | fprintf(stderr, "%s: ", toys.which->name); |
| 13 | if (msg) vfprintf(stderr, msg, va); |
| 14 | else s+=2; |
| 15 | if (err) fprintf(stderr, s, strerror(err)); |
| 16 | putc('\n', stderr); |
Rob Landley | 662a267 | 2013-01-02 02:00:35 -0600 | [diff] [blame] | 17 | if (!toys.exitval) toys.exitval++; |
landley | 09ea7ac | 2006-10-30 01:38:00 -0500 | [diff] [blame] | 18 | } |
| 19 | |
| 20 | void error_msg(char *msg, ...) |
| 21 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 22 | va_list va; |
landley | 09ea7ac | 2006-10-30 01:38:00 -0500 | [diff] [blame] | 23 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 24 | va_start(va, msg); |
| 25 | verror_msg(msg, 0, va); |
| 26 | va_end(va); |
landley | 09ea7ac | 2006-10-30 01:38:00 -0500 | [diff] [blame] | 27 | } |
| 28 | |
| 29 | void perror_msg(char *msg, ...) |
| 30 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 31 | va_list va; |
landley | 09ea7ac | 2006-10-30 01:38:00 -0500 | [diff] [blame] | 32 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 33 | va_start(va, msg); |
| 34 | verror_msg(msg, errno, va); |
| 35 | va_end(va); |
landley | 09ea7ac | 2006-10-30 01:38:00 -0500 | [diff] [blame] | 36 | } |
| 37 | |
landley | 4f344e3 | 2006-10-05 16:18:03 -0400 | [diff] [blame] | 38 | // Die with an error message. |
| 39 | void error_exit(char *msg, ...) |
| 40 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 41 | va_list va; |
landley | 4f344e3 | 2006-10-05 16:18:03 -0400 | [diff] [blame] | 42 | |
Rob Landley | 36ffc5a | 2013-04-14 21:43:22 -0500 | [diff] [blame] | 43 | if (CFG_TOYBOX_HELP && toys.exithelp) show_help(); |
Rob Landley | d06c58d | 2007-10-11 15:36:36 -0500 | [diff] [blame] | 44 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 45 | va_start(va, msg); |
| 46 | verror_msg(msg, 0, va); |
| 47 | va_end(va); |
landley | 09ea7ac | 2006-10-30 01:38:00 -0500 | [diff] [blame] | 48 | |
Rob Landley | 5583030 | 2013-06-16 19:59:51 -0500 | [diff] [blame] | 49 | xexit(); |
landley | 09ea7ac | 2006-10-30 01:38:00 -0500 | [diff] [blame] | 50 | } |
| 51 | |
| 52 | // Die with an error message and strerror(errno) |
| 53 | void perror_exit(char *msg, ...) |
| 54 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 55 | va_list va; |
landley | 09ea7ac | 2006-10-30 01:38:00 -0500 | [diff] [blame] | 56 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 57 | va_start(va, msg); |
| 58 | verror_msg(msg, errno, va); |
| 59 | va_end(va); |
landley | 09ea7ac | 2006-10-30 01:38:00 -0500 | [diff] [blame] | 60 | |
Rob Landley | 5583030 | 2013-06-16 19:59:51 -0500 | [diff] [blame] | 61 | xexit(); |
landley | 4f344e3 | 2006-10-05 16:18:03 -0400 | [diff] [blame] | 62 | } |
| 63 | |
landley | 64b2e23 | 2006-10-30 10:01:19 -0500 | [diff] [blame] | 64 | // Keep reading until full or EOF |
Rob Landley | 9016377 | 2007-01-18 21:54:08 -0500 | [diff] [blame] | 65 | ssize_t readall(int fd, void *buf, size_t len) |
landley | 64b2e23 | 2006-10-30 10:01:19 -0500 | [diff] [blame] | 66 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 67 | size_t count = 0; |
Rob Landley | 15b2315 | 2008-07-18 04:15:59 -0500 | [diff] [blame] | 68 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 69 | while (count<len) { |
Rob Landley | dbbd3d6 | 2013-12-08 13:26:05 -0600 | [diff] [blame] | 70 | int i = read(fd, (char *)buf+count, len-count); |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 71 | if (!i) break; |
| 72 | if (i<0) return i; |
| 73 | count += i; |
| 74 | } |
landley | 64b2e23 | 2006-10-30 10:01:19 -0500 | [diff] [blame] | 75 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 76 | return count; |
landley | 64b2e23 | 2006-10-30 10:01:19 -0500 | [diff] [blame] | 77 | } |
| 78 | |
Rob Landley | f3e452a | 2007-01-08 02:49:39 -0500 | [diff] [blame] | 79 | // Keep writing until done or EOF |
Rob Landley | 9016377 | 2007-01-18 21:54:08 -0500 | [diff] [blame] | 80 | ssize_t writeall(int fd, void *buf, size_t len) |
Rob Landley | f3e452a | 2007-01-08 02:49:39 -0500 | [diff] [blame] | 81 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 82 | size_t count = 0; |
| 83 | while (count<len) { |
Rob Landley | 1fb3ae7 | 2014-02-16 11:09:23 -0600 | [diff] [blame] | 84 | int i = write(fd, count+(char *)buf, len-count); |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 85 | if (i<1) return i; |
| 86 | count += i; |
| 87 | } |
Rob Landley | f3e452a | 2007-01-08 02:49:39 -0500 | [diff] [blame] | 88 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 89 | return count; |
Rob Landley | f3e452a | 2007-01-08 02:49:39 -0500 | [diff] [blame] | 90 | } |
| 91 | |
Rob Landley | bc382be | 2013-09-16 23:41:51 -0500 | [diff] [blame] | 92 | // skip this many bytes of input. Return 0 for success, >0 means this much |
| 93 | // left after input skipped. |
Rob Landley | 2037b83 | 2012-07-15 16:56:20 -0500 | [diff] [blame] | 94 | off_t lskip(int fd, off_t offset) |
| 95 | { |
Rob Landley | bc382be | 2013-09-16 23:41:51 -0500 | [diff] [blame] | 96 | off_t cur = lseek(fd, 0, SEEK_CUR); |
Rob Landley | 2037b83 | 2012-07-15 16:56:20 -0500 | [diff] [blame] | 97 | |
Rob Landley | bc382be | 2013-09-16 23:41:51 -0500 | [diff] [blame] | 98 | if (cur != -1) { |
| 99 | off_t end = lseek(fd, 0, SEEK_END) - cur; |
Rob Landley | 2037b83 | 2012-07-15 16:56:20 -0500 | [diff] [blame] | 100 | |
Rob Landley | bc382be | 2013-09-16 23:41:51 -0500 | [diff] [blame] | 101 | if (end > 0 && end < offset) return offset - end; |
| 102 | end = offset+cur; |
| 103 | if (end == lseek(fd, end, SEEK_SET)) return 0; |
| 104 | perror_exit("lseek"); |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 105 | } |
Rob Landley | bc382be | 2013-09-16 23:41:51 -0500 | [diff] [blame] | 106 | |
| 107 | while (offset>0) { |
| 108 | int try = offset>sizeof(libbuf) ? sizeof(libbuf) : offset, or; |
| 109 | |
| 110 | or = readall(fd, libbuf, try); |
| 111 | if (or < 0) perror_exit("lskip to %lld", (long long)offset); |
Rob Landley | 4dd800c | 2014-02-28 23:10:03 -0600 | [diff] [blame] | 112 | else offset -= or; |
Rob Landley | bc382be | 2013-09-16 23:41:51 -0500 | [diff] [blame] | 113 | if (or < try) break; |
| 114 | } |
| 115 | |
| 116 | return offset; |
Rob Landley | 2037b83 | 2012-07-15 16:56:20 -0500 | [diff] [blame] | 117 | } |
| 118 | |
Rob Landley | ca1b60e | 2014-03-11 20:44:55 -0500 | [diff] [blame] | 119 | // flags: 1=make last dir (with mode lastmode, otherwise skips last component) |
| 120 | // 2=make path (already exists is ok) |
| 121 | // 4=verbose |
| 122 | // returns 0 = path ok, 1 = error |
| 123 | int mkpathat(int atfd, char *dir, mode_t lastmode, int flags) |
| 124 | { |
| 125 | struct stat buf; |
| 126 | char *s; |
| 127 | |
| 128 | // mkdir -p one/two/three is not an error if the path already exists, |
| 129 | // but is if "three" is a file. The others we dereference and catch |
| 130 | // not-a-directory along the way, but the last one we must explicitly |
| 131 | // test for. Might as well do it up front. |
| 132 | |
| 133 | if (!fstatat(atfd, dir, &buf, 0) && !S_ISDIR(buf.st_mode)) { |
| 134 | errno = EEXIST; |
| 135 | return 1; |
| 136 | } |
| 137 | |
Rob Landley | 02f5a30 | 2014-03-24 06:26:49 -0500 | [diff] [blame] | 138 | for (s = dir; ;s++) { |
Rob Landley | ca1b60e | 2014-03-11 20:44:55 -0500 | [diff] [blame] | 139 | char save = 0; |
| 140 | mode_t mode = (0777&~toys.old_umask)|0300; |
| 141 | |
Rob Landley | 02f5a30 | 2014-03-24 06:26:49 -0500 | [diff] [blame] | 142 | // find next '/', but don't try to mkdir "" at start of absolute path |
| 143 | if (*s == '/' && (flags&2) && s != dir) { |
Rob Landley | ca1b60e | 2014-03-11 20:44:55 -0500 | [diff] [blame] | 144 | save = *s; |
| 145 | *s = 0; |
| 146 | } else if (*s) continue; |
| 147 | |
| 148 | // Use the mode from the -m option only for the last directory. |
| 149 | if (!save) { |
| 150 | if (flags&1) mode = lastmode; |
| 151 | else break; |
| 152 | } |
| 153 | |
| 154 | if (mkdirat(atfd, dir, mode)) { |
| 155 | if (!(flags&2) || errno != EEXIST) return 1; |
| 156 | } else if (flags&4) |
| 157 | fprintf(stderr, "%s: created directory '%s'\n", toys.which->name, dir); |
| 158 | |
| 159 | if (!(*s = save)) break; |
| 160 | } |
| 161 | |
| 162 | return 0; |
| 163 | } |
| 164 | |
Rob Landley | fe91e68 | 2012-11-22 21:18:09 -0600 | [diff] [blame] | 165 | // Split a path into linked list of components, tracking head and tail of list. |
| 166 | // Filters out // entries with no contents. |
| 167 | struct string_list **splitpath(char *path, struct string_list **list) |
landley | 00f87f1 | 2006-10-25 18:38:37 -0400 | [diff] [blame] | 168 | { |
Rob Landley | fe91e68 | 2012-11-22 21:18:09 -0600 | [diff] [blame] | 169 | char *new = path; |
landley | 00f87f1 | 2006-10-25 18:38:37 -0400 | [diff] [blame] | 170 | |
Rob Landley | fe91e68 | 2012-11-22 21:18:09 -0600 | [diff] [blame] | 171 | *list = 0; |
| 172 | do { |
| 173 | int len; |
landley | 00f87f1 | 2006-10-25 18:38:37 -0400 | [diff] [blame] | 174 | |
Rob Landley | fe91e68 | 2012-11-22 21:18:09 -0600 | [diff] [blame] | 175 | if (*path && *path != '/') continue; |
| 176 | len = path-new; |
| 177 | if (len > 0) { |
| 178 | *list = xmalloc(sizeof(struct string_list) + len + 1); |
| 179 | (*list)->next = 0; |
| 180 | strncpy((*list)->str, new, len); |
| 181 | (*list)->str[len] = 0; |
| 182 | list = &(*list)->next; |
| 183 | } |
| 184 | new = path+1; |
| 185 | } while (*path++); |
| 186 | |
| 187 | return list; |
| 188 | } |
| 189 | |
Rob Landley | 0a04b3e | 2006-11-03 00:05:52 -0500 | [diff] [blame] | 190 | // Find all file in a colon-separated path with access type "type" (generally |
| 191 | // X_OK or R_OK). Returns a list of absolute paths to each file found, in |
| 192 | // order. |
| 193 | |
| 194 | struct string_list *find_in_path(char *path, char *filename) |
Rob Landley | fa98d01 | 2006-11-02 02:57:27 -0500 | [diff] [blame] | 195 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 196 | struct string_list *rlist = NULL, **prlist=&rlist; |
| 197 | char *cwd = xgetcwd(); |
Rob Landley | fa98d01 | 2006-11-02 02:57:27 -0500 | [diff] [blame] | 198 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 199 | for (;;) { |
| 200 | char *next = path ? strchr(path, ':') : NULL; |
| 201 | int len = next ? next-path : strlen(path); |
| 202 | struct string_list *rnext; |
| 203 | struct stat st; |
Rob Landley | fa98d01 | 2006-11-02 02:57:27 -0500 | [diff] [blame] | 204 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 205 | rnext = xmalloc(sizeof(void *) + strlen(filename) |
| 206 | + (len ? len : strlen(cwd)) + 2); |
| 207 | if (!len) sprintf(rnext->str, "%s/%s", cwd, filename); |
| 208 | else { |
| 209 | char *res = rnext->str; |
| 210 | strncpy(res, path, len); |
| 211 | res += len; |
| 212 | *(res++) = '/'; |
| 213 | strcpy(res, filename); |
| 214 | } |
Rob Landley | fa98d01 | 2006-11-02 02:57:27 -0500 | [diff] [blame] | 215 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 216 | // Confirm it's not a directory. |
| 217 | if (!stat(rnext->str, &st) && S_ISREG(st.st_mode)) { |
| 218 | *prlist = rnext; |
| 219 | rnext->next = NULL; |
| 220 | prlist = &(rnext->next); |
| 221 | } else free(rnext); |
Rob Landley | fa98d01 | 2006-11-02 02:57:27 -0500 | [diff] [blame] | 222 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 223 | if (!next) break; |
| 224 | path += len; |
| 225 | path++; |
| 226 | } |
| 227 | free(cwd); |
landley | 00f87f1 | 2006-10-25 18:38:37 -0400 | [diff] [blame] | 228 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 229 | return rlist; |
landley | 00f87f1 | 2006-10-25 18:38:37 -0400 | [diff] [blame] | 230 | } |
landley | 09ea7ac | 2006-10-30 01:38:00 -0500 | [diff] [blame] | 231 | |
Rob Landley | f575716 | 2007-02-16 21:08:22 -0500 | [diff] [blame] | 232 | // atol() with the kilo/mega/giga/tera/peta/exa extensions. |
| 233 | // (zetta and yotta don't fit in 64 bits.) |
Rob Landley | 5e6dca6 | 2012-02-09 06:09:27 -0600 | [diff] [blame] | 234 | long atolx(char *numstr) |
Rob Landley | f575716 | 2007-02-16 21:08:22 -0500 | [diff] [blame] | 235 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 236 | char *c, *suffixes="bkmgtpe", *end; |
| 237 | long val = strtol(numstr, &c, 0); |
Rob Landley | f575716 | 2007-02-16 21:08:22 -0500 | [diff] [blame] | 238 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 239 | if (*c) { |
| 240 | if (c != numstr && (end = strchr(suffixes, tolower(*c)))) { |
| 241 | int shift = end-suffixes; |
| 242 | if (shift--) val *= 1024L<<(shift*10); |
| 243 | } else { |
| 244 | while (isspace(*c)) c++; |
| 245 | if (*c) error_exit("not integer: %s", numstr); |
| 246 | } |
| 247 | } |
Rob Landley | ad63f4b | 2011-12-12 15:19:52 -0600 | [diff] [blame] | 248 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 249 | return val; |
Rob Landley | f575716 | 2007-02-16 21:08:22 -0500 | [diff] [blame] | 250 | } |
| 251 | |
Rob Landley | b5e7416 | 2013-11-28 21:11:34 -0600 | [diff] [blame] | 252 | long atolx_range(char *numstr, long low, long high) |
| 253 | { |
| 254 | long val = atolx(numstr); |
| 255 | |
| 256 | if (val < low) error_exit("%ld < %ld", val, low); |
| 257 | if (val > high) error_exit("%ld > %ld", val, high); |
| 258 | |
| 259 | return val; |
| 260 | } |
| 261 | |
Rob Landley | eb7ea22 | 2012-04-14 22:30:41 -0500 | [diff] [blame] | 262 | int numlen(long l) |
| 263 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 264 | int len = 0; |
| 265 | while (l) { |
| 266 | l /= 10; |
| 267 | len++; |
| 268 | } |
| 269 | return len; |
Rob Landley | eb7ea22 | 2012-04-14 22:30:41 -0500 | [diff] [blame] | 270 | } |
| 271 | |
Rob Landley | 2037b83 | 2012-07-15 16:56:20 -0500 | [diff] [blame] | 272 | int stridx(char *haystack, char needle) |
| 273 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 274 | char *off; |
Rob Landley | 2037b83 | 2012-07-15 16:56:20 -0500 | [diff] [blame] | 275 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 276 | if (!needle) return -1; |
| 277 | off = strchr(haystack, needle); |
| 278 | if (!off) return -1; |
Rob Landley | 2037b83 | 2012-07-15 16:56:20 -0500 | [diff] [blame] | 279 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 280 | return off-haystack; |
Rob Landley | 2037b83 | 2012-07-15 16:56:20 -0500 | [diff] [blame] | 281 | } |
| 282 | |
Rob Landley | 8115fc1 | 2014-06-09 07:12:49 -0500 | [diff] [blame] | 283 | // If *a starts with b, advance *a past it and return 1, else return 0; |
| 284 | int strstart(char **a, char *b) |
| 285 | { |
| 286 | int len = strlen(b), i = !strncmp(*a, b, len); |
| 287 | |
| 288 | if (i) *a += len; |
| 289 | |
| 290 | return i; |
| 291 | } |
| 292 | |
Rob Landley | 055cfcb | 2007-01-14 20:20:06 -0500 | [diff] [blame] | 293 | // Return how long the file at fd is, if there's any way to determine it. |
| 294 | off_t fdlength(int fd) |
| 295 | { |
Rob Landley | 035f27a | 2013-08-08 02:46:45 -0500 | [diff] [blame] | 296 | struct stat st; |
| 297 | off_t base = 0, range = 1, expand = 1, old; |
| 298 | |
| 299 | if (!fstat(fd, &st) && S_ISREG(st.st_mode)) return st.st_size; |
Rob Landley | 055cfcb | 2007-01-14 20:20:06 -0500 | [diff] [blame] | 300 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 301 | // If the ioctl works for this, return it. |
Rob Landley | 035f27a | 2013-08-08 02:46:45 -0500 | [diff] [blame] | 302 | // TODO: is blocksize still always 512, or do we stat for it? |
| 303 | // unsigned int size; |
| 304 | // if (ioctl(fd, BLKGETSIZE, &size) >= 0) return size*512L; |
Rob Landley | 055cfcb | 2007-01-14 20:20:06 -0500 | [diff] [blame] | 305 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 306 | // If not, do a binary search for the last location we can read. (Some |
| 307 | // block devices don't do BLKGETSIZE right.) This should probably have |
| 308 | // a CONFIG option... |
Rob Landley | 055cfcb | 2007-01-14 20:20:06 -0500 | [diff] [blame] | 309 | |
Rob Landley | 035f27a | 2013-08-08 02:46:45 -0500 | [diff] [blame] | 310 | // If not, do a binary search for the last location we can read. |
| 311 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 312 | old = lseek(fd, 0, SEEK_CUR); |
| 313 | do { |
| 314 | char temp; |
Rob Landley | 035f27a | 2013-08-08 02:46:45 -0500 | [diff] [blame] | 315 | off_t pos = base + range / 2; |
Rob Landley | 055cfcb | 2007-01-14 20:20:06 -0500 | [diff] [blame] | 316 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 317 | if (lseek(fd, pos, 0)>=0 && read(fd, &temp, 1)==1) { |
Rob Landley | 035f27a | 2013-08-08 02:46:45 -0500 | [diff] [blame] | 318 | off_t delta = (pos + 1) - base; |
Rob Landley | 055cfcb | 2007-01-14 20:20:06 -0500 | [diff] [blame] | 319 | |
Rob Landley | 035f27a | 2013-08-08 02:46:45 -0500 | [diff] [blame] | 320 | base += delta; |
| 321 | if (expand) range = (expand <<= 1) - base; |
| 322 | else range -= delta; |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 323 | } else { |
Rob Landley | 035f27a | 2013-08-08 02:46:45 -0500 | [diff] [blame] | 324 | expand = 0; |
| 325 | range = pos - base; |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 326 | } |
Rob Landley | 035f27a | 2013-08-08 02:46:45 -0500 | [diff] [blame] | 327 | } while (range > 0); |
Rob Landley | 055cfcb | 2007-01-14 20:20:06 -0500 | [diff] [blame] | 328 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 329 | lseek(fd, old, SEEK_SET); |
Rob Landley | f15387d | 2008-07-18 05:43:44 -0500 | [diff] [blame] | 330 | |
Rob Landley | 035f27a | 2013-08-08 02:46:45 -0500 | [diff] [blame] | 331 | return base; |
Rob Landley | 055cfcb | 2007-01-14 20:20:06 -0500 | [diff] [blame] | 332 | } |
| 333 | |
Rob Landley | 8fdcfdb | 2013-09-03 17:56:28 -0500 | [diff] [blame] | 334 | // Read contents of file as a single nul-terminated string. |
| 335 | // malloc new one if buf=len=0 |
Ashwini Sharma | 26b2188 | 2014-05-02 06:24:11 -0500 | [diff] [blame] | 336 | char *readfile(char *name, char *ibuf, off_t len) |
Rob Landley | 055cfcb | 2007-01-14 20:20:06 -0500 | [diff] [blame] | 337 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 338 | int fd; |
Ashwini Sharma | 26b2188 | 2014-05-02 06:24:11 -0500 | [diff] [blame] | 339 | char *buf; |
Rob Landley | 055cfcb | 2007-01-14 20:20:06 -0500 | [diff] [blame] | 340 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 341 | fd = open(name, O_RDONLY); |
| 342 | if (fd == -1) return 0; |
Rob Landley | 8fdcfdb | 2013-09-03 17:56:28 -0500 | [diff] [blame] | 343 | |
| 344 | if (len<1) { |
| 345 | len = fdlength(fd); |
| 346 | // proc files don't report a length, so try 1 page minimum. |
| 347 | if (len<4096) len = 4096; |
| 348 | } |
Ashwini Sharma | 26b2188 | 2014-05-02 06:24:11 -0500 | [diff] [blame] | 349 | if (!ibuf) buf = xmalloc(len+1); |
| 350 | else buf = ibuf; |
Rob Landley | 8fdcfdb | 2013-09-03 17:56:28 -0500 | [diff] [blame] | 351 | |
| 352 | len = readall(fd, buf, len-1); |
| 353 | close(fd); |
| 354 | if (len<0) { |
Ashwini Sharma | 26b2188 | 2014-05-02 06:24:11 -0500 | [diff] [blame] | 355 | if (ibuf != buf) free(buf); |
Rob Landley | 8fdcfdb | 2013-09-03 17:56:28 -0500 | [diff] [blame] | 356 | buf = 0; |
| 357 | } else buf[len] = 0; |
Rob Landley | 055cfcb | 2007-01-14 20:20:06 -0500 | [diff] [blame] | 358 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 359 | return buf; |
Rob Landley | 055cfcb | 2007-01-14 20:20:06 -0500 | [diff] [blame] | 360 | } |
| 361 | |
Rob Landley | f015344 | 2013-04-26 02:41:05 -0500 | [diff] [blame] | 362 | // Sleep for this many thousandths of a second |
| 363 | void msleep(long miliseconds) |
| 364 | { |
| 365 | struct timespec ts; |
| 366 | |
| 367 | ts.tv_sec = miliseconds/1000; |
| 368 | ts.tv_nsec = (miliseconds%1000)*1000000; |
| 369 | nanosleep(&ts, &ts); |
| 370 | } |
| 371 | |
Rob Landley | 6b28341 | 2013-06-01 20:41:35 -0500 | [diff] [blame] | 372 | int64_t peek(void *ptr, int size) |
| 373 | { |
| 374 | if (size & 8) { |
Rob Landley | a8b88fe | 2014-03-11 18:11:12 -0500 | [diff] [blame] | 375 | volatile int64_t *p = (int64_t *)ptr; |
Rob Landley | 6b28341 | 2013-06-01 20:41:35 -0500 | [diff] [blame] | 376 | return *p; |
| 377 | } else if (size & 4) { |
Rob Landley | a8b88fe | 2014-03-11 18:11:12 -0500 | [diff] [blame] | 378 | volatile int *p = (int *)ptr; |
Rob Landley | 6b28341 | 2013-06-01 20:41:35 -0500 | [diff] [blame] | 379 | return *p; |
| 380 | } else if (size & 2) { |
Rob Landley | a8b88fe | 2014-03-11 18:11:12 -0500 | [diff] [blame] | 381 | volatile short *p = (short *)ptr; |
Rob Landley | 6b28341 | 2013-06-01 20:41:35 -0500 | [diff] [blame] | 382 | return *p; |
| 383 | } else { |
Rob Landley | a8b88fe | 2014-03-11 18:11:12 -0500 | [diff] [blame] | 384 | volatile char *p = (char *)ptr; |
Rob Landley | 6b28341 | 2013-06-01 20:41:35 -0500 | [diff] [blame] | 385 | return *p; |
| 386 | } |
| 387 | } |
| 388 | |
| 389 | void poke(void *ptr, uint64_t val, int size) |
| 390 | { |
| 391 | if (size & 8) { |
Rob Landley | a8b88fe | 2014-03-11 18:11:12 -0500 | [diff] [blame] | 392 | volatile uint64_t *p = (uint64_t *)ptr; |
Rob Landley | 6b28341 | 2013-06-01 20:41:35 -0500 | [diff] [blame] | 393 | *p = val; |
| 394 | } else if (size & 4) { |
Rob Landley | a8b88fe | 2014-03-11 18:11:12 -0500 | [diff] [blame] | 395 | volatile int *p = (int *)ptr; |
Rob Landley | 6b28341 | 2013-06-01 20:41:35 -0500 | [diff] [blame] | 396 | *p = val; |
| 397 | } else if (size & 2) { |
Rob Landley | a8b88fe | 2014-03-11 18:11:12 -0500 | [diff] [blame] | 398 | volatile short *p = (short *)ptr; |
Rob Landley | 6b28341 | 2013-06-01 20:41:35 -0500 | [diff] [blame] | 399 | *p = val; |
| 400 | } else { |
Rob Landley | a8b88fe | 2014-03-11 18:11:12 -0500 | [diff] [blame] | 401 | volatile char *p = (char *)ptr; |
Rob Landley | 6b28341 | 2013-06-01 20:41:35 -0500 | [diff] [blame] | 402 | *p = val; |
| 403 | } |
| 404 | } |
| 405 | |
Rob Landley | 2bfaaf2 | 2008-07-03 19:19:00 -0500 | [diff] [blame] | 406 | // Iterate through an array of files, opening each one and calling a function |
| 407 | // on that filehandle and name. The special filename "-" means stdin if |
| 408 | // flags is O_RDONLY, stdout otherwise. An empty argument list calls |
| 409 | // function() on just stdin/stdout. |
| 410 | // |
| 411 | // Note: read only filehandles are automatically closed when function() |
| 412 | // returns, but writeable filehandles must be close by function() |
Rob Landley | ad63f4b | 2011-12-12 15:19:52 -0600 | [diff] [blame] | 413 | void loopfiles_rw(char **argv, int flags, int permissions, int failok, |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 414 | void (*function)(int fd, char *name)) |
Rob Landley | 7634b55 | 2007-11-29 17:49:50 -0600 | [diff] [blame] | 415 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 416 | int fd; |
Rob Landley | 7634b55 | 2007-11-29 17:49:50 -0600 | [diff] [blame] | 417 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 418 | // If no arguments, read from stdin. |
| 419 | if (!*argv) function(flags ? 1 : 0, "-"); |
| 420 | else do { |
| 421 | // Filename "-" means read from stdin. |
| 422 | // Inability to open a file prints a warning, but doesn't exit. |
Rob Landley | 7634b55 | 2007-11-29 17:49:50 -0600 | [diff] [blame] | 423 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 424 | if (!strcmp(*argv,"-")) fd=0; |
| 425 | else if (0>(fd = open(*argv, flags, permissions)) && !failok) { |
| 426 | perror_msg("%s", *argv); |
| 427 | toys.exitval = 1; |
| 428 | continue; |
| 429 | } |
| 430 | function(fd, *argv); |
| 431 | if (flags == O_RDONLY) close(fd); |
| 432 | } while (*++argv); |
Rob Landley | 7634b55 | 2007-11-29 17:49:50 -0600 | [diff] [blame] | 433 | } |
Rob Landley | bc07865 | 2007-12-15 21:47:25 -0600 | [diff] [blame] | 434 | |
Rob Landley | ad63f4b | 2011-12-12 15:19:52 -0600 | [diff] [blame] | 435 | // Call loopfiles_rw with O_RDONLY and !failok (common case). |
Rob Landley | 2bfaaf2 | 2008-07-03 19:19:00 -0500 | [diff] [blame] | 436 | void loopfiles(char **argv, void (*function)(int fd, char *name)) |
| 437 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 438 | loopfiles_rw(argv, O_RDONLY, 0, 0, function); |
Rob Landley | 2bfaaf2 | 2008-07-03 19:19:00 -0500 | [diff] [blame] | 439 | } |
| 440 | |
Rob Landley | bc07865 | 2007-12-15 21:47:25 -0600 | [diff] [blame] | 441 | // Slow, but small. |
| 442 | |
Rob Landley | 3fc4e0f | 2008-04-13 00:29:00 -0500 | [diff] [blame] | 443 | char *get_rawline(int fd, long *plen, char end) |
Rob Landley | bc07865 | 2007-12-15 21:47:25 -0600 | [diff] [blame] | 444 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 445 | char c, *buf = NULL; |
| 446 | long len = 0; |
Rob Landley | bc07865 | 2007-12-15 21:47:25 -0600 | [diff] [blame] | 447 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 448 | for (;;) { |
| 449 | if (1>read(fd, &c, 1)) break; |
| 450 | if (!(len & 63)) buf=xrealloc(buf, len+65); |
| 451 | if ((buf[len++]=c) == end) break; |
| 452 | } |
| 453 | if (buf) buf[len]=0; |
| 454 | if (plen) *plen = len; |
Rob Landley | bc07865 | 2007-12-15 21:47:25 -0600 | [diff] [blame] | 455 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 456 | return buf; |
Rob Landley | bc07865 | 2007-12-15 21:47:25 -0600 | [diff] [blame] | 457 | } |
| 458 | |
| 459 | char *get_line(int fd) |
| 460 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 461 | long len; |
| 462 | char *buf = get_rawline(fd, &len, '\n'); |
Rob Landley | bc07865 | 2007-12-15 21:47:25 -0600 | [diff] [blame] | 463 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 464 | if (buf && buf[--len]=='\n') buf[len]=0; |
Rob Landley | bc07865 | 2007-12-15 21:47:25 -0600 | [diff] [blame] | 465 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 466 | return buf; |
Rob Landley | bc07865 | 2007-12-15 21:47:25 -0600 | [diff] [blame] | 467 | } |
| 468 | |
Rob Landley | 67a069d | 2012-06-03 00:32:12 -0500 | [diff] [blame] | 469 | int wfchmodat(int fd, char *name, mode_t mode) |
| 470 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 471 | int rc = fchmodat(fd, name, mode, 0); |
Rob Landley | 67a069d | 2012-06-03 00:32:12 -0500 | [diff] [blame] | 472 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 473 | if (rc) { |
| 474 | perror_msg("chmod '%s' to %04o", name, mode); |
| 475 | toys.exitval=1; |
| 476 | } |
| 477 | return rc; |
Rob Landley | 67a069d | 2012-06-03 00:32:12 -0500 | [diff] [blame] | 478 | } |
| 479 | |
Rob Landley | c52db60 | 2012-07-30 01:01:33 -0500 | [diff] [blame] | 480 | static char *tempfile2zap; |
| 481 | static void tempfile_handler(int i) |
| 482 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 483 | if (1 < (long)tempfile2zap) unlink(tempfile2zap); |
| 484 | _exit(1); |
Rob Landley | c52db60 | 2012-07-30 01:01:33 -0500 | [diff] [blame] | 485 | } |
| 486 | |
Rob Landley | 42ecbab | 2007-12-18 02:02:21 -0600 | [diff] [blame] | 487 | // Open a temporary file to copy an existing file into. |
| 488 | int copy_tempfile(int fdin, char *name, char **tempname) |
| 489 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 490 | struct stat statbuf; |
| 491 | int fd; |
Rob Landley | 42ecbab | 2007-12-18 02:02:21 -0600 | [diff] [blame] | 492 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 493 | *tempname = xstrndup(name, strlen(name)+6); |
| 494 | strcat(*tempname,"XXXXXX"); |
| 495 | if(-1 == (fd = mkstemp(*tempname))) error_exit("no temp file"); |
| 496 | if (!tempfile2zap) sigatexit(tempfile_handler); |
| 497 | tempfile2zap = *tempname; |
Rob Landley | 42ecbab | 2007-12-18 02:02:21 -0600 | [diff] [blame] | 498 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 499 | // Set permissions of output file |
Rob Landley | 42ecbab | 2007-12-18 02:02:21 -0600 | [diff] [blame] | 500 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 501 | fstat(fdin, &statbuf); |
| 502 | fchmod(fd, statbuf.st_mode); |
Rob Landley | 42ecbab | 2007-12-18 02:02:21 -0600 | [diff] [blame] | 503 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 504 | return fd; |
Rob Landley | 42ecbab | 2007-12-18 02:02:21 -0600 | [diff] [blame] | 505 | } |
| 506 | |
| 507 | // Abort the copy and delete the temporary file. |
| 508 | void delete_tempfile(int fdin, int fdout, char **tempname) |
| 509 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 510 | close(fdin); |
| 511 | close(fdout); |
| 512 | unlink(*tempname); |
| 513 | tempfile2zap = (char *)1; |
| 514 | free(*tempname); |
| 515 | *tempname = NULL; |
Rob Landley | 42ecbab | 2007-12-18 02:02:21 -0600 | [diff] [blame] | 516 | } |
| 517 | |
| 518 | // Copy the rest of the data and replace the original with the copy. |
| 519 | void replace_tempfile(int fdin, int fdout, char **tempname) |
| 520 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 521 | char *temp = xstrdup(*tempname); |
Rob Landley | 42ecbab | 2007-12-18 02:02:21 -0600 | [diff] [blame] | 522 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 523 | temp[strlen(temp)-6]=0; |
| 524 | if (fdin != -1) { |
| 525 | xsendfile(fdin, fdout); |
| 526 | xclose(fdin); |
| 527 | } |
| 528 | xclose(fdout); |
| 529 | rename(*tempname, temp); |
| 530 | tempfile2zap = (char *)1; |
| 531 | free(*tempname); |
| 532 | free(temp); |
| 533 | *tempname = NULL; |
Rob Landley | bc07865 | 2007-12-15 21:47:25 -0600 | [diff] [blame] | 534 | } |
Rob Landley | 7e849c5 | 2009-01-03 18:15:18 -0600 | [diff] [blame] | 535 | |
| 536 | // Create a 256 entry CRC32 lookup table. |
| 537 | |
Rob Landley | b15b8fa | 2009-01-05 01:05:43 -0600 | [diff] [blame] | 538 | void crc_init(unsigned int *crc_table, int little_endian) |
Rob Landley | 7e849c5 | 2009-01-03 18:15:18 -0600 | [diff] [blame] | 539 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 540 | unsigned int i; |
Rob Landley | 7e849c5 | 2009-01-03 18:15:18 -0600 | [diff] [blame] | 541 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 542 | // Init the CRC32 table (big endian) |
| 543 | for (i=0; i<256; i++) { |
| 544 | unsigned int j, c = little_endian ? i : i<<24; |
| 545 | for (j=8; j; j--) |
| 546 | if (little_endian) c = (c&1) ? (c>>1)^0xEDB88320 : c>>1; |
| 547 | else c=c&0x80000000 ? (c<<1)^0x04c11db7 : (c<<1); |
| 548 | crc_table[i] = c; |
| 549 | } |
Rob Landley | 7e849c5 | 2009-01-03 18:15:18 -0600 | [diff] [blame] | 550 | } |
Rob Landley | 26e7b5e | 2012-02-02 07:27:35 -0600 | [diff] [blame] | 551 | |
| 552 | // Quick and dirty query size of terminal, doesn't do ANSI probe fallback. |
Rob Landley | 10bdaa4 | 2013-11-07 09:04:50 -0600 | [diff] [blame] | 553 | // set x=80 y=25 before calling to provide defaults. Returns 0 if couldn't |
| 554 | // determine size. |
Rob Landley | 26e7b5e | 2012-02-02 07:27:35 -0600 | [diff] [blame] | 555 | |
Rob Landley | 10bdaa4 | 2013-11-07 09:04:50 -0600 | [diff] [blame] | 556 | int terminal_size(unsigned *xx, unsigned *yy) |
Rob Landley | 26e7b5e | 2012-02-02 07:27:35 -0600 | [diff] [blame] | 557 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 558 | struct winsize ws; |
Rob Landley | 10bdaa4 | 2013-11-07 09:04:50 -0600 | [diff] [blame] | 559 | unsigned i, x = 0, y = 0; |
Rob Landley | c9cc530 | 2013-10-27 00:02:56 -0500 | [diff] [blame] | 560 | char *s; |
Rob Landley | 26e7b5e | 2012-02-02 07:27:35 -0600 | [diff] [blame] | 561 | |
Rob Landley | 10bdaa4 | 2013-11-07 09:04:50 -0600 | [diff] [blame] | 562 | // stdin, stdout, stderr |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 563 | for (i=0; i<3; i++) { |
Rob Landley | c9cc530 | 2013-10-27 00:02:56 -0500 | [diff] [blame] | 564 | memset(&ws, 0, sizeof(ws)); |
| 565 | if (!ioctl(i, TIOCGWINSZ, &ws)) { |
| 566 | if (ws.ws_col) x = ws.ws_col; |
| 567 | if (ws.ws_row) y = ws.ws_row; |
Rob Landley | 26e7b5e | 2012-02-02 07:27:35 -0600 | [diff] [blame] | 568 | |
Rob Landley | c9cc530 | 2013-10-27 00:02:56 -0500 | [diff] [blame] | 569 | break; |
| 570 | } |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 571 | } |
Rob Landley | c9cc530 | 2013-10-27 00:02:56 -0500 | [diff] [blame] | 572 | s = getenv("COLUMNS"); |
| 573 | if (s) sscanf(s, "%u", &x); |
| 574 | s = getenv("ROWS"); |
| 575 | if (s) sscanf(s, "%u", &y); |
Rob Landley | 26e7b5e | 2012-02-02 07:27:35 -0600 | [diff] [blame] | 576 | |
Rob Landley | 10bdaa4 | 2013-11-07 09:04:50 -0600 | [diff] [blame] | 577 | // Never return 0 for either value, leave it at default instead. |
| 578 | if (xx && x) *xx = x; |
| 579 | if (yy && y) *yy = y; |
| 580 | |
| 581 | return x || y; |
Rob Landley | 26e7b5e | 2012-02-02 07:27:35 -0600 | [diff] [blame] | 582 | } |
| 583 | |
Rob Landley | f793d53 | 2012-02-27 21:56:49 -0600 | [diff] [blame] | 584 | int yesno(char *prompt, int def) |
Rob Landley | 26e7b5e | 2012-02-02 07:27:35 -0600 | [diff] [blame] | 585 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 586 | char buf; |
Rob Landley | 26e7b5e | 2012-02-02 07:27:35 -0600 | [diff] [blame] | 587 | |
Rob Landley | db8eb32 | 2012-12-08 02:25:32 -0600 | [diff] [blame] | 588 | fprintf(stderr, "%s (%c/%c):", prompt, def ? 'Y' : 'y', def ? 'n' : 'N'); |
| 589 | fflush(stderr); |
| 590 | while (fread(&buf, 1, 1, stdin)) { |
| 591 | int new; |
Rob Landley | 26e7b5e | 2012-02-02 07:27:35 -0600 | [diff] [blame] | 592 | |
Rob Landley | 2279108 | 2013-01-31 04:13:07 -0600 | [diff] [blame] | 593 | // The letter changes the value, the newline (or space) returns it. |
Rob Landley | db8eb32 | 2012-12-08 02:25:32 -0600 | [diff] [blame] | 594 | if (isspace(buf)) break; |
| 595 | if (-1 != (new = stridx("ny", tolower(buf)))) def = new; |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 596 | } |
Rob Landley | ee00a7f | 2012-03-19 19:19:21 -0500 | [diff] [blame] | 597 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 598 | return def; |
Rob Landley | 26e7b5e | 2012-02-02 07:27:35 -0600 | [diff] [blame] | 599 | } |
Rob Landley | ff9ee8f | 2012-02-18 15:12:41 -0600 | [diff] [blame] | 600 | |
Rob Landley | 2dd50ad | 2012-02-26 13:48:00 -0600 | [diff] [blame] | 601 | struct signame { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 602 | int num; |
| 603 | char *name; |
Rob Landley | 2dd50ad | 2012-02-26 13:48:00 -0600 | [diff] [blame] | 604 | }; |
| 605 | |
| 606 | // Signals required by POSIX 2008: |
| 607 | // http://pubs.opengroup.org/onlinepubs/9699919799/basedefs/signal.h.html |
| 608 | |
| 609 | #define SIGNIFY(x) {SIG##x, #x} |
| 610 | |
| 611 | static struct signame signames[] = { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 612 | SIGNIFY(ABRT), SIGNIFY(ALRM), SIGNIFY(BUS), |
| 613 | SIGNIFY(FPE), SIGNIFY(HUP), SIGNIFY(ILL), SIGNIFY(INT), SIGNIFY(KILL), |
| 614 | SIGNIFY(PIPE), SIGNIFY(QUIT), SIGNIFY(SEGV), SIGNIFY(TERM), |
| 615 | SIGNIFY(USR1), SIGNIFY(USR2), SIGNIFY(SYS), SIGNIFY(TRAP), |
| 616 | SIGNIFY(VTALRM), SIGNIFY(XCPU), SIGNIFY(XFSZ), |
Rob Landley | c52db60 | 2012-07-30 01:01:33 -0500 | [diff] [blame] | 617 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 618 | // Start of non-terminal signals |
Rob Landley | c52db60 | 2012-07-30 01:01:33 -0500 | [diff] [blame] | 619 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 620 | SIGNIFY(CHLD), SIGNIFY(CONT), SIGNIFY(STOP), SIGNIFY(TSTP), |
| 621 | SIGNIFY(TTIN), SIGNIFY(TTOU), SIGNIFY(URG) |
Rob Landley | 2dd50ad | 2012-02-26 13:48:00 -0600 | [diff] [blame] | 622 | }; |
| 623 | |
| 624 | // not in posix: SIGNIFY(STKFLT), SIGNIFY(WINCH), SIGNIFY(IO), SIGNIFY(PWR) |
| 625 | // obsolete: SIGNIFY(PROF) SIGNIFY(POLL) |
| 626 | |
Rob Landley | 1bc5224 | 2014-05-21 07:24:16 -0500 | [diff] [blame] | 627 | // Handler that sets toys.signal, and writes to toys.signalfd if set |
| 628 | void generic_signal(int sig) |
| 629 | { |
| 630 | if (toys.signalfd) { |
| 631 | char c = sig; |
| 632 | |
| 633 | writeall(toys.signalfd, &c, 1); |
| 634 | } |
| 635 | toys.signal = sig; |
| 636 | } |
| 637 | |
Rob Landley | c52db60 | 2012-07-30 01:01:33 -0500 | [diff] [blame] | 638 | // Install the same handler on every signal that defaults to killing the process |
| 639 | void sigatexit(void *handler) |
| 640 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 641 | int i; |
| 642 | for (i=0; signames[i].num != SIGCHLD; i++) signal(signames[i].num, handler); |
Rob Landley | c52db60 | 2012-07-30 01:01:33 -0500 | [diff] [blame] | 643 | } |
Rob Landley | 1bc5224 | 2014-05-21 07:24:16 -0500 | [diff] [blame] | 644 | |
Rob Landley | 2dd50ad | 2012-02-26 13:48:00 -0600 | [diff] [blame] | 645 | // Convert name to signal number. If name == NULL print names. |
| 646 | int sig_to_num(char *pidstr) |
| 647 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 648 | int i; |
Rob Landley | 2dd50ad | 2012-02-26 13:48:00 -0600 | [diff] [blame] | 649 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 650 | if (pidstr) { |
| 651 | char *s; |
| 652 | i = strtol(pidstr, &s, 10); |
| 653 | if (!*s) return i; |
Rob Landley | 2dd50ad | 2012-02-26 13:48:00 -0600 | [diff] [blame] | 654 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 655 | if (!strncasecmp(pidstr, "sig", 3)) pidstr+=3; |
| 656 | } |
| 657 | for (i = 0; i < sizeof(signames)/sizeof(struct signame); i++) |
| 658 | if (!pidstr) xputs(signames[i].name); |
| 659 | else if (!strcasecmp(pidstr, signames[i].name)) return signames[i].num; |
Rob Landley | 2dd50ad | 2012-02-26 13:48:00 -0600 | [diff] [blame] | 660 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 661 | return -1; |
Rob Landley | 2dd50ad | 2012-02-26 13:48:00 -0600 | [diff] [blame] | 662 | } |
| 663 | |
| 664 | char *num_to_sig(int sig) |
| 665 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 666 | int i; |
Rob Landley | 2dd50ad | 2012-02-26 13:48:00 -0600 | [diff] [blame] | 667 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 668 | for (i=0; i<sizeof(signames)/sizeof(struct signame); i++) |
| 669 | if (signames[i].num == sig) return signames[i].name; |
| 670 | return NULL; |
Rob Landley | 2dd50ad | 2012-02-26 13:48:00 -0600 | [diff] [blame] | 671 | } |
Daniel Walter | 05744b3 | 2012-03-19 19:57:56 -0500 | [diff] [blame] | 672 | |
Rob Landley | c52db60 | 2012-07-30 01:01:33 -0500 | [diff] [blame] | 673 | // premute mode bits based on posix mode strings. |
Rob Landley | b6f601e | 2012-05-16 21:11:43 -0500 | [diff] [blame] | 674 | mode_t string_to_mode(char *modestr, mode_t mode) |
Daniel Walter | 05744b3 | 2012-03-19 19:57:56 -0500 | [diff] [blame] | 675 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 676 | char *whos = "ogua", *hows = "=+-", *whats = "xwrstX", *whys = "ogu"; |
| 677 | char *s, *str = modestr; |
Rob Landley | cf6bcb2 | 2012-03-19 20:56:18 -0500 | [diff] [blame] | 678 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 679 | // Handle octal mode |
| 680 | if (isdigit(*str)) { |
| 681 | mode = strtol(str, &s, 8); |
| 682 | if (*s || (mode & ~(07777))) goto barf; |
Rob Landley | b6f601e | 2012-05-16 21:11:43 -0500 | [diff] [blame] | 683 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 684 | return mode; |
| 685 | } |
Rob Landley | cf6bcb2 | 2012-03-19 20:56:18 -0500 | [diff] [blame] | 686 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 687 | // Gaze into the bin of permission... |
| 688 | for (;;) { |
| 689 | int i, j, dowho, dohow, dowhat, amask; |
Rob Landley | cf6bcb2 | 2012-03-19 20:56:18 -0500 | [diff] [blame] | 690 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 691 | dowho = dohow = dowhat = amask = 0; |
Daniel Walter | 05744b3 | 2012-03-19 19:57:56 -0500 | [diff] [blame] | 692 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 693 | // Find the who, how, and what stanzas, in that order |
| 694 | while (*str && (s = strchr(whos, *str))) { |
| 695 | dowho |= 1<<(s-whos); |
| 696 | str++; |
| 697 | } |
| 698 | // If who isn't specified, like "a" but honoring umask. |
| 699 | if (!dowho) { |
| 700 | dowho = 8; |
| 701 | umask(amask=umask(0)); |
| 702 | } |
| 703 | if (!*str || !(s = strchr(hows, *str))) goto barf; |
| 704 | dohow = *(str++); |
Rob Landley | 67a069d | 2012-06-03 00:32:12 -0500 | [diff] [blame] | 705 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 706 | if (!dohow) goto barf; |
| 707 | while (*str && (s = strchr(whats, *str))) { |
| 708 | dowhat |= 1<<(s-whats); |
| 709 | str++; |
| 710 | } |
Rob Landley | b6f601e | 2012-05-16 21:11:43 -0500 | [diff] [blame] | 711 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 712 | // Convert X to x for directory or if already executable somewhere |
| 713 | if ((dowhat&32) && (S_ISDIR(mode) || (mode&0111))) dowhat |= 1; |
Rob Landley | b6f601e | 2012-05-16 21:11:43 -0500 | [diff] [blame] | 714 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 715 | // Copy mode from another category? |
| 716 | if (!dowhat && *str && (s = strchr(whys, *str))) { |
| 717 | dowhat = (mode>>(3*(s-whys)))&7; |
| 718 | str++; |
| 719 | } |
Rob Landley | b6f601e | 2012-05-16 21:11:43 -0500 | [diff] [blame] | 720 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 721 | // Are we ready to do a thing yet? |
| 722 | if (*str && *(str++) != ',') goto barf; |
Rob Landley | b6f601e | 2012-05-16 21:11:43 -0500 | [diff] [blame] | 723 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 724 | // Ok, apply the bits to the mode. |
| 725 | for (i=0; i<4; i++) { |
| 726 | for (j=0; j<3; j++) { |
| 727 | mode_t bit = 0; |
| 728 | int where = 1<<((3*i)+j); |
Rob Landley | 31f49e7 | 2012-07-21 22:45:05 -0500 | [diff] [blame] | 729 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 730 | if (amask & where) continue; |
Rob Landley | b6f601e | 2012-05-16 21:11:43 -0500 | [diff] [blame] | 731 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 732 | // Figure out new value at this location |
| 733 | if (i == 3) { |
| 734 | // suid/sticky bit. |
| 735 | if (j) { |
| 736 | if ((dowhat & 8) && (dowho&(8|(1<<i)))) bit++; |
| 737 | } else if (dowhat & 16) bit++; |
| 738 | } else { |
| 739 | if (!(dowho&(8|(1<<i)))) continue; |
| 740 | if (dowhat&(1<<j)) bit++; |
| 741 | } |
Rob Landley | b6f601e | 2012-05-16 21:11:43 -0500 | [diff] [blame] | 742 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 743 | // When selection active, modify bit |
Rob Landley | b6f601e | 2012-05-16 21:11:43 -0500 | [diff] [blame] | 744 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 745 | if (dohow == '=' || (bit && dohow == '-')) mode &= ~where; |
| 746 | if (bit && dohow != '-') mode |= where; |
| 747 | } |
| 748 | } |
Rob Landley | b6f601e | 2012-05-16 21:11:43 -0500 | [diff] [blame] | 749 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 750 | if (!*str) break; |
| 751 | } |
| 752 | return mode; |
Rob Landley | b6f601e | 2012-05-16 21:11:43 -0500 | [diff] [blame] | 753 | barf: |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 754 | error_exit("bad mode '%s'", modestr); |
Daniel Walter | 05744b3 | 2012-03-19 19:57:56 -0500 | [diff] [blame] | 755 | } |
Ashwini Kumar | 1a0eedf | 2012-08-26 21:17:00 -0500 | [diff] [blame] | 756 | |
Rob Landley | 5a26a86 | 2013-06-02 00:24:24 -0500 | [diff] [blame] | 757 | // Format access mode into a drwxrwxrwx string |
| 758 | void mode_to_string(mode_t mode, char *buf) |
Felix Janda | 5a221e6 | 2013-04-22 22:29:43 +0200 | [diff] [blame] | 759 | { |
| 760 | char c, d; |
| 761 | int i, bit; |
| 762 | |
Rob Landley | 5a26a86 | 2013-06-02 00:24:24 -0500 | [diff] [blame] | 763 | buf[10]=0; |
Felix Janda | 5a221e6 | 2013-04-22 22:29:43 +0200 | [diff] [blame] | 764 | for (i=0; i<9; i++) { |
| 765 | bit = mode & (1<<i); |
| 766 | c = i%3; |
| 767 | if (!c && (mode & (1<<((d=i/3)+9)))) { |
| 768 | c = "tss"[d]; |
| 769 | if (!bit) c &= ~0x20; |
| 770 | } else c = bit ? "xwr"[c] : '-'; |
Rob Landley | 5a26a86 | 2013-06-02 00:24:24 -0500 | [diff] [blame] | 771 | buf[9-i] = c; |
Felix Janda | 5a221e6 | 2013-04-22 22:29:43 +0200 | [diff] [blame] | 772 | } |
| 773 | |
| 774 | if (S_ISDIR(mode)) c = 'd'; |
| 775 | else if (S_ISBLK(mode)) c = 'b'; |
| 776 | else if (S_ISCHR(mode)) c = 'c'; |
| 777 | else if (S_ISLNK(mode)) c = 'l'; |
| 778 | else if (S_ISFIFO(mode)) c = 'p'; |
| 779 | else if (S_ISSOCK(mode)) c = 's'; |
| 780 | else c = '-'; |
Rob Landley | 5a26a86 | 2013-06-02 00:24:24 -0500 | [diff] [blame] | 781 | *buf = c; |
Felix Janda | 5a221e6 | 2013-04-22 22:29:43 +0200 | [diff] [blame] | 782 | } |
Rob Landley | db1009d | 2013-12-19 09:32:30 -0600 | [diff] [blame] | 783 | |
| 784 | // Execute a callback for each PID that matches a process name from a list. |
| 785 | void names_to_pid(char **names, int (*callback)(pid_t pid, char *name)) |
| 786 | { |
| 787 | DIR *dp; |
| 788 | struct dirent *entry; |
| 789 | |
| 790 | if (!(dp = opendir("/proc"))) perror_exit("opendir"); |
| 791 | |
| 792 | while ((entry = readdir(dp))) { |
| 793 | unsigned u; |
| 794 | char *cmd, **curname; |
| 795 | |
| 796 | if (!(u = atoi(entry->d_name))) continue; |
| 797 | sprintf(libbuf, "/proc/%u/cmdline", u); |
| 798 | if (!(cmd = readfile(libbuf, libbuf, sizeof(libbuf)))) continue; |
| 799 | |
| 800 | for (curname = names; *curname; curname++) |
| 801 | if (**curname == '/' ? !strcmp(cmd, *curname) |
| 802 | : !strcmp(basename(cmd), basename(*curname))) |
| 803 | if (callback(u, *curname)) break; |
| 804 | if (*curname) break; |
| 805 | } |
| 806 | closedir(dp); |
| 807 | } |
Rob Landley | 48c172b | 2014-05-06 06:31:28 -0500 | [diff] [blame] | 808 | |
| 809 | // display first few digits of number with power of two units, except we're |
| 810 | // actually just counting decimal digits and showing mil/bil/trillions. |
| 811 | int human_readable(char *buf, unsigned long long num) |
| 812 | { |
| 813 | int end, len; |
| 814 | |
| 815 | len = sprintf(buf, "%lld", num); |
| 816 | end = ((len-1)%3)+1; |
| 817 | len /= 3; |
| 818 | |
| 819 | if (len && end == 1) { |
| 820 | buf[2] = buf[1]; |
| 821 | buf[1] = '.'; |
| 822 | end = 3; |
| 823 | } |
| 824 | buf[end++] = ' '; |
| 825 | if (len) buf[end++] = " KMGTPE"[len]; |
| 826 | buf[end++] = 'B'; |
| 827 | buf[end++] = 0; |
| 828 | |
| 829 | return end; |
| 830 | } |