//头文件
#pragma once
#include "BaseSocket.h"
class CServerSocket :
public CBaseSocket
{
private:
BOOL m_bConnect;
BOOL m_bSend;
char m_recvBuff[100];
char m_sendBuff[100];
int m_nSendLen;
public:
CServerSocket();
~CServerSocket();
BOOL CreateServerSocket(int nPort=6000, int nListen=5);
void Listening();
void CloseServerSocket();
void RecvData(char *pBuff);
void SendData(char *pBuff, int nLen);
};
//cpp文件
#include "stdafx.h"
#include "ServerSocket.h"
UINT ListenThread(LPVOID lp)
{
CServerSocket *pServerSocket = (CServerSocket*)lp;
pServerSocket->Listening();
return 0;
}
CServerSocket::CServerSocket()
{
m_bConnect = FALSE;
m_bSend = FALSE;
m_nSendLen = 0;
}
CServerSocket::~CServerSocket()
{
}
BOOL CServerSocket::CreateServerSocket(int nPort, int nListen)
{
m_socket = socket(AF_INET, SOCK_STREAM, 0);
SOCKADDR_IN addrSrv;
addrSrv.sin_addr.S_un.S_addr = htonl(INADDR_ANY);
addrSrv.sin_family = AF_INET;
addrSrv.sin_port = htons(nPort);
bind(m_socket, (SOCKADDR*)&addrSrv, sizeof(SOCKADDR));
listen(m_socket, nListen);
m_bConnect = TRUE;
AfxBeginThread(ListenThread, this);
return 0;
}
void CServerSocket::Listening()
{
SOCKADDR_IN addrClient;
int len = sizeof(SOCKADDR);
while (m_bConnect)
{
SOCKET sockClient = accept(m_socket, (SOCKADDR*)&addrClient, &len);
recv(sockClient, m_recvBuff, 100, 0);
if (m_bSend)
{
send(sockClient, m_sendBuff, m_nSendLen, 0);
m_bSend = FALSE;
}
}
CloseSocket();
}
void CServerSocket::CloseServerSocket()
{
m_bConnect = FALSE;
}
void CServerSocket::RecvData(char *pBuff)
{
pBuff = m_recvBuff;
}
void CServerSocket::SendData(char *pBuff, int nLen)
{
memcpy(m_sendBuff, pBuff, nLen);
m_nSendLen = nLen;
m_bSend = TRUE;
}