Linux网络编程示例之多进程

server.c

#include <stdio.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <errno.h>
#include <signal.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <fcntl.h>
#include <stdlib.h>
#include <string.h>

int fd;
char buff[512];
void sig_handler(int signo)
{
    if (signo == SIGINT) {
        printf("server ctrl+c,stop server\n");
        close(fd);
        exit(0);//结束整个进程
    }

    if(signo == SIGCHLD)
	{
		printf("child process deaded...\n");
		wait(0);
	}
}

int main(int argc,char* argv[])
{
    if (argc < 2) {
        perror("usage port");
        return 0;
    }

    //接收信号，ctrl+z结束程序
    if (signal(SIGINT,sig_handler) == SIG_ERR) {
        perror("server signal error");
        return 0;
    }

    int err = 0;
    //创建socket fd
    if ((fd = socket(AF_INET,SOCK_STREAM,0)) == -1) {
        perror("server socket error");
        return 0;
    }

    //绑定ip和端口号
    struct sockaddr_in serveraddr;
	memset(&serveraddr,0,sizeof(serveraddr));
	serveraddr.sin_family = AF_INET;
	serveraddr.sin_port = htons(atoi(argv[1]));//port
	serveraddr.sin_addr.s_addr = INADDR_ANY;
    if ((err = bind(fd,(struct sockaddr*)&serveraddr,sizeof(serveraddr))) == -1) {
        perror("server bind error");
        close(fd);
        return 0;
    }

    //listen 监听
    if ((err = listen(fd,10)) == -1) {
        perror("server listen error");
        close(fd);
        return 0;
    }

    //accept 等待连接
    struct sockaddr_in clientaddr;
    memset(&clientaddr,0,sizeof(clientaddr));
    socklen_t len = sizeof(clientaddr);
    while (1) {
        int client_fd;
        if ((client_fd = accept(fd,(struct sockaddr*)&clientaddr,&len)) == -1) {
            perror("server accept error");
            continue;
        }

        //启动子进程调用IO函数和客户端进行通信
        pid_t pid = fork();
        if (pid < 0) {
            perror("fork error");
            continue;
        } else if (pid == 0) {
            //child
            //读取客户端发送来的数据,并把它写会客户端
            while (1) {
                memset(buff,0,sizeof(buff));
                // printf("start read and write...\n");
                size_t size;
                if ((size = read(client_fd,buff,sizeof(buff))) <= 0) {
                    perror("server read error");
                    continue;
                } else {
                    printf("> %s\n",buff);
                    //将读取到的数据在发送回客户端
                    if (write(client_fd,buff,sizeof(buff)) < 0) {
                        perror("server write data to client error");
                    }
                }
            }
        } else {
            //parent
            //关闭accept建立连接的套接字
            //fork()之后会在子进程复制一份accept建立连接的套接子，独立于父进程，此处关闭的是属于父进程的acdept套接字
            close(client_fd);
        }
    }
    return 0;
}

client.c

#include <stdio.h>
#include <stdlib.h>
#include <memory.h>
#include <errno.h>
#include <unistd.h>
#include <sys/socket.h>
#include <arpa/inet.h>

int fd;
char buff[512];
static char serversay[] = "                              server say:";
static char clientsay[] = "client say:";

int main(int argc,char* argv[])
{
    if (argc < 3) {
		perror("usage error ip & port");
		return 0;
	}

	if ((fd = socket(AF_INET,SOCK_STREAM,0)) == -1) {
		perror("socket error");
		return 0;
	}

	//填充服务器地址，进行连接请求
	struct sockaddr_in seraddr;
    memset(&seraddr,0,sizeof(seraddr));
	seraddr.sin_family = AF_INET;
	seraddr.sin_port = htons(atoi(argv[2]));
	inet_pton(AF_INET,argv[1],&seraddr.sin_addr.s_addr);
	if (connect(fd,(struct sockaddr*)&seraddr,sizeof(seraddr)) == -1) {
		perror("connect error");
		close(fd);
		return 0;
	}

	//连接成功，与服务端通信，进行读写
	while (1) {
		write(STDOUT_FILENO,clientsay,sizeof(clientsay));
		//从终端读取数据
		memset(buff,0,sizeof(buff));
		if (read(STDIN_FILENO,buff,sizeof(buff)) > 0) {
			//读取成功，将数据发往服务端
			if (write(fd,buff,sizeof(buff)) > 0) {
				memset(buff,0,sizeof(buff));
				if (read(fd,buff,sizeof(buff)) > 0) {
					write(STDOUT_FILENO,serversay,sizeof(serversay));
					write(STDOUT_FILENO,buff,sizeof(buff));
				}
			}
		}
	}

    return 0;
}

执行：