UNPv1第五章:TCP客户服务器程序实例

来源:互联网 发布:跳舞毯安装软件 编辑:程序博客网 时间:2024/06/07 03:04

现在写一个完整的TCP客户-服务器程序例子

这个简单的例子是完成下述功能的一个回射服务器:
1. 客户从标准输入读一行文本,写到服务器上
2. 服务器读入此行,并回射给客户
3. 客户读此回射行写到标准输出
这里写图片描述
客户端代码:

#include <stdio.h>#include <stdlib.h>#include <string.h>#include <sys/socket.h>#include <arpa/inet.h>#include <netinet/in.h>#define SRV_PORT 8888#define MAXLINE 4096void str_cli(FILE *fp, int sockfd);int main(int argc, char **argv){    if(argc != 2)    {        printf("usage:tcpcli <ip address>\n");        exit(0);    }    int sockfd = socket(AF_INET, SOCK_STREAM, 0);    if(sockfd < 0)    {        perror("create socket error.");    }    struct sockaddr_in srvaddr;    bzero(&srvaddr, sizeof(srvaddr));    srvaddr.sin_family = AF_INET;    srvaddr.sin_port = htons(SRV_PORT);    if(inet_pton(AF_INET, argv[1], &srvaddr.sin_addr) <= 0)    {        printf("address error %s\n", argv[1]);        exit(0);    }    if(connect(sockfd, (struct sockaddr *)&srvaddr, sizeof(srvaddr)) < 0 )    {        perror("connect error");    }    str_cli(stdin, sockfd);    exit(0);}void str_cli(FILE *fp, int sockfd){    char sendline[MAXLINE];    char readline[MAXLINE];    while(fgets(sendline, MAXLINE, fp))    {        if( write(sockfd, sendline, strlen(sendline)) != strlen(sendline) )        {            perror("send data error");        }        if( read(sockfd, readline, MAXLINE) == 0)        {            perror("recv data error");        }        fputs(readline,stdout);    }}

服务器端代码:

#include <stdio.h>#include <stdlib.h>#include <string.h>#include <sys/socket.h>#include <arpa/inet.h>#include <netinet/in.h>#define SRV_PORT 8888#define MAXLINE 4096void str_echo(int fd);int main(int argc, char **argv){    int listenfd = socket(AF_INET, SOCK_STREAM, 0);    if(listenfd < 0)    {        perror("create socket error.");    }    struct sockaddr_in srvaddr;    bzero(&srvaddr, sizeof(srvaddr));    srvaddr.sin_family = AF_INET;    srvaddr.sin_addr.s_addr = htonl(INADDR_ANY);    srvaddr.sin_port = htons(SRV_PORT);    if(bind(listenfd, (struct sockaddr*)&srvaddr, sizeof(srvaddr)) < 0)    {        perror("bind error.");    }    if(listen(listenfd, 1023) < 0)    {        perror("listen error.");    }    struct sockaddr_in cliaddr;    for(; ;)    {        socklen_t clilen = sizeof(cliaddr);        int connfd = accept(listenfd, (struct sockaddr*)&cliaddr, &clilen);        if(connfd < 0)        {            perror("accept error.");        }        pid_t childpid;        if( (childpid = fork()) == 0 )        {            close(listenfd);            str_echo(connfd);            exit(0);        }        close(connfd);    }    return 0;}void str_echo(int sockfd){    char line[MAXLINE];    while(read(sockfd, line, MAXLINE) != 0)    {        if(write(sockfd, line, strlen(line)) != strlen(line))        {            perror("write error");        }    }}

accept返回前客户/服务器的状态
这里写图片描述
accept返回后客户/服务器的状态
(connfd = Accept(listenfd, (SA *) &cliaddr, &clilen);)
这里写图片描述
fork返回后客户/服务器的状态.
注意,此时listen和connfd这两个描述符都在父进程和子进程共享
这里写图片描述
在下一步是由父进程关闭已连接套接字,由子进程关闭监听套接字
父子进程关闭相应套接字后客户/服务器的状态
这里写图片描述

wait和waitpid

可以调用如下两个函数处理已终止的子进程

#include <sys/wait.h>pid_t wait (int *statloc);pid_t waitpid (pid_t pid, int *statloc, int options);//返回值:成功返回进程ID,出错返回返回0或-1;

对于参数pid 想等待的进程ID号。-1表示等待第一个结束的子进程,options附加选项,常用的是WNOHANG,告知内核在没有以终止子进程时不要阻塞
函数wait和waitpid均返回两个值: 函数的返回值是终止子进程的进程ID号,子进程的终止状态(一个整数)则是通过指针statloc返回的。
wait和waitpid的区别: wait 等待第一个结束的子进程,如果没有结束的子进程,wait将阻塞。waitpid 通过参数设置,可以在没有子进程结束时waitpid不阻塞。

0 0
原创粉丝点击