build/mkarchive.c

   1 /*
   2  * Copyright (c) 2011 Josef 'Jeff' Sipek
   3  */
   4 #include <stdio.h>
   5 #include <stdlib.h>
   6 #include <string.h>
   7 #include <assert.h>
   8 #include <libgen.h>
   9
  10 typedef unsigned char u8;
  11
  12 struct cpio_hdr {
  13         u8      magic[6];
  14         u8      dev[6];
  15         u8      ino[6];
  16         u8      mode[6];
  17         u8      uid[6];
  18         u8      gid[6];
  19         u8      nlink[6];
  20         u8      rdev[6];
  21         u8      mtime[11];
  22         u8      namesize[6];
  23         u8      filesize[11];
  24         u8      data[0];
  25 };
  26
  27 struct buffer {
  28         struct buffer *next;
  29         int     len;
  30         u8      data[0];
  31 };
  32
  33 static void storeoct(u8 *buf, int len, unsigned val)
  34 {
  35         len--;
  36
  37         while(val) {
  38                 buf[len] = '0' + (val % 8);
  39                 val /= 8;
  40                 len--;
  41         }
  42 }
  43
  44 static void mkhdr(struct cpio_hdr *hdr, char *fn, unsigned filesize)
  45 {
  46         memset(hdr, '0', sizeof(struct cpio_hdr));
  47
  48         hdr->magic[1] = '7';
  49         hdr->magic[3] = '7';
  50         hdr->magic[5] = '7';
  51
  52         hdr->nlink[5] = '1';
  53
  54         storeoct(hdr->namesize, 6, strlen(fn)+1);
  55         storeoct(hdr->filesize, 11, filesize);
  56 }
  57
  58 static unsigned __text(char *fn, int lrecl)
  59 {
  60         struct cpio_hdr hdr;
  61         struct buffer *first, *last, *cur, *tmp;
  62         FILE *f;
  63         int l;
  64         unsigned flen;
  65
  66         if (!lrecl) {
  67                 fprintf(stderr, "Error: zero lrecl is bogus\n");
  68                 exit(1);
  69         }
  70
  71         f = fopen(fn, "rb");
  72         if (!f) {
  73                 fprintf(stderr, "Error: could not open file '%s'\n",
  74                         fn);
  75                 exit(1);
  76         }
  77
  78         flen  = 0;
  79         first = NULL;
  80         last  = NULL;
  81         for(;;) {
  82                 cur = malloc(sizeof(struct buffer) + lrecl + 1);
  83                 if (!cur) {
  84                         fprintf(stderr, "Error: OOM\n");
  85                         exit(2);
  86                 }
  87
  88                 if (!fgets((char*)cur->data, lrecl+1, f))
  89                         break;
  90
  91                 l = strlen((char*)cur->data);
  92
  93                 if (l && cur->data[l-1])
  94                         l--;
  95
  96                 assert(l <= lrecl);
  97
  98                 memset(cur->data+l, ' ', lrecl-l);
  99
 100                 cur->next = NULL;
 101                 if (!first)
 102                         first = cur;
 103                 if (last)
 104                         last->next = cur;
 105                 last = cur;
 106
 107                 flen += lrecl;
 108         }
 109
 110         fclose(f);
 111
 112         mkhdr(&hdr, basename(fn), flen);
 113         fwrite(&hdr, sizeof(hdr), 1, stdout);
 114         fwrite(basename(fn), strlen(basename(fn))+1, 1, stdout);
 115
 116         for(cur=first; cur; cur=tmp) {
 117                 fwrite(cur->data, 1, lrecl, stdout);
 118                 tmp = cur->next;
 119                 free(cur);
 120         }
 121
 122         return flen + strlen(basename(fn)) + 1 + sizeof(hdr);
 123 }
 124
 125 #define BUF_LEN 4096
 126 static unsigned __bin(char *fn)
 127 {
 128         struct cpio_hdr hdr;
 129         struct buffer *first, *last, *cur, *tmp;
 130         FILE *f;
 131         unsigned flen;
 132         size_t s;
 133
 134         f = fopen(fn, "rb");
 135         if (!f) {
 136                 fprintf(stderr, "Error: could not open file '%s'\n",
 137                         fn);
 138                 exit(1);
 139         }
 140
 141         flen  = 0;
 142         first = NULL;
 143         last  = NULL;
 144         for(;;) {
 145                 cur = malloc(sizeof(struct buffer) + BUF_LEN);
 146                 if (!cur) {
 147                         fprintf(stderr, "Error: OOM\n");
 148                         exit(2);
 149                 }
 150
 151                 memset(cur->data, 0, BUF_LEN);
 152                 if (!(s = fread(cur->data, 1, BUF_LEN, f)))
 153                         break;
 154
 155                 cur->next = NULL;
 156                 cur->len  = s;
 157                 if (!first)
 158                         first = cur;
 159                 if (last)
 160                         last->next = cur;
 161                 last = cur;
 162
 163                 flen += s;
 164         }
 165
 166         fclose(f);
 167
 168         mkhdr(&hdr, basename(fn), flen);
 169         fwrite(&hdr, sizeof(hdr), 1, stdout);
 170         fwrite(basename(fn), strlen(basename(fn))+1, 1, stdout);
 171
 172         for(cur=first; cur; cur=tmp) {
 173                 fwrite(cur->data, 1, cur->len, stdout);
 174                 tmp = cur->next;
 175                 free(cur);
 176         }
 177
 178         return flen + strlen(basename(fn)) + 1 + sizeof(hdr);
 179 }
 180
 181 static void mktail(unsigned len)
 182 {
 183         struct cpio_hdr hdr;
 184
 185         mkhdr(&hdr, "TRAILER!!!", 0);
 186         fwrite(&hdr, sizeof(hdr), 1, stdout);
 187         printf("TRAILER!!!");
 188
 189         while(len % 512) {
 190                 fwrite("", 1, 1, stdout); // write a nul char
 191                 len++;
 192         }
 193 }
 194
 195 int main(int argc, char **argv)
 196 {
 197         int lrecl;
 198         char *fn;
 199         int i;
 200         unsigned len;
 201
 202         i=1;
 203         len=0;
 204         while((i+1) < argc) {
 205                 fn = argv[i];
 206                 i++;
 207
 208                 if (!strcmp(argv[i], "text")) {
 209                         lrecl = atoi(argv[i+1]);
 210                         i += 2;
 211
 212                         len += __text(fn, lrecl);
 213                 } else if (!strcmp(argv[i], "bin")) {
 214                         len += __bin(fn);
 215                         i++;
 216                 } else {
 217                         fprintf(stderr, "Error: cannot understand type '%s'\n",
 218                                 argv[i]);
 219                         return 1;
 220                 }
 221         }
 222
 223         mktail(len);
 224
 225         return 0;
 226 }