rev |
line source |
pascal@13974
|
1 #include <windows.h>
|
pascal@14261
|
2 #include <winnt.h>
|
pascal@14261
|
3 #include <sys/types.h>
|
pascal@14261
|
4 #include <sys/stat.h>
|
pascal@14261
|
5 #include <fcntl.h>
|
pascal@13972
|
6
|
pascal@14261
|
7 #define BOOTSTRAP_SECTOR_COUNT_OFFSET 28
|
pascal@14261
|
8
|
pascal@14261
|
9 static int fullread(int fd, char *p, int n)
|
pascal@13972
|
10 {
|
pascal@14261
|
11 int i, j;
|
pascal@14261
|
12 for (i = 0; n > 0; i += j, n -= j) {
|
pascal@14261
|
13 j = read(fd, p + i, n);
|
pascal@14261
|
14 if (j <= 0) break;
|
pascal@14261
|
15 }
|
pascal@14261
|
16 return i;
|
pascal@13972
|
17 }
|
pascal@14261
|
18
|
pascal@14261
|
19 static int fullwrite(int fd, char *p, int n)
|
pascal@14261
|
20 {
|
pascal@14261
|
21 int i, j;
|
pascal@14261
|
22 for (i = 0; n > 0; i += j, n -= j) {
|
pascal@14261
|
23 j = write(fd, p + i, n);
|
pascal@14261
|
24 if (j <= 0) break;
|
pascal@14261
|
25 }
|
pascal@14261
|
26 return i;
|
pascal@14261
|
27 }
|
pascal@14261
|
28
|
pascal@14261
|
29 static void exec16bits(char *isoFileName)
|
pascal@14261
|
30 {
|
pascal@14261
|
31 int fdiso, fdtmp, i;
|
pascal@14261
|
32 long buffer[512/4];
|
pascal@14261
|
33 char tmpFileName[MAX_PATH], *p;
|
pascal@14261
|
34
|
pascal@14261
|
35 strcpy(tmpFileName, isoFileName);
|
pascal@14261
|
36 p = strrchr(tmpFileName, '\\');
|
pascal@14261
|
37 if (!p++) p = tmpFileName;
|
pascal@14261
|
38 strcpy(p, "tazboot.exe");
|
pascal@14261
|
39 fdiso = open(isoFileName, O_RDONLY|O_BINARY);
|
pascal@14261
|
40 fdtmp = open(tmpFileName, O_WRONLY|O_BINARY|O_CREAT,0555);
|
pascal@14261
|
41 for (i = 0; i < 0x8000; i += sizeof(buffer)) {
|
pascal@14261
|
42 fullread(fdiso, (char *) buffer, sizeof(buffer));
|
pascal@14261
|
43 if (i == 0) buffer[15] = 0; // kill PE header
|
pascal@14261
|
44 fullwrite(fdtmp, (char *) buffer, sizeof(buffer));
|
pascal@14261
|
45 }
|
pascal@14261
|
46 close(fdiso);
|
pascal@14261
|
47 close(fdtmp);
|
pascal@14261
|
48 execl(tmpFileName, isoFileName);
|
pascal@14261
|
49 }
|
pascal@14261
|
50
|
pascal@14261
|
51 static int iswinnt(void)
|
pascal@14261
|
52 {
|
pascal@14261
|
53 OSVERSIONINFOA Version;
|
pascal@14261
|
54 Version.dwOSVersionInfoSize = sizeof(Version);
|
pascal@14261
|
55 return (GetVersionEx(&Version) &&
|
pascal@14261
|
56 Version.dwPlatformId != VER_PLATFORM_WIN32_WINDOWS); // not Win9x
|
pascal@14261
|
57 }
|
pascal@14261
|
58
|
pascal@14261
|
59 static int ishybrid(char *isoFileName)
|
pascal@14261
|
60 {
|
pascal@14261
|
61 int fdiso;
|
pascal@14261
|
62 char buffer[2048];
|
pascal@14261
|
63 unsigned long magic = 0;
|
pascal@14261
|
64
|
pascal@14261
|
65 fdiso = open(isoFileName, O_RDONLY|O_BINARY);
|
pascal@14261
|
66 if (lseek(fdiso, 17 * 2048L, SEEK_SET) != -1 &&
|
pascal@14261
|
67 fullread(fdiso, buffer, 2048) == 2048 &&
|
pascal@14261
|
68 strncmp(buffer+23,"EL TORITO SPECIFICATION",23) == 0) {
|
pascal@14261
|
69 unsigned long lba = * (unsigned long *) (buffer + 71);
|
pascal@14261
|
70
|
pascal@14261
|
71 if (lseek(fdiso, lba * 2048L, SEEK_SET) != -1 &&
|
pascal@14261
|
72 fullread(fdiso, buffer, 2048) == 2048 &&
|
pascal@14261
|
73 * (unsigned long *) (buffer + 0) == 1 &&
|
pascal@14261
|
74 * (unsigned long *) (buffer + 30) == 0x88AA55) {
|
pascal@14261
|
75 lba = * (unsigned long *) (buffer + 40);
|
pascal@14261
|
76 if (lseek(fdiso, lba * 2048L, SEEK_SET) != -1 &&
|
pascal@14261
|
77 fullread(fdiso, buffer, 2048) == 2048)
|
pascal@14261
|
78 magic = * (unsigned long *) (buffer + 64);
|
pascal@14261
|
79 }
|
pascal@14261
|
80 }
|
pascal@14261
|
81 close(fdiso);
|
pascal@14261
|
82 return (magic == 1886961915);
|
pascal@14261
|
83 }
|
pascal@14261
|
84
|
pascal@14261
|
85 #define MODE_READ 0
|
pascal@14261
|
86 #define MODE_WRITE 1
|
pascal@14261
|
87 static int rdwrsector(int mode, int drive, unsigned long startingsector,
|
pascal@14261
|
88 unsigned long count, char *buffer)
|
pascal@14261
|
89 {
|
pascal@14261
|
90 HANDLE hDevice;
|
pascal@14261
|
91 DWORD result;
|
pascal@14261
|
92 char devname[sizeof("\\\\.\\PhysicalDrive0")];
|
pascal@14261
|
93
|
pascal@14261
|
94 if (drive >= 128) {
|
pascal@14261
|
95 strcpy(devname, "\\\\.\\PhysicalDrive0");
|
pascal@14261
|
96 devname[17] += drive - 128;
|
pascal@14261
|
97 }
|
pascal@14261
|
98 else {
|
pascal@14261
|
99 strcpy(devname, "\\\\.\\A:");
|
pascal@14261
|
100 devname[4] += drive;
|
pascal@14261
|
101 }
|
pascal@14261
|
102 hDevice = CreateFile (devname, GENERIC_READ,
|
pascal@14261
|
103 FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_EXISTING, 0, NULL);
|
pascal@14261
|
104 if (hDevice == INVALID_HANDLE_VALUE)
|
pascal@14261
|
105 return -1;
|
pascal@14261
|
106 SetFilePointer(hDevice, (startingsector*512), NULL, FILE_BEGIN);
|
pascal@14261
|
107 if (mode == MODE_READ) {
|
pascal@14261
|
108 if (!ReadFile(hDevice, buffer, 512*count, &result, NULL))
|
pascal@14261
|
109 result = -1;
|
pascal@14261
|
110 }
|
pascal@14261
|
111 else {
|
pascal@14261
|
112 if (!WriteFile(hDevice, buffer, 512*count, &result, NULL))
|
pascal@14261
|
113 result = -1;
|
pascal@14261
|
114 }
|
pascal@14261
|
115 CloseHandle(hDevice);
|
pascal@14261
|
116 return result;
|
pascal@14261
|
117 }
|
pascal@14261
|
118
|
pascal@14261
|
119 static void rawrite(unsigned long drive, char *isoFileName)
|
pascal@14261
|
120 {
|
pascal@14261
|
121 int fdiso, s, dev, isohybrid = -1;
|
pascal@14261
|
122 char buffer[2048];
|
pascal@14261
|
123
|
pascal@14261
|
124 if (drive == 0) return;
|
pascal@14261
|
125 for (dev = 128; (drive & 1) == 0; dev++)
|
pascal@14261
|
126 drive >>= 1;
|
pascal@14261
|
127 fdiso = open(isoFileName, O_RDONLY|O_BINARY);
|
pascal@14261
|
128 for (s = 0;;) {
|
pascal@14261
|
129 int s, n = fullread(fdiso, buffer, sizeof(buffer));
|
pascal@14261
|
130 if (n <= 0) break;
|
pascal@14261
|
131 n = (n+511)/512;
|
pascal@14261
|
132 if (s == 0) isohybrid = buffer[69];
|
pascal@14261
|
133 rdwrsector(MODE_WRITE, dev, s, n, buffer);
|
pascal@14261
|
134 if (s == isohybrid)
|
pascal@14261
|
135 rdwrsector(MODE_WRITE, dev, 0, 1, buffer);
|
pascal@14261
|
136 s += n;
|
pascal@14261
|
137 }
|
pascal@14261
|
138 close(fdiso);
|
pascal@14261
|
139 }
|
pascal@14261
|
140
|
pascal@14261
|
141 static unsigned long drives(void)
|
pascal@14261
|
142 {
|
pascal@14261
|
143 int i, mask, result;
|
pascal@14261
|
144 char buffer[512];
|
pascal@14261
|
145 for (i = result = 0, mask = 1; i < 8; i++, mask <<= 1) {
|
pascal@14261
|
146 if (rdwrsector(MODE_READ, i+128, 0, 1, buffer) != -1)
|
pascal@14261
|
147 result |= mask;
|
pascal@14261
|
148 }
|
pascal@14261
|
149 return result;
|
pascal@14261
|
150 }
|
pascal@14261
|
151
|
pascal@14261
|
152 static void writefloppy(char *isoFileName)
|
pascal@14261
|
153 {
|
pascal@14261
|
154 char buffer[512];
|
pascal@14261
|
155 int i, n, fd;
|
pascal@14261
|
156
|
pascal@14261
|
157 buffer[BOOTSTRAP_SECTOR_COUNT_OFFSET] = 0;
|
pascal@14261
|
158 fd = open(isoFileName, O_RDONLY|O_BINARY);
|
pascal@14261
|
159 if (fd != -1) {
|
pascal@14261
|
160 read(fd, buffer, sizeof(buffer));
|
pascal@14261
|
161 n = buffer[BOOTSTRAP_SECTOR_COUNT_OFFSET];
|
pascal@14261
|
162 if (n != 0 &&
|
pascal@14261
|
163 lseek(fd, * (unsigned long *) (buffer + 60) - (512 * n),
|
pascal@14261
|
164 SEEK_SET) != -1) {
|
pascal@14261
|
165 for (i = 0; i < n; i++) {
|
pascal@14261
|
166 fullread(fd, buffer, 512);
|
pascal@14261
|
167 if (i == 1) strncpy(buffer, isoFileName, 512);
|
pascal@14261
|
168 rdwrsector(MODE_WRITE, 0, i, 1, buffer);
|
pascal@14261
|
169 }
|
pascal@14261
|
170 }
|
pascal@14261
|
171 close(fd);
|
pascal@14261
|
172 }
|
pascal@14261
|
173 }
|
pascal@14261
|
174
|
pascal@14261
|
175 //TODO #define VCPI_LINUX_LOADER The DOS/EXE loader can boot in VM86 using VCPI API
|
pascal@14261
|
176 int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance,
|
pascal@14261
|
177 LPSTR lpCmdLine, int nCmdShow)
|
pascal@14261
|
178 {
|
pascal@14261
|
179 char isoFileName[MAX_PATH];
|
pascal@14261
|
180 char header[32];
|
pascal@14261
|
181 int fd;
|
pascal@14261
|
182 int usbkeyicon = MB_ICONASTERISK;
|
pascal@14261
|
183 char *usbkeymsg = "You can either:\n\n"
|
pascal@14261
|
184 "- create a SliTaz USB boot key or a boot memory card.\n"
|
pascal@14261
|
185 " (note that this will be read only like a CDROM.\n"
|
pascal@14261
|
186 " The Slitaz utility 'tazusb' can be used later to create\n"
|
pascal@14261
|
187 " a true read/write USB key).\n\n"
|
pascal@14261
|
188 "- create a SliTaz bootstrap floppy for the ISO image\n"
|
pascal@14261
|
189 " on the hard disk.\n"
|
pascal@14261
|
190 "\nDo you want to create a boot key now ?";
|
pascal@14261
|
191
|
pascal@14261
|
192 GetModuleFileName(hInstance, isoFileName, MAX_PATH);
|
pascal@14261
|
193 if (!iswinnt()) {
|
pascal@14261
|
194 #ifdef VCPI_LINUX_LOADER
|
pascal@14261
|
195 exec16bits(isoFileName);
|
pascal@14261
|
196 #else
|
pascal@14261
|
197 MessageBox(NULL,"No support for Win9x yet.\n"
|
pascal@14261
|
198 "Retry in DOS mode without emm386.\n",
|
pascal@14261
|
199 "Sorry", MB_OK|MB_ICONERROR);
|
pascal@14261
|
200 exit(1);
|
pascal@14261
|
201 #endif
|
pascal@14261
|
202 }
|
pascal@14261
|
203 if (!ishybrid(isoFileName)) {
|
pascal@14261
|
204 if (MessageBox(NULL,"Not an isolinux hybrid ISO.\n"
|
pascal@14261
|
205 "This ISO image will not boot\n"
|
pascal@14261
|
206 "from the media you will create !",
|
pascal@14261
|
207 "Will not boot !",
|
pascal@14261
|
208 MB_OKCANCEL|MB_ICONWARNING) == IDCANCEL)
|
pascal@14261
|
209 exit(0);
|
pascal@14261
|
210 }
|
pascal@14261
|
211 header[BOOTSTRAP_SECTOR_COUNT_OFFSET] = 0;
|
pascal@14261
|
212 fd = open(isoFileName, O_RDONLY|O_BINARY);
|
pascal@14261
|
213 if (fd != -1) {
|
pascal@14261
|
214 read(fd, header, sizeof(header));
|
pascal@14261
|
215 close(fd);
|
pascal@14261
|
216 }
|
pascal@14261
|
217 if (header[BOOTSTRAP_SECTOR_COUNT_OFFSET] == 0) { // No floppy bootstrap available
|
pascal@14261
|
218 usbkeyicon = MB_ICONQUESTION;
|
pascal@14261
|
219 usbkeymsg = "Do you want to create a boot key ?";
|
pascal@14261
|
220 }
|
pascal@14261
|
221 if (MessageBox(NULL,usbkeymsg, "Create a boot stick ?",
|
pascal@14261
|
222 MB_YESNO|usbkeyicon) == IDYES) {
|
pascal@14261
|
223 unsigned long base, new;
|
pascal@14261
|
224 int retry;
|
pascal@14261
|
225
|
pascal@14261
|
226 if (MessageBox(NULL,"Step 1: unplug the USB stick.",
|
pascal@14261
|
227 "Drive detection 1/2",
|
pascal@14261
|
228 MB_OKCANCEL|MB_ICONEXCLAMATION) == IDCANCEL)
|
pascal@14261
|
229 exit(0);
|
pascal@14261
|
230 base = drives();
|
pascal@14261
|
231 if (MessageBox(NULL,"Step 2: plug the USB stick in.",
|
pascal@14261
|
232 "Drive detection 2/2",
|
pascal@14261
|
233 MB_OKCANCEL|MB_ICONEXCLAMATION) == IDCANCEL)
|
pascal@14261
|
234 exit(0);
|
pascal@14261
|
235 retry = 0;
|
pascal@14261
|
236 do {
|
pascal@14261
|
237 Sleep(1000); // ms
|
pascal@14261
|
238 new = drives();
|
pascal@14261
|
239 } while (new == base && retry++ < 10);
|
pascal@14261
|
240 if (new == base) {
|
pascal@14261
|
241 MessageBox(NULL,"No USB stick found.","Sorry",
|
pascal@14261
|
242 MB_OK|MB_ICONERROR);
|
pascal@14261
|
243 exit(1);
|
pascal@14261
|
244 }
|
pascal@14261
|
245 rawrite(base ^ new, isoFileName);
|
pascal@14261
|
246 }
|
pascal@14261
|
247 if (header[BOOTSTRAP_SECTOR_COUNT_OFFSET] != 0 &&
|
pascal@14261
|
248 MessageBox(NULL,"Do you want to create a bootstrap floppy ?",
|
pascal@14261
|
249 "Create a bootstrap floppy ?",
|
pascal@14261
|
250 MB_YESNO|MB_ICONQUESTION) == IDYES &&
|
pascal@14261
|
251 MessageBox(NULL,"Insert a floppy disk in drive now",
|
pascal@14261
|
252 "Insert floppy",
|
pascal@14261
|
253 MB_OKCANCEL|MB_ICONEXCLAMATION) != IDCANCEL) {
|
pascal@14261
|
254
|
pascal@14261
|
255 // Create a 9k bootstrap with vfat, ext2 & ntfs drivers
|
pascal@14261
|
256 // to boot the ISO image on hard disk
|
pascal@14261
|
257 writefloppy(isoFileName);
|
pascal@14261
|
258 }
|
pascal@14261
|
259 }
|