Code

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