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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
// This file is part of par2cmdline (a PAR 2.0 compatible file verification and
// repair tool). See http://parchive.sourceforge.net for details of PAR 2.0.
//
// Copyright (c) 2003 Peter Brian Clements
//
// par2cmdline is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 2 of the License, or
// (at your option) any later version.
//
// par2cmdline is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software
// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
#include "par2cmdline.h"
#ifdef _MSC_VER
#ifdef _DEBUG
#undef THIS_FILE
static char THIS_FILE[]=__FILE__;
#define new DEBUG_NEW
#endif
#endif
VerificationHashTable::VerificationHashTable(void)
{
hashmask = 0;
hashtable = 0;
}
VerificationHashTable::~VerificationHashTable(void)
{
// Destroy the hash table
if (hashtable)
{
for (unsigned int entry=0; entry<=hashmask; entry++)
{
delete hashtable[entry];
}
}
delete [] hashtable;
}
// Allocate the hash table with a reasonable size
void VerificationHashTable::SetLimit(u32 limit)
{
// Pick a good size for the hash table
hashmask = 256;
while (hashmask < limit && hashmask < 65536)
{
hashmask <<= 1;
}
// Allocate and clear the hash table
hashtable = new VerificationHashEntry*[hashmask];
memset(hashtable, 0, hashmask * sizeof(hashtable[0]));
hashmask--;
}
// Load data from a verification packet
void VerificationHashTable::Load(Par2RepairerSourceFile *sourcefile, u64 blocksize)
{
VerificationHashEntry *preventry = 0;
// Get information from the sourcefile
VerificationPacket *verificationpacket = sourcefile->GetVerificationPacket();
u32 blockcount = verificationpacket->BlockCount();
// Iterate throught the data blocks for the source file and the verification
// entries in the verification packet.
vector<DataBlock>::iterator sourceblocks = sourcefile->SourceBlocks();
const FILEVERIFICATIONENTRY *verificationentry = verificationpacket->VerificationEntry(0);
u32 blocknumber = 0;
while (blocknumber<blockcount)
{
DataBlock &datablock = *sourceblocks;
// Create a new VerificationHashEntry with the details for the current
// data block and verification entry.
VerificationHashEntry *entry = new VerificationHashEntry(sourcefile,
&datablock,
blocknumber == 0,
verificationentry);
// Insert the entry in the hash table
entry->Insert(&hashtable[entry->Checksum() & hashmask]);
// Make the previous entry point forwards to this one
if (preventry)
{
preventry->Next(entry);
}
preventry = entry;
++blocknumber;
++sourceblocks;
++verificationentry;
}
}
|
