2014-04-10 01:22:57 +02:00
|
|
|
/*****************************************************************************
|
|
|
|
* Copyright (c) 2014 Ted John
|
|
|
|
* OpenRCT2, an open source clone of Roller Coaster Tycoon 2.
|
|
|
|
*
|
|
|
|
* This file is part of OpenRCT2.
|
|
|
|
*
|
|
|
|
* OpenRCT2 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 3 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, see <http://www.gnu.org/licenses/>.
|
|
|
|
*****************************************************************************/
|
|
|
|
|
2014-05-25 11:46:11 +02:00
|
|
|
#include <windows.h>
|
2014-05-19 21:10:32 +02:00
|
|
|
#include <string.h>
|
2014-10-06 18:36:58 +02:00
|
|
|
#include "../addresses.h"
|
2014-04-10 01:22:57 +02:00
|
|
|
#include "sawyercoding.h"
|
|
|
|
|
2014-09-29 13:43:12 +02:00
|
|
|
static int decode_chunk_rle(uint8* src_buffer, uint8* dst_buffer, int length);
|
2014-04-10 01:22:57 +02:00
|
|
|
static int decode_chunk_repeat(char *buffer, int length);
|
|
|
|
static void decode_chunk_rotate(char *buffer, int length);
|
|
|
|
|
2014-10-03 21:07:21 +02:00
|
|
|
int encode_chunk_rle(char *src_buffer, char *dst_buffer, int length);
|
|
|
|
void encode_chunk_rotate(char *buffer, int length);
|
2014-10-04 12:41:16 +02:00
|
|
|
|
|
|
|
int sawyercoding_calculate_checksum(uint8* buffer, uint32 length){
|
|
|
|
int checksum = 0;
|
|
|
|
do {
|
|
|
|
int bufferSize = min(length , 1024);
|
|
|
|
|
|
|
|
for (int i = 0; i < bufferSize; i++)
|
|
|
|
checksum += buffer[i];
|
|
|
|
length -= bufferSize;
|
|
|
|
} while (length != 0);
|
|
|
|
|
|
|
|
return checksum;
|
|
|
|
}
|
|
|
|
|
2014-05-04 18:34:13 +02:00
|
|
|
/**
|
|
|
|
*
|
|
|
|
* rct2: 0x00676FD2
|
|
|
|
*/
|
2014-05-24 23:14:42 +02:00
|
|
|
int sawyercoding_validate_checksum(FILE *file)
|
2014-05-04 18:34:13 +02:00
|
|
|
{
|
2014-05-24 23:14:42 +02:00
|
|
|
uint32 i, checksum, fileChecksum, dataSize, bufferSize;
|
2014-05-04 18:34:13 +02:00
|
|
|
uint8 buffer[1024];
|
|
|
|
|
|
|
|
// Get data size
|
2014-05-24 23:14:42 +02:00
|
|
|
fseek(file, 0, SEEK_END);
|
|
|
|
dataSize = ftell(file);
|
|
|
|
if (dataSize < 8)
|
2014-05-12 02:45:45 +02:00
|
|
|
return 0;
|
2014-05-04 18:34:13 +02:00
|
|
|
dataSize -= 4;
|
|
|
|
|
|
|
|
// Calculate checksum
|
2014-05-24 23:14:42 +02:00
|
|
|
fseek(file, 0, SEEK_SET);
|
2014-05-04 18:34:13 +02:00
|
|
|
checksum = 0;
|
|
|
|
do {
|
|
|
|
bufferSize = min(dataSize, 1024);
|
2014-05-24 23:14:42 +02:00
|
|
|
if (fread(buffer, bufferSize, 1, file) != 1)
|
2014-05-04 18:34:13 +02:00
|
|
|
return 0;
|
|
|
|
|
|
|
|
for (i = 0; i < bufferSize; i++)
|
|
|
|
checksum += buffer[i];
|
|
|
|
dataSize -= bufferSize;
|
|
|
|
} while (dataSize != 0);
|
|
|
|
|
|
|
|
// Read file checksum
|
2014-05-24 23:14:42 +02:00
|
|
|
if (fread(&fileChecksum, sizeof(fileChecksum), 1, file) != 1)
|
2014-05-04 18:34:13 +02:00
|
|
|
return 0;
|
|
|
|
|
|
|
|
// Reset file position
|
2014-05-24 23:14:42 +02:00
|
|
|
fseek(file, 0, SEEK_SET);
|
2014-05-04 18:34:13 +02:00
|
|
|
|
|
|
|
// Validate
|
|
|
|
return checksum == fileChecksum;
|
|
|
|
}
|
|
|
|
|
2014-09-27 13:05:40 +02:00
|
|
|
|
2014-04-10 01:22:57 +02:00
|
|
|
/**
|
|
|
|
*
|
|
|
|
* rct2: 0x0067685F
|
|
|
|
* buffer (esi)
|
|
|
|
*/
|
2014-05-24 23:14:42 +02:00
|
|
|
int sawyercoding_read_chunk(FILE *file, uint8 *buffer)
|
2014-04-10 01:22:57 +02:00
|
|
|
{
|
2014-04-10 04:23:12 +02:00
|
|
|
sawyercoding_chunk_header chunkHeader;
|
2014-04-10 01:22:57 +02:00
|
|
|
|
2014-04-10 04:23:12 +02:00
|
|
|
// Read chunk header
|
2014-05-24 23:14:42 +02:00
|
|
|
if (fread(&chunkHeader, sizeof(sawyercoding_chunk_header), 1, file) != 1) {
|
|
|
|
RCT2_ERROR("Unable to read chunk header!");
|
|
|
|
return -1;
|
2014-05-21 00:07:57 +02:00
|
|
|
}
|
2014-04-10 01:22:57 +02:00
|
|
|
|
2014-09-29 13:43:12 +02:00
|
|
|
uint8* src_buffer = malloc(chunkHeader.length);
|
|
|
|
|
2014-04-10 01:22:57 +02:00
|
|
|
// Read chunk data
|
2014-09-29 13:43:12 +02:00
|
|
|
if (fread(src_buffer, chunkHeader.length, 1, file) != 1) {
|
|
|
|
free(src_buffer);
|
2014-05-24 23:14:42 +02:00
|
|
|
RCT2_ERROR("Unable to read chunk data!");
|
|
|
|
return -1;
|
2014-05-21 00:07:57 +02:00
|
|
|
}
|
2014-04-10 01:22:57 +02:00
|
|
|
|
|
|
|
// Decode chunk data
|
2014-04-10 04:23:12 +02:00
|
|
|
switch (chunkHeader.encoding) {
|
2014-09-29 13:43:12 +02:00
|
|
|
case CHUNK_ENCODING_NONE:
|
|
|
|
memcpy(buffer, src_buffer, chunkHeader.length);
|
|
|
|
break;
|
2014-04-10 01:22:57 +02:00
|
|
|
case CHUNK_ENCODING_RLE:
|
2014-09-29 13:43:12 +02:00
|
|
|
chunkHeader.length = decode_chunk_rle(src_buffer, buffer, chunkHeader.length);
|
2014-04-10 01:22:57 +02:00
|
|
|
break;
|
|
|
|
case CHUNK_ENCODING_RLECOMPRESSED:
|
2014-09-29 13:43:12 +02:00
|
|
|
chunkHeader.length = decode_chunk_rle(src_buffer, buffer, chunkHeader.length);
|
2014-04-10 04:23:12 +02:00
|
|
|
chunkHeader.length = decode_chunk_repeat(buffer, chunkHeader.length);
|
2014-04-10 01:22:57 +02:00
|
|
|
break;
|
|
|
|
case CHUNK_ENCODING_ROTATE:
|
2014-09-29 13:43:12 +02:00
|
|
|
memcpy(buffer, src_buffer, chunkHeader.length);
|
2014-04-10 04:23:12 +02:00
|
|
|
decode_chunk_rotate(buffer, chunkHeader.length);
|
2014-04-10 01:22:57 +02:00
|
|
|
break;
|
|
|
|
}
|
2014-09-29 13:43:12 +02:00
|
|
|
free(src_buffer);
|
2014-04-10 01:22:57 +02:00
|
|
|
// Set length
|
2014-04-10 04:23:12 +02:00
|
|
|
RCT2_GLOBAL(0x009E3828, uint32) = chunkHeader.length;
|
|
|
|
return chunkHeader.length;
|
2014-04-10 01:22:57 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
*
|
|
|
|
* rct2: 0x0067693A
|
|
|
|
*/
|
2014-09-29 13:43:12 +02:00
|
|
|
static int decode_chunk_rle(uint8* src_buffer, uint8* dst_buffer, int length)
|
2014-04-10 01:22:57 +02:00
|
|
|
{
|
|
|
|
int i, j, count;
|
2014-09-29 13:43:12 +02:00
|
|
|
uint8 *dst, rleCodeByte;
|
2014-04-10 01:22:57 +02:00
|
|
|
|
2014-09-29 13:43:12 +02:00
|
|
|
dst = dst_buffer;
|
2014-04-10 01:22:57 +02:00
|
|
|
|
|
|
|
for (i = 0; i < length; i++) {
|
2014-09-29 13:43:12 +02:00
|
|
|
rleCodeByte = src_buffer[i];
|
2014-04-10 01:22:57 +02:00
|
|
|
if (rleCodeByte & 128) {
|
|
|
|
i++;
|
2014-04-10 18:08:41 +02:00
|
|
|
count = 257 - rleCodeByte;
|
2014-04-10 01:22:57 +02:00
|
|
|
for (j = 0; j < count; j++)
|
2014-09-29 13:43:12 +02:00
|
|
|
*dst++ = src_buffer[i];
|
2014-04-10 01:22:57 +02:00
|
|
|
} else {
|
2014-04-10 18:08:41 +02:00
|
|
|
for (j = 0; j <= rleCodeByte; j++)
|
2014-09-29 13:43:12 +02:00
|
|
|
*dst++ = src_buffer[++i];
|
2014-04-10 01:22:57 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Return final size
|
2014-09-29 20:07:25 +02:00
|
|
|
return dst - dst_buffer;
|
2014-04-10 01:22:57 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
*
|
|
|
|
* rct2: 0x006769F1
|
|
|
|
*/
|
|
|
|
static int decode_chunk_repeat(char *buffer, int length)
|
|
|
|
{
|
|
|
|
int i, j, count;
|
2014-05-21 00:07:57 +02:00
|
|
|
unsigned char *src, *dst, *copyOffset;
|
2014-04-10 01:22:57 +02:00
|
|
|
|
|
|
|
// Backup buffer
|
|
|
|
src = malloc(length);
|
|
|
|
memcpy(src, buffer, length);
|
|
|
|
dst = buffer;
|
|
|
|
|
|
|
|
for (i = 0; i < length; i++) {
|
|
|
|
if (src[i] == 0xFF) {
|
|
|
|
*dst++ = src[++i];
|
|
|
|
} else {
|
2014-04-10 18:08:41 +02:00
|
|
|
count = (src[i] & 7) + 1;
|
2014-04-10 01:22:57 +02:00
|
|
|
copyOffset = dst + (int)(src[i] >> 3) - 32;
|
|
|
|
for (j = 0; j < count; j++)
|
|
|
|
*dst++ = *copyOffset++;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Free backup buffer
|
|
|
|
free(src);
|
|
|
|
|
|
|
|
// Return final size
|
2014-05-21 00:07:57 +02:00
|
|
|
return (char*)dst - buffer;
|
2014-04-10 01:22:57 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
*
|
|
|
|
* rct2: 0x006768F4
|
|
|
|
*/
|
|
|
|
static void decode_chunk_rotate(char *buffer, int length)
|
|
|
|
{
|
|
|
|
int i, code = 1;
|
|
|
|
for (i = 0; i < length; i++) {
|
|
|
|
buffer[i] = ror8(buffer[i], code);
|
|
|
|
code = (code + 2) % 8;
|
|
|
|
}
|
2014-04-21 11:27:48 +02:00
|
|
|
}
|
2014-10-01 22:49:37 +02:00
|
|
|
|
|
|
|
/**
|
|
|
|
*
|
|
|
|
* rct2: 0x006762E1
|
|
|
|
*
|
|
|
|
*/
|
2014-10-04 12:41:16 +02:00
|
|
|
int sawyercoding_write_chunk_buffer(uint8 *dst_file, uint8* buffer, sawyercoding_chunk_header chunkHeader){
|
|
|
|
uint8* encode_buffer;
|
2014-10-03 21:07:21 +02:00
|
|
|
|
2014-10-01 22:49:37 +02:00
|
|
|
switch (chunkHeader.encoding){
|
|
|
|
case CHUNK_ENCODING_NONE:
|
2014-10-04 12:41:16 +02:00
|
|
|
memcpy(dst_file, &chunkHeader, sizeof(sawyercoding_chunk_header));
|
|
|
|
dst_file += sizeof(sawyercoding_chunk_header);
|
|
|
|
memcpy(dst_file, buffer, chunkHeader.length);
|
|
|
|
//fwrite(&chunkHeader, sizeof(sawyercoding_chunk_header), 1, file);
|
|
|
|
//fwrite(buffer, 1, chunkHeader.length, file);
|
2014-10-01 22:49:37 +02:00
|
|
|
break;
|
|
|
|
case CHUNK_ENCODING_RLE:
|
2014-10-04 12:41:16 +02:00
|
|
|
encode_buffer = malloc(0x600000);
|
|
|
|
chunkHeader.length = encode_chunk_rle(buffer, encode_buffer, chunkHeader.length);
|
|
|
|
memcpy(dst_file, &chunkHeader, sizeof(sawyercoding_chunk_header));
|
|
|
|
dst_file += sizeof(sawyercoding_chunk_header);
|
|
|
|
memcpy(dst_file, encode_buffer, chunkHeader.length);
|
|
|
|
|
|
|
|
free(encode_buffer);
|
2014-10-01 22:49:37 +02:00
|
|
|
break;
|
|
|
|
case CHUNK_ENCODING_RLECOMPRESSED:
|
2014-10-04 12:41:16 +02:00
|
|
|
RCT2_ERROR("This has not been implemented");
|
|
|
|
return -1;
|
2014-10-01 22:49:37 +02:00
|
|
|
//chunkHeader.length = decode_chunk_rle(src_buffer, buffer, chunkHeader.length);
|
|
|
|
//chunkHeader.length = decode_chunk_repeat(buffer, chunkHeader.length);
|
|
|
|
break;
|
|
|
|
case CHUNK_ENCODING_ROTATE:
|
2014-10-03 21:07:21 +02:00
|
|
|
encode_chunk_rotate(buffer, chunkHeader.length);
|
2014-10-04 12:41:16 +02:00
|
|
|
memcpy(dst_file, &chunkHeader, sizeof(sawyercoding_chunk_header));
|
|
|
|
dst_file += sizeof(sawyercoding_chunk_header);
|
|
|
|
memcpy(dst_file, buffer, chunkHeader.length);
|
2014-10-01 22:49:37 +02:00
|
|
|
break;
|
|
|
|
}
|
2014-10-03 21:07:21 +02:00
|
|
|
|
2014-10-04 12:41:16 +02:00
|
|
|
return chunkHeader.length + sizeof(sawyercoding_chunk_header);
|
2014-10-03 21:07:21 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Ensure dst_buffer is bigger than src_buffer then resize afterwards
|
|
|
|
* returns length of dst_buffer
|
|
|
|
*/
|
|
|
|
int encode_chunk_rle(char *src_buffer, char *dst_buffer, int length)
|
|
|
|
{
|
|
|
|
char* src = src_buffer;
|
|
|
|
char* dst = dst_buffer;
|
|
|
|
char* end_src = src + length;
|
|
|
|
uint8 count = 0;
|
|
|
|
char* src_norm_start = src;
|
|
|
|
|
|
|
|
while (src < end_src - 1){
|
|
|
|
|
2014-10-04 12:41:16 +02:00
|
|
|
if ((count && *src == src[1]) || count > 120){
|
|
|
|
*dst++ = count - 1;
|
2014-10-03 21:07:21 +02:00
|
|
|
for (; count != 0; --count){
|
|
|
|
*dst++ = *src_norm_start++;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if (*src == src[1]){
|
2014-10-04 12:41:16 +02:00
|
|
|
for (; (count < 120) && ((src + count) < end_src); count++){
|
2014-10-03 21:07:21 +02:00
|
|
|
if (*src != src[count]) break;
|
|
|
|
}
|
|
|
|
*dst++ = 257 - count;
|
|
|
|
*dst++ = *src;
|
|
|
|
src += count;
|
|
|
|
src_norm_start = src;
|
|
|
|
count = 0;
|
|
|
|
}
|
|
|
|
else{
|
|
|
|
count++;
|
|
|
|
src++;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if (src == end_src - 1)count++;
|
|
|
|
if (count){
|
2014-10-04 12:41:16 +02:00
|
|
|
*dst++ = count - 1;
|
2014-10-03 21:07:21 +02:00
|
|
|
for (; count != 0; --count){
|
|
|
|
*dst++ = *src_norm_start++;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return dst - dst_buffer;
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
*
|
|
|
|
*
|
|
|
|
*/
|
|
|
|
void encode_chunk_rotate(char *buffer, int length)
|
|
|
|
{
|
|
|
|
int i, code = 1;
|
|
|
|
for (i = 0; i < length; i++) {
|
|
|
|
buffer[i] = rol8(buffer[i], code);
|
|
|
|
code = (code + 2) % 8;
|
|
|
|
}
|
|
|
|
}
|
2014-10-10 23:50:22 +02:00
|
|
|
|
|
|
|
int sawyercoding_decode_sv4(char *src, char *dst, int length)
|
|
|
|
{
|
|
|
|
// (0 to length - 4): RLE chunk
|
|
|
|
// (length - 4 to length): checksum
|
|
|
|
return decode_chunk_rle(src, dst, length - 4);
|
|
|
|
}
|
|
|
|
|
|
|
|
int sawyercoding_decode_sc4(char *src, char *dst, int length)
|
|
|
|
{
|
|
|
|
int decodedLength, i;
|
|
|
|
uint32 *code;
|
|
|
|
|
|
|
|
// Uncompress
|
|
|
|
decodedLength = decode_chunk_rle(src, dst, length - 4);
|
|
|
|
|
|
|
|
// Decode
|
|
|
|
for (i = 0x60018; i <= min(decodedLength - 1, 0x1F8353); i++)
|
|
|
|
dst[i] = dst[i] ^ 0x9C;
|
|
|
|
|
|
|
|
for (i = 0x60018; i <= min(decodedLength - 1, 0x1F8350); i += 4) {
|
|
|
|
dst[i + 1] = ror8(dst[i + 1], 3);
|
|
|
|
|
|
|
|
code = (uint32*)&dst[i];
|
|
|
|
*code = rol32(*code, 9);
|
|
|
|
}
|
|
|
|
|
|
|
|
return decodedLength;
|
|
|
|
}
|
|
|
|
|
|
|
|
int sawyercoding_encode_sv4(char *src, char *dst, int length)
|
|
|
|
{
|
|
|
|
int encodedLength, checksum;
|
|
|
|
|
|
|
|
// Encode
|
|
|
|
encodedLength = encode_chunk_rle(src, dst, length);
|
|
|
|
|
|
|
|
// Append checksum
|
|
|
|
checksum = sawyercoding_calculate_checksum(dst, encodedLength);
|
|
|
|
*((uint32*)&dst[encodedLength]) = checksum;
|
|
|
|
|
|
|
|
return encodedLength + 4;
|
|
|
|
}
|