X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fnetvis.cpp;h=0e2378fd689ac6ee9efb96a6d77a0ff9af10efc9;hb=e4b57e21ec2bc7f99c182eea23ecec297e111799;hp=dce083706470e17b1600765d4a94ac9a415c433f;hpb=26fb816da7078d687f7540e2e0a2d525801592f7;p=netvis.git diff --git a/source/netvis.cpp b/source/netvis.cpp index dce0837..0e2378f 100644 --- a/source/netvis.cpp +++ b/source/netvis.cpp @@ -9,11 +9,7 @@ Distributed unter the GPL #include #include #include -#include -#include -#include -#include -#include +#include #include #include #include @@ -23,7 +19,8 @@ Distributed unter the GPL #include #include #include -#include +#include +#include #include #include #include "history.h" @@ -46,20 +43,44 @@ NetVis::NetVis(int argc, char **argv): frames(0) { if(argc<2) - throw UsageError("No interface given"); + 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 Exception(err); + throw runtime_error(err); if(pcap_setnonblock(pcap, true, err)==-1) - throw Exception(err); + throw runtime_error(err); + + 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(a->addr)->sin_addr.s_addr)); + if(a->netmask) + addr.set_mask(Address(ntohl(reinterpret_cast(a->netmask)->sin_addr.s_addr))); + localnets.push_back(addr); + } + else if(a->addr->sa_family==AF_INET6) + { + Address addr(reinterpret_cast(a->addr)->sin6_addr); + if(a->netmask) + addr.set_mask(Address(reinterpret_cast(a->netmask)->sin6_addr)); + localnets.push_back(addr); + } + } + } - pcap_lookupnet(iface.c_str(), &localnet, &localnet_mask, err); - localnet = ntohl(localnet); - localnet_mask = ntohl(localnet_mask); + pcap_freealldevs(devs); resolver = new Resolver; @@ -68,8 +89,7 @@ NetVis::NetVis(int argc, char **argv): wnd->signal_close.connect(sigc::bind(sigc::mem_fun(this, &NetVis::exit), 0)); wnd->show(); - GL::enable(GL::BLEND); - GL::blend_func(GL::SRC_ALPHA, GL::ONE_MINUS_SRC_ALPHA); + GL::Blend::alpha().bind(); font = new GL::Font; DataFile::load(*font, "dejavu-10.font"); @@ -81,15 +101,16 @@ NetVis::NetVis(int argc, char **argv): NetVis::~NetVis() { + delete history; delete resolver; delete font; delete wnd; pcap_close(pcap); - for(map::iterator i=hosts.begin(); i!=hosts.end(); ++i) + for(map::iterator i=hosts.begin(); i!=hosts.end(); ++i) delete i->second; - for(map::iterator i=disabled_hosts.begin(); i!=disabled_hosts.end(); ++i) + for(map::iterator i=disabled_hosts.begin(); i!=disabled_hosts.end(); ++i) delete i->second; for(map::iterator i=ports.begin(); i!=ports.end(); ++i) delete i->second; @@ -121,13 +142,13 @@ void NetVis::tick() float min_activity = numeric_limits::max(); - for(map::iterator i=hosts.begin(); i!=hosts.end(); ++i) + for(map::iterator i=hosts.begin(); i!=hosts.end(); ++i) { i->second->tick(dt); min_activity = min(min_activity, i->second->get_activity()); } - float del_limit = pow(10, 6-0.1*(max_hosts-hosts.size()-disabled_hosts.size())); - for(map::iterator i=disabled_hosts.begin(); i!=disabled_hosts.end();) + float del_limit = pow(10, 6-0.1*static_cast(max_hosts-hosts.size()-disabled_hosts.size())); + for(map::iterator i=disabled_hosts.begin(); i!=disabled_hosts.end();) { i->second->tick(dt); @@ -152,7 +173,7 @@ void NetVis::tick() if(hosts.size()>max_visible_hosts) { list activity; - for(map::iterator i=hosts.begin(); i!=hosts.end(); ++i) + for(map::iterator i=hosts.begin(); i!=hosts.end(); ++i) activity.push_back(i->second->get_activity()); activity.sort(); @@ -160,7 +181,7 @@ void NetVis::tick() advance(j, activity.size()-max_visible_hosts); float limit = *j; - for(map::iterator i=hosts.begin(); i!=hosts.end();) + for(map::iterator i=hosts.begin(); i!=hosts.end();) { if(i->second->get_activity()::iterator i=ports.begin(); i!=ports.end();) + { + i->second->tick(dt); + + if(!i->second->is_registered() && i->second->get_activity()<0.1) + { + delete i->second; + ports.erase(i++); + } + else + ++i; + } + for(list::iterator i=packets.begin(); i!=packets.end();) { (*i)->tick(dt); @@ -193,15 +227,12 @@ void NetVis::tick() void NetVis::render() { - GL::clear(GL::COLOR_BUFFER_BIT); + GL::Framebuffer::system().clear(GL::COLOR_BUFFER_BIT); - GL::matrix_mode(GL::PROJECTION); - GL::load_identity(); - GL::ortho_centered(1024, 768); - GL::matrix_mode(GL::MODELVIEW); - GL::load_identity(); + GL::MatrixStack::projection() = GL::Matrix::ortho_centered(1024, 768); + GL::MatrixStack::modelview() = GL::Matrix(); - for(map::iterator i=hosts.begin(); i!=hosts.end(); ++i) + for(map::iterator i=hosts.begin(); i!=hosts.end(); ++i) i->second->render(); { GL::Immediate imm((GL::COLOR4_UBYTE, GL::VERTEX2)); @@ -211,34 +242,35 @@ void NetVis::render() imm.end(); } - GL::push_matrix(); - GL::translate(-500, 360, 0); + GL::MatrixStack::modelview() = GL::Matrix::translation(-500, 360, 0); unsigned n = 0; - for(map::iterator i=ports.begin(); (i!=ports.end() && n<50); ++i, ++n) + for(map::iterator i=ports.begin(); (i!=ports.end() && n<50); ++i) { - i->second->render(); - GL::translate(0, -12, 0); + float act = i->second->get_activity(); + if((i->second->is_registered() && act>1) || act>200) + { + i->second->render(); + GL::MatrixStack::modelview() *= GL::Matrix::translation(0, -12, 0); + ++n; + } } - GL::pop_matrix(); - GL::push_matrix(); - GL::translate(-500, -360, 0); - GL::scale_uniform(10); + GL::MatrixStack::modelview() = GL::Matrix::translation(-500, -348, 0); + GL::MatrixStack::modelview() *= GL::Matrix::scaling(10); font->draw_string(format("%d hosts", hosts.size()+disabled_hosts.size())); - GL::translate(0, -1.2, 0); + GL::MatrixStack::modelview() *= GL::Matrix::translation(0, -1.2, 0); + font->draw_string(format("%d ports", ports.size())); + GL::MatrixStack::modelview() *= GL::Matrix::translation(0, -1.2, 0); font->draw_string(format("%.2f fps", fps)); GL::Texture::unbind(); - GL::pop_matrix(); - GL::push_matrix(); - GL::translate(170, -370, 0); + GL::MatrixStack::modelview() = GL::Matrix::translation(170, -370, 0); history->render(); - GL::pop_matrix(); } -Host &NetVis::get_host(unsigned a) +Host &NetVis::get_host(const Address &a) { - map::iterator i = hosts.find(a); + map::iterator i = hosts.find(a); if(i!=hosts.end()) return *i->second; @@ -247,8 +279,9 @@ Host &NetVis::get_host(unsigned a) return *i->second; Host *host = new Host(*this, a); - if((a&localnet_mask)==localnet) - host->set_local(true); + for(list
::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) @@ -257,7 +290,7 @@ Host &NetVis::get_host(unsigned a) return *host; } -const Port &NetVis::get_port(unsigned number) +Port &NetVis::get_port(unsigned number) { map::iterator i = ports.find(number); if(i!=ports.end()) @@ -267,48 +300,131 @@ const Port &NetVis::get_port(unsigned number) return *port; } -void NetVis::capture_handler(unsigned char *user, const pcap_pkthdr *, const unsigned char *data) +void NetVis::capture_handler(unsigned char *user, const pcap_pkthdr *cap, const unsigned char *data) { NetVis *self = reinterpret_cast(user); + CaptureContext ctx; + ctx.cap_hdr = cap; const ethhdr *eth = reinterpret_cast(data); - if(ntohs(eth->h_proto)==ETH_P_IP) + self->handle_ethernet(ctx, eth, cap->caplen); +} + +void NetVis::handle_ethernet(CaptureContext &ctx, const ethhdr *eth, unsigned len) +{ + ctx.size = ctx.cap_hdr->len-sizeof(ethhdr); + + int proto = ntohs(eth->h_proto); + if(proto==ETH_P_IP) { const iphdr *ip = reinterpret_cast(eth+1); + handle_ipv4(ctx, ip, len-sizeof(ethhdr)); + } + else if(proto==ETH_P_IPV6) + { + const ip6_hdr *ip6 = reinterpret_cast(eth+1); + handle_ipv6(ctx, ip6, len-sizeof(ethhdr)); + } + else + IO::print("Unknown protocol in eth: %d\n", proto); +} - unsigned size = ntohs(ip->tot_len); - unsigned port = 0; - if(ip->protocol==IPPROTO_TCP) - { - const tcphdr *tcp = reinterpret_cast(ip+1); - port = min(ntohs(tcp->source), ntohs(tcp->dest)); - } - else if(ip->protocol==IPPROTO_UDP) - { - const udphdr *udp = reinterpret_cast(ip+1); - port = min(ntohs(udp->source), ntohs(udp->dest)); - } - Host &shost = self->get_host(ntohl(ip->saddr)); - Host *dhost = 0; - if((ntohl(ip->daddr)&0xFF)!=0xFF) - dhost = &self->get_host(ntohl(ip->daddr)); +void NetVis::handle_ipv4(CaptureContext &ctx, const iphdr *ip, unsigned len) +{ + ctx.src_host = &get_host(ntohl(ip->saddr)); + if((ntohl(ip->daddr)&0xFF)!=0xFF) + ctx.dst_host = &get_host(ntohl(ip->daddr)); + + if(ip->protocol==IPPROTO_TCP) + { + const tcphdr *tcp = reinterpret_cast(ip+1); + handle_tcp(ctx, tcp, len-sizeof(iphdr)); + } + else if(ip->protocol==IPPROTO_UDP) + { + const udphdr *udp = reinterpret_cast(ip+1); + handle_udp(ctx, udp, 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(ip6+1); + handle_tcp(ctx, tcp, len-sizeof(ip6_hdr)); + } + else if(ip6->ip6_nxt==IPPROTO_UDP) + { + const udphdr *udp = reinterpret_cast(ip6+1); + handle_udp(ctx, udp, len-sizeof(ip6_hdr)); + } + else + IO::print("Unknown next header in ip6: %d\n", ip6->ip6_nxt); +} - float throttle = shost.send_packet(); - if(throttle<1) +void NetVis::handle_tcp(CaptureContext &ctx, const tcphdr *tcp, unsigned) +{ + ctx.src_port = &get_port(ntohs(tcp->source)); + ctx.dst_port = &get_port(ntohs(tcp->dest)); + handle_packet(ctx); +} + +void NetVis::handle_udp(CaptureContext &ctx, const udphdr *udp, unsigned) +{ + ctx.src_port = &get_port(ntohs(udp->source)); + ctx.dst_port = &get_port(ntohs(udp->dest)); + handle_packet(ctx); +} + +void NetVis::handle_packet(CaptureContext &ctx) +{ + Port *port = 0; + if(ctx.src_port && ctx.dst_port) + { + if(ctx.src_port->is_registered()!=ctx.dst_port->is_registered()) { - self->packets.push_back(new Packet(shost, dhost, self->get_port(port).get_color(), size)); - self->packets.back()->tick(-throttle*Msp::Time::sec); + if(ctx.src_port->is_registered()) + port = ctx.src_port; + else + port = ctx.dst_port; } + else if(ctx.src_port->get_number()get_number()) + port = ctx.src_port; + else + port = ctx.dst_port; + } + else + port = &get_port(0); - shost.add_activity(size); - if(dhost) - dhost->add_activity(size); - - if((ntohl(ip->saddr)&self->localnet_mask)==self->localnet) - self->history->activity(0, size); - else if((ntohl(ip->daddr)&self->localnet_mask)==self->localnet) - self->history->activity(size, 0); + float throttle = ctx.src_host->send_packet(); + if(throttle<1) + { + packets.push_back(new Packet(*ctx.src_host, ctx.dst_host, port->get_color(), ctx.size)); + packets.back()->tick(-throttle*Msp::Time::sec); } + + ctx.src_host->add_activity(ctx.size); + if(ctx.dst_host) + ctx.dst_host->add_activity(ctx.size); + + if(ctx.src_port) + ctx.src_port->add_activity(ctx.size); + if(ctx.dst_port) + ctx.dst_port->add_activity(ctx.size); + + bool local_src = ctx.src_host->is_local(); + bool local_dst = (ctx.dst_host && ctx.dst_host->is_local()); + if(local_src && !local_dst) + history->activity(0, ctx.size); + else if(local_dst && !local_src) + history->activity(ctx.size, 0); } void NetVis::sighandler(int) @@ -316,4 +432,12 @@ void NetVis::sighandler(int) exit(0); } -Application::RegApp NetVis::reg; + +NetVis::CaptureContext::CaptureContext(): + cap_hdr(0), + src_host(0), + src_port(0), + dst_host(0), + dst_port(0), + size(0) +{ }