winSocket编程之TCP聊天小工具

///////////////////////////////////////////////// // TCPClient.cpp文件 #include "TCPClient.h" #include "resource.h" #include <string> // 定义网络事件通知消息 #define WM_SOCKET WM_USER + 1 #define MAX_BUF 1024 CMyApp theApp; BOOL CMyApp::InitInstance() { // 初始化Winsock库 WSADATA wsaData; WORD sockVersion = MAKEWORD(2, 0); ::WSAStartup(sockVersion, &wsaData); // 弹出主窗口对话框 CMainDialog dlg; m_pMainWnd = &dlg; dlg.DoModal(); // 释放Winsock库 ::WSACleanup(); return FALSE; } CMainDialog::CMainDialog(CWnd* pParentWnd):CDialog(IDD_MAINDIALOG, pParentWnd) { } BEGIN_MESSAGE_MAP(CMainDialog, CDialog) ON_BN_CLICKED(IDC_CONNECT, OnButtonConnect) ON_BN_CLICKED(IDC_SEND, OnButtonSend) ON_BN_CLICKED(IDC_CLEAR, OnButtonClear) ON_MESSAGE(WM_SOCKET, OnSocket) //ON_BN_CLICKED(IDC_SUREIP, &CMainDialog::OnBnClickedSureIP) ON_LBN_DBLCLK(IDC_LIST2, &CMainDialog::OnLbnDblclkList2) ON_BN_CLICKED(IDC_BUTTON1, &CMainDialog::OnBnClickedClearHaoyou) ON_BN_CLICKED(IDC_BUTTON2, &CMainDialog::OnBnClickedClearRecv) ON_LBN_SELCHANGE(IDC_LIST2, &CMainDialog::OnLbnSelchangeHaoYou) ON_BN_CLICKED(IDC_BUTTON3, &CMainDialog::OnBnClickedAdd) ON_BN_CLICKED(IDC_BUTTON4, &CMainDialog::OnBnClickedRepeatSelect) END_MESSAGE_MAP() BOOL CMainDialog::OnInitDialog() { CDialog::OnInitDialog(); // 设置图标 SetIcon(theApp.LoadIcon(IDI_MAIN), FALSE); // 创建状态栏，设置它的属性 m_bar.Create(WS_CHILD|WS_VISIBLE|SBS_SIZEGRIP, CRect(0, 0, 0, 0), this, 101); m_bar.SetBkColor(RGB(0xa6, 0xca, 0xf0)); // 背景色 int arWidth[] = { 200, -1 }; m_bar.SetParts(2, arWidth); // 分栏 m_bar.SetText("TCP聊天小软件v1.0！", 1, 0); // 第一个栏的文本 m_bar.SetText(" 空闲", 0, 0); // 第二个栏的文本 // 设置列表框控件到m_listInfo对象的关联 listBox.SubclassDlgItem(IDC_LIST2, this); // 初始化发送按钮和发送编辑框的状态 GetDlgItem(IDC_SEND)->EnableWindow(FALSE); GetDlgItem(IDC_TEXT)->EnableWindow(FALSE); // 初始化连接套节字 m_socket = INVALID_SOCKET; return TRUE; } void CMainDialog::OnCancel() { if(m_socket != INVALID_SOCKET) ::closesocket(m_socket); CDialog::OnCancel(); } void CMainDialog::OnButtonClear() { GetDlgItem(IDC_INFO)->SetWindowText(""); } void CMainDialog::OnButtonConnect() { if(m_socket == INVALID_SOCKET) // 连接服务器 { //初始化 for (int i = 0; i < 56; i++) { deIpss[i] = ""; } deipcount = 0; // 取得服务器地址 CString sAddr; GetDlgItem(IDC_ADDR)->GetWindowText(sAddr); if(sAddr.IsEmpty()) { MessageBox("请输入服务器地址！"); return; } // 取得端口号 CString sPort; GetDlgItem(IDC_PORT)->GetWindowText(sPort); int nPort = atoi(sPort); if(nPort < 1 || nPort > 65535) { MessageBox("端口号错误！"); return; } // 试图连接服务器 if(!Connect(sAddr, nPort)) { MessageBox("连接服务器出错！"); return; } // 设置用户界面 GetDlgItem(IDC_CONNECT)->SetWindowText("取消"); m_bar.SetText(" 正在连接……", 0, 0); } else // 断开服务器 { // 关闭套节字 ::closesocket(m_socket); m_socket = INVALID_SOCKET; listBox.ResetContent(); // 清零 GetDlgItem(IDC_SELECT_SURE)->SetWindowText(""); // 设置用户界面 GetDlgItem(IDC_CONNECT)->SetWindowText("连接服务器"); m_bar.SetText(" 空闲", 0, 0); GetDlgItem(IDC_ADDR)->EnableWindow(TRUE); GetDlgItem(IDC_PORT)->EnableWindow(TRUE); GetDlgItem(IDC_TEXT)->EnableWindow(FALSE); GetDlgItem(IDC_SEND)->EnableWindow(FALSE); } } long CMainDialog::OnSocket(WPARAM wParam, LPARAM lParam) { // 取得有事件发生的套节字句柄 SOCKET s = wParam; // 查看是否出错 if(WSAGETSELECTERROR(lParam)) { if(m_socket != SOCKET_ERROR) OnButtonConnect(); m_bar.SetText(" 连接出错！", 0, 0); return 0; } // 处理发生的事件 switch(WSAGETSELECTEVENT(lParam)) { case FD_CONNECT: // 套节字正确的连接到服务器 { // 设置用户界面 GetDlgItem(IDC_CONNECT)->SetWindowText("断开连接"); GetDlgItem(IDC_ADDR)->EnableWindow(FALSE); GetDlgItem(IDC_PORT)->EnableWindow(FALSE); GetDlgItem(IDC_TEXT)->EnableWindow(TRUE); GetDlgItem(IDC_SEND)->EnableWindow(TRUE); m_bar.SetText(" 已经连接到服务器", 0, 0); } break; case FD_READ: // 套接字接受到对方发送过来的数据包 { // GetDlgItem(IDC_RECV)->SetWindowText(""); // 从服务器接受数据 char szText[1024] = { 0 }; ::recv(s, szText, 1024, 0); std::string str = szText; //解析一下字符串 int index = -2; /////////////////////////////////////////////////// for (int i = 0; i < sizeof(szText); i++) { if (szText[i] == '^') { index = 0; //找到^ 作为数据消息代表 break; } } if (szText[0] == 'd') { index = 1; //找到delete 删除特定好友 } /*char tt[20]; wsprintf(tt, "%d", index); MessageBox(tt);*/ if (index == -2) { //如果没有找到^，也没有找到delete 只是传送的ip集合 std::string a = str.substr(0, str.find("\r\n", 0)); CString realtext; realtext.Format("%s", a.c_str()); bool isRongYu = false; for (int i = 0; i < listBox.GetCount(); i++) { CString tmpstr = ""; listBox.GetText(i, tmpstr); if (tmpstr == realtext) { isRongYu = true; break; } } if (!isRongYu) { listBox.AddString(CString(realtext)); } } else if(index == 0) { //消息 std::string a = str.substr(0, str.find('\r\n', 0)); // std::string b = str.substr(a.length() + 1, str.find('\r\n', 0) - str.find("^", 0)); CString realtext; realtext.Format("%s", a.c_str()); AddStringToRecvList(CString(realtext)+"\r\n"); } else if (index == 1) { //删除指定ip std::string b = str.substr(1, str.find('\r\n', 0) - 1); CString realtext; // MessageBox(realtext); realtext.Format("%s", b.c_str()); bool isfounndIp = false; int item; for (int i = 0; i < listBox.GetCount(); i++) { CString tmpstr = ""; listBox.GetText(i, tmpstr); if (tmpstr == realtext) { isfounndIp = true; item = i; break; } } if (isfounndIp) { // MessageBox("delete" + realtext); listBox.DeleteString(item); } } } break; case FD_CLOSE: OnButtonConnect(); break; } return 0; } void CMainDialog::OnButtonSend() { if(m_socket == INVALID_SOCKET) { return; } CString sText; CString tmIp = ""; GetDlgItem(IDC_TEXT)->GetWindowText(sText); tmIp = deIp + "^"; sText = tmIp + sText; //MessageBox(sText); // 添加一个“回车换行” // 注意，添加它并不是必须的，但是如果使用本软件作为客户端调试网络协议， // 比如SMTP、FTP等，就要添加它了。因为这些协议都要求使用“回车换行”作为一个命令的结束标记 sText += "\r\n"; // AddStringToSendList(sText); // 发送数据到服务器 //MessageBox(sText); //Sleep(50); if(::send(m_socket, sText, sText.GetLength(), 0) != -1) { AddStringToSendList(sText); GetDlgItem(IDC_TEXT)->SetWindowText(""); } } BOOL CMainDialog::Connect(LPCTSTR pszRemoteAddr, u_short nPort) { // 创建套节字 m_socket = ::socket(AF_INET, SOCK_STREAM, IPPROTO_TCP); if(m_socket == INVALID_SOCKET) { return FALSE; } // 设置socket为窗口通知消息类型 ::WSAAsyncSelect(m_socket, m_hWnd, WM_SOCKET, FD_CONNECT | FD_CLOSE | FD_WRITE | FD_READ); // 假定szAddr是IP地址 ULONG uAddr = ::inet_addr(pszRemoteAddr); if(uAddr == INADDR_NONE) { // 不是IP地址，就认为这是主机名称 // 从主机名取得IP地址 hostent* pHost = ::gethostbyname(pszRemoteAddr); if(pHost == NULL) { ::closesocket(m_socket); m_socket = INVALID_SOCKET; return FALSE; } // 得到以网络字节顺序排列的IP地址 uAddr = ((struct in_addr*)*(pHost->h_addr_list))->s_addr; } // 填写服务器地址信息 sockaddr_in remote; remote.sin_addr.S_un.S_addr = uAddr; remote.sin_family = AF_INET;