TheCloud/panda3d
1
0
Fork 0
mirror of https://github.com/panda3d/panda3d.git synced 2025-10-03 02:15:43 -04:00
Code Issues Packages Projects Releases Wiki Activity

patcher cleanup

Browse Source
This commit is contained in:
David Rose 2009-08-31 06:10:55 +00:00
parent 9a68233e6f
commit b9b6dc8f0d
3 changed files with 62 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
panda/src/express/patchfile.I
View File

@ -17,11 +17,13 @@
////////////////////////////////////////////////////////////////////
// Function: Patchfile::get_progress
// Access: Published
// Description:
// Description: Returns a value in the range 0..1, representing the
// amount of progress through the patchfile, during a
// session.
////////////////////////////////////////////////////////////////////
INLINE float Patchfile::
get_progress() const {
if (false == _initiated) {
if (!_initiated) {
express_cat.warning()
<< "Patchfile::get_progress() - Patch has not been initiated" << endl;
return 0.0f;
@ -39,6 +41,9 @@ get_progress() const {
// subfile-by-subfile basis. If this flag is false, the
// Patchfile will always patch the file on a full-file
// basis.
//
// This has effect only when building patches; it is not
// used for applying patches.
////////////////////////////////////////////////////////////////////
INLINE void Patchfile::
set_allow_multifile(bool allow_multifile) {
@ -48,7 +53,7 @@ set_allow_multifile(bool allow_multifile) {
////////////////////////////////////////////////////////////////////
// Function: Patchfile::get_allow_multifile
// Access: Published
// Description:
// Description: See set_allow_multifile().
////////////////////////////////////////////////////////////////////
INLINE bool Patchfile::
get_allow_multifile() {

75
panda/src/express/patchfile.cxx
View File

@ -23,8 +23,8 @@
#include "streamWriter.h"
#include "multifile.h"
#include "hashVal.h"
#include "virtualFileSystem.h"
#include <stdio.h> // for tempnam
#include <string.h> // for strstr
#ifdef HAVE_TAR
@ -32,10 +32,6 @@
#include <fcntl.h> // for O_RDONLY
#endif // HAVE_TAR
#ifdef WIN32_VC
#define tempnam _tempnam
#endif
#ifdef HAVE_TAR
istream *Patchfile::_tar_istream = NULL;
#endif // HAVE_TAR
@ -130,6 +126,10 @@ init(PT(Buffer) buffer) {
_version_number = 0;
_allow_multifile = true;
_patch_stream = NULL;
_origfile_stream = NULL;
reset_footprint_length();
}
@ -147,6 +147,9 @@ Patchfile::
if (_initiated) {
cleanup();
}
nassertv(_patch_stream == NULL);
nassertv(_origfile_stream == NULL);
}
////////////////////////////////////////////////////////////////////
@ -164,8 +167,15 @@ cleanup() {
}
// close files
_origfile_stream.close();
_patch_stream.close();
VirtualFileSystem *vfs = VirtualFileSystem::get_global_ptr();
if (_origfile_stream != NULL) {
vfs->close_read_file(_origfile_stream);
_origfile_stream = NULL;
}
if (_patch_stream != NULL) {
vfs->close_read_file(_patch_stream);
_patch_stream = NULL;
}
_write_stream.close();
_initiated = false;
@ -216,10 +226,14 @@ initiate(const Filename &patch_file, const Filename &orig_file,
nassertr(orig_file != target_file, EU_error_abort);
VirtualFileSystem *vfs = VirtualFileSystem::get_global_ptr();
// Open the original file for read
nassertr(_origfile_stream == NULL, EU_error_abort);
_orig_file = orig_file;
_orig_file.set_binary();
if (!_orig_file.open_read(_origfile_stream)) {
_origfile_stream = vfs->open_read_file(_orig_file, false);
if (_origfile_stream == NULL) {
express_cat.error()
<< "Patchfile::initiate() - Failed to open file: " << _orig_file << endl;
return get_write_error();
@ -264,7 +278,11 @@ read_header(const Filename &patch_file) {
}
int result = internal_read_header(patch_file);
_patch_stream.close();
if (_patch_stream != NULL) {
VirtualFileSystem *vfs = VirtualFileSystem::get_global_ptr();
vfs->close_read_file(_patch_stream);
_patch_stream = NULL;
}
return result;
}
@ -296,7 +314,9 @@ run() {
return EU_error_abort;
}
StreamReader patch_reader(_patch_stream);
nassertr(_patch_stream != NULL, EU_error_abort);
nassertr(_origfile_stream != NULL, EU_error_abort);
StreamReader patch_reader(*_patch_stream);
buflen = _buffer->get_length();
bytes_read = 0;
@ -306,7 +326,7 @@ run() {
// read # of ADD bytes
nassertr(_buffer->get_length() >= (int)sizeof(ADD_length), false);
ADD_length = patch_reader.get_uint16();
if (_patch_stream.fail()) {
if (_patch_stream->fail()) {
express_cat.error()
<< "Truncated patch file.\n";
return EU_error_file_invalid;
@ -330,8 +350,8 @@ run() {
PN_uint32 bytes_left = (PN_uint32)ADD_length;
while (bytes_left > 0) {
PN_uint32 bytes_this_time = (PN_uint32) min(bytes_left, (PN_uint32) buflen);
_patch_stream.read(_buffer->_buffer, bytes_this_time);
if (_patch_stream.fail()) {
_patch_stream->read(_buffer->_buffer, bytes_this_time);
if (_patch_stream->fail()) {
express_cat.error()
<< "Truncated patch file.\n";
return EU_error_file_invalid;
@ -344,7 +364,7 @@ run() {
// read # of COPY bytes
nassertr(_buffer->get_length() >= (int)sizeof(COPY_length), false);
COPY_length = patch_reader.get_uint16();
if (_patch_stream.fail()) {
if (_patch_stream->fail()) {
express_cat.error()
<< "Truncated patch file.\n";
return EU_error_file_invalid;
@ -363,7 +383,7 @@ run() {
// read copy offset
nassertr(_buffer->get_length() >= (int)sizeof(COPY_offset), false);
COPY_offset = patch_reader.get_int32();
if (_patch_stream.fail()) {
if (_patch_stream->fail()) {
express_cat.error()
<< "Truncated patch file.\n";
return EU_error_file_invalid;
@ -371,11 +391,11 @@ run() {
// seek to the copy source pos
if (_version_number < 2) {
_origfile_stream.seekg(COPY_offset, ios::beg);
_origfile_stream->seekg(COPY_offset, ios::beg);
} else {
_origfile_stream.seekg(COPY_offset, ios::cur);
_origfile_stream->seekg(COPY_offset, ios::cur);
}
if (_origfile_stream.fail()) {
if (_origfile_stream->fail()) {
express_cat.error()
<< "Invalid copy offset in patch file.\n";
return EU_error_file_invalid;
@ -384,7 +404,7 @@ run() {
if (express_cat.is_spam()) {
express_cat.spam()
<< "COPY: " << COPY_length << " bytes from offset "
<< COPY_offset << " (from " << _origfile_stream.tellg()
<< COPY_offset << " (from " << _origfile_stream->tellg()
<< " to " << _write_stream.tellp() << ")"
<< endl;
}
@ -394,8 +414,8 @@ run() {
while (bytes_left > 0) {
PN_uint32 bytes_this_time = (PN_uint32) min(bytes_left, (PN_uint32) buflen);
_origfile_stream.read(_buffer->_buffer, bytes_this_time);
if (_origfile_stream.fail()) {
_origfile_stream->read(_buffer->_buffer, bytes_this_time);
if (_origfile_stream->fail()) {
express_cat.error()
<< "Invalid copy length in patch file.\n";
return EU_error_file_invalid;
@ -421,7 +441,11 @@ run() {
MD5_actual.hash_file(_output_file);
if (_MD5_ofResult != MD5_actual) {
// Whoops, patching screwed up somehow.
_origfile_stream.close();
if (_origfile_stream != NULL) {
VirtualFileSystem *vfs = VirtualFileSystem::get_global_ptr();
vfs->close_read_file(_origfile_stream);
_origfile_stream = NULL;
}
_write_stream.close();
express_cat.info()
@ -545,9 +569,12 @@ apply(Filename &patch_file, Filename &orig_file, const Filename &target_file) {
int Patchfile::
internal_read_header(const Filename &patch_file) {
// Open the patch file for read
VirtualFileSystem *vfs = VirtualFileSystem::get_global_ptr();
nassertr(_patch_stream == NULL, EU_error_abort);
_patch_file = patch_file;
_patch_file.set_binary();
if (!_patch_file.open_read(_patch_stream)) {
_patch_stream = vfs->open_read_file(_patch_file, true);
if (_patch_stream == NULL) {
express_cat.error()
<< "Patchfile::initiate() - Failed to open file: " << _patch_file << endl;
return get_write_error();
@ -556,7 +583,7 @@ internal_read_header(const Filename &patch_file) {
/////////////
// read header, make sure the patch file is valid
StreamReader patch_reader(_patch_stream);
StreamReader patch_reader(*_patch_stream);
// check the magic number
nassertr(_buffer->get_length() >= _v0_header_length, false);

6
panda/src/express/patchfile.h
View File

@ -165,7 +165,7 @@ private:
PN_uint32 _cache_copy_start;
PN_uint32 _cache_copy_length;
protected:
private:
PT(Buffer) _buffer; // this is the work buffer for apply -- used to prevent virtual memory swapping
// async patch apply state variables
@ -180,9 +180,9 @@ protected:
PN_uint32 _total_bytes_to_process;
PN_uint32 _total_bytes_processed;
pifstream _patch_stream;
istream *_patch_stream;
pofstream _write_stream;
pifstream _origfile_stream;
istream *_origfile_stream;
Filename _patch_file;
Filename _orig_file;