diff options
Diffstat (limited to 'libglouglou')
-rw-r--r-- | libglouglou/libggnet.c | 13 | ||||
-rw-r--r-- | libglouglou/libggnet.h | 3 | ||||
-rw-r--r-- | libglouglou/libggnet_traceroute.c | 2 | ||||
-rw-r--r-- | libglouglou/libglouglou.c | 2 | ||||
-rw-r--r-- | libglouglou/libglouglou.h | 2 |
5 files changed, 15 insertions, 7 deletions
diff --git a/libglouglou/libggnet.c b/libglouglou/libggnet.c index 5f9810b..160ac8b 100644 --- a/libglouglou/libggnet.c +++ b/libglouglou/libggnet.c @@ -118,7 +118,7 @@ ggnet_free(struct ggnet *net) } struct ggnet_node * -ggnet_node_add(struct ggnet *net, struct in_addr *addr) +ggnet_node_add(struct ggnet *net, struct in_addr *addr, int allow_grouping) { struct ggnet_node *n; @@ -133,16 +133,17 @@ ggnet_node_add(struct ggnet *net, struct in_addr *addr) n->net = net; n->addr.s_addr = addr->s_addr; n->lastseen = net->time; + //n->allow_grouping = allow_grouping; XXX IN PROGRESS why segfault ?? LIST_INSERT_HEAD(&net->node_list, n, entry); net->node_count++; - if (net->use_grouping) + if (net->use_grouping && allow_grouping) nodegroup_set(net, n); if (net->use_dns) n->dns_req = ggnet_dns_reverse(net->ggdns, &n->addr, _cb_dns_reverse, n); if (net->use_traceroute) { struct addr ip; addr_aton(inet_ntoa(*addr), &ip); - printf("XXX calling trace on %s\n", inet_ntoa(*addr)); + printf("XXX calling trace on %x\n", ip.addr_ip); n->traceroute_req = ggnet_traceroute_trace_tcp(net->ggtraceroute, &ip, 80, _cb_traceroute_trace, NULL, n); } @@ -228,7 +229,7 @@ ggnet_conn_add(struct ggnet *net, struct in_addr *src, int src_port, struct ggnet_node *dstnode; int id; - if (net->debug) + // XXX if (net->debug) printf("ggnet_conn_add, %x:%d->%x:%d %d [%d]\n", src->s_addr, src_port, dst->s_addr, dst_port, proto, size); @@ -245,13 +246,13 @@ ggnet_conn_add(struct ggnet *net, struct in_addr *src, int src_port, srcnode = ggnet_node_find(net, src); if (!srcnode) - srcnode = ggnet_node_add(net, src); + srcnode = ggnet_node_add(net, src, 1); srcnode->used++; if (srcnode->group) srcnode->group->conn_count++; dstnode = ggnet_node_find(net, dst); if (!dstnode) - dstnode = ggnet_node_add(net, dst); + dstnode = ggnet_node_add(net, dst, 1); dstnode->used++; if (dstnode->group) dstnode->group->conn_count++; diff --git a/libglouglou/libggnet.h b/libglouglou/libggnet.h index fecf2bb..95d27d6 100644 --- a/libglouglou/libggnet.h +++ b/libglouglou/libggnet.h @@ -52,6 +52,7 @@ struct ggnet_node { struct ggnet_node *path_parent; /* traceroute parent */ time_t lastseen; int used; /* by ggnet_conn */ + int allow_grouping; struct ggnet_nodegroup *group; /* XXX for now only one group */ char fqdn[GGNET_DNSNAME_MAX]; struct ggnet_dns_req *dns_req; @@ -137,7 +138,7 @@ void ggnet_set_traceroute(struct ggnet *, int, char *, struct ggnet_traceroute_req *)); void ggnet_debug_set(struct ggnet *, int); void ggnet_free(struct ggnet *); -struct ggnet_node *ggnet_node_add(struct ggnet *, struct in_addr *); +struct ggnet_node *ggnet_node_add(struct ggnet *, struct in_addr *, int); void ggnet_node_del(struct ggnet *, struct ggnet_node *); struct ggnet_node *ggnet_node_find(struct ggnet *, struct in_addr *); void *ggnet_node_usrdata_get(struct ggnet_node *); diff --git a/libglouglou/libggnet_traceroute.c b/libglouglou/libggnet_traceroute.c index aca1e32..8c189f0 100644 --- a/libglouglou/libggnet_traceroute.c +++ b/libglouglou/libggnet_traceroute.c @@ -221,6 +221,8 @@ ggnet_traceroute_trace(struct ggnet_traceroute *ggtr, struct addr *ip, req->params = params; TAILQ_INIT(&req->hops_list); + printf("XXX tracing to %s\n", addr_ntoa(&req->target->ip)); + req->run.ev_recv = event_new(ggtr->ev_base, ggtr->pcap_fd, EV_READ, _cb_recv, req); event_add(req->run.ev_recv, NULL); diff --git a/libglouglou/libglouglou.c b/libglouglou/libglouglou.c index 9f41313..d17137a 100644 --- a/libglouglou/libglouglou.c +++ b/libglouglou/libglouglou.c @@ -569,6 +569,7 @@ pkt_decode(char **buf, int *buf_len) break; case PACKET_TRACE: newpkt.trace_addr = ntohl(pkt->trace_addr); + newpkt.trace_src = ntohl(pkt->trace_src); newpkt.trace_hopcount = pkt->trace_hopcount; if (newpkt.trace_hopcount > GG_PKTARG_MAX) invalid("type trace hopcount"); @@ -666,6 +667,7 @@ pkt_encode(struct gg_packet *pkt, struct gg_packet *newpkt) if (pkt->trace_hopcount > GG_PKTARG_MAX) goto invalid; newpkt->trace_addr = htonl(pkt->trace_addr); + newpkt->trace_src = htonl(pkt->trace_src); newpkt->trace_hopcount = pkt->trace_hopcount; for (i=0; i<newpkt->trace_hopcount; i++) { newpkt->trace_hops[i].addr = htonl(pkt->trace_hops[i].addr); diff --git a/libglouglou/libglouglou.h b/libglouglou/libglouglou.h index ac57595..c5fb252 100644 --- a/libglouglou/libglouglou.h +++ b/libglouglou/libglouglou.h @@ -74,6 +74,7 @@ struct __attribute__((packed)) gg_packet { } name; struct __attribute__((packed)) trace { u_int32_t addr; + u_int32_t src; u_int8_t hopcount; struct gg_packet_tracehop hops[GG_PKTARG_MAX]; } trace; @@ -106,6 +107,7 @@ struct __attribute__((packed)) gg_packet { #define name_len pdat.name.len #define name_fqdn pdat.name.fqdn #define trace_addr pdat.trace.addr +#define trace_src pdat.trace.src #define trace_hopcount pdat.trace.hopcount #define trace_hops pdat.trace.hops #define fork_pid pdat.fork.pid |