Skip to content

Commit faa5fcb

Browse files
author
Requiem
committed
whitelisted x86_32 from boot logo detections
1 parent 0f95280 commit faa5fcb

File tree

3 files changed

+133
-112
lines changed

3 files changed

+133
-112
lines changed

auxiliary/boot_logo_hasher.cpp

Lines changed: 29 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -28,33 +28,50 @@
2828

2929
static inline uint32_t crc32c_bmp(const char* filename) {
3030
std::ifstream in{ filename, std::ios::binary | std::ios::ate };
31+
if (!in) return 0;
3132
auto size = in.tellg();
32-
in.seekg(0);
33-
std::vector<uint8_t> buf(size);
33+
in.seekg(0, std::ios::beg);
34+
35+
std::vector<uint8_t> buf(static_cast<size_t>(size));
3436
in.read(reinterpret_cast<char*>(buf.data()), size);
3537

36-
// BMP offset at bytes 10–13
38+
// offset in BMP header (bytes 10–13)
3739
uint32_t offset = *reinterpret_cast<uint32_t*>(buf.data() + 10);
3840
uint8_t* bmp = buf.data() + offset;
39-
size_t len = buf.size() - offset;
41+
size_t len = buf.size() - offset;
4042

41-
uint64_t crcReg = 0xFFFFFFFFull;
42-
size_t qwords = len / 8;
43-
auto ptr = reinterpret_cast<uint64_t*>(bmp);
44-
for (size_t i = 0; i < qwords; ++i)
45-
crcReg = _mm_crc32_u64(crcReg, ptr[i]);
43+
#if defined(_M_X64) || defined(__x86_64__)
44+
uint64_t crc64 = 0xFFFFFFFFull;
45+
auto q64 = len / 8;
46+
auto const* p64 = reinterpret_cast<uint64_t const*>(bmp);
47+
for (size_t i = 0; i < q64; ++i)
48+
crc64 = _mm_crc32_u64(crc64, p64[i]);
4649

47-
uint32_t crc = static_cast<uint32_t>(crcReg);
48-
auto tail = reinterpret_cast<uint8_t*>(ptr + qwords);
50+
uint32_t crc = static_cast<uint32_t>(crc64);
51+
auto const* tail = reinterpret_cast<uint8_t const*>(p64 + q64);
4952
for (size_t i = 0, r = len & 7; i < r; ++i)
5053
crc = _mm_crc32_u8(crc, tail[i]);
54+
#else
55+
uint32_t crc = 0xFFFFFFFFu;
56+
auto q32 = len / 4;
57+
auto const* p32 = reinterpret_cast<uint32_t const*>(bmp);
58+
for (size_t i = 0; i < q32; ++i)
59+
crc = _mm_crc32_u32(crc, p32[i]);
60+
61+
auto const* tail = reinterpret_cast<uint8_t const*>(p32 + q32);
62+
for (size_t i = 0, r = len & 3; i < r; ++i)
63+
crc = _mm_crc32_u8(crc, tail[i]);
64+
#endif
5165

5266
return crc ^ 0xFFFFFFFFu;
5367
}
5468

5569
int main(int argc, char** argv) {
5670
for (int i = 1; i < argc; ++i) {
5771
uint32_t h = crc32c_bmp(argv[i]);
58-
std::cout << argv[i] << ": 0x" << std::hex << std::uppercase << h << "\n";
72+
std::cout << argv[i]
73+
<< ": 0x" << std::hex << std::uppercase << h
74+
<< std::dec << "\n";
5975
}
76+
return 0;
6077
}

0 commit comments

Comments
 (0)