UNIX DGRAM socket实例

服务端

#include <sys/socket.h>
#include <sys/un.h> 
#include <linux/types.h>
#include <sys/types.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <string.h>
#include <stdlib.h>
#include <stdio.h>
#include <signal.h>
#include <stddef.h>
#include <ctype.h> 
 
#define SRC_ADDR "/var/run/uds_test.socket"

static int serve_new_conn(int sk)
{
    int rd, wr;
    char recv_buf[1024];
    struct sockaddr_un client;
 
    printf("New connection\n");
 
    while (1) {
        rd = recvfrom(sk, recv_buf, sizeof(recv_buf), 0, NULL, NULL);
        if (!rd)
            break;
 
        if (rd < 0) {
            perror("Can't read socket");
            return 1;
        }

        recv_buf[rd]='\0';
 
        wr = 0;
        wr = sendto(sk, recv_buf, sizeof(recv_buf), 0, (struct sockaddr *)&client, sizeof(client));
        if (wr <= 0) {
            perror("Can't write socket");
            return 1;
        }
    }
 
    printf("Done\n");
    return 0;
}

int main(int argc, char **argv)
{
    int sk, port, size, ret;
 
    struct sockaddr_un addr, cliun; 
    socklen_t cliun_len;
    memset(&addr, 0, sizeof(addr)); 
    addr.sun_family = AF_UNIX; 
    strcpy(addr.sun_path, SRC_ADDR);
    size = offsetof(struct sockaddr_un, sun_path) + strlen(addr.sun_path); 
    /*
     * Let kids die themselves
     */
  
    signal(SIGCHLD, SIG_IGN);
  
    sk = socket(AF_UNIX, SOCK_DGRAM, 0);
    if (sk < 0) {
        perror("Can't create socket");
        return -1;
    }
  
    printf("Binding to port %s\n", SRC_ADDR);
  
    unlink(SRC_ADDR);
     
    ret = bind(sk, (struct sockaddr *)&addr, size);
    if (ret < 0) {
        perror("Can't bind socket");
        return -1;
    }
    
    printf("Waiting for connections\n");
    ret = serve_new_conn(sk);
    exit(ret);
}

客户端

#include <sys/socket.h>
#include <sys/un.h> 
#include <linux/types.h>
#include <sys/types.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <string.h>
#include <stdlib.h>
#include <stdio.h>
#include <signal.h>
#include <stddef.h>
#include <ctype.h> 

#define DST_ADDR "/var/run/uds_test.socket"

int main(int argc, char **argv)
{
    int sk, port, ret, val = 1, rval = 0, len = 0;
    struct sockaddr_un addr,cliun;
  
    sk = socket(AF_UNIX, SOCK_DGRAM, 0);
    if (sk < 0) {
        perror("Can't create socket");
        return -1;
    }
 
    // 一般显式调用bind函数，以便服务器区分不同客户端 
    // criu 需要读取此文件
    memset(&cliun, 0, sizeof(cliun)); 
    cliun.sun_family = AF_UNIX; 
    strcpy(cliun.sun_path, DST_ADDR); 
    len = offsetof(struct sockaddr_un, sun_path) + strlen(cliun.sun_path); 
    unlink(cliun.sun_path); 
    if (bind(sk, (struct sockaddr *)&cliun, len) < 0) { 
        perror("bind error"); 
        return -1;
    } 
     
    port = atoi(argv[2]);
    printf("Connecting to %s\n", DST_ADDR);
 
    memset(&addr, 0, sizeof(addr)); 
    addr.sun_family = AF_UNIX; 
    strcpy(addr.sun_path, DST_ADDR);
     
    ret = connect(sk, (struct sockaddr *)&addr, sizeof(addr));
    if (ret < 0) {
        perror("Can't connect");
        return -1;
    }
  
    while (1) {
        ret=sendto(sk, &val, sizeof(val), 0, (struct sockaddr *)&addr, sizeof(addr));
        if (ret < 0) {
            perror("send data error\n");
            close(sk);
            exit(1);
        }

        ret=recvfrom(sk, &rval, sizeof(rval), 0, NULL, NULL);
        if (ret < 0) {
            perror("recv data error\n");
            close(sk);
            exit(1);
        }
        printf("PP %d -> %d\n", val, rval);
        sleep(2);
        val++;
    }
}