tcp/ip c++ 即时通讯聊天室(二）客户端实现
客户端代码实现：（大佬请飘过）
一下三个文件分别对应上一期的三个头文件：
- Client_LAN 负责程序的初始化
- ClientCon_LAN 负责发送数据
- ClientDlg_LAN 负责使用MFC创建聊天室的对话框和处理事件，并显示发送来的数据
具体代码原理看注释，这里小编为了方便将注释用英文编写，部分采用中文，若有看不懂请在评论区留言或者有道翻译
写码不易，关注收藏一下呗，谢谢各位！！！
第三篇文章（服务端实现）
https://blog.csdn.net/qq_42662283/article/details/106493917
Client_LAN.cpp
// Client.cpp : Defines the class behaviors for the application.
#include "stdafx.h"
#include "Client_LAN.h"
#include "ClientDlg_LAN.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#endif
// CClientApp
BEGIN_MESSAGE_MAP(CClientApp, CWinApp)
ON_COMMAND(ID_HELP, &CWinApp::OnHelp)
END_MESSAGE_MAP()
// CClientApp construction
CClientApp::CClientApp()
{
// support Restart Manager
m_dwRestartManagerSupportFlags = AFX_RESTART_MANAGER_SUPPORT_RESTART;
// TODO: add construction code here,
// Place all significant initialization in InitInstance
}
// The one and only CClientApp object
CClientApp theApp;
// CClientApp initialization
BOOL CClientApp::InitInstance()
{
// InitCommonControlsEx() is required on Windows XP if an application
// manifest specifies use of ComCtl32.dll version 6 or later to enable
// visual styles. Otherwise, any window creation will fail.
INITCOMMONCONTROLSEX InitCtrls;
InitCtrls.dwSize = sizeof(InitCtrls);
// Set this to include all the common control classes you want to use
// in your application.
InitCtrls.dwICC = ICC_WIN95_CLASSES;
InitCommonControlsEx(&InitCtrls);
CWinApp::InitInstance();
AfxEnableControlContainer();
// Create the shell manager, in case the dialog contains
// any shell tree view or shell list view controls.
CShellManager *pShellManager = new CShellManager;
// Standard initialization
// If you are not using these features and wish to reduce the size
// of your final executable, you should remove from the following
// the specific initialization routines you do not need
// Change the registry key under which our settings are stored
// TODO: You should modify this string to be something appropriate
// such as the name of your company or organization
SetRegistryKey(_T("Local AppWizard-Generated Applications"));
CClientDlg dlg;
m_pMainWnd = &dlg;
INT_PTR nResponse = dlg.DoModal();
if (nResponse == IDOK)
{
// TODO: Place code here to handle when the dialog is
// dismissed with OK
}
else if (nResponse == IDCANCEL)
{
// TODO: Place code here to handle when the dialog is
// dismissed with Cancel
}
// Delete the shell manager created above.
if (pShellManager != NULL)
{
delete pShellManager;
}
// Since the dialog has been closed, return FALSE so that we exit the
// application, rather than start the application's message pump.
return FALSE;
}
ClientCon_LAN.cpp
#include "stdafx.h"
#include "ClientCon_LAN.h"
#include "ClientDlg_LAN.h"
#include "FileUtil.h"
#pragma comment(lib,"ws2_32.lib") //Winsock Library
ClientCon::ClientCon(CClientDlg *dlg)
{
m_pClient = dlg;
isFile = false;
isReady = false;
sClient = 1;
}
ClientCon::~ClientCon(void)
{
closesocket(sClient);
WSACleanup();
}
void ClientCon::StartConnect(string sAddress, int iPort, string sUsername)
{
const int REC_BUFERSIZE = 2048;
struct sockaddr_in server;
char *message , server_reply[REC_BUFERSIZE + 1];
int recv_size;
m_pUser = sUsername;
printf("\r\nInitialising Winsock...");
if (WSAStartup(MAKEWORD(2,2),&wsa) != 0)
{
printf("Failed. Error Code : %d",WSAGetLastError());
return;
}
printf("Initialised.\r\n");
//Create a socket(SOCK_STREAM: connection-oriented)
if((sClient = socket(AF_INET , SOCK_STREAM , 0 )) == INVALID_SOCKET)
{
printf("Could not create socket : %d" , WSAGetLastError());
}
printf("Socket created.\r\n");
server.sin_addr.s_addr = inet_addr(sAddress.c_str());
server.sin_family = AF_INET;
server.sin_port = htons( iPort );
//Connect to remote server
if (connect(sClient, (struct sockaddr *)&server , sizeof(server)) < 0)
{
puts("connect error");
return;
}
//Receive data from the server
while((recv_size = recv(sClient, server_reply , REC_BUFERSIZE, 0)) != SOCKET_ERROR)
{
//Add a NULL terminating character to make it a proper string before printing
server_reply[recv_size] = '\0';
string sTempMsg1 = string(server_reply);
string sTempMsg = string(server_reply) + "\r\n";
//判断是否包含文件.txt标志
if (!(isFile) && (sTempMsg1.find(".txt") != string::npos))
//if(isFile)
{
isFile = true;
file = fu.createFile(server_reply);//创建文件准备存放数据
if (file != NULL)//成功创建文件
{
bool isOver = false;
int i;
while (true)
{
/*fd_set recvfd;
struct timeval tv = { 3, 0 };
FD_ZERO(&recvfd);
FD_SET(sClient, &recvfd);
//检测有没有数据到达socket,有->接收数据，无->break
int result = select(0, &recvfd, NULL, NULL, &tv);*/
//接收数据
recv_size = recv(sClient, server_reply, 2048, 0);
server_reply[recv_size] = '\0';//添加文件结尾标志
Sleep(500);
//判断是否含有规定的文件结束标志
for (i = 0; i < (recv_size - 4); i++)
{
if (server_reply[i] == 'O' &&
server_reply[i + 1] == 'V' &&
server_reply[i + 2] == 'E' &&
server_reply[i + 3] == 'R' &&
server_reply[i + 4] == '!')
{
isOver = true;
break;
}
}
if (isOver)
{
recv_size = (i == recv_size - 4) ? recv_size : (i - 1);
recv_size = (recv_size < 0) ? 0 : recv_size;
fwrite(server_reply, 1, recv_size, file);
break;
}
else
{
//循环一直写入文件
fwrite(server_reply, 1, recv_size, file);
}
}
fclose(file);
isFile = false;
}
}
else
{
if (sTempMsg1 == "READY!")
{
isReady = true;
}
else if(sTempMsg1.length() != 0)//控制信息不打印
{
m_pClient->ShowServerInfo(sTempMsg);
}
}
}
}
void ClientCon::SendData(string sMessage, bool showName)
{
isReady = false;
if (sMessage.length() > 0)
{
string sTemp = m_pUser + ">>" + sMessage + "\r\n";
if (send(sClient, sTemp.c_str(), sTemp.size(), 0) < 0)
{
puts("Send failed");
return;
}
else
{
sTemp = "您（" + m_pUser + ")>>" + sMessage + "\r\n";
//显示自己发送的消息
m_pClient->ShowServerInfo(sTemp);
}
}
}
void ClientCon::SendFile(FILE* file, CString filename)
{
FileUtil fu;
CString a = CString(".txt");
filename.Append(a);//加上后缀
char *data = fu.CString2Char(filename);
isReady = false;
send(sClient, data, filename.GetLength(), 0);//先发送文件名，让接收端做好接受准备
while (!isReady);
isReady = false;
Sleep(100);
const int SEND_BUFSIZE = 512;
char sendData[SEND_BUFSIZE + 1];
char recData[20] = {'0'};
int nCount;//nCount：实际读取的字符数量
while ((nCount = fread(sendData, 1, SEND_BUFSIZE, file)) > 0)//每次读取512个字符一起发送
{
//sendData[nCount] = '\0';
send(sClient, sendData, nCount, 0);
while (!isReady);//阻塞等待接收方发送确认
isReady = false;
}
Sleep(500);
send(sClient, "OVER!", 5, 0);//发送文件发送完毕标志
fclose(file);
}
ClientDlg_LAN.cpp
// ClientDlg.cpp : implementation file
//
#include "stdafx.h"
#include "ClientDlg_LAN.h"
#include "ClientCon_LAN.h"
#include "FileUtil.h"
#include 第三篇文章（服务端实现）
https://blog.csdn.net/qq_42662283/article/details/106493917