【Linux篇】项目：多方收发消息

一、编程实现 

服务端

#include <stdio.h>
#include <sys/types.h>          /* See NOTES */
#include <sys/socket.h>
#include <stdlib.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <string.h>
#include <unistd.h>

int main(int argc,char *argv[])
{
    if(argc != 3){
        printf("参数错误\n");
        exit(-1);
    }

    int s_fd;
    int c_fd;
    int n_read;
    char readBuf[128];
    char msg[128] = {0};
    int mark = 0;

    struct sockaddr_in s_addr;
    struct sockaddr_in c_addr;

    memset(&s_addr,0,sizeof(struct sockaddr_in));
    memset(&c_addr,0,sizeof(struct sockaddr_in));

    //1.socket
    s_fd = socket(AF_INET,SOCK_STREAM,0);
    if(s_fd == -1){
        perror("socket");
        exit(-1);
    }

    //2.bind
    s_addr.sin_family = AF_INET;
    s_addr.sin_port = htons(atoi(argv[2]));
    inet_aton(argv[1],&s_addr.sin_addr);

    bind(s_fd,(struct sockaddr *)&s_addr,sizeof(struct sockaddr_in));
    //3. listen
    listen(s_fd,10);
    //4.accept
    int clen = sizeof(struct sockaddr_in);
    while(1){
        c_fd = accept(s_fd,(struct sockaddr *)&c_addr,&clen);
        if(c_fd == -1){
            perror("accept");
        }
        
        mark++;
        printf("get connect:%s\n",inet_ntoa(c_addr.sin_addr));

        if(fork() == 0){
            if(fork() == 0){
                while(1){ //不断发送，类似心跳包
                    memset(msg,0,sizeof(msg));
                    sprintf(msg,"welcome No.%d client",mark);
                    //6.write
                    write(c_fd,msg,strlen(msg));
                    sleep(3);
                }
            }
            while(1){ //不断读取
                memset(readBuf,0,sizeof(readBuf));
                //5.read
                n_read = read(c_fd,readBuf,128);
                if(n_read == -1){
                    perror("read");
                }else{
                    printf("get message:%s\n",readBuf);
                }
            }
            break;
        }
    }
    return 0;
}

客户端

#include <stdio.h>
#include <sys/types.h>          /* See NOTES */
#include <sys/socket.h>
#include <stdlib.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <string.h>
#include <unistd.h>

int main(int argc,char *argv[])
{
    if(argc != 3){
        printf("参数错误\n");
        exit(-1);
    }

    int c_fd;
    int n_read;
    char readBuf[128];
    char msg[128] = {0};

    struct sockaddr_in c_addr;

    memset(&c_addr,0,sizeof(struct sockaddr_in));

    //1.socket
    c_fd = socket(AF_INET,SOCK_STREAM,0);
    if(c_fd == -1){
        perror("socket");
        exit(-1);
    }
    c_addr.sin_family = AF_INET;
    c_addr.sin_port = htons(atoi(argv[2]));
    inet_aton(argv[1],&c_addr.sin_addr);
    //2.connect
    if(connect(c_fd,(struct sockaddr * )&c_addr,sizeof(struct sockaddr))== -1){
        perror("connect");        
        exit(-1);
    }
    while(1){
        if(fork() == 0){
            while(1){ //不断发送
                memset(msg,0,sizeof(msg));
                printf("input: ");
                fgets(msg,sizeof(msg),stdin);//下面4行代码是解决用fgets函数运行换新行的问题 
                size_t length = strlen(msg);
                if (length > 0 && msg[length - 1] == '\n') {
                    msg[length - 1] = '\0';
                }
                //3.send
                write(c_fd,msg,strlen(msg));
            }
        }
        while(1){ //不断读取
            memset(readBuf,0,sizeof(readBuf));
            //4.read
            n_read = read(c_fd,readBuf,128);
            if(n_read == -1){
                perror("read");
                exit(-1);
            }else{
                printf("get message from server:%s\n",readBuf);
            }   
        }
    }

    return 0;
}

二、项目展示