use function predefinition

Author: makemek

main.cpp

@@ -11,6 +11,53 @@
 #include "threadInfo.h"
 #include "ntinfo.h"
 
+std::vector<DWORD> threadList(DWORD pid);
+DWORD GetThreadStartAddress(HANDLE processHandle, HANDLE hThread);
+
+bool isGameAvail;
+
+int main(int argc, char** argv) {
+	std::string gameName = "osu!";
+	LPCSTR LGameName = "osu!";
+	
+	HWND hGameWindow = NULL;
+	DWORD dwProcID = NULL;
+	HANDLE hProcHandle = NULL;
+	isGameAvail = false;
+
+	// keep polling until target process is opened
+	std::cout << "Looking for " << gameName << std::endl;
+	while (!isGameAvail) {
+		hGameWindow = FindWindow(NULL, LGameName);
+		GetWindowThreadProcessId(hGameWindow, &dwProcID);
+		if (!dwProcID)
+			continue;
+
+		std::cout << "Found it PID " << dwProcID << std::endl;
+		std::cout << "Grabbing handle" << std::endl;
+		hProcHandle = OpenProcess(PROCESS_ALL_ACCESS, FALSE, dwProcID);
+
+		if (hProcHandle == INVALID_HANDLE_VALUE || hProcHandle == NULL) {
+			std::cerr << "Failed to open process -- invalid handle" << std::endl;
+		}
+		else {
+			std::cout << "Success" << std::endl;
+			isGameAvail = true;
+		}
+	}
+	
+	std::vector<DWORD> threadId = threadList(dwProcID);
+	int stackNum = 0;
+	for (auto it = threadId.begin(); it != threadId.end(); ++it) {
+		HANDLE threadHandle = OpenThread(THREAD_GET_CONTEXT | THREAD_QUERY_INFORMATION, FALSE, *it);
+		DWORD threadStartAddress = GetThreadStartAddress(hProcHandle, threadHandle);
+		printf("TID: 0x%04x = THREADSTACK%2d BASE ADDRESS: 0x%04x\n", *it, stackNum, threadStartAddress);
+		stackNum++;
+	}
+
+	return EXIT_SUCCESS;
+}
+
 std::vector<DWORD> threadList(DWORD pid) {
 	// solution from http://stackoverflow.com/questions/1206878/enumerating-threads-in-windows
 	std::vector<DWORD> vect = std::vector<DWORD>();
@@ -25,12 +72,12 @@ std::vector<DWORD> threadList(DWORD pid) {
 			if (te.dwSize >= FIELD_OFFSET(THREADENTRY32, th32OwnerProcessID) +
 				sizeof(te.th32OwnerProcessID)) {
 
-				
+
 				if (te.th32OwnerProcessID == pid) {
 					printf("PID: %04d Thread ID: 0x%04x\n", te.th32OwnerProcessID, te.th32ThreadID);
 					vect.push_back(te.th32ThreadID);
 				}
-					
+
 			}
 			te.dwSize = sizeof(te);
 		} while (Thread32Next(h, &te));
@@ -45,7 +92,7 @@ DWORD GetThreadStartAddress(HANDLE processHandle, HANDLE hThread) {
 	DWORD stacktop = 0, result = 0;
 
 	MODULEINFO mi;
-		
+
 	GetModuleInformation(processHandle, LoadLibrary("kernel32.dll"), &mi, sizeof(mi));
 	stacktop = (DWORD)GetThreadStackTopAddress_x86(processHandle, hThread);
 
@@ -76,15 +123,15 @@ DWORD GetThreadStartAddress(HANDLE processHandle, HANDLE hThread) {
 		//find the stack entry pointing to the function that calls "ExitXXXXXThread"
 		//Fun thing to note: It's the first entry that points to a address in kernel32
 
-		DWORD* buf32 = new DWORD[4096/4];
-		
+		DWORD* buf32 = new DWORD[4096 / 4];
+
 		if (ReadProcessMemory(processHandle, (LPCVOID)(stacktop - 4096), buf32, 4096, NULL)) {
 			for (int i = 4096 / 4 - 1; i >= 0; --i) {
 				if (buf32[i] >= (DWORD)mi.lpBaseOfDll && buf32[i] <= (DWORD)mi.lpBaseOfDll + mi.SizeOfImage) {
 					result = stacktop - 4096 + i * 4;
 					break;
 				}
-					
+
 			}
 		}
 
@@ -93,47 +140,3 @@ DWORD GetThreadStartAddress(HANDLE processHandle, HANDLE hThread) {
 
 	return result;
 }
-
-bool isGameAvail;
-
-int main(int argc, char** argv) {
-	std::string gameName = "osu!";
-	LPCSTR LGameName = "osu!";
-	
-	HWND hGameWindow = NULL;
-	DWORD dwProcID = NULL;
-	HANDLE hProcHandle = NULL;
-	isGameAvail = false;
-
-	// keep polling until target process is opened
-	std::cout << "Looking for " << gameName << std::endl;
-	while (!isGameAvail) {
-		hGameWindow = FindWindow(NULL, LGameName);
-		GetWindowThreadProcessId(hGameWindow, &dwProcID);
-		if (!dwProcID)
-			continue;
-
-		std::cout << "Found it PID " << dwProcID << std::endl;
-		std::cout << "Grabbing handle" << std::endl;
-		hProcHandle = OpenProcess(PROCESS_ALL_ACCESS, FALSE, dwProcID);
-
-		if (hProcHandle == INVALID_HANDLE_VALUE || hProcHandle == NULL) {
-			std::cerr << "Failed to open process -- invalid handle" << std::endl;
-		}
-		else {
-			std::cout << "Success" << std::endl;
-			isGameAvail = true;
-		}
-	}
-	
-	std::vector<DWORD> threadId = threadList(dwProcID);
-	int stackNum = 0;
-	for (auto it = threadId.begin(); it != threadId.end(); ++it) {
-		HANDLE threadHandle = OpenThread(THREAD_GET_CONTEXT | THREAD_QUERY_INFORMATION, FALSE, *it);
-		DWORD threadStartAddress = GetThreadStartAddress(hProcHandle, threadHandle);
-		printf("TID: 0x%04x = THREADSTACK%2d BASE ADDRESS: 0x%04x\n", *it, stackNum, threadStartAddress);
-		stackNum++;
-	}
-
-	return EXIT_SUCCESS;
-}