/*
* Copyright (C) 2002 - 2006 Tomasz Kojm <tkojm@clamav.net>
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License version 2 as
* published by the Free Software Foundation.
*
* 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., 51 Franklin Street, Fifth Floor, Boston,
* MA 02110-1301, USA.
*/
#if HAVE_CONFIG_H
#include "clamav-config.h"
#endif
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#ifdef HAVE_UNISTD_H
#include <unistd.h>
#endif
#ifdef C_WINDOWS
#include <fcntl.h>
#else
#include <sys/time.h>
#endif
#include <time.h>
#ifdef C_LINUX
#include <sys/resource.h>
#endif
#include "clamscan_opt.h"
#include "others.h"
#include "global.h"
#include "manager.h"
#include "treewalk.h"
#include "shared/misc.h"
#include "shared/output.h"
#include "shared/options.h"
void help(void);
#if defined(C_WINDOWS) && defined(CL_DEBUG)
#include <crtdbg.h>
#endif
struct s_info info;
short recursion = 0, printinfected = 0, bell = 0;
int main(int argc, char **argv)
{
int ds, dms, ret;
double mb;
struct timeval t1, t2;
#ifndef C_WINDOWS
struct timezone tz;
#endif
struct optstruct *opt;
const char *pt;
#if defined(C_WINDOWS) && defined(CL_THREAD_SAFE)
if(!pthread_win32_process_attach_np()) {
mprintf("!Can't start the win32 pthreads layer\n");
return 72;
}
#endif
opt = opt_parse(argc, argv, clamscan_shortopt, clamscan_longopt, NULL);
if(!opt) {
mprintf("!Can't parse the command line\n");
return 40;
}
if(opt_check(opt, "verbose")) {
mprintf_verbose = 1;
logg_verbose = 1;
}
if(opt_check(opt, "quiet"))
mprintf_quiet = 1;
if(opt_check(opt, "stdout"))
mprintf_stdout = 1;
if(opt_check(opt, "debug")) {
#if defined(C_LINUX)
/* njh@bandsman.co.uk: create a dump if needed */
struct rlimit rlim;
rlim.rlim_cur = rlim.rlim_max = RLIM_INFINITY;
if(setrlimit(RLIMIT_CORE, &rlim) < 0)
perror("setrlimit");
#endif
cl_debug(); /* enable debug messages */
}
if(opt_check(opt, "version")) {
opt_free(opt);
print_version();
return 0;
}
if(opt_check(opt, "help")) {
opt_free(opt);
help();
return 0;
}
if(opt_check(opt, "recursive"))
recursion = 1;
if(opt_check(opt, "infected"))
printinfected = 1;
if(opt_check(opt, "bell"))
bell = 1;
if(opt_check(opt, "tempdir"))
cl_settempdir(opt_arg(opt, "tempdir"), 0);
if(opt_check(opt, "leave-temps"))
cl_settempdir(NULL, 1);
/* initialize logger */
if(opt_check(opt, "log")) {
logg_file = opt_arg(opt, "log");
if(logg("#\n-------------------------------------------------------------------------------\n\n")) {
mprintf("!Problem with internal logger.\n");
opt_free(opt);
return 62;
}
} else
logg_file = NULL;
/* validate some numerical options */
if(opt_check(opt, "max-space")) {
pt = opt_arg(opt, "max-space");
if(!strchr(pt, 'M') && !strchr(pt, 'm')) {
if(!isnumb(pt)) {
logg("!--max-space requires a natural number\n");
opt_free(opt);
return 40;
}
}
}
if(opt_check(opt, "max-files")) {
if(!isnumb(opt_arg(opt, "max-files"))) {
logg("!--max-files requires a natural number\n");
opt_free(opt);
return 40;
}
}
if(opt_check(opt, "max-recursion")) {
if(!isnumb(opt_arg(opt, "max-recursion"))) {
logg("!--max-recursion requires a natural number\n");
opt_free(opt);
return 40;
}
}
if(opt_check(opt, "max-mail-recursion")) {
if(!isnumb(opt_arg(opt, "max-mail-recursion"))) {
logg("!--max-mail-recursion requires a natural number\n");
opt_free(opt);
return 40;
}
}
if(opt_check(opt, "max-dir-recursion")) {
if(!isnumb(opt_arg(opt, "max-dir-recursion"))) {
logg("!--max-dir-recursion requires a natural number\n");
opt_free(opt);
return 40;
}
}
if(opt_check(opt, "max-ratio")) {
if(!isnumb(opt_arg(opt, "max-ratio"))) {
logg("!--max-ratio requires a natural number\n");
opt_free(opt);
return 40;
}
}
memset(&info, 0, sizeof(struct s_info));
#ifdef C_WINDOWS
_set_fmode(_O_BINARY);
#ifdef CL_DEBUG
{
_CrtSetReportMode(_CRT_ERROR, _CRTDBG_MODE_FILE);
_CrtSetReportFile(_CRT_ERROR, _CRTDBG_FILE_STDERR);
}
#endif
gettimeofday(&t1, NULL);
#else
gettimeofday(&t1, &tz);
#endif
ret = scanmanager(opt);
if(!opt_check(opt, "disable-summary") && !opt_check(opt, "no-summary")) {
#ifdef C_WINDOWS
gettimeofday(&t2, NULL);
#else
gettimeofday(&t2, &tz);
#endif
ds = t2.tv_sec - t1.tv_sec;
dms = t2.tv_usec - t1.tv_usec;
ds -= (dms < 0) ? (1):(0);
dms += (dms < 0) ? (1000000):(0);
logg("\n----------- SCAN SUMMARY -----------\n");
logg("Known viruses: %u\n", info.sigs);
if(opt_check(opt, "ncore"))
logg("Engine version: %s [ncore]\n", cl_retver());
else
logg("Engine version: %s\n", cl_retver());
logg("Scanned directories: %u\n", info.dirs);
logg("Scanned files: %u\n", info.files);
logg("Infected files: %u\n", info.ifiles);
if(info.notremoved) {
logg("Not removed: %u\n", info.notremoved);
}
if(info.notmoved) {
logg("Not %s: %u\n", opt_check(opt, "copy") ? "moved" : "copied", info.notmoved);
}
mb = info.blocks * (CL_COUNT_PRECISION / 1024) / 1024.0;
logg("Data scanned: %2.2lf MB\n", mb);
logg("Time: %u.%3.3u sec (%u m %u s)\n", ds, dms/1000, ds/60, ds%60);
}
opt_free(opt);
#if defined(C_WINDOWS) && defined(CL_THREAD_SAFE)
if(!pthread_win32_process_detach_np()) {
logg("!Can't stop the win32 pthreads layer\n");
return 72;
}
#endif
return ret;
}
void help(void)
{
mprintf_stdout = 1;
mprintf("\n");
mprintf(" Clam AntiVirus Scanner "VERSION"\n");
mprintf(" (C) 2002 - 2007 ClamAV Team - http://www.clamav.net/team\n\n");
mprintf(" --help -h Print this help screen\n");
mprintf(" --version -V Print version number\n");
mprintf(" --verbose -v Be verbose\n");
mprintf(" --debug Enable libclamav's debug messages\n");
mprintf(" --quiet Only output error messages\n");
mprintf(" --stdout Write to stdout instead of stderr\n");
mprintf(" --no-summary Disable summary at end of scanning\n");
mprintf(" --infected -i Only print infected files\n");
mprintf(" --bell Sound bell on virus detection\n");
mprintf("\n");
mprintf(" --tempdir=DIRECTORY Create temporary files in DIRECTORY\n");
mprintf(" --leave-temps Do not remove temporary files\n");
mprintf(" --database=FILE/DIR -d FILE/DIR Load virus database from FILE or load\n");
mprintf(" all .cvd and .db[2] files from DIR\n");
mprintf(" --log=FILE -l FILE Save scan report to FILE\n");
mprintf(" --recursive -r Scan subdirectories recursively\n");
mprintf(" --remove Remove infected files. Be careful!\n");
mprintf(" --move=DIRECTORY Move infected files into DIRECTORY\n");
mprintf(" --copy=DIRECTORY Copy infected files into DIRECTORY\n");
#ifdef HAVE_REGEX_H
mprintf(" --exclude=REGEX Don't scan file names matching REGEX\n");
mprintf(" --exclude-dir=REGEX Don't scan directories matching REGEX\n");
mprintf(" --include=REGEX Only scan file names matching REGEX\n");
mprintf(" --include-dir=REGEX Only scan directories matching REGEX\n");
#else
mprintf(" --exclude=PATT Don't scan file names containing PATT\n");
mprintf(" --exclude-dir=PATT Don't scan directories containing PATT\n");
mprintf(" --include=PATT Only scan file names containing PATT\n");
mprintf(" --include-dir=PATT Only scan directories containing PATT\n");
#endif
#ifdef HAVE_NCORE
mprintf("\n --ncore Use hardware acceleration\n");
#endif
mprintf("\n");
mprintf(" --no-mail Disable mail file support\n");
mprintf(" --no-phishing-sigs Disable signature-based phishing detection\n");
mprintf(" --no-phishing-scan-urls Disable url-based phishing detection\n");
mprintf(" --no-phishing-restrictedscan Enable phishing detection for all domains (might lead to false positives!)\n");
mprintf(" --phishing-ssl Always block SSL mismatches in URLs (phishing module)\n");
mprintf(" --phishing-cloak Always block cloaked URLs (phishing module)\n");
mprintf(" --no-algorithmic Disable algorithmic detection\n");
mprintf(" --no-pe Disable PE analysis\n");
mprintf(" --no-elf Disable ELF support\n");
mprintf(" --no-ole2 Disable OLE2 support\n");
mprintf(" --no-pdf Disable PDF support\n");
mprintf(" --no-html Disable HTML support\n");
mprintf(" --no-archive Disable libclamav archive support\n");
mprintf(" --detect-broken Try to detect broken executable files\n");
mprintf(" --block-encrypted Block encrypted archives\n");
mprintf(" --block-max Block archives that exceed limits\n");
mprintf(" --mail-follow-urls Download and scan URLs\n");
mprintf("\n");
mprintf(" --max-space=#n Only extract first #n kilobytes from\n");
mprintf(" archived files\n");
mprintf(" --max-files=#n Only extract first #n files from\n");
mprintf(" archives\n");
mprintf(" --max-ratio=#n Maximum compression ratio limit\n");
mprintf(" --max-recursion=#n Maximum archive recursion level\n");
mprintf(" --max-dir-recursion=#n Maximum directory recursion level\n");
mprintf(" --max-mail-recursion=#n Maximum mail recursion level\n");
mprintf(" --unzip[=FULLPATH] Enable support for .zip files\n");
mprintf(" --unrar[=FULLPATH] Enable support for .rar files\n");
mprintf(" --arj[=FULLPATH] Enable support for .arj files\n");
mprintf(" --unzoo[=FULLPATH] Enable support for .zoo files\n");
mprintf(" --lha[=FULLPATH] Enable support for .lha files\n");
mprintf(" --jar[=FULLPATH] Enable support for .jar files\n");
mprintf(" --tar[=FULLPATH] Enable support for .tar files\n");
mprintf(" --deb[=FULLPATH to ar] Enable support for .deb files\n");
mprintf(" --tgz[=FULLPATH] Enable support for .tar.gz, .tgz files\n\n");
}