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 }
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);
105 return;
106 }
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;
144 }
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 }
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
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;
212 }
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;
255 }
258 1;