2.2 TCP协议下实现两人聊天

head.h

//TCP协议下实现两人聊天
#ifndef _HEAD_H_
#define _HEAD_H_

#include <stdio.h>
#include<stdlib.h>
#include<string.h>
#include<error.h>
#include<errno.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<netinet/in.h>
#include<netinet/ip.h>
#include<arpa/inet.h>
#include<signal.h>
#define error_exit(_errmsg_) error(EXIT_FAILURE,errno,_errmsg_)
#define SER_PORT 50000
#define SER_ADDR "192.168.239.129"

#endif

Makefile

ALL:client server
client:client.c
    gcc $< -o $@
server:server.c
    gcc $< -o $@

clean:
    rm client server

client.c

#include"head.h"
int main()//client
{    
    int sockfd=0;
    char buff[1024]={0};
    struct sockaddr_in seraddr;
    seraddr.sin_family=AF_INET;
    seraddr.sin_port=htons(SER_PORT);
    seraddr.sin_addr.s_addr=inet_addr(SER_ADDR);
    //创建一个用来通信的文件描述符
    if(-1==(sockfd=socket(AF_INET,SOCK_STREAM,0)))
        error_exit("fail to socket!");
    //发送一个链接请求，有绑定功能
    if(-1==connect(sockfd,(struct sockaddr* )&seraddr,sizeof(seraddr)))
        error_exit("fail to connect!");
    //创建父子进程
    pid_t pid,pid1;
    if(-1==(pid=fork()))
        error_exit("fail to fork!");
    if(pid==0)//子进程负责发送
    {
        while(1)
        {    //从终端读取数据到buff/ fgets 读到字符串尾加'\0'
            fgets(buff,sizeof(buff),stdin);
            //send 无绑定功能,不绑定客户端自己的IP地址和端口号发送出去的。
            if(-1==send(sockfd,buff,strlen(buff)+1,0))
                error_exit("fail to send!");
            if(!strncmp(buff,"quit",4))break;//发quit退出
        }
        pid1=getppid();
    }
    else if(pid>0)//父进程负责接收
    {
        while(1)
        {
            //收到服务器端回复，可以不显示对方IP地址,recv有阻塞功能
            if(-1==recv(sockfd,buff,sizeof(buff),0))
                error_exit("fail to recv!");
            //printf("%s:%d-->buff=%s\n",inet_ntoa(cltaddr.sin_addr),
            //        ntohs(cltaddr.sin_port),buff);
            if(!strncmp(buff,"quit",4))break;
            printf("RECV:%s",buff);//接收显示
        }
        pid1=pid;
    }
    kill(pid1,SIGKILL);//杀死父子进程
    close(sockfd);//关闭套接字文件描述符
    return 0;
}

server.c

#include"head.h"
int main()//server
{    
    int sockfd=0;
    int confd=0;
    char buff[1024]={0};
    struct sockaddr_in seraddr;
    seraddr.sin_family=AF_INET;
    seraddr.sin_port=htons(SER_PORT);
    seraddr.sin_addr.s_addr=inet_addr(SER_ADDR);
    //创建一个用来通信的文件描述符
    if(-1==(sockfd=socket(AF_INET,SOCK_STREAM,0)))
        error_exit("fail to socket!");
    //将sockfd和IP地址绑定
    if(-1==bind(sockfd,(struct sockaddr*)&seraddr,sizeof(seraddr)))//socklen_t addrlen;
        error_exit("fail to bind!");
    //指定链接请求队列的长度，listen无阻塞功能
    if(-1==listen(sockfd,10))
        error_exit("fail to listen!");
    //与链接请求队列中第一个链接请求建立TCP握手,accept有阻塞等待功能
    if(-1==(confd=accept(sockfd,NULL,NULL)))
        error_exit("fail to accept!");
    //创建父子进程
    pid_t pid,pid1;
    if(-1==(pid=fork()))
        error_exit("fail to fork!");
    if(pid==0)//子进程负责发送
    {
        while(1)
        {    //从终端读取数据到buff/ fgets 读到字符串尾加'\0'
            fgets(buff,sizeof(buff),stdin);
            //send 无绑定功能,不绑定客户端自己的IP地址和端口号发送出去的。
            if(-1==send(confd,buff,strlen(buff)+1,0))
                error_exit("fail to send!");
            if(!strncmp(buff,"quit",4))break;//发quit退出
        }
        pid1=getppid();
    }
    else if(pid>0)//父进程负责接收
    {
        while(1)
        {
            //收到客户端信息，可以不显示对方IP地址,recv有阻塞功能
            if(-1==recv(confd,buff,sizeof(buff),0))
                error_exit("fail to recv!");
            //printf("%s:%d-->buff=%s\n",inet_ntoa(cltaddr.sin_addr),
            //        ntohs(cltaddr.sin_port),buff);
            if(!strncmp(buff,"quit",4))break;
            printf("RECV:%s",buff);//接收显示
        }
        pid1=pid;
    }
    kill(pid1,SIGKILL);//杀死父子进程
    close(sockfd);//关闭套接字文件描述符
    close(confd);//关闭各子进程文件描述符
    return 0;
}