#!/usr/bin/perl -w

use strict;

use Getopt::Long;



sub usage() {
    print STDERR "usage:
    --dev {e. g. ath0}
    --ssid
    --channel
    --gateway
    --mode {a/b/g}
    --ap ( act as an access point. off by default)
    --txf (enable/disable tx-feedback. on by default)
";
    exit 1;
}

sub mac_addr_from_dev($) {
    my $device = $_[0];
    my $output = `/sbin/ifconfig $device 2>&1`;
    if ($output =~ /^$device: error/) {
        return "";
    }
    my @tmp = split(/\s+/, $output);
    my $mac = $tmp[4];
    $mac =~ s/-/:/g;
    my @hex = split(/:/, $mac);
    
    return uc (join ":", @hex[0 .. 5]);
}




sub mac_addr_to_ip($) {
    my $mac = $_[0];
    my @hex = split(/:/, $mac);
    if (scalar(@hex) != 6) {
        return "0";
    }
    # convert the hex digits to decimals                                            
    my $x = hex($hex[3]);
    my $y = hex($hex[4]);
    my $z = hex($hex[5]);

    return "$x.$y.$z";

}

my $dev;
my $channel = 3;
my $ssid;
my $mode = "g";
my $gateway = 0;
my $ap = 0;
#my $txf = 1;
my $txf = 0;
my $interval = 10000;
my $rate_control = "static-2";
my $kernel = 0;
GetOptions('device=s' => \$dev,
	   'channel=i' => \$channel,
	   'ssid=s' => \$ssid,
	   'mode=s' => \$mode,
	   'gateway' => \$gateway,
	   'ap!' => \$ap,
	   'txf!' => \$txf,
	   'rate-control=s' => \$rate_control,
	   'kernel!' => \$kernel,
	   ) or usage();




sub dev_exists($) {
    my $device = $_[0];
    my $output = `/sbin/ifconfig $device 2>&1`;
    if ($output =~ /^$device: error/) {
        return 0;
    }
    return 1;
}

if ( -f "/.green") {
    $dev = "ath0";
    if (!dev_exists("ath0")) {
	system "/sbin/insmod /home/roofnet /home/roofnet/kernel/ath_hal.o > /dev/null 2>&1";
	system "/sbin/insmod /home/roofnet /home/roofnet/kernel/ath_pci.o > /dev/null 2>&1";
    }
    my $wlan_mac = mac_addr_from_dev("wlan0");
    system "/sbin/ifconfig ath0 down > /dev/null 2>&1";
    my $mac = "00:00:00:00:0" . substr($wlan_mac, length($wlan_mac) - 4, 4);
    
    system "/sbin/ifconfig ath0 hw ether $mac";

}



if (! defined $dev) {
    usage();
}


if ($gateway) {
    $gateway = "true";
} else{
    $gateway = "false";
}
if ($dev =~ /wlan/) {
    $mode= "b";
}

my $hostname = `hostname`;
my $wireless_mac = mac_addr_from_dev($dev);
my $suffix;
$suffix = mac_addr_to_ip($wireless_mac);




my $srcr_ip = "5." . $suffix;
my $etx_ip = "4." . $suffix;
my $safe_ip = "6." . $suffix;
my $rate_ip = "7." . $suffix;


my $srcr_nm = "255.0.0.0";
my $srcr_net = "5.0.0.0";
my $srcr_bcast = "5.255.255.255";

if (! defined $ssid) {
    $ssid = "roofnet.$srcr_ip";
}
if ($wireless_mac eq "" or 
    $wireless_mac eq "00:00:00:00:00:00") {
    print STDERR "got invalid mac address!";
    exit -1;
}
system "/sbin/ifconfig $dev up";
my $iwconfig = "/home/roofnet/bin/iwconfig";

if (-f "/sbin/iwconfig") {
    $iwconfig = "/sbin/iwconfig";
}

system "/sbin/ifconfig $dev down";
#system "$iwconfig $dev mode ad-hoc";
system "$iwconfig $dev mode monitor";
system "/sbin/ifconfig $dev up";

#system "/sbin/ifconfig $dev $safe_ip";
system "/sbin/ifconfig $dev mtu 1800";


if ($dev =~ /wlan/) {
#    system "/home/roofnet/bin/prism2_param $dev ptype 6";
#    system "/home/roofnet/bin/prism2_param $dev pseudo_ibss 1";
#    system "/home/roofnet/bin/prism2_param $dev monitor_type 1";
  #  system "$iwconfig $dev essid $ssid";
    system "$iwconfig $dev rts off";
    system "$iwconfig $dev retry 17";
    # make sure we broadcast at a fixed power
#    system "/home/roofnet/bin/prism2_param $dev alc 0";
    system "$iwconfig $dev txpower 19";

}

if ($dev =~ /ath/) {
    system "/sbin/ifconfig ath0 txqueuelen 5";
}

system "$iwconfig $dev channel $channel";

system "/sbin/modprobe tun > /dev/null 2>&1";
my $probes = "2 60 2 1500 4 1500 11 1500 22 1500";

if ($mode =~ /g/) {
    $probes = "2 60 12 60 2 1500 4 1500 11 1500 22 1500 12 1500 18 1500 24 1500 36 1500 48 1500 72 1500 96 1500";
} elsif ($mode =~ /a/) {
    $probes = "12 60 2 60 12 1500 24 1500 48 1500 72 1500 96 1500 108 1500";
}



my $srcr_es_ethtype = "0941";  # broadcast probes
my $srcr_forwarder_ethtype = "0943"; # data
my $srcr_ethtype = "0944";  # queries and replies
my $srcr_gw_ethtype = "092c"; # gateway ads

if ($mode =~ /g/) {
    print "rates :: AvailableRates(DEFAULT 2 4 11 12 18 22 24 36 48 72 96 108,
$wireless_mac 2 4 11 12 18 22 24 36 48 72 96 108);\n\n";
} elsif ($mode =~ /a/) {
    print "rates :: AvailableRates(DEFAULT 12,
$wireless_mac 12);\n\n";
} else {
  # print "rates :: AvailableRates(DEFAULT 2 4 11 22);\n\n";
 print "rates :: AvailableRates(DEFAULT 11 22,
$wireless_mac 11 22);\n\n";
}

system "cat /home/wifi/Downloads/click-1.8.0/conf/wifi/srcr.click";
#"cat /home/roofnet/scripts/srcr.click";

print <<EOF;
// has one input and one output
// takes and spits out ip packets

elementclass LinuxHost {
    \$dev, \$ip, \$nm, \$mac |
    input -> ToHost(\$dev);
    FromHost(\$dev, \$ip/\$nm, ETHER \$mac) -> output;
}


EOF

if ($kernel) {
print <<EOF;

elementclass LinuxIPHost {
    \$dev, \$ip, \$nm |

  input -> CheckIPHeader()
//    -> EtherEncap(0x0800, 1:1:1:1:1:1, 2:2:2:2:2:2) 
 -> EtherEncap(0x0800, 00:25:86:20:ea:90, 00:25:86:a7:01:d3) 
    -> SetPacketType(HOST) 
    -> to_host :: ToHost(\$dev);


  from_host :: FromHost(\$dev, \$ip/\$nm) 
    -> fromhost_cl :: Classifier(12/0806, 12/0800);


  // arp packets
  fromhost_cl[0] 
   -> ARPResponder(0.0.0.0/0 1:1:1:1:1:1) 
// -> ARPResponder(5.167.1.0/24 00:25:86:20:ea:90) 
    -> SetPacketType(HOST) 
    -> ToHost();
  
  // IP packets
  fromhost_cl[1]
    -> Strip(14)
    -> CheckIPHeader 
    -> GetIPAddress(16) 
    -> MarkIPHeader(0)
    -> output;

}




elementclass SniffDevice {
    \$device, \$promisc|

  from_dev :: FromDevice(\$device, PROMISC \$promisc)
  -> t1 :: Tee
  -> output;

    t1 [1] -> ToHostSniffers(\$device);

  input
  -> t2 :: PullTee
  -> to_dev :: ToDevice(\$device);

    t2 [1] -> ToHostSniffers(\$device);
}

EOF
} else {
    print <<EOF;
// has one input and one output
// takes and spits out ip packets
elementclass LinuxIPHost {
    \$dev, \$ip, \$nm |

input -> KernelTun(\$ip/\$nm, MTU 1500, DEVNAME \$dev, ETHER 00:01:02:03:04:05) 
//input -> KernelTap(\$ip/\$nm, MTU 1500, DEVNAME \$dev, ETHER 00:25:86:20:ea:90) 
  -> MarkIPHeader(0)
  -> CheckIPHeader()
  -> output;

}

elementclass SniffDevice {
    \$device, \$promisc|
  from_dev :: FromDevice(\$device, PROMISC \$promisc) -> output;
  input -> to_dev :: ToDevice(\$device);
}

EOF
    }

if (!$kernel) {
    print "control :: ControlSocket(\"TCP\", 7777);\n";
    print "chatter :: ChatterSocket(\"TCP\", 7778);\n";
}

print <<EOF;
//sniff_dev :: SniffDevice($dev, false);
sniff_dev :: SniffDevice($dev, true);

sched :: PrioSched()
//-> prism2_encap :: Prism2Encap()
//->athdesc_encap :: AthdescEncap()
-> set_power :: SetTXPower(POWER 63)
//-> extra_encap :: ExtraEncap()
-> SetTXRate(RATE 22)
-> radiotap_encap :: RadiotapEncap()
// -> Print ("to_dev", TIMESTAMP true)
-> sniff_dev;

route_q :: NotifierQueue(50) 
-> [0] sched;

data_q :: NotifierQueue(50)
-> data_static_rate :: SetTXRate(RATE 11)
//-> data_madwifi_rate :: MadwifiRate(OFFSET 4,
//			       ALT_RATE true,
//			       RT rates,
////			       ACTIVE true)
//				ACTIVE false)
//-> data_arf_rate :: AutoRateFallback(OFFSET 4,
//				STEPUP 25,
//				RT rates,
////				ACTIVE false)
//				ACTIVE true)
//-> data_probe_rate :: ProbeTXRate(OFFSET 4,
//			     WINDOW 5000,
////			     ALT_RATE true,
//			     RT rates,
//			     ACTIVE false)
-> [1] sched;

//Idle -> [1] data_probe_rate;
//Idle -> [1] data_madwifi_rate;
//Idle -> [1] data_arf_rate;
Idle -> Discard;

rate_q :: NotifierQueue(50)
-> rate_static_rate :: SetTXRate(RATE 11)
//-> rate_madwifi_rate :: MadwifiRate(OFFSET 4,
//			       ALT_RATE true,
//			       RT rates,
//			       ACTIVE false)
////			       ACTIVE true)
//-> rate_arf_rate :: AutoRateFallback(OFFSET 4,
//				STEPUP 25,
//				RT rates,
////				ACTIVE false)
//				ACTIVE true)
//-> rate_probe_rate :: ProbeTXRate(OFFSET 4,
//			     WINDOW 20000,
////			     ALT_RATE true,
//			     RT rates,
//			     ACTIVE false)
-> [2] sched;

//Idle -> [1] rate_probe_rate;
//Idle -> [1] rate_madwifi_rate;
//Idle -> [1] rate_arf_rate;
Idle -> Discard;


srcr :: srcr_ett($srcr_ip, $srcr_nm, $wireless_mac, $gateway, 
		 "$probes");



// make sure this is listed first so it gets tap0
srcr_host :: LinuxIPHost(srcr, $srcr_ip, $srcr_nm)
-> [1] srcr;


route_encap :: WifiEncap(0x0, 00:00:00:00:00:00)
//route_encap :: WifiEncap(0x1, 00:25:86:a7:01:d3)
->  route_q;
data_encap :: WifiEncap(0x0, 00:00:00:00:00:00)
//data_encap :: WifiEncap(0x1, 00:25:86:a7:01:d3)
-> data_q;

//route_encap :: Null
//->  route_q;
//data_encap :: Null
//-> data_q;


srcr [0] -> route_encap;   // queries, replies
srcr [1] -> route_encap;   // bcast_stats
srcr [2] -> data_encap;    // data
srcr [3] -> srcr_host; // data to me


EOF


print <<EOF;

sniff_dev 
//-> prism2_decap :: Prism2Decap()
//-> athdesc_decap :: AthdescDecap()
//-> extra_decap :: ExtraDecap()
-> radiotap_decap :: RadiotapDecap()
-> phyerr_filter :: FilterPhyErr()
-> PrintWifi(fromdev)
-> tx_filter :: FilterTX()
-> dupe :: WifiDupeFilter() 
-> wifi_cl :: Classifier(0/00%0c 0/80%f0, //beacons
			 0/08%0c, //data
			 );


wifi_cl [0] 
-> bs :: BeaconScanner(RT rates) 
-> Discard;

wifi_cl [1]
-> WifiDecap()
//-> HostEtherFilter($wireless_mac, DROP_OTHER true, DROP_OWN true)
-> HostEtherFilter($wireless_mac, DROP_OTHER false, DROP_OWN true) 
-> rxstats :: RXStats()
-> ncl :: Classifier(
		     12/09??,
		     12/0106,
		     12/0100,
		     -);


ncl [0] -> srcr;

rate_host :: LinuxHost(rate, $rate_ip, 255.0.0.0, $wireless_mac);

ncl[1] 
-> StoreData(12,\\<0806>) 
-> rate_host;

ncl[2] 
-> StoreData(12,\\<0800>) 
-> rate_host;

ncl[3] 
EOF

if ($kernel) {
    print "-> ToHost(ath0);\n";
} else {
    print "-> Discard;\n";
}

print <<EOF;

rate_host
-> arp_cl :: Classifier(12/0806,
			12/0800);
arp_cl [0] 
-> StoreData(12, \\<0106>)
-> WifiEncap(0x0, 00:00:00:00:00:00)
//-> WifiEncap(0x1, 00:25:86:a7:01:d3)
-> SetTXRate(RATE 11)
-> route_q;

arp_cl [1] 
-> StoreData(12, \\<0100>)
-> SetTXRate(RATE 11)
-> WifiEncap(0x0, 00:00:00:00:00:00)
//-> WifiEncap(0x1, 00:25:86:a7:01:d3)
-> rate_q;

EOF
    if ($txf) {
print <<EOF;
tx_filter [1] 
-> PrintWifi(txf)
-> rate_cl :: Classifier(30/0100, 
			 -);


rate_cl 
-> txf_t :: Tee(4)
-> PushAnno() 
EOF
if ($kernel) {
    print "-> ToHost(rate-txf);\n";
    print "FromHost(rate-txf, 1.1.1.1/32)\n";
}
print <<EOF;
-> Discard;


rate_cl [1] 
-> txf_t2 :: Tee(3);

//txf_t2 [0] -> [1] data_arf_rate;
//txf_t2 [1] -> [1] data_madwifi_rate;
txf_t2 [0] -> Discard;
txf_t2 [1] -> Discard;
txf_t2 [2] -> Discard;
//txf_t2 [2] -> [1] data_probe_rate;

//txf_t [1] -> [1] rate_arf_rate;
//txf_t [2] -> [1] rate_madwifi_rate;
txf_t [1] -> Discard;
txf_t [2] -> Discard;
txf_t [3] -> Discard;
//txf_t [3] -> [1] rate_probe_rate;

EOF
}