host/ihost-perl/ihost.pl

#!/usr/bin/perl -w

# 
# i-scream central monitoring system
# http://www.i-scream.org.uk
# Copyright (C) 2000-2002 i-scream
# 
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
# 
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
# 
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
#

# -----------------------------------------------------------
# Perl i-scream Host.
# http://www.i-scream.org.uk
#
# An all-in-one script to act as an i-scream host on
# a typical Unix/Linux box.
#
# $Author: tdb $
# $Id: ihost.pl,v 1.50 2002/05/21 16:47:11 tdb Exp $
#------------------------------------------------------------

$| = 1;

use strict;
use IO::Socket;
use Sys::Hostname;

use vars qw (
             $filter_manager_addr
             $filter_manager_port
             $seq_no
             $udp_update_time
             $tcp_update_time
             $last_udp_time
             $last_tcp_time
             $last_modified
             $udp_port
             $tcp_port
             $filter_addr
             $file_list
             $fqdn
             $pidfile
             $retry_wait
             $ostype
             $key
             @data
            );

if (@ARGV != 2) {
    die "Usage: ihost.pl [i-scream filter manager] [TCP port]\n";
}

$filter_manager_addr = $ARGV[0];
$filter_manager_port = $ARGV[1];

$seq_no = 1;
$retry_wait = 60;

# work out our platform, if we can.
$ostype = `uname -s`;
chomp $ostype;
$ostype = "unknown" if not defined $ostype;

# write our PID to a file
# use home dir by default
#$pidfile = $ENV{"HOME"};
# or drop it in /var/tmp if we can't find HOME
$pidfile = "/var/tmp" if not defined $pidfile;
$pidfile .= "/.ihost.pid";
&write_pid();

&tcp_configure();
&send_tcp_heartbeat();
&send_udp_packet();

$last_udp_time = time;
$last_tcp_time = time;
while (1) {
    my($time) = time;
    if ($time >= $last_udp_time + $udp_update_time) {
        &send_udp_packet();
        $last_udp_time = $time;
    }
    if ($time >= $last_tcp_time + $tcp_update_time) {
        &send_tcp_heartbeat();
        $last_tcp_time = $time;
    }
    my($next_udp) = $udp_update_time - $time + $last_udp_time;
    my($next_tcp) = $tcp_update_time - $time + $last_tcp_time;
    my($delay);
    if ($next_udp < $next_tcp) {
        $delay = $next_udp
    }
    else {
        $delay = $next_tcp;
    }
    sleep $delay;
}

# we'll probably never get here...
`rm -f $pidfile`;
exit(0);


#-----------------------------------------------------------------------
# wait_then_retry
# Waits for the period of time specified in $retry_wait, then attempts
# to reconfigure with the server.
#-----------------------------------------------------------------------
sub wait_then_retry() {
    print "Will retry configuration with filter manager in $retry_wait seconds.\n";
    sleep $retry_wait;
}


#-----------------------------------------------------------------------
# tcp_configure
# Establishes a TCP connection to the specified i-scream filter manager.
# The host then requests details from the server, such as the intervals
# at which to send UDP packets.
#-----------------------------------------------------------------------
sub tcp_configure() {
    
    while (1) {
        my($sock) = new IO::Socket::INET(
                                         PeerAddr => $filter_manager_addr,
                                         PeerPort => $filter_manager_port,
                                         Proto => 'tcp'
                                        ) or die "Cannot connect!";
        if (!defined $sock) {
            print "IHOST ERROR: Could not connect to $filter_manager_addr:$filter_manager_port.\n";
            print "Please check that there is an i-scream server at this address.\n";
            wait_then_retry();
            next;
        }

        # Now run through the configuration process...
        my($response);
    
        print $sock "STARTCONFIG\n";
        $response = <$sock>;
        if ($response && !($response eq "OK\n")) {
            print "The i-scream server rejected the STARTCONFIG command.\n";
            close($sock);
            wait_then_retry();
            next;
        }

        print "Config started okay.\n";
    
        print $sock "LASTMODIFIED\n";
        $response = <$sock>;
        if (!$response || $response eq "ERROR\n") {
            print "The i-scream server did not provide the LASTMODIFIED value.\n";
            close($sock);
            wait_then_retry();
            next;
        }
        chomp $response;
        $last_modified = $response;
    
        print "Config last modified: ". (scalar localtime $last_modified/1000) . "\n";

        print $sock "FILELIST\n";
        $response = <$sock>;
        if (!$response || $response eq "ERROR\n") {
            print "The i-scream server did not provide a configuration file list.\n";
            close($sock);
            wait_then_retry();
            next;
        }
        chomp $response;
        $file_list = $response;

        print "File list obtained: $file_list\n";

        print $sock "FQDN\n";
        $response = <$sock>;
        if (!$response || $response eq "ERROR\n") {
            print "The i-scream server did not tell us our FQDN.\n";
            close($sock);
            wait_then_retry();
            next;
        }
        chomp $response;
        $fqdn = $response;

        print "FQDN returned: $fqdn\n";

        print $sock "UDPUpdateTime\n";
        $response = <$sock>;
        if (!$response || $response eq "ERROR\n") {
            print "The i-scream server did not give us a UDPUpdateTime.\n";
            close($sock);
            wait_then_retry();
            next;
        }
        chomp $response;
        $udp_update_time = $response;

        print $sock "TCPUpdateTime\n";
        $response = <$sock>;
        if (!$response || $response eq "ERROR\n") {
            print "The i-scream server did not give us a TCPUpdateTime.\n";
            close($sock);
            wait_then_retry();
            next;
        }
        chomp $response;
        $tcp_update_time = $response;
    
        print "UDP packet period: $udp_update_time seconds.\nTCP heartbeat period: $tcp_update_time seconds.\n";

        print $sock "ENDCONFIG\n";
        $response = <$sock>;
        if ($response && !($response eq "OK\n")) {
            print "ENDCONFIG command to server failed.  Terminated.\n";
            close($sock);
            wait_then_retry();
            next;
        }

        print "Config ended.\n";
    
        print $sock "FILTER\n";
        $response = <$sock>;
        if (!$response) {
            print "Failed: Could not get a filter address from the filter manager.\n";
            close($sock);
            wait_then_retry();
            next;
        }
        chomp $response;
        if ($response eq "ERROR") {
            print "There are no active configured filters for your host.\n";
            close($sock);
            wait_then_retry();
            next;
        }
        $response =~ /^(.*);(.*);(.*)/;
        ($filter_addr, $udp_port, $tcp_port) = ($1, $2, $3);
        unless (defined($filter_addr) && defined($udp_port) && defined($tcp_port)) {
            print "Failed: Filter address response from server did not make sense: $response\n";
            close($sock);
            wait_then_retry();
            next;
        }
    
        print "Got filter data ($filter_addr, $udp_port, $tcp_port)\n";

        print $sock "END\n";
        $response = <$sock>;
        if ($response && ($response eq "OK\n")) {
            print "Host successfully configured via TCP.\n"
        }
        else {
            print "The server failed the host configuration on the END command.\n";
            close($sock);
            wait_then_retry();
            next;
        }
    
        close($sock);

        print "Configuration finished sucessfully!\n";
        last;
    }
    return;
}


#-----------------------------------------------------------------------
# send_udp_packet
# Sends a UDP packet to an i-scream filter.
# The packet contains XML markup describing some of the machine's state.
# Receipt of UDP packets is not guaranteed.
#-----------------------------------------------------------------------
sub send_udp_packet() {

    my($plugins_dir) = "plugins";
    
    opendir PLUGINS, $plugins_dir;
    my(@plugins) = readdir PLUGINS;
    foreach my $plugin (@plugins) {
        push @data, `$plugins_dir/$plugin $ostype` if -x "$plugins_dir/$plugin" && -f "$plugins_dir/$plugin";
    }
    
    # get some extra data
    my($date) = time;
    my($ip);
    $ip = inet_ntoa(scalar(gethostbyname(hostname())) || 'localhost') or $ip = 'localhost';
    
    # add some extra data to the array
    push(@data, "packet.attributes.seq_no $seq_no");
    push(@data, "packet.attributes.machine_name $fqdn");
    push(@data, "packet.attributes.date $date");
    push(@data, "packet.attributes.type data");
    push(@data, "packet.attributes.ip $ip");
    push(@data, "packet.attributes.key $key");
    
    # sort the data
    @data = sort(grep(!/^$/, grep(/^packet\./, @data)));

    # turn the array into some nice XML
    my($xml) = &make_xml("", "");

    my($sock) = new IO::Socket::INET (
                                      PeerPort => $udp_port,
                                      PeerAddr => $filter_addr,
                                      Proto => 'udp'
                                     ) or die "Could not send UDP: $!\n";
    
    print $sock $xml or die "Could not send UDP packet: $!\n";
    close($sock);
    $seq_no++;
    print "-";
    
    return;
}


#-----------------------------------------------------------------------
# send_tcp_heartbeat
# Establishes a TCP connection to an i-scream filter.
# The heartbeat is used as a guaranteed "I'm alive" delivery mechanism.
# If we need to reconfigure, then we complete the heartbeat before
# doing so.
#-----------------------------------------------------------------------
sub send_tcp_heartbeat() {

    my ($doReconfigure) = 0;

    my($sock) = new IO::Socket::INET(
                                     PeerAddr => $filter_addr,
                                     PeerPort => $tcp_port,
                                     Proto => 'tcp'
                                    ) or return;
    if (!defined $sock) {
        print "IHOST WARNING: Failed to deliver a heartbeat to the i-scream filter.\n";
        &tcp_configure();
        return;
    }

    # Now run through the configuration process.
    my($response);

    print $sock "HEARTBEAT\n";
    $response = <$sock>;
    if (!$response eq "OK\n") {
        close($sock);
        print "Server gave wrong response to HEARTBEAT: $response\n";
        &tcp_configure();
        return;
    }
    
    print $sock "CONFIG\n";
    $response = <$sock>;
    if (!$response eq "OK\n") {
        close($sock);
        print "Server gave wrong response to CONFIG: $response\n";
        &tcp_configure();
        return;
    }

    print $sock "$file_list\n";
    $response = <$sock>;
    if (!$response eq "OK\n") {
        close($sock);
        print "Server gave wrong response to file list: $response\n";
        &tcp_configure();
        return;
    }
    
    print $sock "$last_modified\n";
    $response = <$sock>;
    if ($response eq "ERROR\n") {
        close($sock);
        print "Server configuration changed.  Reconfiguring with filter manager.\n";
        $doReconfigure = 1;
    }
    if (!$response eq "OK\n") {
        close($sock);
        print "Server gave wrong response to HEARTBEAT: $response\n";
        &tcp_configure();
        return;
    }
    
    print $sock "KEY\n";
    $key = <$sock>;
    
    print $sock "ENDHEARTBEAT\n";
    $response = <$sock>;
    if (!$response eq "OK\n") {
        close($sock);
        print "Server gave wrong response to ENDHEARTBEAT: $response\n";
        &tcp_configure();
        return;
    }
    
    close($sock);
    print "^";
    
    &tcp_configure() if $doReconfigure;
    
    return;
}


#-----------------------------------------------------------------------
# write_pid
# Writes the PID (process ID) of this instance to $pidfile.
# This is then used by a seperate script to check (and restart) ihost.
#-----------------------------------------------------------------------
sub write_pid() {
    open PID, ">$pidfile";
    print PID $$;
    close PID;
    
    return;
}

#-----------------------------------------------------------------------
# make_xml
# Turns an array of plugins data into an XML string.
#-----------------------------------------------------------------------
sub make_xml() {
    my($curlevel, $curline) = @_;
    my($xmltemp) = ""; my($curtag) = ""; my($attributes) = "";
    while(1) {
        $curline = shift(@data) if $curline eq "";
        return $xmltemp if not defined $curline;
        chomp $curline;
        # dealing with nest (or attributes)
        if($curline =~ /^$curlevel([^\.\s]+\.)/) {
            $curtag=$1;
            if($curline =~ /^$curlevel$curtag([^\.\s]+)\s+(.*)$/) {
                $xmltemp .= &make_xml("$curlevel$curtag", $curline);
            }
            elsif($curline =~ /^$curlevel$curtag(attributes)\.([^\.\s]+)\s+(.*)$/) {
                $attributes .= " $2=\"$3\"";
            }
            else {
                $xmltemp .= &make_xml("$curlevel$curtag", $curline);
            }
            my($nextline) = $data[0]; chomp $nextline if defined $nextline;
            $curtag =~ s/(.*)\./$1/;
            if((defined $nextline) && ($nextline =~ /^$curlevel$curtag\./)) {
                $curline = ""; 
            }
            else {
                $xmltemp = "<$curtag$attributes>$xmltemp</$curtag>" unless $curtag eq "";
                return $xmltemp;
            }
        }
        # dealing with value
        elsif($curline =~ /^$curlevel([^\.\s]+)\s+(.*)$/) {
            $curtag=$1;
            $xmltemp=$2;           
            my($nextline) = $data[0]; chomp $nextline if defined $nextline;
            if(defined $nextline && ($nextline =~ /^$curlevel$curtag\./ || $nextline =~ /^$curlevel$curtag\s+/)) {
                $curline = ""; 
            }
            else {
                $xmltemp = "<$curtag$attributes>$xmltemp</$curtag>" unless $curtag eq "";
                return $xmltemp;
            }
        }
        # dealing with a null value
        elsif($curline =~ /^$curlevel([^\.\s]+)$/) {
            # simply adding a space makes the above elsif deal with it :)
            # just level with an empty tag in the XML
            $curline .= " ";
        }
        # failing all that, skip the line
        else {
            $curline = "";
        }
    }
}
Revision:	1.51
Committed:	Fri Mar 28 16:30:30 2003 UTC (21 years, 8 months ago) by tdb
Content type:	text/plain
Branch:	MAIN
CVS Tags:	HEAD
Changes since 1.50:	+1 -1 lines
State:	*FILE REMOVED*
Log Message:	Removed some un-used code from CVS. We can always resurrect this later if someone feels they want to work on it. Gone are the old perl ihost which isn't needed now, winhost which is broken and shows no sign of being fixed, and DBReporter. If someone wants to revive them, I'll undelete them :-)
#	User	Rev	Content
1	pjm2	1.6	#!/usr/bin/perl -w
2	pjm2	1.1
3	tdb	1.49	#
4			# i-scream central monitoring system
5	tdb	1.50	# http://www.i-scream.org.uk
6	tdb	1.49	# Copyright (C) 2000-2002 i-scream
7			#
8			# This program is free software; you can redistribute it and/or
9			# modify it under the terms of the GNU General Public License
10			# as published by the Free Software Foundation; either version 2
11			# of the License, or (at your option) any later version.
12			#
13			# This program is distributed in the hope that it will be useful,
14			# but WITHOUT ANY WARRANTY; without even the implied warranty of
15			# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16			# GNU General Public License for more details.
17			#
18			# You should have received a copy of the GNU General Public License
19			# along with this program; if not, write to the Free Software
20			# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
21			#
22
23	pjm2	1.1	# -----------------------------------------------------------
24			# Perl i-scream Host.
25	pjm2	1.12	# http://www.i-scream.org.uk
26	pjm2	1.1	#
27			# An all-in-one script to act as an i-scream host on
28	tdb	1.36	# a typical Unix/Linux box.
29	pjm2	1.1	#
30	tdb	1.44	# $Author: tdb $
31	tdb	1.51	# $Id: ihost.pl,v 1.50 2002/05/21 16:47:11 tdb Exp $
32	pjm2	1.1	#------------------------------------------------------------
33
34			$\| = 1;
35
36			use strict;
37			use IO::Socket;
38			use Sys::Hostname;
39
40			use vars qw (
41			$filter_manager_addr
42			$filter_manager_port
43			$seq_no
44			$udp_update_time
45			$tcp_update_time
46			$last_udp_time
47			$last_tcp_time
48			$last_modified
49			$udp_port
50			$tcp_port
51			$filter_addr
52			$file_list
53	pjm2	1.17	$fqdn
54	tdb	1.25	$pidfile
55	pjm2	1.28	$retry_wait
56	tdb	1.44	$ostype
57	tdb	1.47	$key
58	tdb	1.37	@data
59	pjm2	1.1	);
60
61			if (@ARGV != 2) {
62			die "Usage: ihost.pl [i-scream filter manager] [TCP port]\n";
63			}
64
65			$filter_manager_addr = $ARGV[0];
66			$filter_manager_port = $ARGV[1];
67
68			$seq_no = 1;
69	pjm2	1.33	$retry_wait = 60;
70	pjm2	1.1
71	tdb	1.44	# work out our platform, if we can.
72			$ostype = `uname -s`;
73			chomp $ostype;
74			$ostype = "unknown" if not defined $ostype;
75
76	tdb	1.25	# write our PID to a file
77	tdb	1.41	# use home dir by default
78	tdb	1.42	#$pidfile = $ENV{"HOME"};
79	tdb	1.41	# or drop it in /var/tmp if we can't find HOME
80			$pidfile = "/var/tmp" if not defined $pidfile;
81			$pidfile .= "/.ihost.pid";
82	tdb	1.25	&write_pid();
83
84	pjm2	1.1	&tcp_configure();
85	tdb	1.47	&send_tcp_heartbeat();
86	pjm2	1.1	&send_udp_packet();
87
88			$last_udp_time = time;
89			$last_tcp_time = time;
90			while (1) {
91	pjm2	1.9	my($time) = time;
92			if ($time >= $last_udp_time + $udp_update_time) {
93	pjm2	1.1	&send_udp_packet();
94	pjm2	1.9	$last_udp_time = $time;
95	pjm2	1.1	}
96	pjm2	1.9	if ($time >= $last_tcp_time + $tcp_update_time) {
97	pjm2	1.1	&send_tcp_heartbeat();
98	pjm2	1.9	$last_tcp_time = $time;
99	pjm2	1.1	}
100	pjm2	1.9	my($next_udp) = $udp_update_time - $time + $last_udp_time;
101			my($next_tcp) = $tcp_update_time - $time + $last_tcp_time;
102			my($delay);
103			if ($next_udp < $next_tcp) {
104			$delay = $next_udp
105			}
106			else {
107			$delay = $next_tcp;
108			}
109	tdb	1.35	sleep $delay;
110	pjm2	1.1	}
111
112	tdb	1.25	# we'll probably never get here...
113			`rm -f $pidfile`;
114	pjm2	1.1	exit(0);
115
116	pjm2	1.12
117			#-----------------------------------------------------------------------
118	pjm2	1.28	# wait_then_retry
119			# Waits for the period of time specified in $retry_wait, then attempts
120			# to reconfigure with the server.
121			#-----------------------------------------------------------------------
122			sub wait_then_retry() {
123			print "Will retry configuration with filter manager in $retry_wait seconds.\n";
124	tdb	1.35	sleep $retry_wait;
125	pjm2	1.28	}
126
127
128			#-----------------------------------------------------------------------
129	pjm2	1.12	# tcp_configure
130			# Establishes a TCP connection to the specified i-scream filter manager.
131			# The host then requests details from the server, such as the intervals
132			# at which to send UDP packets.
133			#-----------------------------------------------------------------------
134	pjm2	1.1	sub tcp_configure() {
135
136	pjm2	1.31	while (1) {
137	pjm2	1.28	my($sock) = new IO::Socket::INET(
138			PeerAddr => $filter_manager_addr,
139			PeerPort => $filter_manager_port,
140			Proto => 'tcp'
141			) or die "Cannot connect!";
142			if (!defined $sock) {
143			print "IHOST ERROR: Could not connect to $filter_manager_addr:$filter_manager_port.\n";
144			print "Please check that there is an i-scream server at this address.\n";
145			wait_then_retry();
146			next;
147			}
148
149			# Now run through the configuration process...
150			my($response);
151
152			print $sock "STARTCONFIG\n";
153			$response = <$sock>;
154	pjm2	1.32	if ($response && !($response eq "OK\n")) {
155			print "The i-scream server rejected the STARTCONFIG command.\n";
156	pjm2	1.28	close($sock);
157			wait_then_retry();
158			next;
159			}
160
161			print "Config started okay.\n";
162
163			print $sock "LASTMODIFIED\n";
164			$response = <$sock>;
165	tdb	1.48	if (!$response \|\| $response eq "ERROR\n") {
166			print "The i-scream server did not provide the LASTMODIFIED value.\n";
167	pjm2	1.32	close($sock);
168			wait_then_retry();
169			next;
170			}
171	tdb	1.48	chomp $response;
172	pjm2	1.28	$last_modified = $response;
173
174			print "Config last modified: ". (scalar localtime $last_modified/1000) . "\n";
175
176			print $sock "FILELIST\n";
177			$response = <$sock>;
178	tdb	1.48	if (!$response \|\| $response eq "ERROR\n") {
179	pjm2	1.32	print "The i-scream server did not provide a configuration file list.\n";
180			close($sock);
181			wait_then_retry();
182			next;
183			}
184	tdb	1.48	chomp $response;
185	pjm2	1.28	$file_list = $response;
186
187			print "File list obtained: $file_list\n";
188
189			print $sock "FQDN\n";
190			$response = <$sock>;
191	tdb	1.48	if (!$response \|\| $response eq "ERROR\n") {
192	pjm2	1.32	print "The i-scream server did not tell us our FQDN.\n";
193			close($sock);
194			wait_then_retry();
195			next;
196			}
197	tdb	1.48	chomp $response;
198	pjm2	1.28	$fqdn = $response;
199
200			print "FQDN returned: $fqdn\n";
201
202			print $sock "UDPUpdateTime\n";
203			$response = <$sock>;
204	tdb	1.48	if (!$response \|\| $response eq "ERROR\n") {
205	pjm2	1.32	print "The i-scream server did not give us a UDPUpdateTime.\n";
206			close($sock);
207			wait_then_retry();
208			next;
209			}
210	tdb	1.48	chomp $response;
211	pjm2	1.28	$udp_update_time = $response;
212
213			print $sock "TCPUpdateTime\n";
214			$response = <$sock>;
215	tdb	1.48	if (!$response \|\| $response eq "ERROR\n") {
216	pjm2	1.32	print "The i-scream server did not give us a TCPUpdateTime.\n";
217			close($sock);
218			wait_then_retry();
219			next;
220			}
221	tdb	1.48	chomp $response;
222	pjm2	1.28	$tcp_update_time = $response;
223
224			print "UDP packet period: $udp_update_time seconds.\nTCP heartbeat period: $tcp_update_time seconds.\n";
225
226			print $sock "ENDCONFIG\n";
227			$response = <$sock>;
228	pjm2	1.32	if ($response && !($response eq "OK\n")) {
229	pjm2	1.28	print "ENDCONFIG command to server failed. Terminated.\n";
230			close($sock);
231			wait_then_retry();
232			next;
233			}
234
235			print "Config ended.\n";
236
237			print $sock "FILTER\n";
238			$response = <$sock>;
239	pjm2	1.32	if (!$response) {
240	pjm2	1.28	print "Failed: Could not get a filter address from the filter manager.\n";
241			close($sock);
242			wait_then_retry();
243			next;
244			}
245	tdb	1.48	chomp $response;
246	pjm2	1.28	if ($response eq "ERROR") {
247			print "There are no active configured filters for your host.\n";
248			close($sock);
249			wait_then_retry();
250			next;
251			}
252	tdb	1.48	$response =~ /^(.);(.);(.*)/;
253	pjm2	1.28	($filter_addr, $udp_port, $tcp_port) = ($1, $2, $3);
254			unless (defined($filter_addr) && defined($udp_port) && defined($tcp_port)) {
255			print "Failed: Filter address response from server did not make sense: $response\n";
256			close($sock);
257			wait_then_retry();
258			next;
259			}
260
261			print "Got filter data ($filter_addr, $udp_port, $tcp_port)\n";
262
263			print $sock "END\n";
264			$response = <$sock>;
265	pjm2	1.32	if ($response && ($response eq "OK\n")) {
266	pjm2	1.28	print "Host successfully configured via TCP.\n"
267			}
268			else {
269	pjm2	1.32	print "The server failed the host configuration on the END command.\n";
270	pjm2	1.28	close($sock);
271			wait_then_retry();
272			next;
273			}
274	pjm2	1.1
275	pjm2	1.28	close($sock);
276	pjm2	1.1
277	pjm2	1.28	print "Configuration finished sucessfully!\n";
278			last;
279	pjm2	1.1	}
280	pjm2	1.12	return;
281	pjm2	1.1	}
282
283	pjm2	1.12
284			#-----------------------------------------------------------------------
285			# send_udp_packet
286			# Sends a UDP packet to an i-scream filter.
287			# The packet contains XML markup describing some of the machine's state.
288			# Receipt of UDP packets is not guaranteed.
289			#-----------------------------------------------------------------------
290	pjm2	1.1	sub send_udp_packet() {
291
292	tdb	1.37	my($plugins_dir) = "plugins";
293
294			opendir PLUGINS, $plugins_dir;
295			my(@plugins) = readdir PLUGINS;
296			foreach my $plugin (@plugins) {
297	tdb	1.44	push @data, `$plugins_dir/$plugin $ostype` if -x "$plugins_dir/$plugin" && -f "$plugins_dir/$plugin";
298	tdb	1.37	}
299	pjm2	1.1
300	tdb	1.36	# get some extra data
301	pjm2	1.1	my($date) = time;
302	pjm2	1.33	my($ip);
303			$ip = inet_ntoa(scalar(gethostbyname(hostname())) \|\| 'localhost') or $ip = 'localhost';
304	tdb	1.36
305			# add some extra data to the array
306	tdb	1.43	push(@data, "packet.attributes.seq_no $seq_no");
307			push(@data, "packet.attributes.machine_name $fqdn");
308			push(@data, "packet.attributes.date $date");
309			push(@data, "packet.attributes.type data");
310			push(@data, "packet.attributes.ip $ip");
311	tdb	1.47	push(@data, "packet.attributes.key $key");
312	tdb	1.39
313			# sort the data
314	tdb	1.43	@data = sort(grep(!/^$/, grep(/^packet\./, @data)));
315
316	tdb	1.36	# turn the array into some nice XML
317			my($xml) = &make_xml("", "");
318	tdb	1.43
319	pjm2	1.1	my($sock) = new IO::Socket::INET (
320			PeerPort => $udp_port,
321			PeerAddr => $filter_addr,
322			Proto => 'udp'
323	pjm2	1.21	) or die "Could not send UDP: $!\n";
324	tdb	1.36
325	pjm2	1.1	print $sock $xml or die "Could not send UDP packet: $!\n";
326			close($sock);
327			$seq_no++;
328	tdb	1.38	print "-";
329	pjm2	1.12
330			return;
331	pjm2	1.1	}
332
333	pjm2	1.12
334			#-----------------------------------------------------------------------
335			# send_tcp_heartbeat
336			# Establishes a TCP connection to an i-scream filter.
337			# The heartbeat is used as a guaranteed "I'm alive" delivery mechanism.
338	pjm2	1.34	# If we need to reconfigure, then we complete the heartbeat before
339			# doing so.
340	pjm2	1.12	#-----------------------------------------------------------------------
341	pjm2	1.1	sub send_tcp_heartbeat() {
342
343	pjm2	1.34	my ($doReconfigure) = 0;
344
345	pjm2	1.1	my($sock) = new IO::Socket::INET(
346			PeerAddr => $filter_addr,
347			PeerPort => $tcp_port,
348			Proto => 'tcp'
349	pjm2	1.21	) or return;
350	pjm2	1.11	if (!defined $sock) {
351			print "IHOST WARNING: Failed to deliver a heartbeat to the i-scream filter.\n";
352	pjm2	1.27	&tcp_configure();
353	pjm2	1.11	return;
354			}
355	pjm2	1.1
356			# Now run through the configuration process.
357			my($response);
358
359			print $sock "HEARTBEAT\n";
360			$response = <$sock>;
361	pjm2	1.32	if (!$response eq "OK\n") {
362	pjm2	1.1	close($sock);
363			print "Server gave wrong response to HEARTBEAT: $response\n";
364	pjm2	1.27	&tcp_configure();
365	pjm2	1.1	return;
366			}
367
368			print $sock "CONFIG\n";
369			$response = <$sock>;
370	pjm2	1.32	if (!$response eq "OK\n") {
371	pjm2	1.1	close($sock);
372			print "Server gave wrong response to CONFIG: $response\n";
373	pjm2	1.27	&tcp_configure();
374	pjm2	1.1	return;
375			}
376
377			print $sock "$file_list\n";
378			$response = <$sock>;
379	pjm2	1.32	if (!$response eq "OK\n") {
380	pjm2	1.1	close($sock);
381			print "Server gave wrong response to file list: $response\n";
382	pjm2	1.27	&tcp_configure();
383	pjm2	1.1	return;
384			}
385
386			print $sock "$last_modified\n";
387			$response = <$sock>;
388	pjm2	1.32	if ($response eq "ERROR\n") {
389	pjm2	1.1	close($sock);
390	pjm2	1.27	print "Server configuration changed. Reconfiguring with filter manager.\n";
391	pjm2	1.34	$doReconfigure = 1;
392	pjm2	1.1	}
393	pjm2	1.32	if (!$response eq "OK\n") {
394	pjm2	1.1	close($sock);
395			print "Server gave wrong response to HEARTBEAT: $response\n";
396	pjm2	1.27	&tcp_configure();
397	pjm2	1.1	return;
398			}
399	tdb	1.47
400			print $sock "KEY\n";
401			$key = <$sock>;
402	pjm2	1.1
403			print $sock "ENDHEARTBEAT\n";
404	pjm2	1.23	$response = <$sock>;
405	pjm2	1.32	if (!$response eq "OK\n") {
406	pjm2	1.1	close($sock);
407			print "Server gave wrong response to ENDHEARTBEAT: $response\n";
408	pjm2	1.27	&tcp_configure();
409	pjm2	1.1	return;
410			}
411
412			close($sock);
413			print "^";
414	pjm2	1.34
415			&tcp_configure() if $doReconfigure;
416	tdb	1.25
417			return;
418			}
419
420	tdb	1.36
421	tdb	1.25	#-----------------------------------------------------------------------
422			# write_pid
423			# Writes the PID (process ID) of this instance to $pidfile.
424			# This is then used by a seperate script to check (and restart) ihost.
425			#-----------------------------------------------------------------------
426			sub write_pid() {
427			open PID, ">$pidfile";
428			print PID $$;
429			close PID;
430	pjm2	1.12
431			return;
432	tdb	1.36	}
433
434			#-----------------------------------------------------------------------
435			# make_xml
436	tdb	1.37	# Turns an array of plugins data into an XML string.
437	tdb	1.36	#-----------------------------------------------------------------------
438			sub make_xml() {
439			my($curlevel, $curline) = @_;
440			my($xmltemp) = ""; my($curtag) = ""; my($attributes) = "";
441	tdb	1.40	while(1) {
442	tdb	1.43	$curline = shift(@data) if $curline eq "";
443			return $xmltemp if not defined $curline;
444			chomp $curline;
445			# dealing with nest (or attributes)
446	tdb	1.36	if($curline =~ /^$curlevel([^\.\s]+\.)/) {
447			$curtag=$1;
448	tdb	1.43	if($curline =~ /^$curlevel$curtag([^\.\s]+)\s+(.*)$/) {
449			$xmltemp .= &make_xml("$curlevel$curtag", $curline);
450			}
451			elsif($curline =~ /^$curlevel$curtag(attributes)\.([^\.\s]+)\s+(.*)$/) {
452			$attributes .= " $2=\"$3\"";
453			}
454			else {
455			$xmltemp .= &make_xml("$curlevel$curtag", $curline);
456			}
457			my($nextline) = $data[0]; chomp $nextline if defined $nextline;
458	tdb	1.45	$curtag =~ s/(.*)\./$1/;
459			if((defined $nextline) && ($nextline =~ /^$curlevel$curtag\./)) {
460	tdb	1.43	$curline = "";
461			}
462			else {
463			$xmltemp = "<$curtag$attributes>$xmltemp</$curtag>" unless $curtag eq "";
464			return $xmltemp;
465			}
466	tdb	1.36	}
467	tdb	1.43	# dealing with value
468			elsif($curline =~ /^$curlevel([^\.\s]+)\s+(.*)$/) {
469			$curtag=$1;
470			$xmltemp=$2;
471			my($nextline) = $data[0]; chomp $nextline if defined $nextline;
472			if(defined $nextline && ($nextline =~ /^$curlevel$curtag\./ \|\| $nextline =~ /^$curlevel$curtag\s+/)) {
473			$curline = "";
474			}
475			else {
476			$xmltemp = "<$curtag$attributes>$xmltemp</$curtag>" unless $curtag eq "";
477			return $xmltemp;
478			}
479	tdb	1.36	}
480	tdb	1.46	# dealing with a null value
481			elsif($curline =~ /^$curlevel([^\.\s]+)$/) {
482			# simply adding a space makes the above elsif deal with it :)
483			# just level with an empty tag in the XML
484			$curline .= " ";
485			}
486			# failing all that, skip the line
487			else {
488			$curline = "";
489			}
490	tdb	1.36	}
491	pjm2	1.1	}