Code

Make sure a newline is passed to the socket
[gosa.git] / trunk / gosa-si / modules / ArpHandler.pm
1 package ArpHandler;
3 use Exporter;
4 @ISA = ("Exporter");
6 use strict;
7 use warnings;
8 use GOSA::GosaSupportDaemon;
9 use POSIX;
10 use Fcntl;
11 use Net::LDAP;
12 use Net::LDAP::LDIF;
13 use Net::LDAP::Entry;
14 use Net::DNS;
15 use Switch;
16 use Data::Dumper;
17 use Socket;
19 # Don't start if some of the modules are missing
20 my $start_service=1;
21 my $lookup_vendor=1;
22 BEGIN{
23         unless(eval('use Socket qw(PF_INET SOCK_DGRAM inet_ntoa sockaddr_in)')) {
24                 $start_service=0;
25         }
26         unless(eval('use POE qw(Component::Pcap Component::ArpWatch)')) {
27                 $start_service=0;
28         }
29         unless(eval('use Net::MAC::Vendor')) {
30                 $lookup_vendor=0;
31         }
32 }
34 END{}
36 my ($timeout, $mailto, $mailfrom, $user, $group);
37 my ($arp_enabled, $arp_interface, $ldap_uri, $ldap_base, $ldap_admin_dn, $ldap_admin_password);
38 my $hosts_database={};
39 my $ldap;
41 my %cfg_defaults =
42 (
43     "ArpHandler" => {
44         "enabled"             => [\$arp_enabled,         "true"],
45         "interface"           => [\$arp_interface,       "all"],
46     },
47     "server" => {
48         "ldap-uri"            => [\$ldap_uri,            ""],
49         "ldap-base"           => [\$ldap_base,           ""],
50         "ldap-admin-dn"       => [\$ldap_admin_dn,       ""],
51         "ldap-admin-password" => [\$ldap_admin_password, ""],
52     },
53 );
55 #===  FUNCTION  ================================================================
56 #         NAME:  read_configfile
57 #   PARAMETERS:  cfg_file - string -
58 #      RETURNS:  nothing
59 #  DESCRIPTION:  read cfg_file and set variables
60 #===============================================================================
61 sub local_read_configfile {
62         my $cfg;
63         if( defined( $main::cfg_file) && ( (-s $main::cfg_file) > 0 )) {
64                 if( -r $main::cfg_file ) {
65                         $cfg = Config::IniFiles->new( -file => $main::cfg_file );
66                 } else {
67                         print STDERR "Couldn't read config file!";
68                 }
69         } else {
70                 $cfg = Config::IniFiles->new() ;
71         }
72         foreach my $section (keys %cfg_defaults) {
73                 foreach my $param (keys %{$cfg_defaults{ $section }}) {
74                         my $pinfo = $cfg_defaults{ $section }{ $param };
75                         ${@$pinfo[0]} = $cfg->val( $section, $param, @$pinfo[1] );
76                 }
77         }
78 }
80 sub get_module_info {
81         my @info = (undef,
82                 undef,
83         );
85         &local_read_configfile();
86         # Don't start if some of the modules are missing
87         if(($arp_enabled eq 'true') && $start_service) {
88                 if($lookup_vendor) {
89                         eval("Net::MAC::Vendor::load_cache('file:///usr/lib/gosa-si/modules/oui.txt')");
90                         if($@) {
91                                 &main::daemon_log("Loading OUI cache file failed! MAC Vendor lookup disabled", 1);
92                                 $lookup_vendor=0;
93                         } else {
94                                 &main::daemon_log("Loading OUI cache file suceeded!", 6);
95                         }
96                 }
97                 my $ldap_handle = &main::get_ldap_handle();
99                 # When interface is not configured (or 'all'), start arpwatch on all possible interfaces
100                 if ((!defined($arp_interface)) || $arp_interface eq 'all') {
101                         foreach my $device(&get_interfaces) {
102                                 # TODO: Need a better workaround for IPv4-to-IPv6 bridges
103                                 if($device =~ m/^sit\d+$/) {
104                                         next;
105                                 }
107                                 # If device has a valid mac address
108                                 # TODO: Check if this should be the right way
109                                 if(not(&get_mac($device) eq "00:00:00:00:00:00")) {
110                                         &main::daemon_log("Starting ArpWatch on $device", 1);
111                                         POE::Session->create( 
112                                                 inline_states => {
113                                                         _start => sub {
114                                                                 &start(@_,$device);
115                                                         },
116                                                         _stop => sub {
117                                                                 $_[KERNEL]->post( sprintf("arp_watch_$device") => 'shutdown' )
118                                                         },
119                                                         got_packet => \&got_packet,
120                                                 },
121                                         );
122                                 }
123                         }
124                 } else {
125                         foreach my $device(split(/[\s,]+/, $arp_interface)) {
126                                 &main::daemon_log("Starting ArpWatch on $device", 1);
127                                 POE::Session->create( 
128                                         inline_states => {
129                                                 _start => sub {
130                                                         &start(@_,$device);
131                                                 },
132                                                 _stop => sub {
133                                                         $_[KERNEL]->post( sprintf("arp_watch_$device") => 'shutdown' )
134                                                 },
135                                                 got_packet => \&got_packet,
136                                         },
137                                 );
138                         }
139                 }
140         } else {
141                 &main::daemon_log("ArpHandler disabled. Not starting any capture processes");
142         }
143         return \@info;
146 sub process_incoming_msg {
147         return 1;
150 sub start {
151         my $device = (exists($_[ARG0])?$_[ARG0]:'eth0');
152         POE::Component::ArpWatch->spawn( 
153                 Alias => sprintf("arp_watch_$device"),
154                 Device => $device, 
155                 Dispatch => 'got_packet',
156                 Session => $_[SESSION],
157         );
159         $_[KERNEL]->post( sprintf("arp_watch_$device") => 'run' );
162 sub got_packet {
163         my ($kernel, $heap, $sender, $packet) = @_[KERNEL, HEAP, SENDER, ARG0];
165         if(     $packet->{source_haddr} eq "00:00:00:00:00:00" || 
166                 $packet->{source_haddr} eq "ff:ff:ff:ff:ff:ff" || 
167                 $packet->{source_ipaddr} eq "0.0.0.0") {
168                 return;
169         }
170         
171         my $capture_device = sprintf "%s", $kernel->alias_list($sender) =~ /^arp_watch_(.*)$/;
173         my $ldap_handle = &main::get_ldap_handle(); 
174     if(!exists($hosts_database->{$packet->{source_haddr}})) {
175                 my $dnsname= gethostbyaddr(inet_aton($packet->{source_ipaddr}), AF_INET) || $packet->{source_ipaddr};
176                 my $ldap_result=&get_host_from_ldap($packet->{source_haddr});
177                 if(exists($ldap_result->{dn})) {
178                         $hosts_database->{$packet->{source_haddr}}=$ldap_result;
179                         $hosts_database->{$packet->{source_haddr}}->{dnsname}= $dnsname;
180                         if(!exists($ldap_result->{ipHostNumber})) {
181                                 $hosts_database->{$packet->{source_haddr}}->{ipHostNumber}=$packet->{source_ipaddr};
182                         } else {
183                                 if(!($ldap_result->{ipHostNumber} eq $packet->{source_ipaddr})) {
184                                         &main::daemon_log(
185                                                 "Current IP Address ".$packet->{source_ipaddr}.
186                                                 " of host ".$hosts_database->{$packet->{source_haddr}}->{dnsname}.
187                                                 " differs from LDAP (".$ldap_result->{ipHostNumber}.")", 4);
188                                 }
189                         }
190                         $hosts_database->{$packet->{source_haddr}}->{dnsname}=$dnsname;
191                         &main::daemon_log("Host was found in LDAP as ".$ldap_result->{dn}, 8);
192                 } else {
193                         $hosts_database->{$packet->{source_haddr}}={
194                                 macAddress => $packet->{source_haddr},
195                                 ipHostNumber => $packet->{source_ipaddr},
196                                 dnsname => $dnsname,
197                                 cn => (($dnsname =~ /^(\d){1,3}\.(\d){1,3}\.(\d){1,3}\.(\d){1,3}/) ? $dnsname : sprintf "%s", $dnsname =~ /([^\.]+)\./),
198                                 macVendor => (($lookup_vendor) ? &get_vendor_for_mac($packet->{source_haddr}) : "Unknown Vendor"),
199                         };
200                         &main::daemon_log("Host was not found in LDAP (".($hosts_database->{$packet->{source_haddr}}->{dnsname}).")",6);
201                         &main::daemon_log(
202                                 "New Host ".($hosts_database->{$packet->{source_haddr}}->{dnsname}).
203                                 ": ".$hosts_database->{$packet->{source_haddr}}->{ipHostNumber}.
204                                 "/".$hosts_database->{$packet->{source_haddr}}->{macAddress},4);
205                         &add_ldap_entry(
206                                 $ldap_handle, 
207                                 $ldap_base, 
208                                 $hosts_database->{$packet->{source_haddr}}->{macAddress},
209                                 'new-system',
210                                 $hosts_database->{$packet->{source_haddr}}->{ipHostNumber},
211                                 'interface',
212                                 $hosts_database->{$packet->{source_haddr}}->{macVendor});
213                 }
214                 $hosts_database->{$packet->{source_haddr}}->{device}= $capture_device;
215         } else {
216                 if(!($hosts_database->{$packet->{source_haddr}}->{ipHostNumber} eq $packet->{source_ipaddr})) {
217                         &main::daemon_log(
218                                 "IP Address change of MAC ".$packet->{source_haddr}.
219                                 ": ".$hosts_database->{$packet->{source_haddr}}->{ipHostNumber}.
220                                 "->".$packet->{source_ipaddr}, 4);
221                         $hosts_database->{$packet->{source_haddr}}->{ipHostNumber}= $packet->{source_ipaddr};
222                         &change_ldap_entry(
223                                 $ldap_handle, 
224                                 $ldap_base, 
225                                 $hosts_database->{$packet->{source_haddr}}->{macAddress},
226                                 'ip-changed',
227                                 $hosts_database->{$packet->{source_haddr}}->{ipHostNumber},
228                         );
230                 }
231                 &main::daemon_log("Host already in cache (".($hosts_database->{$packet->{source_haddr}}->{device})."->".($hosts_database->{$packet->{source_haddr}}->{dnsname}).")",8);
232         }
233
235 sub get_host_from_ldap {
236         my $mac=shift;
237         my $result={};
238                 
239     my $ldap_handle = &main::get_ldap_handle();     
240         if(defined($ldap_handle)) {
241                 my $ldap_result= &search_ldap_entry(
242                         $ldap_handle,
243                         $ldap_base,
244                         "(|(macAddress=$mac)(dhcpHWAddress=ethernet $mac))"
245                 );
247                 if(defined($ldap_result) && $ldap_result->count==1) {
248                         if(exists($ldap_result->{entries}[0]) && 
249                                 exists($ldap_result->{entries}[0]->{asn}->{objectName}) && 
250                                 exists($ldap_result->{entries}[0]->{asn}->{attributes})) {
252                                 for my $attribute(@{$ldap_result->{entries}[0]->{asn}->{attributes}}) {
253                                         if($attribute->{type} eq 'cn') {
254                                                 $result->{cn} = $attribute->{vals}[0];
255                                         }
256                                         if($attribute->{type} eq 'macAddress') {
257                                                 $result->{macAddress} = $attribute->{vals}[0];
258                                         }
259                                         if($attribute->{type} eq 'dhcpHWAddress') {
260                                                 $result->{dhcpHWAddress} = $attribute->{vals}[0];
261                                         }
262                                         if($attribute->{type} eq 'ipHostNumber') {
263                                                 $result->{ipHostNumber} = $attribute->{vals}[0];
264                                         }
265                                 }
266                         }
267                         $result->{dn} = $ldap_result->{entries}[0]->{asn}->{objectName};
268                 }
269         }
271         return $result;
275 #===  FUNCTION  ================================================================
276 #         NAME:  get_mac 
277 #   PARAMETERS:  interface name (i.e. eth0)
278 #      RETURNS:  (mac address) 
279 #  DESCRIPTION:  Uses ioctl to get mac address directly from system.
280 #===============================================================================
281 sub get_mac {
282         my $ifreq= shift;
283         my $result;
284         if ($ifreq && length($ifreq) > 0) { 
285                 if($ifreq eq "all") {
286                         $result = "00:00:00:00:00:00";
287                 } else {
288                         my $SIOCGIFHWADDR= 0x8927;     # man 2 ioctl_list
290                         socket SOCKET, PF_INET, SOCK_DGRAM, getprotobyname('ip')
291                                 or die "socket: $!";
293                         if(ioctl SOCKET, $SIOCGIFHWADDR, $ifreq) {
294                                 my ($if, $mac)= unpack 'h36 H12', $ifreq;
296                                 if (length($mac) > 0) {
297                                         $mac=~ m/^([0-9a-f][0-9a-f])([0-9a-f][0-9a-f])([0-9a-f][0-9a-f])([0-9a-f][0-9a-f])([0-9a-f][0-9a-f])([0-9a-f][0-9a-f])$/;
298                                         $mac= sprintf("%s:%s:%s:%s:%s:%s", $1, $2, $3, $4, $5, $6);
299                                         $result = $mac;
300                                 }
301                         }
302                 }
303         }
304         return $result;
307 sub get_vendor_for_mac {
308         my $mac=shift;
309         my $result="Unknown Vendor";
311         if(defined($mac)) {
312                 my $vendor= Net::MAC::Vendor::fetch_oui_from_cache(Net::MAC::Vendor::normalize_mac($mac));
313                 if(length($vendor) > 0) {
314                         $result= @{$vendor}[0];
315                 }
316                 &main::daemon_log("Looking up Vendor for MAC ".$mac.": $result", 4);
317         }
319         return $result;
322 #===  FUNCTION  ================================================================
323 #         NAME:  add_ldap_entry
324 #      PURPOSE:  adds an element to ldap-tree
325 #   PARAMETERS:  
326 #      RETURNS:  none
327 #  DESCRIPTION:  ????
328 #       THROWS:  no exceptions
329 #     COMMENTS:  none
330 #     SEE ALSO:  n/a/bin
331 #===============================================================================
332 sub add_ldap_entry {
333         my ($ldap_tree, $ldap_base, $mac, $gotoSysStatus, $ip, $interface, $desc) = @_;
334         if(defined($ldap_tree)) {
335                 my $dn = "cn=".$hosts_database->{$mac}->{cn}.",ou=incoming,$ldap_base";
336                 my $s_res = &search_ldap_entry($ldap_tree, $ldap_base, "(|(macAddress=$mac)(dhcpHWAddress=ethernet $mac))");
337                 my $c_res = (defined($s_res))?$s_res->count:0;
338                 if($c_res == 1) {
339                         &main::daemon_log("WARNING: macAddress $mac already in LDAP", 1);
340                         return;
341                 } elsif($c_res > 0) {
342                         &main::daemon_log("ERROR: macAddress $mac exists $c_res times in LDAP", 1);
343                         return;
344                 }
346                 # create LDAP entry 
347                 my $entry = Net::LDAP::Entry->new( $dn );
348                 $entry->dn($dn);
349                 $entry->add("objectClass" => "GOhard");
350                 $entry->add("cn" => $hosts_database->{$mac}->{cn});
351                 $entry->add("macAddress" => $mac);
352                 if(defined $gotoSysStatus) {$entry->add("gotoSysStatus" => $gotoSysStatus)}
353                 if(defined $ip) {$entry->add("ipHostNumber" => $ip) }
354                 #if(defined $interface) { }
355                 if(defined $desc) {$entry->add("description" => $desc) }
357                 # submit entry to LDAP
358                 my $result = $entry->update ($ldap_tree); 
360                 # for $result->code constants please look at Net::LDAP::Constant
361                 if($result->code == 68) {   # entry already exists 
362                         &main::daemon_log("WARNING: $dn ".$result->error, 3);
363                 } elsif($result->code == 0) {   # everything went fine
364                         &main::daemon_log("add entry $dn to ldap", 1);
365                 } else {  # if any other error occur
366                         &main::daemon_log("ERROR: $dn, ".$result->code.", ".$result->error, 1);
367                 }
368         } else {
369                 &main::daemon_log("Not adding new Entry: LDAP disabled", 6);
370         }
371         return;
375 #===  FUNCTION  ================================================================
376 #         NAME:  change_ldap_entry
377 #      PURPOSE:  ????
378 #   PARAMETERS:  ????
379 #      RETURNS:  ????
380 #  DESCRIPTION:  ????
381 #       THROWS:  no exceptions
382 #     COMMENTS:  none
383 #     SEE ALSO:  n/a
384 #===============================================================================
385 sub change_ldap_entry {
386         my ($ldap_tree, $ldap_base, $mac, $gotoSysStatus, $ip) = @_;
388         if(defined($ldap_tree)) {
389                 # check if ldap_entry exists or not
390                 my $s_res = &search_ldap_entry($ldap_tree, $ldap_base, "(|(macAddress=$mac)(dhcpHWAddress=ethernet $mac))");
391                 my $c_res = (defined $s_res)?$s_res->count:0;
392                 if($c_res == 0) {
393                         &main::daemon_log("WARNING: macAddress $mac not in LDAP", 1);
394                         return;
395                 } elsif($c_res > 1) {
396                         &main::daemon_log("ERROR: macAddress $mac exists $c_res times in LDAP", 1);
397                         return;
398                 }
400                 my $s_res_entry = $s_res->pop_entry();
401                 my $dn = $s_res_entry->dn();
402                 my $replace = {
403                         'gotoSysStatus' => $gotoSysStatus,
404                 };
405                 if (defined($ip)) {
406                         $replace->{'ipHostNumber'} = $ip;
407                 }
408                 my $result = $ldap_tree->modify( $dn, replace => $replace );
410                 # for $result->code constants please look at Net::LDAP::Constant
411                 if($result->code == 32) {   # entry doesnt exists 
412                         &add_ldap_entry($mac, $gotoSysStatus);
413                 } elsif($result->code == 0) {   # everything went fine
414                         &main::daemon_log("entry $dn changed successful", 1);
415                 } else {  # if any other error occur
416                         &main::daemon_log("ERROR: $dn, ".$result->code.", ".$result->error, 1);
417                 }
418         } else {
419                 &main::daemon_log("Not changing Entry: LDAP disabled", 6);
420         }
422         return;
425 #===  FUNCTION  ================================================================
426 #         NAME:  search_ldap_entry
427 #      PURPOSE:  ????
428 #   PARAMETERS:  [Net::LDAP] $ldap_tree - object of an ldap-tree
429 #                string $sub_tree - dn of the subtree the search is performed
430 #                string $search_string - either a string or a Net::LDAP::Filter object
431 #      RETURNS:  [Net::LDAP::Search] $msg - result object of the performed search
432 #  DESCRIPTION:  ????
433 #       THROWS:  no exceptions
434 #     COMMENTS:  none
435 #     SEE ALSO:  n/a
436 #===============================================================================
437 sub search_ldap_entry {
438         my ($ldap_tree, $sub_tree, $search_string) = @_;
439         my $msg;
440         if(defined($ldap_tree)) {
441                 $msg = $ldap_tree->search( # perform a search
442                         base   => $sub_tree,
443                         filter => $search_string,
444                 ) or &main::daemon_log("cannot perform search at ldap: $@", 1);
445         }
446         return $msg;
449 # vim:ts=4:shiftwidth:expandtab
450 1;