[project] Rename winvblock__uint32 back to UINT32
[people/sha0/winvblock.git] / src / winvblock / probe.c
1 /**
2  * Copyright (C) 2009-2011, Shao Miller <shao.miller@yrdsb.edu.on.ca>.
3  * Copyright 2006-2008, V.
4  * For WinAoE contact information, see http://winaoe.org/
5  *
6  * This file is part of WinVBlock, derived from WinAoE.
7  *
8  * WinVBlock is free software: you can redistribute it and/or modify
9  * it under the terms of the GNU General Public License as published by
10  * the Free Software Foundation, either version 3 of the License, or
11  * (at your option) any later version.
12  *
13  * WinVBlock is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16  * GNU General Public License for more details.
17  *
18  * You should have received a copy of the GNU General Public License
19  * along with WinVBlock.  If not, see <http://www.gnu.org/licenses/>.
20  */
21
22 /**
23  * @file
24  *
25  * Boot-time disk probing specifics.
26  */
27
28 #include <ntddk.h>
29
30 #include "winvblock.h"
31 #include "wv_string.h"
32 #include "portable.h"
33 #include "driver.h"
34 #include "bus.h"
35 #include "device.h"
36 #include "disk.h"
37 #include "mount.h"
38 #include "debug.h"
39 #include "bus.h"
40 #include "ramdisk.h"
41 #include "memdisk.h"
42 #include "grub4dos.h"
43 #include "filedisk.h"
44 #include "probe.h"
45
46 WV_SP_PROBE_SAFE_MBR_HOOK STDCALL WvProbeGetSafeHook(
47     IN PUCHAR PhysicalMemory,
48     IN WV_SP_PROBE_INT_VECTOR InterruptVector
49   ) {
50     UINT32 int13_hook;
51     WV_SP_PROBE_SAFE_MBR_HOOK safe_mbr_hook;
52     UCHAR sig[9] = {0};
53     UCHAR ven_id[9] = {0};
54
55     int13_hook = (((UINT32) InterruptVector->Segment) << 4) +
56       ((UINT32) InterruptVector->Offset);
57     safe_mbr_hook = (WV_SP_PROBE_SAFE_MBR_HOOK) (PhysicalMemory + int13_hook);
58     RtlCopyMemory(sig, safe_mbr_hook->Signature, 8);
59     RtlCopyMemory(ven_id, safe_mbr_hook->VendorId, 8);
60     DBG("INT 0x13 Segment: 0x%04x\n", InterruptVector->Segment);
61     DBG("INT 0x13 Offset: 0x%04x\n", InterruptVector->Offset);
62     DBG("INT 0x13 Hook: 0x%08x\n", int13_hook);
63     DBG("INT 0x13 Safe Hook Signature: %s\n", sig);
64     if (!wv_memcmpeq(sig, "$INT13SF", sizeof "$INT13SF" - 1)) {
65         DBG("Invalid INT 0x13 Safe Hook Signature; End of chain\n");
66         return NULL;
67       }
68     return safe_mbr_hook;
69   }
70
71 void WvProbeDisks(void) {
72     static winvblock__bool probed = FALSE;
73
74     if (probed)
75       return;
76     WvMemdiskFind();
77     ramdisk_grub4dos__find();
78     filedisk_grub4dos__find();
79     probed = TRUE;
80   }