Subversion Repositories HomeAutomation

#!/usr/bin/perl -w

################################################

#

# author: arune @ efnet

#

################################################

use IO::Socket;

use integer;

use threads;

use Cwd qw(abs_path);

use File::Basename;

use Getopt::Long;

use IO::Select;

use Fcntl ':seek';

$host = "localhost";

$port = 1201;

# Get hostname and port from command line arguments

if ( @ARGV > 0 ) {

	GetOptions(	"host=s" 	=> \$host,			#

			"port=i"	=> \$port, 			#

						);

} 

my $path = dirname(abs_path($0));

# Read devices from separate file

if (-e $path."/devices.pl") {

	require $path."/devices.pl";

} else {

	print "Error: No devices.pl was found in ".$path.". \n\n";

	exit 1;

}

$biosfile = "bios.inc";

open (FP, "+<".$biosfile) or die "Error: Cannot open file $biosfile\n";

@file = <FP>;

close FP;

# Check bios.inc for the settings that should be updated

my $foundidrow = 0;

my $foundmcurow = 0;

my $foundhfrow = 0;

my $foundlfrow = 0;

my $foundefrow = 0;

my $foundbootrow = 0;

foreach $file (@file) {

	if ($file =~ m/^NODE_HW_ID=/) {

		print "\nCurrent id-setting: ".$file."\n";

		$foundidrow = 1;

	}

	if ($file =~ m/^MCU=/) {

		$foundmcurow = 1;

	}

	if ($file =~ m/^HIGHFUSE=/) {

		$foundhfrow = 1;

	}

	if ($file =~ m/^LOWFUSE=/) {

		$foundlfrow = 1;

	}

	if ($file =~ m/^EXTFUSE=/) {

		$foundefrow = 1;

	}

	if ($file =~ m/^BOOT_START=/) {

		$foundbootrow = 1;

	}

}

if ($foundidrow == 0) {

	print "Error: Did not find NODE_HW_ID= in file $biosfile\n";

	exit 1;

}

if ($foundmcurow == 0) {

	print "Error: Did not find MCU= in file $biosfile\n";

	exit 1;

}

if ($foundhfrow == 0) {

	print "Error: Did not find HIGHFUSE= in file $biosfile\n";

	exit 1;

}

if ($foundlfrow == 0) {

	print "Error: Did not find LOWFUSE= in file $biosfile\n";

	exit 1;

}

if ($foundefrow == 0) {

	print "Error: Did not find EXTFUSE= in file $biosfile\n";

	exit 1;

}

if ($foundbootrow == 0) {

	print "Error: Did not find BOOT_START= in file $biosfile\n";

	exit 1;

}

my $hwid = "";

my @device = ();

#print "/usr/bin/atomic -s $host -p $port -c \"Node_WaitForInformation\"\n";

#my @lines = qx{/usr/bin/atomic -s $host -p $port -c \"Node_GetInformation 0xB23F54EA\"};

#my @lines = qx{/usr/bin/atomic -s $host -p $port -c \"Node_WaitForInformation\"};

$socket = atomd_initialize($host, $port);

atomd_send_command($socket, "Node_WaitForInformation");

$return = atomd_read_command_response($socket);

my @lines = split(/\n/, $return);

foreach (@lines)

{

	my $line = $_;

	print($line."\n");

	chomp($line);

	if (index($line, "Id") != -1)

	{

		my $temp = substr($line, index($line, ":") + 1);

		$temp =~ s/^\s+//;

		$temp =~ s/\s+$//;

		my @hwidChars = split(//, $temp);

		$hwid = $hwidChars[0]. $hwidChars[1]. $hwidChars[8]. $hwidChars[9]. $hwidChars[6]. $hwidChars[7]. $hwidChars[4]. $hwidChars[5]. $hwidChars[2]. $hwidChars[3];

		print "Setting HWID in bios.inc to ".$hwid."\n";

	}

	elsif (index($line, "Device Type") != -1)

	{

		$devicename = substr($line, index($line, ":") + 1);

		$devicename =~ s/^\s+//;

		$devicename =~ s/\s+$//;

		@device = $devices[0];

		foreach (@devices)

		{

			if (@{$_}[0] eq $devicename)

			{

				@device = @{$_};

			}

		}

		if ($device[0] eq "unknown") 

		{

			print "\nWarning: Unknown device type, will not set MCU settings\n\n";

		}

		else

		{

			print "Setting device type to ".$device[0]."\n";

		}

	}

}

if ($hwid eq "")

{

	exit(1);

}

# Open bios.inc and replace all settings

open (FP, "<".$biosfile) or die "Cannot open file $biosfile\n";

@file = <FP>;

close FP;

open (FP, ">".$biosfile) or die "Cannot open file $biosfile\n";

foreach $file (@file) {

	$file =~ s/^NODE_HW_ID=.*/NODE_HW_ID=$hwid/g;

	if ($device[0] ne "unknown") 

	{

		$file =~ s/^MCU=.*/MCU=$device[0]/g;

		$file =~ s/^HIGHFUSE=.*/HIGHFUSE=$device[1]/g;

		$file =~ s/^LOWFUSE=.*/LOWFUSE=$device[2]/g;

		$file =~ s/^EXTFUSE=.*/EXTFUSE=$device[3]/g;

		$file =~ s/^BOOT_START=.*/BOOT_START=$device[4]/g;

	}

	print FP $file;

}

close FP;

exit(0);

# TODO: Move all atomic-protocol functions to a separate file/lib

sub atomd_connect

{

	($host, $port) = @_;

	$socket = IO::Socket::INET->new(

			Proto    => "tcp",

			PeerAddr => $host,

			PeerPort => $port,

			Blocking => 1,

		)

		or die "Error: Cannot connect to port $port at $host\n";

	return $socket;

}

sub atomd_read_packet

{

	($socket) = @_;

	$command="";

	#while (length($command)<4)

	#{

		#sysread blocks and reads at least 1 byte before returning

	#	$bytes = sysread($socket, $buf, 1);

	#	$command .= $buf;

	#}

	read($socket, $command, 4);

	$payload_length="";

	#while (length($payload_length)<4)

	#{

	#	$bytes = sysread($socket, $buf, 1);

	#	$payload_length .= $buf;

	#}

	read($socket, $payload_length, 4);

	$payload_length = int($payload_length);

	$payload="";

	if ($payload_length > 0)

	{

		#while (length($payload) < $payload_length)

		#{

		#	$bytes = sysread($socket, $buf, 1);

		#	$payload .= $buf;

		#}

		read($socket, $payload, $payload_length);

	}

	$payload_length = sprintf("%04d", $payload_length);

	#print "areadpacket ".$command . $payload_length . $payload."\n";

	return $command . $payload_length . $payload;

}

sub atomd_write_packet

{

	($socket, $command, $payload) = @_;

	$payload_length = sprintf("%04d", length($payload)+1);

	$packet = $command.$payload_length.$payload.chr(0);

	#print "awritepacket ".$packet."\n";

	print $socket $packet;

}

sub atomd_data_available

{

	($socket) = @_;

	#Get the current buffer position

	$position = tell($socket);

	#Set the current position to end of buffer

	seek($socket, -1, SEEK_END);

	#Compare end postion to stored position

	$has_data = tell($socket) > $position;

	#Restore buffer position

	seek($socket, $position, SEEK_SET);

	return $has_data;

}

sub atomd_initialize

{

	($host, $port) = @_;

	$socket = atomd_connect($host, $port);

	#while (!atomd_data_available($socket)) {}

	#while (atomd_data_available($socket))

	#{

	#	$packet = atomd_read_packet($socket); # Read initial prompt

	#}

	# Read initial prompt

	$s = IO::Select->new();

	$s->add($socket);

	$s->can_read(1);

	sysread($socket, $data, 9000);

	$data="";

	#print "ainit".$data."\n";

	return $socket;

}

sub atomd_send_command

{

	($socket, $command) = @_;

	atomd_write_packet($socket, "RESP", $command);

}

sub atomd_read_command_response

{

	($socket) = @_;

	$response = "";

	while (1)

	{

		$packet = atomd_read_packet($socket);

		if (substr($packet, 0, 4) ne "TEXT")

		{

			last;

		}

		$response .= substr($packet, 8) . "\n";

	}

	return $response;

}