e3aaff8e |
{"database", 1, 0, 'd'},
{"whole-file", 0, 0, 'w'}, /* not used */
{"force", 0, 0, 0},
{"recursive", 0, 0, 'r'},
{"disable-summary", 0, 0, 0},
{"infected", 0, 0, 'i'},
{"log", 1, 0, 'l'},
{"log-verbose", 0, 0, 0},
{"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-recursion", 1, 0, 0},
{"disable-archive", 0, 0, 0},
{"mbox", 0, 0, 'm'},
{"stdout", 0, 0, 0},
{"unzip", 2, 0, 0},
{"unrar", 2, 0, 0},
{"unace", 2, 0, 0},
{"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*) mmalloc(sizeof(struct optstruct));
opt->optlist = NULL;
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))
register_char_option(opt, ret);
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, " ", 1);
}
} else
/* FIXME !!! Without this, we have segfault */
opt->filename=(char*)mcalloc(1, sizeof(char));
clamscan(opt);
return(0);
}
void register_char_option(struct optstruct *opt, char ch)
{
struct optnode *newnode;
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;
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 || !opt->optlist)
return;
mprintf("*Freeing option list... ");
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);
}
free(opt->filename);
free(opt);
mprintf("*done\n");
} |