[FREELDR] Minor fixes.
[reactos.git] / boot / freeldr / freeldr / ntldr / headless.c
1 /*
2 * PROJECT: ReactOS Boot Loader
3 * LICENSE: BSD - See COPYING.ARM in the top level directory
4 * FILE: boot/freeldr/freeldr/windows/headless.c
5 * PURPOSE: Provides support for Windows Emergency Management Services
6 * PROGRAMMERS: ReactOS Portable Systems Group
7 */
8
9 /* INCLUDES *******************************************************************/
10
11 #include <freeldr.h>
12 #include <cportlib/cportlib.h>
13
14 /* Note: Move these to some smbios.h header */
15 #define SYSID_TYPE_UUID "_UUID_"
16 #define SYSID_UUID_DATA_SIZE 16
17 #include <pshpack1.h>
18 typedef struct _SYSID_UUID_ENTRY
19 {
20 UCHAR Type[6];
21 UCHAR Checksum;
22 USHORT Length;
23 UCHAR UUID[SYSID_UUID_DATA_SIZE];
24 } SYSID_UUID_ENTRY, *PSYSID_UUID_ENTRY;
25 #include <poppack.h>
26
27 /* GLOBALS ********************************************************************/
28
29 HEADLESS_LOADER_BLOCK LoaderRedirectionInformation;
30 BOOLEAN WinLdrTerminalConnected;
31 ULONG WinLdrTerminalDeviceId;
32 ULONG WinLdrTerminalDelay;
33
34 CPPORT Port[4] =
35 {
36 {NULL, 0, TRUE},
37 {NULL, 0, TRUE},
38 {NULL, 0, TRUE},
39 {NULL, 0, TRUE}
40 };
41
42 /* FUNCTIONS ******************************************************************/
43
44 VOID
45 WinLdrLoadGUID(OUT PGUID SystemGuid)
46 {
47 PSYSID_UUID_ENTRY CurrentAddress;
48
49 CurrentAddress = (PSYSID_UUID_ENTRY)0xE0000;
50 while (CurrentAddress < (PSYSID_UUID_ENTRY)0x100000)
51 {
52 if (RtlCompareMemory(&CurrentAddress->Type, SYSID_TYPE_UUID, 6) == 6)
53 {
54 RtlCopyMemory(SystemGuid, &CurrentAddress->UUID, SYSID_UUID_DATA_SIZE);
55 return;
56 }
57 CurrentAddress = (PSYSID_UUID_ENTRY)((ULONG_PTR)CurrentAddress + 1);
58 }
59
60 RtlZeroMemory(SystemGuid, SYSID_UUID_DATA_SIZE);
61 }
62
63 BOOLEAN
64 WinLdrPortInitialize(IN ULONG BaudRate,
65 IN ULONG PortNumber,
66 IN PUCHAR PortAddress,
67 IN BOOLEAN TerminalConnected,
68 OUT PULONG PortId)
69 {
70 /* Set default baud rate */
71 if (BaudRate == 0) BaudRate = 19200;
72
73 /* Check if port or address given */
74 if (PortNumber)
75 {
76 /* Pick correct address for port */
77 if (!PortAddress)
78 {
79 switch (PortNumber)
80 {
81 case 1:
82 PortAddress = (PUCHAR)0x3F8;
83 break;
84
85 case 2:
86 PortAddress = (PUCHAR)0x2F8;
87 break;
88
89 case 3:
90 PortAddress = (PUCHAR)0x3E8;
91 break;
92
93 default:
94 PortNumber = 4;
95 PortAddress = (PUCHAR)0x2E8;
96 }
97 }
98 }
99 else
100 {
101 /* Pick correct port for address */
102 PortAddress = (PUCHAR)0x2F8;
103 if (CpDoesPortExist(PortAddress))
104 {
105 PortNumber = 2;
106 }
107 else
108 {
109 PortAddress = (PUCHAR)0x3F8;
110 if (!CpDoesPortExist(PortAddress)) return FALSE;
111 PortNumber = 1;
112 }
113 }
114
115 /* Not yet supported */
116 ASSERT(LoaderRedirectionInformation.IsMMIODevice == FALSE);
117
118 /* Check if port exists */
119 if ((CpDoesPortExist(PortAddress)) || (CpDoesPortExist(PortAddress)))
120 {
121 /* Initialize port for first time, or re-initialize if specified */
122 if (((TerminalConnected) && (Port[PortNumber - 1].Address)) ||
123 !(Port[PortNumber - 1].Address))
124 {
125 /* Initialize the port, return it */
126 CpInitialize(&Port[PortNumber - 1], PortAddress, BaudRate);
127 *PortId = PortNumber - 1;
128 return TRUE;
129 }
130 }
131
132 return FALSE;
133 }
134
135 VOID
136 WinLdrPortPutByte(IN ULONG PortId,
137 IN UCHAR Byte)
138 {
139 CpPutByte(&Port[PortId], Byte);
140 }
141
142 BOOLEAN
143 WinLdrPortGetByte(IN ULONG PortId,
144 OUT PUCHAR Byte)
145 {
146 return CpGetByte(&Port[PortId], Byte, TRUE, FALSE) == CP_GET_SUCCESS;
147 }
148
149 BOOLEAN
150 WinLdrPortPollOnly(IN ULONG PortId)
151 {
152 UCHAR Dummy;
153
154 return CpGetByte(&Port[PortId], &Dummy, FALSE, TRUE) == CP_GET_SUCCESS;
155 }
156
157 VOID
158 WinLdrEnableFifo(IN ULONG PortId,
159 IN BOOLEAN Enable)
160 {
161 CpEnableFifo(Port[PortId].Address, Enable);
162 }
163
164 VOID
165 WinLdrInitializeHeadlessPort(VOID)
166 {
167 ULONG PortNumber, BaudRate;
168 PUCHAR PortAddress;
169 PCHAR AnsiReset = "\x1B[m";
170 ULONG i;
171
172 PortNumber = LoaderRedirectionInformation.PortNumber;
173 PortAddress = LoaderRedirectionInformation.PortAddress;
174 BaudRate = LoaderRedirectionInformation.BaudRate;
175
176 /* Pick a port address */
177 if (PortNumber)
178 {
179 if (!PortAddress)
180 {
181 switch (PortNumber)
182 {
183 case 2:
184 LoaderRedirectionInformation.PortAddress = (PUCHAR)0x2F8;
185 break;
186
187 case 3:
188 LoaderRedirectionInformation.PortAddress = (PUCHAR)0x3E8;
189 break;
190
191 case 4:
192 LoaderRedirectionInformation.PortAddress = (PUCHAR)0x2E8;
193 break;
194
195 default:
196 LoaderRedirectionInformation.PortAddress = (PUCHAR)0x3F8;
197 break;
198 }
199 }
200 }
201 else
202 {
203 /* No number, so no EMS */
204 WinLdrTerminalConnected = FALSE;
205 return;
206 }
207
208 /* Call arch code to initialize the port */
209 PortAddress = LoaderRedirectionInformation.PortAddress;
210 WinLdrTerminalConnected = WinLdrPortInitialize(
211 BaudRate,
212 PortNumber,
213 PortAddress,
214 WinLdrTerminalConnected,
215 &WinLdrTerminalDeviceId);
216
217 if (WinLdrTerminalConnected)
218 {
219 /* Port seems usable, set it up and get the BIOS GUID */
220 WinLdrEnableFifo(WinLdrTerminalDeviceId, TRUE);
221
222 WinLdrLoadGUID(&LoaderRedirectionInformation.SystemGUID);
223
224 /* Calculate delay in us based on the baud, assume 9600 if none given */
225 if (!BaudRate)
226 {
227 BaudRate = 9600;
228 LoaderRedirectionInformation.BaudRate = BaudRate;
229 }
230
231 WinLdrTerminalDelay = (10 * 1000 * 1000) / (BaudRate / 10) / 6;
232
233 /* Sent an ANSI reset sequence to get the terminal up and running */
234 for (i = 0; i < strlen(AnsiReset); i++)
235 {
236 WinLdrPortPutByte(WinLdrTerminalDeviceId, AnsiReset[i]);
237 StallExecutionProcessor(WinLdrTerminalDelay);
238 }
239 }
240 }
241
242 VOID
243 WinLdrSetupEms(IN PCHAR BootOptions)
244 {
245 PCHAR Settings, RedirectPort;
246
247 /* Start fresh */
248 RtlZeroMemory(&LoaderRedirectionInformation, sizeof(HEADLESS_LOADER_BLOCK));
249 LoaderRedirectionInformation.PciDeviceId = PCI_INVALID_VENDORID;
250
251 /* Use a direction port if one was given, or use ACPI to detect one instead */
252 Settings = strstr(BootOptions, "/redirect=");
253 if (Settings)
254 {
255 RedirectPort = strstr(Settings, "com");
256 if (RedirectPort)
257 {
258 RedirectPort += sizeof("com") - 1;
259 LoaderRedirectionInformation.PortNumber = atoi(RedirectPort);
260 LoaderRedirectionInformation.TerminalType = 1; // HeadlessSerialPort
261 }
262 else
263 {
264 RedirectPort = strstr(Settings, "usebiossettings");
265 if (RedirectPort)
266 {
267 UiDrawStatusText("ACPI SRT Table Not Supported...");
268 return;
269 }
270 else
271 {
272 LoaderRedirectionInformation.PortAddress = (PUCHAR)strtoul(Settings, 0, 16);
273 if (LoaderRedirectionInformation.PortAddress)
274 {
275 LoaderRedirectionInformation.PortNumber = 3;
276 }
277 }
278 }
279 }
280
281 /* Use a direction baudrate if one was given */
282 Settings = strstr(BootOptions, "/redirectbaudrate=");
283 if (Settings)
284 {
285 if (strstr(Settings, "115200"))
286 {
287 LoaderRedirectionInformation.BaudRate = 115200;
288 }
289 else if (strstr(Settings, "57600"))
290 {
291 LoaderRedirectionInformation.BaudRate = 57600;
292 }
293 else if (strstr(Settings, "19200"))
294 {
295 LoaderRedirectionInformation.BaudRate = 19200;
296 }
297 else
298 {
299 LoaderRedirectionInformation.BaudRate = 9600;
300 }
301 }
302
303 /* Enable headless support if parameters were found */
304 if (LoaderRedirectionInformation.PortNumber)
305 {
306 if (!LoaderRedirectionInformation.BaudRate)
307 {
308 LoaderRedirectionInformation.BaudRate = 9600;
309 }
310
311 WinLdrInitializeHeadlessPort();
312 }
313 }
314
315 /* EOF */