用c++编写一个win32桌面应用程序,可以部署在两台机器上,实现socket通信,互传文件
时间: 2024-02-09 15:10:24 浏览: 67
好的,我来为您提供一个简单的 C++ Win32 Socket 通信程序,可以实现在两台机器上互传文件。
首先,我们需要创建一个 Win32 应用程序项目。在 Visual Studio 中,选择“新建项目”,选择 Visual C++ -> Win32 项目,选择“Win32 应用程序向导”,命名项目并选择“下一步”。
在向导的第一个页面中,选择“应用程序类型”为“对话框”,在下一页中选择“空项目”,然后单击“完成”按钮。
接下来,我们需要添加一些头文件和库文件。在 stdafx.h 文件中添加以下代码:
```c++
#include <WinSock2.h>
#include <WS2tcpip.h>
#include <Windows.h>
#include <CommCtrl.h>
#include <shlobj_core.h>
#include <fstream>
#include <string>
#include <sstream>
#pragma comment(lib, "Ws2_32.lib")
```
然后,我们需要创建一个对话框,用于用户选择文件和显示传输进度。在对话框中添加两个按钮和一个进度条控件。在对话框的头文件中添加以下控件 ID:
```c++
#define IDC_SELECT_FILE 1001
#define IDC_SEND_FILE 1002
#define IDC_PROGRESS_BAR 1003
```
在对话框的 WM_INITDIALOG 消息处理函数中,初始化进度条控件:
```c++
HWND progressBar = GetDlgItem(hwnd, IDC_PROGRESS_BAR);
SendMessage(progressBar, PBM_SETRANGE, 0, MAKELPARAM(0, 100));
```
在选择文件按钮的单击事件处理函数中,使用 Windows API 打开一个文件对话框,让用户选择要发送的文件:
```c++
OPENFILENAME ofn;
ZeroMemory(&ofn, sizeof(ofn));
ofn.lStructSize = sizeof(ofn);
ofn.hwndOwner = hwnd;
ofn.lpstrFilter = _T("All Files (*.*)\0*.*\0");
ofn.lpstrFile = szFileName;
ofn.nMaxFile = MAX_PATH;
ofn.Flags = OFN_EXPLORER | OFN_FILEMUSTEXIST | OFN_HIDEREADONLY;
ofn.lpstrDefExt = _T("txt");
if (GetOpenFileName(&ofn))
{
// 文件选择成功,可以进行发送操作
}
```
在发送文件按钮的单击事件处理函数中,创建一个套接字并连接到目标机器:
```c++
// 创建套接字
SOCKET sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
if (sock == INVALID_SOCKET)
{
MessageBox(hwnd, _T("Failed to create socket"), _T("Error"), MB_OK | MB_ICONERROR);
return;
}
// 连接到目标机器
char szHost[] = "192.168.1.2";
USHORT port = 12345;
sockaddr_in addr = { 0 };
addr.sin_family = AF_INET;
addr.sin_port = htons(port);
inet_pton(AF_INET, szHost, &addr.sin_addr);
if (connect(sock, (sockaddr*)&addr, sizeof(addr)) == SOCKET_ERROR)
{
MessageBox(hwnd, _T("Failed to connect to server"), _T("Error"), MB_OK | MB_ICONERROR);
closesocket(sock);
return;
}
// 发送文件
std::ifstream file(ofn.lpstrFile, std::ios::binary);
if (!file.is_open())
{
MessageBox(hwnd, _T("Failed to open file"), _T("Error"), MB_OK | MB_ICONERROR);
closesocket(sock);
return;
}
char buffer[1024];
while (file.read(buffer, sizeof(buffer)))
{
int nBytesSent = send(sock, buffer, file.gcount(), 0);
if (nBytesSent == SOCKET_ERROR)
{
MessageBox(hwnd, _T("Failed to send file"), _T("Error"), MB_OK | MB_ICONERROR);
break;
}
int nProgress = (int)(file.tellg() * 100 / fileSize);
SendMessage(progressBar, PBM_SETPOS, nProgress, 0);
}
file.close();
// 关闭套接字
closesocket(sock);
```
在接收文件的代码中,先创建一个监听套接字,并等待客户端连接:
```c++
// 创建监听套接字
SOCKET listenSock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
if (listenSock == INVALID_SOCKET)
{
MessageBox(hwnd, _T("Failed to create socket"), _T("Error"), MB_OK | MB_ICONERROR);
return;
}
// 绑定监听套接字
USHORT port = 12345;
sockaddr_in addr = { 0 };
addr.sin_family = AF_INET;
addr.sin_port = htons(port);
addr.sin_addr.s_addr = htonl(INADDR_ANY);
if (bind(listenSock, (sockaddr*)&addr, sizeof(addr)) == SOCKET_ERROR)
{
MessageBox(hwnd, _T("Failed to bind socket"), _T("Error"), MB_OK | MB_ICONERROR);
closesocket(listenSock);
return;
}
// 开始监听
if (listen(listenSock, SOMAXCONN) == SOCKET_ERROR)
{
MessageBox(hwnd, _T("Failed to listen on socket"), _T("Error"), MB_OK | MB_ICONERROR);
closesocket(listenSock);
return;
}
// 等待客户端连接
SOCKET clientSock = accept(listenSock, NULL, NULL);
if (clientSock == INVALID_SOCKET)
{
MessageBox(hwnd, _T("Failed to accept client connection"), _T("Error"), MB_OK | MB_ICONERROR);
closesocket(listenSock);
return;
}
closesocket(listenSock);
```
然后,在接收客户端发送的文件时,使用 recv 函数接收数据,并写入文件:
```c++
// 接收文件
std::ofstream file("received_file.txt", std::ios::binary);
if (!file.is_open())
{
MessageBox(hwnd, _T("Failed to create file"), _T("Error"), MB_OK | MB_ICONERROR);
closesocket(clientSock);
return;
}
char buffer[1024];
int nBytesReceived = 0;
do
{
nBytesReceived = recv(clientSock, buffer, sizeof(buffer), 0);
if (nBytesReceived == SOCKET_ERROR)
{
MessageBox(hwnd, _T("Failed to receive file"), _T("Error"), MB_OK | MB_ICONERROR);
break;
}
file.write(buffer, nBytesReceived);
int nProgress = (int)(file.tellp() * 100 / fileSize);
SendMessage(progressBar, PBM_SETPOS, nProgress, 0);
} while (nBytesReceived > 0);
file.close();
// 关闭套接字
closesocket(clientSock);
```
通过以上代码,我们就可以在两台机器上实现 Socket 通信,互传文件了。需要注意的是,需要在发送和接收方分别运行应用程序,且需要在接收方的防火墙中打开指定的端口,才能正常进行通信。
阅读全文