From 11771cfdeade7fbb283a208138c6561ba779779c Mon Sep 17 00:00:00 2001
From: George Hazan <george.hazan@gmail.com>
Date: Thu, 17 May 2012 14:57:45 +0000
Subject: 'tools' folder renamed

git-svn-id: http://svn.miranda-ng.org/main/trunk@12 1316c22d-e87f-b044-9b9b-93d7a3e3ba9c
---
 tools/rebaser/rebaser.cpp | 138 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 138 insertions(+)
 create mode 100644 tools/rebaser/rebaser.cpp

(limited to 'tools/rebaser/rebaser.cpp')

diff --git a/tools/rebaser/rebaser.cpp b/tools/rebaser/rebaser.cpp
new file mode 100644
index 0000000000..f6c048358b
--- /dev/null
+++ b/tools/rebaser/rebaser.cpp
@@ -0,0 +1,138 @@
+
+#define WIN32_LEAN_AND_MEAN		// Exclude rarely-used stuff from Windows headers
+
+#include <io.h>
+#include <stdio.h>
+#include <stdlib.h>
+
+#include <windows.h>
+
+#include "imagehlp.h"
+
+#define SIZEOF(X) (sizeof(X)/sizeof(X[0]))
+
+DWORD glbOffset = 0x12000000;   // rebased offset by default
+bool	glbCheckMode = false;
+
+//====[ Prints standard Win32 error message ]============================================
+
+void PrintWin32Error( DWORD tErrorCode )
+{
+   WCHAR tBuffer[ 1024 ];
+   FormatMessageW( FORMAT_MESSAGE_FROM_SYSTEM, NULL, GetLastError(), 0, tBuffer, SIZEOF( tBuffer ), NULL );
+   fputws( tBuffer, stdout );
+   fputc( '\n', stdout );
+}
+
+//====[ Trims trailing spaces ]==========================================================
+
+char* rtrim( char *string )
+{
+   if ( string == 0 )
+      return 0;
+
+   char* p = string + strlen( string ) - 1;
+
+   while ( p >= string )
+   {  if ( *p == ' ' || *p == '\t' || *p == '\n' || *p == '\r' )
+         *p-- = 0;
+      else
+         break;
+   }
+
+   return string;
+}
+
+//====[ Main ]===========================================================================
+
+int main( int argc, char** argv )
+{
+	fprintf( stderr, "DLL base address scheduler v.1.2, copyright 2001-2002 (c) George Hazan.\n\n" );
+
+   //----[ Command line parsing ]--------------------------------------------------------
+
+   for ( int i=1; i < argc; i++ )
+   {
+      if ( argv[ i ][ 0 ] != '/' )
+      {
+LBL_Usage:
+         fprintf( stderr, "Usage: Rebaser.exe [/BASE:<hex_digit>] [/CHECK]\n" );
+		   return 200;
+      }
+
+      if ( memicmp( &argv[ i ][ 1 ], "BASE:", 5 ) == 0 )
+         sscanf( &argv[ i ][ 6 ], "%08X", &glbOffset );
+		else if ( memicmp( &argv[ i ][ 1 ], "CHECK", 5 ) == 0 )
+		{
+			glbCheckMode = true;
+			glbOffset = 0x400000;
+		}
+      else goto LBL_Usage;
+   }
+
+   //----[ Command line ok, opening data files ]-----------------------------------------
+
+	fprintf( stdout, "Scanning DLLs...\n" );
+
+	int  locResult = 0;
+	char locFileName[ MAX_PATH ];
+
+	while ( fgets( locFileName, sizeof( locFileName ), stdin ) != NULL )
+	{
+		rtrim( locFileName );
+		fprintf( stdout, "Processing '%s'...\n", locFileName );
+
+		//----| If we need to check files only, check them |-------------------------------
+		if ( glbCheckMode )
+		{
+			DWORD locOldImageSize, locOldBase, locNewSize, locNewBase;
+			if ( !ReBaseImage( locFileName, NULL, FALSE, FALSE, FALSE, FALSE, &locOldImageSize,
+																&locOldBase, &locNewSize, &locNewBase, 0 ))
+				PrintWin32Error( GetLastError() );
+
+			if ( locOldBase == glbOffset )
+			{
+				fprintf( stdout, "Image is not based\n\n" );
+				locResult = 2;
+			}
+
+			continue;
+		}
+
+		//----| Preparing files |----------------------------------------------------------
+LBL_Ok:
+		HANDLE locFileHandle = CreateFile( locFileName, GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, 0 );
+		if ( locFileHandle == INVALID_HANDLE_VALUE )
+		{
+			fprintf( stdout, "File '%s' cannot be opened, skipped.\n\n", locFileName );
+			locResult = 2;
+			continue;
+		}
+
+		FILETIME locCreateTime, locAccessTime, locModifyTime;
+		GetFileTime( locFileHandle, &locCreateTime, &locAccessTime, &locModifyTime );
+		CloseHandle( locFileHandle );
+
+		//----| Processing |---------------------------------------------------------------
+
+		DWORD locOldImageSize, locOldBase, locNewSize;
+		if ( !ReBaseImage( locFileName, NULL, TRUE, FALSE, FALSE, FALSE, &locOldImageSize,
+															&locOldBase, &locNewSize, &glbOffset, 0 ))
+		{
+			PrintWin32Error( GetLastError() );
+			locResult = 2;
+		}
+
+		glbOffset += locNewSize;
+
+		//----| Write down the original file timestamp |-----------------------------------
+
+		locFileHandle = CreateFile( locFileName, GENERIC_WRITE, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, 0 );
+		if ( locFileHandle != INVALID_HANDLE_VALUE )
+		{
+			SetFileTime( locFileHandle, &locCreateTime, &locAccessTime, &locModifyTime );
+			CloseHandle( locFileHandle );
+	}	}
+
+   return locResult;
+}
-- 
cgit v1.2.3