【网络编程实例】C++实现基于TCP的服务器和客户端文件传输

参考《TCP/IP网络编程第7章》

• 实现服务器向客户端发送文件数据，在发送完毕后半关闭socket，并等待客户端返回的消息；
• 注意半关闭shutdown函数的调用；
  
  服务器端代码：

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <sys/socket.h>
#include <sys/unistd.h>
#include <sys/types.h>
#include <sys/errno.h>
#include <netinet/in.h>
#include <signal.h>

#include <iostream>
#include <string>

using std::cout;
using std::cin;
using std::string;
using std::endl;

#define BUFFSIZE 30
#define DEFAULT_PORT 9190    // 指定端口为 9190

int main()
{
    int serv_sock, clnt_sock;
    FILE *fp;
    char buf[BUFFSIZE];
    int read_cnt;

    struct sockaddr_in serv_adr, clnt_adr;
    socklen_t clnt_adr_sz;

    fp = fopen("file_server.cpp", "rb");
    serv_sock = socket(PF_INET, SOCK_STREAM, 0);

    bzero(&serv_adr, sizeof(serv_adr));
    serv_adr.sin_family = AF_INET;
    serv_adr.sin_addr.s_addr = htonl(INADDR_ANY);
    serv_adr.sin_port = htons(DEFAULT_PORT);

    bind(serv_sock, (struct sockaddr*) &serv_adr, sizeof(serv_adr));
    listen(serv_sock, 5);

    clnt_adr_sz = sizeof(clnt_adr);
    clnt_sock = accept(serv_sock, (struct sockaddr*)&clnt_adr, &clnt_adr_sz);

    while(1) 
    {
         //从文件流中读取数据，buffer为接收数据的地址，size为一个单元的大小，count为单元个数，stream为文件流
        //返回实际读取的单元个数 
        read_cnt = fread((void*) buf, 1, BUFFSIZE, fp);
        if(read_cnt < BUFFSIZE)
        {
            write(clnt_sock, buf, read_cnt);
            break;
        }
        write(clnt_sock, buf, BUFFSIZE);

        shutdown(clnt_sock, SHUT_WR); //半关闭，可以读取，但不能向socket中写入
        read(clnt_sock, buf, BUFFSIZE);
        cout << "message from cilent: " << buf << endl;

        fclose(fp);
        close(clnt_sock);
        close(serv_sock);
        return 0;
    }
}

客户端代码：

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <sys/socket.h>
#include <sys/unistd.h>
#include <sys/types.h>
#include <sys/errno.h>
#include <netinet/in.h>
#include <arpa/inet.h>

#include <iostream>
#include <string>

using std::cout;
using std::cin;
using std::string;
using std::endl;

#define BUFFSIZE 2048
#define SERVER_IP "127.0.0.1"    // 指定服务端的IP
#define SERVER_PORT 9190            // 指定服务端的port

int main()
{
    int sock;
    FILE* fp;

    char buf[BUFFSIZE];
    int read_cnt;
    struct sockaddr_in serv_adr;


    fp = fopen("receive.cpp", "wb");
    sock = socket(PF_INET, SOCK_STREAM, 0);

    bzero(&serv_adr, sizeof(serv_adr));
    serv_adr.sin_family = AF_INET;
    serv_adr.sin_addr.s_addr = inet_addr(SERVER_IP);
    serv_adr.sin_port = htons(SERVER_PORT);

    connect(sock, (struct sockaddr*) &serv_adr, sizeof(serv_adr));

    while((read_cnt = read(sock, buf, BUFFSIZE)) != 0) 
        fwrite((void*) buf, 1, read_cnt, fp);

    cout << "receiced file data" << endl;
    write(sock, "thank u" , 10);
    fclose(fp);
    close(sock);
    return 0;

}