基于TCP的回声服务端/客户端：实现文件读写

服务端实现逻辑：

1、接收来自客户端的连接请求
2、接受来自客户端的文件名称信息
3、根据文件名称打开对应文件，并读取文件内容
4、将文件内容发送给客户端

客户端实现逻辑：

1、建立对服务端的TCP连接
2、接收用户需要读取的文件名，并发送给服务端
3、接收来自服务端的文件内容或读取失败的回复信息

服务端代码实现：echo_server.cpp

#include <iostream>
#include <string.h>
#include <fcntl.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/socket.h>
using namespace std;

#define BUF_SIZE 100

int main(int argc, char* argv[])
{
    char message[BUF_SIZE] = { 0 };
    
	// PF_INET -- IPv4协议族 SOCK_STREAM -- 流式控制协议
    int serv_sock = socket(PF_INET, SOCK_STREAM, 0); 
	
	// 初始化地址信息
	struct sockaddr_in serv_addr;
    memset(&serv_addr, 0, sizeof(serv_addr));
    serv_addr.sin_family = AF_INET; // IPv4地址族
    serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);
    serv_addr.sin_port = htons(atoi(argv[1]));
	
	// 地址信息绑定
    bind(serv_sock, (struct sockaddr*)&serv_addr, sizeof(serv_addr));
    
	// 开始监听
	listen(serv_sock, 5);
    
	// 接收来自客户端的连接请求
	struct sockaddr_in clnt_addr;
    socklen_t clnt_adr_sz = sizeof(clnt_addr);
    int clnt_sock = accept(serv_sock, (struct sockaddr*)&clnt_addr, &clnt_addr_sz);
    if(clnt_sock != -1)
    {
        cout << "client connect!!!" << endl;
    }
	
	// 读取来自客户端的信息：即文件名
    read(clnt_sock, message, sizeof(message));
    cout << "file name = " << message << endl;
    
	// 读取文件内容
    int fd = open(message, O_RDONLY);
    if(fd != -1)
    {
        int read_len = 0;
        int write_count = 0;
        while((read_len = read(fd,message, sizeof(message) - 1)) != 0)
        {
             ++write_count;
             message[read_len] = 0;    
             write(clnt_sock, message, read_len);
        }   
        cout << "write over! write count: " << write_count << endl;
    }
    else
    {
        cout << "file open error!!!" << endl;
    }
	
	// 关闭socket
	close(clnt_sock);
    close(serv_sock);
    return 0;
}

客户端代码实现：echo_client.cpp

#include <iostream>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/socket.h>
using namespace std;

#define BUF_SIZE 100

int main(int argc, char* argv[])
{
    char message[BUF_SIZE] = { 0 };
	
	// 建立socket
    int sock = socket(PF_INET, SOCK_STREAM, 0);

	// 初始化地址信息
    struct sockaddr_in serv_addr;
    memset(&serv_addr, 0, sizeof(serv_addr));
    serv_addr.sin_family = AF_INET;
    serv_addr.sin_addr.s_addr = inet_addr(argv[1]);
    serv_addr.sin_port = htons(atoi(argv[2]));
    
	// 建立对服务端的连接
    if(connect(sock,(struct sockaddr*)&serv_addr, sizeof(serv_addr)) != -1)
    {
        cout << "connected..." << endl;
    }
    
	// 用户输入文件名
    string file_name;
    cout << "please input file name you want read: " << endl;
    cin >> file_name;
    cout << "file_name  confirm: " << file_name << endl;
    
	// 向服务端发送文件名称
    write(sock, file_name.c_str(), file_name.length());
    
	// 读取来自服务端的文件内容
    int read_len = 0;
    int read_count = 0;
    while((read_len = read(sock, message, BUF_SIZE - 1)) != 0)
    {
        message[read_len] = 0;
        cout << "=============================" << endl;
        cout << "read count: " << ++read_count << endl;
        cout << "file content: " << message << endl;
        cout << "=============================" << endl;
    }
	
	// 关闭sock请求
    close(sock);
    return 0;
}

运行服务端和客户端程序，注意输入参数：

服务端：可执行程序 + 监听的端口号
客户端：可执行程序 + 服务端IP + 服务端端口号

以下是将要读取的文件内容：

以下是服务端运行结果：

以下是客户端运行结果：

考虑到不同的文件大小不同，所以采用服务端程序循环读取和发送
，客户端也采用循环接收，并且设置count变量，来表示服务端读文件的次数和客户端读取来自服务端消息的次数。

另外，可根据文件大小不同，调整BUF_SIZE的大小，来减少读写的次数。

谢谢阅读