• MFCSocketServerSampledlg.cpp源文件


    // MFCSocketServerSample.cpp : 定义应用程序的类行为。
    //
    
    #include "stdafx.h"
    #include "MFCSocketServerSample.h"
    #include "MFCSocketServerSampleDlg.h"
    
    #ifdef _DEBUG
    #define new DEBUG_NEW
    #endif
    
    
    // CMFCSocketServerSampleApp
    
    BEGIN_MESSAGE_MAP(CMFCSocketServerSampleApp, CWinApp)
        ON_COMMAND(ID_HELP, &CWinApp::OnHelp)
    END_MESSAGE_MAP()
    
    
    // CMFCSocketServerSampleApp 构造
    
    CMFCSocketServerSampleApp::CMFCSocketServerSampleApp()
    {
        // 支持重新启动管理器
        m_dwRestartManagerSupportFlags = AFX_RESTART_MANAGER_SUPPORT_RESTART;
    
        // TODO: 在此处添加构造代码,
        // 将所有重要的初始化放置在 InitInstance 中
    }
    
    
    // 唯一的一个 CMFCSocketServerSampleApp 对象
    
    CMFCSocketServerSampleApp theApp;
    
    
    // CMFCSocketServerSampleApp 初始化
    
    BOOL CMFCSocketServerSampleApp::InitInstance()
    {
        // 如果一个运行在 Windows XP 上的应用程序清单指定要
        // 使用 ComCtl32.dll 版本 6 或更高版本来启用可视化方式,
        //则需要 InitCommonControlsEx()。  否则,将无法创建窗口。
        INITCOMMONCONTROLSEX InitCtrls;
        InitCtrls.dwSize = sizeof(InitCtrls);
        // 将它设置为包括所有要在应用程序中使用的
        // 公共控件类。
        InitCtrls.dwICC = ICC_WIN95_CLASSES;
        InitCommonControlsEx(&InitCtrls);
    
        CWinApp::InitInstance();
    
        if (!AfxSocketInit()) //new added,socket初始化---------------------------
        {
            //AfxMessageBox(IDP_SOCKETS_INIT_FAILED);
            AfxMessageBox("socket initiative failed");
            return FALSE;
        }
    
        AfxEnableControlContainer();
    
        // 创建 shell 管理器,以防对话框包含
        // 任何 shell 树视图控件或 shell 列表视图控件。
        CShellManager *pShellManager = new CShellManager;
    
        // 激活“Windows Native”视觉管理器,以便在 MFC 控件中启用主题
        CMFCVisualManager::SetDefaultManager(RUNTIME_CLASS(CMFCVisualManagerWindows));
    
        // 标准初始化
        // 如果未使用这些功能并希望减小
        // 最终可执行文件的大小,则应移除下列
        // 不需要的特定初始化例程
        // 更改用于存储设置的注册表项
        // TODO: 应适当修改该字符串,
        // 例如修改为公司或组织名
        SetRegistryKey(_T("应用程序向导生成的本地应用程序"));
    
        CMFCSocketServerSampleDlg dlg;
        m_pMainWnd = &dlg;
        INT_PTR nResponse = dlg.DoModal();
        if (nResponse == IDOK)
        {
            // TODO: 在此放置处理何时用
            //  “确定”来关闭对话框的代码
        }
        else if (nResponse == IDCANCEL)
        {
            // TODO: 在此放置处理何时用
            //  “取消”来关闭对话框的代码
        }
        else if (nResponse == -1)
        {
            TRACE(traceAppMsg, 0, "警告: 对话框创建失败,应用程序将意外终止。
    ");
            TRACE(traceAppMsg, 0, "警告: 如果您在对话框上使用 MFC 控件,则无法 #define _AFX_NO_MFC_CONTROLS_IN_DIALOGS。
    ");
        }
    
        // 删除上面创建的 shell 管理器。
        if (pShellManager != NULL)
        {
            delete pShellManager;
        }
    
        // 由于对话框已关闭,所以将返回 FALSE 以便退出应用程序,
        //  而不是启动应用程序的消息泵。
        return FALSE;
    }
    // MFCSocketServerSampleDlg.cpp : 实现文件
    //
    
    #include "stdafx.h"
    #include "MFCSocketServerSample.h"
    #include "MFCSocketServerSampleDlg.h"
    #include "afxdialogex.h"
    
    #ifdef _DEBUG
    #define new DEBUG_NEW
    #endif
    
    
    // 用于应用程序“关于”菜单项的 CAboutDlg 对话框
    
    class CAboutDlg : public CDialogEx
    {
    public:
        CAboutDlg();
    
    // 对话框数据
    #ifdef AFX_DESIGN_TIME
        enum { IDD = IDD_ABOUTBOX };
    #endif
    
        protected:
        virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV 支持
    
    // 实现
    protected:
        DECLARE_MESSAGE_MAP()
    };
    
    CAboutDlg::CAboutDlg() : CDialogEx(IDD_ABOUTBOX)
    {
    }
    
    void CAboutDlg::DoDataExchange(CDataExchange* pDX)
    {
        CDialogEx::DoDataExchange(pDX);
    }
    
    BEGIN_MESSAGE_MAP(CAboutDlg, CDialogEx)
    END_MESSAGE_MAP()
    
    
    // CMFCSocketServerSampleDlg 对话框
    
    
    
    CMFCSocketServerSampleDlg::CMFCSocketServerSampleDlg(CWnd* pParent /*=NULL*/)
        : CDialogEx(IDD_MFCSOCKETSERVERSAMPLE_DIALOG, pParent)
        , m_Port(_T("6666"))
        , m_Log(_T(""))
        , m_editSend(_T(""))
        , m_editRecv(_T(""))
    {
        m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
    }
    
    void CMFCSocketServerSampleDlg::DoDataExchange(CDataExchange* pDX)
    {
        CDialogEx::DoDataExchange(pDX);
        //  DDX_Control(pDX, IDC_EDIT_PORT, m_Port);
        DDX_Text(pDX, IDC_EDIT_PORT, m_Port);
        DDX_Text(pDX, IDC_EDIT_LOG, m_Log);
        DDX_Text(pDX, IDC_EDIT_SEND, m_editSend);
        DDX_Control(pDX, IDC_IPADDRESS1, m_IP);
        DDX_Text(pDX, IDC_EDIT_RECEIVE, m_editRecv);
    }
    
    BEGIN_MESSAGE_MAP(CMFCSocketServerSampleDlg, CDialogEx)
        ON_WM_SYSCOMMAND()
        ON_WM_PAINT()
        ON_WM_QUERYDRAGICON()
        ON_BN_CLICKED(IDC_BUTTON_Listen, &CMFCSocketServerSampleDlg::OnBnClickedButtonListen)
        ON_BN_CLICKED(IDC_BUTTON_SEND, &CMFCSocketServerSampleDlg::OnBnClickedButtonSend)
        ON_BN_CLICKED(IDOK, &CMFCSocketServerSampleDlg::OnBnClickedOk)
        ON_BN_CLICKED(IDCANCEL, &CMFCSocketServerSampleDlg::OnBnClickedCancel)
        ON_MESSAGE(WM_SOCKET_LOG,OnLogMsg)
        ON_MESSAGE(WM_SOCKET_RECEIVE,OnRecvMsg)
    END_MESSAGE_MAP()
    
    
    // CMFCSocketServerSampleDlg 消息处理程序
    
    BOOL CMFCSocketServerSampleDlg::OnInitDialog()
    {
        CDialogEx::OnInitDialog();
    
        // 将“关于...”菜单项添加到系统菜单中。
    
        // IDM_ABOUTBOX 必须在系统命令范围内。
        ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
        ASSERT(IDM_ABOUTBOX < 0xF000);
    
        CMenu* pSysMenu = GetSystemMenu(FALSE);
        if (pSysMenu != NULL)
        {
            BOOL bNameValid;
            CString strAboutMenu;
            bNameValid = strAboutMenu.LoadString(IDS_ABOUTBOX);
            ASSERT(bNameValid);
            if (!strAboutMenu.IsEmpty())
            {
                pSysMenu->AppendMenu(MF_SEPARATOR);
                pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
            }
        }
    
        // 设置此对话框的图标。  当应用程序主窗口不是对话框时,框架将自动
        //  执行此操作
        SetIcon(m_hIcon, TRUE);            // 设置大图标
        SetIcon(m_hIcon, FALSE);        // 设置小图标
    
        // TODO: 在此添加额外的初始化代码
        m_IP.SetWindowText("127.0.0.1");
    //    m_Port = "12356";
    //    UpdateData(FALSE);
        //m_Port.SetString("111");
        
    
        return TRUE;  // 除非将焦点设置到控件,否则返回 TRUE
    }
    
    void CMFCSocketServerSampleDlg::OnSysCommand(UINT nID, LPARAM lParam)
    {
        if ((nID & 0xFFF0) == IDM_ABOUTBOX)
        {
            CAboutDlg dlgAbout;
            dlgAbout.DoModal();
        }
        else
        {
            CDialogEx::OnSysCommand(nID, lParam);
        }
    }
    
    // 如果向对话框添加最小化按钮,则需要下面的代码
    //  来绘制该图标。  对于使用文档/视图模型的 MFC 应用程序,
    //  这将由框架自动完成。
    
    void CMFCSocketServerSampleDlg::OnPaint()
    {
        if (IsIconic())
        {
            CPaintDC dc(this); // 用于绘制的设备上下文
    
            SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);
    
            // 使图标在工作区矩形中居中
            int cxIcon = GetSystemMetrics(SM_CXICON);
            int cyIcon = GetSystemMetrics(SM_CYICON);
            CRect rect;
            GetClientRect(&rect);
            int x = (rect.Width() - cxIcon + 1) / 2;
            int y = (rect.Height() - cyIcon + 1) / 2;
    
            // 绘制图标
            dc.DrawIcon(x, y, m_hIcon);
        }
        else
        {
            CDialogEx::OnPaint();
        }
    }
    
    //当用户拖动最小化窗口时系统调用此函数取得光标
    //显示。
    HCURSOR CMFCSocketServerSampleDlg::OnQueryDragIcon()
    {
        return static_cast<HCURSOR>(m_hIcon);
    }
    
    
    
    void CMFCSocketServerSampleDlg::OnBnClickedButtonListen()
    {
        // TODO: 在此添加控件通知处理程序代码
        
        pSocketServer = new CSocketServer();//fgh
        UpdateData(TRUE);
        pSocketServer->m_hMsgWnd = this->GetSafeHwnd();//获取句柄
        if (pSocketServer->Create(atoi(m_Port),SOCK_STREAM,FD_ACCEPT|FD_CLOSE))
        //if (pSocketServer->Create(88888, SOCK_STREAM, FD_ACCEPT | FD_CLOSE))
            WriteLog("创建服务器SOCKET成功");
        else
            WriteLog("创建服务器SOCKET失败");
        if(pSocketServer->Listen())
            WriteLog("监听服务器成功");
        else
            WriteLog("监听服务器失败");
    
    }
    
    
    void CMFCSocketServerSampleDlg::WriteLog(CString log)
    {
        
        m_Log += log;
        m_Log += "
    ";
        UpdateData(FALSE);
    }
    
    
    void CMFCSocketServerSampleDlg::OnBnClickedButtonSend()
    {
        // TODO: 在此添加控件通知处理程序代码
        if (pSocketServer == NULL)
        {
            WriteLog("Socket没有初始化");
            return;
        }
        if (pSocketServer->m_clientList.GetCount()<= 0)
        {
            WriteLog("没有客户端链接");
            return;
        }
        UpdateData(TRUE);
        int iLen = m_editSend.GetLength();
        BYTE* s = new BYTE[iLen];
        memset(s,0x00,iLen);
        memcpy(s,(LPCTSTR)m_editSend,iLen);
        int iCount = pSocketServer->m_clientList.GetCount();
        for (int i = 0; i < iCount; i++)
        {
            POSITION pos = pSocketServer->m_clientList.FindIndex(i);
            CSocketClient* pClient = (CSocketClient*)pSocketServer->m_clientList.GetAt(pos);
            if (pClient != NULL)
                pClient->Send((unsigned char*)s,iLen);
        }
        WriteLog("发送数据中。。。");
    
    
    }
    
    
    void CMFCSocketServerSampleDlg::OnBnClickedOk()
    {
        // TODO: 在此添加控件通知处理程序代码
        //CDialogEx::OnOK();
    }
    
    
    void CMFCSocketServerSampleDlg::OnBnClickedCancel()
    {
        // TODO: 在此添加控件通知处理程序代码
        CDialogEx::OnCancel();
    }
    
    
    HRESULT CMFCSocketServerSampleDlg::OnRecvMsg(WPARAM dwEvent, LPARAM dwLen)
    {
        //return E_NOTIMPL;
        if (!dwLen)
            return 0;
        BYTE* temp = new BYTE[dwLen + 1];
        memset(temp,0x00,dwLen+1);
        memcpy(temp,(const void*)dwEvent,dwLen);
        CString log,log1;
        log1.Format("%s", (LPCTSTR)temp);
        //if (log1 == "xzcvb
    ")
        if(    !log1.Compare("xzcvb"))  //字符串比较,compare相同就返回0,所以需要加,非的符号
            MessageBox("you are same");
        else
            MessageBox("different +"+log1);
        log.Format("接收到客户端数据=%s
    ",(LPCTSTR)temp);
        if (m_editRecv.GetLength() > 50000)
            m_editRecv = "";
        m_editRecv += log;
        UpdateData(FALSE);
        return 0;
    }
    
    
    HRESULT CMFCSocketServerSampleDlg::OnLogMsg(WPARAM dwEvent, LPARAM dwLen)
    {
        //return E_NOTIMPL;
        if (!dwLen)
            return 0;
        BYTE* temp = new BYTE[dwLen+1];
        memset(temp,0x00,dwLen+1);
        memcpy(temp,(const void*)dwEvent,dwLen);
        WriteLog((LPCTSTR)temp);
        return 0;
    
    }

    SocketClient.cpp

    #include "stdafx.h"
    #include "SocketClient.h"
    
    
    CSocketClient::CSocketClient()
    {
        m_nLength = 0;
        memset(m_szReceBuf,0,sizeof(m_szReceBuf));
        memset(m_szSendBuf,0,sizeof(m_szSendBuf));
        m_bConnect = false;
        m_hWnd = NULL;
        m_strHost.Empty();
        m_strIP.Empty();
    }
    
    
    CSocketClient::~CSocketClient()
    {
        if (m_hSocket != INVALID_SOCKET)
            Close();
    }
    
    
    void CSocketClient::OnSend(int nErrorCode)
    {
        int nSendBytes = Send(m_szSendBuf,strlen(m_szSendBuf),0);
        char* pLog = new char[200];
        sprintf(pLog,"客户端发送%d个数据",nSendBytes);
        if (m_hWnd != NULL)
        {
            ::SendMessage(m_hWnd,WM_SOCKET_LOG,(WPARAM)pLog,strlen(pLog));
            memset(m_szSendBuf,0,sizeof(m_szSendBuf));//把字符串末的所用用0替代
            AsyncSelect(FD_READ|FD_CLOSE);
    
        }
    }
    
    
    void CSocketClient::OnReceive(int nErrorCode)
    {
        m_nLength = Receive((void*)m_szReceBuf,MAXSOCKBUF,0);
        m_szReceBuf[m_nLength] = 0;
        char* recvBuf = new char[MAXSOCKBUF];
        sprintf(recvBuf,(const char*)m_szReceBuf,m_nLength);
        if (m_hMsgWnd != NULL)
            ::SendMessage(m_hMsgWnd,WM_SOCKET_RECEIVE,(WPARAM)recvBuf,strlen(recvBuf));
        CAsyncSocket::OnReceive(nErrorCode);
    }
    
    
    void CSocketClient::OnConnect(int nErrorCode)
    {
        char* pLog = new char[200];
        if (nErrorCode == 0)
        {
            sprintf(pLog, "连接服务器成功");
            m_bConnect = TRUE;
        }
        else
            sprintf(pLog,"连接服务器失败,错误代码%d",nErrorCode);
        if (m_hMsgWnd != NULL)
            ::SendMessage(m_hMsgWnd,WM_SOCKET_LOG,(WPARAM)pLog,strlen(pLog));
    }
    
    
    void CSocketClient::Init()
    {
        memset(m_szReceBuf,0,sizeof(m_szReceBuf));
        m_nLength = MAXSOCKBUF;
    }

    SocketServer.cpp

    #include "stdafx.h"
    #include "SocketServer.h"
    //#include "SocketClient.h"
    
    
    CSocketServer::CSocketServer()
    {
    }
    
    
    CSocketServer::~CSocketServer()
    {
        while (!m_clientList.IsEmpty())
        {
            CSocketClient* client = (CSocketClient*)m_clientList.RemoveHead();
            client->Close();
            //error C2039: “Close”: 不是“CSocketClient”的成员。CSocketClient类没有继承CAsyncSocket
            delete client;
        }
        m_clientList.RemoveAll();
        if (m_hSocket != INVALID_SOCKET)
            Close();
    }
    
    
    //void CSocketServer::OnAccept(int mErrprCpde)
    //{
    //}
    
    
    void CSocketServer::OnAccept(int nErrorCode)
    {
        char* pLog = new char[200];
        //CString* pLog = NULL;
        if (nErrorCode)
        {
            if (nErrorCode == WSAENETDOWN)
                sprintf(pLog, "网络故障!");
            else
                sprintf(pLog,"FD_ACCEPT位置错误");
            return;
        }
        else
        {
            sockaddr address;
            CString IPaddr;
            UINT port;
            int address_len;
            address_len = sizeof(address);
            CSocketClient* pSocket = new CSocketClient();
            pSocket->m_hMsgWnd = m_hMsgWnd;
            if (this->Accept(*pSocket, &address, &address_len))
            {
                pSocket->AsyncSelect(FD_WRITE|FD_READ|FD_CLOSE);
                pSocket->GetPeerName(IPaddr,port);
                pSocket->m_strIP = IPaddr;
                m_clientList.AddTail(pSocket);
                sprintf(pLog, "接收客户端连接:IP=%s;端口=%d", IPaddr, port);
            }
            else
            {
                int Error = GetLastError();
                if (Error == WSAECONNREFUSED)
                    sprintf(pLog, "拒绝连接");
                else
                    wsprintf(pLog,"WSAAccept失败,错误代码:%d",Error);
                delete pSocket;
                return;
            }
    
            
    
        
        }
        if (m_hMsgWnd != NULL)
            ::SendMessage(m_hMsgWnd,WM_SOCKET_LOG,(WPARAM)pLog,strlen(pLog));
        CAsyncSocket::OnAccept(nErrorCode);
    }
    
    
    void CSocketServer::OnClose(int nErrorCode)
    {
        while (!m_clientList.IsEmpty())
        {
            CSocketClient* client = (CSocketClient*)m_clientList.RemoveHead();
            client->Close();
            delete client;
        }
        m_clientList.RemoveAll();
        if (m_hSocket != INVALID_SOCKET)
            Close();
        CAsyncSocket::OnClose(nErrorCode);
    }
    
    
    void CSocketServer::DeleteRemoteSocket(CSocketClient* pSock)
    {
        pSock->Close();
        POSITION pos = m_clientList.GetHeadPosition();
        POSITION temp;
        while (pos != NULL)
        {
            temp = pos;
            CSocketClient* client = (CSocketClient*)m_clientList.GetNext(pos);
            if (client == pSock)
            {
                m_clientList.RemoveAt(temp);
                client->Close();
                delete client;
                break;
            }
        }
        return;
    }

    stdafx.cpp

    // stdafx.cpp : 只包括标准包含文件的源文件
    // MFCSocketServerSample.pch 将作为预编译头
    // stdafx.obj 将包含预编译类型信息
    
    #include "stdafx.h"
  • 相关阅读:
    MVC设计模式
    二十三种设计模式
    描绘质量属性的六个常见属性场景。
    《架构漫谈》读后感
    软件架构师的工作过程
    《架构之美》阅读笔记06
    《架构之美》阅读笔记05
    《架构之法》阅读笔记04
    《架构之美》阅读笔记03
    《构架之美》阅读笔记02
  • 原文地址:https://www.cnblogs.com/txwtech/p/13688598.html
Copyright © 2020-2023  润新知