/*

  *  Copyright (C) 2001-2003 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.
  */
 
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
 #include <clamav.h>
 #define _GNU_SOURCE
 #include "getopt.h"
 

 #if defined(C_LINUX) && defined(CL_DEBUG)
 #include <sys/resource.h>
 #endif
 

 #include "options.h"
 #include "others.h"
 #include "shared.h"
 
 void sigtool(struct optstruct *opt);
 
 int main(int argc, char **argv)
 {
 	int ret, opt_index, i, len;
 	struct optstruct *opt;
 

 	const char *getopt_parameters = "hvVc:s:f:b:i:u:";

 
 	static struct option long_options[] = {
 	    {"help", 0, 0, 'h'},
 	    {"quiet", 0, 0, 0},

 	    {"debug", 0, 0, 0},

 	    {"verbose", 0, 0, 'v'},
 	    {"stdout", 0, 0, 0},
 	    {"version", 0, 0, 'V'},
 	    {"tempdir", 1, 0, 0},
 	    {"hex-dump", 0, 0, 0},
 	    {"command", 1, 0, 'c'},
 	    {"string", 1, 0, 's'},
 	    {"file", 1, 0, 'f'},

 	    {"build", 1, 0, 'b'},

 	    {"server", 1, 0, 0},

 	    {"unpack", 1, 0, 'u'},

 	    {"unpack-current", 1, 0, 0},

 	    {"info", 1, 0, 'i'},

 	    {0, 0, 0, 0}
     	};
 

 #if defined(C_LINUX) && defined(CL_DEBUG)
 	/* 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

 
     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));
 
 
     sigtool(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 *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);
 }
 
 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");
 }