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

2019-02-21 10:39:05 浏览数 (2)

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

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

客户端代码:

代码语言:javascript复制
#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 4096
void 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 %sn", 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);
    }
}

服务器端代码:

代码语言:javascript复制
#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 4096
void 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

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

代码语言:javascript复制
#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 人点赞