1/* $NetBSD: srtconfig.c,v 1.2 2006/12/29 10:39:59 wiz Exp $ */ 2/* This file is in the public domain. */ 3 4#include <stdio.h> 5#include <errno.h> 6#include <fcntl.h> 7#include <stdlib.h> 8#include <strings.h> 9#include <sys/ioctl.h> 10#include <arpa/inet.h> 11#include <net/if_srt.h> 12 13extern const char *__progname; 14 15#define ACT_ERROR 1 /* none of the below */ 16#define ACT_QUERYALL 2 /* srtX */ 17#define ACT_QUERYONE 3 /* srtX N */ 18#define ACT_DEL 4 /* srtX del N */ 19#define ACT_ADD 5 /* srtX add srcaddr mask dstif dstaddr */ 20#define ACT_SET 6 /* srtX set N srcaddr mask dstif dstaddr */ 21#define ACT_FLAGS 7 /* srtX flags */ 22#define ACT_SFLAG 8 /* srtX flags [+|-]flag */ 23#define ACT_DEBUG 9 /* srtX debug */ 24static int action = ACT_ERROR; 25 26static char *txt_dev; 27static char *txt_n; 28static char *txt_flg; 29static char *txt_addr; 30static char *txt_mask; 31static char *txt_dstif; 32static char *txt_dstaddr; 33 34static int devfd; 35 36static struct { 37 const char *name; 38 unsigned int bit; 39 } flagbits[] = { { "mtulock", SSF_MTULOCK }, 40 { 0, 0 } }; 41 42static void handleargs(int ac, char **av) 43{ 44 txt_dev = av[1]; 45 if (ac == 2) 46 { action = ACT_QUERYALL; 47 } 48 else if ((ac == 3) && !strcmp(av[2],"debug")) 49 { action = ACT_DEBUG; 50 } 51 else if ((ac == 3) && !strcmp(av[2],"flags")) 52 { action = ACT_FLAGS; 53 } 54 else if (ac == 3) 55 { action = ACT_QUERYONE; 56 txt_n = av[2]; 57 } 58 else if ((ac == 4) && !strcmp(av[2],"del")) 59 { action = ACT_DEL; 60 txt_n = av[3]; 61 } 62 else if ((ac == 4) && !strcmp(av[2],"flags")) 63 { action = ACT_SFLAG; 64 txt_flg = av[3]; 65 } 66 else if ((ac == 7) && !strcmp(av[2],"add")) 67 { action = ACT_ADD; 68 txt_addr = av[3]; 69 txt_mask = av[4]; 70 txt_dstif = av[5]; 71 txt_dstaddr = av[6]; 72 } 73 else if ((ac == 8) && !strcmp(av[2],"set")) 74 { action = ACT_SET; 75 txt_n = av[3]; 76 txt_addr = av[4]; 77 txt_mask = av[5]; 78 txt_dstif = av[6]; 79 txt_dstaddr = av[7]; 80 } 81 if (action == ACT_ERROR) 82 { fprintf(stderr,"Usage: %s srtX\n",__progname); 83 fprintf(stderr," %s srtX N\n",__progname); 84 fprintf(stderr," %s srtX del N\n",__progname); 85 fprintf(stderr," %s srtX add addr mask dstif dstaddr\n",__progname); 86 fprintf(stderr," %s srtX set N addr mask dstif dstaddr\n",__progname); 87 fprintf(stderr," %s srtX flags {[+|-]flag}\n",__progname); 88 fprintf(stderr," %s srtX debug\n",__progname); 89 exit(1); 90 } 91} 92 93static void open_dev(int how) 94{ 95 if (! index(txt_dev,'/')) 96 { char *tmp; 97 asprintf(&tmp,"/dev/%s",txt_dev); 98 txt_dev = tmp; 99 } 100 devfd = open(txt_dev,how,0); 101 if (devfd < 0) 102 { fprintf(stderr,"%s; can't open %s: %s\n",__progname,txt_dev,strerror(errno)); 103 exit(1); 104 } 105} 106 107static void query_n(int n) 108{ 109 struct srt_rt r; 110 char obuf[64]; 111 112 r.inx = n; 113 if (ioctl(devfd,SRT_GETRT,&r) < 0) 114 { fprintf(stderr,"%s: can't get rt #%d: %s\n",__progname,n,strerror(errno)); 115 return; 116 } 117 printf("%d:",n); 118 printf(" %s",inet_ntop(r.af,&r.srcmatch,&obuf[0],sizeof(obuf))); 119 printf(" /%d",r.srcmask); 120 printf(" %.*s",(int)sizeof(r.u.dstifn),&r.u.dstifn[0]); 121 switch (r.af) 122 { case AF_INET: 123 printf(" %s",inet_ntoa(r.dst.sin.sin_addr)); 124 break; 125 case AF_INET6: 126 printf(" %s",inet_ntop(AF_INET6,&r.dst.sin6.sin6_addr,&obuf[0],sizeof(obuf))); 127 break; 128 default: 129 printf(" ?af%d",r.af); 130 break; 131 } 132 printf("\n"); 133} 134 135static void do_query(int narg) 136{ 137 int i; 138 int n; 139 140 open_dev(O_RDONLY); 141 if (narg >= 0) 142 { query_n(narg); 143 } 144 else 145 { if (ioctl(devfd,SRT_GETNRT,&n) < 0) 146 { fprintf(stderr,"%s: can't get count: %s\n",__progname,strerror(errno)); 147 exit(1); 148 } 149 for (i=0;i<n;i++) query_n(i); 150 } 151} 152 153static void do_del(unsigned int n) 154{ 155 open_dev(O_RDWR); 156 if (ioctl(devfd,SRT_DELRT,&n) < 0) 157 { fprintf(stderr,"%s: can't delete #%u: %s\n",__progname,n,strerror(errno)); 158 exit(1); 159 } 160} 161 162static void do_set(int n) 163{ 164 struct srt_rt r; 165 int w; 166 int maxw; 167 void *dp; 168 169 open_dev(O_RDWR); 170 if (n < 0) 171 { unsigned int v; 172 if (ioctl(devfd,SRT_GETNRT,&v) < 0) 173 { fprintf(stderr,"%s: can't get count: %s\n",__progname,strerror(errno)); 174 exit(1); 175 } 176 n = v; 177 } 178 bzero(&r.dst,sizeof(r.dst)); 179 r.inx = n; 180 if (inet_pton(AF_INET,txt_addr,&r.srcmatch.v4) == 1) 181 { r.af = AF_INET; 182 r.dst.sin.sin_family = AF_INET; 183 r.dst.sin.sin_len = sizeof(r.dst.sin); 184 dp = &r.dst.sin.sin_addr; 185 maxw = 32; 186 } 187 else if (inet_pton(AF_INET6,txt_addr,&r.srcmatch.v6) == 1) 188 { r.af = AF_INET6; 189 r.dst.sin6.sin6_family = AF_INET6; 190 r.dst.sin6.sin6_len = sizeof(r.dst.sin6); 191 dp = &r.dst.sin6.sin6_addr; 192 maxw = 128; 193 } 194 else 195 { fprintf(stderr,"%s: %s: invalid match address\n",__progname,txt_addr); 196 exit(1); 197 } 198 if (txt_mask[0] == '/') txt_mask ++; 199 w = atoi(txt_mask); 200 if ((w < 0) || (w > maxw)) 201 { fprintf(stderr,"%s: %s: out-of-range CIDR width\n",__progname,txt_mask); 202 exit(1); 203 } 204 r.srcmask = w; 205 if (strlen(txt_dstif) > sizeof(r.u.dstifn)-1) 206 { fprintf(stderr,"%s: %s: too long\n",__progname,txt_dstif); 207 exit(1); 208 } 209 strncpy(&r.u.dstifn[0],txt_dstif,sizeof(r.u.dstifn)); 210 if (inet_pton(r.af,txt_dstaddr,dp) != 1) 211 { fprintf(stderr,"%s: %s: invalid destination address\n",__progname,txt_dstaddr); 212 exit(1); 213 } 214 if (ioctl(devfd,SRT_SETRT,&r) < 0) 215 { fprintf(stderr,"%s: can't set route: %s\n",__progname,strerror(errno)); 216 exit(1); 217 } 218} 219 220static void do_flags(void) 221{ 222 unsigned int f; 223 int i; 224 225 open_dev(O_RDONLY); 226 if (ioctl(devfd,SRT_GFLAGS,&f) < 0) 227 { fprintf(stderr,"%s: can't get flags: %s\n",__progname,strerror(errno)); 228 exit(1); 229 } 230 for (i=0;flagbits[i].name;i++) 231 { printf(" %c%s",(f&flagbits[i].bit)?'+':'-',flagbits[i].name); 232 f &= ~flagbits[i].bit; 233 } 234 if (f) printf(" +0x%x",f); 235 printf("\n"); 236} 237 238static void do_sflag(void) 239{ 240 unsigned int f; 241 unsigned int b; 242 int i; 243 244 switch (txt_flg[0]) 245 { case '+': case '-': break; 246 default: 247 fprintf(stderr,"%s: last argument must be +flag or -flag\n",__progname); 248 exit(1); 249 break; 250 } 251 for (i=0;flagbits[i].name;i++) if (!strcmp(flagbits[i].name,txt_flg+1)) break; 252 if (! flagbits[i].name) 253 { fprintf(stderr,"%s: unrecognized flag bit `%s'\n",__progname,txt_flg+1); 254 exit(1); 255 } 256 b = flagbits[i].bit; 257 open_dev(O_RDWR); 258 if (ioctl(devfd,SRT_GFLAGS,&f) < 0) 259 { fprintf(stderr,"%s: can't get flags: %s\n",__progname,strerror(errno)); 260 exit(1); 261 } 262 if (txt_flg[0] == '+') f |= b; else f &= ~b; 263 if (ioctl(devfd,SRT_SFLAGS,&f) < 0) 264 { fprintf(stderr,"%s: can't set flags: %s\n",__progname,strerror(errno)); 265 exit(1); 266 } 267} 268 269static void do_debug(void) 270{ 271 void *vp; 272 273 open_dev(O_RDWR); 274 vp = 0; 275 if (ioctl(devfd,SRT_DEBUG,&vp) < 0) 276 { fprintf(stderr,"%s: can't SRT_DEBUG: %s\n",__progname,strerror(errno)); 277 exit(1); 278 } 279} 280 281int main(int, char **); 282int main(int ac, char **av) 283{ 284 handleargs(ac,av); 285 switch (action) 286 { case ACT_QUERYALL: 287 do_query(-1); 288 break; 289 case ACT_QUERYONE: 290 do_query(atoi(txt_n)); 291 break; 292 case ACT_DEL: 293 do_del(atoi(txt_n)); 294 break; 295 case ACT_ADD: 296 do_set(-1); 297 break; 298 case ACT_SET: 299 do_set(atoi(txt_n)); 300 break; 301 case ACT_FLAGS: 302 do_flags(); 303 break; 304 case ACT_SFLAG: 305 do_sflag(); 306 break; 307 case ACT_DEBUG: 308 do_debug(); 309 break; 310 default: 311 abort(); 312 break; 313 } 314 exit(0); 315} 316