boot/freeldr/freeldr/custom.c

   1 /*
   2  *  FreeLoader
   3  *  Copyright (C) 1998-2003  Brian Palmer  <brianp@sginet.com>
   4  *
   5  *  This program is free software; you can redistribute it and/or modify
   6  *  it under the terms of the GNU General Public License as published by
   7  *  the Free Software Foundation; either version 2 of the License, or
   8  *  (at your option) any later version.
   9  *
  10  *  This program is distributed in the hope that it will be useful,
  11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  *  GNU General Public License for more details.
  14  *
  15  *  You should have received a copy of the GNU General Public License along
  16  *  with this program; if not, write to the Free Software Foundation, Inc.,
  17  *  51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  18  */
  19
  20 /* INCLUDES *******************************************************************/
  21
  22 #include <freeldr.h>
  23
  24 /* GLOBALS ********************************************************************/
  25
  26 #ifdef _M_IX86
  27
  28 const CHAR BootSectorFilePrompt[] = "Enter the boot sector file path.\n\nExamples:\n\\BOOTSECT.DOS\n/boot/bootsect.dos";
  29 const CHAR LinuxKernelPrompt[] = "Enter the Linux kernel image path.\n\nExamples:\n/vmlinuz\n/boot/vmlinuz-2.4.18";
  30 const CHAR LinuxInitrdPrompt[] = "Enter the initrd image path.\n\nExamples:\n/initrd.gz\n/boot/root.img.gz\n\nLeave blank for no initial ram disk.";
  31 const CHAR LinuxCommandLinePrompt[] = "Enter the Linux kernel command line.\n\nExamples:\nroot=/dev/hda1\nroot=/dev/fd0 read-only\nroot=/dev/sdb1 init=/sbin/init";
  32
  33 #endif // _M_IX86
  34
  35 const CHAR BootDrivePrompt[] = "Enter the boot drive.\n\nExamples:\nfd0 - first floppy drive\nhd0 - first hard drive\nhd1 - second hard drive\ncd0 - first CD-ROM drive.\n\nBIOS drive numbers may also be used:\n0 - first floppy drive\n0x80 - first hard drive\n0x81 - second hard drive";
  36 const CHAR BootPartitionPrompt[] = "Enter the boot partition.\n\nEnter 0 for the active (bootable) partition.";
  37 const CHAR ReactOSSystemPathPrompt[] = "Enter the path to your ReactOS system directory.\n\nExamples:\n\\REACTOS\n\\ROS";
  38 const CHAR ReactOSOptionsPrompt[] = "Enter the options you want passed to the kernel.\n\nExamples:\n/DEBUG /DEBUGPORT=COM1 /BAUDRATE=115200\n/FASTDETECT /SOS /NOGUIBOOT\n/BASEVIDEO /MAXMEM=64\n/KERNEL=NTKRNLMP.EXE /HAL=HALMPS.DLL";
  39 const CHAR CustomBootPrompt[] = "Press ENTER to boot your custom boot setup.";
  40
  41 /* FUNCTIONS ******************************************************************/
  42
  43 VOID OptionMenuCustomBoot(VOID)
  44 {
  45     PCSTR CustomBootMenuList[] = {
  46 #ifdef _M_IX86
  47         "Disk",
  48         "Partition",
  49         "Boot Sector File",
  50         "Linux",
  51 #endif
  52         "ReactOS"
  53         };
  54     ULONG SelectedMenuItem;
  55
  56     if (!UiDisplayMenu("Please choose a boot method:", "",
  57                        FALSE,
  58                        CustomBootMenuList,
  59                        sizeof(CustomBootMenuList) / sizeof(CustomBootMenuList[0]),
  60                        0, -1,
  61                        &SelectedMenuItem,
  62                        TRUE,
  63                        NULL))
  64     {
  65         /* The user pressed ESC */
  66         return;
  67     }
  68
  69     switch (SelectedMenuItem)
  70     {
  71 #ifdef _M_IX86
  72         case 0: // Disk
  73             OptionMenuCustomBootDisk();
  74             break;
  75         case 1: // Partition
  76             OptionMenuCustomBootPartition();
  77             break;
  78         case 2: // Boot Sector File
  79             OptionMenuCustomBootBootSectorFile();
  80             break;
  81         case 3: // Linux
  82             OptionMenuCustomBootLinux();
  83             break;
  84         case 4: // ReactOS
  85 #else
  86         case 0:
  87 #endif
  88             OptionMenuCustomBootReactOS();
  89             break;
  90     }
  91 }
  92
  93 #ifdef _M_IX86
  94
  95 VOID OptionMenuCustomBootDisk(VOID)
  96 {
  97     ULONG_PTR SectionId;
  98     CHAR SectionName[100];
  99     CHAR BootDriveString[20];
 100     TIMEINFO* TimeInfo;
 101     OperatingSystemItem OperatingSystem;
 102
 103     RtlZeroMemory(SectionName, sizeof(SectionName));
 104     RtlZeroMemory(BootDriveString, sizeof(BootDriveString));
 105
 106     if (!UiEditBox(BootDrivePrompt, BootDriveString, 20))
 107         return;
 108
 109     /* Generate a unique section name */
 110     TimeInfo = ArcGetTime();
 111     RtlStringCbPrintfA(SectionName, sizeof(SectionName),
 112                        "CustomBootDisk%u%u%u%u%u%u",
 113                        TimeInfo->Year, TimeInfo->Day, TimeInfo->Month,
 114                        TimeInfo->Hour, TimeInfo->Minute, TimeInfo->Second);
 115
 116     /* Add the section */
 117     if (!IniAddSection(SectionName, &SectionId))
 118         return;
 119
 120     /* Add the BootType */
 121     if (!IniAddSettingValueToSection(SectionId, "BootType", "Drive"))
 122         return;
 123
 124     /* Add the BootDrive */
 125     if (!IniAddSettingValueToSection(SectionId, "BootDrive", BootDriveString))
 126         return;
 127
 128     UiMessageBox(CustomBootPrompt);
 129
 130     OperatingSystem.SectionName = SectionName;
 131     OperatingSystem.LoadIdentifier = NULL;
 132     LoadOperatingSystem(&OperatingSystem);
 133 }
 134
 135 VOID OptionMenuCustomBootPartition(VOID)
 136 {
 137     ULONG_PTR SectionId;
 138     CHAR SectionName[100];
 139     CHAR BootDriveString[20];
 140     CHAR BootPartitionString[20];
 141     TIMEINFO* TimeInfo;
 142     OperatingSystemItem OperatingSystem;
 143
 144     RtlZeroMemory(SectionName, sizeof(SectionName));
 145     RtlZeroMemory(BootDriveString, sizeof(BootDriveString));
 146     RtlZeroMemory(BootPartitionString, sizeof(BootPartitionString));
 147
 148     if (!UiEditBox(BootDrivePrompt, BootDriveString, 20))
 149         return;
 150
 151     if (!UiEditBox(BootPartitionPrompt, BootPartitionString, 20))
 152         return;
 153
 154     /* Generate a unique section name */
 155     TimeInfo = ArcGetTime();
 156     RtlStringCbPrintfA(SectionName, sizeof(SectionName),
 157                        "CustomBootPartition%u%u%u%u%u%u",
 158                        TimeInfo->Year, TimeInfo->Day, TimeInfo->Month,
 159                        TimeInfo->Hour, TimeInfo->Minute, TimeInfo->Second);
 160
 161     /* Add the section */
 162     if (!IniAddSection(SectionName, &SectionId))
 163         return;
 164
 165     /* Add the BootType */
 166     if (!IniAddSettingValueToSection(SectionId, "BootType", "Partition"))
 167         return;
 168
 169     /* Add the BootDrive */
 170     if (!IniAddSettingValueToSection(SectionId, "BootDrive", BootDriveString))
 171         return;
 172
 173     /* Add the BootPartition */
 174     if (!IniAddSettingValueToSection(SectionId, "BootPartition", BootPartitionString))
 175         return;
 176
 177     UiMessageBox(CustomBootPrompt);
 178
 179     OperatingSystem.SectionName = SectionName;
 180     OperatingSystem.LoadIdentifier = NULL;
 181     LoadOperatingSystem(&OperatingSystem);
 182 }
 183
 184 VOID OptionMenuCustomBootBootSectorFile(VOID)
 185 {
 186     ULONG_PTR SectionId;
 187     CHAR SectionName[100];
 188     CHAR BootDriveString[20];
 189     CHAR BootPartitionString[20];
 190     CHAR BootSectorFileString[200];
 191     TIMEINFO* TimeInfo;
 192     OperatingSystemItem OperatingSystem;
 193
 194     RtlZeroMemory(SectionName, sizeof(SectionName));
 195     RtlZeroMemory(BootDriveString, sizeof(BootDriveString));
 196     RtlZeroMemory(BootPartitionString, sizeof(BootPartitionString));
 197     RtlZeroMemory(BootSectorFileString, sizeof(BootSectorFileString));
 198
 199     if (!UiEditBox(BootDrivePrompt, BootDriveString, 20))
 200         return;
 201
 202     if (!UiEditBox(BootPartitionPrompt, BootPartitionString, 20))
 203         return;
 204
 205     if (!UiEditBox(BootSectorFilePrompt, BootSectorFileString, 200))
 206         return;
 207
 208     /* Generate a unique section name */
 209     TimeInfo = ArcGetTime();
 210     RtlStringCbPrintfA(SectionName, sizeof(SectionName),
 211                        "CustomBootSectorFile%u%u%u%u%u%u",
 212                        TimeInfo->Year, TimeInfo->Day, TimeInfo->Month,
 213                        TimeInfo->Hour, TimeInfo->Minute, TimeInfo->Second);
 214
 215     /* Add the section */
 216     if (!IniAddSection(SectionName, &SectionId))
 217         return;
 218
 219     /* Add the BootType */
 220     if (!IniAddSettingValueToSection(SectionId, "BootType", "BootSector"))
 221         return;
 222
 223     /* Add the BootDrive */
 224     if (!IniAddSettingValueToSection(SectionId, "BootDrive", BootDriveString))
 225         return;
 226
 227     /* Add the BootPartition */
 228     if (!IniAddSettingValueToSection(SectionId, "BootPartition", BootPartitionString))
 229         return;
 230
 231     /* Add the BootSectorFile */
 232     if (!IniAddSettingValueToSection(SectionId, "BootSectorFile", BootSectorFileString))
 233         return;
 234
 235     UiMessageBox(CustomBootPrompt);
 236
 237     OperatingSystem.SectionName = SectionName;
 238     OperatingSystem.LoadIdentifier = NULL;
 239     LoadOperatingSystem(&OperatingSystem);
 240 }
 241
 242 VOID OptionMenuCustomBootLinux(VOID)
 243 {
 244     ULONG_PTR SectionId;
 245     CHAR SectionName[100];
 246     CHAR BootDriveString[20];
 247     CHAR BootPartitionString[20];
 248     CHAR LinuxKernelString[200];
 249     CHAR LinuxInitrdString[200];
 250     CHAR LinuxCommandLineString[200];
 251     TIMEINFO* TimeInfo;
 252     OperatingSystemItem OperatingSystem;
 253
 254     RtlZeroMemory(SectionName, sizeof(SectionName));
 255     RtlZeroMemory(BootDriveString, sizeof(BootDriveString));
 256     RtlZeroMemory(BootPartitionString, sizeof(BootPartitionString));
 257     RtlZeroMemory(LinuxKernelString, sizeof(LinuxKernelString));
 258     RtlZeroMemory(LinuxInitrdString, sizeof(LinuxInitrdString));
 259     RtlZeroMemory(LinuxCommandLineString, sizeof(LinuxCommandLineString));
 260
 261     if (!UiEditBox(BootDrivePrompt, BootDriveString, 20))
 262         return;
 263
 264     if (!UiEditBox(BootPartitionPrompt, BootPartitionString, 20))
 265         return;
 266
 267     if (!UiEditBox(LinuxKernelPrompt, LinuxKernelString, 200))
 268         return;
 269
 270     if (!UiEditBox(LinuxInitrdPrompt, LinuxInitrdString, 200))
 271         return;
 272
 273     if (!UiEditBox(LinuxCommandLinePrompt, LinuxCommandLineString, 200))
 274         return;
 275
 276     /* Generate a unique section name */
 277     TimeInfo = ArcGetTime();
 278     RtlStringCbPrintfA(SectionName, sizeof(SectionName),
 279                        "CustomLinux%u%u%u%u%u%u",
 280                        TimeInfo->Year, TimeInfo->Day, TimeInfo->Month,
 281                        TimeInfo->Hour, TimeInfo->Minute, TimeInfo->Second);
 282
 283     /* Add the section */
 284     if (!IniAddSection(SectionName, &SectionId))
 285         return;
 286
 287     /* Add the BootType */
 288     if (!IniAddSettingValueToSection(SectionId, "BootType", "Linux"))
 289         return;
 290
 291     /* Add the BootDrive */
 292     if (!IniAddSettingValueToSection(SectionId, "BootDrive", BootDriveString))
 293         return;
 294
 295     /* Add the BootPartition */
 296     if (!IniAddSettingValueToSection(SectionId, "BootPartition", BootPartitionString))
 297         return;
 298
 299     /* Add the Kernel */
 300     if (!IniAddSettingValueToSection(SectionId, "Kernel", LinuxKernelString))
 301         return;
 302
 303     /* Add the Initrd */
 304     if (*LinuxInitrdString)
 305     {
 306         if (!IniAddSettingValueToSection(SectionId, "Initrd", LinuxInitrdString))
 307             return;
 308     }
 309
 310     /* Add the CommandLine */
 311     if (!IniAddSettingValueToSection(SectionId, "CommandLine", LinuxCommandLineString))
 312         return;
 313
 314     UiMessageBox(CustomBootPrompt);
 315
 316     OperatingSystem.SectionName = SectionName;
 317     OperatingSystem.LoadIdentifier = "Custom Linux Setup";
 318     LoadOperatingSystem(&OperatingSystem);
 319 }
 320
 321 #endif // _M_IX86
 322
 323 VOID OptionMenuCustomBootReactOS(VOID)
 324 {
 325     ULONG_PTR SectionId;
 326     CHAR SectionName[100];
 327     CHAR BootDriveString[20];
 328     CHAR BootPartitionString[20];
 329     CHAR ReactOSSystemPath[200];
 330     CHAR ReactOSARCPath[200];
 331     CHAR ReactOSOptions[200];
 332     TIMEINFO* TimeInfo;
 333     OperatingSystemItem OperatingSystem;
 334
 335     RtlZeroMemory(SectionName, sizeof(SectionName));
 336     RtlZeroMemory(BootDriveString, sizeof(BootDriveString));
 337     RtlZeroMemory(BootPartitionString, sizeof(BootPartitionString));
 338     RtlZeroMemory(ReactOSSystemPath, sizeof(ReactOSSystemPath));
 339     RtlZeroMemory(ReactOSOptions, sizeof(ReactOSOptions));
 340
 341     if (!UiEditBox(BootDrivePrompt, BootDriveString, 20))
 342         return;
 343
 344     if (!UiEditBox(BootPartitionPrompt, BootPartitionString, 20))
 345         return;
 346
 347     if (!UiEditBox(ReactOSSystemPathPrompt, ReactOSSystemPath, 200))
 348         return;
 349
 350     if (!UiEditBox(ReactOSOptionsPrompt, ReactOSOptions, 200))
 351         return;
 352
 353     /* Generate a unique section name */
 354     TimeInfo = ArcGetTime();
 355     RtlStringCbPrintfA(SectionName, sizeof(SectionName),
 356                        "CustomReactOS%u%u%u%u%u%u",
 357                        TimeInfo->Year, TimeInfo->Day, TimeInfo->Month,
 358                        TimeInfo->Hour, TimeInfo->Minute, TimeInfo->Second);
 359
 360     /* Add the section */
 361     if (!IniAddSection(SectionName, &SectionId))
 362         return;
 363
 364     /* Add the BootType */
 365     if (!IniAddSettingValueToSection(SectionId, "BootType", "Windows2003"))
 366         return;
 367
 368     /* Construct the ReactOS ARC system path */
 369     ConstructArcPath(ReactOSARCPath, ReactOSSystemPath, DriveMapGetBiosDriveNumber(BootDriveString), atoi(BootPartitionString));
 370
 371     /* Add the system path */
 372     if (!IniAddSettingValueToSection(SectionId, "SystemPath", ReactOSARCPath))
 373         return;
 374
 375     /* Add the CommandLine */
 376     if (!IniAddSettingValueToSection(SectionId, "Options", ReactOSOptions))
 377         return;
 378
 379     UiMessageBox(CustomBootPrompt);
 380
 381     OperatingSystem.SectionName = SectionName;
 382     OperatingSystem.LoadIdentifier = NULL;
 383     LoadOperatingSystem(&OperatingSystem);
 384 }
 385
 386 VOID OptionMenuReboot(VOID)
 387 {
 388     UiMessageBox("The system will now reboot.");
 389
 390 #if defined(__i386__) || defined(_M_AMD64)
 391     DiskStopFloppyMotor();
 392 #endif
 393     Reboot();
 394 }