mirror of https://github.com/Cisco-Talos/clamav
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
407 lines
9.1 KiB
407 lines
9.1 KiB
/*
|
|
* Copyright (C) 2001-2002 Tomasz Kojm <zolw@konarski.edu.pl>
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* 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., 675 Mass Ave, Cambridge, MA 02139, USA.
|
|
*
|
|
* Sat Sep 14 22:18:20 CEST 2002: included getfirst*(), getnext*() functions
|
|
* from Alejandro Dubrovsky <s328940@student.uq.edu.au>
|
|
*/
|
|
|
|
#if HAVE_CONFIG_H
|
|
#include "clamav-config.h"
|
|
#endif
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <clamav.h>
|
|
#define _GNU_SOURCE
|
|
#include "getopt.h"
|
|
|
|
|
|
#include "options.h"
|
|
#include "shared.h"
|
|
#include "memory.h"
|
|
#include "output.h"
|
|
|
|
extern int clamscan(struct optstruct *opt);
|
|
|
|
static char *clamdscan_long[] = { "help", "version", "verbose", "quiet",
|
|
"stdout", "log", "config-file", "no-summary",
|
|
"disable-summary", NULL };
|
|
|
|
static char clamdscan_short[] = { 'h', 'V', 'v', 'l', 0 };
|
|
|
|
int clamdscan_mode = 0;
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
int ret, opt_index, i, len;
|
|
struct optstruct *opt;
|
|
|
|
const char *getopt_parameters = "hvd:wriVl:m";
|
|
|
|
static struct option long_options[] = {
|
|
/*
|
|
* WARNING: For compatibility reasons options marked as "not used"
|
|
* must still be accepted !
|
|
*/
|
|
{"help", 0, 0, 'h'}, /* clamscan + clamdscan */
|
|
{"quiet", 0, 0, 0}, /* clamscan + clamdscan */
|
|
{"stdout", 0, 0, 0}, /* clamscan + clamdscan */
|
|
{"verbose", 0, 0, 'v'}, /* clamscan + clamdscan */
|
|
{"debug", 0, 0, 0},
|
|
{"version", 0, 0, 'V'}, /* clamscan + clamdscan */
|
|
{"tempdir", 1, 0, 0},
|
|
{"leave-temps", 0, 0, 0},
|
|
{"config-file", 1, 0, 0}, /* clamdscan */
|
|
{"database", 1, 0, 'd'},
|
|
{"whole-file", 0, 0, 'w'}, /* not used */
|
|
{"force", 0, 0, 0},
|
|
{"recursive", 0, 0, 'r'},
|
|
{"bell", 0, 0, 0},
|
|
{"disable-summary", 0, 0, 0}, /* obsolete */
|
|
{"no-summary", 0, 0, 0},
|
|
{"infected", 0, 0, 'i'},
|
|
{"log", 1, 0, 'l'},
|
|
{"log-verbose", 0, 0, 0}, /* not used */
|
|
{"threads", 1, 0, 0}, /* not used */
|
|
{"one-virus", 0, 0, 0}, /* not used */
|
|
{"move", 1, 0, 0},
|
|
{"remove", 0, 0, 0},
|
|
{"exclude", 1, 0, 0},
|
|
{"include", 1, 0, 0},
|
|
{"max-files", 1, 0, 0},
|
|
{"max-space", 1, 0, 0},
|
|
{"max-ratio", 1, 0, 0},
|
|
{"max-recursion", 1, 0, 0},
|
|
{"disable-archive", 0, 0, 0},
|
|
{"no-archive", 0, 0, 0},
|
|
{"detect-broken", 0, 0, 0},
|
|
{"block-encrypted", 0, 0, 0},
|
|
{"block-max", 0, 0, 0},
|
|
{"no-pe", 0, 0, 0},
|
|
{"no-ole2", 0, 0, 0},
|
|
{"no-html", 0, 0, 0},
|
|
{"mbox", 0, 0, 'm'}, /* not used */
|
|
{"no-mail", 0, 0, 0},
|
|
{"mail-follow-urls", 0, 0, 0},
|
|
{"unzip", 2, 0, 0},
|
|
{"unrar", 2, 0, 0},
|
|
{"unace", 2, 0, 0}, /* not used */
|
|
{"unarj", 2, 0, 0}, /* not used */
|
|
{"arj", 2, 0, 0},
|
|
{"zoo", 2, 0, 0}, /* not used */
|
|
{"unzoo", 2, 0, 0},
|
|
{"lha", 2, 0, 0},
|
|
{"jar", 2, 0, 0},
|
|
{"tar", 2, 0, 0},
|
|
{"tgz", 2, 0, 0},
|
|
{"deb", 2, 0, 0},
|
|
{0, 0, 0, 0}
|
|
};
|
|
|
|
|
|
opt=(struct optstruct*) mcalloc(1, sizeof(struct optstruct));
|
|
opt->optlist = NULL;
|
|
|
|
if(strstr(argv[0], "clamdscan"))
|
|
clamdscan_mode = 1;
|
|
|
|
while(1) {
|
|
|
|
opt_index=0;
|
|
ret=getopt_long(argc, argv, getopt_parameters, long_options, &opt_index);
|
|
|
|
if (ret == -1)
|
|
break;
|
|
|
|
switch (ret) {
|
|
case 0:
|
|
register_long_option(opt, long_options[opt_index].name);
|
|
break;
|
|
|
|
default:
|
|
if(strchr(getopt_parameters, ret)) {
|
|
if(opt_index)
|
|
register_char_option(opt, ret, long_options[opt_index].name);
|
|
else
|
|
register_char_option(opt, ret, NULL);
|
|
|
|
} else {
|
|
mprintf("!Unknown option passed.\n");
|
|
free_opt(opt);
|
|
exit(40);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (optind < argc) {
|
|
|
|
len=0;
|
|
|
|
/* count length of non-option arguments */
|
|
|
|
for(i=optind; i<argc; i++)
|
|
len+=strlen(argv[i]);
|
|
|
|
len=len+argc-optind-1; /* add spaces between arguments */
|
|
opt->filename=(char*)mcalloc(len + 256, sizeof(char));
|
|
|
|
for(i=optind; i<argc; i++) {
|
|
strncat(opt->filename, argv[i], strlen(argv[i]));
|
|
if(i != argc-1)
|
|
strncat(opt->filename, "\t", 1);
|
|
}
|
|
|
|
}
|
|
ret = clamscan(opt);
|
|
|
|
free_opt(opt);
|
|
|
|
return ret;
|
|
}
|
|
|
|
void register_char_option(struct optstruct *opt, char ch, const char *longname)
|
|
{
|
|
struct optnode *newnode;
|
|
int i, found = 0;
|
|
|
|
|
|
if(clamdscan_mode) {
|
|
for(i = 0; clamdscan_short[i]; i++)
|
|
if(clamdscan_short[i] == ch)
|
|
found = 1;
|
|
|
|
if(!found) {
|
|
if(longname)
|
|
mprintf("WARNING: Ignoring option -%c (--%s): please edit clamav.conf instead.\n", ch, longname);
|
|
else
|
|
mprintf("WARNING: Ignoring option -%c: please edit clamav.conf instead.\n", ch);
|
|
|
|
return;
|
|
}
|
|
}
|
|
|
|
newnode = (struct optnode *) mmalloc(sizeof(struct optnode));
|
|
newnode->optchar = ch;
|
|
if(optarg != NULL) {
|
|
newnode->optarg = (char *) mcalloc(strlen(optarg) + 1, sizeof(char));
|
|
strcpy(newnode->optarg, optarg);
|
|
} else newnode->optarg = NULL;
|
|
|
|
newnode->optname = NULL;
|
|
newnode->next = opt->optlist;
|
|
opt->optlist = newnode;
|
|
}
|
|
|
|
void register_long_option(struct optstruct *opt, const char *optname)
|
|
{
|
|
struct optnode *newnode;
|
|
int i, found = 0;
|
|
|
|
|
|
if(clamdscan_mode) {
|
|
for(i = 0; clamdscan_long[i]; i++)
|
|
if(!strcmp(clamdscan_long[i], optname))
|
|
found = 1;
|
|
|
|
if(!found) {
|
|
mprintf("WARNING: Ignoring option --%s: please edit clamav.conf instead.\n", optname);
|
|
return;
|
|
}
|
|
}
|
|
|
|
newnode = (struct optnode *) mmalloc(sizeof(struct optnode));
|
|
newnode->optchar = 0;
|
|
if(optarg != NULL) {
|
|
newnode->optarg = (char *) mcalloc(strlen(optarg) + 1, sizeof(char));
|
|
strcpy(newnode->optarg, optarg);
|
|
} else newnode->optarg = NULL;
|
|
|
|
newnode->optname = (char *) mcalloc(strlen(optname) + 1, sizeof(char));
|
|
strcpy(newnode->optname, optname);
|
|
newnode->next = opt->optlist;
|
|
opt->optlist = newnode;
|
|
}
|
|
|
|
int optc(const struct optstruct *opt, char ch)
|
|
{
|
|
struct optnode *handler;
|
|
|
|
handler = opt->optlist;
|
|
|
|
while(1) {
|
|
if(handler) {
|
|
if(handler->optchar == ch) return 1;
|
|
} else break;
|
|
handler = handler->next;
|
|
}
|
|
|
|
return(0);
|
|
}
|
|
|
|
int optl(const struct optstruct *opt, const char *optname)
|
|
{
|
|
struct optnode *handler;
|
|
|
|
handler = opt->optlist;
|
|
|
|
while(1) {
|
|
if(handler) {
|
|
if(handler->optname)
|
|
if(!strcmp(handler->optname, optname)) return 1;
|
|
} else break;
|
|
handler = handler->next;
|
|
}
|
|
|
|
return(0);
|
|
}
|
|
|
|
char *getargc(const struct optstruct *opt, char ch)
|
|
{
|
|
struct optnode *handler;
|
|
|
|
handler = opt->optlist;
|
|
|
|
while(1) {
|
|
if(handler) {
|
|
if(handler->optchar == ch) return handler->optarg;
|
|
} else break;
|
|
handler = handler->next;
|
|
}
|
|
|
|
return(NULL);
|
|
}
|
|
|
|
char *getfirstargc(const struct optstruct *opt, char ch, struct optnode **optnode)
|
|
{
|
|
struct optnode *handler;
|
|
|
|
handler = opt->optlist;
|
|
|
|
while(1) {
|
|
if(handler) {
|
|
if(handler->optchar == ch) {
|
|
*optnode = handler;
|
|
return handler->optarg;
|
|
}
|
|
} else break;
|
|
handler = handler->next;
|
|
}
|
|
*optnode = NULL;
|
|
return(NULL);
|
|
}
|
|
|
|
char *getnextargc(struct optnode **optnode, char ch)
|
|
{
|
|
struct optnode *handler;
|
|
|
|
handler = (*optnode)->next;
|
|
|
|
while(1) {
|
|
if(handler) {
|
|
if(handler->optchar == ch) {
|
|
*optnode = handler;
|
|
return handler->optarg;
|
|
}
|
|
} else break;
|
|
handler = handler->next;
|
|
}
|
|
*optnode = NULL;
|
|
return(NULL);
|
|
}
|
|
|
|
char *getargl(const struct optstruct *opt, const char *optname)
|
|
{
|
|
struct optnode *handler;
|
|
|
|
handler = opt->optlist;
|
|
|
|
while(1) {
|
|
if(handler) {
|
|
if(handler->optname)
|
|
if(!strcmp(handler->optname, optname)) return handler->optarg;
|
|
} else break;
|
|
handler = handler->next;
|
|
}
|
|
|
|
return(NULL);
|
|
}
|
|
|
|
char *getfirstargl(const struct optstruct *opt, const char *optname, struct optnode **optnode)
|
|
{
|
|
struct optnode *handler;
|
|
|
|
handler = opt->optlist;
|
|
|
|
while(1) {
|
|
if(handler) {
|
|
if(handler->optname)
|
|
if(!strcmp(handler->optname, optname)) {
|
|
*optnode = handler;
|
|
return handler->optarg;
|
|
}
|
|
} else break;
|
|
handler = handler->next;
|
|
}
|
|
|
|
*optnode = NULL;
|
|
return(NULL);
|
|
}
|
|
|
|
char *getnextargl(struct optnode **optnode, const char *optname)
|
|
{
|
|
struct optnode *handler;
|
|
|
|
handler = (*optnode)->next;
|
|
|
|
while(1) {
|
|
if(handler) {
|
|
if(handler->optname)
|
|
if(!strcmp(handler->optname, optname)) {
|
|
*optnode = handler;
|
|
return handler->optarg;
|
|
}
|
|
} else break;
|
|
handler = handler->next;
|
|
}
|
|
|
|
*optnode = NULL;
|
|
return(NULL);
|
|
}
|
|
|
|
void free_opt(struct optstruct *opt)
|
|
{
|
|
struct optnode *handler, *prev;
|
|
|
|
if(!opt)
|
|
return;
|
|
|
|
handler = opt->optlist;
|
|
|
|
while(handler != NULL) {
|
|
handler->optchar = 0;
|
|
if(handler->optarg) free(handler->optarg);
|
|
if(handler->optname) free(handler->optname);
|
|
prev = handler;
|
|
handler = handler->next;
|
|
free(prev);
|
|
}
|
|
|
|
if (opt->filename)
|
|
free(opt->filename);
|
|
free(opt);
|
|
}
|
|
|