Linux下 UNIX域套接字

在Linux系统中，有很多进程间通信方式，套接字（Socket）就是其中的一种。

但传统的套接字的用法都是基于TCP/IP协议栈的，需要指定IP地址。如果不同主机上的两个进程进行通信，当然这样做没什么问题。但是，如果只需要在一台机器上的两个不同进程间通信，还要用到IP地址就有点大材小用了。

Unix域套接字，专门用来解决这个问题。其API的调用方法基本上和普通TCP/IP的套接字一样，只是有些许差别。

注： UNIX域套接字仅仅复制数据；它们并不执行协议处理，不需要添加或删除网络报头，无需计算检验和，不要产生顺序号，无需发送确认报文。

1. demo - server

#include <stdio.h>
#include <unistd.h>
#include <string.h>
#include <stdlib.h>
#include <errno.h>
#include <sys/types.h>
#include <signal.h>
#include <sys/socket.h>
#include <sys/un.h>


#define UNIX_SOCKET_NAME    "socket_test"

void echo_cli(int sock)
{
    char buf[1024] = {0};
    int ret = 0;

    while(1) {
        ret = read(sock, buf, sizeof(buf));
        if(ret == 0) {
            printf("client %d close...\n", sock);
            break;
        }
        write(sock, buf, strlen(buf));

        memset(buf, 0, sizeof(buf));

    }

    close(sock);


}

int main()
{
    int listenfd = socket(PF_UNIX, SOCK_STREAM, 0);
    if(listenfd < 0) {
        perror("socket error: ");
    }

    unlink(UNIX_SOCKET_NAME);

    struct sockaddr_un serv_addr = {0};
    serv_addr.sun_family = AF_UNIX;
    strncpy(serv_addr.sun_path, UNIX_SOCKET_NAME, sizeof(serv_addr.sun_path));

    int reuse = 1;
    if(setsockopt(listenfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) < 0) {
        perror("setsockopt error: ");
        return 1;
    }

    if(bind(listenfd, (struct sockaddr *)&serv_addr, sizeof(serv_addr)) < 0) {
        perror("bind error: ");
        return 1;
    }

    if(listen(listenfd, SOMAXCONN) < 0) {
        perror("listern error: ");
        return 1;
    }

    pid_t pid = {0};

    while(1) {

        int conn = 0;
        if ((conn = accept(listenfd, NULL, NULL)) < 0) {

            if(errno == EINTR) {
                continue;
            }
            perror("accept error: ");
            break;
        }

        printf("New client connected, conn = %d\n", conn);
        pid = fork();
        if(pid < 0) {
            perror("fork error: ");

        } else if(pid == 0) { //child process
            echo_cli(conn);
            exit(0);

        }

        /* parent process */
        close(conn);
    }

    return 0;
}

2. demo - client

include <stdio.h>
#include <unistd.h>
#include <string.h>
#include <stdlib.h>
#include <error.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/un.h>


#define UNIX_SOCKET_NAME    "socket_test"


void echo_cli(int sock)
{
    char buf1[1024] = {0};
    char buf2[1024] = {0};
    int ret = 0;

    while(fgets(buf1, sizeof(buf1), stdin) != NULL) {
        write(sock, buf1, strlen(buf1));
        ret = read(sock, buf2, sizeof(buf2));
        if(ret == 0) {
            printf("server %d close\n", sock);
            break;
        }

        printf("Get server data: %s\n", buf2);

        memset(buf1, 0, sizeof(buf1));
        memset(buf2, 0, sizeof(buf2));
    }
    close(sock);
}

int main()
{
    int sock = socket(PF_UNIX, SOCK_STREAM, 0);
    if(sock < 0) {
        perror("socket error: \n");
    }

    struct sockaddr_un cli_addr = {0};
    cli_addr.sun_family = AF_UNIX;
    strncpy(cli_addr.sun_path, UNIX_SOCKET_NAME, sizeof(cli_addr.sun_path));

    if(connect(sock, (struct sockaddr *)&cli_addr, sizeof(cli_addr)) < 0) {
        perror("connetc error: ");
    }

    echo_cli(sock);

    return 0;

}

实际上，生成了一个叫socket_test的文件，类似于有名管道