Code

logged in user at client are reported periodically to server
[gosa.git] / gosa-si / gosa-si-client
index ffdae1f57f93d93a8507227aa840b34f9afec015..4e48451073feb9254a50962b422e89e4c502fa67 100755 (executable)
@@ -41,7 +41,7 @@ use File::Basename;
 my $event_dir = "/usr/lib/gosa-si/client/events";
 use lib "/usr/lib/gosa-si/client/events";
 
-my ($cfg_file, %cfg_defaults, $foreground, $verbose, $pid_file, $opts_file, $procid, $pid, $log_file, $fai_logpath);
+my (%cfg_defaults, $foreground, $verbose, $pid_file, $procid, $pid, $log_file, $fai_logpath);
 my ($server_ip, $server_port, $server_key, $server_timeout, $server_domain, $server_key_lifetime);
 my ($client_ip, $client_port, $client_mac_address, $ldap_enabled, $ldap_config, $pam_config, $nss_config);
 my $xml;
@@ -55,8 +55,13 @@ $verbose= 1;
 
 # globalise variables which are used in imported events
 our $cfg_file;
+our $opts_file;
 our $server_address;
 our $client_address;
+our $client_ip;
+our $client_mac_address;
+our $client_dnsname;
+our $client_force_hostname;
 our $server_key;
 
 # default variables
@@ -71,6 +76,9 @@ $gosa_si_client_fifo = "/var/run/gosa-si-client.socket";
 my $delay_set_time = 5;
 our $prg= basename($0);
 
+# all x seconds the client reports logged_in users to gosa-si-server
+my $trigger_logged_in_users_report_delay = 600;
+
 %cfg_defaults = (
 "general" =>
     {"log-file"           => [\$log_file, "/var/run/".$prg.".log"],
@@ -87,6 +95,7 @@ our $prg= basename($0);
      "pam-config"         => [\$pam_config, "/etc/pam_ldap.conf"],
      "nss-config"         => [\$nss_config, "/etc/libnss_ldap.conf"],
      "fai-logpath"         => [\$fai_logpath, "/var/log/fai/fai.log"],
+        "force-hostname"               => [\$client_force_hostname, "false"],
     },
 "server" => {
     "ip"          => [\$server_ip, "127.0.0.1"],
@@ -679,93 +688,89 @@ sub open_socket {
 #  DESCRIPTION:  
 #===============================================================================
 sub register_at_gosa_si_server {
-    my ($kernel) = $_[KERNEL];
-    my $try_to_register = 0;
-
-    if( not $REGISTERED ) {
-        # create new passwd and ciphering object for client-server communication
-        $server_key = &create_passwd();
-
-        my $events = join( ",", keys %{$event_hash} );
-        while(1) {
-
-            if( $try_to_register >= @servers )  {
-                last;
-            }
-
-            # fetch first gosa-si-server from @servers
-            my $server = shift(@servers);
-
-            # append shifted gosa-si-server at the end of @servers, so looking for servers never stop if
-            # a registration never occured
-            push( @servers, $server );
-               
-                       # Check if our ip is resolvable - if not: don't try to register
-                       my $ip= &get_local_ip_for_remote_ip(sprintf("%s", $server =~ /^([0-9\.]*?):.*$/));
-                       my $resolver= Net::DNS::Resolver->new;
-                       my $dnsresult= $resolver->search($ip);
-                       my $dnsname="";
-                       if(!defined($dnsresult)) {
-                               &write_to_file("goto-error-dns:$ip", $fai_logpath);
-                               exit(1);
-                       } else {
-                               $dnsname=$dnsresult->{answer}[0]->{ptrdname};
-                       }
-
-            # create registration msg
-            my $local_ip = &get_local_ip_for_remote_ip(sprintf("%s", $server =~ /^([0-9\.]*?):.*$/));
-            my $local_mac = &get_local_mac_for_remote_ip(sprintf("%s", $server =~ /^([0-9\.]*?):.*$/));
-            my $register_hash = &create_xml_hash("here_i_am", $local_ip.":".$client_port, $server);
-            &add_content2xml_hash($register_hash, "new_passwd", $server_key);
-                       &add_content2xml_hash($register_hash, "mac_address", $local_mac);
-            &add_content2xml_hash($register_hash, "events", $events);
-            &add_content2xml_hash($register_hash, "gotoHardwareChecksum", $gotoHardwareChecksum);
-
-            # send xml hash to server with general server passwd
-            my $res = &send_msg_hash_to_target($register_hash, $server, $default_server_key);
-                       if($res == 0) {
-                # reset try_to_register
-                $try_to_register = 0;
-    
-                               # Set fixed client address and mac address
-                               $client_ip= &get_local_ip_for_remote_ip(sprintf("%s", $server =~ /^([0-9\.]*?):.*$/));
-                               $client_address= "$client_ip:$client_port";
-                $client_mac_address = $local_mac;
-
-                               # Write the MAC address to file
-                               if(stat($opts_file)) {
-                                       unlink($opts_file);
-                               }
-                               my $opts_file_FH;
-                               my $hostname= $dnsname;
-                               $hostname =~ s/\..*$//;
-                               open($opts_file_FH, ">$opts_file");
-                               print $opts_file_FH "MAC=\"$local_mac\"\n";
-                               print $opts_file_FH "IPADDRESS=\"$client_ip\"\n";
-                               print $opts_file_FH "HOSTNAME=\"$hostname\"\n";
-                               print $opts_file_FH "FQDN=\"$dnsname\"\n";
-                               close($opts_file_FH);
-                               last;
-                       } else {
-                $try_to_register++;
-                # wait 1 sec until trying to register again
-                sleep(1);
-                               next;
-                       }
-        }
+  my ($kernel) = $_[KERNEL];
+  my $try_to_register = 0;
+
+  if( not $REGISTERED ) {
+    # create new passwd and ciphering object for client-server communication
+    $server_key = &create_passwd();
+
+    my $events = join( ",", keys %{$event_hash} );
+    while(1) {
+
+      if( $try_to_register >= @servers )  {
+        last;
+      }
+
+      # fetch first gosa-si-server from @servers
+      my $server = shift(@servers);
+
+      # append shifted gosa-si-server at the end of @servers, so looking for servers never stop if
+      # a registration never occured
+      push( @servers, $server );
+
+      # Check if our ip is resolvable - if not: don't try to register
+      my $ip= &get_local_ip_for_remote_ip(sprintf("%s", $server =~ /^([0-9\.]*?):.*$/));
+      my $dnsname= gethostbyaddr(inet_aton($ip), AF_INET);
+      if(!defined($dnsname)) {
+        &write_to_file("goto-error-dns:$ip", $fai_logpath);
+        exit(1);
+      }
+
+      # create registration msg
+      my $local_ip = &get_local_ip_for_remote_ip(sprintf("%s", $server =~ /^([0-9\.]*?):.*$/));
+      my $local_mac = &get_local_mac_for_remote_ip(sprintf("%s", $server =~ /^([0-9\.]*?):.*$/));
+      my $register_hash = &create_xml_hash("here_i_am", $local_ip.":".$client_port, $server);
+      &add_content2xml_hash($register_hash, "new_passwd", $server_key);
+      &add_content2xml_hash($register_hash, "mac_address", $local_mac);
+      &add_content2xml_hash($register_hash, "events", $events);
+      &add_content2xml_hash($register_hash, "gotoHardwareChecksum", $gotoHardwareChecksum);
+
+         # Add $HOSTNAME from ENV if force-hostname is set
+         if( defined($client_force_hostname) && 
+                 $client_force_hostname eq "true") {
+                 if(defined($ENV{HOSTNAME}) &&
+                         length($ENV{HOSTNAME}) >0 ) {
+                         &add_content2xml_hash($register_hash, "force-hostname", $ENV{HOSTNAME});
+                 } else {
+                         &main::daemon_log("force-hostname was set to true, but no \$HOSTNAME was found in Environment!",0);
+                 }
+         }
+
+      # send xml hash to server with general server passwd
+      my $res = &send_msg_hash_to_target($register_hash, $server, $default_server_key);
+      if($res == 0) {
+        # reset try_to_register
+        $try_to_register = 0;
+
+        # Set fixed client address and mac address
+        $client_ip= &get_local_ip_for_remote_ip(sprintf("%s", $server =~ /^([0-9\.]*?):.*$/));
+        $client_address= "$client_ip:$client_port";
+        $client_mac_address = $local_mac;
+               $client_dnsname = $dnsname;
+
+        last;
+      } else {
+        $try_to_register++;
+
+        # wait 1 sec until trying to register again
+        sleep(1);
+        next;
+      }
+    }
 
-        if( $try_to_register >= @servers )  {
-            &write_to_file("gosa-si-no-server-available", $fai_logpath);
-             $kernel->delay_set('register_at_gosa_si_server', $delay_set_time);
-        } 
-        else {
-            daemon_log("INFO: waiting for msg 'register_at_gosa_si_server'",5);
-            $kernel->delay_set('register_at_gosa_si_server', $delay_set_time);
-            # clear old settings and set it again
-            $kernel->delay_set('trigger_new_key', $server_key_lifetime);
-        }
+    if( $try_to_register >= @servers )  {
+      &write_to_file("gosa-si-no-server-available", $fai_logpath);
+      $kernel->delay_set('register_at_gosa_si_server', $delay_set_time);
+    } 
+    else {
+      daemon_log("INFO: waiting for msg 'register_at_gosa_si_server'",5);
+      $kernel->delay_set('register_at_gosa_si_server', $delay_set_time);
+      # clear old settings and set it again
+      $kernel->delay_set('trigger_new_key', $server_key_lifetime);
     }
-    return;
+  }
+  return;
 }
     
 
@@ -926,7 +931,34 @@ sub trigger_new_key {
     &send_msg_to_target($msg, $client_address, $server_key, 'new_key');
 
     $kernel->delay_set('trigger_new_key', $server_key_lifetime);
+}
+
+
+sub trigger_logged_in_users_report {
+    my ($kernel) = $_[KERNEL] ;  
+
+    # just do if client is registered already
+    if( $REGISTERED ) {
+        my $result = qx(/usr/bin/w -hs);
+        my @res_lines;
+
+        if( defined $result ) { 
+            chomp($result);
+            @res_lines = split("\n", $result);
+        }
 
+        my @logged_in_user_list;
+        foreach my $line (@res_lines) {
+            chomp($line);
+            my @line_parts = split(/\s+/, $line); 
+            push(@logged_in_user_list, $line_parts[0]);
+        }
+        system("echo 'CURRENTLY_LOGGED_IN ".join(" ", @logged_in_user_list)."' > /var/run/gosa-si-client.socket"); 
+        $kernel->delay_set('trigger_logged_in_users_report', $trigger_logged_in_users_report_delay);
+    } else {
+        # try it in 1 sec again
+        $kernel->delay_set('trigger_logged_in_users_report', 1);
+    }
 }
 
 
@@ -970,7 +1002,7 @@ sub fifo_got_record {
     my $clmsg_hash = &create_xml_hash("CLMSG_$header", $client_address, $server_address, $content);
     &add_content2xml_hash($clmsg_hash, "macaddress", $client_mac_address);
     my $clmsg = &create_xml_string($clmsg_hash);
-    &send_msg_to_target($clmsg, $server_address, $server_key);
+    &send_msg_to_target($clmsg, $server_address, $server_key, "CLMSG_$header");
     return;
 }
 
@@ -993,6 +1025,13 @@ sub _start {
         $heap->{services}->{ $file_watcher->ID } = $file_name;
         $heap->{watchers}->{ $file_watcher->ID } = $file_watcher;
     }
+    $kernel->yield('trigger_logged_in_users_report'); 
+}
+
+
+sub _default {
+    daemon_log("ERROR: can not handle incoming msg with header '$_[ARG0]'", 1);
+    return;
 }
 
 
@@ -1197,8 +1236,12 @@ POSIX::mkfifo("$gosa_si_client_fifo", "0600");
 POE::Session->create(
        inline_states => {
                _start => \&_start, 
+        _default => \&_default,
         register_at_gosa_si_server => \&register_at_gosa_si_server,
+
+        # trigger periodical tasks
         trigger_new_key => \&trigger_new_key,
+        trigger_logged_in_users_report => \&trigger_logged_in_users_report,
         
         # handle records from each defined file differently
         fifo_record => \&fifo_got_record,