Thread.h

/*
 *	PROJECT: Capture
 *	FILE: Thread.h
 *	AUTHORS: Ramon Steenson (rsteenson@gmail.com) & Christian Seifert (christian.seifert@gmail.com)
 *
 *	Developed by Victoria University of Wellington and the New Zealand Honeynet Alliance
 *
 *	This file is part of Capture.
 *
 *	Capture 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 2 of the License, or
 *  (at your option) any later version.
 *
 *  Capture 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 Capture; if not, write to the Free Software
 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 */
#pragma once
#include <windows.h>
#include <stdio.h>
/*
	Class: Thread

	Class that accepts a pointer to another class that implements the <Runnable> interface. This then wraps the c-style function 
	for creating a thread (threadProc) and creates an OO way to initialise, start, and stop a thread.
*/
/*
	Interface: Runnable

	Workaround to create an OO-style thread
*/
struct Runnable {
  virtual void run() = 0;
};

typedef struct tagTHREADNAME_INFO
{
   DWORD dwType; // must be 0x1000
   LPCSTR szName; // pointer to name (in user addr space)
   DWORD dwThreadID; // thread ID (-1=caller thread)
   DWORD dwFlags; // reserved for future use, must be zero
} THREADNAME_INFO;



class Thread {
public:
	/*
		Constructor: Thread

		Creates a thread object and stores the object which implements the Runnable interface in <_threadObj>
	*/
	Thread(Runnable *ptr) {
		_threadObj = ptr;
	}

	/*
		Function: Start

		Starts a thread
	*/
	DWORD start(char* name) {
		DWORD threadID;
		hThread = CreateThread(0, 0, threadProc, _threadObj, 0, &threadID);
		setThreadName( threadID, name);
		return threadID;
	}

	/* Give a thread a name. Thanks Microsoft. */ 
	void setThreadName( DWORD dwThreadID, LPCSTR szThreadName)
	{
		THREADNAME_INFO info;
		info.dwType = 0x1000;
		info.szName = "hello";
		info.dwThreadID = dwThreadID;
		info.dwFlags = 0;

		__try {
			RaiseException( 0x406D1388, 0, sizeof(info)/sizeof(DWORD), (DWORD*)&info );
		} __except(EXCEPTION_CONTINUE_EXECUTION) {
		}
	}

	/*
		Function: Stop

		Stops the running thread that was created
	*/
	BOOL stop() {
		if(hThread != NULL)
			return TerminateThread(hThread, 0);
		return FALSE;
	}
  
protected:
	/*
		Variable: hThread

		Handle to the thread when <Start> is called
	*/
	HANDLE hThread;

	/*
		Variable: _threadObj

		Pointer to an object which implements the Runnable interface
	*/
	Runnable *_threadObj;

	/*
		Function: threadProc

		Static c-function which can be used in the CreateThread function in windows.h
	*/
	static unsigned long __stdcall threadProc(void* ptr) {
		((Runnable*)ptr)->run();
		return 0;
	}   
};