// ZXPortMap.h: interface for the CZXPortMap class.
//
//////////////////////////////////////////////////////////////////////

#if !defined(AFX_ZXPORTMAP_H__A43DC211_66E7_4744_AFD9_AD91C1B36A9A__INCLUDED_)
#define AFX_ZXPORTMAP_H__A43DC211_66E7_4744_AFD9_AD91C1B36A9A__INCLUDED_

#include <WINSOCK2.H>

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

#define MAXBUFSIZE 8192
#define ADDRSIZE 32

extern BYTE bSerStat[128];

struct SOCKINFO
{
	SOCKET ClientSock;
	SOCKET ServerSock;
};
struct ADDRESS
{
	char szIP[ADDRSIZE];
	WORD wPort;
	SOCKET s;
};
template<typename T>
class STACK
{
#define MAXSTACK 1024*2
	
private:
	int top;
	T Data[MAXSTACK];
public:
	
	STACK()
	{
		top = -1;
	}
	
	bool IsEmpty()
	{
		return top < 0;
	}
	
	bool IsFull()
	{
		return top >= MAXSTACK;
	}
	
	bool Push(T data)
	{
		if(IsFull())
			return false;
		top++;
		Data[top] = data;
		return true;
	}
	
	T Pop()
	{
		return Data[top--];
	}
	
};/////////////////////stack end
//////////////////////////////////////////////////////////////////Transfer some Parameters
template<typename X, typename Y>
class TransferParam
{
public:
	X GlobalData;
	STACK<Y> LocalData;
public:
	TransferParam();
	virtual ~TransferParam();
	bool Push(Y data);
	Y Pop();
	
};

template<typename X, typename Y>
TransferParam<X, Y>::TransferParam()
{
	memset(this, 0, sizeof(TransferParam));
}

template<typename X, typename Y>
TransferParam<X, Y>::~TransferParam()
{
	
}

template<typename X, typename Y>
bool TransferParam<X, Y>::Push(Y data)
{
	return LocalData.Push(data);
}

template<typename X, typename Y>
Y TransferParam<X, Y>::Pop()
{
	return LocalData.Pop(data);
}


class CZXPortMap  
{
public:
	DWORD nCtrlPort, nServerPort;
	static SOCKET SocketNum[1024];
	static int nSocket;
	static int nTimes;
	static 	HANDLE hThreadPort[1024];
	static bool bIsRunning;
	static int nThreadNum;
	CZXPortMap();
	virtual ~CZXPortMap();
	static int DataSend(SOCKET s, char *DataBuf, int DataLen);//将DataBuf中的DataLen个字节发到s去
	static DWORD WINAPI TransmitData(LPVOID lParam);
	static DWORD WINAPI PortTransfer_1(LPVOID lParam);
	static DWORD WINAPI PortTransfer_2(LPVOID lParam);
	static DWORD WINAPI PortTransfer_3(LPVOID lParam);
	static SOCKET ConnectHost(DWORD dwIP, WORD wPort);
	static SOCKET ConnectHost(char *szIP, WORD wPort);
	static SOCKET CreateSocket(DWORD dwIP, WORD wPort);
	static SOCKET CreateTmpSocket(WORD *wPort);
	BOOL InitSocket();
	BOOL PortTransfer_1(WORD ListenPort, char *szIP, WORD wPort);;
	BOOL PortTransfer_2(char *szCtrlIP, WORD wCtrlPort, char *szIP, WORD wPort);
	BOOL PortTransfer_3(WORD wCtrlPort, WORD wServerPort);
	void Stop();
};


#endif // !defined(AFX_ZXPORTMAP_H__A43DC211_66E7_4744_AFD9_AD91C1B36A9A__INCLUDED_)