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

syslinux/iso2exe: add NT+ support
author Pascal Bellard <pascal.bellard@slitaz.org>
date Sat Mar 30 18:36:46 2013 +0100 (2013-03-30)
parents ab3f0098073a
children 0c880b8c93be
rev   line source
pascal@14150 1 #include <sys/types.h>
pascal@14150 2 #include <fcntl.h>
pascal@14150 3 #include <stdio.h>
pascal@14150 4 #include "iso2exe.h"
pascal@14150 5
pascal@14150 6 static int fd;
pascal@14150 7 static char buffer[2048];
pascal@14150 8
pascal@14150 9 static void quit(char *msg)
pascal@14150 10 {
pascal@14150 11 fprintf(stderr,"%s.\n", msg);
pascal@14150 12 exit(1);
pascal@14150 13 }
pascal@14150 14
pascal@14150 15 static void readsector(unsigned long sector)
pascal@14150 16 {
pascal@14150 17 if (lseek(fd, sector * sizeof(buffer), SEEK_SET) == -1 ||
pascal@14150 18 read(fd, buffer, sizeof(buffer)) != sizeof(buffer))
pascal@14150 19 quit("read sector failure");
pascal@14150 20 }
pascal@14150 21
pascal@14150 22 int main(int argc, char *argv[])
pascal@14150 23 {
pascal@14150 24 #define heads 64
pascal@14150 25 #define sectors 32
pascal@14150 26 #define partition 446
pascal@14150 27 #define trksz (512 * heads * sectors)
pascal@14150 28 unsigned long size, catalog, lba;
pascal@14261 29 int cylinders, i, j, isohybrid;
pascal@14150 30 unsigned n;
pascal@14261 31 char tazlitoinfo[31*1024];
pascal@14150 32 #ifndef WIN32
pascal@14150 33 char *bootiso;
pascal@14150 34 for (bootiso = (char *) main;
pascal@14150 35 bootiso[0] != 'M' || bootiso[1] != 'Z' || bootiso[2] != 0xEB;
pascal@14150 36 bootiso++) if (bootiso < (char *) main) quit("bootiso not found");
pascal@14150 37 #endif
pascal@14150 38 if (argc < 2)
pascal@14150 39 quit("Usage : isohybrid.exe file.iso");
pascal@14150 40 fd = open(argv[1],O_RDWR|O_BINARY);
pascal@14150 41 if (fd == -1)
pascal@14150 42 quit("Can't open rw");
pascal@14150 43
pascal@14150 44 // Install hybridiso boot sector
pascal@14150 45 readsector(17UL);
pascal@14150 46 if (strncmp(buffer+7, "EL TORITO SPECIFICATION", 23))
pascal@14150 47 quit("No EL TORITO boot record found");
pascal@14150 48 catalog = * (unsigned long *) (buffer + 71);
pascal@14150 49 readsector(catalog);
pascal@14150 50 if (* (unsigned long *) buffer != 1 ||
pascal@14150 51 * (unsigned long *) (buffer + 30) != 0x88AA55)
pascal@14150 52 quit("invalid boot catalog.");
pascal@14150 53 lba = * (unsigned long *) (buffer + 40);
pascal@14150 54 readsector(lba);
pascal@14150 55 if (* (unsigned long *) (buffer + 64) != 1886961915)
pascal@14150 56 quit("no isolinux.bin hybrid signature in bootloader");
pascal@14261 57 isohybrid = bootiso[69] * 512;
pascal@14261 58 * (unsigned long *) &bootiso[isohybrid + 432] = lba * 4;
pascal@14261 59 * (unsigned long *) &bootiso[isohybrid + 440] = rand();
pascal@14261 60 * (unsigned long *) &bootiso[isohybrid + partition] = 0x10080;
pascal@14261 61 * (unsigned short *) &bootiso[isohybrid + 510] = 0xAA55;
pascal@14151 62 size = lseek(fd, 0UL, SEEK_END);
pascal@14150 63 cylinders = (size + trksz - 1) / trksz;
pascal@14261 64 bootiso[isohybrid + partition + 4] = 23; // "Windows hidden IFS"
pascal@14261 65 bootiso[isohybrid + partition + 5] = heads - 1;
pascal@14261 66 bootiso[isohybrid + partition + 6] = (((cylinders - 1) & 0x300) >> 2) + sectors;
pascal@14261 67 bootiso[isohybrid + partition + 7] = (cylinders - 1) & 0xFF;
pascal@14261 68 * (unsigned long *) &bootiso[isohybrid + partition + 8] = 0;
pascal@14261 69 * (unsigned long *) &bootiso[isohybrid + partition + 12] = cylinders * sectors * heads;
pascal@14150 70
pascal@14150 71 // Install iso2exe boot sector
pascal@14150 72 * (unsigned short *) (bootiso + 26) = rand();
pascal@14150 73
pascal@14261 74 // read tazlito flavor data
pascal@14261 75 lseek(fd, 1024UL, SEEK_SET);
pascal@14261 76 read(fd, tazlitoinfo, sizeof(tazlitoinfo));
pascal@14261 77
pascal@14150 78 // Update iso image
pascal@14261 79 n = (bootiso[69] + 1) * 512;
pascal@14151 80 lseek(fd, 0UL, SEEK_SET);
pascal@14261 81 write(fd, bootiso, n); // EXE/PE + isohybrid mbr
pascal@14261 82 write(fd, tazlitoinfo, ((0x8000U - BOOTISOSZ) > sizeof(tazlitoinfo))
pascal@14261 83 ? sizeof(tazlitoinfo) : (0x8000U - BOOTISOSZ));
pascal@14261 84 write(fd, bootiso + n, BOOTISOSZ - n); // COM + rootfs + EXE/DOS
pascal@14150 85
pascal@14150 86 // Compute the checksum
pascal@14151 87 lseek(fd, 0UL, SEEK_SET);
pascal@14150 88 for (i = 66, n = 0, j = 0; j < 16; j++, i = 0) {
pascal@14150 89 if (read(fd, buffer, sizeof(buffer)) != sizeof(buffer))
pascal@14150 90 goto nochksum;
pascal@14150 91 for (; i < sizeof(buffer); i += 2)
pascal@14150 92 n += * (unsigned short *) (buffer + i);
pascal@14150 93 }
pascal@14150 94 * (unsigned short *) (bootiso + 64) = -n;
pascal@14151 95 lseek(fd, 0UL, SEEK_SET);
pascal@14150 96 write(fd, bootiso, 512);
pascal@14150 97 nochksum:
pascal@14150 98 close(fd);
pascal@14261 99 printf("Note you can create a USB key with %s.\n"
pascal@14261 100 "Simply rename it to a .exe file and run it.\n", argv[1]);
pascal@14150 101 }