<?php
/**
 * PDBSKiP v0.0.0.1
 *
 * @description PDB parse && repack
 * @author secator
 * @link https://secator.com/
 * @date 24.10.2024
 */


if (empty($_SERVER['argv'][1])) {
	echo 'usage: ' . basename(__FILE__) . ' chrome.dll.pdb';
	exit;
}

if (!is_file($_SERVER['argv'][1]) || !file_exists($_SERVER['argv'][1])) {
	echo 'file not found';
	exit;
}

// MSF заголовок, для unpack и pack
// size => 0x38
$msf = array(
	'signature' => 'a32',       // Microsoft C/C++ MSF 7.00 ...
	'page_size' => 'V',         // количество байт в странице/блоке
	'free_page' => 'V',         // не используется (значение 1 или 2)
	'pages' => 'V',             // всего страниц в файле
	'directory_size' => 'V',    // размер основной страницы
	'reserved' => 'V',          // reserved
	'directory' => 'V',         // на какой странице основная страница
);


$from = fopen($_SERVER['argv'][1], 'r+');

$header = unpack(implode('/', array_map(function ($key, $value) { return $value . $key; }, array_keys($msf),array_values($msf))), fread($from, 0x38));
if ($header['signature'] != "Microsoft C/C++ MSF 7.00\r\n\x1ADS\0\0\0") {
	echo 'Bad format';
	exit;
}

echo 'Read ' . $_SERVER['argv'][1] . PHP_EOL;

$original = array();

// количество основных страниц описания потоков
$directory_pages = ceil($header['directory_size'] / $header['page_size']);

// читаем все номера основных страниц
fseek($from, $header['directory'] * $header['page_size']);

$original['directory'] = array();
for ($i = 0; $i < $directory_pages; $i++) {
	$original['directory'][] = implode(unpack('V', fread($from, 4)));
}

// переход на описание всех потоков
$directory_current = 0;
fseek($from, $original['directory'][ $directory_current] * $header['page_size']);
$data = fread($from, $header['page_size']);

// количество всех потоков
$original['streams'] = implode(unpack('V', substr($data, 0, 4)));
$data = substr($data, 4);

// размер в байтах каждого потока
$original['stream_sizes'] = array();
for ($i = 0; $i < $original['streams']; $i++) {

	$original['stream_sizes'][] = implode(unpack('V', substr($data, 0, 4)));
	$data = substr($data,4);
	if (empty($data)) {
		
		$directory_current++;
		fseek($from, $original['directory'][ $directory_current] * $header['page_size']);
		$data = fread($from, $header['page_size']);
	}
}

// номера страниц каждого потока
$original['streams_pages'] = array();
for ($i = 0; $i < $original['streams']; $i++) {
	// сколько страниц в текущем потоке
	$pages = ceil($original['stream_sizes'][ $i ] / $header['page_size']);
	if ($pages > 0) {
		
		$list = array();
		for ($j = 0; $j < $pages; $j++) {
			$list[] = implode(unpack('V', substr($data, 0, 4)));
			$data   = substr($data, 4);
			if (empty($data)) {
				$directory_current++;
				fseek($from, $original['directory'][ $directory_current] * $header['page_size']);
				$data = fread($from, $header['page_size']);
			}
		}
		$original['streams_pages'][] = $list;
	} else {
		$original['streams_pages'][] = array();
	}
}

// что будем пропускать

$skipping = array();
// пропускаем TPI (2), тогда нужны все потоки
$skipping[2] = array(
	'name' => '.skip-tpi',
	'streams' => array_keys($original['streams_pages'])
);
// пропускаем DBI (3), тогда хватит первых десяти
$skipping[3] = array(
	'name' => '.skip-dbi',
	'streams' => array(0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10)
);


function align($to, $size ) {
	if (ftell($to) % $size) {
		for ($i = 0, $ic = $size - (ftell($to) % $size); $i < $ic; $i++) {
			fwrite($to, chr(0x00));
		}
	}
	return ;
}



foreach ($skipping as $index => $skip) {
	
	$filename = substr_replace($_SERVER['argv'][1], $skip['name'], (int)strrpos($_SERVER['argv'][1], '.'), 0);
	echo 'Pack ' . $filename . PHP_EOL;
	
	$to = fopen($filename, 'w+');
	
	// первые блоки в нули, потом перезапишем
	for ($i = 0; $i < $header['page_size'] * ($header['free_page'] + 1); $i++) {
		fwrite($to, chr(0x00));
	}
	
	//
	$streams = array(
		count($skip['streams']),
		'',
	);
	$bytes = 8;
	
	// 0 - описание всех потоков
	$stream_number = array(0 => array());
	
	for ($i = 1, $ic = count($skip['streams']); $i < $ic; $i++) {
		$bytes += 4;
		
		$pages = $original['streams_pages'][ $i ];
		
		$stream_number[ $i ] = array();
		
		if ($i === $index) {
			$streams[] = 0;
			continue;
		}
		
		$streams[] = $original['stream_sizes'][ $i ];
		
		for ($j = 0, $jc = count($pages); $j < $jc; $j++) {
			$bytes += 4;
			
			$stream_number[ $i ][] = ftell($to) / $header['page_size'];
			
			fseek($from, $pages[ $j ] * $header['page_size']);
			$data = fread($from, $header['page_size']);
			fwrite($to, $data);
		}
	}
	
	// количество страниц в описании потоков
	$pages = ceil($bytes / $header['page_size']);
	
	$directory = ftell($to) / $header['page_size'];
	for ($i = 0; $i < $pages; $i++) {
		$stream_number[0][] = $directory + $i;
	}
	
	// количество потоков
	$streams[1] = $bytes;
	
	// размеры потоков
	foreach ($streams as $data) {
		fwrite($to, pack('V', $data));
	}
	
	// номера страниц
	foreach ($stream_number as $size) {
		foreach ($size as $data) {
			fwrite($to, pack('V', $data));
		}
	}
	
	// выравниваем страницу
	align($to, $header['page_size']);
	
	// записываем основную страницу
	$directory_pages = ftell($to) / $header['page_size'];
	for ($i = 0; $i < $pages; $i++) {
		fwrite($to, pack('V', $directory + $i));
	}
	
	// выравниваем страницу
	align($to, $header['page_size']);
	
	// переписываем заголовок
	$header['pages']          = $directory_pages + 1;
	$header['directory']      = $directory_pages;
	$header['directory_size'] = $pages * $header['page_size'];
	
	fseek($to, 0);
	fwrite($to, pack(implode($msf), ...array_values($header)));
	
	fclose($to);
}