throw usage_error("No interface given");
iface = argv[1];
- char err[1024];
+ char err[PCAP_ERRBUF_SIZE];
+ pcap_if_t *devs;
+ if(pcap_findalldevs(&devs, err)==-1)
+ throw runtime_error(err);
+
pcap = pcap_open_live(iface.c_str(), 128, true, 0, err);
if(!pcap)
throw runtime_error(err);
if(pcap_setnonblock(pcap, true, err)==-1)
throw runtime_error(err);
- unsigned local_addr, local_mask;
- pcap_lookupnet(iface.c_str(), &local_addr, &local_mask, err);
- localnet = Address(ntohl(local_addr));
- localnet.set_mask(Address(ntohl(local_mask)));
+ for(pcap_if_t *d=devs; d; d=d->next)
+ if(iface==d->name)
+ {
+ for(pcap_addr_t *a=d->addresses; a; a=a->next)
+ {
+ if(a->addr->sa_family==AF_INET)
+ {
+ Address addr(ntohl(reinterpret_cast<sockaddr_in *>(a->addr)->sin_addr.s_addr));
+ if(a->netmask)
+ addr.set_mask(Address(ntohl(reinterpret_cast<sockaddr_in *>(a->netmask)->sin_addr.s_addr)));
+ localnets.push_back(addr);
+ }
+ else if(a->addr->sa_family==AF_INET6)
+ {
+ Address addr(reinterpret_cast<sockaddr_in6 *>(a->addr)->sin6_addr);
+ if(a->netmask)
+ addr.set_mask(Address(reinterpret_cast<sockaddr_in6 *>(a->netmask)->sin6_addr));
+ localnets.push_back(addr);
+ }
+ }
+ }
+
+ pcap_freealldevs(devs);
resolver = new Resolver;
return *i->second;
Host *host = new Host(*this, a);
- if(localnet.masked_match(a))
- host->set_local(true);
+ for(list<Address>::const_iterator j=localnets.begin(); j!=localnets.end(); ++j)
+ if(j->masked_match(a))
+ host->set_local(true);
resolver->push(host);
host->set_position(Vector2(rand()*400.0/RAND_MAX-200.0, rand()*400.0/RAND_MAX-200.0));
for(unsigned j=0; j<100; ++j)
if(i!=ports.end())
return *i->second;
Port *port = new Port(*this, number);
+ if(number>=0x10000)
+ {
+ if((number&0xFF)==IPPROTO_ICMP)
+ port->set_name("icmp");
+ else if((number&0xFF)==IPPROTO_ICMPV6)
+ port->set_name("icmp6");
+ }
ports[number] = port;
return *port;
}
const iphdr *ip = reinterpret_cast<const iphdr *>(eth+1);
handle_ipv4(ctx, ip, len-sizeof(ethhdr));
}
+ else if(proto==ETH_P_IPV6)
+ {
+ const ip6_hdr *ip6 = reinterpret_cast<const ip6_hdr *>(eth+1);
+ handle_ipv6(ctx, ip6, len-sizeof(ethhdr));
+ }
else
IO::print("Unknown protocol in eth: %d\n", proto);
}
const udphdr *udp = reinterpret_cast<const udphdr *>(ip+1);
handle_udp(ctx, udp, len-sizeof(iphdr));
}
+ else if(ip->protocol==IPPROTO_ICMP)
+ {
+ const icmphdr *icmp = reinterpret_cast<const icmphdr *>(ip+1);
+ handle_icmp(ctx, icmp, len-sizeof(iphdr));
+ }
else
IO::print("Unknown protocol in ip: %d\n", ip->protocol);
}
+void NetVis::handle_ipv6(CaptureContext &ctx, const ip6_hdr *ip6, unsigned len)
+{
+ ctx.src_host = &get_host(ip6->ip6_src);
+ if(!IN6_IS_ADDR_MULTICAST(ip6->ip6_dst.s6_addr))
+ ctx.dst_host = &get_host(ip6->ip6_dst);
+
+ if(ip6->ip6_nxt==IPPROTO_TCP)
+ {
+ const tcphdr *tcp = reinterpret_cast<const tcphdr *>(ip6+1);
+ handle_tcp(ctx, tcp, len-sizeof(ip6_hdr));
+ }
+ else if(ip6->ip6_nxt==IPPROTO_UDP)
+ {
+ const udphdr *udp = reinterpret_cast<const udphdr *>(ip6+1);
+ handle_udp(ctx, udp, len-sizeof(ip6_hdr));
+ }
+ else if(ip6->ip6_nxt==IPPROTO_ICMPV6)
+ {
+ const icmp6_hdr *icmp6 = reinterpret_cast<const icmp6_hdr *>(ip6+1);
+ handle_icmp6(ctx, icmp6, len-sizeof(ip6_hdr));
+ }
+ else
+ IO::print("Unknown next header in ip6: %d\n", ip6->ip6_nxt);
+}
void NetVis::handle_tcp(CaptureContext &ctx, const tcphdr *tcp, unsigned)
{
handle_packet(ctx);
}
+void NetVis::handle_icmp(CaptureContext &ctx, const icmphdr *, unsigned)
+{
+ ctx.src_port = &get_port(0x10000|IPPROTO_ICMP);
+ ctx.dst_port = ctx.src_port;
+ handle_packet(ctx);
+}
+
+void NetVis::handle_icmp6(CaptureContext &ctx, const icmp6_hdr *, unsigned)
+{
+ ctx.src_port = &get_port(0x10000|IPPROTO_ICMPV6);
+ ctx.dst_port = ctx.src_port;
+ handle_packet(ctx);
+}
+
void NetVis::handle_packet(CaptureContext &ctx)
{
Port *port = 0;