点对点通信实例

　　实现点对点通信，双方都能发送数据、接收数据，双方维护套接字。read对方数据和read键盘输入目前使用两个进程来实现。

　　服务器进程：

#include<unistd.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<string.h>
#include<stdlib.h>
#include<stdio.h>
#include<errno.h>
#include<netinet/in.h>
#include<arpa/inet.h>
#include<signal.h>
#define ERR_EXIT(m)
    do
    {
        perror(m);
        exit(EXIT_FAILURE);
    }while(0)
void handler(int sig)
{
    printf("recv a sig=%d
",sig);
    printf("ser child close");
    exit(EXIT_SUCCESS);//退出子进程

}
int main(void)
{
    int listenfd;
    if((listenfd=socket(PF_INET,SOCK_STREAM,IPPROTO_TCP))<0)
        ERR_EXIT("socket error");
    //if((listenfd=socket(PF_INET,SOCK_STREAM,0))<0)
    

    //本地协议地址赋给一个套接字
    struct sockaddr_in servaddr;
    memset(&servaddr,0,sizeof(servaddr));
    servaddr.sin_family=AF_INET;
    servaddr.sin_port=htons(5188);
    servaddr.sin_addr.s_addr=htonl(INADDR_ANY);//表示本机地址
    //servaddr.sin_addr.s_addr=inet_addr("127.0.0.1");
    //inet_aton("127.0.0.1",&servaddr.sin_addr);
    
    //开启地址重复使用,关闭服务器再打开不用等待TIME_WAIT
    int on=1;
    if(setsockopt(listenfd,SOL_SOCKET,SO_REUSEADDR,&on,sizeof(on))<0)
        ERR_EXIT("setsockopt error");
    //绑定本地套接字
    if(bind(listenfd,(struct sockaddr*)&servaddr,sizeof(servaddr))<0)
        ERR_EXIT("bind error");
    if(listen(listenfd,SOMAXCONN)<0)//设置监听套接字(被动套接字)
        ERR_EXIT("listen error");
    
    struct sockaddr_in peeraddr;//对方套接字地址
    socklen_t peerlen=sizeof(peeraddr);
    int conn;//已连接套接字(主动套接字)
    if((conn=accept(listenfd,(struct sockaddr*)&peeraddr,&peerlen))<0)
        ERR_EXIT("accept error");
    printf("IP=%s,port=%d
",inet_ntoa(peeraddr.sin_addr),ntohs(peeraddr.sin_port));
    pid_t pid;
    pid=fork();
    if(pid==-1)
        ERR_EXIT("fork");
    if(pid==0)
    {
        signal(SIGUSR1,handler);
        //子进程获取键盘输入,发送数据
        char sendbuf[1024]={0};
        while(fgets(sendbuf,sizeof(sendbuf),stdin)!=NULL)
        {
            write(conn,sendbuf,strlen(sendbuf));
            memset(sendbuf,0,sizeof(sendbuf));
        }
        //printf("child close
");
        exit(EXIT_SUCCESS);
    }
    else
    {
      //父进程获取对方数据
        char recvbuf[1024];
        while(1)
        {
            memset(recvbuf,0,sizeof(recvbuf));
            int ret=read(conn,recvbuf,sizeof(recvbuf));
            if(ret==-1)
                ERR_EXIT("read");
            else if(ret==0)
                {
                    printf("peer closed");
                    break;
                }
            fputs(recvbuf,stdout);
        }
        //printf("ser father close");
        kill(pid,SIGUSR1);//通知子进程
        exit(EXIT_SUCCESS);
    }
    return 0;
}

　　客户端程序代码：

#include<unistd.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<string.h>
#include<stdlib.h>
#include<stdio.h>
#include<errno.h>
#include<netinet/in.h>
#include<arpa/inet.h>
#include<signal.h>
#define ERR_EXIT(m)
    do
    {
        perror(m);
        exit(EXIT_FAILURE);
    }while(0)
void handler(int sig)
{
    printf("recv a sig=%d
",sig);
    exit(EXIT_SUCCESS);
}
int main(void)
{
    int sock;//客户端创建套接字
    if((sock=socket(PF_INET,SOCK_STREAM,IPPROTO_TCP))<0)
        ERR_EXIT("socket error");
    
    struct sockaddr_in servaddr;//本地协议地址赋给一个套接字
    memset(&servaddr,0,sizeof(servaddr));
    servaddr.sin_family=AF_INET;
    servaddr.sin_port=htons(5188);
    
    servaddr.sin_addr.s_addr=inet_addr("127.0.0.1");//服务器段地址
    //inet_aton("127.0.0.1",&servaddr.sin_addr);
    
    if(connect(sock,(struct sockaddr*)&servaddr,sizeof(servaddr))<0)
        ERR_EXIT("connect");


    pid_t pid;
    pid=fork();
    if(pid==-1)
        ERR_EXIT("fork");
    if(pid==0)
    {
        char recvbuf[1024];
        //子进程接收对方数据，服务器端也能发送数据。
        while(1)
        {
            memset(recvbuf,0,sizeof(recvbuf));
            int ret=read(sock,recvbuf,sizeof(recvbuf));
            if(ret==-1)
                ERR_EXIT("read");
            else if(ret==0)
                {
                    printf("peer closed
");
                    break;
                }
            fputs(recvbuf,stdout);
        }
        close(sock);
        //break跳出循环的话，说明服务端关闭了.那么给父进程发送信号通知父进程
        //父进程是给对方发送数据的，对方关闭了，那么父进程收到信号后，退出进程，不再发送数据。
        kill(getppid(),SIGUSR1);
        exit(EXIT_SUCCESS);
    }
    else
    {
        signal(SIGUSR1,handler);//如果没有这个，父进程不会退出
        //父进程发送数据
        char sendbuf[1024]={0};
        while(fgets(sendbuf,sizeof(sendbuf),stdin)!=NULL)//默认有换行符
        {
            write(sock,sendbuf,strlen(sendbuf));
            memset(sendbuf,0,sizeof(sendbuf));
        }
        close(sock);
        printf("cli father close
");
        exit(EXIT_SUCCESS);
    }
    return 0;
}