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) { |
| 70 | int i = read(fd, buf+count, len-count); |
| 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) { |
| 84 | int i = write(fd, buf+count, len-count); |
| 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 | 2037b83 | 2012-07-15 16:56:20 -0500 | [diff] [blame] | 92 | off_t lskip(int fd, off_t offset) |
| 93 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 94 | off_t and = lseek(fd, offset, SEEK_CUR); |
Rob Landley | 2037b83 | 2012-07-15 16:56:20 -0500 | [diff] [blame] | 95 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 96 | if (and != -1 && offset >= lseek(fd, offset, SEEK_END) |
| 97 | && offset+and == lseek(fd, offset+and, SEEK_SET)) return 0; |
| 98 | else { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 99 | while (offset>0) { |
Rob Landley | 8fdcfdb | 2013-09-03 17:56:28 -0500 | [diff] [blame^] | 100 | int try = offset>sizeof(libbuf) ? sizeof(libbuf) : offset, or; |
Rob Landley | 2037b83 | 2012-07-15 16:56:20 -0500 | [diff] [blame] | 101 | |
Rob Landley | 8fdcfdb | 2013-09-03 17:56:28 -0500 | [diff] [blame^] | 102 | or = readall(fd, libbuf, try); |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 103 | if (or < 0) perror_msg("lskip to %lld", (long long)offset); |
| 104 | else offset -= try; |
| 105 | if (or < try) break; |
| 106 | } |
Rob Landley | 2037b83 | 2012-07-15 16:56:20 -0500 | [diff] [blame] | 107 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 108 | return offset; |
| 109 | } |
Rob Landley | 2037b83 | 2012-07-15 16:56:20 -0500 | [diff] [blame] | 110 | } |
| 111 | |
Rob Landley | fe91e68 | 2012-11-22 21:18:09 -0600 | [diff] [blame] | 112 | // Split a path into linked list of components, tracking head and tail of list. |
| 113 | // Filters out // entries with no contents. |
| 114 | struct string_list **splitpath(char *path, struct string_list **list) |
landley | 00f87f1 | 2006-10-25 18:38:37 -0400 | [diff] [blame] | 115 | { |
Rob Landley | fe91e68 | 2012-11-22 21:18:09 -0600 | [diff] [blame] | 116 | char *new = path; |
landley | 00f87f1 | 2006-10-25 18:38:37 -0400 | [diff] [blame] | 117 | |
Rob Landley | fe91e68 | 2012-11-22 21:18:09 -0600 | [diff] [blame] | 118 | *list = 0; |
| 119 | do { |
| 120 | int len; |
landley | 00f87f1 | 2006-10-25 18:38:37 -0400 | [diff] [blame] | 121 | |
Rob Landley | fe91e68 | 2012-11-22 21:18:09 -0600 | [diff] [blame] | 122 | if (*path && *path != '/') continue; |
| 123 | len = path-new; |
| 124 | if (len > 0) { |
| 125 | *list = xmalloc(sizeof(struct string_list) + len + 1); |
| 126 | (*list)->next = 0; |
| 127 | strncpy((*list)->str, new, len); |
| 128 | (*list)->str[len] = 0; |
| 129 | list = &(*list)->next; |
| 130 | } |
| 131 | new = path+1; |
| 132 | } while (*path++); |
| 133 | |
| 134 | return list; |
| 135 | } |
| 136 | |
Rob Landley | 0a04b3e | 2006-11-03 00:05:52 -0500 | [diff] [blame] | 137 | // Find all file in a colon-separated path with access type "type" (generally |
| 138 | // X_OK or R_OK). Returns a list of absolute paths to each file found, in |
| 139 | // order. |
| 140 | |
| 141 | struct string_list *find_in_path(char *path, char *filename) |
Rob Landley | fa98d01 | 2006-11-02 02:57:27 -0500 | [diff] [blame] | 142 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 143 | struct string_list *rlist = NULL, **prlist=&rlist; |
| 144 | char *cwd = xgetcwd(); |
Rob Landley | fa98d01 | 2006-11-02 02:57:27 -0500 | [diff] [blame] | 145 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 146 | for (;;) { |
| 147 | char *next = path ? strchr(path, ':') : NULL; |
| 148 | int len = next ? next-path : strlen(path); |
| 149 | struct string_list *rnext; |
| 150 | struct stat st; |
Rob Landley | fa98d01 | 2006-11-02 02:57:27 -0500 | [diff] [blame] | 151 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 152 | rnext = xmalloc(sizeof(void *) + strlen(filename) |
| 153 | + (len ? len : strlen(cwd)) + 2); |
| 154 | if (!len) sprintf(rnext->str, "%s/%s", cwd, filename); |
| 155 | else { |
| 156 | char *res = rnext->str; |
| 157 | strncpy(res, path, len); |
| 158 | res += len; |
| 159 | *(res++) = '/'; |
| 160 | strcpy(res, filename); |
| 161 | } |
Rob Landley | fa98d01 | 2006-11-02 02:57:27 -0500 | [diff] [blame] | 162 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 163 | // Confirm it's not a directory. |
| 164 | if (!stat(rnext->str, &st) && S_ISREG(st.st_mode)) { |
| 165 | *prlist = rnext; |
| 166 | rnext->next = NULL; |
| 167 | prlist = &(rnext->next); |
| 168 | } else free(rnext); |
Rob Landley | fa98d01 | 2006-11-02 02:57:27 -0500 | [diff] [blame] | 169 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 170 | if (!next) break; |
| 171 | path += len; |
| 172 | path++; |
| 173 | } |
| 174 | free(cwd); |
landley | 00f87f1 | 2006-10-25 18:38:37 -0400 | [diff] [blame] | 175 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 176 | return rlist; |
landley | 00f87f1 | 2006-10-25 18:38:37 -0400 | [diff] [blame] | 177 | } |
landley | 09ea7ac | 2006-10-30 01:38:00 -0500 | [diff] [blame] | 178 | |
Rob Landley | f575716 | 2007-02-16 21:08:22 -0500 | [diff] [blame] | 179 | // atol() with the kilo/mega/giga/tera/peta/exa extensions. |
| 180 | // (zetta and yotta don't fit in 64 bits.) |
Rob Landley | 5e6dca6 | 2012-02-09 06:09:27 -0600 | [diff] [blame] | 181 | long atolx(char *numstr) |
Rob Landley | f575716 | 2007-02-16 21:08:22 -0500 | [diff] [blame] | 182 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 183 | char *c, *suffixes="bkmgtpe", *end; |
| 184 | long val = strtol(numstr, &c, 0); |
Rob Landley | f575716 | 2007-02-16 21:08:22 -0500 | [diff] [blame] | 185 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 186 | if (*c) { |
| 187 | if (c != numstr && (end = strchr(suffixes, tolower(*c)))) { |
| 188 | int shift = end-suffixes; |
| 189 | if (shift--) val *= 1024L<<(shift*10); |
| 190 | } else { |
| 191 | while (isspace(*c)) c++; |
| 192 | if (*c) error_exit("not integer: %s", numstr); |
| 193 | } |
| 194 | } |
Rob Landley | ad63f4b | 2011-12-12 15:19:52 -0600 | [diff] [blame] | 195 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 196 | return val; |
Rob Landley | f575716 | 2007-02-16 21:08:22 -0500 | [diff] [blame] | 197 | } |
| 198 | |
Rob Landley | eb7ea22 | 2012-04-14 22:30:41 -0500 | [diff] [blame] | 199 | int numlen(long l) |
| 200 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 201 | int len = 0; |
| 202 | while (l) { |
| 203 | l /= 10; |
| 204 | len++; |
| 205 | } |
| 206 | return len; |
Rob Landley | eb7ea22 | 2012-04-14 22:30:41 -0500 | [diff] [blame] | 207 | } |
| 208 | |
Rob Landley | 2037b83 | 2012-07-15 16:56:20 -0500 | [diff] [blame] | 209 | int stridx(char *haystack, char needle) |
| 210 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 211 | char *off; |
Rob Landley | 2037b83 | 2012-07-15 16:56:20 -0500 | [diff] [blame] | 212 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 213 | if (!needle) return -1; |
| 214 | off = strchr(haystack, needle); |
| 215 | if (!off) return -1; |
Rob Landley | 2037b83 | 2012-07-15 16:56:20 -0500 | [diff] [blame] | 216 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 217 | return off-haystack; |
Rob Landley | 2037b83 | 2012-07-15 16:56:20 -0500 | [diff] [blame] | 218 | } |
| 219 | |
Rob Landley | 055cfcb | 2007-01-14 20:20:06 -0500 | [diff] [blame] | 220 | // Return how long the file at fd is, if there's any way to determine it. |
| 221 | off_t fdlength(int fd) |
| 222 | { |
Rob Landley | 035f27a | 2013-08-08 02:46:45 -0500 | [diff] [blame] | 223 | struct stat st; |
| 224 | off_t base = 0, range = 1, expand = 1, old; |
| 225 | |
| 226 | 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] | 227 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 228 | // If the ioctl works for this, return it. |
Rob Landley | 035f27a | 2013-08-08 02:46:45 -0500 | [diff] [blame] | 229 | // TODO: is blocksize still always 512, or do we stat for it? |
| 230 | // unsigned int size; |
| 231 | // if (ioctl(fd, BLKGETSIZE, &size) >= 0) return size*512L; |
Rob Landley | 055cfcb | 2007-01-14 20:20:06 -0500 | [diff] [blame] | 232 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 233 | // If not, do a binary search for the last location we can read. (Some |
| 234 | // block devices don't do BLKGETSIZE right.) This should probably have |
| 235 | // a CONFIG option... |
Rob Landley | 055cfcb | 2007-01-14 20:20:06 -0500 | [diff] [blame] | 236 | |
Rob Landley | 035f27a | 2013-08-08 02:46:45 -0500 | [diff] [blame] | 237 | // If not, do a binary search for the last location we can read. |
| 238 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 239 | old = lseek(fd, 0, SEEK_CUR); |
| 240 | do { |
| 241 | char temp; |
Rob Landley | 035f27a | 2013-08-08 02:46:45 -0500 | [diff] [blame] | 242 | off_t pos = base + range / 2; |
Rob Landley | 055cfcb | 2007-01-14 20:20:06 -0500 | [diff] [blame] | 243 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 244 | if (lseek(fd, pos, 0)>=0 && read(fd, &temp, 1)==1) { |
Rob Landley | 035f27a | 2013-08-08 02:46:45 -0500 | [diff] [blame] | 245 | off_t delta = (pos + 1) - base; |
Rob Landley | 055cfcb | 2007-01-14 20:20:06 -0500 | [diff] [blame] | 246 | |
Rob Landley | 035f27a | 2013-08-08 02:46:45 -0500 | [diff] [blame] | 247 | base += delta; |
| 248 | if (expand) range = (expand <<= 1) - base; |
| 249 | else range -= delta; |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 250 | } else { |
Rob Landley | 035f27a | 2013-08-08 02:46:45 -0500 | [diff] [blame] | 251 | expand = 0; |
| 252 | range = pos - base; |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 253 | } |
Rob Landley | 035f27a | 2013-08-08 02:46:45 -0500 | [diff] [blame] | 254 | } while (range > 0); |
Rob Landley | 055cfcb | 2007-01-14 20:20:06 -0500 | [diff] [blame] | 255 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 256 | lseek(fd, old, SEEK_SET); |
Rob Landley | f15387d | 2008-07-18 05:43:44 -0500 | [diff] [blame] | 257 | |
Rob Landley | 035f27a | 2013-08-08 02:46:45 -0500 | [diff] [blame] | 258 | return base; |
Rob Landley | 055cfcb | 2007-01-14 20:20:06 -0500 | [diff] [blame] | 259 | } |
| 260 | |
Rob Landley | 8fdcfdb | 2013-09-03 17:56:28 -0500 | [diff] [blame^] | 261 | // Read contents of file as a single nul-terminated string. |
| 262 | // malloc new one if buf=len=0 |
| 263 | char *readfile(char *name, char *buf, off_t len) |
Rob Landley | 055cfcb | 2007-01-14 20:20:06 -0500 | [diff] [blame] | 264 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 265 | int fd; |
Rob Landley | 055cfcb | 2007-01-14 20:20:06 -0500 | [diff] [blame] | 266 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 267 | fd = open(name, O_RDONLY); |
| 268 | if (fd == -1) return 0; |
Rob Landley | 8fdcfdb | 2013-09-03 17:56:28 -0500 | [diff] [blame^] | 269 | |
| 270 | if (len<1) { |
| 271 | len = fdlength(fd); |
| 272 | // proc files don't report a length, so try 1 page minimum. |
| 273 | if (len<4096) len = 4096; |
| 274 | } |
| 275 | if (!buf) buf = xmalloc(len+1); |
| 276 | |
| 277 | len = readall(fd, buf, len-1); |
| 278 | close(fd); |
| 279 | if (len<0) { |
| 280 | free(buf); |
| 281 | buf = 0; |
| 282 | } else buf[len] = 0; |
Rob Landley | 055cfcb | 2007-01-14 20:20:06 -0500 | [diff] [blame] | 283 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 284 | return buf; |
Rob Landley | 055cfcb | 2007-01-14 20:20:06 -0500 | [diff] [blame] | 285 | } |
| 286 | |
Rob Landley | f015344 | 2013-04-26 02:41:05 -0500 | [diff] [blame] | 287 | // Sleep for this many thousandths of a second |
| 288 | void msleep(long miliseconds) |
| 289 | { |
| 290 | struct timespec ts; |
| 291 | |
| 292 | ts.tv_sec = miliseconds/1000; |
| 293 | ts.tv_nsec = (miliseconds%1000)*1000000; |
| 294 | nanosleep(&ts, &ts); |
| 295 | } |
| 296 | |
Rob Landley | 6b28341 | 2013-06-01 20:41:35 -0500 | [diff] [blame] | 297 | int64_t peek(void *ptr, int size) |
| 298 | { |
| 299 | if (size & 8) { |
| 300 | int64_t *p = (int64_t *)ptr; |
| 301 | return *p; |
| 302 | } else if (size & 4) { |
| 303 | int *p = (int *)ptr; |
| 304 | return *p; |
| 305 | } else if (size & 2) { |
| 306 | short *p = (short *)ptr; |
| 307 | return *p; |
| 308 | } else { |
| 309 | char *p = (char *)ptr; |
| 310 | return *p; |
| 311 | } |
| 312 | } |
| 313 | |
| 314 | void poke(void *ptr, uint64_t val, int size) |
| 315 | { |
| 316 | if (size & 8) { |
| 317 | uint64_t *p = (uint64_t *)ptr; |
| 318 | *p = val; |
| 319 | } else if (size & 4) { |
| 320 | int *p = (int *)ptr; |
| 321 | *p = val; |
| 322 | } else if (size & 2) { |
| 323 | short *p = (short *)ptr; |
| 324 | *p = val; |
| 325 | } else { |
| 326 | char *p = (char *)ptr; |
| 327 | *p = val; |
| 328 | } |
| 329 | } |
| 330 | |
Rob Landley | 2bfaaf2 | 2008-07-03 19:19:00 -0500 | [diff] [blame] | 331 | // Iterate through an array of files, opening each one and calling a function |
| 332 | // on that filehandle and name. The special filename "-" means stdin if |
| 333 | // flags is O_RDONLY, stdout otherwise. An empty argument list calls |
| 334 | // function() on just stdin/stdout. |
| 335 | // |
| 336 | // Note: read only filehandles are automatically closed when function() |
| 337 | // returns, but writeable filehandles must be close by function() |
Rob Landley | ad63f4b | 2011-12-12 15:19:52 -0600 | [diff] [blame] | 338 | void loopfiles_rw(char **argv, int flags, int permissions, int failok, |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 339 | void (*function)(int fd, char *name)) |
Rob Landley | 7634b55 | 2007-11-29 17:49:50 -0600 | [diff] [blame] | 340 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 341 | int fd; |
Rob Landley | 7634b55 | 2007-11-29 17:49:50 -0600 | [diff] [blame] | 342 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 343 | // If no arguments, read from stdin. |
| 344 | if (!*argv) function(flags ? 1 : 0, "-"); |
| 345 | else do { |
| 346 | // Filename "-" means read from stdin. |
| 347 | // Inability to open a file prints a warning, but doesn't exit. |
Rob Landley | 7634b55 | 2007-11-29 17:49:50 -0600 | [diff] [blame] | 348 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 349 | if (!strcmp(*argv,"-")) fd=0; |
| 350 | else if (0>(fd = open(*argv, flags, permissions)) && !failok) { |
| 351 | perror_msg("%s", *argv); |
| 352 | toys.exitval = 1; |
| 353 | continue; |
| 354 | } |
| 355 | function(fd, *argv); |
| 356 | if (flags == O_RDONLY) close(fd); |
| 357 | } while (*++argv); |
Rob Landley | 7634b55 | 2007-11-29 17:49:50 -0600 | [diff] [blame] | 358 | } |
Rob Landley | bc07865 | 2007-12-15 21:47:25 -0600 | [diff] [blame] | 359 | |
Rob Landley | ad63f4b | 2011-12-12 15:19:52 -0600 | [diff] [blame] | 360 | // Call loopfiles_rw with O_RDONLY and !failok (common case). |
Rob Landley | 2bfaaf2 | 2008-07-03 19:19:00 -0500 | [diff] [blame] | 361 | void loopfiles(char **argv, void (*function)(int fd, char *name)) |
| 362 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 363 | loopfiles_rw(argv, O_RDONLY, 0, 0, function); |
Rob Landley | 2bfaaf2 | 2008-07-03 19:19:00 -0500 | [diff] [blame] | 364 | } |
| 365 | |
Rob Landley | bc07865 | 2007-12-15 21:47:25 -0600 | [diff] [blame] | 366 | // Slow, but small. |
| 367 | |
Rob Landley | 3fc4e0f | 2008-04-13 00:29:00 -0500 | [diff] [blame] | 368 | char *get_rawline(int fd, long *plen, char end) |
Rob Landley | bc07865 | 2007-12-15 21:47:25 -0600 | [diff] [blame] | 369 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 370 | char c, *buf = NULL; |
| 371 | long len = 0; |
Rob Landley | bc07865 | 2007-12-15 21:47:25 -0600 | [diff] [blame] | 372 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 373 | for (;;) { |
| 374 | if (1>read(fd, &c, 1)) break; |
| 375 | if (!(len & 63)) buf=xrealloc(buf, len+65); |
| 376 | if ((buf[len++]=c) == end) break; |
| 377 | } |
| 378 | if (buf) buf[len]=0; |
| 379 | if (plen) *plen = len; |
Rob Landley | bc07865 | 2007-12-15 21:47:25 -0600 | [diff] [blame] | 380 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 381 | return buf; |
Rob Landley | bc07865 | 2007-12-15 21:47:25 -0600 | [diff] [blame] | 382 | } |
| 383 | |
| 384 | char *get_line(int fd) |
| 385 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 386 | long len; |
| 387 | char *buf = get_rawline(fd, &len, '\n'); |
Rob Landley | bc07865 | 2007-12-15 21:47:25 -0600 | [diff] [blame] | 388 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 389 | if (buf && buf[--len]=='\n') buf[len]=0; |
Rob Landley | bc07865 | 2007-12-15 21:47:25 -0600 | [diff] [blame] | 390 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 391 | return buf; |
Rob Landley | bc07865 | 2007-12-15 21:47:25 -0600 | [diff] [blame] | 392 | } |
| 393 | |
Rob Landley | 67a069d | 2012-06-03 00:32:12 -0500 | [diff] [blame] | 394 | int wfchmodat(int fd, char *name, mode_t mode) |
| 395 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 396 | int rc = fchmodat(fd, name, mode, 0); |
Rob Landley | 67a069d | 2012-06-03 00:32:12 -0500 | [diff] [blame] | 397 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 398 | if (rc) { |
| 399 | perror_msg("chmod '%s' to %04o", name, mode); |
| 400 | toys.exitval=1; |
| 401 | } |
| 402 | return rc; |
Rob Landley | 67a069d | 2012-06-03 00:32:12 -0500 | [diff] [blame] | 403 | } |
| 404 | |
Rob Landley | c52db60 | 2012-07-30 01:01:33 -0500 | [diff] [blame] | 405 | static char *tempfile2zap; |
| 406 | static void tempfile_handler(int i) |
| 407 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 408 | if (1 < (long)tempfile2zap) unlink(tempfile2zap); |
| 409 | _exit(1); |
Rob Landley | c52db60 | 2012-07-30 01:01:33 -0500 | [diff] [blame] | 410 | } |
| 411 | |
Rob Landley | 42ecbab | 2007-12-18 02:02:21 -0600 | [diff] [blame] | 412 | // Open a temporary file to copy an existing file into. |
| 413 | int copy_tempfile(int fdin, char *name, char **tempname) |
| 414 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 415 | struct stat statbuf; |
| 416 | int fd; |
Rob Landley | 42ecbab | 2007-12-18 02:02:21 -0600 | [diff] [blame] | 417 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 418 | *tempname = xstrndup(name, strlen(name)+6); |
| 419 | strcat(*tempname,"XXXXXX"); |
| 420 | if(-1 == (fd = mkstemp(*tempname))) error_exit("no temp file"); |
| 421 | if (!tempfile2zap) sigatexit(tempfile_handler); |
| 422 | tempfile2zap = *tempname; |
Rob Landley | 42ecbab | 2007-12-18 02:02:21 -0600 | [diff] [blame] | 423 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 424 | // Set permissions of output file |
Rob Landley | 42ecbab | 2007-12-18 02:02:21 -0600 | [diff] [blame] | 425 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 426 | fstat(fdin, &statbuf); |
| 427 | fchmod(fd, statbuf.st_mode); |
Rob Landley | 42ecbab | 2007-12-18 02:02:21 -0600 | [diff] [blame] | 428 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 429 | return fd; |
Rob Landley | 42ecbab | 2007-12-18 02:02:21 -0600 | [diff] [blame] | 430 | } |
| 431 | |
| 432 | // Abort the copy and delete the temporary file. |
| 433 | void delete_tempfile(int fdin, int fdout, char **tempname) |
| 434 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 435 | close(fdin); |
| 436 | close(fdout); |
| 437 | unlink(*tempname); |
| 438 | tempfile2zap = (char *)1; |
| 439 | free(*tempname); |
| 440 | *tempname = NULL; |
Rob Landley | 42ecbab | 2007-12-18 02:02:21 -0600 | [diff] [blame] | 441 | } |
| 442 | |
| 443 | // Copy the rest of the data and replace the original with the copy. |
| 444 | void replace_tempfile(int fdin, int fdout, char **tempname) |
| 445 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 446 | char *temp = xstrdup(*tempname); |
Rob Landley | 42ecbab | 2007-12-18 02:02:21 -0600 | [diff] [blame] | 447 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 448 | temp[strlen(temp)-6]=0; |
| 449 | if (fdin != -1) { |
| 450 | xsendfile(fdin, fdout); |
| 451 | xclose(fdin); |
| 452 | } |
| 453 | xclose(fdout); |
| 454 | rename(*tempname, temp); |
| 455 | tempfile2zap = (char *)1; |
| 456 | free(*tempname); |
| 457 | free(temp); |
| 458 | *tempname = NULL; |
Rob Landley | bc07865 | 2007-12-15 21:47:25 -0600 | [diff] [blame] | 459 | } |
Rob Landley | 7e849c5 | 2009-01-03 18:15:18 -0600 | [diff] [blame] | 460 | |
| 461 | // Create a 256 entry CRC32 lookup table. |
| 462 | |
Rob Landley | b15b8fa | 2009-01-05 01:05:43 -0600 | [diff] [blame] | 463 | void crc_init(unsigned int *crc_table, int little_endian) |
Rob Landley | 7e849c5 | 2009-01-03 18:15:18 -0600 | [diff] [blame] | 464 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 465 | unsigned int i; |
Rob Landley | 7e849c5 | 2009-01-03 18:15:18 -0600 | [diff] [blame] | 466 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 467 | // Init the CRC32 table (big endian) |
| 468 | for (i=0; i<256; i++) { |
| 469 | unsigned int j, c = little_endian ? i : i<<24; |
| 470 | for (j=8; j; j--) |
| 471 | if (little_endian) c = (c&1) ? (c>>1)^0xEDB88320 : c>>1; |
| 472 | else c=c&0x80000000 ? (c<<1)^0x04c11db7 : (c<<1); |
| 473 | crc_table[i] = c; |
| 474 | } |
Rob Landley | 7e849c5 | 2009-01-03 18:15:18 -0600 | [diff] [blame] | 475 | } |
Rob Landley | 26e7b5e | 2012-02-02 07:27:35 -0600 | [diff] [blame] | 476 | |
| 477 | // Quick and dirty query size of terminal, doesn't do ANSI probe fallback. |
| 478 | // set *x=0 and *y=0 before calling to detect failure to set either, or |
| 479 | // x=80 y=25 to provide defaults |
| 480 | |
| 481 | void terminal_size(unsigned *x, unsigned *y) |
| 482 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 483 | struct winsize ws; |
| 484 | int i; |
Rob Landley | 26e7b5e | 2012-02-02 07:27:35 -0600 | [diff] [blame] | 485 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 486 | //memset(&ws, 0, sizeof(ws)); |
| 487 | for (i=0; i<3; i++) { |
| 488 | if (ioctl(i, TIOCGWINSZ, &ws)) continue; |
| 489 | if (x) *x = ws.ws_col; |
| 490 | if (y) *y = ws.ws_row; |
| 491 | } |
| 492 | if (x) { |
| 493 | char *s = getenv("COLUMNS"); |
Rob Landley | 26e7b5e | 2012-02-02 07:27:35 -0600 | [diff] [blame] | 494 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 495 | i = s ? atoi(s) : 0; |
| 496 | if (i>0) *x = i; |
| 497 | } |
| 498 | if (y) { |
| 499 | char *s = getenv("ROWS"); |
Rob Landley | 26e7b5e | 2012-02-02 07:27:35 -0600 | [diff] [blame] | 500 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 501 | i = s ? atoi(s) : 0; |
| 502 | if (i>0) *y = i; |
| 503 | } |
Rob Landley | 26e7b5e | 2012-02-02 07:27:35 -0600 | [diff] [blame] | 504 | } |
| 505 | |
Rob Landley | f793d53 | 2012-02-27 21:56:49 -0600 | [diff] [blame] | 506 | int yesno(char *prompt, int def) |
Rob Landley | 26e7b5e | 2012-02-02 07:27:35 -0600 | [diff] [blame] | 507 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 508 | char buf; |
Rob Landley | 26e7b5e | 2012-02-02 07:27:35 -0600 | [diff] [blame] | 509 | |
Rob Landley | db8eb32 | 2012-12-08 02:25:32 -0600 | [diff] [blame] | 510 | fprintf(stderr, "%s (%c/%c):", prompt, def ? 'Y' : 'y', def ? 'n' : 'N'); |
| 511 | fflush(stderr); |
| 512 | while (fread(&buf, 1, 1, stdin)) { |
| 513 | int new; |
Rob Landley | 26e7b5e | 2012-02-02 07:27:35 -0600 | [diff] [blame] | 514 | |
Rob Landley | 2279108 | 2013-01-31 04:13:07 -0600 | [diff] [blame] | 515 | // The letter changes the value, the newline (or space) returns it. |
Rob Landley | db8eb32 | 2012-12-08 02:25:32 -0600 | [diff] [blame] | 516 | if (isspace(buf)) break; |
| 517 | if (-1 != (new = stridx("ny", tolower(buf)))) def = new; |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 518 | } |
Rob Landley | ee00a7f | 2012-03-19 19:19:21 -0500 | [diff] [blame] | 519 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 520 | return def; |
Rob Landley | 26e7b5e | 2012-02-02 07:27:35 -0600 | [diff] [blame] | 521 | } |
Rob Landley | ff9ee8f | 2012-02-18 15:12:41 -0600 | [diff] [blame] | 522 | |
Rob Landley | 2dd50ad | 2012-02-26 13:48:00 -0600 | [diff] [blame] | 523 | struct signame { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 524 | int num; |
| 525 | char *name; |
Rob Landley | 2dd50ad | 2012-02-26 13:48:00 -0600 | [diff] [blame] | 526 | }; |
| 527 | |
| 528 | // Signals required by POSIX 2008: |
| 529 | // http://pubs.opengroup.org/onlinepubs/9699919799/basedefs/signal.h.html |
| 530 | |
| 531 | #define SIGNIFY(x) {SIG##x, #x} |
| 532 | |
| 533 | static struct signame signames[] = { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 534 | SIGNIFY(ABRT), SIGNIFY(ALRM), SIGNIFY(BUS), |
| 535 | SIGNIFY(FPE), SIGNIFY(HUP), SIGNIFY(ILL), SIGNIFY(INT), SIGNIFY(KILL), |
| 536 | SIGNIFY(PIPE), SIGNIFY(QUIT), SIGNIFY(SEGV), SIGNIFY(TERM), |
| 537 | SIGNIFY(USR1), SIGNIFY(USR2), SIGNIFY(SYS), SIGNIFY(TRAP), |
| 538 | SIGNIFY(VTALRM), SIGNIFY(XCPU), SIGNIFY(XFSZ), |
Rob Landley | c52db60 | 2012-07-30 01:01:33 -0500 | [diff] [blame] | 539 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 540 | // Start of non-terminal signals |
Rob Landley | c52db60 | 2012-07-30 01:01:33 -0500 | [diff] [blame] | 541 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 542 | SIGNIFY(CHLD), SIGNIFY(CONT), SIGNIFY(STOP), SIGNIFY(TSTP), |
| 543 | SIGNIFY(TTIN), SIGNIFY(TTOU), SIGNIFY(URG) |
Rob Landley | 2dd50ad | 2012-02-26 13:48:00 -0600 | [diff] [blame] | 544 | }; |
| 545 | |
| 546 | // not in posix: SIGNIFY(STKFLT), SIGNIFY(WINCH), SIGNIFY(IO), SIGNIFY(PWR) |
| 547 | // obsolete: SIGNIFY(PROF) SIGNIFY(POLL) |
| 548 | |
Rob Landley | c52db60 | 2012-07-30 01:01:33 -0500 | [diff] [blame] | 549 | // Install the same handler on every signal that defaults to killing the process |
| 550 | void sigatexit(void *handler) |
| 551 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 552 | int i; |
| 553 | 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] | 554 | } |
Rob Landley | 2dd50ad | 2012-02-26 13:48:00 -0600 | [diff] [blame] | 555 | // Convert name to signal number. If name == NULL print names. |
| 556 | int sig_to_num(char *pidstr) |
| 557 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 558 | int i; |
Rob Landley | 2dd50ad | 2012-02-26 13:48:00 -0600 | [diff] [blame] | 559 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 560 | if (pidstr) { |
| 561 | char *s; |
| 562 | i = strtol(pidstr, &s, 10); |
| 563 | if (!*s) return i; |
Rob Landley | 2dd50ad | 2012-02-26 13:48:00 -0600 | [diff] [blame] | 564 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 565 | if (!strncasecmp(pidstr, "sig", 3)) pidstr+=3; |
| 566 | } |
| 567 | for (i = 0; i < sizeof(signames)/sizeof(struct signame); i++) |
| 568 | if (!pidstr) xputs(signames[i].name); |
| 569 | else if (!strcasecmp(pidstr, signames[i].name)) return signames[i].num; |
Rob Landley | 2dd50ad | 2012-02-26 13:48:00 -0600 | [diff] [blame] | 570 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 571 | return -1; |
Rob Landley | 2dd50ad | 2012-02-26 13:48:00 -0600 | [diff] [blame] | 572 | } |
| 573 | |
| 574 | char *num_to_sig(int sig) |
| 575 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 576 | int i; |
Rob Landley | 2dd50ad | 2012-02-26 13:48:00 -0600 | [diff] [blame] | 577 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 578 | for (i=0; i<sizeof(signames)/sizeof(struct signame); i++) |
| 579 | if (signames[i].num == sig) return signames[i].name; |
| 580 | return NULL; |
Rob Landley | 2dd50ad | 2012-02-26 13:48:00 -0600 | [diff] [blame] | 581 | } |
Daniel Walter | 05744b3 | 2012-03-19 19:57:56 -0500 | [diff] [blame] | 582 | |
Rob Landley | c52db60 | 2012-07-30 01:01:33 -0500 | [diff] [blame] | 583 | // premute mode bits based on posix mode strings. |
Rob Landley | b6f601e | 2012-05-16 21:11:43 -0500 | [diff] [blame] | 584 | mode_t string_to_mode(char *modestr, mode_t mode) |
Daniel Walter | 05744b3 | 2012-03-19 19:57:56 -0500 | [diff] [blame] | 585 | { |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 586 | char *whos = "ogua", *hows = "=+-", *whats = "xwrstX", *whys = "ogu"; |
| 587 | char *s, *str = modestr; |
Rob Landley | cf6bcb2 | 2012-03-19 20:56:18 -0500 | [diff] [blame] | 588 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 589 | // Handle octal mode |
| 590 | if (isdigit(*str)) { |
| 591 | mode = strtol(str, &s, 8); |
| 592 | if (*s || (mode & ~(07777))) goto barf; |
Rob Landley | b6f601e | 2012-05-16 21:11:43 -0500 | [diff] [blame] | 593 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 594 | return mode; |
| 595 | } |
Rob Landley | cf6bcb2 | 2012-03-19 20:56:18 -0500 | [diff] [blame] | 596 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 597 | // Gaze into the bin of permission... |
| 598 | for (;;) { |
| 599 | int i, j, dowho, dohow, dowhat, amask; |
Rob Landley | cf6bcb2 | 2012-03-19 20:56:18 -0500 | [diff] [blame] | 600 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 601 | dowho = dohow = dowhat = amask = 0; |
Daniel Walter | 05744b3 | 2012-03-19 19:57:56 -0500 | [diff] [blame] | 602 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 603 | // Find the who, how, and what stanzas, in that order |
| 604 | while (*str && (s = strchr(whos, *str))) { |
| 605 | dowho |= 1<<(s-whos); |
| 606 | str++; |
| 607 | } |
| 608 | // If who isn't specified, like "a" but honoring umask. |
| 609 | if (!dowho) { |
| 610 | dowho = 8; |
| 611 | umask(amask=umask(0)); |
| 612 | } |
| 613 | if (!*str || !(s = strchr(hows, *str))) goto barf; |
| 614 | dohow = *(str++); |
Rob Landley | 67a069d | 2012-06-03 00:32:12 -0500 | [diff] [blame] | 615 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 616 | if (!dohow) goto barf; |
| 617 | while (*str && (s = strchr(whats, *str))) { |
| 618 | dowhat |= 1<<(s-whats); |
| 619 | str++; |
| 620 | } |
Rob Landley | b6f601e | 2012-05-16 21:11:43 -0500 | [diff] [blame] | 621 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 622 | // Convert X to x for directory or if already executable somewhere |
| 623 | if ((dowhat&32) && (S_ISDIR(mode) || (mode&0111))) dowhat |= 1; |
Rob Landley | b6f601e | 2012-05-16 21:11:43 -0500 | [diff] [blame] | 624 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 625 | // Copy mode from another category? |
| 626 | if (!dowhat && *str && (s = strchr(whys, *str))) { |
| 627 | dowhat = (mode>>(3*(s-whys)))&7; |
| 628 | str++; |
| 629 | } |
Rob Landley | b6f601e | 2012-05-16 21:11:43 -0500 | [diff] [blame] | 630 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 631 | // Are we ready to do a thing yet? |
| 632 | if (*str && *(str++) != ',') goto barf; |
Rob Landley | b6f601e | 2012-05-16 21:11:43 -0500 | [diff] [blame] | 633 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 634 | // Ok, apply the bits to the mode. |
| 635 | for (i=0; i<4; i++) { |
| 636 | for (j=0; j<3; j++) { |
| 637 | mode_t bit = 0; |
| 638 | int where = 1<<((3*i)+j); |
Rob Landley | 31f49e7 | 2012-07-21 22:45:05 -0500 | [diff] [blame] | 639 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 640 | if (amask & where) continue; |
Rob Landley | b6f601e | 2012-05-16 21:11:43 -0500 | [diff] [blame] | 641 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 642 | // Figure out new value at this location |
| 643 | if (i == 3) { |
| 644 | // suid/sticky bit. |
| 645 | if (j) { |
| 646 | if ((dowhat & 8) && (dowho&(8|(1<<i)))) bit++; |
| 647 | } else if (dowhat & 16) bit++; |
| 648 | } else { |
| 649 | if (!(dowho&(8|(1<<i)))) continue; |
| 650 | if (dowhat&(1<<j)) bit++; |
| 651 | } |
Rob Landley | b6f601e | 2012-05-16 21:11:43 -0500 | [diff] [blame] | 652 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 653 | // When selection active, modify bit |
Rob Landley | b6f601e | 2012-05-16 21:11:43 -0500 | [diff] [blame] | 654 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 655 | if (dohow == '=' || (bit && dohow == '-')) mode &= ~where; |
| 656 | if (bit && dohow != '-') mode |= where; |
| 657 | } |
| 658 | } |
Rob Landley | b6f601e | 2012-05-16 21:11:43 -0500 | [diff] [blame] | 659 | |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 660 | if (!*str) break; |
| 661 | } |
| 662 | return mode; |
Rob Landley | b6f601e | 2012-05-16 21:11:43 -0500 | [diff] [blame] | 663 | barf: |
Rob Landley | 7aa651a | 2012-11-13 17:14:08 -0600 | [diff] [blame] | 664 | error_exit("bad mode '%s'", modestr); |
Daniel Walter | 05744b3 | 2012-03-19 19:57:56 -0500 | [diff] [blame] | 665 | } |
Ashwini Kumar | 1a0eedf | 2012-08-26 21:17:00 -0500 | [diff] [blame] | 666 | |
Rob Landley | 5a26a86 | 2013-06-02 00:24:24 -0500 | [diff] [blame] | 667 | // Format access mode into a drwxrwxrwx string |
| 668 | void mode_to_string(mode_t mode, char *buf) |
Felix Janda | 5a221e6 | 2013-04-22 22:29:43 +0200 | [diff] [blame] | 669 | { |
| 670 | char c, d; |
| 671 | int i, bit; |
| 672 | |
Rob Landley | 5a26a86 | 2013-06-02 00:24:24 -0500 | [diff] [blame] | 673 | buf[10]=0; |
Felix Janda | 5a221e6 | 2013-04-22 22:29:43 +0200 | [diff] [blame] | 674 | for (i=0; i<9; i++) { |
| 675 | bit = mode & (1<<i); |
| 676 | c = i%3; |
| 677 | if (!c && (mode & (1<<((d=i/3)+9)))) { |
| 678 | c = "tss"[d]; |
| 679 | if (!bit) c &= ~0x20; |
| 680 | } else c = bit ? "xwr"[c] : '-'; |
Rob Landley | 5a26a86 | 2013-06-02 00:24:24 -0500 | [diff] [blame] | 681 | buf[9-i] = c; |
Felix Janda | 5a221e6 | 2013-04-22 22:29:43 +0200 | [diff] [blame] | 682 | } |
| 683 | |
| 684 | if (S_ISDIR(mode)) c = 'd'; |
| 685 | else if (S_ISBLK(mode)) c = 'b'; |
| 686 | else if (S_ISCHR(mode)) c = 'c'; |
| 687 | else if (S_ISLNK(mode)) c = 'l'; |
| 688 | else if (S_ISFIFO(mode)) c = 'p'; |
| 689 | else if (S_ISSOCK(mode)) c = 's'; |
| 690 | else c = '-'; |
Rob Landley | 5a26a86 | 2013-06-02 00:24:24 -0500 | [diff] [blame] | 691 | *buf = c; |
Felix Janda | 5a221e6 | 2013-04-22 22:29:43 +0200 | [diff] [blame] | 692 | } |