MFC를 이용한 범용 서버/데이타 소켓 작성

// Server socket

#define UM_ACCEPT (WM_USER+10)

class CServerSocket : public CSocket {
public:
void Init(CWnd *pWnd, int nPortNum);

protected:
CWnd *m_pWnd;
virtual void OnAccept(int nErrorCode);
};

void CServerSocket::Init(CWnd *pWnd, int nPortNum) {
m_pWnd = pWnd;
Create(nPortNum);
Listen();
}

void CServerSocket::OnAccept(int nErrorCode) {
m_pWnd->SendMessage(UM_ACCEPT);
CSocket::OnAccept(nErrorCode);
}

// Data socket
#define UM_DATARECEIVE (WM_USER+11)
class CData; // Forward declaration

class CDataSocket : public CSocket {
public:
CDataSocket();
virtual ~CDataSocket();

void Init(CWnd *pWnd);
void Send(CData *pData);
void Receive(CData *pData);
void operator << (CData &data); void operator >> (CData &data);
void Flush();

CArchive *m_pArchiveIn;
CArchive *m_pArchiveOut;

protected:
CWnd *m_pWnd;
CSocketFile *m_pFile;

virtual void OnReceive(int nErrorCode);
virtual void OnClose(int nErrorCode);
};

void CDataSocket::Init(CWnd *pWnd) {
m_pWnd = pWnd;
m_pFile = new CSocketFile(this);
m_pArchiveIn = new CArchive(m_pFile, CArchive::load);
m_pArchiveOut = new CArchive(m_pFile, CArchive::store);
}

void CDataSocket::Receive(CData *pData) {
if (m_pArchiveIn != NULL)
pData->Serialize(*m_pArchiveIn);
}

void CDataSocket::operator >> (CData &data) {
  Receive(&data);
}

void CDataSocket::Send(CData *pData) {
if (m_pArchiveOut != NULL)
pData->Serialize(*m_pArchiveOut);
}

void CDataSocket::Flush() {
if (m_pArchiveOut != NULL)
m_pArchiveOut->Flush();
}

void CDataSocket::operator << (CData &data) {
  Send(&data);
Flush();
}

void CDataSocket::OnReceive(int nErrorCode) {
  m_pWnd->SendMessage(UM_DATARECEIVE);
  CSocket::OnReceive(nErrorCode);
}

void CDataSocket::OnClose(int nErrorCode) {
  if (m_pFile != NULL) {
        delete m_pFile;
      m_pFile = NULL;
  }
    if (m_pArchiveIn != NULL) {
        m_pArchiveIn->Abort();
      delete m_pArchiveIn;
      m_pArchiveIn = NULL;
  }
    if (m_pArchiveOut != NULL) {
        m_pArchiveOut->Abort();
      delete m_pArchiveOut;
      m_pArchiveOut = NULL;
  }
    CSocket::OnClose(nErrorCode);
}

// CData

class CData : public CObject {
public:
  CData();
  virtual ~CData();
  virtual void Serialize(CArchive &ar);

  CString m_strData; // Real data from here
};

void CData::Serialize(CArchive &ar) {
    if (ar.IsStoring()) {
        ar << m_strData;
  }
    else {
        ar >> m_strData;
  }
}

 

[출처] MFC를 이용한 범용 서버/데이타 소켓 작성|작성자 엘리트