wok-stable diff gen-init-cpio/stuff/gen_init_cpio.c @ rev 6460
Up: libwebkit (1.3.4)
author | Alexander Medvedev <devl547@gmail.com> |
---|---|
date | Wed Sep 29 22:50:12 2010 +0000 (2010-09-29) |
parents | |
children |
line diff
1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000 1.2 +++ b/gen-init-cpio/stuff/gen_init_cpio.c Wed Sep 29 22:50:12 2010 +0000 1.3 @@ -0,0 +1,592 @@ 1.4 +#include <stdio.h> 1.5 +#include <stdlib.h> 1.6 +#include <sys/types.h> 1.7 +#include <sys/stat.h> 1.8 +#include <string.h> 1.9 +#include <unistd.h> 1.10 +#include <time.h> 1.11 +#include <fcntl.h> 1.12 +#include <errno.h> 1.13 +#include <ctype.h> 1.14 +#include <limits.h> 1.15 + 1.16 +/* 1.17 + * Original work by Jeff Garzik 1.18 + * 1.19 + * External file lists, symlink, pipe and fifo support by Thayne Harbaugh 1.20 + * Hard link support by Luciano Rocha 1.21 + */ 1.22 + 1.23 +#define xstr(s) #s 1.24 +#define str(s) xstr(s) 1.25 + 1.26 +static unsigned int offset; 1.27 +static unsigned int ino = 721; 1.28 + 1.29 +struct file_handler { 1.30 + const char *type; 1.31 + int (*handler)(const char *line); 1.32 +}; 1.33 + 1.34 +static void push_string(const char *name) 1.35 +{ 1.36 + unsigned int name_len = strlen(name) + 1; 1.37 + 1.38 + fputs(name, stdout); 1.39 + putchar(0); 1.40 + offset += name_len; 1.41 +} 1.42 + 1.43 +static void push_pad (void) 1.44 +{ 1.45 + while (offset & 3) { 1.46 + putchar(0); 1.47 + offset++; 1.48 + } 1.49 +} 1.50 + 1.51 +static void push_rest(const char *name) 1.52 +{ 1.53 + unsigned int name_len = strlen(name) + 1; 1.54 + unsigned int tmp_ofs; 1.55 + 1.56 + fputs(name, stdout); 1.57 + putchar(0); 1.58 + offset += name_len; 1.59 + 1.60 + tmp_ofs = name_len + 110; 1.61 + while (tmp_ofs & 3) { 1.62 + putchar(0); 1.63 + offset++; 1.64 + tmp_ofs++; 1.65 + } 1.66 +} 1.67 + 1.68 +static void push_hdr(const char *s) 1.69 +{ 1.70 + fputs(s, stdout); 1.71 + offset += 110; 1.72 +} 1.73 + 1.74 +static void cpio_trailer(void) 1.75 +{ 1.76 + char s[256]; 1.77 + const char name[] = "TRAILER!!!"; 1.78 + 1.79 + sprintf(s, "%s%08X%08X%08lX%08lX%08X%08lX" 1.80 + "%08X%08X%08X%08X%08X%08X%08X", 1.81 + "070701", /* magic */ 1.82 + 0, /* ino */ 1.83 + 0, /* mode */ 1.84 + (long) 0, /* uid */ 1.85 + (long) 0, /* gid */ 1.86 + 1, /* nlink */ 1.87 + (long) 0, /* mtime */ 1.88 + 0, /* filesize */ 1.89 + 0, /* major */ 1.90 + 0, /* minor */ 1.91 + 0, /* rmajor */ 1.92 + 0, /* rminor */ 1.93 + (unsigned)strlen(name)+1, /* namesize */ 1.94 + 0); /* chksum */ 1.95 + push_hdr(s); 1.96 + push_rest(name); 1.97 + 1.98 + while (offset % 512) { 1.99 + putchar(0); 1.100 + offset++; 1.101 + } 1.102 +} 1.103 + 1.104 +static int cpio_mkslink(const char *name, const char *target, 1.105 + unsigned int mode, uid_t uid, gid_t gid) 1.106 +{ 1.107 + char s[256]; 1.108 + time_t mtime = time(NULL); 1.109 + 1.110 + sprintf(s,"%s%08X%08X%08lX%08lX%08X%08lX" 1.111 + "%08X%08X%08X%08X%08X%08X%08X", 1.112 + "070701", /* magic */ 1.113 + ino++, /* ino */ 1.114 + S_IFLNK | mode, /* mode */ 1.115 + (long) uid, /* uid */ 1.116 + (long) gid, /* gid */ 1.117 + 1, /* nlink */ 1.118 + (long) mtime, /* mtime */ 1.119 + (unsigned)strlen(target)+1, /* filesize */ 1.120 + 3, /* major */ 1.121 + 1, /* minor */ 1.122 + 0, /* rmajor */ 1.123 + 0, /* rminor */ 1.124 + (unsigned)strlen(name) + 1,/* namesize */ 1.125 + 0); /* chksum */ 1.126 + push_hdr(s); 1.127 + push_string(name); 1.128 + push_pad(); 1.129 + push_string(target); 1.130 + push_pad(); 1.131 + return 0; 1.132 +} 1.133 + 1.134 +static int cpio_mkslink_line(const char *line) 1.135 +{ 1.136 + char name[PATH_MAX + 1]; 1.137 + char target[PATH_MAX + 1]; 1.138 + unsigned int mode; 1.139 + int uid; 1.140 + int gid; 1.141 + int rc = -1; 1.142 + 1.143 + if (5 != sscanf(line, "%" str(PATH_MAX) "s %" str(PATH_MAX) "s %o %d %d", name, target, &mode, &uid, &gid)) { 1.144 + fprintf(stderr, "Unrecognized dir format '%s'", line); 1.145 + goto fail; 1.146 + } 1.147 + rc = cpio_mkslink(name, target, mode, uid, gid); 1.148 + fail: 1.149 + return rc; 1.150 +} 1.151 + 1.152 +static int cpio_mkgeneric(const char *name, unsigned int mode, 1.153 + uid_t uid, gid_t gid) 1.154 +{ 1.155 + char s[256]; 1.156 + time_t mtime = time(NULL); 1.157 + 1.158 + sprintf(s,"%s%08X%08X%08lX%08lX%08X%08lX" 1.159 + "%08X%08X%08X%08X%08X%08X%08X", 1.160 + "070701", /* magic */ 1.161 + ino++, /* ino */ 1.162 + mode, /* mode */ 1.163 + (long) uid, /* uid */ 1.164 + (long) gid, /* gid */ 1.165 + 2, /* nlink */ 1.166 + (long) mtime, /* mtime */ 1.167 + 0, /* filesize */ 1.168 + 3, /* major */ 1.169 + 1, /* minor */ 1.170 + 0, /* rmajor */ 1.171 + 0, /* rminor */ 1.172 + (unsigned)strlen(name) + 1,/* namesize */ 1.173 + 0); /* chksum */ 1.174 + push_hdr(s); 1.175 + push_rest(name); 1.176 + return 0; 1.177 +} 1.178 + 1.179 +enum generic_types { 1.180 + GT_DIR, 1.181 + GT_PIPE, 1.182 + GT_SOCK 1.183 +}; 1.184 + 1.185 +struct generic_type { 1.186 + const char *type; 1.187 + mode_t mode; 1.188 +}; 1.189 + 1.190 +static struct generic_type generic_type_table[] = { 1.191 + [GT_DIR] = { 1.192 + .type = "dir", 1.193 + .mode = S_IFDIR 1.194 + }, 1.195 + [GT_PIPE] = { 1.196 + .type = "pipe", 1.197 + .mode = S_IFIFO 1.198 + }, 1.199 + [GT_SOCK] = { 1.200 + .type = "sock", 1.201 + .mode = S_IFSOCK 1.202 + } 1.203 +}; 1.204 + 1.205 +static int cpio_mkgeneric_line(const char *line, enum generic_types gt) 1.206 +{ 1.207 + char name[PATH_MAX + 1]; 1.208 + unsigned int mode; 1.209 + int uid; 1.210 + int gid; 1.211 + int rc = -1; 1.212 + 1.213 + if (4 != sscanf(line, "%" str(PATH_MAX) "s %o %d %d", name, &mode, &uid, &gid)) { 1.214 + fprintf(stderr, "Unrecognized %s format '%s'", 1.215 + line, generic_type_table[gt].type); 1.216 + goto fail; 1.217 + } 1.218 + mode |= generic_type_table[gt].mode; 1.219 + rc = cpio_mkgeneric(name, mode, uid, gid); 1.220 + fail: 1.221 + return rc; 1.222 +} 1.223 + 1.224 +static int cpio_mkdir_line(const char *line) 1.225 +{ 1.226 + return cpio_mkgeneric_line(line, GT_DIR); 1.227 +} 1.228 + 1.229 +static int cpio_mkpipe_line(const char *line) 1.230 +{ 1.231 + return cpio_mkgeneric_line(line, GT_PIPE); 1.232 +} 1.233 + 1.234 +static int cpio_mksock_line(const char *line) 1.235 +{ 1.236 + return cpio_mkgeneric_line(line, GT_SOCK); 1.237 +} 1.238 + 1.239 +static int cpio_mknod(const char *name, unsigned int mode, 1.240 + uid_t uid, gid_t gid, char dev_type, 1.241 + unsigned int maj, unsigned int min) 1.242 +{ 1.243 + char s[256]; 1.244 + time_t mtime = time(NULL); 1.245 + 1.246 + if (dev_type == 'b') 1.247 + mode |= S_IFBLK; 1.248 + else 1.249 + mode |= S_IFCHR; 1.250 + 1.251 + sprintf(s,"%s%08X%08X%08lX%08lX%08X%08lX" 1.252 + "%08X%08X%08X%08X%08X%08X%08X", 1.253 + "070701", /* magic */ 1.254 + ino++, /* ino */ 1.255 + mode, /* mode */ 1.256 + (long) uid, /* uid */ 1.257 + (long) gid, /* gid */ 1.258 + 1, /* nlink */ 1.259 + (long) mtime, /* mtime */ 1.260 + 0, /* filesize */ 1.261 + 3, /* major */ 1.262 + 1, /* minor */ 1.263 + maj, /* rmajor */ 1.264 + min, /* rminor */ 1.265 + (unsigned)strlen(name) + 1,/* namesize */ 1.266 + 0); /* chksum */ 1.267 + push_hdr(s); 1.268 + push_rest(name); 1.269 + return 0; 1.270 +} 1.271 + 1.272 +static int cpio_mknod_line(const char *line) 1.273 +{ 1.274 + char name[PATH_MAX + 1]; 1.275 + unsigned int mode; 1.276 + int uid; 1.277 + int gid; 1.278 + char dev_type; 1.279 + unsigned int maj; 1.280 + unsigned int min; 1.281 + int rc = -1; 1.282 + 1.283 + if (7 != sscanf(line, "%" str(PATH_MAX) "s %o %d %d %c %u %u", 1.284 + name, &mode, &uid, &gid, &dev_type, &maj, &min)) { 1.285 + fprintf(stderr, "Unrecognized nod format '%s'", line); 1.286 + goto fail; 1.287 + } 1.288 + rc = cpio_mknod(name, mode, uid, gid, dev_type, maj, min); 1.289 + fail: 1.290 + return rc; 1.291 +} 1.292 + 1.293 +static int cpio_mkfile(const char *name, const char *location, 1.294 + unsigned int mode, uid_t uid, gid_t gid, 1.295 + unsigned int nlinks) 1.296 +{ 1.297 + char s[256]; 1.298 + char *filebuf = NULL; 1.299 + struct stat buf; 1.300 + long size; 1.301 + int file = -1; 1.302 + int retval; 1.303 + int rc = -1; 1.304 + int namesize; 1.305 + int i; 1.306 + 1.307 + mode |= S_IFREG; 1.308 + 1.309 + retval = stat (location, &buf); 1.310 + if (retval) { 1.311 + fprintf (stderr, "File %s could not be located\n", location); 1.312 + goto error; 1.313 + } 1.314 + 1.315 + file = open (location, O_RDONLY); 1.316 + if (file < 0) { 1.317 + fprintf (stderr, "File %s could not be opened for reading\n", location); 1.318 + goto error; 1.319 + } 1.320 + 1.321 + filebuf = malloc(buf.st_size); 1.322 + if (!filebuf) { 1.323 + fprintf (stderr, "out of memory\n"); 1.324 + goto error; 1.325 + } 1.326 + 1.327 + retval = read (file, filebuf, buf.st_size); 1.328 + if (retval < 0) { 1.329 + fprintf (stderr, "Can not read %s file\n", location); 1.330 + goto error; 1.331 + } 1.332 + 1.333 + size = 0; 1.334 + for (i = 1; i <= nlinks; i++) { 1.335 + /* data goes on last link */ 1.336 + if (i == nlinks) size = buf.st_size; 1.337 + 1.338 + namesize = strlen(name) + 1; 1.339 + sprintf(s,"%s%08X%08X%08lX%08lX%08X%08lX" 1.340 + "%08lX%08X%08X%08X%08X%08X%08X", 1.341 + "070701", /* magic */ 1.342 + ino, /* ino */ 1.343 + mode, /* mode */ 1.344 + (long) uid, /* uid */ 1.345 + (long) gid, /* gid */ 1.346 + nlinks, /* nlink */ 1.347 + (long) buf.st_mtime, /* mtime */ 1.348 + size, /* filesize */ 1.349 + 3, /* major */ 1.350 + 1, /* minor */ 1.351 + 0, /* rmajor */ 1.352 + 0, /* rminor */ 1.353 + namesize, /* namesize */ 1.354 + 0); /* chksum */ 1.355 + push_hdr(s); 1.356 + push_string(name); 1.357 + push_pad(); 1.358 + 1.359 + if (size) { 1.360 + fwrite(filebuf, size, 1, stdout); 1.361 + offset += size; 1.362 + push_pad(); 1.363 + } 1.364 + 1.365 + name += namesize; 1.366 + } 1.367 + ino++; 1.368 + rc = 0; 1.369 + 1.370 +error: 1.371 + if (filebuf) free(filebuf); 1.372 + if (file >= 0) close(file); 1.373 + return rc; 1.374 +} 1.375 + 1.376 +static char *cpio_replace_env(char *new_location) 1.377 +{ 1.378 + char expanded[PATH_MAX + 1]; 1.379 + char env_var[PATH_MAX + 1]; 1.380 + char *start; 1.381 + char *end; 1.382 + 1.383 + for (start = NULL; (start = strstr(new_location, "${")); ) { 1.384 + end = strchr(start, '}'); 1.385 + if (start < end) { 1.386 + *env_var = *expanded = '\0'; 1.387 + strncat(env_var, start + 2, end - start - 2); 1.388 + strncat(expanded, new_location, start - new_location); 1.389 + strncat(expanded, getenv(env_var), PATH_MAX); 1.390 + strncat(expanded, end + 1, PATH_MAX); 1.391 + strncpy(new_location, expanded, PATH_MAX); 1.392 + } else 1.393 + break; 1.394 + } 1.395 + 1.396 + return new_location; 1.397 +} 1.398 + 1.399 + 1.400 +static int cpio_mkfile_line(const char *line) 1.401 +{ 1.402 + char name[PATH_MAX + 1]; 1.403 + char *dname = NULL; /* malloc'ed buffer for hard links */ 1.404 + char location[PATH_MAX + 1]; 1.405 + unsigned int mode; 1.406 + int uid; 1.407 + int gid; 1.408 + int nlinks = 1; 1.409 + int end = 0, dname_len = 0; 1.410 + int rc = -1; 1.411 + 1.412 + if (5 > sscanf(line, "%" str(PATH_MAX) "s %" str(PATH_MAX) 1.413 + "s %o %d %d %n", 1.414 + name, location, &mode, &uid, &gid, &end)) { 1.415 + fprintf(stderr, "Unrecognized file format '%s'", line); 1.416 + goto fail; 1.417 + } 1.418 + if (end && isgraph(line[end])) { 1.419 + int len; 1.420 + int nend; 1.421 + 1.422 + dname = malloc(strlen(line)); 1.423 + if (!dname) { 1.424 + fprintf (stderr, "out of memory (%d)\n", dname_len); 1.425 + goto fail; 1.426 + } 1.427 + 1.428 + dname_len = strlen(name) + 1; 1.429 + memcpy(dname, name, dname_len); 1.430 + 1.431 + do { 1.432 + nend = 0; 1.433 + if (sscanf(line + end, "%" str(PATH_MAX) "s %n", 1.434 + name, &nend) < 1) 1.435 + break; 1.436 + len = strlen(name) + 1; 1.437 + memcpy(dname + dname_len, name, len); 1.438 + dname_len += len; 1.439 + nlinks++; 1.440 + end += nend; 1.441 + } while (isgraph(line[end])); 1.442 + } else { 1.443 + dname = name; 1.444 + } 1.445 + rc = cpio_mkfile(dname, cpio_replace_env(location), 1.446 + mode, uid, gid, nlinks); 1.447 + fail: 1.448 + if (dname_len) free(dname); 1.449 + return rc; 1.450 +} 1.451 + 1.452 +static void usage(const char *prog) 1.453 +{ 1.454 + fprintf(stderr, "Usage:\n" 1.455 + "\t%s <cpio_list>\n" 1.456 + "\n" 1.457 + "<cpio_list> is a file containing newline separated entries that\n" 1.458 + "describe the files to be included in the initramfs archive:\n" 1.459 + "\n" 1.460 + "# a comment\n" 1.461 + "file <name> <location> <mode> <uid> <gid> [<hard links>]\n" 1.462 + "dir <name> <mode> <uid> <gid>\n" 1.463 + "nod <name> <mode> <uid> <gid> <dev_type> <maj> <min>\n" 1.464 + "slink <name> <target> <mode> <uid> <gid>\n" 1.465 + "pipe <name> <mode> <uid> <gid>\n" 1.466 + "sock <name> <mode> <uid> <gid>\n" 1.467 + "\n" 1.468 + "<name> name of the file/dir/nod/etc in the archive\n" 1.469 + "<location> location of the file in the current filesystem\n" 1.470 + " expands shell variables quoted with ${}\n" 1.471 + "<target> link target\n" 1.472 + "<mode> mode/permissions of the file\n" 1.473 + "<uid> user id (0=root)\n" 1.474 + "<gid> group id (0=root)\n" 1.475 + "<dev_type> device type (b=block, c=character)\n" 1.476 + "<maj> major number of nod\n" 1.477 + "<min> minor number of nod\n" 1.478 + "<hard links> space separated list of other links to file\n" 1.479 + "\n" 1.480 + "example:\n" 1.481 + "# A simple initramfs\n" 1.482 + "dir /dev 0755 0 0\n" 1.483 + "nod /dev/console 0600 0 0 c 5 1\n" 1.484 + "dir /root 0700 0 0\n" 1.485 + "dir /sbin 0755 0 0\n" 1.486 + "file /sbin/kinit /usr/src/klibc/kinit/kinit 0755 0 0\n", 1.487 + prog); 1.488 +} 1.489 + 1.490 +struct file_handler file_handler_table[] = { 1.491 + { 1.492 + .type = "file", 1.493 + .handler = cpio_mkfile_line, 1.494 + }, { 1.495 + .type = "nod", 1.496 + .handler = cpio_mknod_line, 1.497 + }, { 1.498 + .type = "dir", 1.499 + .handler = cpio_mkdir_line, 1.500 + }, { 1.501 + .type = "slink", 1.502 + .handler = cpio_mkslink_line, 1.503 + }, { 1.504 + .type = "pipe", 1.505 + .handler = cpio_mkpipe_line, 1.506 + }, { 1.507 + .type = "sock", 1.508 + .handler = cpio_mksock_line, 1.509 + }, { 1.510 + .type = NULL, 1.511 + .handler = NULL, 1.512 + } 1.513 +}; 1.514 + 1.515 +#define LINE_SIZE (2 * PATH_MAX + 50) 1.516 + 1.517 +int main (int argc, char *argv[]) 1.518 +{ 1.519 + FILE *cpio_list; 1.520 + char line[LINE_SIZE]; 1.521 + char *args, *type; 1.522 + int ec = 0; 1.523 + int line_nr = 0; 1.524 + 1.525 + if (2 != argc) { 1.526 + usage(argv[0]); 1.527 + exit(1); 1.528 + } 1.529 + 1.530 + if (!strcmp(argv[1], "-")) 1.531 + cpio_list = stdin; 1.532 + else if (! (cpio_list = fopen(argv[1], "r"))) { 1.533 + fprintf(stderr, "ERROR: unable to open '%s': %s\n\n", 1.534 + argv[1], strerror(errno)); 1.535 + usage(argv[0]); 1.536 + exit(1); 1.537 + } 1.538 + 1.539 + while (fgets(line, LINE_SIZE, cpio_list)) { 1.540 + int type_idx; 1.541 + size_t slen = strlen(line); 1.542 + 1.543 + line_nr++; 1.544 + 1.545 + if ('#' == *line) { 1.546 + /* comment - skip to next line */ 1.547 + continue; 1.548 + } 1.549 + 1.550 + if (! (type = strtok(line, " \t"))) { 1.551 + fprintf(stderr, 1.552 + "ERROR: incorrect format, could not locate file type line %d: '%s'\n", 1.553 + line_nr, line); 1.554 + ec = -1; 1.555 + break; 1.556 + } 1.557 + 1.558 + if ('\n' == *type) { 1.559 + /* a blank line */ 1.560 + continue; 1.561 + } 1.562 + 1.563 + if (slen == strlen(type)) { 1.564 + /* must be an empty line */ 1.565 + continue; 1.566 + } 1.567 + 1.568 + if (! (args = strtok(NULL, "\n"))) { 1.569 + fprintf(stderr, 1.570 + "ERROR: incorrect format, newline required line %d: '%s'\n", 1.571 + line_nr, line); 1.572 + ec = -1; 1.573 + } 1.574 + 1.575 + for (type_idx = 0; file_handler_table[type_idx].type; type_idx++) { 1.576 + int rc; 1.577 + if (! strcmp(line, file_handler_table[type_idx].type)) { 1.578 + if ((rc = file_handler_table[type_idx].handler(args))) { 1.579 + ec = rc; 1.580 + fprintf(stderr, " line %d\n", line_nr); 1.581 + } 1.582 + break; 1.583 + } 1.584 + } 1.585 + 1.586 + if (NULL == file_handler_table[type_idx].type) { 1.587 + fprintf(stderr, "unknown file type line %d: '%s'\n", 1.588 + line_nr, line); 1.589 + } 1.590 + } 1.591 + if (ec == 0) 1.592 + cpio_trailer(); 1.593 + 1.594 + exit(ec); 1.595 +}