【UNP学习笔记】一个简单的服务器/客户端程序

照着书上的代码写了一个简单的socket程序，实现的功能是客户端向服务器发送一个字符串，服务器向客户端返回字符串的大小。写这个程序主要目的是熟悉基本的socket函数的使用，以及学习使用makefile。
测试时发现了一个有趣的现象：关闭服务器程序后，如果马上再次运行服务器程序就会返回bind error错误。这是由于关闭程序后TCP连接并没有马上终止，而是要执行TCP“四次挥手”，这时的服务器端处于TIME_WAIT状态，所以此时的端口仍然被占用。这时候如果立即执行服务器程序，bind函数就会因为端口被占用而失败。所以要等待几分钟直到上一个TCP连接完全终止，然后再运行新的服务端程序。

下面贴上代码

tcp_server.c

#include <stdio.h>#include <stdlib.h>#include <strings.h>#include <sys/types.h>#include <sys/socket.h>#include <arpa/inet.h>#include <unistd.h>#define PORT 8888 //侦听端口地址#define BACKLOG 2 //侦听队列长度void process_conn_server(int s);int main(int argc, char *argv[]){    int ss, sc;    struct sockaddr_in server_addr;    struct sockaddr_in client_addr;    int err;    pid_t pid;    //建立一个流式套接字    ss = socket(AF_INET, SOCK_STREAM, 0);    if(ss < 0){        printf("socket error\n");        return -1;    }    //设置服务器地址    bzero(&server_addr, sizeof(server_addr)); //清零    server_addr.sin_family = AF_INET; //地址族    server_addr.sin_addr.s_addr = htonl(INADDR_ANY); //本地地址    server_addr.sin_port = htons(PORT); //服务器端口    //绑定地址到套接字    err = bind(ss, (struct sockaddr*)&server_addr, sizeof(server_addr));    if(err < 0){        printf("bind error\n");        return -1;    }    //设置侦听    err = listen(ss, BACKLOG);    if(err < 0){        printf("listen error\n");        return -1;    }    //主循环    for(;;){        socklen_t addrlen = sizeof(struct sockaddr);        sc = accept(ss, (struct sockaddr*)&client_addr, &addrlen);        if(sc < 0){            continue;        }        //建立新进程处理到来的连接        pid = fork();        if(pid == 0){            close(ss);            process_conn_server(sc);        }else{            close(sc);        }    }}

tvp_client.c

#include <stdio.h>#include <stdlib.h>#include <strings.h>#include <sys/types.h>#include <sys/socket.h>#include <arpa/inet.h>#include <unistd.h>#define PORT 8888void process_conn_client(int s);int main(int argc, char* argv[]){    int s;    struct sockaddr_in server_addr; //服务器地址结构    s = socket(AF_INET, SOCK_STREAM, 0); //建立流式套接字    if(s < 0){        printf("socket error");        return -1;    }    //设置服务器地址    bzero(&server_addr, sizeof(server_addr));    server_addr.sin_family = AF_INET;    server_addr.sin_addr.s_addr = htonl(INADDR_ANY);    server_addr.sin_port = htons(PORT);    //将用户输入的点分十进制地址转换为二进制值    inet_pton(AF_INET, argv[1], &server_addr.sin_addr);    //连接服务器    connect(s, (struct sockaddr*)&server_addr, sizeof(struct sockaddr));    process_conn_client(s); //客户端处理过程    close(s); //关闭连接    return 0;}

tcp_process.c

#include <stdio.h>#include <string.h>#include <unistd.h>//服务器对客户端的处理void process_conn_server(int s){    ssize_t size = 0;    char buffer[1024]; //数据缓冲区    for(;;){        size = read(s, buffer, 1024); //从套接字读取数据到缓冲区        if(size == 0)            return;        //构建响应字符，为接收到客户端字节的数量        sprintf(buffer, "%ld bytes altogether\n", size);        write(s, buffer, strlen(buffer)+1); //发送给客户端    }}//客户端的处理过程void process_conn_client(int s){    ssize_t size = 0;    char buffer[1024]; //数据缓冲区    for(;;){        size = read(0, buffer, 1024); //从标准输入读取数据到缓冲区        if(size > 0){            write(s, buffer, size); //发送给服务器            size = read(s, buffer, 1024); //从服务器读取数据            write(1, buffer, size); //写入到标准输出        }    }}

Makefile文件

all:client serverclient:tcp_process.o tcp_client.o    gcc -o client tcp_process.o tcp_client.oserver:tcp_process.o tcp_server.o    gcc -o server tcp_process.o tcp_server.otcp_process.o:    gcc -c tcp_process.c -o tcp_process.oclean:    rm -f client server *.o