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;
}