1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
//////////////////////////////////////////////////////////////////////////
// Lan functions
#ifndef __lan_h__
#define __lan_h__
#define MAKE_PORT(x) (HIBYTE(x)|(LOBYTE(x)<<8))
#define MAX_INTERNAL_IP 32
#define PORT_NUMBER MAKE_PORT(34074)
//! Class for operating with LAN
class CLan
{
public:
//! constructor
CLan();
//! destructor
~CLan();
//! Helper function - returns status
int GetStatus() { return m_status; }
//! Helper function - returns mode
int GetMode() { return m_mode; }
//! Getting host addresses count
int GetHostAddrCount() { return m_hostAddrCount; }
//! Getting host addresses
in_addr GetHostAddress(int ind) { return m_hostAddr[ind]; }
//! Get current host address
in_addr GetCurHostAddress() { return m_curAddr; }
protected:
//! Lan status
enum enumStatus
{
LS_OK, //!< no problems
LS_WRONG_WINSOCK, //!< not found winsock of propper version
LS_CANT_CREATE_SOCKET, //!< can not create income socket
LS_CANT_GET_HOSTADDR, //!< can not find host address
LS_CANT_TURN_ON_BROADCAST, //!< can not allow broadcast messages for socket
LS_CANT_BIND_SOCKET, //!< can not bind socket to the address
LS_CANT_START_LISTEN, //!< can not start listen on TCP socket
LS_CANT_CREATE_THREADS, //!< can not create threads for listen and accept
};
//! Lan mode
enum enumMode
{
LM_OFF, //!< Winsock is turned off
LM_ON, //!< Winsock is on
LM_LISTEN, //!< Listening for incoming messages
};
//! Starts winsock
void Startup();
//! Stops winsock
void Shutdown();
//! Listen
void StartListen();
//! Stop Listen
void StopListen();
//! Set current host address
void SetCurHostAddress(in_addr addr);
//! Send packet
void SendPacket(in_addr addr, const u_char* mes, int len);
//! Send broadcast packet
void SendPacketBroadcast(const u_char* mes, int len);
//! Event - called when packet is received
virtual void OnRecvPacket(u_char* mes, int len, in_addr from) { };
//! Event - called when new incoming tcp connection is created (new thread is created)
virtual void OnInTCPConnection(u_long addr, SOCKET m_socket) { };
//! Event - called when new outgoing tcp connection is created )new thread is created)
virtual void OnOutTCPConnection(u_long addr, SOCKET m_socket, LPVOID lpParameter) {};
//! Creates new outgoing TCP connection
SOCKET CreateTCPConnection(u_long addr, LPVOID lpParameter);
private:
//! Launches Listen procedure when in new thread
static DWORD WINAPI ListenProc(LPVOID lpParameter);
//! Listnes for incoming messages
void Listen();
//! Listen thread handle
HANDLE m_hListenThread;
//! Structure passed to new TCP connection thread
struct TTCPConnect
{
CLan* m_lan;
u_long m_addr;
SOCKET m_socket;
LPVOID m_lpParameter;
};
//! Launches accept procedure for TCP connections in new thread
static DWORD WINAPI AcceptTCPProc(LPVOID lpParameter);
//! Accepts TCP connections
void AcceptTCP();
//! Accept TCP thread handle
HANDLE m_hAcceptTCPThread;
//! Semaphore for killing accept thread
CRITICAL_SECTION m_csAcceptTCPThread;
//! Called when new income TCP connection is created
static DWORD WINAPI OnInTCPConnectionProc(LPVOID lpParameter);
//! Called when new ougoing TCP connectio is created
static DWORD WINAPI OnOutTCPConnectionProc(LPVOID lpParameter);
//! Stores retrieved host addresses
in_addr m_hostAddr[MAX_INTERNAL_IP];
//! Current address count
int m_hostAddrCount;
//! Stores current host address
in_addr m_curAddr;
//! Socket for income messages
SOCKET m_income;
//! Socket for income files
SOCKET m_filesoc;
//! Current status
int m_status;
//! Current mode
int m_mode;
};
#endif //__lan_h__
|
