【unix网络编程】5-2回射服务器程序

先写一个简单的程序，这个程序完成了基本的功能，但有很多地方还需要改进

服务器程序：

#include "unp.h"

void str_echo(int sockfd);

int main(int argc,char **argv)
{
    int listenfd,connfd;
    struct sockaddr_in cliaddr,servaddr;
    pid_t childpid;
    socklen_t len;

    listenfd = Socket(AF_INET,SOCK_STREAM,0);

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

    Bind(listenfd,(SA *)&servaddr,sizeof(servaddr));

    Listen(listenfd,LISTENQ);

    for(;;){
	len = sizeof(cliaddr);
        connfd = Accept(listenfd,(SA *)&cliaddr,&len);
        Close(listenfd);
        str_echo(connfd);
        Close(connfd);
        exit(0);
    }
    Close(connfd);

    exit(0);
}

void str_echo(int sockfd)
{
    ssize_t n;
    char buf[MAXLINE];

again:      //读取的时候，不要管换行符，读多少写多少
    while( (n = read(sockfd,buf,MAXLINE)) > 0){
        Writen(sockfd,buf,n);   //将读进来的数据写回，这里要用自己定义的Writen，防止只发送了一部分数据就不发送了
    }
    if(n < 0 && errno == EINTR) //忽略因为意外的中断引起的errno
        goto again;
    else if(n < 0)
        err_sys("str_echo:read error\n");
}

客户程序

#include "unp.h"


void str_cli(FILE *fp,int sockfd);


int main(int argc,char **argv)
{
    int sockfd;
    struct sockaddr_in servaddr;

    if(argc != 2)
    	err_quit("usage: tcpcli<ipaddress>");
    
    sockfd = Socket(AF_INET,SOCK_STREAM,0);

    bzero(&servaddr,sizeof(servaddr));
    servaddr.sin_family = AF_INET;
    servaddr.sin_port = htons(SERV_PORT);
    Inet_pton(AF_INET,argv[1],&servaddr.sin_addr);

    Connect(sockfd,(SA *)&servaddr,sizeof(servaddr));

    str_cli(stdin,sockfd);

    exit(0);
}

void str_cli(FILE *fp,int sockfd)
{
    char sendline[MAXLINE],recvline[MAXLINE];

    while(Fgets(sendline,MAXLINE,fp) != NULL){
        Writen(sockfd,sendline,strlen(sendline));   //这里的最后一个参数要用有效的个数！不要包含自动添加的空字符

        if(Readline(sockfd,recvline,MAXLINE) == 0){
            err_quit("str_cli:server terminated prematurely");
        }
        Fputs(recvline,stdout);
    }
}