/*- * Copyright (c) 2004-2005 Gleb Smirnoff * Copyright (c) 2001-2003 Roman V. Palagin * All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions * are met: * 1. Redistributions of source code must retain the above copyright * notice, this list of conditions and the following disclaimer. * 2. Redistributions in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer in the * documentation and/or other materials provided with the distribution. * * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF * SUCH DAMAGE. * * $SourceForge: flowctl.c,v 1.15 2004/08/31 20:24:58 glebius Exp $ */ #ifndef lint static const char rcs_id[] = "@(#) $FreeBSD$"; #endif #include #include #include #include #include #include #include #include #include #include #include #include #include #include #define CISCO_SH_FLOW_HEADER "SrcIf SrcIPaddress DstIf DstIPaddress Pr SrcP DstP Pkts\n" #define CISCO_SH_FLOW "%-13s %-15s %-13s %-15s %2u %4.4x %4.4x %6lu\n" #define CISCO_SH_VERB_FLOW_HEADER "SrcIf SrcIPaddress DstIf DstIPaddress Pr TOS Flgs Pkts\n" \ "Port Msk AS Port Msk AS NextHop B/Pk Active\n" #define CISCO_SH_VERB_FLOW "%-14s %-15s %-14s %-15s %2u %3x %4x %6lu\n" \ "%4.4x /%-2u %-5u %4.4x /%-2u %-5u %-15s %9u %8u\n\n" static int flow_cache_print(struct ngnf_flows *recs); static int flow_cache_print_verbose(struct ngnf_flows *recs); static int ctl_show(int, char **); static void help(void); static void execute_command(int, char **); struct ip_ctl_cmd { char *cmd_name; int (*cmd_func)(int argc, char **argv); }; struct ip_ctl_cmd cmds[] = { {"show", ctl_show}, {NULL, NULL}, }; int cs; char ng_nodename[NG_PATHSIZ]; int main(int argc, char **argv) { int c; char sname[NG_NODESIZ]; int rcvbuf = SORCVBUF_SIZE; char *ng_name; /* parse options */ while ((c = getopt(argc, argv, "d:")) != -1) { switch (c) { case 'd': /* set libnetgraph debug level. */ NgSetDebug(atoi(optarg)); break; } } argc -= optind; argv += optind; ng_name = argv[0]; if (ng_name == NULL) help(); argc--; argv++; snprintf(ng_nodename, sizeof(ng_nodename), "%s:", ng_name); /* create control socket. */ snprintf(sname, sizeof(sname), "flowctl%i", getpid()); if (NgMkSockNode(sname, &cs, NULL) == -1) err(1, "NgMkSockNode"); /* set receive buffer size */ if (setsockopt(cs, SOL_SOCKET, SO_RCVBUF, &rcvbuf, sizeof(int)) == -1) err(1, "setsockopt(SOL_SOCKET, SO_RCVBUF)"); /* parse and execute command */ execute_command(argc, argv); close(cs); exit(0); } static void execute_command(int argc, char **argv) { int cindex = -1; int i; if (!argc) help(); for (i = 0; cmds[i].cmd_name != NULL; i++) if (!strncmp(argv[0], cmds[i].cmd_name, strlen(argv[0]))) { if (cindex != -1) errx(1, "ambiguous command: %s", argv[0]); cindex = i; } if (cindex == -1) errx(1, "bad command: %s", argv[0]); argc--; argv++; (*cmds[cindex].cmd_func)(argc, argv); } static int ctl_show(int argc, char **argv) { struct ng_mesg *ng_mesg; struct ngnf_flows *data; char path[NG_PATHSIZ]; int token, nread, last = 0; int verbose = 0; if (argc > 0 && !strncmp(argv[0], "verbose", strlen(argv[0]))) verbose = 1; ng_mesg = alloca(SORCVBUF_SIZE); if (verbose) printf(CISCO_SH_VERB_FLOW_HEADER); else printf(CISCO_SH_FLOW_HEADER); for (;;) { /* request set of accounting records */ token = NgSendMsg(cs, ng_nodename, NGM_NETFLOW_COOKIE, NGM_NETFLOW_SHOW, (void *)&last, sizeof(last)); if (token == -1) err(1, "NgSendMsg(NGM_NETFLOW_SHOW)"); /* read reply */ nread = NgRecvMsg(cs, ng_mesg, SORCVBUF_SIZE, path); if (nread == -1) err(1, "NgRecvMsg() failed"); if (ng_mesg->header.token != token) err(1, "NgRecvMsg(NGM_NETFLOW_SHOW): token mismatch"); data = (struct ngnf_flows*)ng_mesg->data; if ((ng_mesg->header.arglen < (sizeof(*data))) || (ng_mesg->header.arglen < (sizeof(*data) + (data->nentries * sizeof(struct flow_entry_data))))) err(1, "NgRecvMsg(NGM_NETFLOW_SHOW): arglen too small"); if (verbose) (void )flow_cache_print_verbose(data); else (void )flow_cache_print(data); if (data->last != 0) last = data->last; else break; } return (0); } static int flow_cache_print(struct ngnf_flows *recs) { struct flow_entry_data *fle; char src[INET_ADDRSTRLEN], dst[INET_ADDRSTRLEN]; char src_if[IFNAMSIZ], dst_if[IFNAMSIZ]; int i; /* quick check */ if (recs->nentries == 0) return (0); fle = recs->entries; for (i = 0; i < recs->nentries; i++, fle++) { inet_ntop(AF_INET, &fle->r.r_src, src, sizeof(src)); inet_ntop(AF_INET, &fle->r.r_dst, dst, sizeof(dst)); printf(CISCO_SH_FLOW, if_indextoname(fle->fle_i_ifx, src_if), src, if_indextoname(fle->fle_o_ifx, dst_if), dst, fle->r.r_ip_p, ntohs(fle->r.r_sport), ntohs(fle->r.r_dport), fle->packets); } return (i); } static int flow_cache_print_verbose(struct ngnf_flows *recs) { struct flow_entry_data *fle; char src[INET_ADDRSTRLEN], dst[INET_ADDRSTRLEN], next[INET_ADDRSTRLEN]; char src_if[IFNAMSIZ], dst_if[IFNAMSIZ]; int i; /* quick check */ if (recs->nentries == 0) return (0); fle = recs->entries; for (i = 0; i < recs->nentries; i++, fle++) { inet_ntop(AF_INET, &fle->r.r_src, src, sizeof(src)); inet_ntop(AF_INET, &fle->r.r_dst, dst, sizeof(dst)); inet_ntop(AF_INET, &fle->next_hop, next, sizeof(next)); printf(CISCO_SH_VERB_FLOW, if_indextoname(fle->fle_i_ifx, src_if), src, if_indextoname(fle->fle_o_ifx, dst_if), dst, fle->r.r_ip_p, fle->r.r_tos, fle->tcp_flags, fle->packets, ntohs(fle->r.r_sport), fle->src_mask, 0, ntohs(fle->r.r_dport), fle->dst_mask, 0, next, (u_int)(fle->bytes / fle->packets), 0); } return (i); } static void help(void) { extern char *__progname; fprintf(stderr, "usage: %s [-d level] nodename command\n", __progname); exit (0); }