1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3 * Copyright (c) 2022 Huawei Device Co., Ltd.
4 *
5 * Description: Demo example of configuring NewIP route.
6 *
7 * Author: Yang Yanjun <yangyanjun@huawei.com>
8 *
9 * Data: 2022-09-06
10 */
11 #include <sys/socket.h>
12 #include <sys/ioctl.h>
13 #include <errno.h>
14 #include <string.h>
15 #include <stdio.h>
16 #include <unistd.h>
17 #include <stdint.h>
18 #include <linux/route.h>
19
20 #include "nip_uapi.h"
21 #include "nip_lib.h"
22 #include "newip_route.h"
23
24 /* get ifindex based on the device name
25 * struct ifreq ifr;
26 * struct nip_ifreq ifrn;
27 * ioctl(fd, SIOGIFINDEX, &ifr);
28 * ifr.ifr_ifindex; ===> ifindex
29 */
nip_route_add(int ifindex,const unsigned char * dst_addr,uint8_t dst_addr_len,const unsigned char * gateway_addr,uint8_t gateway_addr_len)30 int nip_route_add(int ifindex, const unsigned char *dst_addr, uint8_t dst_addr_len,
31 const unsigned char *gateway_addr, uint8_t gateway_addr_len)
32 {
33 int fd, ret;
34 struct nip_rtmsg rt;
35
36 fd = socket(AF_NINET, SOCK_DGRAM, 0);
37 if (fd < 0)
38 return -1;
39
40 memset(&rt, 0, sizeof(rt));
41 rt.rtmsg_ifindex = ifindex;
42 rt.rtmsg_flags = RTF_UP;
43 rt.rtmsg_dst.bitlen = dst_addr_len * BITS_PER_BYTE;
44 memcpy(rt.rtmsg_dst.NIP_ADDR_FIELD8, dst_addr, dst_addr_len);
45
46 if (gateway_addr) {
47 rt.rtmsg_gateway.bitlen = gateway_addr_len * BITS_PER_BYTE;
48 memcpy(rt.rtmsg_gateway.NIP_ADDR_FIELD8, gateway_addr, gateway_addr_len);
49 rt.rtmsg_flags |= RTF_GATEWAY;
50 }
51
52 ret = ioctl(fd, SIOCADDRT, &rt);
53 if (ret < 0 && errno != EEXIST) { // ignore File Exists error
54 close(fd);
55 return -1;
56 }
57
58 close(fd);
59 return 0;
60 }
61
main(int argc,char ** argv)62 int main(int argc, char **argv)
63 {
64 int ret;
65 int ifindex = 0;
66 uint8_t client_addr[INDEX_1] = {0x50}; // 1-byte address of the client: 0x50
67 uint8_t server_addr[INDEX_2] = {0xDE, 0x00}; // 2-byte address of the server: 0xDE00
68 uint8_t *dst_addr;
69 uint8_t dst_addr_len;
70
71 if (argc == DEMO_INPUT_1) {
72 if (!strcmp(*(argv + 1), "server")) {
73 printf("server cfg route, dst-addr=0x%02x\n", client_addr[INDEX_0]);
74 dst_addr = client_addr;
75 dst_addr_len = 1;
76 } else if (!strcmp(*(argv + 1), "client")) {
77 printf("client cfg route, dst-addr=0x%02x%02x\n",
78 server_addr[INDEX_0], server_addr[INDEX_1]);
79 dst_addr = server_addr;
80 dst_addr_len = INDEX_2;
81 } else {
82 printf("invalid route cfg input\n");
83 return -1;
84 }
85 } else {
86 printf("unsupport route cfg input\n");
87 return -1;
88 }
89
90 ret = nip_get_ifindex(NIC_NAME, &ifindex);
91 if (ret != 0) {
92 printf("get %s ifindex fail, ret=%d\n", NIC_NAME, ret);
93 return -1;
94 }
95
96 ret = nip_route_add(ifindex, dst_addr, dst_addr_len, NULL, 0);
97 if (ret != 0) {
98 printf("get %s ifindex fail, ret=%d\n", NIC_NAME, ret);
99 return -1;
100 }
101
102 printf("%s %s(ifindex=%d) cfg route success\n", *argv, NIC_NAME, ifindex);
103 return 0;
104 }
105
106