gosa-si/server/events/server_server_com.pm

   1 package server_server_com;
   2 use Exporter;
   3 @ISA = qw(Exporter);
   4 my @events = (
   5     'new_server',
   6     'confirm_new_server',
   7     'new_foreign_client',
   8     'trigger_wake',
   9     'foreign_job_updates',
  10     );
  11 @EXPORT = @events;
  12
  13 use strict;
  14 use warnings;
  15 use Data::Dumper;
  16 use GOSA::GosaSupportDaemon;
  17 use Time::HiRes qw( usleep);
  18 use Socket;
  19
  20
  21 BEGIN {}
  22
  23 END {}
  24
  25 ### Start ######################################################################
  26
  27 sub get_events {
  28     return \@events;
  29 }
  30
  31
  32 sub foreign_job_updates {
  33     my ($msg, $msg_hash, $session_id) = @_ ;
  34     my $header = @{$msg_hash->{'header'}}[0];
  35     my $source = @{$msg_hash->{'source'}}[0];
  36     my $target = @{$msg_hash->{'target'}}[0];
  37
  38     my @act_keys = keys %$msg_hash;
  39     my @jobs;
  40     foreach my $key (@act_keys) {
  41         if ($key =~ /answer\d+/ ) { push(@jobs, $key); }
  42     }
  43
  44     foreach my $foreign_job (@jobs) {
  45
  46         # add job to job queue
  47         my $func_dic = {table=>$main::job_queue_tn,
  48             primkey=>['macaddress', 'headertag'],
  49             timestamp=>@{@{$msg_hash->{$foreign_job}}[0]->{'timestamp'}}[0],
  50             status=>@{@{$msg_hash->{$foreign_job}}[0]->{'status'}}[0],
  51             result=>@{@{$msg_hash->{$foreign_job}}[0]->{'result'}}[0],
  52             progress=>@{@{$msg_hash->{$foreign_job}}[0]->{'progress'}}[0],
  53             headertag=>@{@{$msg_hash->{$foreign_job}}[0]->{'headertag'}}[0],
  54             targettag=>@{@{$msg_hash->{$foreign_job}}[0]->{'targettag'}}[0],
  55             xmlmessage=>@{@{$msg_hash->{$foreign_job}}[0]->{'xmlmessage'}}[0],
  56             macaddress=>@{@{$msg_hash->{$foreign_job}}[0]->{'macaddress'}}[0],
  57             plainname=>@{@{$msg_hash->{$foreign_job}}[0]->{'plainname'}}[0],
  58             siserver=>$source,
  59             modified=>"0",
  60         };
  61         my $res = $main::job_db->add_dbentry($func_dic);
  62         if (not $res == 0) {
  63             &main::daemon_log("$session_id ERROR: ServerPackages: process_job_msg: $res", 1);
  64         } else {
  65             &main::daemon_log("$session_id INFO: ServerPackages: $header, job '".@{@{$msg_hash->{$foreign_job}}[0]->{'headertag'}}[0].
  66                     "' successfully added to job queue", 5);
  67         }
  68     }
  69
  70     return;
  71 }
  72
  73
  74 sub new_server {
  75     my ($msg, $msg_hash, $session_id) = @_ ;
  76     my $header = @{$msg_hash->{'header'}}[0];
  77     my $source = @{$msg_hash->{'source'}}[0];
  78     my $target = @{$msg_hash->{'target'}}[0];
  79     my $key = @{$msg_hash->{'key'}}[0];
  80     my $mac = exists $msg_hash->{'macaddress'} ? @{$msg_hash->{'macaddress'}}[0] : "" ;
  81     my @clients = exists $msg_hash->{'client'} ? @{$msg_hash->{'client'}} : qw();
  82     my @loaded_modules = exists $msg_hash->{'loaded_modules'} ? @{$msg_hash->{'loaded_modules'}} : qw();
  83
  84     # sanity check
  85     if (ref $key eq 'HASH') {
  86         &main::daemon_log("$session_id ERROR: 'new_server'-message from host '$source' contains no key!", 1);
  87         return;
  88     }
  89     # add foreign server to known_server_db
  90     my $func_dic = {table=>$main::known_server_tn,
  91         primkey=>['hostname'],
  92         hostname => $source,
  93         macaddress => $mac,
  94         status => "new_server",
  95         hostkey => $key,
  96         loaded_modules => join(',', @loaded_modules),
  97         timestamp=>&get_time(),
  98     };
  99     my $res = $main::known_server_db->add_dbentry($func_dic);
 100     if (not $res == 0) {
 101         &main::daemon_log("$session_id ERROR: server_server_com.pm: cannot add server to known_server_db: $res", 1);
 102     } else {
 103         &main::daemon_log("$session_id INFO: server_server_com.pm: server '$source' successfully added to known_server_db", 5);
 104     }
 105
 106     # delete all entries at foreign_clients_db coresponding to this server
 107     my $del_sql = "DELETE FROM $main::foreign_clients_tn WHERE regserver='$source' ";
 108     my $del_res = $main::foreign_clients_db->exec_statement($del_sql);
 109
 110     # add clients of foreign server to known_foreign_clients_db
 111     my @sql_list;
 112     foreach my $client (@clients) {
 113         my @client_details = split(/,/, $client);
 114
 115         # workaround to avoid double entries in foreign_clients_db
 116         my $del_sql = "DELETE FROM $main::foreign_clients_tn WHERE hostname='".$client_details[0]."'";
 117         push(@sql_list, $del_sql);
 118
 119         my $sql = "INSERT INTO $main::foreign_clients_tn VALUES ("
 120             ."'".$client_details[0]."',"   # hostname
 121             ."'".$client_details[1]."',"   # macaddress
 122             ."'".$source."',"              # regserver
 123             ."'".&get_time()."')";         # timestamp
 124         push(@sql_list, $sql);
 125     }
 126     if (@sql_list) {
 127                 my $len = @sql_list;
 128                 $len /= 2;
 129         &main::daemon_log("$session_id DEBUG: Inserting ".$len." entries to foreign_clients_db", 8);
 130         my $res = $main::foreign_clients_db->exec_statementlist(\@sql_list);
 131     }
 132
 133     # fetch all registered clients
 134     my $client_sql = "SELECT * FROM $main::known_clients_tn";
 135     my $client_res = $main::known_clients_db->exec_statement($client_sql);
 136
 137
 138     # add already connected clients to registration message
 139     my $myhash = &create_xml_hash('confirm_new_server', $main::server_address, $source);
 140     &add_content2xml_hash($myhash, 'key', $key);
 141     map(&add_content2xml_hash($myhash, 'client', @{$_}[0].",".@{$_}[4]), @$client_res);
 142
 143     # add locally loaded gosa-si modules to registration message
 144     my $loaded_modules = {};
 145     while (my ($package, $pck_info) = each %$main::known_modules) {
 146         foreach my $act_module (keys(%{@$pck_info[2]})) {
 147             $loaded_modules->{$act_module} = "";
 148         }
 149     }
 150     map(&add_content2xml_hash($myhash, "loaded_modules", $_), keys(%$loaded_modules));
 151
 152     # add macaddress to registration message
 153     my ($host_ip, $host_port) = split(/:/, $source);
 154     my $local_ip = &get_local_ip_for_remote_ip($host_ip);
 155     my $network_interface= &get_interface_for_ip($local_ip);
 156     my $host_mac = &get_mac_for_interface($network_interface);
 157     &add_content2xml_hash($myhash, 'macaddress', $host_mac);
 158
 159     # build registration message and send it
 160     my $out_msg = &create_xml_string($myhash);
 161     my $error =  &main::send_msg_to_target($out_msg, $source, $main::ServerPackages_key, 'confirm_new_server', $session_id);
 162
 163     return;
 164 }
 165
 166
 167 sub confirm_new_server {
 168     my ($msg, $msg_hash, $session_id) = @_ ;
 169     my $header = @{$msg_hash->{'header'}}[0];
 170     my $source = @{$msg_hash->{'source'}}[0];
 171     my $key = @{$msg_hash->{'key'}}[0];
 172     my $mac = exists $msg_hash->{'macaddress'} ? @{$msg_hash->{'macaddress'}}[0] : "" ;
 173     my @clients = exists $msg_hash->{'client'} ? @{$msg_hash->{'client'}} : qw();
 174     my @loaded_modules = exists $msg_hash->{'loaded_modules'} ? @{$msg_hash->{'loaded_modules'}} : qw();
 175
 176     my $sql = "UPDATE $main::known_server_tn".
 177         " SET status='$header', hostkey='$key', loaded_modules='".join(",",@loaded_modules)."', macaddress='$mac'".
 178         " WHERE hostname='$source'";
 179     my $res = $main::known_server_db->update_dbentry($sql);
 180
 181     # add clients of foreign server to known_foreign_clients_db
 182     my @sql_list;
 183     foreach my $client (@clients) {
 184         my @client_details = split(/,/, $client);
 185
 186         # workaround to avoid double entries in foreign_clients_db
 187         my $del_sql = "DELETE FROM $main::foreign_clients_tn WHERE hostname='".$client_details[0]."'";
 188         push(@sql_list, $del_sql);
 189
 190         my $sql = "INSERT INTO $main::foreign_clients_tn VALUES ("
 191             ."'".$client_details[0]."',"   # hostname
 192             ."'".$client_details[1]."',"   # macaddress
 193             ."'".$source."',"              # regserver
 194             ."'".&get_time()."')";         # timestamp
 195         push(@sql_list, $sql);
 196     }
 197     if (@sql_list) {
 198                 my $len = @sql_list;
 199                 $len /= 2;
 200         &main::daemon_log("$session_id DEBUG: Inserting ".$len." entries to foreign_clients_db", 8);
 201         my $res = $main::foreign_clients_db->exec_statementlist(\@sql_list);
 202     }
 203
 204
 205     return;
 206 }
 207
 208
 209 sub new_foreign_client {
 210     my ($msg, $msg_hash, $session_id) = @_ ;
 211     my $header = @{$msg_hash->{'header'}}[0];
 212     my $source = @{$msg_hash->{'source'}}[0];
 213     my $hostname = @{$msg_hash->{'client'}}[0];
 214     my $macaddress = @{$msg_hash->{'macaddress'}}[0];
 215         # if new client is known in known_clients_db
 216         my $check_sql = "SELECT * FROM $main::known_clients_tn WHERE (macaddress LIKE '$macaddress')";
 217         my $check_res = $main::known_clients_db->select_dbentry($check_sql);
 218
 219         if( (keys(%$check_res) == 1) ) {
 220                         my $host_key = $check_res->{1}->{'hostkey'};
 221
 222                         # check if new client is still alive
 223                         my $client_hash = &create_xml_hash("ping", $main::server_address, $hostname);
 224                         &add_content2xml_hash($client_hash, 'session_id', $session_id);
 225                         my $client_msg = &create_xml_string($client_hash);
 226                         my $error = &main::send_msg_to_target($client_msg, $hostname, $host_key, 'ping', $session_id);
 227                         my $message_id;
 228                         my $i = 0;
 229                         while (1) {
 230                                         $i++;
 231                                         my $sql = "SELECT * FROM $main::incoming_tn WHERE headertag='answer_$session_id'";
 232                                         my $res = $main::incoming_db->exec_statement($sql);
 233                                         if (ref @$res[0] eq "ARRAY") {
 234                                                         $message_id = @{@$res[0]}[0];
 235                                                         last;
 236                                         }
 237
 238                                         # do not run into a endless loop
 239                                         if ($i > 50) { last; }
 240                                         usleep(100000);
 241                         }
 242
 243                         # client is alive
 244                         # -> new_foreign_client will be ignored
 245                         if (defined $message_id) {
 246                                 &main::daemon_log("$session_id ERROR: At new_foreign_clients: host '$hostname' is reported as a new foreign client, ".
 247                                                                 "but the host is still registered at this server. So, the new_foreign_client-msg will be ignored: $msg", 1);
 248                         }
 249         }
 250
 251
 252         # new client is not found in known_clients_db or
 253         # new client is dead -> new_client-msg from foreign server is valid
 254         # -> client will be deleted from known_clients_db
 255         # -> inserted to foreign_clients_db
 256
 257         my $del_sql = "DELETE FROM $main::known_clients_tn WHERE (hostname='$hostname')";
 258         my $del_res = $main::known_clients_db->exec_statement($del_sql);
 259     my $func_dic = { table => $main::foreign_clients_tn,
 260         primkey => ['hostname'],
 261         hostname =>   $hostname,
 262         macaddress => $macaddress,
 263         regserver =>  $source,
 264         timestamp =>  &get_time(),
 265     };
 266     my $res = $main::foreign_clients_db->add_dbentry($func_dic);
 267     if (not $res == 0) {
 268         &main::daemon_log("$session_id ERROR: server_server_com.pm: cannot add server to foreign_clients_db: $res", 1);
 269     } else {
 270         &main::daemon_log("$session_id INFO: server_server_com.pm: client '$hostname' successfully added to foreign_clients_db", 5);
 271     }
 272
 273     return;
 274 }
 275
 276
 277 sub trigger_wake {
 278     my ($msg, $msg_hash, $session_id) = @_ ;
 279
 280     foreach (@{$msg_hash->{macAddress}}){
 281         &main::daemon_log("$session_id INFO: trigger wake for $_", 5);
 282         my $host    = $_;
 283         my $ipaddr  = '255.255.255.255';
 284         my $port    = getservbyname('discard', 'udp');
 285         if (not defined $port) {
 286                 &main::daemon_log("$session_id ERROR: cannot determine port for wol $_: 'getservbyname('discard', 'udp')' failed!",1);
 287                 next;
 288         }
 289
 290         my ($raddr, $them, $proto);
 291         my ($hwaddr, $hwaddr_re, $pkt);
 292
 293         # get the hardware address (ethernet address)
 294         $hwaddr_re = join(':', ('[0-9A-Fa-f]{1,2}') x 6);
 295         if ($host =~ m/^$hwaddr_re$/) {
 296           $hwaddr = $host;
 297         } else {
 298           &main::daemon_log("$session_id ERROR: trigger_wake called with non mac address", 1);
 299         }
 300
 301         # Generate magic sequence
 302         foreach (split /:/, $hwaddr) {
 303                 $pkt .= chr(hex($_));
 304         }
 305         $pkt = chr(0xFF) x 6 . $pkt x 16 . $main::wake_on_lan_passwd;
 306
 307         # Allocate socket and send packet
 308
 309         $raddr = gethostbyname($ipaddr);
 310         if (not defined $raddr) {
 311                 &main::daemon_log("$session_id ERROR: cannot determine raddr for wol $_: 'gethostbyname($ipaddr)' failed!", 1);
 312                 next;
 313         }
 314
 315         $them = pack_sockaddr_in($port, $raddr);
 316         $proto = getprotobyname('udp');
 317
 318         socket(S, AF_INET, SOCK_DGRAM, $proto) or die "socket : $!";
 319         setsockopt(S, SOL_SOCKET, SO_BROADCAST, 1) or die "setsockopt : $!";
 320         send(S, $pkt, 0, $them) or die "send : $!";
 321         close S;
 322     }
 323
 324     return;
 325 }
 326
 327
 328 1;