platform_system_core/toolbox/route.c
Chung-yih Wang f309f77adc AI 146558: am: CL 146260 Add the net-based routing entries.
e.g. route add net 172.16.0.0 mask 255.240.0.0 gw 172.19.248.10
  Original author: cywang

Automated import of CL 146558
2009-04-18 16:16:21 -07:00

126 lines
3.7 KiB
C

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <stdarg.h>
#include <errno.h>
#include <string.h>
#include <ctype.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <linux/if.h>
#include <linux/sockios.h>
#include <arpa/inet.h>
#include <linux/route.h>
static void die(const char *fmt, ...)
{
va_list p;
va_start(p, fmt);
fprintf(stderr,"error(%s): ", strerror(errno));
fprintf(stderr, fmt, p);
va_end(p);
exit(-1);
}
static inline void init_sockaddr_in(struct sockaddr_in *sin, const char *addr)
{
sin->sin_family = AF_INET;
sin->sin_port = 0;
sin->sin_addr.s_addr = inet_addr(addr);
}
#define ADVANCE(argc, argv) do { argc--, argv++; } while(0)
#define EXPECT_NEXT(argc, argv) do { \
ADVANCE(argc, argv); \
if (0 == argc) { \
errno = EINVAL; \
die("expecting one more argument"); \
} \
} while(0)
/* current support the following routing entries */
/* route add default dev wlan0 */
/* route add default gw 192.168.20.1 dev wlan0 */
/* route add net 192.168.1.1 netmask 255.255.255.0 gw 172.24.192.10 */
int route_main(int argc, char *argv[])
{
struct ifreq ifr;
int s,i;
struct rtentry rt;
struct sockaddr_in ina;
if(argc == 0) return 0;
strncpy(ifr.ifr_name, argv[0], IFNAMSIZ);
ifr.ifr_name[IFNAMSIZ-1] = 0;
ADVANCE(argc, argv);
if((s = socket(AF_INET, SOCK_DGRAM, 0)) < 0) {
die("cannot open control socket\n");
}
while(argc > 0){
if(!strcmp(argv[0], "add")) {
EXPECT_NEXT(argc, argv);
if(!strcmp(argv[0], "default")) {
EXPECT_NEXT(argc, argv);
memset((char *) &rt, 0, sizeof(struct rtentry));
rt.rt_dst.sa_family = AF_INET;
if(!strcmp(argv[0], "dev")) {
EXPECT_NEXT(argc, argv);
rt.rt_flags = RTF_UP | RTF_HOST;
rt.rt_dev = argv[0];
if (ioctl(s, SIOCADDRT, &rt) < 0) die("SIOCADDRT\n");
}else if(!strcmp(argv[0], "gw")) {
EXPECT_NEXT(argc, argv);
rt.rt_flags = RTF_UP | RTF_GATEWAY;
init_sockaddr_in((struct sockaddr_in *)&(rt.rt_genmask), "0.0.0.0");
if(isdigit(argv[0][0])){
init_sockaddr_in((struct sockaddr_in *)&(rt.rt_gateway), argv[0]);
}else{
die("expecting an IP address for parameter \"gw\"\n");
}
EXPECT_NEXT(argc, argv);
if(!strcmp(argv[0], "dev")) {
EXPECT_NEXT(argc, argv);
rt.rt_dev = argv[0];
if (ioctl(s, SIOCADDRT, &rt) < 0){
die("SIOCADDRT\n");
}
}
}
} else {
char keywords[3][5] = { "net", "mask", "gw" };
struct sockaddr_in *paddr[3] = { &rt.rt_dst, &rt.rt_genmask, &rt.rt_gateway };
int k = 0;
memset((char *) &rt, 0, sizeof(struct rtentry));
rt.rt_flags = RTF_UP | RTF_GATEWAY;
do {
if(!strcmp(argv[0], keywords[k])) {
EXPECT_NEXT(argc, argv);
if(isdigit(argv[0][0])) {
init_sockaddr_in(paddr[k], argv[0]);
} else {
die("expecting an IP/MASK address for parameter %s\n", keywords[k]);
}
if(k < 2) EXPECT_NEXT(argc, argv);
} else {
die("expecting keyword(s)\n");
}
} while(++k < 3);
if(ioctl(s, SIOCADDRT, &rt) < 0) {
die("SIOCADDRT\n");
}
}
}
ADVANCE(argc, argv);
}
return 0;
}