dns_query.ml

(* vim:sw=4 ts=4 sts=4 expandtab
*)
(* Copyright 2012, Cedric Cellier
 *
 * This file is part of RobiNet.
 *
 * RobiNet is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * RobiNet is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with RobiNet.  If not, see <http://www.gnu.org/licenses/>.
 *)
(*
  This test program performs an HTTP GET from an unknown IP and an unknown MAC addr,
  which is correct enough to get an actual response from the server.
*)
open Batteries
open Tools

let iface = Pcap.openif "eth0"

let main =
    let src_ip_str  = ref "192.168.1.66"
    and src_eth_str = ref "12:34:56:78:9a:bc"
    and dst_ip_str  = ref "192.168.1.254"
    and gw_eth_str  = ref None
    and search      = ref "local"
    and names       = ref []
    in
    Arg.parse [ "-src-ip",  Arg.Set_string src_ip_str,  "IP to use as the client" ;
                "-src-mac", Arg.Set_string src_eth_str, "MAC to use as the client" ;
                "-dst-ip",  Arg.Set_string dst_ip_str,  "IP to send the request to (ie. name server)" ;
                "-gw",      Arg.String (fun gw ->
                                         gw_eth_str := Some [ Ip.Addr.zero, Ip.Addr.zero,
                                                              Some (Eth.gw_addr_of_string gw) ]),
                                                        "Gateway MAC address" ]
              (fun name -> names := name :: !names)
              "Perform a DNS A query with faked addresses" ;
    let emit bits =
        hexstring_of_bitstring bits |> Printf.printf "Injecting '%s'\n" ;
        Pcap.inject iface bits in
    let host = Host.make_static ~on:true "requester" ?gw:!gw_eth_str ~nameserver:(Ip.Addr.of_string !dst_ip_str) ~search_sfx:!search ~netmask:Ip.Addr.all_ones (Eth.Addr.of_string !src_eth_str) (Ip.Addr.of_string !src_ip_str) in
    host.Host.dev.set_read emit ;
    List.iter (Clock.asap (fun name ->
        host.Host.gethostbyname name (function
        | None -> ()
        | Some ips ->
            List.print (fun oc ip -> Printf.fprintf oc "%s\n" (Ip.Addr.to_dotted_string ip)) stdout ips))) !names ;
    ignore (Pcap.sniffer iface host.Host.dev.write) ;
    Clock.run false