reactos/boot/freeldr/install/linux/finstext2.c

   1 #include <stdio.h>
   2 #include <string.h>
   3 #include "finstext2.h"
   4 #include "ext2.h"       // #defines ext2_data
   5 #include <linux/hdreg.h>
   6 #include <linux/ext3_fs.h>
   7 #include <sys/ioctl.h>
   8 #include <sys/types.h>
   9 #include <sys/stat.h>
  10 #include <fcntl.h>
  11 #include <unistd.h>
  12
  13 #define MAJOR_VERSION   1
  14 #define MINOR_VERSION   0
  15
  16 PEXT2_BOOTCODE                  Ext2BootCode = (PEXT2_BOOTCODE)ext2_data;
  17 struct ext3_super_block Ext2SuperBlock;
  18 struct hd_geometry              Ext2DriveGeometry;
  19 unsigned char                   Ext2ExistingBootCode[1024];
  20
  21 char                                    BlockDevice[260];
  22 int                                             BlockDeviceSpecified = 0;
  23 int                                             BlockDeviceFileDescriptor;
  24
  25 int main(int argc, char *argv[])
  26 {
  27         int             Index;
  28         char    ch;
  29
  30         // Verify that we got enough command line parameters
  31         if (argc < 2)
  32         {
  33                 printf("finstext2 block_device [-v]\n\n");
  34                 printf("block_device    Specifies the ext2/3 volume to install to (i.e. /dev/hda1)\n");
  35                 printf("-v              Displays the version\n\n");
  36         }
  37
  38         // Parse the command line parameters
  39         for (Index=1; Index<argc; Index++)
  40         {
  41                 if (strcmp(argv[Index], "-v") == 0)
  42                 {
  43                         printf("FreeLoader Ext2/3 Installer v%d.%d\n", MAJOR_VERSION, MINOR_VERSION);
  44                 }
  45                 else
  46                 {
  47                         strcpy(BlockDevice, argv[Index]);
  48                         BlockDeviceSpecified = 1;
  49                 }
  50         }
  51
  52         if (BlockDeviceSpecified)
  53         {
  54                 BlockDeviceFileDescriptor = open(BlockDevice, O_RDWR|O_SYNC);
  55
  56                 if (BlockDeviceFileDescriptor == -1)
  57                 {
  58                         printf("Couldn't open block device %s\n", BlockDevice);
  59                         return 1;
  60                 }
  61
  62                 if (read(BlockDeviceFileDescriptor, Ext2ExistingBootCode, (size_t)1024) != (ssize_t)1024)
  63                 {
  64                         close(BlockDeviceFileDescriptor);
  65                         printf("Couldn't read existing boot code from %s\n", BlockDevice);
  66                         return 1;
  67                 }
  68
  69                 for (Index=0; Index<1024; Index++)
  70                 {
  71                         if (Ext2ExistingBootCode[Index] != 0x00)
  72                         {
  73                                 printf("This EXT2/3 volume has existing boot code.\n");
  74                                 printf("Do you want to overwrite it? [y/n] ");
  75                                 scanf("%c", &ch);
  76
  77                                 if (ch == 'n' || ch == 'N')
  78                                 {
  79                                         close(BlockDeviceFileDescriptor);
  80                                         printf("Cancelled.\n");
  81                                         return 0;
  82                                 }
  83
  84                                 break;
  85                         }
  86                 }
  87
  88                 if (read(BlockDeviceFileDescriptor, &Ext2SuperBlock, (size_t)1024) != (ssize_t)1024)
  89                 {
  90                         close(BlockDeviceFileDescriptor);
  91                         printf("Couldn't read super block from %s\n", BlockDevice);
  92                         return 1;
  93                 }
  94
  95                 if (Ext2SuperBlock.s_magic != EXT3_SUPER_MAGIC)
  96                 {
  97                         close(BlockDeviceFileDescriptor);
  98                         printf("Block device %s is not a EXT2/3 volume or has an invalid super block.\n", BlockDevice);
  99                         return 1;
 100                 }
 101
 102                 if (ioctl(BlockDeviceFileDescriptor, HDIO_GETGEO, &Ext2DriveGeometry) != 0)
 103                 {
 104                         close(BlockDeviceFileDescriptor);
 105                         printf("Couldn't get drive geometry from block device %s\n", BlockDevice);
 106                         return 1;
 107                 }
 108
 109                 printf("Heads: %d\n", Ext2DriveGeometry.heads);
 110                 printf("Sectors: %d\n", Ext2DriveGeometry.sectors);
 111                 printf("Cylinders: %d\n", Ext2DriveGeometry.cylinders);
 112                 printf("Start: %d\n", Ext2DriveGeometry.start);
 113
 114                 Ext2BootCode->BootDrive = 0xff;
 115                 Ext2BootCode->BootPartition = 0x00;
 116                 //Ext2BootCode->SectorsPerTrack = Ext2DriveGeometry.sectors;
 117                 //Ext2BootCode->NumberOfHeads = Ext2DriveGeometry.heads;
 118                 Ext2BootCode->Ext2VolumeStartSector = Ext2DriveGeometry.start;
 119                 Ext2BootCode->Ext2BlockSizeInBytes = 1024 << Ext2SuperBlock.s_log_block_size;
 120                 Ext2BootCode->Ext2BlockSize = Ext2BootCode->Ext2BlockSizeInBytes / 512;
 121                 Ext2BootCode->Ext2PointersPerBlock = Ext2BootCode->Ext2BlockSizeInBytes / 4;
 122                 Ext2BootCode->Ext2GroupDescPerBlock = Ext2BootCode->Ext2BlockSizeInBytes / 32;
 123                 Ext2BootCode->Ext2FirstDataBlock = Ext2SuperBlock.s_first_data_block;
 124                 Ext2BootCode->Ext2InodesPerGroup = Ext2SuperBlock.s_inodes_per_group;
 125                 Ext2BootCode->Ext2InodesPerBlock = Ext2BootCode->Ext2BlockSizeInBytes / 128;
 126
 127                 if (lseek(BlockDeviceFileDescriptor, (off_t)0, SEEK_SET) == (off_t)-1)
 128                 {
 129                         close(BlockDeviceFileDescriptor);
 130                         printf("Couldn't write boot code on %s\n", BlockDevice);
 131                         return 1;
 132                 }
 133
 134                 if (write(BlockDeviceFileDescriptor, Ext2BootCode, (size_t)1024) != (ssize_t)1024)
 135                 {
 136                         close(BlockDeviceFileDescriptor);
 137                         printf("Couldn't write boot code on %s\n", BlockDevice);
 138                         return 1;
 139                 }
 140
 141                 close(BlockDeviceFileDescriptor);
 142
 143                 printf("Boot code written successfully!\n");
 144         }
 145
 146         return 0;
 147 }