ViewVC Help
View File | Revision Log | Show Annotations | Revision Graph | Root Listing
root/i-scream/projects/cms/source/host/ihost-perl/ihost.pl
Revision: 1.24
Committed: Tue Mar 6 08:38:42 2001 UTC (23 years, 6 months ago) by pjm2
Content type: text/plain
Branch: MAIN
Changes since 1.23: +9 -1 lines
Log Message:
The host now acts sensibly when the server either does not return anything
upon sending FILTER or if it returns a filter address that cannot be
correctly split into a hostname and pair of ports.

File Contents

# User Rev Content
1 pjm2 1.6 #!/usr/bin/perl -w
2 pjm2 1.1
3     # -----------------------------------------------------------
4     # Perl i-scream Host.
5 pjm2 1.12 # http://www.i-scream.org.uk
6 pjm2 1.1 #
7     # An all-in-one script to act as an i-scream host on
8     # a typical Unix/Linux box. You may adapt the data-gathering
9     # methods as you see fit.
10     # - pjm2@ukc.ac.uk
11     #
12 pjm2 1.22 # $Author: pjm2 $
13 pjm2 1.24 # $Id: ihost.pl,v 1.23 2001/03/01 18:53:42 pjm2 Exp $
14 pjm2 1.1 #------------------------------------------------------------
15    
16     $| = 1;
17    
18     use strict;
19     use IO::Socket;
20     use Sys::Hostname;
21    
22     use vars qw (
23     $filter_manager_addr
24     $filter_manager_port
25     $seq_no
26     $udp_update_time
27     $tcp_update_time
28     $last_udp_time
29     $last_tcp_time
30     $last_modified
31     $udp_port
32     $tcp_port
33     $filter_addr
34     $file_list
35 pjm2 1.17 $fqdn
36 pjm2 1.1 );
37    
38     if (@ARGV != 2) {
39     die "Usage: ihost.pl [i-scream filter manager] [TCP port]\n";
40     }
41    
42     $filter_manager_addr = $ARGV[0];
43     $filter_manager_port = $ARGV[1];
44    
45     $seq_no = 1;
46    
47     &tcp_configure();
48     &send_udp_packet();
49    
50     $last_udp_time = time;
51     $last_tcp_time = time;
52     while (1) {
53 pjm2 1.9 my($time) = time;
54     if ($time >= $last_udp_time + $udp_update_time) {
55 pjm2 1.1 &send_udp_packet();
56 pjm2 1.9 $last_udp_time = $time;
57 pjm2 1.1 }
58 pjm2 1.9 if ($time >= $last_tcp_time + $tcp_update_time) {
59 pjm2 1.1 &send_tcp_heartbeat();
60 pjm2 1.9 $last_tcp_time = $time;
61 pjm2 1.1 }
62 pjm2 1.9 my($next_udp) = $udp_update_time - $time + $last_udp_time;
63     my($next_tcp) = $tcp_update_time - $time + $last_tcp_time;
64     my($delay);
65     if ($next_udp < $next_tcp) {
66     $delay = $next_udp
67     }
68     else {
69     $delay = $next_tcp;
70     }
71     `sleep $delay`;
72 pjm2 1.1 }
73    
74     exit(0);
75    
76 pjm2 1.12
77     #-----------------------------------------------------------------------
78     # tcp_configure
79     # Establishes a TCP connection to the specified i-scream filter manager.
80     # The host then requests details from the server, such as the intervals
81     # at which to send UDP packets.
82     #-----------------------------------------------------------------------
83 pjm2 1.1 sub tcp_configure() {
84    
85     my($sock) = new IO::Socket::INET(
86     PeerAddr => $filter_manager_addr,
87     PeerPort => $filter_manager_port,
88     Proto => 'tcp'
89 pjm2 1.21 ) or die "Cannot connect!";
90 pjm2 1.11 if (!defined $sock) {
91     print "IHOST ERROR: Could not connect to $filter_manager_addr:$filter_manager_port.\n";
92     print "Please check that there is an i-scream server at this address.\n";
93     print "Program ended.\n";
94     exit(1);
95     }
96 pjm2 1.1
97 pjm2 1.12 # Now run through the configuration process...
98 pjm2 1.1 my($response);
99    
100     print $sock "STARTCONFIG\n";
101     $response = <$sock>;
102     if (!chop $response eq "OK") {
103     print "The i-scream server rejected the STARTCONFIG command. Terminated.";
104     exit(1);
105     }
106    
107     print "Config started okay.\n";
108    
109     print $sock "LASTMODIFIED\n";
110     $response = <$sock>;
111     chop $response;
112     $last_modified = $response;
113    
114 pjm2 1.7 print "Config last modified: ". (scalar localtime $last_modified/1000) . "\n";
115 pjm2 1.1
116     print $sock "FILELIST\n";
117     $response = <$sock>;
118     chop $response;
119     $file_list = $response;
120    
121     print "File list obtained: $file_list\n";
122 pjm2 1.17
123     print $sock "FQDN\n";
124     $response = <$sock>;
125     chop $response;
126     $fqdn = $response;
127    
128     print "FQDN returned: $fqdn\n";
129 pjm2 1.1
130     print $sock "UDPUpdateTime\n";
131     $response = <$sock>;
132     chop $response;
133     $udp_update_time = $response;
134    
135     print $sock "TCPUpdateTime\n";
136     $response = <$sock>;
137     chop $response;
138     $tcp_update_time = $response;
139    
140 pjm2 1.7 print "UDP packet period: $udp_update_time seconds.\nTCP heartbeat period: $tcp_update_time seconds.\n";
141 pjm2 1.1
142     print $sock "ENDCONFIG\n";
143     $response = <$sock>;
144     chomp $response;
145     if (!$response eq "OK") {
146     print "ENDCONFIG command to server failed. Terminated.\n";
147     exit(1);
148     }
149    
150     print "Config ended.\n";
151    
152     print $sock "FILTER\n";
153     $response = <$sock>;
154 pjm2 1.24 unless (defined($response)) {
155     print "Failed: Could not get a filter address from the filter manager.\n";
156     exit(1);
157     }
158 pjm2 1.1 chop $response;
159     $response =~ /(.*);(.*);(.*)/;
160     ($filter_addr, $udp_port, $tcp_port) = ($1, $2, $3);
161 pjm2 1.24 unless (defined($filter_addr) && defined($udp_port) && defined($tcp_port)) {
162     print "Failed: Filter address response from server did not make sense: $response\n";
163     exit(1);
164     }
165 pjm2 1.1
166     print "Got filter data ($filter_addr, $udp_port, $tcp_port)\n";
167    
168     print $sock "END\n";
169     $response = <$sock>;
170     chop $response;
171     if ($response eq "OK") {
172     print "Host successfully configured via TCP.\n"
173     }
174     else {
175     print "The server failed the host configuration on the END command.";
176     exit(1);
177     }
178    
179     close($sock);
180    
181     print "Configuration finished sucessfully!\n";
182 pjm2 1.12
183     return;
184 pjm2 1.1 }
185    
186 pjm2 1.12
187    
188    
189     #-----------------------------------------------------------------------
190     # send_udp_packet
191     # Sends a UDP packet to an i-scream filter.
192     # The packet contains XML markup describing some of the machine's state.
193     # Receipt of UDP packets is not guaranteed.
194     #-----------------------------------------------------------------------
195 pjm2 1.1 sub send_udp_packet() {
196    
197     my(@statgrab) = `./statgrab.pl`;
198     my(%packet);
199 pjm2 1.13 for (my($i) = 0; $i <= $#statgrab; $i++) {
200 pjm2 1.1 $statgrab[$i] =~ /^([^\s]*) (.*)$/;
201     $packet{$1} = $2;
202     }
203    
204     my($date) = time;
205    
206     my($disk_info) = "<disk>";
207     my($i) = 0;
208     while (defined $packet{"packet.disk.p$i.attributes.mount"}) {
209 pjm2 1.15 $disk_info .= "<p$i";
210     $disk_info .= " name=\"" . $packet{"packet.disk.p$i.attributes.name"} . "\"";
211     $disk_info .= " kbytes=\"" . $packet{"packet.disk.p$i.attributes.kbytes"} . "\"";
212     $disk_info .= " used=\"" . $packet{"packet.disk.p$i.attributes.used"} . "\"";
213     $disk_info .= " avail=\"" . $packet{"packet.disk.p$i.attributes.avail"} . "\"";
214     $disk_info .= " mount=\"" . $packet{"packet.disk.p$i.attributes.mount"} . "\"";
215     $disk_info .= "></p$i>";
216 pjm2 1.1 ++$i;
217     }
218     $disk_info .= "</disk>";
219 pjm2 1.13
220 tdb 1.19 my($ip) = inet_ntoa(scalar(gethostbyname(hostname())) || 'localhost');
221 pjm2 1.1
222     # Build the XML packet this way, as we can clearly
223     # see the structure and contents... I like this ;-)
224 pjm2 1.12 # [Note that the server rejects UDP packets that are
225     # larger than 8196 bytes]
226 pjm2 1.1 my($xml) = <<EOF;
227    
228 tdb 1.19 <packet seq_no="$seq_no" machine_name="$fqdn" date="$date" type="data" ip="$ip">
229 pjm2 1.1 <load>
230     <load1>$packet{"packet.load.load1"}</load1>
231     <load5>$packet{"packet.load.load5"}</load5>
232     <load15>$packet{"packet.load.load15"}</load15>
233     </load>
234     <os>
235     <name>$packet{"packet.os.name"}</name>
236     <release>$packet{"packet.os.release"}</release>
237     <platform>$packet{"packet.os.platform"}</platform>
238     <sysname>$packet{"packet.os.sysname"}</sysname>
239     <version>$packet{"packet.os.version"}</version>
240 tdb 1.10 <uptime>$packet{"packet.os.uptime"}</uptime>
241 pjm2 1.1 </os>
242     <users>
243     <count>$packet{"packet.users.count"}</count>
244     <list>$packet{"packet.users.list"}</list>
245     </users>
246     <processes>
247     <total>$packet{"packet.processes.total"}</total>
248     <sleeping>$packet{"packet.processes.sleeping"}</sleeping>
249     <zombie>$packet{"packet.processes.zombie"}</zombie>
250     <stopped>$packet{"packet.processes.stopped"}</stopped>
251     <cpu>$packet{"packet.processes.cpu"}</cpu>
252     </processes>
253     <cpu>
254     <idle>$packet{"packet.cpu.idle"}</idle>
255     <user>$packet{"packet.cpu.user"}</user>
256     <kernel>$packet{"packet.cpu.kernel"}</kernel>
257     <iowait>$packet{"packet.cpu.iowait"}</iowait>
258     <swap>$packet{"packet.cpu.swap"}</swap>
259     </cpu>
260     <memory>
261 pjm2 1.13 <total>$packet{"packet.memory.total"}</total>
262 pjm2 1.1 <free>$packet{"packet.memory.free"}</free>
263     </memory>
264     <swap>
265 pjm2 1.13 <total>$packet{"packet.swap.total"}</total>
266     <free>$packet{"packet.swap.free"}</free>
267 pjm2 1.1 </swap>
268     $disk_info
269     </packet>
270    
271     EOF
272    
273 pjm2 1.12 # Make the packet smaller by stripping out newlines and leading spaces.
274 pjm2 1.1 $xml =~ s/\n\s*//g;
275 tdb 1.19
276 pjm2 1.1 my($sock) = new IO::Socket::INET (
277     PeerPort => $udp_port,
278     PeerAddr => $filter_addr,
279     Proto => 'udp'
280 pjm2 1.21 ) or die "Could not send UDP: $!\n";
281 pjm2 1.1
282     print $sock $xml or die "Could not send UDP packet: $!\n";
283     close($sock);
284     $seq_no++;
285     print "-";
286 pjm2 1.12
287     return;
288 pjm2 1.1 }
289    
290 pjm2 1.12
291    
292    
293     #-----------------------------------------------------------------------
294     # send_tcp_heartbeat
295     # Establishes a TCP connection to an i-scream filter.
296     # The heartbeat is used as a guaranteed "I'm alive" delivery mechanism.
297     #-----------------------------------------------------------------------
298 pjm2 1.1 sub send_tcp_heartbeat() {
299    
300     my($sock) = new IO::Socket::INET(
301     PeerAddr => $filter_addr,
302     PeerPort => $tcp_port,
303     Proto => 'tcp'
304 pjm2 1.21 ) or return;
305 pjm2 1.11 if (!defined $sock) {
306     print "IHOST WARNING: Failed to deliver a heartbeat to the i-scream filter.\n";
307     return;
308     }
309 pjm2 1.1
310     # Now run through the configuration process.
311     my($response);
312    
313     print $sock "HEARTBEAT\n";
314     $response = <$sock>;
315 pjm2 1.23 return if (!defined $response);
316     chop $response;
317 pjm2 1.1 if (!$response eq "OK") {
318     close($sock);
319     print "Server gave wrong response to HEARTBEAT: $response\n";
320     return;
321     }
322    
323     print $sock "CONFIG\n";
324     $response = <$sock>;
325 pjm2 1.23 return if (!defined $response);
326     chop $response;
327 pjm2 1.1 if (!$response eq "OK") {
328     close($sock);
329     print "Server gave wrong response to CONFIG: $response\n";
330     return;
331     }
332    
333     print $sock "$file_list\n";
334     $response = <$sock>;
335 pjm2 1.23 return if (!defined $response);
336     chop $response;
337 pjm2 1.1 if (!$response eq "OK") {
338     close($sock);
339     print "Server gave wrong response to file list: $response\n";
340     return;
341     }
342    
343     print $sock "$last_modified\n";
344     $response = <$sock>;
345 pjm2 1.23 return if (!defined $response);
346     chop $response;
347 pjm2 1.1 if ($response eq "ERROR") {
348     close($sock);
349     &tcp_configure();
350     return;
351     }
352     if (!$response eq "OK") {
353     close($sock);
354     print "Server gave wrong response to HEARTBEAT: $response\n";
355     return;
356     }
357    
358     print $sock "ENDHEARTBEAT\n";
359 pjm2 1.23 $response = <$sock>;
360     return if (!defined $response);
361 pjm2 1.1 chop $response;
362     if (!$response eq "OK") {
363     close($sock);
364     print "Server gave wrong response to ENDHEARTBEAT: $response\n";
365     return;
366     }
367    
368     close($sock);
369     print "^";
370 pjm2 1.12
371     return;
372 pjm2 1.1 }