wok annotate syslinux/stuff/iso2exe/iso2exe.c @ rev 18779

syslinux/isohybrid.exe: fix config md5
author Pascal Bellard <pascal.bellard@slitaz.org>
date Fri Jan 01 19:26:32 2016 +0100 (2016-01-01)
parents ea9f391548cf
children ec7b15533c2d
rev   line source
pascal@17451 1 #ifdef __TURBOC__
pascal@17451 2 #include <io.h>
pascal@17451 3 #endif
pascal@14150 4 #include <sys/types.h>
pascal@14150 5 #include <fcntl.h>
pascal@14150 6 #include <stdio.h>
pascal@17451 7 #include <stdlib.h>
pascal@17451 8 #include <string.h>
pascal@14268 9 #ifdef WIN32
pascal@14268 10 #include <windows.h>
pascal@14268 11 #endif
pascal@18775 12 #ifdef __MSDOS__
pascal@18775 13 #define ftruncate(a,b)
pascal@18775 14 #endif
pascal@18775 15 #ifdef __MINGW32__
pascal@18775 16 #define ftruncate chsize
pascal@18775 17 #endif
pascal@18775 18 #if !defined(__MSDOS__) && !defined(WIN32)
pascal@18775 19 #define O_BINARY 0
pascal@18775 20 #endif
pascal@17591 21 typedef unsigned char uint8_t;
pascal@17591 22 typedef unsigned long uint32_t;
pascal@14150 23 #include "iso2exe.h"
pascal@14150 24
pascal@17591 25 static int fd, forced, uninstall, status = 1;
pascal@18775 26 static char *append, *initrd;
pascal@17489 27 static char tazlitoinfo[0x8000U - BOOTISOSZ];
pascal@14264 28 #define buffer tazlitoinfo
pascal@14264 29 #define BUFFERSZ 2048
pascal@17451 30 #define WORD(n) * (unsigned short *) (n)
pascal@17451 31 #define LONG(n) * (unsigned long *) (n)
pascal@14150 32
pascal@14150 33 static void readsector(unsigned long sector)
pascal@14150 34 {
pascal@14264 35 if (lseek(fd, sector * BUFFERSZ, SEEK_SET) == -1 ||
pascal@14268 36 read(fd, buffer, BUFFERSZ) != BUFFERSZ) {
pascal@17451 37 puts(bootiso+READSECTORERR);
pascal@14268 38 exit(1);
pascal@14268 39 }
pascal@14150 40 }
pascal@14150 41
pascal@18775 42 static unsigned long getcustomsector(void)
pascal@18775 43 {
pascal@18775 44 readsector(16UL);
pascal@18775 45 return 16UL + LONG(buffer + 80);
pascal@18775 46 }
pascal@18775 47
pascal@17810 48 static int skipmd5 = 0;
pascal@17591 49 #define ALIGN1
pascal@17591 50
pascal@17591 51 typedef struct {
pascal@17591 52 uint32_t l;
pascal@17591 53 uint32_t h;
pascal@17591 54 } uint64_t;
pascal@17591 55 static uint8_t wbuffer[64]; /* always correctly aligned for uint64_t */
pascal@17591 56 static uint64_t total64; /* must be directly before hash[] */
pascal@17591 57 static uint32_t hash[8]; /* 4 elements for md5, 5 for sha1, 8 for sha256 */
pascal@17591 58
pascal@17591 59 //#define rotl32(x,n) (((x) << (n)) | ((x) >> (32 - (n))))
pascal@17591 60 static uint32_t rotl32(uint32_t x, unsigned n)
pascal@17591 61 {
pascal@17591 62 return (x << n) | (x >> (32 - n));
pascal@17591 63 }
pascal@17591 64
pascal@17591 65 static void md5_process_block64(void);
pascal@17591 66
pascal@17591 67 /* Feed data through a temporary buffer.
pascal@17591 68 * The internal buffer remembers previous data until it has 64
pascal@17591 69 * bytes worth to pass on.
pascal@17591 70 */
pascal@17591 71 static void common64_hash(const void *buffer, size_t len)
pascal@17591 72 {
pascal@17591 73 unsigned bufpos = total64.l & 63;
pascal@17591 74
pascal@17591 75 total64.l += len; if (total64.l < len) total64.h++;
pascal@17591 76
pascal@17591 77 while (1) {
pascal@17591 78 unsigned remaining = 64 - bufpos;
pascal@17591 79 if (remaining > len)
pascal@17591 80 remaining = len;
pascal@17591 81 /* Copy data into aligned buffer */
pascal@17591 82 memcpy(wbuffer + bufpos, buffer, remaining);
pascal@17591 83 len -= remaining;
pascal@17591 84 buffer = (const char *)buffer + remaining;
pascal@17591 85 bufpos += remaining;
pascal@17591 86 /* clever way to do "if (bufpos != 64) break; ... ; bufpos = 0;" */
pascal@17591 87 bufpos -= 64;
pascal@17591 88 if (bufpos != 0)
pascal@17591 89 break;
pascal@17591 90 /* Buffer is filled up, process it */
pascal@17591 91 md5_process_block64();
pascal@17591 92 /*bufpos = 0; - already is */
pascal@17591 93 }
pascal@17591 94 }
pascal@17591 95
pascal@17591 96 /* Process the remaining bytes in the buffer */
pascal@17591 97 static void common64_end(void)
pascal@17591 98 {
pascal@17591 99 unsigned bufpos = total64.l & 63;
pascal@17591 100 /* Pad the buffer to the next 64-byte boundary with 0x80,0,0,0... */
pascal@17591 101 wbuffer[bufpos++] = 0x80;
pascal@17591 102
pascal@17591 103 /* This loop iterates either once or twice, no more, no less */
pascal@17591 104 while (1) {
pascal@17591 105 unsigned remaining = 64 - bufpos;
pascal@17591 106 memset(wbuffer + bufpos, 0, remaining);
pascal@17591 107 /* Do we have enough space for the length count? */
pascal@17591 108 if (remaining >= 8) {
pascal@17591 109 /* Store the 64-bit counter of bits in the buffer */
pascal@17591 110 //uint64_t t = total64 << 3;
pascal@17591 111 uint32_t *t = (uint32_t *) (&wbuffer[64 - 8]);
pascal@17591 112 /* wbuffer is suitably aligned for this */
pascal@17591 113 //*(uint64_t *) (&wbuffer[64 - 8]) = t;
pascal@17591 114 t[0] = total64.l << 3;
pascal@17591 115 t[1] = (total64.h << 3) | (total64.l >> 29);
pascal@17591 116 }
pascal@17591 117 md5_process_block64();
pascal@17591 118 if (remaining >= 8)
pascal@17591 119 break;
pascal@17591 120 bufpos = 0;
pascal@17591 121 }
pascal@17591 122 }
pascal@17591 123
pascal@17591 124 /* These are the four functions used in the four steps of the MD5 algorithm
pascal@17591 125 * and defined in the RFC 1321. The first function is a little bit optimized
pascal@17591 126 * (as found in Colin Plumbs public domain implementation).
pascal@17591 127 * #define FF(b, c, d) ((b & c) | (~b & d))
pascal@17591 128 */
pascal@17591 129 #undef FF
pascal@17591 130 #undef FG
pascal@17591 131 #undef FH
pascal@17591 132 #undef FI
pascal@17591 133 #define FF(b, c, d) (d ^ (b & (c ^ d)))
pascal@17591 134 #define FG(b, c, d) FF(d, b, c)
pascal@17591 135 #define FH(b, c, d) (b ^ c ^ d)
pascal@17591 136 #define FI(b, c, d) (c ^ (b | ~d))
pascal@17591 137
pascal@17591 138 /* Hash a single block, 64 bytes long and 4-byte aligned */
pascal@17591 139 static void md5_process_block64(void)
pascal@17591 140 {
pascal@17591 141 uint32_t *words = (void*) wbuffer;
pascal@17591 142 uint32_t A = hash[0];
pascal@17591 143 uint32_t B = hash[1];
pascal@17591 144 uint32_t C = hash[2];
pascal@17591 145 uint32_t D = hash[3];
pascal@17591 146
pascal@17591 147 const uint32_t *pc;
pascal@17591 148 const char *pp;
pascal@17591 149 const char *ps;
pascal@17591 150 int i;
pascal@17591 151 uint32_t temp;
pascal@17591 152
pascal@17591 153
pascal@17591 154 pc = C_array;
pascal@17591 155 pp = P_array;
pascal@17591 156 ps = S_array - 4;
pascal@17591 157
pascal@17591 158 for (i = 0; i < 64; i++) {
pascal@17591 159 if ((i & 0x0f) == 0)
pascal@17591 160 ps += 4;
pascal@17591 161 temp = A;
pascal@17591 162 switch (i >> 4) {
pascal@17591 163 case 0:
pascal@17591 164 temp += FF(B, C, D);
pascal@17591 165 break;
pascal@17591 166 case 1:
pascal@17591 167 temp += FG(B, C, D);
pascal@17591 168 break;
pascal@17591 169 case 2:
pascal@17591 170 temp += FH(B, C, D);
pascal@17591 171 break;
pascal@17591 172 case 3:
pascal@17591 173 temp += FI(B, C, D);
pascal@17591 174 }
pascal@17591 175 temp += words[(int) (*pp++)] + *pc++;
pascal@17591 176 temp = rotl32(temp, ps[i & 3]);
pascal@17591 177 temp += B;
pascal@17591 178 A = D;
pascal@17591 179 D = C;
pascal@17591 180 C = B;
pascal@17591 181 B = temp;
pascal@17591 182 }
pascal@17591 183 /* Add checksum to the starting values */
pascal@17591 184 hash[0] += A;
pascal@17591 185 hash[1] += B;
pascal@17591 186 hash[2] += C;
pascal@17591 187 hash[3] += D;
pascal@17591 188
pascal@17591 189 }
pascal@17591 190 #undef FF
pascal@17591 191 #undef FG
pascal@17591 192 #undef FH
pascal@17591 193 #undef FI
pascal@17591 194
pascal@17591 195 /* Initialize structure containing state of computation.
pascal@17591 196 * (RFC 1321, 3.3: Step 3)
pascal@17591 197 */
pascal@17591 198 static void md5_begin(void)
pascal@17591 199 {
pascal@17591 200 hash[0] = 0x67452301;
pascal@17591 201 hash[1] = 0xefcdab89;
pascal@17591 202 hash[2] = 0x98badcfe;
pascal@17591 203 hash[3] = 0x10325476;
pascal@17591 204 total64.l = total64.h = 0;
pascal@17591 205 }
pascal@17591 206
pascal@17591 207 /* Used also for sha1 and sha256 */
pascal@17591 208 #define md5_hash common64_hash
pascal@17591 209
pascal@17591 210 /* Process the remaining bytes in the buffer and put result from CTX
pascal@17591 211 * in first 16 bytes following RESBUF. The result is always in little
pascal@17591 212 * endian byte order, so that a byte-wise output yields to the wanted
pascal@17591 213 * ASCII representation of the message digest.
pascal@17591 214 */
pascal@17591 215 #define md5_end common64_end
pascal@17591 216
pascal@18775 217 static int writenhash(void *buffer, size_t len)
pascal@18775 218 {
pascal@18775 219 md5_hash(buffer, len);
pascal@18775 220 return write(fd, buffer, len);
pascal@18775 221 }
pascal@18775 222
pascal@17591 223 static void md5sum(void)
pascal@17591 224 {
pascal@17591 225 unsigned long sectors = 0;
pascal@17591 226 int count;
pascal@17591 227
pascal@17591 228 lseek(fd, 32768UL, SEEK_SET);
pascal@17591 229
pascal@17591 230 md5_begin();
pascal@17591 231 while ((count = read(fd, buffer, BUFFERSZ)) > 0) {
pascal@17591 232 if (sectors == 0)
pascal@17591 233 sectors = LONG(buffer + 80);
pascal@17591 234 md5_hash(buffer, count);
pascal@17591 235 if (--sectors == 0)
pascal@17591 236 break;
pascal@17591 237 }
pascal@17591 238
pascal@17591 239 if (count < 0)
pascal@17591 240 return;
pascal@17591 241
pascal@17591 242 md5_end();
pascal@17591 243
pascal@17591 244 lseek(fd, 32752UL, SEEK_SET);
pascal@17591 245 write(fd, hash, 16);
pascal@17591 246 memcpy(bootiso + BOOTISOSZ - 16, hash, 16);
pascal@17591 247 }
pascal@17591 248
pascal@17591 249 static unsigned chksum(unsigned start, unsigned stop)
pascal@17591 250 {
pascal@17591 251 unsigned i, n = 0;
pascal@17591 252
pascal@17591 253 lseek(fd, 0UL /* (unsigned long) (start / BUFFERSZ) */, SEEK_SET);
pascal@17591 254 while (1) {
pascal@17591 255 if (read(fd, buffer, BUFFERSZ) != BUFFERSZ)
pascal@17591 256 return 0;
pascal@17591 257 for (i = start % BUFFERSZ; i < BUFFERSZ; i += 2, start += 2) {
pascal@17591 258 if (start >= stop)
pascal@17591 259 return - n;
pascal@17591 260 n += WORD(buffer + i);
pascal@17591 261 }
pascal@17591 262 }
pascal@17591 263 }
pascal@17591 264
pascal@18779 265 static void clear_config(unsigned i)
pascal@18779 266 {
pascal@18779 267 for (;i % 512; i++) {
pascal@18779 268 /* clear custom config */
pascal@18779 269 write(fd, buffer + 2048, 2048);
pascal@18779 270 }
pascal@18779 271 }
pascal@18779 272
pascal@17451 273 static unsigned install(char *filename)
pascal@14150 274 {
pascal@14150 275 #define heads 64
pascal@14150 276 #define sectors 32
pascal@17489 277 #define partition (446+16)
pascal@17451 278 #define trksz (512UL * heads * sectors)
pascal@14150 279 unsigned long size, catalog, lba;
pascal@14261 280 int cylinders, i, j, isohybrid;
pascal@14150 281 unsigned n;
pascal@18775 282 #ifdef __MSDOS__
pascal@14268 283 for (bootiso = (char *) install;
pascal@17451 284 bootiso[0] != 'M' || bootiso[1] != 'Z' || bootiso[2] != '\xEB';
pascal@17451 285 bootiso++) if (bootiso < (char *) install) {
pascal@17451 286 bootiso = "No bootiso data";
pascal@17451 287 return 0;
pascal@17451 288 }
pascal@14150 289 #endif
pascal@14268 290 if (!filename)
pascal@17451 291 return USAGE;
pascal@14268 292 fd = open(filename,O_RDWR|O_BINARY);
pascal@14150 293 if (fd == -1)
pascal@17451 294 return OPENERR;
pascal@14150 295
pascal@17591 296 if (uninstall) {
pascal@17591 297 struct { char check[sizeof(tazlitoinfo) - BUFFERSZ - 1024]; };
pascal@17591 298 readsector(0UL);
pascal@18775 299 n = BUFFERSZ; /* fill with zeros */
pascal@17591 300 if (WORD(buffer) == 23117) {
pascal@18775 301 /* restore isolinux hybrid boot */
pascal@17810 302 readsector((unsigned long) buffer[417]);
pascal@18775 303 n = 0; /* fill with hybrid boot */
pascal@17591 304 }
pascal@17591 305 lseek(fd, 0UL, SEEK_SET);
pascal@17591 306 for (i = 0; i < 32; i++, n = BUFFERSZ) {
pascal@17591 307 write(fd, buffer + n, 1024);
pascal@17591 308 }
pascal@18775 309 i = getcustomsector();
pascal@18775 310 lseek(fd, i * 2048UL, SEEK_SET);
pascal@18779 311 clear_config(i);
pascal@18775 312 ftruncate(fd, i * 2048UL);
pascal@17591 313 close(fd);
pascal@17591 314 status = 0;
pascal@17591 315 return UNINSTALLMSG;
pascal@17591 316 }
pascal@17591 317
pascal@18775 318 if (append || initrd) {
pascal@18775 319 unsigned long pos = getcustomsector() * 2048UL;
pascal@18775 320 lseek(fd, pos, SEEK_SET);
pascal@18779 321 clear_config(pos);
pascal@18779 322 lseek(fd, pos, SEEK_SET);
pascal@18775 323 write(fd, "#!boot 00000000000000000000000000000000\n", 40);
pascal@18775 324 md5_begin();
pascal@18775 325 if (append) {
pascal@18775 326 writenhash("append=", 7);
pascal@18775 327 writenhash(append, strlen(append));
pascal@18775 328 writenhash("\n", 1);
pascal@18775 329 }
pascal@18775 330 if (initrd) {
pascal@18775 331 char number[16], *p;
pascal@18775 332 unsigned long end, x;
pascal@18775 333 int data = open(initrd,O_RDONLY|O_BINARY);
pascal@18775 334 if (data == -1)
pascal@18775 335 return OPENINITRDERR;
pascal@18775 336 for (end = 0;; end += i) {
pascal@18775 337 i = read(data, buffer, BUFFERSZ);
pascal@18775 338 if (i <= 0)
pascal@18775 339 break;
pascal@18775 340 }
pascal@18775 341 p = number + sizeof(number) -1;
pascal@18775 342 x = end; *p-- = '\n';
pascal@18775 343 do {
pascal@18775 344 *p-- = '0' + (x % 10);
pascal@18775 345 x /= 10;
pascal@18775 346 } while (x);
pascal@18775 347 if (*++p != '0') {
pascal@18775 348 writenhash("initrd:", 7);
pascal@18775 349 i = number - p + sizeof(number);
pascal@18775 350 writenhash(p, i);
pascal@18775 351 lseek(data, 0UL, SEEK_SET);
pascal@18775 352 do {
pascal@18775 353 i = read(data, buffer, BUFFERSZ);
pascal@18775 354 if (i <= 0)
pascal@18775 355 break;
pascal@18775 356 if (i > end)
pascal@18775 357 i = end;
pascal@18775 358 writenhash(buffer, i);
pascal@18775 359 end -= i;
pascal@18775 360 } while (end != 0);
pascal@18775 361 }
pascal@18775 362 close(data);
pascal@18775 363 }
pascal@18775 364 md5_end();
pascal@18775 365 {
pascal@18779 366 static char h[] = "0123456789abcdef";
pascal@18775 367 char string[32], *s = string + 30;
pascal@18775 368 unsigned char *p = (void *) hash;
pascal@18775 369
pascal@18775 370 lseek(fd, 7 + pos, SEEK_SET);
pascal@18775 371 for (p += 15; s >= string; p--, s -= 2) {
pascal@18775 372 s[1] = h[ *p & 15 ];
pascal@18775 373 s[0] = h[ *p >> 4 ];
pascal@18775 374 }
pascal@18775 375 write(fd, string, 32);
pascal@18775 376 }
pascal@18775 377 }
pascal@18775 378
pascal@16055 379 if (forced == 0) {
pascal@16055 380 status = 2;
pascal@17451 381 /* Install hybridiso boot sector */
pascal@16055 382 readsector(17UL);
pascal@17451 383 if (strncmp(buffer+7, bootiso+ELTORITOERR+ELTORITOOFS, 23))
pascal@17451 384 return ELTORITOERR;
pascal@17451 385 catalog = LONG(buffer + 71);
pascal@16055 386 readsector(catalog);
pascal@17451 387 if (LONG(buffer) != 1 || LONG(buffer + 30) != 0x88AA55UL)
pascal@17451 388 return CATALOGERR;
pascal@17451 389 lba = LONG(buffer + 40);
pascal@16055 390 readsector(lba);
pascal@17451 391 if (LONG(buffer + 64) != 1886961915UL)
pascal@17451 392 return HYBRIDERR;
pascal@17810 393 isohybrid = bootiso[417] * 512;
pascal@17451 394 LONG(bootiso + isohybrid + 432) = lba * 4;
pascal@17451 395 LONG(bootiso + isohybrid + 440) = rand();
pascal@17451 396 LONG(bootiso + isohybrid + partition) = 0x10080UL;
pascal@17451 397 WORD(bootiso + isohybrid + 510) = 0xAA55U;
pascal@17489 398 #if 0
pascal@16055 399 size = lseek(fd, 0UL, SEEK_END);
pascal@17489 400 size += 0x000FFFFFUL;
pascal@17489 401 size &= 0xFFF00000UL;
pascal@17489 402 #else
pascal@17489 403 for (size = 0x000FFFFFUL; /* 1M - 1 */
pascal@17489 404 read(fd, tazlitoinfo, 1024) == 1024;
pascal@17489 405 size += 1024);
pascal@17489 406 size &= 0xFFF00000UL; /* round */
pascal@17489 407 #endif
pascal@17489 408 cylinders = (size >> 20) - 1;
pascal@17451 409 bootiso[isohybrid + partition + 4] = 23; /* "Windows hidden IFS" */
pascal@16055 410 bootiso[isohybrid + partition + 5] = heads - 1;
pascal@17489 411 bootiso[isohybrid + partition + 6] = ((cylinders & 0x300) >> 2) + sectors;
pascal@17489 412 bootiso[isohybrid + partition + 7] = cylinders & 0xFF;
pascal@17451 413 LONG(bootiso + isohybrid + partition + 8) = 0;
pascal@17489 414 LONG(bootiso + isohybrid + partition + 12) = (size >> 9);
pascal@14150 415
pascal@17451 416 /* Copy the partition table */
pascal@16055 417 memcpy(bootiso + 0x1BE, bootiso + isohybrid + 0x1BE, 66);
pascal@16055 418 }
pascal@14266 419
pascal@17451 420 /* Install iso2exe boot sector */
pascal@17810 421 LONG(bootiso + 440) = time(NULL);
pascal@14150 422
pascal@17451 423 /* read tazlito flavor data */
pascal@14261 424 lseek(fd, 1024UL, SEEK_SET);
pascal@14261 425 read(fd, tazlitoinfo, sizeof(tazlitoinfo));
pascal@14261 426
pascal@17451 427 /* Update iso image */
pascal@17810 428 n = (bootiso[417] + 1) * 512;
pascal@14151 429 lseek(fd, 0UL, SEEK_SET);
pascal@17451 430 write(fd, bootiso, n); /* EXE/PE + isohybrid mbr */
pascal@17489 431 write(fd, tazlitoinfo, sizeof(tazlitoinfo));
pascal@17451 432 write(fd, bootiso + n, BOOTISOSZ - n); /* COM + rootfs + EXE/DOS */
pascal@14150 433
pascal@17810 434 /* Compute the boot checksums */
pascal@17810 435 if (!skipmd5) {
pascal@17591 436 puts(bootiso + MD5MSG);
pascal@17591 437 md5sum();
pascal@17591 438 lseek(fd, 0UL, SEEK_SET);
pascal@17591 439 write(fd, bootiso, 512);
pascal@17810 440 n = WORD(bootiso + 2) - 512*(WORD(bootiso + 4) - 1);
pascal@17810 441 WORD(bootiso + 18) = chksum(0, (unsigned short) n) - 1;
pascal@17591 442 }
pascal@17810 443 lseek(fd, 0UL, SEEK_SET);
pascal@17810 444 write(fd, bootiso, 512);
pascal@14150 445 close(fd);
pascal@14268 446 status = 0;
pascal@17451 447 return SUCCESSMSG;
pascal@14150 448 }
pascal@14268 449
pascal@14268 450 int main(int argc, char *argv[])
pascal@14268 451 {
pascal@17591 452 int i;
pascal@18775 453 char *s;
pascal@18775 454
pascal@18775 455 while (argc > 2) {
pascal@18775 456 s = argv[1];
pascal@18775 457 if (*s != '-') break;
pascal@18775 458 while (*s == '-') s++;
pascal@18775 459 switch (*s | 0x20) {
pascal@18775 460 case 'a' : append=argv[2]; break;
pascal@18775 461 case 'i' : initrd=argv[2]; break;
pascal@18775 462 }
pascal@18775 463 argv += 2;
pascal@18775 464 argc -= 2;
pascal@18775 465 }
pascal@17591 466 for (i = 2; i < argc; i++) {
pascal@17591 467 char *s = argv[i];
pascal@17591 468 while ((unsigned)(*s - '-') <= ('/' - '-')) s++;
pascal@17591 469 switch (*s | 0x20) {
pascal@17591 470 case 'f' : forced++; break;
pascal@17810 471 case 'q' : skipmd5++; break;
pascal@17591 472 case 'u' : uninstall++; break;
pascal@17591 473 }
pascal@17591 474 }
pascal@17451 475 puts(bootiso + install(argv[1]));
pascal@16055 476 if (status > 1)
pascal@17451 477 puts(bootiso + FORCEMSG);
pascal@14268 478 #ifdef WIN32
pascal@14268 479 Sleep(2000);
pascal@14268 480 #endif
pascal@14268 481 return status;
pascal@14268 482 }