使用 XFRM_MSG_NEWSA 在用户层添加 SA 的 C 代码示例

下面是一个简单的 C 代码示例，用于在用户层使用 XFRM_MSG_NEWSA 新增 SA：

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/socket.h>
#include <linux/xfrm.h>

// 定义 XFRM_MSG_NEWSA 消息结构体
struct xfrm_userpolicy_info sa_info = {
    .sel.family = AF_INET6,
    .sel.sport = 0,
    .sel.dport = 0,
    .sel.prefixlen_s = 0,
    .sel.prefixlen_d = 0,
    .sel.saddr.a6 = {0},
    .sel.daddr.a6 = {0},
    .dir = XFRM_POLICY_OUT,
    .action = XFRM_POLICY_ALLOW,
    .share = XFRM_SHARE_UNIQUE,
    .sel.family = AF_INET6,
    .index = 0,
    .type = XFRM_POLICY_TYPE_MAIN,
    .id.daddr.a6 = {0},
    .id.spi = 0,
    .priority = 0,
    .sel.sport = 0,
    .sel.dport = 0,
    .sel.saddr.a6 = {0},
    .sel.daddr.a6 = {0},
    .lft.soft_add_expires_seconds = 0,
    .lft.soft_bytes = 0,
    .lft.hard_add_expires_seconds = 0,
    .lft.hard_bytes = 0,
    .flags = 0
};

int main() {
    int sock_fd = socket(PF_NETLINK, SOCK_RAW, NETLINK_XFRM);
    if (sock_fd < 0) {
        perror("socket");
        exit(1);
    }
    // 绑定到 XFRM Netlink Socket
    struct sockaddr_nl local = {
        .nl_family = AF_NETLINK,
        .nl_groups = XFRMNLGRP_NONE,
        .nl_pid = getpid()
    };
    if (bind(sock_fd, (struct sockaddr*)&local, sizeof(local)) < 0) {
        perror("bind");
        exit(1);
    }
    // 构造 XFRM_MSG_NEWSA 消息
    struct nlmsghdr *nlh = NULL;
    struct msghdr msg = {0};
    struct iovec iov = {0};
    char buf[1024] = {0};
    nlh = (struct nlmsghdr *)buf;
    nlh->nlmsg_len = NLMSG_LENGTH(sizeof(struct xfrm_userpolicy_info));
    nlh->nlmsg_flags = NLM_F_REQUEST | NLM_F_CREATE;
    nlh->nlmsg_type = XFRM_MSG_NEWSA;
    nlh->nlmsg_seq = 0;
    nlh->nlmsg_pid = getpid();
    memcpy(NLMSG_DATA(nlh), &sa_info, sizeof(struct xfrm_userpolicy_info));
    iov.iov_base = nlh;
    iov.iov_len = nlh->nlmsg_len;
    msg.msg_iov = &iov;
    msg.msg_iovlen = 1;
    // 发送 XFRM_MSG_NEWSA 消息
    if (sendmsg(sock_fd, &msg, 0) < 0) {
        perror("sendmsg");
        exit(1);
    }
    close(sock_fd);
    return 0;
}

这段代码中，我们首先创建了一个 XFRM Netlink Socket，并将其绑定到本地地址。然后，我们构造了一个 XFRM_MSG_NEWSA 消息，将其发送到内核，并使用 XFRM_MSG_NEWSA 消息中指定的参数创建了一个新的 SA。请注意，这里的 SA_INFO 结构体是一个示例，您需要根据实际需求填充更多的字段。您还需要确保在编译时链接 -lxfrm 库。