File: recoverypacket.cpp

package info (click to toggle)
libpar2 0.4-4
  • links: PTS, VCS
  • area: main
  • in suites: jessie, jessie-kfreebsd
  • size: 2,368 kB
  • ctags: 1,145
  • sloc: sh: 11,118; cpp: 8,657; makefile: 44
file content (127 lines) | stat: -rw-r--r-- 4,150 bytes parent folder | download | duplicates (2)
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
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
//  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

RecoveryPacket::RecoveryPacket(void)
{
  diskfile = NULL;
  offset = 0;
  packetcontext = NULL;
}

RecoveryPacket::~RecoveryPacket(void)
{
  delete packetcontext;
}

// Create a recovery packet. 

// The packet header can be almost completely filled in using the supplied
// information and computation of the hash of the packet started immediately.
// The hash will be updated as new data is written to the packet.

void RecoveryPacket::Create(DiskFile      *_diskfile, 
                            u64            _offset, 
                            u64            _blocksize, 
                            u32            _exponent, 
                            const MD5Hash &_setid)
{
  diskfile = _diskfile;
  offset = _offset;

  // Record everything we know in the packet.
  packet.header.magic  = packet_magic;
  packet.header.length = sizeof(packet) + _blocksize;
  //packet.header.hash;  // Not known yet.
  packet.header.setid  = _setid;
  packet.header.type   = recoveryblockpacket_type;
  packet.exponent      = _exponent;

  // Start computation of the packet hash
  packetcontext = new MD5Context;
  packetcontext->Update(&packet.header.setid, 
                        sizeof(RECOVERYBLOCKPACKET)-offsetof(RECOVERYBLOCKPACKET, header.setid));

  // Set the data block to immediatly follow the header on disk
  datablock.SetLocation(_diskfile, _offset + sizeof(packet));
  datablock.SetLength(_blocksize);
}

// Write data from the buffer to the data block on disk
bool RecoveryPacket::WriteData(u64 position, 
                               size_t size, 
                               const void *buffer)
{
  // Update the packet hash
  packetcontext->Update(buffer, size);

  // Write the data to the data block
  size_t wrote;
  return datablock.WriteData(position, size, buffer, wrote);
}

// Write the header of the packet to disk
bool RecoveryPacket::WriteHeader(void)
{
  // Finish computing the packet hash
  packetcontext->Final(packet.header.hash);

  // Write the header to disk
  return diskfile->Write(offset, &packet, sizeof(packet));  
}

// Load the recovery packet from disk.
//
// The header of the packet will already have been read from disk. The only
// thing that actually needs to be read is the exponent value.
// The recovery data is not read from disk at this point. Its location
// is recovered in the DataBlock object.

bool RecoveryPacket::Load(DiskFile      *_diskfile, 
                          u64            _offset, 
                          PACKET_HEADER &_header)
{
  diskfile = _diskfile;
  offset = _offset;

  // Is the packet actually large enough
  if (_header.length <= sizeof(packet))
  {
    return false;
  }

  // Save the fixed header
  packet.header = _header;

  // Set the data block to immediatly follow the header on disk
  datablock.SetLocation(diskfile, offset + sizeof(packet));
  datablock.SetLength(packet.header.length - sizeof(packet));

  // Read the rest of the packet header
  return diskfile->Read(offset + sizeof(packet.header), &packet.exponent, sizeof(packet)-sizeof(packet.header));
}