ViewVC Help
View File | Revision Log | Show Annotations | Revision Graph | Root Listing
root/i-scream/projects/cms/source/host/ihost-perl/ihost.pl
(Generate patch)

Comparing projects/cms/source/host/ihost-perl/ihost.pl (file contents):
Revision 1.2 by pjm2, Fri Jan 26 17:26:11 2001 UTC vs.
Revision 1.22 by pjm2, Thu Mar 1 10:49:27 2001 UTC

# Line 1 | Line 1
1 < #!/usr/local/bin/perl -w
1 > #!/usr/bin/perl -w
2  
3   # -----------------------------------------------------------
4   # Perl i-scream Host.
5 + # http://www.i-scream.org.uk
6   #
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
# Line 31 | Line 32 | use vars qw (
32               $tcp_port
33               $filter_addr
34               $file_list
35 +             $fqdn
36              );
37  
38   if (@ARGV != 2) {
# Line 48 | Line 50 | $seq_no = 1;
50   $last_udp_time = time;
51   $last_tcp_time = time;
52   while (1) {
53 <    if (time >= $last_udp_time + $udp_update_time) {
53 >    my($time) = time;
54 >    if ($time >= $last_udp_time + $udp_update_time) {
55          &send_udp_packet();
56 <        $last_udp_time = time;
56 >        $last_udp_time = $time;
57      }
58 <    if (time >= $last_tcp_time + $tcp_update_time) {
58 >    if ($time >= $last_tcp_time + $tcp_update_time) {
59          &send_tcp_heartbeat();
60 <        $last_tcp_time = time;
60 >        $last_tcp_time = $time;
61      }
62 <    `sleep 1`;
62 >    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   }
73  
74   exit(0);
75  
76 +
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   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 <                                    ) or die "Could not perform configuration via TCP: $!\n";
89 >                                    ) or die "Cannot connect!";
90 >    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  
97 <    die "Could not connect to the i-scream filter manager: $!\n" unless $sock;
73 <
74 <    # Now run through the configuration process.
97 >    # Now run through the configuration process...
98      my($response);
99      
100      print $sock "STARTCONFIG\n";
# Line 88 | Line 111 | sub tcp_configure() {
111      chop $response;
112      $last_modified = $response;
113      
114 <    print "Config last modified: ". scalar localtime $last_modified . "\n";
114 >    print "Config last modified: ". (scalar localtime $last_modified/1000) . "\n";
115  
116      print $sock "FILELIST\n";
117      $response = <$sock>;
# Line 97 | Line 120 | sub tcp_configure() {
120  
121      print "File list obtained: $file_list\n";
122  
123 +    print $sock "FQDN\n";
124 +    $response = <$sock>;
125 +    chop $response;
126 +    $fqdn = $response;
127 +
128 +    print "FQDN returned: $fqdn\n";
129 +
130      print $sock "UDPUpdateTime\n";
131      $response = <$sock>;
132      chop $response;
# Line 107 | Line 137 | sub tcp_configure() {
137      chop $response;
138      $tcp_update_time = $response;
139      
140 <    print "UDP packet period: $udp_update_time seconds.\nTCP heartbeat period: $tcp_update_time.\n";
140 >    print "UDP packet period: $udp_update_time seconds.\nTCP heartbeat period: $tcp_update_time seconds.\n";
141  
142      print $sock "ENDCONFIG\n";
143      $response = <$sock>;
# Line 141 | Line 171 | sub tcp_configure() {
171      close($sock);
172  
173      print "Configuration finished sucessfully!\n";
174 +    
175 +    return;
176   }
177  
178 +
179 +
180 +
181 + #-----------------------------------------------------------------------
182 + # send_udp_packet
183 + # Sends a UDP packet to an i-scream filter.
184 + # The packet contains XML markup describing some of the machine's state.
185 + # Receipt of UDP packets is not guaranteed.
186 + #-----------------------------------------------------------------------
187   sub send_udp_packet() {
188  
189      my(@statgrab) = `./statgrab.pl`;
190      my(%packet);
191 <    for (my($i) = 0; $i < $#statgrab; $i++) {
191 >    for (my($i) = 0; $i <= $#statgrab; $i++) {
192          $statgrab[$i] =~ /^([^\s]*) (.*)$/;
193          $packet{$1} = $2;
194      }
# Line 157 | Line 198 | sub send_udp_packet() {
198      my($disk_info) = "<disk>";
199      my($i) = 0;
200      while (defined $packet{"packet.disk.p$i.attributes.mount"}) {
201 <        $disk_info .= "<p$i>";
202 <        $disk_info .= qq/<name>$packet{"packet.disk.p$i.attributes.name"}<\/name>/;
203 <        $disk_info .= qq/<kbytes>$packet{"packet.disk.p$i.attributes.kbytes"}<\/kbytes>/;
204 <        $disk_info .= qq/<used>$packet{"packet.disk.p$i.attributes.used"}<\/used>/;
205 <        $disk_info .= qq/<avail>$packet{"packet.disk.p$i.attributes.avail"}<\/avail>/;
206 <        $disk_info .= qq/<mount>$packet{"packet.disk.p$i.attributes.mount"}<\/mount>/;
207 <        $disk_info .= "</p$i>";
201 >        $disk_info .= "<p$i";
202 >        $disk_info .= " name=\"" . $packet{"packet.disk.p$i.attributes.name"} . "\"";
203 >        $disk_info .= " kbytes=\"" . $packet{"packet.disk.p$i.attributes.kbytes"} . "\"";
204 >        $disk_info .= " used=\"" . $packet{"packet.disk.p$i.attributes.used"} . "\"";
205 >        $disk_info .= " avail=\"" . $packet{"packet.disk.p$i.attributes.avail"} . "\"";
206 >        $disk_info .= " mount=\"" . $packet{"packet.disk.p$i.attributes.mount"} . "\"";
207 >        $disk_info .= "></p$i>";
208          ++$i;
209      }
210      $disk_info .= "</disk>";
211  
212 <    my($hostname) = hostname();
172 <    $hostname =~ s/^([^\.]*?).*$/$1/;
173 <    my($domainname) = `cat /etc/resolv.conf`;
174 <    $domainname =~ s/domain\s*([^\s]*?)/$1/;
175 <    my($machine_name) = "$hostname.$domainname";
176 <    my($ip) = inet_ntoa(scalar(gethostbyname($hostname)) || 'localhost');
212 >    my($ip) = inet_ntoa(scalar(gethostbyname(hostname())) || 'localhost');
213  
214      # Build the XML packet this way, as we can clearly
215      # see the structure and contents... I like this ;-)
216 +    # [Note that the server rejects UDP packets that are
217 +    # larger than 8196 bytes]
218      my($xml) = <<EOF;
219      
220 < <packet seq_no="$seq_no" machine_name="$machine_name" date="$date" type="data" ip="$ip">
220 > <packet seq_no="$seq_no" machine_name="$fqdn" date="$date" type="data" ip="$ip">
221      <load>
222          <load1>$packet{"packet.load.load1"}</load1>
223          <load5>$packet{"packet.load.load5"}</load5>
# Line 191 | Line 229 | sub send_udp_packet() {
229          <platform>$packet{"packet.os.platform"}</platform>
230          <sysname>$packet{"packet.os.sysname"}</sysname>
231          <version>$packet{"packet.os.version"}</version>
232 +        <uptime>$packet{"packet.os.uptime"}</uptime>
233      </os>
234      <users>
235          <count>$packet{"packet.users.count"}</count>
# Line 211 | Line 250 | sub send_udp_packet() {
250          <swap>$packet{"packet.cpu.swap"}</swap>
251      </cpu>
252      <memory>
253 <        <total>$packet{"packet.memory.real"}</total>
253 >        <total>$packet{"packet.memory.total"}</total>
254          <free>$packet{"packet.memory.free"}</free>
255      </memory>
256      <swap>
257 <        <total>$packet{"packet.memory.swap_total"}</total>
258 <        <free>$packet{"packet.memory.swap_free"}</free>
257 >        <total>$packet{"packet.swap.total"}</total>
258 >        <free>$packet{"packet.swap.free"}</free>
259      </swap>
260      $disk_info
261   </packet>
262  
263   EOF
264  
265 +    # Make the packet smaller by stripping out newlines and leading spaces.
266      $xml =~ s/\n\s*//g;
267 <    
267 >
268      my($sock) = new IO::Socket::INET (
269                                        PeerPort => $udp_port,
270                                        PeerAddr => $filter_addr,
271                                        Proto => 'udp'
272 <                                     ) or die "Socket: $!\n";
272 >                                     ) or die "Could not send UDP: $!\n";
273  
274      print $sock $xml or die "Could not send UDP packet: $!\n";
275      close($sock);
276      $seq_no++;
277      print "-";
278 +    
279 +    return;
280   }
281  
282 +
283 +
284 +
285 + #-----------------------------------------------------------------------
286 + # send_tcp_heartbeat
287 + # Establishes a TCP connection to an i-scream filter.
288 + # The heartbeat is used as a guaranteed "I'm alive" delivery mechanism.
289 + #-----------------------------------------------------------------------
290   sub send_tcp_heartbeat() {
291  
292      my($sock) = new IO::Socket::INET(
293                                       PeerAddr => $filter_addr,
294                                       PeerPort => $tcp_port,
295                                       Proto => 'tcp'
296 <                                    ) or die "Could not perform heartbeat via TCP: $!\n";
296 >                                    ) or return;
297 >    if (!defined $sock) {
298 >        print "IHOST WARNING: Failed to deliver a heartbeat to the i-scream filter.\n";
299 >        return;
300 >    }
301  
248    die "Could not connect to the i-scream filter: $!\n" unless $sock;
249
302      # Now run through the configuration process.
303      my($response);
304  
305      print $sock "HEARTBEAT\n";
306      $response = <$sock>;
307 <    chop $response;
307 >    chop $response or return;
308      if (!$response eq "OK") {
309          close($sock);
310          print "Server gave wrong response to HEARTBEAT: $response\n";
# Line 261 | Line 313 | sub send_tcp_heartbeat() {
313      
314      print $sock "CONFIG\n";
315      $response = <$sock>;
316 <    chop $response;
316 >    chop $response or return;
317      if (!$response eq "OK") {
318          close($sock);
319          print "Server gave wrong response to CONFIG: $response\n";
# Line 270 | Line 322 | sub send_tcp_heartbeat() {
322  
323      print $sock "$file_list\n";
324      $response = <$sock>;
325 <    chop $response;
325 >    chop $response or return;
326      if (!$response eq "OK") {
327          close($sock);
328          print "Server gave wrong response to file list: $response\n";
# Line 279 | Line 331 | sub send_tcp_heartbeat() {
331      
332      print $sock "$last_modified\n";
333      $response = <$sock>;
334 <    chop $response;
334 >    chop $response or return;
335      if ($response eq "ERROR") {
336          close($sock);
337          &tcp_configure();
# Line 292 | Line 344 | sub send_tcp_heartbeat() {
344      }
345      
346      print $sock "ENDHEARTBEAT\n";
347 <    $response = <$sock>;
347 >    $response = <$sock> or return;
348      chop $response;
349      if (!$response eq "OK") {
350          close($sock);
# Line 302 | Line 354 | sub send_tcp_heartbeat() {
354      
355      close($sock);
356      print "^";
357 +    
358 +    return;
359   }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines