以下是一个使用C++在Linux下通过socket实现TCP客户端和服务端进行文件收发的示例代码:
客户端代码:
#include <iostream>
#include <fstream>
#include <string>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#define BUFFER_SIZE 1024
void sendFile(int socketFd, const std::string& filename) {
std::ifstream file(filename, std::ios::binary);
if (!file) {
std::cerr << "Failed to open file: " << filename << std::endl;
return;
}
char buffer[BUFFER_SIZE];
while (!file.eof()) {
file.read(buffer, BUFFER_SIZE);
ssize_t bytesRead = file.gcount();
ssize_t bytesSent = send(socketFd, buffer, bytesRead, 0);
if (bytesSent != bytesRead) {
std::cerr << "Failed to send file" << std::endl;
break;
}
}
file.close();
// 发送文件结束标志
char fileEndFlag = '\0';
send(socketFd, &fileEndFlag, sizeof(char), 0);
}
int main() {
std::string serverIp = "127.0.0.1";
int serverPort = 8888;
std::string filename = "test.txt";
int socketFd = socket(AF_INET, SOCK_STREAM, 0);
if (socketFd == -1) {
std::cerr << "Failed to create socket" << std::endl;
return 1;
}
struct sockaddr_in serverAddr{};
serverAddr.sin_family = AF_INET;
serverAddr.sin_port = htons(serverPort);
if (inet_pton(AF_INET, serverIp.c_str(), &(serverAddr.sin_addr)) <= 0) {
std::cerr << "Invalid server IP address" << std::endl;
return 1;
}
if (connect(socketFd, reinterpret_cast<struct sockaddr*>(&serverAddr), sizeof(serverAddr)) == -1) {
std::cerr << "Failed to connect to the server" << std::endl;
return 1;
}
sendFile(socketFd, filename);
close(socketFd);
return 0;
}
服务端代码:
#include <iostream>
#include <fstream>
#include <string>
#include <sys/socket.h>
#include <netinet/in.h>
#include <unistd.h>
#define BUFFER_SIZE 1024
void receiveFile(int clientSocket, const std::string& filename) {
std::ofstream file(filename, std::ios::binary);
if (!file) {
std::cerr << "Failed to create file: " << filename << std::endl;
return;
}
char buffer[BUFFER_SIZE];
while (true) {
ssize_t bytesRead = recv(clientSocket, buffer, BUFFER_SIZE, 0);
if (bytesRead <= 0) {
break;
}
file.write(buffer, bytesRead);
}
file.close();
}
int main() {
int serverPort = 8888;
std::string filename = "received.txt";
int listenSocket = socket(AF_INET, SOCK_STREAM, 0);
if (listenSocket == -1) {
std::cerr << "Failed to create socket" << std::endl;
return 1;
}
struct sockaddr_in serverAddr{};
serverAddr.sin_family = AF_INET;
serverAddr.sin_port = htons(serverPort);
serverAddr.sin_addr.s_addr = htonl(INADDR_ANY);
if (bind(listenSocket, reinterpret_cast<struct sockaddr*>(&serverAddr), sizeof(serverAddr)) == -1) {
std::cerr << "Failed to bind socket" << std::endl;
return 1;
}
if (listen(listenSocket, 5) == -1) {
std::cerr << "Failed to listen on socket" << std::endl;
return 1;
}
while (true) {
struct sockaddr_in clientAddr{};
socklen_t clientAddrLen = sizeof(clientAddr);
int clientSocket = accept(listenSocket, reinterpret_cast<struct sockaddr*>(&clientAddr), &clientAddrLen);
if (clientSocket == -1) {
std::cerr << "Failed to accept client connection" << std::endl;
continue;
}
receiveFile(clientSocket, filename);
close(clientSocket);
}
close(listenSocket);
return 0;
}
这两个程序使用了多线程来实现并发处理多个客户端的连接和文件收发。在服务端代码中,每当有新的客户端连接时,会创建一个新的线程来处理该客户端的请求。在客户端代码中,通过socket连接到服务端,并使用send
函数发送文件内容到服务端。
请确保将代码中的文件名、IP地址和端口号等参数设置为正确的值。运行服务端程序后,可以通过运行多个客户端程序来测试并发连接和文件收发的功能。