tcp.h

来自「网络泡泡被.net管理」· C头文件 代码 · 共 107 行

// TCP.h: interface for the TCP_Session class.
//
//////////////////////////////////////////////////////////////////////

#if !defined(AFX_TCP_H__BAB99BDE_FD81_46A4_B0A7_FAC6CE9FD2BD__INCLUDED_)
#define AFX_TCP_H__BAB99BDE_FD81_46A4_B0A7_FAC6CE9FD2BD__INCLUDED_

#include "net_lib.h"
#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000



class NET_LIB_API TCP_Session  : public NET_Session
{
	friend TCP_Socket;
public:
	virtual void Connect();
//	void Connect();
	virtual bool RunTransfersThread();
	TCP_Socket*		get_TCP_Socket()			{return (TCP_Socket*)m_ptr_net_socket;}
	SOCKET			get_SOCKET()				{return m_socket;}


	bool			PushSendPacket(NET_Packet* ptr_packet);
	NET_Packet*		PopSendPacket();

	int				recv(NET_Packet* pPacket);
	int				send(NET_Packet* pPacket, bool bIC=false);

	TCP_Session(uint32 uid, TCP_Socket* ptr_tcp_socket);
	virtual ~TCP_Session();

protected:
	SOCKET		m_socket;
//	TCP_Socket*	m_ptr_tcp_socket;


	std::list<NET_Packet*>			m_lst_send_packet;				//发送列表
	TYPE_CS							m_lock_send_packet;				//发送列表锁

};


class NET_LIB_API TCP_Socket : public NET_Socket  
{
public:
	TCP_Socket();
	virtual ~TCP_Socket();
public:
	void Ping(uint32 uid=0);
	bool RunTransfersThread();
	bool change_session_uid(uint32 uid, uint32 uid_to);
	void CheckConnect(time_t time);
	uint16 getPort();
	uint32 getIP();
    bool CanRead();	
	bool CanWrite();
	void CloseSession(uint32 uid);
	int GetSessionCount();
	void ThreadInit();
	void ThreadEnd();
	//添加发送数据包，因此要发送者分配数据部对象，发送后将由线程自动删除
	virtual bool		PushSendPacket(uint32 uid_sendto, NET_Packet* ptr_packet);

	virtual bool		PushRecvPacket(NET_Packet* ptr_packet, time_t time);
	virtual NET_Packet*	PopRecvPacket();


	TCP_Session* CreateSession(sockaddr_in* ptr_addr);
	TCP_Session* CreateUniqueSession(uint32 ip, uint16 port, uint32 uid_unique, const char* szAddr=NULL);
	TCP_Session* CreateSession(uint32 ip, uint16 port, const char* szAddr=NULL);
	TCP_Session* get_TCP_Session(uint32 uid);

	NET_Session* get_NET_Session(uint32 uid);
	

	virtual void Shutdown();
	//id在100以内为内部使用
	bool Create(uint16 port, uint32 uid_limit_min=100, uint32 uid_limit_max=100000);

	SOCKET		get_SOCKET()	{return m_socket;}

//	void Unlock();
//	bool Lock(uint32 uid);

	bool LockSessionMap();
	void UnlockSessionMap();

	void CleanupSessionMap();
protected:

	
//	SOCKET			m_socket;


	std::map<uint32, TCP_Session*>	m_map_session;
	TYPE_CS							m_lock_map_session;


};



#endif // !defined(AFX_TCP_H__BAB99BDE_FD81_46A4_B0A7_FAC6CE9FD2BD__INCLUDED_)