Code

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