wok-6.x view gen-init-cpio/stuff/gen_init_cpio.c @ rev 6193

Up: buildbot to 0.8.1.
author Christopher Rogers <slaxemulator@gmail.com>
date Fri Sep 10 21:44:31 2010 +0000 (2010-09-10)
parents
children
line source
1 #include <stdio.h>
2 #include <stdlib.h>
3 #include <sys/types.h>
4 #include <sys/stat.h>
5 #include <string.h>
6 #include <unistd.h>
7 #include <time.h>
8 #include <fcntl.h>
9 #include <errno.h>
10 #include <ctype.h>
11 #include <limits.h>
12
13 /*
14 * Original work by Jeff Garzik
15 *
16 * External file lists, symlink, pipe and fifo support by Thayne Harbaugh
17 * Hard link support by Luciano Rocha
18 */
19
20 #define xstr(s) #s
21 #define str(s) xstr(s)
22
23 static unsigned int offset;
24 static unsigned int ino = 721;
25
26 struct file_handler {
27 const char *type;
28 int (*handler)(const char *line);
29 };
30
31 static void push_string(const char *name)
32 {
33 unsigned int name_len = strlen(name) + 1;
34
35 fputs(name, stdout);
36 putchar(0);
37 offset += name_len;
38 }
39
40 static void push_pad (void)
41 {
42 while (offset & 3) {
43 putchar(0);
44 offset++;
45 }
46 }
47
48 static void push_rest(const char *name)
49 {
50 unsigned int name_len = strlen(name) + 1;
51 unsigned int tmp_ofs;
52
53 fputs(name, stdout);
54 putchar(0);
55 offset += name_len;
56
57 tmp_ofs = name_len + 110;
58 while (tmp_ofs & 3) {
59 putchar(0);
60 offset++;
61 tmp_ofs++;
62 }
63 }
64
65 static void push_hdr(const char *s)
66 {
67 fputs(s, stdout);
68 offset += 110;
69 }
70
71 static void cpio_trailer(void)
72 {
73 char s[256];
74 const char name[] = "TRAILER!!!";
75
76 sprintf(s, "%s%08X%08X%08lX%08lX%08X%08lX"
77 "%08X%08X%08X%08X%08X%08X%08X",
78 "070701", /* magic */
79 0, /* ino */
80 0, /* mode */
81 (long) 0, /* uid */
82 (long) 0, /* gid */
83 1, /* nlink */
84 (long) 0, /* mtime */
85 0, /* filesize */
86 0, /* major */
87 0, /* minor */
88 0, /* rmajor */
89 0, /* rminor */
90 (unsigned)strlen(name)+1, /* namesize */
91 0); /* chksum */
92 push_hdr(s);
93 push_rest(name);
94
95 while (offset % 512) {
96 putchar(0);
97 offset++;
98 }
99 }
100
101 static int cpio_mkslink(const char *name, const char *target,
102 unsigned int mode, uid_t uid, gid_t gid)
103 {
104 char s[256];
105 time_t mtime = time(NULL);
106
107 sprintf(s,"%s%08X%08X%08lX%08lX%08X%08lX"
108 "%08X%08X%08X%08X%08X%08X%08X",
109 "070701", /* magic */
110 ino++, /* ino */
111 S_IFLNK | mode, /* mode */
112 (long) uid, /* uid */
113 (long) gid, /* gid */
114 1, /* nlink */
115 (long) mtime, /* mtime */
116 (unsigned)strlen(target)+1, /* filesize */
117 3, /* major */
118 1, /* minor */
119 0, /* rmajor */
120 0, /* rminor */
121 (unsigned)strlen(name) + 1,/* namesize */
122 0); /* chksum */
123 push_hdr(s);
124 push_string(name);
125 push_pad();
126 push_string(target);
127 push_pad();
128 return 0;
129 }
130
131 static int cpio_mkslink_line(const char *line)
132 {
133 char name[PATH_MAX + 1];
134 char target[PATH_MAX + 1];
135 unsigned int mode;
136 int uid;
137 int gid;
138 int rc = -1;
139
140 if (5 != sscanf(line, "%" str(PATH_MAX) "s %" str(PATH_MAX) "s %o %d %d", name, target, &mode, &uid, &gid)) {
141 fprintf(stderr, "Unrecognized dir format '%s'", line);
142 goto fail;
143 }
144 rc = cpio_mkslink(name, target, mode, uid, gid);
145 fail:
146 return rc;
147 }
148
149 static int cpio_mkgeneric(const char *name, unsigned int mode,
150 uid_t uid, gid_t gid)
151 {
152 char s[256];
153 time_t mtime = time(NULL);
154
155 sprintf(s,"%s%08X%08X%08lX%08lX%08X%08lX"
156 "%08X%08X%08X%08X%08X%08X%08X",
157 "070701", /* magic */
158 ino++, /* ino */
159 mode, /* mode */
160 (long) uid, /* uid */
161 (long) gid, /* gid */
162 2, /* nlink */
163 (long) mtime, /* mtime */
164 0, /* filesize */
165 3, /* major */
166 1, /* minor */
167 0, /* rmajor */
168 0, /* rminor */
169 (unsigned)strlen(name) + 1,/* namesize */
170 0); /* chksum */
171 push_hdr(s);
172 push_rest(name);
173 return 0;
174 }
175
176 enum generic_types {
177 GT_DIR,
178 GT_PIPE,
179 GT_SOCK
180 };
181
182 struct generic_type {
183 const char *type;
184 mode_t mode;
185 };
186
187 static struct generic_type generic_type_table[] = {
188 [GT_DIR] = {
189 .type = "dir",
190 .mode = S_IFDIR
191 },
192 [GT_PIPE] = {
193 .type = "pipe",
194 .mode = S_IFIFO
195 },
196 [GT_SOCK] = {
197 .type = "sock",
198 .mode = S_IFSOCK
199 }
200 };
201
202 static int cpio_mkgeneric_line(const char *line, enum generic_types gt)
203 {
204 char name[PATH_MAX + 1];
205 unsigned int mode;
206 int uid;
207 int gid;
208 int rc = -1;
209
210 if (4 != sscanf(line, "%" str(PATH_MAX) "s %o %d %d", name, &mode, &uid, &gid)) {
211 fprintf(stderr, "Unrecognized %s format '%s'",
212 line, generic_type_table[gt].type);
213 goto fail;
214 }
215 mode |= generic_type_table[gt].mode;
216 rc = cpio_mkgeneric(name, mode, uid, gid);
217 fail:
218 return rc;
219 }
220
221 static int cpio_mkdir_line(const char *line)
222 {
223 return cpio_mkgeneric_line(line, GT_DIR);
224 }
225
226 static int cpio_mkpipe_line(const char *line)
227 {
228 return cpio_mkgeneric_line(line, GT_PIPE);
229 }
230
231 static int cpio_mksock_line(const char *line)
232 {
233 return cpio_mkgeneric_line(line, GT_SOCK);
234 }
235
236 static int cpio_mknod(const char *name, unsigned int mode,
237 uid_t uid, gid_t gid, char dev_type,
238 unsigned int maj, unsigned int min)
239 {
240 char s[256];
241 time_t mtime = time(NULL);
242
243 if (dev_type == 'b')
244 mode |= S_IFBLK;
245 else
246 mode |= S_IFCHR;
247
248 sprintf(s,"%s%08X%08X%08lX%08lX%08X%08lX"
249 "%08X%08X%08X%08X%08X%08X%08X",
250 "070701", /* magic */
251 ino++, /* ino */
252 mode, /* mode */
253 (long) uid, /* uid */
254 (long) gid, /* gid */
255 1, /* nlink */
256 (long) mtime, /* mtime */
257 0, /* filesize */
258 3, /* major */
259 1, /* minor */
260 maj, /* rmajor */
261 min, /* rminor */
262 (unsigned)strlen(name) + 1,/* namesize */
263 0); /* chksum */
264 push_hdr(s);
265 push_rest(name);
266 return 0;
267 }
268
269 static int cpio_mknod_line(const char *line)
270 {
271 char name[PATH_MAX + 1];
272 unsigned int mode;
273 int uid;
274 int gid;
275 char dev_type;
276 unsigned int maj;
277 unsigned int min;
278 int rc = -1;
279
280 if (7 != sscanf(line, "%" str(PATH_MAX) "s %o %d %d %c %u %u",
281 name, &mode, &uid, &gid, &dev_type, &maj, &min)) {
282 fprintf(stderr, "Unrecognized nod format '%s'", line);
283 goto fail;
284 }
285 rc = cpio_mknod(name, mode, uid, gid, dev_type, maj, min);
286 fail:
287 return rc;
288 }
289
290 static int cpio_mkfile(const char *name, const char *location,
291 unsigned int mode, uid_t uid, gid_t gid,
292 unsigned int nlinks)
293 {
294 char s[256];
295 char *filebuf = NULL;
296 struct stat buf;
297 long size;
298 int file = -1;
299 int retval;
300 int rc = -1;
301 int namesize;
302 int i;
303
304 mode |= S_IFREG;
305
306 retval = stat (location, &buf);
307 if (retval) {
308 fprintf (stderr, "File %s could not be located\n", location);
309 goto error;
310 }
311
312 file = open (location, O_RDONLY);
313 if (file < 0) {
314 fprintf (stderr, "File %s could not be opened for reading\n", location);
315 goto error;
316 }
317
318 filebuf = malloc(buf.st_size);
319 if (!filebuf) {
320 fprintf (stderr, "out of memory\n");
321 goto error;
322 }
323
324 retval = read (file, filebuf, buf.st_size);
325 if (retval < 0) {
326 fprintf (stderr, "Can not read %s file\n", location);
327 goto error;
328 }
329
330 size = 0;
331 for (i = 1; i <= nlinks; i++) {
332 /* data goes on last link */
333 if (i == nlinks) size = buf.st_size;
334
335 namesize = strlen(name) + 1;
336 sprintf(s,"%s%08X%08X%08lX%08lX%08X%08lX"
337 "%08lX%08X%08X%08X%08X%08X%08X",
338 "070701", /* magic */
339 ino, /* ino */
340 mode, /* mode */
341 (long) uid, /* uid */
342 (long) gid, /* gid */
343 nlinks, /* nlink */
344 (long) buf.st_mtime, /* mtime */
345 size, /* filesize */
346 3, /* major */
347 1, /* minor */
348 0, /* rmajor */
349 0, /* rminor */
350 namesize, /* namesize */
351 0); /* chksum */
352 push_hdr(s);
353 push_string(name);
354 push_pad();
355
356 if (size) {
357 fwrite(filebuf, size, 1, stdout);
358 offset += size;
359 push_pad();
360 }
361
362 name += namesize;
363 }
364 ino++;
365 rc = 0;
366
367 error:
368 if (filebuf) free(filebuf);
369 if (file >= 0) close(file);
370 return rc;
371 }
372
373 static char *cpio_replace_env(char *new_location)
374 {
375 char expanded[PATH_MAX + 1];
376 char env_var[PATH_MAX + 1];
377 char *start;
378 char *end;
379
380 for (start = NULL; (start = strstr(new_location, "${")); ) {
381 end = strchr(start, '}');
382 if (start < end) {
383 *env_var = *expanded = '\0';
384 strncat(env_var, start + 2, end - start - 2);
385 strncat(expanded, new_location, start - new_location);
386 strncat(expanded, getenv(env_var), PATH_MAX);
387 strncat(expanded, end + 1, PATH_MAX);
388 strncpy(new_location, expanded, PATH_MAX);
389 } else
390 break;
391 }
392
393 return new_location;
394 }
395
396
397 static int cpio_mkfile_line(const char *line)
398 {
399 char name[PATH_MAX + 1];
400 char *dname = NULL; /* malloc'ed buffer for hard links */
401 char location[PATH_MAX + 1];
402 unsigned int mode;
403 int uid;
404 int gid;
405 int nlinks = 1;
406 int end = 0, dname_len = 0;
407 int rc = -1;
408
409 if (5 > sscanf(line, "%" str(PATH_MAX) "s %" str(PATH_MAX)
410 "s %o %d %d %n",
411 name, location, &mode, &uid, &gid, &end)) {
412 fprintf(stderr, "Unrecognized file format '%s'", line);
413 goto fail;
414 }
415 if (end && isgraph(line[end])) {
416 int len;
417 int nend;
418
419 dname = malloc(strlen(line));
420 if (!dname) {
421 fprintf (stderr, "out of memory (%d)\n", dname_len);
422 goto fail;
423 }
424
425 dname_len = strlen(name) + 1;
426 memcpy(dname, name, dname_len);
427
428 do {
429 nend = 0;
430 if (sscanf(line + end, "%" str(PATH_MAX) "s %n",
431 name, &nend) < 1)
432 break;
433 len = strlen(name) + 1;
434 memcpy(dname + dname_len, name, len);
435 dname_len += len;
436 nlinks++;
437 end += nend;
438 } while (isgraph(line[end]));
439 } else {
440 dname = name;
441 }
442 rc = cpio_mkfile(dname, cpio_replace_env(location),
443 mode, uid, gid, nlinks);
444 fail:
445 if (dname_len) free(dname);
446 return rc;
447 }
448
449 static void usage(const char *prog)
450 {
451 fprintf(stderr, "Usage:\n"
452 "\t%s <cpio_list>\n"
453 "\n"
454 "<cpio_list> is a file containing newline separated entries that\n"
455 "describe the files to be included in the initramfs archive:\n"
456 "\n"
457 "# a comment\n"
458 "file <name> <location> <mode> <uid> <gid> [<hard links>]\n"
459 "dir <name> <mode> <uid> <gid>\n"
460 "nod <name> <mode> <uid> <gid> <dev_type> <maj> <min>\n"
461 "slink <name> <target> <mode> <uid> <gid>\n"
462 "pipe <name> <mode> <uid> <gid>\n"
463 "sock <name> <mode> <uid> <gid>\n"
464 "\n"
465 "<name> name of the file/dir/nod/etc in the archive\n"
466 "<location> location of the file in the current filesystem\n"
467 " expands shell variables quoted with ${}\n"
468 "<target> link target\n"
469 "<mode> mode/permissions of the file\n"
470 "<uid> user id (0=root)\n"
471 "<gid> group id (0=root)\n"
472 "<dev_type> device type (b=block, c=character)\n"
473 "<maj> major number of nod\n"
474 "<min> minor number of nod\n"
475 "<hard links> space separated list of other links to file\n"
476 "\n"
477 "example:\n"
478 "# A simple initramfs\n"
479 "dir /dev 0755 0 0\n"
480 "nod /dev/console 0600 0 0 c 5 1\n"
481 "dir /root 0700 0 0\n"
482 "dir /sbin 0755 0 0\n"
483 "file /sbin/kinit /usr/src/klibc/kinit/kinit 0755 0 0\n",
484 prog);
485 }
486
487 struct file_handler file_handler_table[] = {
488 {
489 .type = "file",
490 .handler = cpio_mkfile_line,
491 }, {
492 .type = "nod",
493 .handler = cpio_mknod_line,
494 }, {
495 .type = "dir",
496 .handler = cpio_mkdir_line,
497 }, {
498 .type = "slink",
499 .handler = cpio_mkslink_line,
500 }, {
501 .type = "pipe",
502 .handler = cpio_mkpipe_line,
503 }, {
504 .type = "sock",
505 .handler = cpio_mksock_line,
506 }, {
507 .type = NULL,
508 .handler = NULL,
509 }
510 };
511
512 #define LINE_SIZE (2 * PATH_MAX + 50)
513
514 int main (int argc, char *argv[])
515 {
516 FILE *cpio_list;
517 char line[LINE_SIZE];
518 char *args, *type;
519 int ec = 0;
520 int line_nr = 0;
521
522 if (2 != argc) {
523 usage(argv[0]);
524 exit(1);
525 }
526
527 if (!strcmp(argv[1], "-"))
528 cpio_list = stdin;
529 else if (! (cpio_list = fopen(argv[1], "r"))) {
530 fprintf(stderr, "ERROR: unable to open '%s': %s\n\n",
531 argv[1], strerror(errno));
532 usage(argv[0]);
533 exit(1);
534 }
535
536 while (fgets(line, LINE_SIZE, cpio_list)) {
537 int type_idx;
538 size_t slen = strlen(line);
539
540 line_nr++;
541
542 if ('#' == *line) {
543 /* comment - skip to next line */
544 continue;
545 }
546
547 if (! (type = strtok(line, " \t"))) {
548 fprintf(stderr,
549 "ERROR: incorrect format, could not locate file type line %d: '%s'\n",
550 line_nr, line);
551 ec = -1;
552 break;
553 }
554
555 if ('\n' == *type) {
556 /* a blank line */
557 continue;
558 }
559
560 if (slen == strlen(type)) {
561 /* must be an empty line */
562 continue;
563 }
564
565 if (! (args = strtok(NULL, "\n"))) {
566 fprintf(stderr,
567 "ERROR: incorrect format, newline required line %d: '%s'\n",
568 line_nr, line);
569 ec = -1;
570 }
571
572 for (type_idx = 0; file_handler_table[type_idx].type; type_idx++) {
573 int rc;
574 if (! strcmp(line, file_handler_table[type_idx].type)) {
575 if ((rc = file_handler_table[type_idx].handler(args))) {
576 ec = rc;
577 fprintf(stderr, " line %d\n", line_nr);
578 }
579 break;
580 }
581 }
582
583 if (NULL == file_handler_table[type_idx].type) {
584 fprintf(stderr, "unknown file type line %d: '%s'\n",
585 line_nr, line);
586 }
587 }
588 if (ec == 0)
589 cpio_trailer();
590
591 exit(ec);
592 }