mirror
/
ddr-tools
mirror of https://github.com/987123879113/ddr-tools
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Convert tabs to spaces.

Browse Source
fix-windows-compile
Wesley Castro 5 years ago
parent 9d84d76855
commit 1c59e7c7cd
1 changed files with 119 additions and 119 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 238
      tcb-extract.c

238
tcb-extract.c
Unescape View File

@ -1,16 +1,16 @@
#include <stdio.h>
#include <stdlib.h>
#include <sys/stat.h>
#include <string.h>
#include <unistd.h>
#include "BemaniLZ/BemaniLZ.h"
#if defined(_WIN32)
#define MKDIR(a) _mkdir(a);
#else
#define MKDIR(a) mkdir(a, 0777);
#endif
#include <stdio.h>
#include <stdlib.h>
#include <sys/stat.h>
#include <string.h>
#include <unistd.h>
#include "BemaniLZ/BemaniLZ.h"
#if defined(_WIN32)
#define MKDIR(a) _mkdir(a);
#else
#define MKDIR(a) mkdir(a, 0777);
#endif
int search_for_tcbs(const unsigned char *data, size_t dataLength, uint32_t *offsets, char *isCompressed)
{
int numFound = 0;
@ -44,118 +44,118 @@ int search_for_tcbs(const unsigned char *data, size_t dataLength, uint32_t *offs
return numFound;
}
int main(int argc, char *argv[])
{
if(argc != 3)
{
printf("TCB Extractor\n");
printf("Usage:\n");
printf("%s <MODE> <file containing TCBs>\n", argv[0]);
printf("Modes:\n");
printf("1 - Bruteforce search for compressed and uncompressed TCBs\n");
printf("2 - Extract from file beginning with table (compressed entries)\n");
printf("3 - Extract from file beginning with table (uncompressed entries)\n");
return EXIT_FAILURE;
}
FILE *compressed = fopen(argv[2], "rb");
if(!compressed)
{
int main(int argc, char *argv[])
{
if(argc != 3)
{
printf("TCB Extractor\n");
printf("Usage:\n");
printf("%s <MODE> <file containing TCBs>\n", argv[0]);
printf("Modes:\n");
printf("1 - Bruteforce search for compressed and uncompressed TCBs\n");
printf("2 - Extract from file beginning with table (compressed entries)\n");
printf("3 - Extract from file beginning with table (uncompressed entries)\n");
return EXIT_FAILURE;
}
FILE *compressed = fopen(argv[2], "rb");
if(!compressed)
{
printf("Error opening %s\n", argv[2]);
return EXIT_FAILURE;
}
/* Get input data */
return EXIT_FAILURE;
}
/* Get input data */
printf("Loading %s...\n", argv[2]);
fseek(compressed, 0, SEEK_END);
int length = ftell(compressed);
fseek(compressed, 0, SEEK_SET);
unsigned char *compressedData = malloc(length);
fread(compressedData, length, 1, compressed);
fclose(compressed);
/* Search for TCBs */
fseek(compressed, 0, SEEK_END);
int length = ftell(compressed);
fseek(compressed, 0, SEEK_SET);
unsigned char *compressedData = malloc(length);
fread(compressedData, length, 1, compressed);
fclose(compressed);
/* Search for TCBs */
printf("Searching for TCBs...\n");
uint32_t offsets[8000];
char isCompressed[8000] = {0};
int numEntries = 0;
// file has table
if(argv[1][0] == '2' || argv[1][0] == '3')
{
int offset = 0;
uint32_t firstEntry = (uint32_t)compressedData[0];
if(firstEntry % 4)
{
printf("WARNING: First entry is number of entries to follow (%d).\n", firstEntry);
}
printf("first entry at 0x%08X\n", firstEntry);
memcpy(offsets, compressedData, firstEntry);
numEntries = firstEntry/4;
if(argv[1][0] == '2')
{
for(int i = 0; i < numEntries; i++)
{
isCompressed[i] = 1;
}
}
}
else // no table, need to search for TCBs
{
int numEntries = 0;
// file has table
if(argv[1][0] == '2' || argv[1][0] == '3')
{
int offset = 0;
uint32_t firstEntry = (uint32_t)compressedData[0];
if(firstEntry % 4)
{
printf("WARNING: First entry is number of entries to follow (%d).\n", firstEntry);
}
printf("first entry at 0x%08X\n", firstEntry);
memcpy(offsets, compressedData, firstEntry);
numEntries = firstEntry/4;
if(argv[1][0] == '2')
{
for(int i = 0; i < numEntries; i++)
{
isCompressed[i] = 1;
}
}
}
else // no table, need to search for TCBs
{
numEntries = search_for_tcbs(compressedData, length, offsets, isCompressed);
}
offsets[numEntries] = length;
printf("Found %d TCBs\n", numEntries);
int numCompressed = 0;
for(int i = 0; i < sizeof(isCompressed); i++)
{
if(isCompressed[i] == 1)
numCompressed++;
}
printf("%d are compressed\n", numCompressed);
/* Decompress and save TCBs */
char outFolderName[100];
snprintf(outFolderName, 100, "_%s", argv[2]);
MKDIR(outFolderName);
chdir(outFolderName);
uint8_t *buffer = malloc(1024*1024*10);
for(unsigned int i = 0; i < numEntries; i++)
{
if (offsets[i+1] == 0)
break;
}
offsets[numEntries] = length;
printf("Found %d TCBs\n", numEntries);
int numCompressed = 0;
for(int i = 0; i < sizeof(isCompressed); i++)
{
if(isCompressed[i] == 1)
numCompressed++;
}
printf("%d are compressed\n", numCompressed);
/* Decompress and save TCBs */
char outFolderName[100];
snprintf(outFolderName, 100, "_%s", argv[2]);
MKDIR(outFolderName);
chdir(outFolderName);
uint8_t *buffer = malloc(1024*1024*10);
for(unsigned int i = 0; i < numEntries; i++)
{
if (offsets[i+1] == 0)
break;
printf("Extracting files: %04d/%d [%d%%]\r", i+1, numEntries, ((i+1)*100)/numEntries);
int len = offsets[i+1] - offsets[i];
char filename[100];
snprintf(filename, 100, "%08X.tcb", offsets[i]);
FILE *out = fopen(filename, "wb");
if(isCompressed[i])
{
int decLen = decompress(compressedData + offsets[i], len, buffer, 1024*1024*10);
fwrite(buffer, decLen, 1, out);
}
else
{
fwrite(compressedData + offsets[i], len, 1, out);
}
fclose(out);
}
int len = offsets[i+1] - offsets[i];
char filename[100];
snprintf(filename, 100, "%08X.tcb", offsets[i]);
FILE *out = fopen(filename, "wb");
if(isCompressed[i])
{
int decLen = decompress(compressedData + offsets[i], len, buffer, 1024*1024*10);
fwrite(buffer, decLen, 1, out);
}
else
{
fwrite(compressedData + offsets[i], len, 1, out);
}
fclose(out);
}
printf("\n");
free(compressedData);
free(buffer);
return 0;
}
free(compressedData);
free(buffer);
return 0;
}

Write Preview
Loading…
Cancel
Save