forked from virtualmin/virtualmin-gpl
-
Notifications
You must be signed in to change notification settings - Fork 0
/
clam-wrapper.pl
executable file
·56 lines (50 loc) · 1.29 KB
/
clam-wrapper.pl
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
#!/usr/local/bin/perl
# Runs clamdscan, and exits with 1 if the program reports a virus was found
# or 0 if not was found or some error happened
$prog = join(" ", map { quotemeta($_) } @ARGV);
$temp = "/tmp/clamwrapper.$$";
unlink($temp);
# Check if we are using the streaming client, which has slightly different
# semantics
$stream_client = $ARGV[0] =~ /clamd-stream-client/ ? 1 : 0;
$fullprog = $prog;
if (!$stream_client) {
$fullprog .= " -";
}
# Feed email to clamscan
$SIG{'PIPE'} = 'ignore';
$clampid = open(INPUT, "|$fullprog >$temp");
while(read(STDIN, $buf, 32768) > 0) {
print INPUT $buf;
}
# Wait at most 30 seconds for a response
$timed_out = 0;
$SIG{'ALRM'} = sub { $timed_out++ };
alarm(30);
close(INPUT);
alarm(0);
if ($timed_out) {
print STDERR "Virus scanner failed to response within 30 seconds\n";
kill('KILL', $clampid);
unlink($temp);
exit(0);
}
# Read back status from clamscan, and exit non-zero if a virus was found
open(OUTPUT, "<".$temp);
while(<OUTPUT>) {
$out .= $_;
}
close(OUTPUT);
unlink($temp);
if ($stream_client && $out =~ /\S/) {
# Using clamd-stream-client - output contains a virus name if found
exit(1);
}
elsif (!$stream_client && $out =~ /FOUND/) {
# Using clamscan or clamdscan - output contains FOUND if a virus
# was detected
exit(1);
}
else {
exit(0);
}