aboutsummaryrefslogtreecommitdiff
path: root/engines/wintermute/Base/file/BPkgFile.cpp
blob: 08a92b52f65f9dd924e197b548c1675c1add2408 (plain)
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
/* ScummVM - Graphic Adventure Engine
 *
 * ScummVM is the legal property of its developers, whose names
 * are too numerous to list here. Please refer to the COPYRIGHT
 * file distributed with this source distribution.
 *
 * This program 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.

 * This program 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.
 *
 */

/*
 * This file is based on WME Lite.
 * http://dead-code.org/redir.php?target=wmelite
 * Copyright (c) 2011 Jan Nedoma
 */

#include "engines/wintermute/dcgf.h"
#include "engines/wintermute/Base/BPackage.h"
#include "engines/wintermute/Base/file/BPkgFile.h"
#include "engines/wintermute/Base/BGame.h"
#include "engines/wintermute/Base/BFileManager.h"
#include "common/util.h"
#include "common/file.h"
#include "common/stream.h"
#include "common/substream.h"
#include "common/zlib.h"

namespace WinterMute {

// HACK: wrapCompressedStream might set the size to 0, so we need a way to override it.
class CBPkgFile : public Common::SeekableReadStream {
	uint32 _size;
	Common::SeekableReadStream *_stream;
public:
	CBPkgFile(Common::SeekableReadStream *stream, uint32 knownLength) : _size(knownLength), _stream(stream) {}
	virtual ~CBPkgFile() {
		delete _stream;
	}
	virtual uint32 read(void *dataPtr, uint32 dataSize) {
		return _stream->read(dataPtr, dataSize);
	}
	virtual bool eos() const {
		return _stream->eos();
	}
	virtual int32 pos() const {
		return _stream->pos();
	}
	virtual int32 size() const {
		return _size;
	}
	virtual bool seek(int32 offset, int whence = SEEK_SET) {
		return _stream->seek(offset, whence);
	}
};

Common::SeekableReadStream *openPkgFile(const Common::String &filename, CBFileManager *fileManager) {
	CBFileEntry *fileEntry;
	Common::SeekableReadStream *file = NULL;
	char fileName[MAX_PATH];
	strcpy(fileName, filename.c_str());

	// correct slashes
	for (uint32 i = 0; i < strlen(fileName); i++) {
		if (fileName[i] == '/') fileName[i] = '\\';
	}

	fileEntry = fileManager->getPackageEntry(fileName);
	if (!fileEntry) return NULL;

	file = fileEntry->_package->getFilePointer();
	if (!file) return NULL;

	// TODO: Cleanup
	bool compressed = (fileEntry->_compressedLength != 0);
	/* _size = fileEntry->_length; */

	if (compressed) {
		// TODO: Really, most of this logic might be doable directly in the fileEntry?
		// But for now, this should get us rolling atleast.
		file = Common::wrapCompressedReadStream(new Common::SeekableSubReadStream(file, fileEntry->_offset, fileEntry->_offset + fileEntry->_length, DisposeAfterUse::YES));
	} else {
		file = new Common::SeekableSubReadStream(file, fileEntry->_offset, fileEntry->_offset + fileEntry->_length, DisposeAfterUse::YES);
	}
	if (file->size() == 0) {
		file = new CBPkgFile(file, fileEntry->_length);
	}

	file->seek(0);

	return file;
}

} // end of namespace WinterMute