1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
//
// Copyright (c) Microsoft. All rights reserved.
// Licensed under the MIT license. See LICENSE file in the project root for full license information.
//
//----------------------------------------------------------
// TOCFile.cpp - Abstraction for reading a TOC file
//----------------------------------------------------------
#include "standardpch.h"
#include "tocfile.h"
#include "logging.h"
// Tries to load a Table of Contents
void TOCFile::LoadToc(const char *inputFileName, bool validate)
{
HANDLE hIndex = CreateFileA(inputFileName, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL | FILE_FLAG_SEQUENTIAL_SCAN, NULL);
if (hIndex == INVALID_HANDLE_VALUE)
{
LogError("Failed to open file '%s'. GetLastError()=%u", inputFileName, GetLastError());
return;
}
// Now read the index file
LARGE_INTEGER val; // I'm abusing LARGE_INTEGER here...
DWORD read;
if (!ReadFile(hIndex, &val, sizeof(val), &read, nullptr) ||
(val.u.LowPart != *(DWORD *)("INDX")))
{
CloseHandle(hIndex);
LogWarning("The index file %s is invalid: it seems to be missing the starting sentinel/length", inputFileName);
return;
}
this->m_tocCount = val.u.HighPart;
this->m_tocArray = new TOCElement[this->m_tocCount];
// Read the whole array
if (!ReadFile(hIndex, &this->m_tocArray[0], (DWORD)(this->m_tocCount * sizeof(TOCElement)), &read, nullptr) ||
(read != (DWORD)(this->m_tocCount * sizeof(TOCElement))))
{
CloseHandle(hIndex);
this->Clear();
LogWarning("The index file %s is invalid: it appears to be truncated.", inputFileName);
return;
}
// Get the last 4 byte token (more abuse of LARGE_INTEGER)
if (!ReadFile(hIndex, &val.u.HighPart, sizeof(DWORD), &read, nullptr) ||
(read != sizeof(DWORD)) ||
(val.u.LowPart != val.u.HighPart))
{
CloseHandle(hIndex);
this->Clear();
LogWarning("The index file %s is invalid: it appears to be missing the ending sentinel.", inputFileName);
return;
}
CloseHandle(hIndex);
if (validate)
{
int lastNum = -1;
// Quickly validate that the index is sorted
for (size_t i = 0; i < this->m_tocCount; i++)
{
int nextNum = this->m_tocArray[i].Number;
if (nextNum <= lastNum)
{
// It wasn't sorted: abort
this->Clear();
LogWarning("The index file %s is invalid: it is not sorted.", inputFileName);
return;
}
lastNum = nextNum;
}
}
}
|
