Syslinux-mingw.c

/* ----------------------------------------------------------------------- * * *   Copyright 2003 Lars Munch Christensen - All Rights Reserved *   Copyright 1998-2008 H. Peter Anvin - All Rights Reserved * *   Based on the Linux installer program for SYSLINUX by H. Peter Anvin * *   This program is free software; you can redistribute it and/or modify *   it under the terms of the GNU General Public License as published by *   the Free Software Foundation, Inc., 53 Temple Place Ste 330, *   Boston MA 02111-1307, USA; either version 2 of the License, or *   (at your option) any later version; incorporated herein by reference. * * ----------------------------------------------------------------------- *//* * syslinux-mingw.c - Win2k/WinXP installer program for SYSLINUX */#include <windows.h>#include <stdio.h>#include <ctype.h>#include "syslinux.h"#include "libfat.h"#ifdef __GNUC__# define noreturn void __attribute__((noreturn))#else# define noreturn void#endifvoid error(char* msg);/* Begin stuff for MBR code */#include <winioctl.h>#define SECTOR_SIZE 512#define PART_TABLE  0x1be#define PART_SIZE   0x10#define PART_COUNT  4#define PART_ACTIVE 0x80// The following struct should be in the ntddstor.h file, but I didn't have it.// TODO: Make this a conditional compilationtypedef struct _STORAGE_DEVICE_NUMBER {  DEVICE_TYPE  DeviceType;  ULONG  DeviceNumber;  ULONG  PartitionNumber;} STORAGE_DEVICE_NUMBER, *PSTORAGE_DEVICE_NUMBER;BOOL GetStorageDeviceNumberByHandle( HANDLE handle, const STORAGE_DEVICE_NUMBER *sdn ) {  BOOL result = FALSE;  DWORD count;  if ( DeviceIoControl( handle, IOCTL_STORAGE_GET_DEVICE_NUMBER, NULL,0, (LPVOID)sdn, sizeof( *sdn ), &count, NULL ) ) {    result = TRUE;  }  else {    error("GetDriveNumber: DeviceIoControl failed");  }  return( result );}int GetBytesPerSector( HANDLE drive ) {  int result = 0;  DISK_GEOMETRY g;  DWORD count;  if ( DeviceIoControl( drive, IOCTL_DISK_GET_DRIVE_GEOMETRY, NULL, 0,&g, sizeof( g ), &count, NULL ) ) {    result = g.BytesPerSector;  }  return( result );}BOOL FixMBR(int driveNum, int partitionNum, int write_mbr, int set_active) {  BOOL result = TRUE;  HANDLE drive;  char driveName[128];  sprintf( driveName, "\\\\.\\PHYSICALDRIVE%d", driveNum );  drive = CreateFile( driveName,      GENERIC_READ | GENERIC_WRITE,      FILE_SHARE_WRITE | FILE_SHARE_READ,      NULL,      OPEN_EXISTING,      0,      NULL );  if( drive == INVALID_HANDLE_VALUE ) {    error("Accessing physical drive");    result = FALSE;  }  if( result ) {    unsigned char sector[SECTOR_SIZE];    DWORD howMany;    if( GetBytesPerSector( drive ) != SECTOR_SIZE ) {      fprintf(stderr, "Error: Sector size of this drive is %d; must be %d\n",      GetBytesPerSector( drive ), SECTOR_SIZE );      result = FALSE;    }    if ( result ) {      if ( ReadFile( drive, sector, sizeof( sector ), &howMany, NULL ) == 0 ) {error("Reading raw drive");result = FALSE;      } else if ( howMany != sizeof( sector ) ) {fprintf(stderr, "Error: ReadFile on drive only got %d of %d bytes\n",(int)howMany, sizeof( sector ) );result = FALSE;      }    }    // Copy over the MBR code if specified (-m)    if ( write_mbr ) {      if ( result ) {if ( syslinux_mbr_len >= PART_TABLE ) {  fprintf(stderr, "Error: MBR will not fit; not writing\n" );  result = FALSE;} else {  memcpy( sector, syslinux_mbr, syslinux_mbr_len );}      }    }    // Check that our partition is active if specified (-a)    if ( set_active ) {      if ( sector[ PART_TABLE + ( PART_SIZE * ( partitionNum - 1 ) ) ] != 0x80 ) {int p;for ( p = 0; p < PART_COUNT; p++ )  sector[ PART_TABLE + ( PART_SIZE * p ) ] = ( p == partitionNum - 1 ? 0x80 : 0 );      }    }    if ( result ) {      SetFilePointer( drive, 0, NULL, FILE_BEGIN );      if ( WriteFile( drive, sector, sizeof( sector ), &howMany, NULL ) == 0 ) {error("Writing MBR");result = FALSE;      } else if ( howMany != sizeof( sector ) ) {fprintf(stderr, "Error: WriteFile on drive only wrote %d of %d bytes\n",(int)howMany, sizeof( sector ) );result = FALSE;      }    }    if( !CloseHandle( drive ) ) {      error("CloseFile on drive");      result = FALSE;    }  }  return( result );}/* End stuff for MBR code */const char *program;/* Name of program */const char *drive;/* Drive to install to *//* * Check Windows version. * * On Windows Me/98/95 you cannot open a directory, physical disk, or * volume using CreateFile. */int checkver(void){  OSVERSIONINFO osvi;  osvi.dwOSVersionInfoSize = sizeof(OSVERSIONINFO);  GetVersionEx(&osvi);  return  (osvi.dwPlatformId == VER_PLATFORM_WIN32_NT) &&          ((osvi.dwMajorVersion > 4) ||          ((osvi.dwMajorVersion == 4) && (osvi.dwMinorVersion == 0)));}/* * Windows error function */void error(char* msg){  LPVOID lpMsgBuf;  /* Format the Windows error message */  FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER |FORMAT_MESSAGE_FROM_SYSTEM |FORMAT_MESSAGE_IGNORE_INSERTS,NULL, GetLastError(),MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), // Default language(LPTSTR) &lpMsgBuf, 0, NULL );  /* Print it */  fprintf(stderr, "%s: %s", msg, (char*) lpMsgBuf);  /* Free the buffer */  LocalFree(lpMsgBuf);}/* * Wrapper for ReadFile suitable for libfat */int libfat_readfile(intptr_t pp, void *buf, size_t secsize, libfat_sector_t sector){  uint64_t offset = (uint64_t)sector * secsize;  LONG loword = (LONG)offset;  LONG hiword  = (LONG)(offset >> 32);  LONG hiwordx = hiword;  DWORD bytes_read;  if ( SetFilePointer((HANDLE)pp, loword, &hiwordx, FILE_BEGIN) != loword ||       hiword != hiwordx ||       !ReadFile((HANDLE)pp, buf, secsize, &bytes_read, NULL) ||       bytes_read != secsize ) {    fprintf(stderr, "Cannot read sector %u\n", sector);    exit(1);  }  return secsize;}noreturn usage(void){  fprintf(stderr, "Usage: syslinux.exe [-sfmar][-d directory] <drive>: [bootsecfile]\n");  exit(1);}int main(int argc, char *argv[]){  HANDLE f_handle, d_handle;  DWORD bytes_read;  DWORD bytes_written;  DWORD drives;  UINT drive_type;  static unsigned char sectbuf[512];  char **argp, *opt;  static char drive_name[] = "\\\\.\\?:";  static char drive_root[] = "?:\\";  static char ldlinux_name[] = "?:\\ldlinux.sys" ;  const char *errmsg;  struct libfat_filesystem *fs;  libfat_sector_t s, *secp, sectors[65]; /* 65 is maximum possible */  uint32_t ldlinux_cluster;  int nsectors;  const char *bootsecfile = NULL;  const char *subdir = NULL;  int force = 0;/* -f (force) option */  int mbr = 0;/* -m (MBR) option */  int setactive = 0;/* -a (set partition active) */  int stupid = 0;/* -s (stupid) option */  int raid_mode = 0;/* -r (RAID) option */  (void)argc;  if (!checkver()) {    fprintf(stderr, "You need to be running at least Windows NT; use syslinux.com instead.\n");    exit(1);  }  program = argv[0];  drive = NULL;  for ( argp = argv+1 ; *argp ; argp++ ) {    if ( **argp == '-' ) {      opt = *argp + 1;      if ( !*opt )usage();      while ( *opt ) {switch ( *opt ) {case 's':/* Use "safe, slow and stupid" code */  stupid = 1;  break;case 'r':/* RAID mode */  raid_mode = 1;  break;case 'f':/* Force install */  force = 1;  break;case 'm':/* Install MBR */  mbr = 1;  break;case 'a':/* Mark this partition active */  setactive = 1;  break;case 'd':  if ( argp[1] )    subdir = *++argp;  break;default:  usage();  break;}opt++;      }    } else {      if ( bootsecfile )usage();      else if ( drive )bootsecfile = *argp;      elsedrive = *argp;    }  }  if ( !drive || !isalpha(drive[0]) || drive[1] != ':' || drive[2] )    usage();  /* Test if drive exists */  drives = GetLogicalDrives();  if(!(drives & ( 1 << (tolower(drive[0]) - 'a')))) {    fprintf(stderr, "No such drive %c:\n", drive[0]);    exit(1);  }  /* Determines the drive type */  drive_name[4]   = drive[0];  ldlinux_name[0] = drive[0];  drive_root[0]   = drive[0];  drive_type = GetDriveType(drive_root);  /* Test for removeable media */  if ((drive_type == DRIVE_FIXED) && (force == 0)) {    fprintf(stderr, "Not a removable drive (use -f to override) \n");    exit(1);  }  /* Test for unsupported media */  if ((drive_type != DRIVE_FIXED) && (drive_type != DRIVE_REMOVABLE)) {    fprintf(stderr, "Unsupported media\n");    exit(1);  }  /*   * First open the drive   */  d_handle = CreateFile(drive_name, GENERIC_READ | GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_EXISTING, 0, NULL );  if (d_handle == INVALID_HANDLE_VALUE) {    error("Could not open drive");    exit(1);  }  /*   * Make sure we can read the boot sector   */  if ( !ReadFile(d_handle, sectbuf, 512, &bytes_read, NULL) ) {    error("Reading boot sector");    exit(1);  }  if (bytes_read != 512) {    fprintf(stderr, "Could not read the whole boot sector\n");    exit(1);  }  /* Check to see that what we got was indeed an MS-DOS boot sector/superblock */  if( (errmsg = syslinux_check_bootsect(sectbuf)) ) {    fprintf(stderr, "%s\n", errmsg);    exit(1);  }  /* Change to normal attributes to enable deletion */  /* Just ignore error if the file do not exists */  SetFileAttributes(ldlinux_name, FILE_ATTRIBUTE_NORMAL);  /* Delete the file */  /* Just ignore error if the file do not exists */  DeleteFile(ldlinux_name);  /* Create ldlinux.sys file */  f_handle = CreateFile(ldlinux_name, GENERIC_READ | GENERIC_WRITE,FILE_SHARE_READ | FILE_SHARE_WRITE,NULL, CREATE_ALWAYS,FILE_ATTRIBUTE_READONLY | FILE_ATTRIBUTE_SYSTEM |FILE_ATTRIBUTE_HIDDEN,NULL );  if (f_handle == INVALID_HANDLE_VALUE) {    error("Unable to create ldlinux.sys");    exit(1);  }  /* Write ldlinux.sys file */  if (!WriteFile(f_handle, syslinux_ldlinux, syslinux_ldlinux_len, &bytes_written, NULL)) {    error("Could not write ldlinux.sys");    exit(1);  }  if (bytes_written != syslinux_ldlinux_len) {    fprintf(stderr, "Could not write whole ldlinux.sys\n");    exit(1);  }  /* Now flush the media */  if(!FlushFileBuffers(f_handle)) {    error("FlushFileBuffers failed");    exit(1);  }  /* Map the file (is there a better way to do this?) */  fs = libfat_open(libfat_readfile, (intptr_t)d_handle);  ldlinux_cluster = libfat_searchdir(fs, 0, "LDLINUX SYS", NULL);  secp = sectors;  nsectors = 0;  s = libfat_clustertosector(fs, ldlinux_cluster);  while ( s && nsectors < 65 ) {    *secp++ = s;    nsectors++;    s = libfat_nextsector(fs, s);  }  libfat_close(fs);  /*   * Patch ldlinux.sys and the boot sector   */  syslinux_patch(sectors, nsectors, stupid, raid_mode);  /*   * Rewrite the file   */  if ( SetFilePointer(f_handle, 0, NULL, FILE_BEGIN) != 0 ||       !WriteFile(f_handle, syslinux_ldlinux, syslinux_ldlinux_len, &bytes_written, NULL) ||       bytes_written != syslinux_ldlinux_len ) {    error("Could not write ldlinux.sys");    exit(1);  }  /* If desired, fix the MBR */  if( mbr || setactive ) {    STORAGE_DEVICE_NUMBER sdn;    if( GetStorageDeviceNumberByHandle( d_handle, &sdn ) ) {      if( !FixMBR(sdn.DeviceNumber, sdn.PartitionNumber, mbr, setactive) ) {        fprintf(stderr, "Did not successfully update the MBR; continuing...\n");      }    } else {      fprintf(stderr, "Could not find device number for updating MBR; continuing...\n");    }  }  /* Close file */  CloseHandle(f_handle);  /* Move the file to the desired location */  if (subdir) {    char new_ldlinux_name[strlen(subdir)+16];    char *cp = new_ldlinux_name+3;    const char *sd;    int slash = 1;    new_ldlinux_name[0] = drive[0];    new_ldlinux_name[1] = ':';    new_ldlinux_name[2] = '\\';    for (sd = subdir; *sd; sd++) {      char c = *sd;      if (c == '/' || c == '\\') {if (slash)  continue;c = '\\';slash = 1;      } else {slash = 0;      }      *cp++ = c;    }    /* Skip if subdirectory == root */    if (cp > new_ldlinux_name+3) {      if (!slash)*cp++ = '\\';      memcpy(cp, "ldlinux.sys", 12);      /* Delete any previous file */      SetFileAttributes(new_ldlinux_name, FILE_ATTRIBUTE_NORMAL);      DeleteFile(new_ldlinux_name);      if (!MoveFile(ldlinux_name, new_ldlinux_name))SetFileAttributes(ldlinux_name, FILE_ATTRIBUTE_READONLY |  FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN);      elseSetFileAttributes(new_ldlinux_name, FILE_ATTRIBUTE_READONLY |  FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN);    }  }  /* Make the syslinux boot sector */  syslinux_make_bootsect(sectbuf);  /* Write the syslinux boot sector into the boot sector */  if ( bootsecfile ) {    f_handle = CreateFile(bootsecfile, GENERIC_READ | GENERIC_WRITE,  FILE_SHARE_READ | FILE_SHARE_WRITE,  NULL, CREATE_ALWAYS,  FILE_ATTRIBUTE_ARCHIVE,  NULL );    if (f_handle == INVALID_HANDLE_VALUE) {      error("Unable to create bootsector file");      exit(1);    }    if (!WriteFile(f_handle, sectbuf, 512, &bytes_written, NULL)) {      error("Could not write boot sector file");      exit(1);    }    CloseHandle(f_handle);  } else {    SetFilePointer(d_handle, 0, NULL, FILE_BEGIN);    WriteFile( d_handle, sectbuf, 512, &bytes_written, NULL ) ;  }  if(bytes_written != 512) {    fprintf(stderr, "Could not write the whole boot sector\n");    exit(1);  }  /* Close file */  CloseHandle(d_handle);  /* Done! */  return 0;}