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