// コンストラクタ
CServerASock::CServerASock() : m_dlgP(NULL)
{
}
// コンストラクタオーバーロード
CServerASock::CServerASock(CAsyncServerDlg *dlgP) : m_dlgP(dlgP)
{
}
// Createオーバーライド
BOOL CServerASock::Create(CAsyncServerDlg *dlgP, UINT nSocketPort)
{
    m_dlgP = dlgP;
    return CAsyncSocket::Create(nSocketPort);
}
// Accept通知
void CServerASock::OnAccept(int nErrorCode)
{
    if (m_dlgP) m_dlgP->OnAccept(nErrorCode);
    CAsyncSocket::OnAccept(nErrorCode);
}
// Receive通知
void CServerASock::OnReceive(int nErrorCode)
{
    if (m_dlgP) m_dlgP->OnReceive(nErrorCode);
    CAsyncSocket::OnReceive(nErrorCode);
}
// Send通知
void CServerASock::OnSend(int nErrorCode)
{
    if (m_dlgP) m_dlgP->OnSend(nErrorCode);
    CAsyncSocket::OnSend(nErrorCode);
}

// "待ち受け"ボタン押下
void CAsyncServerDlg::OnBnClickedBtnListen()
{
    unsigned int    port = 0;
    int             err = 0;
    
    UpdateData();
    m_lsnSock.Close();
    // ポート取得
    if (!err) if (_stscanf_s(m_xvEditPort, _T("%d"), &port) != 1) err = 1;
    // ソケット作成
    if (!err) if (!m_lsnSock.Create(this, port)) err = 1;
    // 接続要求を待機
    if (!err) if (!m_lsnSock.Listen()) err = 1;
    // エラー表示
    if (err) DispErr(m_lsnSock.GetLastError());

    return;
}

// エラー表示
void CAsyncServerDlg::DispErr(int code)
{
    LPVOID        lpMsgBuf = NULL;
    
    ::FormatMessage(
        FORMAT_MESSAGE_ALLOCATE_BUFFER |
        FORMAT_MESSAGE_FROM_SYSTEM |
        FORMAT_MESSAGE_IGNORE_INSERTS,
        NULL, code,
        MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
        (LPTSTR)&lpMsgBuf, 0, NULL);

    m_xvEditLog = _T("Error : ");
    m_xvEditLog += static_cast<LPTSTR>(lpMsgBuf);
    UpdateData(FALSE);
    LocalFree(lpMsgBuf);

    return;
}

// Accept通知
void CAsyncServerDlg::OnAccept(int nErrorCode)
{
    int            err = 0;
    
    if (nErrorCode) err = 1;
    // コネクション用ソケット作成
    if (!err)
    {
        if (m_conSockP)
        {
            m_conSockP->Close();
            delete m_conSockP;
            m_conSockP = NULL;
        }
        m_conSockP = new CServerASock(this);
        if (!m_conSockP) err = 1;
    }
    // 接続受け入れ
    if (!err)
    {
        if (!m_lsnSock.Accept(*m_conSockP)) err = 1;
        if (err) if (m_lsnSock.GetLastError() == WSAEWOULDBLOCK) err = 0;
    }
    if (err)
    {
        DispErr(nErrorCode ? nErrorCode : m_lsnSock.GetLastError());
        m_lsnSock.Close();
    }
    UpdateData(FALSE);
}

// Receive通知
void CAsyncServerDlg::OnReceive(int nErrorCode)
{
    int          recv;
    LPSTR        byteP = NULL;
    int          err = 0;
    
    if (nErrorCode) err = 1;
    if (!m_conSockP) err = 1;
    // 受信(20バイト固定)
    if (!err)
    {
        byteP = m_recvStr.GetBuffer(21);
        if (m_recvSum >= 20) m_recvSum = 0;
        // Receive()は一回のみ
        if (m_recvSum < 20)
        {
            recv = m_conSockP->Receive(byteP +m_recvSum, 20 -m_recvSum);
            if (recv == SOCKET_ERROR || recv == 0) err = 1;
            if (!err) m_recvSum += recv;
            // WSAEWOULDBLOCKはエラーとしない。次のOnReceiveまで制御を戻す。
            if (err) if (m_conSockP->GetLastError() == WSAEWOULDBLOCK) err = 0;
        }
        byteP[20] = '\0';
        m_recvStr.ReleaseBuffer();
    }
    if (!err)
    {
        if (m_recvSum >= 20)
        {
            m_recvSum = 0;
            m_xvEditLog += _T("Recv : ");
            m_xvEditLog += m_recvStr +_T("\r\n");
            UpdateData(FALSE);
            Reply();    // 指定サイズを受信したので、返信
        }
    }
    if (err)
    {
        if (m_conSockP)
        {
            DispErr(nErrorCode ? nErrorCode : m_conSockP->GetLastError());
            m_conSockP->Close();
            delete m_conSockP;
            m_conSockP = NULL;
        }
    }
    UpdateData(FALSE);
}

// Send通知
void CAsyncServerDlg::OnSend(int nErrorCode)
{
    int            err = 0;
    
    if (nErrorCode) err = 1;
    if (!m_conSockP) err = 1;
    if (!err) if (m_sendSum) Reply();    // 続きを送信
    if (err)
    {
        if (m_conSockP)
        {
            DispErr(nErrorCode ? nErrorCode : m_conSockP->GetLastError());
            m_conSockP->Close();
            delete m_conSockP;
            m_conSockP = NULL;
        }
    }
}

// 返信
void CAsyncServerDlg::Reply(void)
{
    int               send;
    LPCSTR            byteCP = NULL;
    int               err = 0;

    UpdateData();
    if (!m_conSockP) err = 1;
    // 送信(20バイト固定)
    if (!err)
    {
        m_sendStr = _T("Welcome!");
        while (m_sendStr.GetLength() < 20) m_sendStr += _T(" ");
        m_sendStr = m_sendStr.Left(20);
        
        if (m_sendSum >= 20) m_sendSum = 0;
        // Send()は20バイト送るまで繰り返す
        while (m_sendSum < 20)
        {
            byteCP = static_cast<LPCSTR>(m_sendStr) +m_sendSum;
            send = m_conSockP->Send(byteCP, 20 -m_sendSum);
            if (send == SOCKET_ERROR) err = 1;
            // WSAEWOULDBLOCKはエラーとしない。次のOnSendまで制御を戻す。
            if (err) if (m_conSockP->GetLastError() == WSAEWOULDBLOCK) {err = 0; break;}
            if (err) break;

            m_sendSum += send;
        }
    }
    if (!err)
    {
        if (m_sendSum >= 20)
        {
            m_sendSum = 0;
            m_xvEditLog += _T("Send : ");
            m_xvEditLog += m_sendStr +_T("\r\n");
        }
    }
    // エラー表示
    if (err)
    {
        if (m_conSockP)
        {
            DispErr(nErrorCode ? nErrorCode : m_conSockP->GetLastError());
            m_conSockP->Close();
            delete m_conSockP;
            m_conSockP = NULL;
        }
    }
    UpdateData(FALSE);

    return;
}