45 |
|
$filter_manager_port = $ARGV[1]; |
46 |
|
|
47 |
|
$seq_no = 1; |
48 |
< |
$retry_wait = 6; |
48 |
> |
$retry_wait = 60; |
49 |
|
|
50 |
|
# write our PID to a file |
51 |
|
$pidfile = "/var/tmp/ihost.pid"; |
102 |
|
#----------------------------------------------------------------------- |
103 |
|
sub tcp_configure() { |
104 |
|
|
105 |
< |
while (1 == 1) { |
105 |
> |
while (1) { |
106 |
|
my($sock) = new IO::Socket::INET( |
107 |
|
PeerAddr => $filter_manager_addr, |
108 |
|
PeerPort => $filter_manager_port, |
111 |
|
if (!defined $sock) { |
112 |
|
print "IHOST ERROR: Could not connect to $filter_manager_addr:$filter_manager_port.\n"; |
113 |
|
print "Please check that there is an i-scream server at this address.\n"; |
114 |
– |
close($sock); |
114 |
|
wait_then_retry(); |
115 |
|
next; |
116 |
|
} |
120 |
|
|
121 |
|
print $sock "STARTCONFIG\n"; |
122 |
|
$response = <$sock>; |
123 |
< |
if (!chop $response eq "OK") { |
124 |
< |
print "The i-scream server rejected the STARTCONFIG command. Terminated."; |
123 |
> |
if ($response && !($response eq "OK\n")) { |
124 |
> |
print "The i-scream server rejected the STARTCONFIG command.\n"; |
125 |
|
close($sock); |
126 |
|
wait_then_retry(); |
127 |
|
next; |
131 |
|
|
132 |
|
print $sock "LASTMODIFIED\n"; |
133 |
|
$response = <$sock>; |
134 |
+ |
if (!$response) { |
135 |
+ |
print "The i-scream server did not return anything for the LASTMODIFIED command.\n"; |
136 |
+ |
close($sock); |
137 |
+ |
wait_then_retry(); |
138 |
+ |
next; |
139 |
+ |
} |
140 |
|
chop $response; |
141 |
|
$last_modified = $response; |
142 |
|
|
144 |
|
|
145 |
|
print $sock "FILELIST\n"; |
146 |
|
$response = <$sock>; |
147 |
+ |
if (!$response) { |
148 |
+ |
print "The i-scream server did not provide a configuration file list.\n"; |
149 |
+ |
close($sock); |
150 |
+ |
wait_then_retry(); |
151 |
+ |
next; |
152 |
+ |
} |
153 |
|
chop $response; |
154 |
|
$file_list = $response; |
155 |
|
|
157 |
|
|
158 |
|
print $sock "FQDN\n"; |
159 |
|
$response = <$sock>; |
160 |
+ |
if (!$response) { |
161 |
+ |
print "The i-scream server did not tell us our FQDN.\n"; |
162 |
+ |
close($sock); |
163 |
+ |
wait_then_retry(); |
164 |
+ |
next; |
165 |
+ |
} |
166 |
|
chop $response; |
167 |
|
$fqdn = $response; |
168 |
|
|
170 |
|
|
171 |
|
print $sock "UDPUpdateTime\n"; |
172 |
|
$response = <$sock>; |
173 |
+ |
if (!$response) { |
174 |
+ |
print "The i-scream server did not give us a UDPUpdateTime.\n"; |
175 |
+ |
close($sock); |
176 |
+ |
wait_then_retry(); |
177 |
+ |
next; |
178 |
+ |
} |
179 |
|
chop $response; |
180 |
|
$udp_update_time = $response; |
181 |
|
|
182 |
|
print $sock "TCPUpdateTime\n"; |
183 |
|
$response = <$sock>; |
184 |
+ |
if (!$response) { |
185 |
+ |
print "The i-scream server did not give us a TCPUpdateTime.\n"; |
186 |
+ |
close($sock); |
187 |
+ |
wait_then_retry(); |
188 |
+ |
next; |
189 |
+ |
} |
190 |
|
chop $response; |
191 |
|
$tcp_update_time = $response; |
192 |
|
|
194 |
|
|
195 |
|
print $sock "ENDCONFIG\n"; |
196 |
|
$response = <$sock>; |
197 |
< |
chomp $response; |
169 |
< |
if (!$response eq "OK") { |
197 |
> |
if ($response && !($response eq "OK\n")) { |
198 |
|
print "ENDCONFIG command to server failed. Terminated.\n"; |
199 |
|
close($sock); |
200 |
|
wait_then_retry(); |
205 |
|
|
206 |
|
print $sock "FILTER\n"; |
207 |
|
$response = <$sock>; |
208 |
< |
unless (defined($response)) { |
208 |
> |
if (!$response) { |
209 |
|
print "Failed: Could not get a filter address from the filter manager.\n"; |
210 |
|
close($sock); |
211 |
|
wait_then_retry(); |
212 |
|
next; |
213 |
|
} |
214 |
|
chop $response; |
215 |
< |
$response =~ /(.*);(.*);(.*)/; |
215 |
> |
$response =~ /^(.*);(.*);(.*)/; |
216 |
|
if ($response eq "ERROR") { |
217 |
|
print "There are no active configured filters for your host.\n"; |
218 |
|
close($sock); |
231 |
|
|
232 |
|
print $sock "END\n"; |
233 |
|
$response = <$sock>; |
234 |
< |
chop $response; |
207 |
< |
if ($response eq "OK") { |
234 |
> |
if ($response && ($response eq "OK\n")) { |
235 |
|
print "Host successfully configured via TCP.\n" |
236 |
|
} |
237 |
|
else { |
238 |
< |
print "The server failed the host configuration on the END command."; |
238 |
> |
print "The server failed the host configuration on the END command.\n"; |
239 |
|
close($sock); |
240 |
|
wait_then_retry(); |
241 |
|
next; |
283 |
|
} |
284 |
|
$disk_info .= "</disk>"; |
285 |
|
|
286 |
< |
my($ip) = inet_ntoa(scalar(gethostbyname(hostname())) || 'localhost'); |
286 |
> |
my($ip); |
287 |
> |
$ip = inet_ntoa(scalar(gethostbyname(hostname())) || 'localhost') or $ip = 'localhost'; |
288 |
|
|
289 |
|
# Build the XML packet this way, as we can clearly |
290 |
|
# see the structure and contents... I like this ;-) |
380 |
|
|
381 |
|
print $sock "HEARTBEAT\n"; |
382 |
|
$response = <$sock>; |
383 |
< |
return if (!defined $response); |
356 |
< |
chop $response; |
357 |
< |
if (!$response eq "OK") { |
383 |
> |
if (!$response eq "OK\n") { |
384 |
|
close($sock); |
385 |
|
print "Server gave wrong response to HEARTBEAT: $response\n"; |
386 |
|
&tcp_configure(); |
389 |
|
|
390 |
|
print $sock "CONFIG\n"; |
391 |
|
$response = <$sock>; |
392 |
< |
return if (!defined $response); |
367 |
< |
chop $response; |
368 |
< |
if (!$response eq "OK") { |
392 |
> |
if (!$response eq "OK\n") { |
393 |
|
close($sock); |
394 |
|
print "Server gave wrong response to CONFIG: $response\n"; |
395 |
|
&tcp_configure(); |
398 |
|
|
399 |
|
print $sock "$file_list\n"; |
400 |
|
$response = <$sock>; |
401 |
< |
return if (!defined $response); |
378 |
< |
chop $response; |
379 |
< |
if (!$response eq "OK") { |
401 |
> |
if (!$response eq "OK\n") { |
402 |
|
close($sock); |
403 |
|
print "Server gave wrong response to file list: $response\n"; |
404 |
|
&tcp_configure(); |
407 |
|
|
408 |
|
print $sock "$last_modified\n"; |
409 |
|
$response = <$sock>; |
410 |
< |
return if (!defined $response); |
389 |
< |
chop $response; |
390 |
< |
if ($response eq "ERROR") { |
410 |
> |
if ($response eq "ERROR\n") { |
411 |
|
close($sock); |
412 |
|
print "Server configuration changed. Reconfiguring with filter manager.\n"; |
413 |
|
&tcp_configure(); |
414 |
|
return; |
415 |
|
} |
416 |
< |
if (!$response eq "OK") { |
416 |
> |
if (!$response eq "OK\n") { |
417 |
|
close($sock); |
418 |
|
print "Server gave wrong response to HEARTBEAT: $response\n"; |
419 |
|
&tcp_configure(); |
422 |
|
|
423 |
|
print $sock "ENDHEARTBEAT\n"; |
424 |
|
$response = <$sock>; |
425 |
< |
return if (!defined $response); |
406 |
< |
chop $response; |
407 |
< |
if (!$response eq "OK") { |
425 |
> |
if (!$response eq "OK\n") { |
426 |
|
close($sock); |
427 |
|
print "Server gave wrong response to ENDHEARTBEAT: $response\n"; |
428 |
|
&tcp_configure(); |