socketthreadmanager.h

一个可订制ip packet的程序

/*
 *
 *
 *  Copyright (c) 2000 Barak Weichselbaum <[e-mail]>
 *  All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 *
 *
 * Contact info:
 * Site: http://www.komodia.com
 * Email: [e-mail]
 */

#if !defined(AFX_SOCKETTHREADMANAGER_H__74FE76EA_CB87_4F58_8F81_AA23A971A332__INCLUDED_)
#define AFX_SOCKETTHREADMANAGER_H__74FE76EA_CB87_4F58_8F81_AA23A971A332__INCLUDED_

#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000

#include "SpoofBase.h"

//##ModelId=3B43E6F400F0
class CSocketThreadManager : public CSpoofBase  
{
public:
	//Less socket in the system
	//##ModelId=3B43E6F400FB
	void DecreaseSocketCount(HWND hWindowHandle);

	//Get the window handle
	//##ModelId=3B43E6F40105
	HWND GetWindowHandle();

	//ctor and dtor
	//##ModelId=3B43E6F40106
	CSocketThreadManager(int iThreadCount,HINSTANCE hInstance);
	//##ModelId=3B43E6F40118
	virtual ~CSocketThreadManager();
private:
	//Get the socket array position by the window handle
	//##ModelId=3B43E6F4011A
	int GetIndexByHWND(HWND hHandle);
	
	//Get the freeiest thread
	//##ModelId=3B43E6F40123
	int GetMostAvailableThread();

	//Our thread function
	//##ModelId=3B43E6F4012C
	static DWORD WINAPI SocketThread(LPVOID lpParameter);

	//Spawn the threads
	//##ModelId=3B43E6F40137
	void SpawnThreads();

	//Our thread data
	//##ModelId=3B43E6F40186
	typedef struct _ThreadData
	{
		HWND      hWindowHandle;
		int       iSocketCount;
		HANDLE    hThreadHandle;
		DWORD     dwThreadID;
		HINSTANCE hInstance;
		HANDLE    hEvent;
	} ThreadData;

	//Our thread count
	//##ModelId=3B43E6F40140
	int m_iThreadCount;

	//Our windows struct
	//##ModelId=3B43E6F40155
	ThreadData* m_pThreadData;

	//Our instance
	//##ModelId=3B43E6F40168
	HINSTANCE m_hInstance;

	//Our critical section
	//##ModelId=3B43E6F40172
	CRITICAL_SECTION m_pCSection;
};

#endif // !defined(AFX_SOCKETTHREADMANAGER_H__74FE76EA_CB87_4F58_8F81_AA23A971A332__INCLUDED_)