1 | /* $Id: biossums.c 69500 2017-10-28 15:14:05Z vboxsync $ */
|
---|
2 | /** @file
|
---|
3 | * Tool for modifying a BIOS image to write the BIOS checksum.
|
---|
4 | */
|
---|
5 |
|
---|
6 | /*
|
---|
7 | * Copyright (C) 2006-2017 Oracle Corporation
|
---|
8 | *
|
---|
9 | * This file is part of VirtualBox Open Source Edition (OSE), as
|
---|
10 | * available from http://www.virtualbox.org. This file is free software;
|
---|
11 | * you can redistribute it and/or modify it under the terms of the GNU
|
---|
12 | * General Public License (GPL) as published by the Free Software
|
---|
13 | * Foundation, in version 2 as it comes in the "COPYING" file of the
|
---|
14 | * VirtualBox OSE distribution. VirtualBox OSE is distributed in the
|
---|
15 | * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
|
---|
16 | */
|
---|
17 |
|
---|
18 | #include <stdlib.h>
|
---|
19 | #include <stdio.h>
|
---|
20 | #include <string.h>
|
---|
21 | #include <stdarg.h>
|
---|
22 | #include <errno.h>
|
---|
23 | #ifndef RT_OS_WINDOWS
|
---|
24 | # include <unistd.h> /* unlink */
|
---|
25 | #endif
|
---|
26 |
|
---|
27 | typedef unsigned char uint8_t;
|
---|
28 |
|
---|
29 | static uint8_t abBios[64*1024];
|
---|
30 | static FILE *g_pIn = NULL;
|
---|
31 | static FILE *g_pOut = NULL;
|
---|
32 | static const char *g_pszOutFile = NULL;
|
---|
33 | static const char *g_argv0;
|
---|
34 |
|
---|
35 | /**
|
---|
36 | * Find where the filename starts in the given path.
|
---|
37 | */
|
---|
38 | static const char *name(const char *pszPath)
|
---|
39 | {
|
---|
40 | const char *psz = strrchr(pszPath, '/');
|
---|
41 | #if defined(_MSC_VER) || defined(__OS2__)
|
---|
42 | const char *psz2 = strrchr(pszPath, '\\');
|
---|
43 | if (!psz2)
|
---|
44 | psz2 = strrchr(pszPath, ':');
|
---|
45 | if (psz2 && (!psz || psz2 > psz))
|
---|
46 | psz = psz2;
|
---|
47 | #endif
|
---|
48 | return psz ? psz + 1 : pszPath;
|
---|
49 | }
|
---|
50 |
|
---|
51 | /**
|
---|
52 | * Report an error.
|
---|
53 | */
|
---|
54 | static int fatal(const char *pszFormat, ...)
|
---|
55 | {
|
---|
56 | va_list va;
|
---|
57 |
|
---|
58 | fprintf(stderr, "%s: ", name(g_argv0));
|
---|
59 |
|
---|
60 | va_start(va, pszFormat);
|
---|
61 | vfprintf(stderr, pszFormat, va);
|
---|
62 | va_end(va);
|
---|
63 |
|
---|
64 | /* clean up */
|
---|
65 | if (g_pIn)
|
---|
66 | fclose(g_pIn);
|
---|
67 | if (g_pOut)
|
---|
68 | fclose(g_pOut);
|
---|
69 | if (g_pszOutFile)
|
---|
70 | unlink(g_pszOutFile);
|
---|
71 |
|
---|
72 | return 1;
|
---|
73 | }
|
---|
74 |
|
---|
75 | /**
|
---|
76 | * Calculate the checksum.
|
---|
77 | */
|
---|
78 | static uint8_t calculateChecksum(uint8_t *pb, size_t cb, size_t iChecksum)
|
---|
79 | {
|
---|
80 | uint8_t u8Sum = 0;
|
---|
81 | size_t i;
|
---|
82 |
|
---|
83 | for (i = 0; i < cb; i++)
|
---|
84 | if (i != iChecksum)
|
---|
85 | u8Sum += pb[i];
|
---|
86 |
|
---|
87 | return -u8Sum;
|
---|
88 | }
|
---|
89 |
|
---|
90 | /**
|
---|
91 | * Find a header in the binary.
|
---|
92 | *
|
---|
93 | * @param pb Where to search for the signature
|
---|
94 | * @param cb Size of the search area
|
---|
95 | * @param pbHeader Pointer to the start of the signature
|
---|
96 | * @returns 0 if signature was not found, 1 if found or
|
---|
97 | * 2 if more than one signature was found */
|
---|
98 | static int searchHeader(uint8_t *pb, size_t cb, const char *pszHeader, uint8_t **pbHeader)
|
---|
99 | {
|
---|
100 | int fFound = 0;
|
---|
101 | unsigned int i;
|
---|
102 | size_t cbSignature = strlen(pszHeader);
|
---|
103 |
|
---|
104 | for (i = 0; i < cb; i += 16)
|
---|
105 | if (!memcmp(pb + i, pszHeader, cbSignature))
|
---|
106 | {
|
---|
107 | if (fFound++)
|
---|
108 | return 2;
|
---|
109 | *pbHeader = pb + i;
|
---|
110 | }
|
---|
111 |
|
---|
112 | return fFound;
|
---|
113 | }
|
---|
114 |
|
---|
115 | int main(int argc, char **argv)
|
---|
116 | {
|
---|
117 | FILE *pIn, *pOut;
|
---|
118 | size_t cbIn, cbOut;
|
---|
119 | int fAdapterBios = 0;
|
---|
120 |
|
---|
121 | g_argv0 = argv[0];
|
---|
122 |
|
---|
123 | if (argc != 3)
|
---|
124 | return fatal("Input file name and output file name required.\n");
|
---|
125 |
|
---|
126 | pIn = g_pIn = fopen(argv[1], "rb");
|
---|
127 | if (!pIn)
|
---|
128 | return fatal("Error opening '%s' for reading (%s).\n", argv[1], strerror(errno));
|
---|
129 |
|
---|
130 | pOut = g_pOut = fopen(argv[2], "wb");
|
---|
131 | if (!pOut)
|
---|
132 | return fatal("Error opening '%s' for writing (%s).\n", argv[2], strerror(errno));
|
---|
133 | g_pszOutFile = argv[2];
|
---|
134 |
|
---|
135 | /* safety precaution (aka. complete paranoia :-) */
|
---|
136 | memset(abBios, 0, sizeof(abBios));
|
---|
137 |
|
---|
138 | cbIn = fread(abBios, 1, sizeof(abBios), pIn);
|
---|
139 | if (ferror(pIn))
|
---|
140 | return fatal("Error reading from '%s' (%s).\n", argv[1], strerror(errno));
|
---|
141 | g_pIn = NULL;
|
---|
142 | fclose(pIn);
|
---|
143 |
|
---|
144 | fAdapterBios = abBios[0] == 0x55 && abBios[1] == 0xaa;
|
---|
145 |
|
---|
146 | /* align size to page size */
|
---|
147 | if ((cbIn % 4096) != 0)
|
---|
148 | cbIn = (cbIn + 4095) & ~4095;
|
---|
149 |
|
---|
150 | if (!fAdapterBios && cbIn != 64*1024)
|
---|
151 | return fatal("Size of system BIOS is not 64KB!\n");
|
---|
152 |
|
---|
153 | if (fAdapterBios)
|
---|
154 | {
|
---|
155 | /* adapter BIOS */
|
---|
156 |
|
---|
157 | /* set the length indicator */
|
---|
158 | abBios[2] = (uint8_t)(cbIn / 512);
|
---|
159 | }
|
---|
160 | else
|
---|
161 | {
|
---|
162 | /* system BIOS */
|
---|
163 | size_t cbChecksum;
|
---|
164 | uint8_t u8Checksum;
|
---|
165 | uint8_t *pbHeader;
|
---|
166 |
|
---|
167 | /* Set the BIOS32 header checksum. */
|
---|
168 | switch (searchHeader(abBios, cbIn, "_32_", &pbHeader))
|
---|
169 | {
|
---|
170 | case 0:
|
---|
171 | return fatal("No BIOS32 header not found!\n");
|
---|
172 | case 2:
|
---|
173 | return fatal("More than one BIOS32 header found!\n");
|
---|
174 | case 1:
|
---|
175 | cbChecksum = (size_t)pbHeader[9] * 16;
|
---|
176 | u8Checksum = calculateChecksum(pbHeader, cbChecksum, 10);
|
---|
177 | pbHeader[10] = u8Checksum;
|
---|
178 | break;
|
---|
179 | }
|
---|
180 |
|
---|
181 | /* Set the PIR header checksum according to PCI IRQ Routing table
|
---|
182 | * specification version 1.0, Microsoft Corporation, 1996 */
|
---|
183 | switch (searchHeader(abBios, cbIn, "$PIR", &pbHeader))
|
---|
184 | {
|
---|
185 | case 0:
|
---|
186 | return fatal("No PCI IRQ routing table found!\n");
|
---|
187 | case 2:
|
---|
188 | return fatal("More than one PCI IRQ routing table found!\n");
|
---|
189 | case 1:
|
---|
190 | cbChecksum = (size_t)pbHeader[6] + (size_t)pbHeader[7] * 256;
|
---|
191 | u8Checksum = calculateChecksum(pbHeader, cbChecksum, 31);
|
---|
192 | pbHeader[31] = u8Checksum;
|
---|
193 | break;
|
---|
194 | }
|
---|
195 |
|
---|
196 | /* Set the SMBIOS header checksum according to System Management BIOS
|
---|
197 | * Reference Specification Version 2.5, DSP0134. */
|
---|
198 | switch (searchHeader(abBios, cbIn, "_SM_", &pbHeader))
|
---|
199 | {
|
---|
200 | case 0:
|
---|
201 | return fatal("No SMBIOS header found!\n");
|
---|
202 | case 2:
|
---|
203 | return fatal("More than one SMBIOS header found!\n");
|
---|
204 | case 1:
|
---|
205 | /* at first fix the DMI header starting at SMBIOS header offset 16 */
|
---|
206 | u8Checksum = calculateChecksum(pbHeader+16, 15, 5);
|
---|
207 | pbHeader[21] = u8Checksum;
|
---|
208 |
|
---|
209 | /* now fix the checksum of the whole SMBIOS header */
|
---|
210 | cbChecksum = (size_t)pbHeader[5];
|
---|
211 | u8Checksum = calculateChecksum(pbHeader, cbChecksum, 4);
|
---|
212 | pbHeader[4] = u8Checksum;
|
---|
213 | break;
|
---|
214 | }
|
---|
215 | }
|
---|
216 |
|
---|
217 | /* set the BIOS checksum */
|
---|
218 | abBios[cbIn-1] = calculateChecksum(abBios, cbIn, cbIn - 1);
|
---|
219 |
|
---|
220 | cbOut = fwrite(abBios, 1, cbIn, pOut);
|
---|
221 | if (ferror(pOut))
|
---|
222 | return fatal("Error writing to '%s' (%s).\n", g_pszOutFile, strerror(errno));
|
---|
223 | g_pOut = NULL;
|
---|
224 | if (fclose(pOut))
|
---|
225 | return fatal("Error closing '%s' (%s).\n", g_pszOutFile, strerror(errno));
|
---|
226 |
|
---|
227 | return 0;
|
---|
228 | }
|
---|
229 |
|
---|