Linux系统编程之socket编程

• 实现一个基于socket网络的客户端和服务端的双向收发消息；

• 客户端和服务端都会去fork一个子进程，子进程负责去发送消息，父进程负责去读取消息；

一、服务端

1. server端的流程

2. server端的demo

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

int main(int argc, char **argv)
{
    struct sockaddr_in s_addr;
    struct sockaddr_in c_addr;
    int s_fd;
    int c_fd;      
    char msg[128] = {0};
    char readBuf[128];
    int n_read;
   
    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]));
                            //端口号，5000～9000之间，
                            //调用htons返回网络字节序的值
    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);//监听10个连接
    
    //4. accept
    int c_len = sizeof(struct sockaddr_in);
    while(1){
        c_fd = accept(s_fd, (struct sockaddr *)&c_addr, &c_len);
        if(c_fd == -1){
            perror("accept");
        }
        printf("get connect : %s\n",inet_ntoa(c_addr.sin_addr));
        
        if(fork() == 0){
            while(1){
                memset(msg, 0, sizeof(msg));
                fgets(msg, 128, stdin);
                write(c_fd, msg, strlen(msg));
            }
        }
        while(1){
            n_read = read(c_fd, readBuf, 128);
            if(n_read == -1){
                perror("read");
            }
            else{
                printf("get message from client : %s",readBuf);
            }
        }    
        break;
    }
    return 0;
}

二、客户端

1. client端的流程

2. client的demo

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

int main(int argc, char **argv)
{
    struct sockaddr_in c_addr;
    char msg[128] = {0};  
    char readBuf[128];
    int n_read;
    int c_fd;

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

    //1. socket
    c_fd = socket(AF_INET, SOCK_STREAM, 0);//选择IPV4的协议
                                           //使用TCP/IP的socket协议
                                           //自动配合前两项协议                    
    if(c_fd == -1){
        perror("socket");
        exit(-1);
    }
  
    //2. connect
    c_addr.sin_family = AF_INET;
    c_addr.sin_port = htons(atoi(argv[2]));
    inet_aton(argv[1], &c_addr.sin_addr);
    if(!connect(c_fd,(struct sockaddr *)&c_addr,sizeof(struct sockaddr_in))){
        perror("connect");
        exit(-1);
    }
    while(1){
        //3. send
        if(fork() == 0){
            while(1){
                memset(msg, 0, sizeof(msg));
                fgets(msg, 128, stdin);
                write(c_fd, msg, strlen(msg));
            }
        }
        //4. read
        while(1){
            n_read = read(c_fd, readBuf,128);
            if(n_read == -1){
                perror("read");
            }
            else{
                printf("get message from server : %s",readBuf);
            }    
        }
    }
    return 0;
}

三、双方的运行效果