Linux网络编程之简易的点对点聊天程序程序

本程序基于上一篇echo_server的实现,稍作修改,增加了发送消息,服务端客户端各自运行两个进程,一个进程负责发送消息,另一个进程负责接收消息。
代码如下:

  • p2p_server.c
#include <sys/socket.h>
#include <sys/types.h>
#include <string.h>
#include <unistd.h>
#include <memory.h>
#include <errno.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <stdio.h>
#include <stdlib.h>

#define ERR_EXIT(m) \
    do{         \
         perror(m);       \
        exit(EXIT_FAILURE);\
    }while(0)

int main(int argc, char *argv[])
{
    int socket_fd;
    if((socket_fd=socket(PF_INET,SOCK_STREAM,0))==-1)
    {
        ERR_EXIT("socket");
    }


    struct sockaddr_in servaddr;
    memset(&servaddr,0,sizeof(servaddr));
    servaddr.sin_family=AF_INET;
    servaddr.sin_port=htons(5678);
    servaddr.sin_addr.s_addr=inet_addr("127.0.0.1");

    /*
       servaddr.sin_addr.s_addr=htonl(INADDR_ANY);
       inet_aton("127.0.0.1",&servaddr.sin_addr);
     */

    int on=1;
    if(setsockopt(socket_fd,SOL_SOCKET,SO_REUSEADDR,&on,sizeof(on))==-1){
        ERR_EXIT("setsockopt");
    }

    if(bind(socket_fd,(struct sockaddr*)&servaddr,sizeof(servaddr))==-1)
        ERR_EXIT("bind");


    if(listen(socket_fd,SOMAXCONN)==-1)
        ERR_EXIT("listen");

    struct sockaddr_in peeraddr;
    memset(&peeraddr,0,sizeof(peeraddr));
    socklen_t len=sizeof(peeraddr);
    int conn;

    if((conn=accept(socket_fd,(struct sockaddr*)&peeraddr,&len))==-1)
        ERR_EXIT("accept");
    printf("ip:%s\tport:%d\n",inet_ntoa(peeraddr.sin_addr),ntohs(peeraddr.sin_port));

    pid_t pid;
    pid=fork();
    if(pid<0){
        ERR_EXIT("fork");
    }
    else if(pid==0){
        //child process receive message
        close(socket_fd);
        char recebuf[1024]={};
        while(1){
            memset(recebuf,0,sizeof(recebuf));
            int rlen=read(conn,recebuf,sizeof(recebuf));
            if(rlen==-1){
                ERR_EXIT("receive");
            }
            else if(rlen==0){
                printf("close:\n");
                close(conn);
                exit(EXIT_SUCCESS);
            }
            else{
                fputs(recebuf,stdout);
                write(conn,recebuf,rlen);
            }

        }

        close(conn);


    }else{
        //parent process send message
        char sendbuf[1024]={};
        while(fgets(sendbuf,sizeof(sendbuf),stdin)!=NULL){
          write(conn,sendbuf,sizeof(sendbuf));
            fputs(sendbuf,stdout);
            memset(sendbuf,0,sizeof(sendbuf));
        }

        close(conn);
        close(socket_fd);

    }
    return 0;
}
  • p2p_client.c
#include <sys/socket.h>
#include <sys/types.h>
#include <string.h>
#include <unistd.h>
#include <memory.h>
#include <errno.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <stdio.h>
#include <stdlib.h>

#define ERR_EXIT(m)\
        do{\
        perror(m);\
        exit(EXIT_FAILURE);\
}  while(0)


int main(int argc, char *argv[])
{

    if(argc<2){
        ERR_EXIT("too little arguments");
    }
    int socket_fd;
    if((socket_fd=socket(AF_INET,SOCK_STREAM,0))==-1){
        ERR_EXIT("socket");
    }

    struct sockaddr_in servaddr;
    servaddr.sin_family=AF_INET;
    servaddr.sin_port=htons(5678);
    servaddr.sin_addr.s_addr=inet_addr(argv[1]);

    if(connect(socket_fd,(struct sockaddr*)&servaddr,sizeof(servaddr))==-1){
        ERR_EXIT("connect");
    }


    pid_t pid;
    pid=fork();

    if(pid==-1){
        ERR_EXIT("fork");
    }
    else if(pid==0)
    {
        //child process receive message
        char recvbuf[1024]={};
        int ret;
        while(1){
            ret=read(socket_fd,recvbuf,sizeof(recvbuf));
            if(ret<0)
            {
                ERR_EXIT("read failure");
            }
            else if(ret==0){
                printf("close\n");
                exit(EXIT_SUCCESS);
            }else{
               fputs(recvbuf,stdout);

            }
            memset(recvbuf,0,sizeof(recvbuf));
        }

    }else {
        //parent process send message
        char sendbuf[1024]={};

        while(fgets(sendbuf,sizeof(sendbuf),stdin)!=NULL){
            write(socket_fd,sendbuf,strlen(sendbuf));
            memset(sendbuf,0,sizeof(sendbuf));
        }

    }

    close(socket_fd);

    return 0;
}
  • 运行截图
    p2pchat
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值