Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2025-04-05 21:21:33 +00:00
Code Issues Projects Releases Packages Wiki Activity

[USETUP][SETUPLIB] Move the bootsup.c code into the setuplib, and perform the necessary adjustments.

Browse source 
Move the bootsup.c code into the setuplib, as the code is generic
enough to be used by both USETUP and the 1st-stage GUI installer.

svn path=/branches/setup_improvements/; revision=75674
This commit is contained in:
Hermès Bélusca-Maïto 2017-08-26 11:42:32 +00:00
parent 5467cc9493
commit 95a34ef6fc
No known key found for this signature in database
GPG key ID: 3B2539C65E7B93D0
7 changed files with 147 additions and 111 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
base/setup/lib/CMakeLists.txt
View file

@ -2,6 +2,7 @@
list(APPEND SOURCE list(APPEND SOURCE
arcname.c arcname.c
bldrsup.c bldrsup.c
bootsup.c
filesup.c filesup.c
fsutil.c fsutil.c
genlist.c genlist.c

199
base/setup/usetup/bootsup.c → base/setup/lib/bootsup.c
View file

@ -1,17 +1,31 @@
/* /*
* COPYRIGHT: See COPYING in the top level directory * COPYRIGHT: See COPYING in the top level directory
* PROJECT: ReactOS text-mode setup * PROJECT: ReactOS Setup Library
* FILE: base/setup/usetup/bootsup.c * FILE: base/setup/lib/bootsup.c
* PURPOSE: Bootloader support functions * PURPOSE: Bootloader support functions
* PROGRAMMERS: ... * PROGRAMMERS: ...
* Hermes Belusca-Maito (hermes.belusca@sfr.fr) * Hermes Belusca-Maito (hermes.belusca@sfr.fr)
*/ */
#include "usetup.h" /* INCLUDES *****************************************************************/
#include "precomp.h"
#include "bldrsup.h"
#include "filesup.h"
#include "fsutil.h"
#include "partlist.h"
#include "setuplib.h" // HAXX for IsUnattendedSetup!!
#include "bootsup.h"
#define NDEBUG #define NDEBUG
#include <debug.h> #include <debug.h>
/* TYPEDEFS *****************************************************************/
/* /*
* BIG FIXME!! * BIG FIXME!!
* =========== * ===========
@ -31,66 +45,66 @@
#include <pshpack1.h> #include <pshpack1.h>
typedef struct _FAT_BOOTSECTOR typedef struct _FAT_BOOTSECTOR
{ {
UCHAR JumpBoot[3]; // Jump instruction to boot code UCHAR JumpBoot[3]; // Jump instruction to boot code
CHAR OemName[8]; // "MSWIN4.1" for MS formatted volumes CHAR OemName[8]; // "MSWIN4.1" for MS formatted volumes
USHORT BytesPerSector; // Bytes per sector USHORT BytesPerSector; // Bytes per sector
UCHAR SectorsPerCluster; // Number of sectors in a cluster UCHAR SectorsPerCluster; // Number of sectors in a cluster
USHORT ReservedSectors; // Reserved sectors, usually 1 (the bootsector) USHORT ReservedSectors; // Reserved sectors, usually 1 (the bootsector)
UCHAR NumberOfFats; // Number of FAT tables UCHAR NumberOfFats; // Number of FAT tables
USHORT RootDirEntries; // Number of root directory entries (fat12/16) USHORT RootDirEntries; // Number of root directory entries (fat12/16)
USHORT TotalSectors; // Number of total sectors on the drive, 16-bit USHORT TotalSectors; // Number of total sectors on the drive, 16-bit
UCHAR MediaDescriptor; // Media descriptor byte UCHAR MediaDescriptor; // Media descriptor byte
USHORT SectorsPerFat; // Sectors per FAT table (fat12/16) USHORT SectorsPerFat; // Sectors per FAT table (fat12/16)
USHORT SectorsPerTrack; // Number of sectors in a track USHORT SectorsPerTrack; // Number of sectors in a track
USHORT NumberOfHeads; // Number of heads on the disk USHORT NumberOfHeads; // Number of heads on the disk
ULONG HiddenSectors; // Hidden sectors (sectors before the partition start like the partition table) ULONG HiddenSectors; // Hidden sectors (sectors before the partition start like the partition table)
ULONG TotalSectorsBig; // This field is the new 32-bit total count of sectors on the volume ULONG TotalSectorsBig; // This field is the new 32-bit total count of sectors on the volume
UCHAR DriveNumber; // Int 0x13 drive number (e.g. 0x80) UCHAR DriveNumber; // Int 0x13 drive number (e.g. 0x80)
UCHAR Reserved1; // Reserved (used by Windows NT). Code that formats FAT volumes should always set this byte to 0. UCHAR Reserved1; // Reserved (used by Windows NT). Code that formats FAT volumes should always set this byte to 0.
UCHAR BootSignature; // Extended boot signature (0x29). This is a signature byte that indicates that the following three fields in the boot sector are present. UCHAR BootSignature; // Extended boot signature (0x29). This is a signature byte that indicates that the following three fields in the boot sector are present.
ULONG VolumeSerialNumber; // Volume serial number ULONG VolumeSerialNumber; // Volume serial number
CHAR VolumeLabel[11]; // Volume label. This field matches the 11-byte volume label recorded in the root directory CHAR VolumeLabel[11]; // Volume label. This field matches the 11-byte volume label recorded in the root directory
CHAR FileSystemType[8]; // One of the strings "FAT12 ", "FAT16 ", or "FAT " CHAR FileSystemType[8]; // One of the strings "FAT12 ", "FAT16 ", or "FAT "
UCHAR BootCodeAndData[448]; // The remainder of the boot sector UCHAR BootCodeAndData[448]; // The remainder of the boot sector
USHORT BootSectorMagic; // 0xAA55 USHORT BootSectorMagic; // 0xAA55
} FAT_BOOTSECTOR, *PFAT_BOOTSECTOR; } FAT_BOOTSECTOR, *PFAT_BOOTSECTOR;
typedef struct _FAT32_BOOTSECTOR typedef struct _FAT32_BOOTSECTOR
{ {
UCHAR JumpBoot[3]; // Jump instruction to boot code UCHAR JumpBoot[3]; // Jump instruction to boot code
CHAR OemName[8]; // "MSWIN4.1" for MS formatted volumes CHAR OemName[8]; // "MSWIN4.1" for MS formatted volumes
USHORT BytesPerSector; // Bytes per sector USHORT BytesPerSector; // Bytes per sector
UCHAR SectorsPerCluster; // Number of sectors in a cluster UCHAR SectorsPerCluster; // Number of sectors in a cluster
USHORT ReservedSectors; // Reserved sectors, usually 1 (the bootsector) USHORT ReservedSectors; // Reserved sectors, usually 1 (the bootsector)
UCHAR NumberOfFats; // Number of FAT tables UCHAR NumberOfFats; // Number of FAT tables
USHORT RootDirEntries; // Number of root directory entries (fat12/16) USHORT RootDirEntries; // Number of root directory entries (fat12/16)
USHORT TotalSectors; // Number of total sectors on the drive, 16-bit USHORT TotalSectors; // Number of total sectors on the drive, 16-bit
UCHAR MediaDescriptor; // Media descriptor byte UCHAR MediaDescriptor; // Media descriptor byte
USHORT SectorsPerFat; // Sectors per FAT table (fat12/16) USHORT SectorsPerFat; // Sectors per FAT table (fat12/16)
USHORT SectorsPerTrack; // Number of sectors in a track USHORT SectorsPerTrack; // Number of sectors in a track
USHORT NumberOfHeads; // Number of heads on the disk USHORT NumberOfHeads; // Number of heads on the disk
ULONG HiddenSectors; // Hidden sectors (sectors before the partition start like the partition table) ULONG HiddenSectors; // Hidden sectors (sectors before the partition start like the partition table)
ULONG TotalSectorsBig; // This field is the new 32-bit total count of sectors on the volume ULONG TotalSectorsBig; // This field is the new 32-bit total count of sectors on the volume
ULONG SectorsPerFatBig; // This field is the FAT32 32-bit count of sectors occupied by ONE FAT. BPB_FATSz16 must be 0 ULONG SectorsPerFatBig; // This field is the FAT32 32-bit count of sectors occupied by ONE FAT. BPB_FATSz16 must be 0
USHORT ExtendedFlags; // Extended flags (fat32) USHORT ExtendedFlags; // Extended flags (fat32)
USHORT FileSystemVersion; // File system version (fat32) USHORT FileSystemVersion; // File system version (fat32)
ULONG RootDirStartCluster; // Starting cluster of the root directory (fat32) ULONG RootDirStartCluster; // Starting cluster of the root directory (fat32)
USHORT FsInfo; // Sector number of FSINFO structure in the reserved area of the FAT32 volume. Usually 1. USHORT FsInfo; // Sector number of FSINFO structure in the reserved area of the FAT32 volume. Usually 1.
USHORT BackupBootSector; // If non-zero, indicates the sector number in the reserved area of the volume of a copy of the boot record. Usually 6. USHORT BackupBootSector; // If non-zero, indicates the sector number in the reserved area of the volume of a copy of the boot record. Usually 6.
UCHAR Reserved[12]; // Reserved for future expansion UCHAR Reserved[12]; // Reserved for future expansion
UCHAR DriveNumber; // Int 0x13 drive number (e.g. 0x80) UCHAR DriveNumber; // Int 0x13 drive number (e.g. 0x80)
UCHAR Reserved1; // Reserved (used by Windows NT). Code that formats FAT volumes should always set this byte to 0. UCHAR Reserved1; // Reserved (used by Windows NT). Code that formats FAT volumes should always set this byte to 0.
UCHAR BootSignature; // Extended boot signature (0x29). This is a signature byte that indicates that the following three fields in the boot sector are present. UCHAR BootSignature; // Extended boot signature (0x29). This is a signature byte that indicates that the following three fields in the boot sector are present.
ULONG VolumeSerialNumber; // Volume serial number ULONG VolumeSerialNumber; // Volume serial number
CHAR VolumeLabel[11]; // Volume label. This field matches the 11-byte volume label recorded in the root directory CHAR VolumeLabel[11]; // Volume label. This field matches the 11-byte volume label recorded in the root directory
CHAR FileSystemType[8]; // Always set to the string "FAT32 " CHAR FileSystemType[8]; // Always set to the string "FAT32 "
UCHAR BootCodeAndData[420]; // The remainder of the boot sector UCHAR BootCodeAndData[420]; // The remainder of the boot sector
USHORT BootSectorMagic; // 0xAA55 USHORT BootSectorMagic; // 0xAA55
} FAT32_BOOTSECTOR, *PFAT32_BOOTSECTOR; } FAT32_BOOTSECTOR, *PFAT32_BOOTSECTOR;
@ -515,6 +529,7 @@ UpdateBootIni(
} }
static
BOOLEAN BOOLEAN
IsThereAValidBootSector( IsThereAValidBootSector(
IN PCWSTR RootPath) IN PCWSTR RootPath)
@ -591,6 +606,7 @@ Quit:
return IsValid; // Status; return IsValid; // Status;
} }
static
NTSTATUS NTSTATUS
SaveBootSector( SaveBootSector(
IN PCWSTR RootPath, IN PCWSTR RootPath,
@ -693,8 +709,10 @@ SaveBootSector(
return Status; return Status;
} }
static
NTSTATUS NTSTATUS
InstallMbrBootCodeToDisk( InstallMbrBootCodeToDiskHelper(
IN PCWSTR SrcPath, IN PCWSTR SrcPath,
IN PCWSTR RootPath) IN PCWSTR RootPath)
{ {
@ -853,6 +871,48 @@ InstallMbrBootCodeToDisk(
return Status; return Status;
} }
NTSTATUS
InstallMbrBootCodeToDisk(
IN PUNICODE_STRING SystemRootPath,
IN PUNICODE_STRING SourceRootPath,
IN PCWSTR DestinationDevicePathBuffer)
{
NTSTATUS Status;
WCHAR SourceMbrPathBuffer[MAX_PATH];
WCHAR DstPath[MAX_PATH];
#if 0
WCHAR DestinationDevicePathBuffer[MAX_PATH];
StringCchPrintfW(DestinationDevicePathBuffer, ARRAYSIZE(DestinationDevicePathBuffer),
L"\\Device\\Harddisk%d\\Partition0",
DiskNumber);
#endif
CombinePaths(SourceMbrPathBuffer, ARRAYSIZE(SourceMbrPathBuffer), 2,
SourceRootPath->Buffer, L"\\loader\\dosmbr.bin");
if (IsThereAValidBootSector(DestinationDevicePathBuffer))
{
/* Save current MBR */
CombinePaths(DstPath, ARRAYSIZE(DstPath), 2,
SystemRootPath->Buffer, L"mbr.old");
DPRINT1("Save MBR: %S ==> %S\n", DestinationDevicePathBuffer, DstPath);
Status = SaveBootSector(DestinationDevicePathBuffer, DstPath, sizeof(PARTITION_SECTOR));
if (!NT_SUCCESS(Status))
{
DPRINT1("SaveBootSector() failed (Status %lx)\n", Status);
// Don't care if we succeeded or not saving the old MBR, just go ahead.
}
}
DPRINT1("Install MBR bootcode: %S ==> %S\n",
SourceMbrPathBuffer, DestinationDevicePathBuffer);
return InstallMbrBootCodeToDiskHelper(SourceMbrPathBuffer,
DestinationDevicePathBuffer);
}
static static
NTSTATUS NTSTATUS
@ -1767,13 +1827,14 @@ InstallBtrfsBootCodeToDisk(
return Status; return Status;
} }
static static
NTSTATUS NTSTATUS
InstallFatBootcodeToPartition( InstallFatBootcodeToPartition(
PUNICODE_STRING SystemRootPath, IN PUNICODE_STRING SystemRootPath,
PUNICODE_STRING SourceRootPath, IN PUNICODE_STRING SourceRootPath,
PUNICODE_STRING DestinationArcPath, IN PUNICODE_STRING DestinationArcPath,
UCHAR PartitionType) IN UCHAR PartitionType)
{ {
NTSTATUS Status; NTSTATUS Status;
BOOLEAN DoesFreeLdrExist; BOOLEAN DoesFreeLdrExist;
@ -2083,10 +2144,10 @@ InstallFatBootcodeToPartition(
static static
NTSTATUS NTSTATUS
InstallBtrfsBootcodeToPartition( InstallBtrfsBootcodeToPartition(
PUNICODE_STRING SystemRootPath, IN PUNICODE_STRING SystemRootPath,
PUNICODE_STRING SourceRootPath, IN PUNICODE_STRING SourceRootPath,
PUNICODE_STRING DestinationArcPath, IN PUNICODE_STRING DestinationArcPath,
UCHAR PartitionType) IN UCHAR PartitionType)
{ {
NTSTATUS Status; NTSTATUS Status;
BOOLEAN DoesFreeLdrExist; BOOLEAN DoesFreeLdrExist;
@ -2190,10 +2251,10 @@ InstallBtrfsBootcodeToPartition(
NTSTATUS NTSTATUS
InstallVBRToPartition( InstallVBRToPartition(
PUNICODE_STRING SystemRootPath, IN PUNICODE_STRING SystemRootPath,
PUNICODE_STRING SourceRootPath, IN PUNICODE_STRING SourceRootPath,
PUNICODE_STRING DestinationArcPath, IN PUNICODE_STRING DestinationArcPath,
UCHAR PartitionType) IN UCHAR PartitionType)
{ {
switch (PartitionType) switch (PartitionType)
{ {
@ -2233,8 +2294,8 @@ InstallVBRToPartition(
NTSTATUS NTSTATUS
InstallFatBootcodeToFloppy( InstallFatBootcodeToFloppy(
PUNICODE_STRING SourceRootPath, IN PUNICODE_STRING SourceRootPath,
PUNICODE_STRING DestinationArcPath) IN PUNICODE_STRING DestinationArcPath)
{ {
NTSTATUS Status; NTSTATUS Status;
PFILE_SYSTEM FatFS; PFILE_SYSTEM FatFS;

30
base/setup/usetup/bootsup.h → base/setup/lib/bootsup.h
View file

@ -18,38 +18,30 @@
*/ */
/* /*
* COPYRIGHT: See COPYING in the top level directory * COPYRIGHT: See COPYING in the top level directory
* PROJECT: ReactOS text-mode setup * PROJECT: ReactOS Setup Library
* FILE: base/setup/usetup/bootsup.h * FILE: base/setup/lib/bootsup.h
* PURPOSE: Bootloader support functions * PURPOSE: Bootloader support functions
* PROGRAMMER: * PROGRAMMER:
*/ */
#pragma once #pragma once
BOOLEAN
IsThereAValidBootSector(PWSTR RootPath);
NTSTATUS
SaveBootSector(
PWSTR RootPath,
PWSTR DstPath,
ULONG Length);
NTSTATUS NTSTATUS
InstallMbrBootCodeToDisk( InstallMbrBootCodeToDisk(
PWSTR SrcPath, IN PUNICODE_STRING SystemRootPath,
PWSTR RootPath); IN PUNICODE_STRING SourceRootPath,
IN PCWSTR DestinationDevicePathBuffer);
NTSTATUS NTSTATUS
InstallVBRToPartition( InstallVBRToPartition(
PUNICODE_STRING SystemRootPath, IN PUNICODE_STRING SystemRootPath,
PUNICODE_STRING SourceRootPath, IN PUNICODE_STRING SourceRootPath,
PUNICODE_STRING DestinationArcPath, IN PUNICODE_STRING DestinationArcPath,
UCHAR PartitionType); IN UCHAR PartitionType);
NTSTATUS NTSTATUS
InstallFatBootcodeToFloppy( InstallFatBootcodeToFloppy(
PUNICODE_STRING SourceRootPath, IN PUNICODE_STRING SourceRootPath,
PUNICODE_STRING DestinationArcPath); IN PUNICODE_STRING DestinationArcPath);
/* EOF */ /* EOF */

2
base/setup/lib/fsutil.h
View file

@ -6,6 +6,8 @@
* Copyright 2017-2018 Hermes Belusca-Maito * Copyright 2017-2018 Hermes Belusca-Maito
*/ */
#pragma once
#include <fmifs/fmifs.h> #include <fmifs/fmifs.h>
typedef struct _FILE_SYSTEM typedef struct _FILE_SYSTEM

1
base/setup/lib/setuplib.h
View file

@ -29,6 +29,7 @@ extern HANDLE ProcessHeap;
#include "ntverrsrc.h" #include "ntverrsrc.h"
// #include "arcname.h" // #include "arcname.h"
#include "bldrsup.h" #include "bldrsup.h"
#include "bootsup.h"
#include "filesup.h" #include "filesup.h"
#include "fsutil.h" #include "fsutil.h"
#include "genlist.h" #include "genlist.h"

1
base/setup/usetup/CMakeLists.txt
View file

@ -10,7 +10,6 @@ include_directories(
${REACTOS_SOURCE_DIR}/sdk/include/reactos/drivers) ${REACTOS_SOURCE_DIR}/sdk/include/reactos/drivers)
list(APPEND SOURCE list(APPEND SOURCE
bootsup.c
cabinet.c cabinet.c
chkdsk.c chkdsk.c
cmdcons.c cmdcons.c

24
base/setup/usetup/usetup.c
View file

@ -4601,8 +4601,6 @@ BootLoaderHarddiskMbrPage(PINPUT_RECORD Ir)
{ {
NTSTATUS Status; NTSTATUS Status;
WCHAR DestinationDevicePathBuffer[MAX_PATH]; WCHAR DestinationDevicePathBuffer[MAX_PATH];
WCHAR SourceMbrPathBuffer[MAX_PATH];
WCHAR DstPath[MAX_PATH];
/* Step 1: Write the VBR */ /* Step 1: Write the VBR */
Status = InstallVBRToPartition(&USetupData.SystemRootPath, Status = InstallVBRToPartition(&USetupData.SystemRootPath,
@ -4619,26 +4617,8 @@ BootLoaderHarddiskMbrPage(PINPUT_RECORD Ir)
StringCchPrintfW(DestinationDevicePathBuffer, ARRAYSIZE(DestinationDevicePathBuffer), StringCchPrintfW(DestinationDevicePathBuffer, ARRAYSIZE(DestinationDevicePathBuffer),
L"\\Device\\Harddisk%d\\Partition0", L"\\Device\\Harddisk%d\\Partition0",
PartitionList->SystemPartition->DiskEntry->DiskNumber); PartitionList->SystemPartition->DiskEntry->DiskNumber);
Status = InstallMbrBootCodeToDisk(&USetupData.SystemRootPath,
CombinePaths(SourceMbrPathBuffer, ARRAYSIZE(SourceMbrPathBuffer), 2, USetupData.SourceRootPath.Buffer, L"\\loader\\dosmbr.bin"); &USetupData.SourceRootPath,
if (IsThereAValidBootSector(DestinationDevicePathBuffer))
{
/* Save current MBR */
CombinePaths(DstPath, ARRAYSIZE(DstPath), 2, USetupData.SystemRootPath.Buffer, L"mbr.old");
DPRINT1("Save MBR: %S ==> %S\n", DestinationDevicePathBuffer, DstPath);
Status = SaveBootSector(DestinationDevicePathBuffer, DstPath, sizeof(PARTITION_SECTOR));
if (!NT_SUCCESS(Status))
{
DPRINT1("SaveBootSector() failed (Status %lx)\n", Status);
// Don't care if we succeeded or not saving the old MBR, just go ahead.
}
}
DPRINT1("Install MBR bootcode: %S ==> %S\n",
SourceMbrPathBuffer, DestinationDevicePathBuffer);
Status = InstallMbrBootCodeToDisk(SourceMbrPathBuffer,
DestinationDevicePathBuffer); DestinationDevicePathBuffer);
if (!NT_SUCCESS(Status)) if (!NT_SUCCESS(Status))
{ {