1 /* ----------------------------------------------------------------------- *
3 * Copyright 2003 Lars Munch Christensen - All Rights Reserved
4 * Copyright 1998-2004 H. Peter Anvin - All Rights Reserved
6 * Based on the Linux installer program for SYSLINUX by H. Peter Anvin
8 * This program is free software; you can redistribute it and/or modify
9 * it under the terms of the GNU General Public License as published by
10 * the Free Software Foundation, Inc., 53 Temple Place Ste 330,
11 * Boston MA 02111-1307, USA; either version 2 of the License, or
12 * (at your option) any later version; incorporated herein by reference.
14 * ----------------------------------------------------------------------- */
17 * syslinux-mingw.c - Win2k/WinXP installer program for SYSLINUX
28 # define noreturn void __attribute__((noreturn))
30 # define noreturn void
33 void error(char* msg);
35 /* Begin stuff for MBR code */
39 #define SECTOR_SIZE 512
40 #define PART_TABLE 0x1be
41 #define PART_SIZE 0x10
43 #define PART_ACTIVE 0x80
45 // The following struct should be in the ntddstor.h file, but I didn't have it.
46 // TODO: Make this a conditional compilation
47 typedef struct _STORAGE_DEVICE_NUMBER {
48 DEVICE_TYPE DeviceType;
50 ULONG PartitionNumber;
51 } STORAGE_DEVICE_NUMBER, *PSTORAGE_DEVICE_NUMBER;
53 BOOL GetStorageDeviceNumberByHandle( HANDLE handle, const STORAGE_DEVICE_NUMBER *sdn ) {
57 if ( DeviceIoControl( handle, IOCTL_STORAGE_GET_DEVICE_NUMBER, NULL,
58 0, (LPVOID)sdn, sizeof( *sdn ), &count, NULL ) ) {
62 error("GetDriveNumber: DeviceIoControl failed");
68 int GetBytesPerSector( HANDLE drive ) {
73 if ( DeviceIoControl( drive, IOCTL_DISK_GET_DRIVE_GEOMETRY, NULL, 0,
74 &g, sizeof( g ), &count, NULL ) ) {
75 result = g.BytesPerSector;
81 BOOL FixMBR(int driveNum, int partitionNum, int write_mbr, int set_active) {
87 sprintf( driveName, "\\\\.\\PHYSICALDRIVE%d", driveNum );
89 drive = CreateFile( driveName,
90 GENERIC_READ | GENERIC_WRITE,
91 FILE_SHARE_WRITE | FILE_SHARE_READ,
97 if( drive == INVALID_HANDLE_VALUE ) {
98 error("Accessing physical drive");
103 unsigned char sector[SECTOR_SIZE];
106 if( GetBytesPerSector( drive ) != SECTOR_SIZE ) {
107 fprintf(stderr, "Error: Sector size of this drive is %d; must be %d\n",
108 GetBytesPerSector( drive ), SECTOR_SIZE );
113 if ( ReadFile( drive, sector, sizeof( sector ), &howMany, NULL ) == 0 ) {
114 error("Reading raw drive");
116 } else if ( howMany != sizeof( sector ) ) {
117 fprintf(stderr, "Error: ReadFile on drive only got %d of %d bytes\n",
118 (int)howMany, sizeof( sector ) );
123 // Copy over the MBR code if specified (-m)
126 if ( syslinux_mbr_len >= PART_TABLE ) {
127 fprintf(stderr, "Error: MBR will not fit; not writing\n" );
130 memcpy( sector, syslinux_mbr, syslinux_mbr_len );
135 // Check that our partition is active if specified (-a)
137 if ( sector[ PART_TABLE + ( PART_SIZE * ( partitionNum - 1 ) ) ] != 0x80 ) {
139 for ( p = 0; p < PART_COUNT; p++ )
140 sector[ PART_TABLE + ( PART_SIZE * p ) ] = ( p == partitionNum - 1 ? 0x80 : 0 );
145 SetFilePointer( drive, 0, NULL, FILE_BEGIN );
147 if ( WriteFile( drive, sector, sizeof( sector ), &howMany, NULL ) == 0 ) {
148 error("Writing MBR");
150 } else if ( howMany != sizeof( sector ) ) {
151 fprintf(stderr, "Error: WriteFile on drive only wrote %d of %d bytes\n",
152 (int)howMany, sizeof( sector ) );
157 if( !CloseHandle( drive ) ) {
158 error("CloseFile on drive");
166 /* End stuff for MBR code */
168 const char *program; /* Name of program */
169 const char *drive; /* Drive to install to */
172 * Check Windows version.
174 * On Windows Me/98/95 you cannot open a directory, physical disk, or
175 * volume using CreateFile.
181 osvi.dwOSVersionInfoSize = sizeof(OSVERSIONINFO);
184 return (osvi.dwPlatformId == VER_PLATFORM_WIN32_NT) &&
185 ((osvi.dwMajorVersion > 4) ||
186 ((osvi.dwMajorVersion == 4) && (osvi.dwMinorVersion == 0)));
190 * Windows error function
192 void error(char* msg)
196 /* Format the Windows error message */
198 FORMAT_MESSAGE_ALLOCATE_BUFFER |
199 FORMAT_MESSAGE_FROM_SYSTEM |
200 FORMAT_MESSAGE_IGNORE_INSERTS,
201 NULL, GetLastError(),
202 MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), // Default language
203 (LPTSTR) &lpMsgBuf, 0, NULL );
206 fprintf(stderr, "%s: %s", msg, (char*) lpMsgBuf);
208 /* Free the buffer */
213 * Wrapper for ReadFile suitable for libfat
215 int libfat_readfile(intptr_t pp, void *buf, size_t secsize, libfat_sector_t sector)
217 uint64_t offset = (uint64_t)sector * secsize;
218 LONG loword = (LONG)offset;
219 LONG hiword = (LONG)(offset >> 32);
220 LONG hiwordx = hiword;
223 if ( SetFilePointer((HANDLE)pp, loword, &hiwordx, FILE_BEGIN) != loword ||
225 !ReadFile((HANDLE)pp, buf, secsize, &bytes_read, NULL) ||
226 bytes_read != secsize ) {
227 fprintf(stderr, "Cannot read sector %u\n", sector);
236 fprintf(stderr, "Usage: syslinux.exe [-sfma] <drive>: [bootsecfile]\n");
240 int main(int argc, char *argv[])
242 HANDLE f_handle, d_handle;
248 static unsigned char sectbuf[512];
250 static char drive_name[] = "\\\\.\\?:";
251 static char drive_root[] = "?:\\";
252 static char ldlinux_name[] = "?:\\ldlinux.sys" ;
254 struct libfat_filesystem *fs;
255 libfat_sector_t s, *secp, sectors[65]; /* 65 is maximum possible */
256 uint32_t ldlinux_cluster;
258 const char *bootsecfile = NULL;
260 int force = 0; /* -f (force) option */
261 int mbr = 0; /* -m (MBR) option */
262 int setactive = 0; /* -a (set partition active) */
267 fprintf(stderr, "You need to be running at least Windows NT; use syslinux.com instead.\n");
274 for ( argp = argv+1 ; *argp ; argp++ ) {
275 if ( **argp == '-' ) {
282 case 's': /* Use "safe, slow and stupid" code */
283 syslinux_make_stupid();
285 case 'f': /* Force install */
288 case 'm': /* Install MBR */
291 case 'a': /* Mark this partition active */
310 if ( !drive || !isalpha(drive[0]) || drive[1] != ':' || drive[2] )
313 /* Test if drive exists */
314 drives = GetLogicalDrives();
315 if(!(drives & ( 1 << (tolower(drive[0]) - 'a')))) {
316 fprintf(stderr, "No such drive %c:\n", drive[0]);
320 /* Determines the drive type */
321 drive_name[4] = drive[0];
322 ldlinux_name[0] = drive[0];
323 drive_root[0] = drive[0];
324 drive_type = GetDriveType(drive_root);
326 /* Test for removeable media */
327 if ((drive_type == DRIVE_FIXED) && (force == 0)) {
328 fprintf(stderr, "Not a removable drive (use -f to override) \n");
332 /* Test for unsupported media */
333 if ((drive_type != DRIVE_FIXED) && (drive_type != DRIVE_REMOVABLE)) {
334 fprintf(stderr, "Unsupported media\n");
339 * First open the drive
341 d_handle = CreateFile(drive_name, GENERIC_READ | GENERIC_WRITE,
342 FILE_SHARE_READ | FILE_SHARE_WRITE,
343 NULL, OPEN_EXISTING, 0, NULL );
345 if (d_handle == INVALID_HANDLE_VALUE) {
346 error("Could not open drive");
351 * Make sure we can read the boot sector
353 if ( !ReadFile(d_handle, sectbuf, 512, &bytes_read, NULL) ) {
354 error("Reading boot sector");
357 if (bytes_read != 512) {
358 fprintf(stderr, "Could not read the whole boot sector\n");
362 /* Check to see that what we got was indeed an MS-DOS boot sector/superblock */
363 if( (errmsg = syslinux_check_bootsect(sectbuf)) ) {
364 fprintf(stderr, "%s\n", errmsg);
368 /* Change to normal attributes to enable deletion */
369 /* Just ignore error if the file do not exists */
370 SetFileAttributes(ldlinux_name, FILE_ATTRIBUTE_NORMAL);
372 /* Delete the file */
373 /* Just ignore error if the file do not exists */
374 DeleteFile(ldlinux_name);
376 /* Create ldlinux.sys file */
377 f_handle = CreateFile(ldlinux_name, GENERIC_READ | GENERIC_WRITE,
378 FILE_SHARE_READ | FILE_SHARE_WRITE,
380 FILE_ATTRIBUTE_READONLY | FILE_ATTRIBUTE_SYSTEM |
381 FILE_ATTRIBUTE_HIDDEN,
384 if (f_handle == INVALID_HANDLE_VALUE) {
385 error("Unable to create ldlinux.sys");
389 /* Write ldlinux.sys file */
390 if (!WriteFile(f_handle, syslinux_ldlinux, syslinux_ldlinux_len, &bytes_written, NULL)) {
391 error("Could not write ldlinux.sys");
395 if (bytes_written != syslinux_ldlinux_len) {
396 fprintf(stderr, "Could not write whole ldlinux.sys\n");
400 /* Now flush the media */
401 if(!FlushFileBuffers(f_handle)) {
402 error("FlushFileBuffers failed");
406 /* Map the file (is there a better way to do this?) */
407 fs = libfat_open(libfat_readfile, (intptr_t)d_handle);
408 ldlinux_cluster = libfat_searchdir(fs, 0, "LDLINUX SYS", NULL);
411 s = libfat_clustertosector(fs, ldlinux_cluster);
412 while ( s && nsectors < 65 ) {
415 s = libfat_nextsector(fs, s);
420 * Patch ldlinux.sys and the boot sector
422 syslinux_patch(sectors, nsectors);
427 if ( SetFilePointer(f_handle, 0, NULL, FILE_BEGIN) != 0 ||
428 !WriteFile(f_handle, syslinux_ldlinux, syslinux_ldlinux_len, &bytes_written, NULL) ||
429 bytes_written != syslinux_ldlinux_len ) {
430 error("Could not write ldlinux.sys");
434 /* If desired, fix the MBR */
435 if( mbr || setactive ) {
436 STORAGE_DEVICE_NUMBER sdn;
437 if( GetStorageDeviceNumberByHandle( d_handle, &sdn ) ) {
438 if( !FixMBR(sdn.DeviceNumber, sdn.PartitionNumber, mbr, setactive) ) {
439 fprintf(stderr, "Did not successfully update the MBR; continuing...\n");
442 fprintf(stderr, "Could not find device number for updating MBR; continuing...\n");
447 CloseHandle(f_handle);
449 /* Make the syslinux boot sector */
450 syslinux_make_bootsect(sectbuf);
452 /* Write the syslinux boot sector into the boot sector */
454 f_handle = CreateFile(bootsecfile, GENERIC_READ | GENERIC_WRITE,
455 FILE_SHARE_READ | FILE_SHARE_WRITE,
457 FILE_ATTRIBUTE_ARCHIVE,
459 if (f_handle == INVALID_HANDLE_VALUE) {
460 error("Unable to create bootsector file");
463 if (!WriteFile(f_handle, sectbuf, 512, &bytes_written, NULL)) {
464 error("Could not write boot sector file");
467 CloseHandle(f_handle);
469 SetFilePointer(d_handle, 0, NULL, FILE_BEGIN);
470 WriteFile( d_handle, sectbuf, 512, &bytes_written, NULL ) ;
473 if(bytes_written != 512) {
474 fprintf(stderr, "Could not write the whole boot sector\n");
479 CloseHandle(d_handle);