WebSVN – HomeAutomation – Diff – /trunk/PcSoftware/scripts/setConfig/setAtomConfig.pl

 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 $hwid = "";
 my @device = ();
-print "/usr/bin/atomic -s $host -p $port -c \"Node_WaitForInformation\"\n";
+#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\"};
+#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($_);
+    print($line."\n");
     chomp($line);
     if (index($line, "Id") != -1)
+    {
         my $temp = 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/^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("127.0.0.1", 1202);
+    $socket = atomd_connect($host, $port);
-    while (atomd_data_available($socket))
-    {
-        $packet = atomd_read_packet($socket); # Read initial prompt
-    }
+    #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;
+}

Subversion Repositories HomeAutomation

(root)/trunk/PcSoftware/scripts/setConfig/setAtomConfig.pl @ 2011 – Rev 1881 → 1882