fd737752c901257cdfb2aa14eaee1f65e335a8cd
[reactos.git] / boot / freeldr / freeldr / windows / setupldr2.c
1 /*
2 * FreeLoader
3 *
4 * Copyright (C) 2009 Aleksey Bragin <aleksey@reactos.org>
5 *
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation; either version 2 of the License, or
9 * (at your option) any later version.
10 *
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
15 *
16 * You should have received a copy of the GNU General Public License along
17 * with this program; if not, write to the Free Software Foundation, Inc.,
18 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
19 */
20
21 #include <freeldr.h>
22
23 #include <ndk/ldrtypes.h>
24 #include <arc/setupblk.h>
25
26 #include <debug.h>
27
28 // TODO: Move to .h
29 VOID AllocateAndInitLPB(PLOADER_PARAMETER_BLOCK *OutLoaderBlock);
30 BOOLEAN WinLdrLoadBootDrivers(PLOADER_PARAMETER_BLOCK LoaderBlock, LPSTR BootPath);
31 void WinLdrSetupForNt(PLOADER_PARAMETER_BLOCK LoaderBlock,
32 PVOID *GdtIdt,
33 ULONG *PcrBasePage,
34 ULONG *TssBasePage);
35 VOID
36 WinLdrInitializePhase1(PLOADER_PARAMETER_BLOCK LoaderBlock,
37 PCHAR Options,
38 PCHAR SystemPath,
39 PCHAR BootPath,
40 USHORT VersionToBoot);
41 BOOLEAN
42 WinLdrLoadNLSData(IN OUT PLOADER_PARAMETER_BLOCK LoaderBlock,
43 IN LPCSTR DirectoryPath,
44 IN LPCSTR AnsiFileName,
45 IN LPCSTR OemFileName,
46 IN LPCSTR LanguageFileName);
47 BOOLEAN
48 WinLdrAddDriverToList(LIST_ENTRY *BootDriverListHead,
49 LPWSTR RegistryPath,
50 LPWSTR ImagePath,
51 LPWSTR ServiceName);
52
53
54 //FIXME: Do a better way to retrieve Arc disk information
55 extern ULONG reactos_disk_count;
56 extern ARC_DISK_SIGNATURE reactos_arc_disk_info[];
57 extern char reactos_arc_strings[32][256];
58
59 extern BOOLEAN UseRealHeap;
60 extern ULONG LoaderPagesSpanned;
61
62
63 VOID
64 SetupLdrLoadNlsData(PLOADER_PARAMETER_BLOCK LoaderBlock, HINF InfHandle, LPCSTR SearchPath)
65 {
66 INFCONTEXT InfContext;
67 BOOLEAN Status;
68 LPCSTR AnsiName, OemName, LangName;
69
70 /* Get ANSI codepage file */
71 if (!InfFindFirstLine(InfHandle, "NLS", "AnsiCodepage", &InfContext))
72 {
73 printf("Failed to find 'NLS/AnsiCodepage'\n");
74 return;
75 }
76 if (!InfGetDataField(&InfContext, 1, &AnsiName))
77 {
78 printf("Failed to get load options\n");
79 return;
80 }
81
82 /* Get OEM codepage file */
83 if (!InfFindFirstLine(InfHandle, "NLS", "OemCodepage", &InfContext))
84 {
85 printf("Failed to find 'NLS/AnsiCodepage'\n");
86 return;
87 }
88 if (!InfGetDataField(&InfContext, 1, &OemName))
89 {
90 printf("Failed to get load options\n");
91 return;
92 }
93
94 if (!InfFindFirstLine(InfHandle, "NLS", "UnicodeCasetable", &InfContext))
95 {
96 printf("Failed to find 'NLS/AnsiCodepage'\n");
97 return;
98 }
99 if (!InfGetDataField(&InfContext, 1, &LangName))
100 {
101 printf("Failed to get load options\n");
102 return;
103 }
104
105 Status = WinLdrLoadNLSData(LoaderBlock, SearchPath, AnsiName, OemName, LangName);
106 DPRINTM(DPRINT_WINDOWS, "NLS data loaded with status %d\n", Status);
107 }
108
109 VOID
110 SetupLdrScanBootDrivers(PLOADER_PARAMETER_BLOCK LoaderBlock, HINF InfHandle, LPCSTR SearchPath)
111 {
112 INFCONTEXT InfContext;
113 BOOLEAN Status;
114 LPCSTR Media, DriverName;
115 WCHAR ServiceName[256];
116 WCHAR ImagePath[256];
117
118 /* Open inf section */
119 if (!InfFindFirstLine(InfHandle, "SourceDisksFiles", NULL, &InfContext))
120 return;
121
122 /* Load all listed boot drivers */
123 do
124 {
125 if (InfGetDataField(&InfContext, 7, &Media) &&
126 InfGetDataField(&InfContext, 0, &DriverName))
127 {
128 if (strcmp(Media, "x") == 0)
129 {
130 /* Convert name to widechar */
131 swprintf(ServiceName, L"%S", DriverName);
132
133 /* Remove .sys extension */
134 ServiceName[wcslen(ServiceName) - 4] = 0;
135
136 /* Prepare image path */
137 swprintf(ImagePath, L"%S", DriverName);
138
139 /* Add it to the list */
140 Status = WinLdrAddDriverToList(&LoaderBlock->BootDriverListHead,
141 L"\\Registry\\Machine\\System\\CurrentControlSet\\Services\\",
142 ImagePath,
143 ServiceName);
144
145 if (!Status)
146 {
147 DPRINTM(DPRINT_WINDOWS, "could not add boot driver %s, %s\n", SearchPath, DriverName);
148 return;
149 }
150 }
151 }
152 } while (InfFindNextLine(&InfContext, &InfContext));
153 }
154
155 VOID LoadReactOSSetup2(VOID)
156 {
157 CHAR SystemPath[512], SearchPath[512];
158 CHAR FileName[512];
159 CHAR BootPath[512];
160 LPCSTR LoadOptions, BootOptions;
161 BOOLEAN BootFromFloppy;
162 #if DBG
163 LPCSTR DbgOptions;
164 #endif
165 PVOID NtosBase = NULL, HalBase = NULL, KdComBase = NULL;
166 BOOLEAN Status;
167 ULONG i, ErrorLine;
168 HINF InfHandle;
169 INFCONTEXT InfContext;
170 PLOADER_PARAMETER_BLOCK LoaderBlock, LoaderBlockVA;
171 PSETUP_LOADER_BLOCK SetupBlock;
172 KERNEL_ENTRY_POINT KiSystemStartup;
173 PLDR_DATA_TABLE_ENTRY KernelDTE, HalDTE, KdComDTE = NULL;
174 // Mm-related things
175 PVOID GdtIdt;
176 ULONG PcrBasePage=0;
177 ULONG TssBasePage=0;
178 LPCSTR SourcePath;
179 LPCSTR SourcePaths[] =
180 {
181 "", /* Only for floppy boot */
182 #if defined(_M_IX86)
183 "\\I386",
184 #elif defined(_M_MPPC)
185 "\\PPC",
186 #elif defined(_M_MRX000)
187 "\\MIPS",
188 #endif
189 "\\reactos",
190 NULL
191 };
192
193 /* Get boot path */
194 MachDiskGetBootPath(SystemPath, sizeof(SystemPath));
195
196 /* And check if we booted from floppy */
197 BootFromFloppy = strstr(SystemPath, "fdisk") != NULL;
198
199 /* Open 'txtsetup.sif' from any of source paths */
200 for (i = BootFromFloppy ? 0 : 1; ; i++)
201 {
202 SourcePath = SourcePaths[i];
203 if (!SourcePath)
204 {
205 printf("Failed to open 'txtsetup.sif'\n");
206 return;
207 }
208 sprintf(FileName, "%s\\txtsetup.sif", SourcePath);
209 if (InfOpenFile (&InfHandle, FileName, &ErrorLine))
210 {
211 sprintf(BootPath, "%s%s\\", SystemPath, SourcePath);
212 break;
213 }
214 }
215
216 /* Get Load options - debug and non-debug */
217 if (!InfFindFirstLine(InfHandle,
218 "SetupData",
219 "OsLoadOptions",
220 &InfContext))
221 {
222 printf("Failed to find 'SetupData/OsLoadOptions'\n");
223 return;
224 }
225
226 if (!InfGetDataField (&InfContext, 1, &LoadOptions))
227 {
228 printf("Failed to get load options\n");
229 return;
230 }
231
232 BootOptions = LoadOptions;
233
234 #if DBG
235 /* Get debug load options and use them */
236 if (InfFindFirstLine(InfHandle,
237 "SetupData",
238 "DbgOsLoadOptions",
239 &InfContext))
240 {
241 if (!InfGetDataField(&InfContext, 1, &DbgOptions))
242 DbgOptions = "";
243 else
244 BootOptions = DbgOptions;
245 }
246 #endif
247
248 DPRINTM(DPRINT_WINDOWS,"BootOptions: '%s'\n", BootOptions);
249
250 SetupUiInitialize();
251 UiDrawStatusText("");
252 UiDrawStatusText("Detecting Hardware...");
253
254 /* Let user know we started loading */
255 UiDrawStatusText("Loading...");
256
257 /* Construct the system path */
258 sprintf(SystemPath, "%s\\", SourcePath);
259
260 DPRINTM(DPRINT_WINDOWS,"BootPath: '%s', SystemPath: '%s'\n", BootPath, SystemPath);
261
262 /* Allocate and minimalistic-initialize LPB */
263 AllocateAndInitLPB(&LoaderBlock);
264
265 /* Allocate and initialize setup loader block */
266 SetupBlock = MmHeapAlloc(sizeof(SETUP_LOADER_BLOCK));
267 RtlZeroMemory(SetupBlock, sizeof(SETUP_LOADER_BLOCK));
268 LoaderBlock->SetupLdrBlock = SetupBlock;
269
270 /* Set textmode setup flag */
271 SetupBlock->Flags = SETUPLDR_TEXT_MODE;
272
273 /* Detect hardware */
274 UseRealHeap = TRUE;
275 LoaderBlock->ConfigurationRoot = MachHwDetect();
276
277 strcpy(FileName, "\\ArcName\\");
278
279 /* Load kernel */
280 strcpy(FileName+strlen("\\ArcName\\"), BootPath);
281 strcat(FileName, "NTOSKRNL.EXE");
282 Status = WinLdrLoadImage(FileName+strlen("\\ArcName\\"), LoaderSystemCode, &NtosBase);
283 DPRINTM(DPRINT_WINDOWS, "Ntos loaded with status %d at %p\n", Status, NtosBase);
284 WinLdrAllocateDataTableEntry(LoaderBlock, "ntoskrnl.exe",
285 FileName, NtosBase, &KernelDTE);
286
287 /* Load HAL */
288 strcpy(FileName+strlen("\\ArcName\\"), BootPath);
289 strcat(FileName, "HAL.DLL");
290 Status = WinLdrLoadImage(FileName+strlen("\\ArcName\\"), LoaderHalCode, &HalBase);
291 DPRINTM(DPRINT_WINDOWS, "HAL loaded with status %d at %p\n", Status, HalBase);
292 WinLdrAllocateDataTableEntry(LoaderBlock, "hal.dll",
293 FileName, HalBase, &HalDTE);
294
295 /* Load kernel-debugger support dll */
296 strcpy(FileName+strlen("\\ArcName\\"), BootPath);
297 strcat(FileName, "KDCOM.DLL");
298 Status = WinLdrLoadImage(FileName+strlen("\\ArcName\\"), LoaderBootDriver, &KdComBase);
299 DPRINTM(DPRINT_WINDOWS, "KdCom loaded with status %d at %p\n", Status, KdComBase);
300 WinLdrAllocateDataTableEntry(LoaderBlock, "kdcom.dll",
301 FileName, KdComBase, &KdComDTE);
302
303 /* Load all referenced DLLs for kernel, HAL and kdcom.dll */
304 strcpy(SearchPath, BootPath);
305 WinLdrScanImportDescriptorTable(LoaderBlock, SearchPath, KernelDTE);
306 WinLdrScanImportDescriptorTable(LoaderBlock, SearchPath, HalDTE);
307 if (KdComDTE)
308 WinLdrScanImportDescriptorTable(LoaderBlock, SearchPath, KdComDTE);
309
310 /* Load NLS data */
311 SetupLdrLoadNlsData(LoaderBlock, InfHandle, BootPath);
312
313 /* Get a list of boot drivers */
314 SetupLdrScanBootDrivers(LoaderBlock, InfHandle, BootPath);
315
316 /* Load boot drivers */
317 Status = WinLdrLoadBootDrivers(LoaderBlock, BootPath);
318 DPRINTM(DPRINT_WINDOWS, "Boot drivers loaded with status %d\n", Status);
319
320 /* Alloc PCR, TSS, do magic things with the GDT/IDT */
321 WinLdrSetupForNt(LoaderBlock, &GdtIdt, &PcrBasePage, &TssBasePage);
322
323 /* Initialize Phase 1 - no drivers loading anymore */
324 WinLdrInitializePhase1(LoaderBlock, (PCHAR)BootOptions, SystemPath, BootPath, _WIN32_WINNT_WS03);
325
326 /* Save entry-point pointer and Loader block VAs */
327 KiSystemStartup = (KERNEL_ENTRY_POINT)KernelDTE->EntryPoint;
328 LoaderBlockVA = PaToVa(LoaderBlock);
329
330 /* "Stop all motors", change videomode */
331 MachPrepareForReactOS(TRUE);
332
333 /* Debugging... */
334 //DumpMemoryAllocMap();
335
336 /* Turn on paging mode of CPU*/
337 WinLdrTurnOnPaging(LoaderBlock, PcrBasePage, TssBasePage, GdtIdt);
338
339 /* Save final value of LoaderPagesSpanned */
340 LoaderBlock->Extension->LoaderPagesSpanned = LoaderPagesSpanned;
341
342 DPRINTM(DPRINT_WINDOWS, "Hello from paged mode, KiSystemStartup %p, LoaderBlockVA %p!\n",
343 KiSystemStartup, LoaderBlockVA);
344
345 //WinLdrpDumpMemoryDescriptors(LoaderBlockVA);
346 //WinLdrpDumpBootDriver(LoaderBlockVA);
347 //WinLdrpDumpArcDisks(LoaderBlockVA);
348
349 /*asm(".intel_syntax noprefix\n");
350 asm("test1:\n");
351 asm("jmp test1\n");
352 asm(".att_syntax\n");*/
353
354 /* Pass control */
355 (*KiSystemStartup)(LoaderBlockVA);
356
357 return;
358 }