使用netlink给指定的接口配置IP地址

下面的代码是用国内的chatgpt镜像地址写出的代码,可以编译通过,但是运行错误,大家看看:

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h>
#include <unistd.h>
#include <sys/socket.h>
#include <linux/netlink.h>
#include <linux/rtnetlink.h>
#include <arpa/inet.h>
#include <net/if.h>

#define BUF_SIZE 4096

int main(int argc, char **argv) {
    int ifIndex;
    struct in_addr addr;
    uint32_t prefixLen;

    if (argc != 3) {
        fprintf(stderr, "Usage: %s <ifname> <address/prefixlen>\n", argv[0]);
        exit(EXIT_FAILURE);
    }

    char *ifName = argv[1];
    char *addrStr = strtok(argv[2], "/");
    char *prefixLenStr = strtok(NULL, "/");


    if (inet_pton(AF_INET, addrStr, &addr) != 1) {
        fprintf(stderr, "Invalid IPv4 address: %s\n", addrStr);
        exit(EXIT_FAILURE);
    }

    prefixLen = atoi(prefixLenStr);

    ifIndex = if_nametoindex(ifName);
    if (ifIndex == 0) {
        fprintf(stderr, "Invalid interface name: %s\n", ifName);
        exit(EXIT_FAILURE);
    }

    struct sockaddr_nl srcAddr = {0};
    srcAddr.nl_family = AF_NETLINK;

    struct msghdr msg = {0};
    char buf[BUF_SIZE] = {0};

    struct nlmsghdr *nlh = (struct nlmsghdr *)buf;
    nlh->nlmsg_len = NLMSG_LENGTH(sizeof(struct ifaddrmsg));
    nlh->nlmsg_type = RTM_NEWADDR;
    nlh->nlmsg_flags = NLM_F_REQUEST | NLM_F_ACK;
    nlh->nlmsg_seq = 1;
    nlh->nlmsg_pid = getpid();

    struct ifaddrmsg *ifa = (struct ifaddrmsg *)NLMSG_DATA(nlh);
    ifa->ifa_family = AF_INET;
    ifa->ifa_prefixlen = prefixLen;
    ifa->ifa_index = ifIndex;
    ifa->ifa_flags = IFA_F_PERMANENT | IFA_F_SECONDARY;


    struct rtattr *rta = (struct rtattr *)((char *)ifa + sizeof(struct ifaddrmsg));
    rta->rta_len = RTA_LENGTH(4);
    rta->rta_type = IFA_LOCAL;
    memcpy(RTA_DATA(rta), &addr, 4);

    nlh->nlmsg_len = NLMSG_ALIGN(nlh->nlmsg_len) + rta->rta_len;

    msg.msg_name = (void *)&srcAddr;
    msg.msg_namelen = sizeof(srcAddr);
    msg.msg_iov = (struct iovec *)buf;
    msg.msg_iovlen = 1;

    int sockFd = socket(AF_NETLINK, SOCK_RAW, NETLINK_ROUTE);
    if (sockFd == -1) {
        perror("socket");
        exit(EXIT_FAILURE);
    }

    if (bind(sockFd, (struct sockaddr *)&srcAddr, sizeof(srcAddr)) == -1) {
        perror("bind");
        close(sockFd);
        exit(EXIT_FAILURE);
    }

    if (sendmsg(sockFd, &msg, 0) == -1) {
        perror("sendmsg");
        close(sockFd);
        exit(EXIT_FAILURE);
    }

    memset(buf, 0, BUF_SIZE);

    while (1) {
        if (recv(sockFd, buf, BUF_SIZE, 0) == -1) {
            perror("recv");
            close(sockFd);
            exit(EXIT_FAILURE);
        }

        struct nlmsghdr *nlhRecv = (struct nlmsghdr *)buf;

        while (NLMSG_OK(nlhRecv, BUF_SIZE)) {
            switch (nlhRecv->nlmsg_type) {
                case NLMSG_ERROR:
                    fprintf(stderr, "Error: %s\n", strerror(-nlhRecv->nlmsg_len));
                    close(sockFd);
                    exit(EXIT_FAILURE);
                    break;
                case RTM_NEWADDR:
                    printf("IPv4 address configured successfully\n");
                    close(sockFd);
                    exit(EXIT_SUCCESS);
                    break;
                default:
                    fprintf(stderr, "Unknown message type: %d\n", nlhRecv->nlmsg_type);
                    break;
            }

            nlhRecv = (struct nlmsghdr *)((char *)nlhRecv + NLMSG_ALIGN(nlhRecv->nlmsg_len));
        }
    }

    return 0;
}

 

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值