/*

  *  Copyright (C) 2002 - 2004 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 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.
  */
 
 #ifndef __CLAMAV_H
 #define __CLAMAV_H
 
 #include <sys/types.h>
 #include <sys/stat.h>
 #include <unistd.h>
  
 #ifdef __cplusplus
 extern "C"
 {
 #endif
  
 
 #define CL_NUM_CHILDS 256
 #define CL_MIN_LENGTH 2
 
 #define CL_COUNT_PRECISION 4096
 
 
 /* return codes */
 #define CL_CLEAN	0   /* virus not found */
 #define CL_VIRUS	1   /* virus found */

 #define CL_EMAXREC	10  /* recursion level limit exceeded */
 #define CL_EMAXSIZE	11  /* size limit exceeded */
 #define CL_EMAXFILES	12  /* files limit exceeded */
 #define CL_ERAR		100 /* rar handler error */
 #define CL_EZIP		101 /* zip handler error */
 #define	CL_EMALFZIP	102 /* malformed zip */
 #define CL_EGZIP	103 /* gzip handler error */
 #define CL_EBZIP	104 /* bzip2 handler error */

 #define CL_EOLE2	105 /* OLE2 handler error */

 #define CL_EMSCOMP	106 /* compress.exe handler error */

 #define CL_EMSCAB	107 /* MS CAB module error */

 #define CL_EACCES	200 /* access denied */
 #define CL_ENULLARG	300 /* null argument error */
 

 #define CL_ETMPFILE	-1  /* tmpfile() failed */
 #define CL_EFSYNC	-2  /* fsync() failed */
 #define CL_EMEM		-3  /* memory allocation error */
 #define CL_EOPEN	-4  /* file open error */
 #define CL_EMALFDB	-5  /* malformed database */
 #define CL_EPATSHORT	-6  /* pattern too short */
 #define CL_ETMPDIR	-7  /* mkdir() failed */
 #define CL_ECVD		-8  /* not a CVD file (or broken) */
 #define CL_ECVDEXTR	-9  /* CVD extraction failure */
 #define CL_EMD5		-10 /* MD5 verification error */
 #define CL_EDSIG	-11 /* digital signature verification error */

 #define CL_EIO		-12 /* general I/O error */

 
 /* options */

 #define CL_RAW		0
 #define CL_ARCHIVE	1
 #define CL_MAIL		2
 #define CL_DISABLERAR	4
 #define CL_OLE2		8

 #define CL_ENCRYPTED    16

 #define CL_HTML		32

 #define CL_PE		64

 struct cli_patt {

     short int *pattern;
     unsigned int length;
     char *virname;

     unsigned short int sigid, parts, partno, type;

     struct cli_patt *next;

 };
 
 struct cl_node {
     char islast;

     struct cli_patt *list;

     struct cl_node *trans[CL_NUM_CHILDS], *fail;
 

     /* FIXME: these variables are only used in a root node */

     unsigned int maxpatlen, partsigs;
     unsigned int nodes;
     struct cl_node **nodetable;
 };
 
 struct cl_limits {

     int maxreclevel; /* maximal recursion level */
     int maxfiles; /* maximal number of files to be
 		   * scanned within an archive
 		   */
     int maxratio; /* maximal compression ratio */
     short archivememlim; /* limit memory usage for bzip2 (0/1) */
     long int maxfilesize; /* files in an archive larger than
 			   * this value will not be scanned
 			   */

 };
 
 struct cl_stat {
     char *dir;
     int no;
     struct stat *stattab;
 };
 

 struct cl_cvd {
     char *time;	    /* 2 */
     int version;    /* 3 */
     int sigs;	    /* 4 */
     short int fl;   /* 5 */
     char *md5;	    /* 6 */
     char *dsig;	    /* 7 */
     char *builder;  /* 8 */
 };
 

 /* file scanning */

 extern int cl_scanbuff(const char *buffer, unsigned int length, const char **virname, const struct cl_node *root);

 extern int cl_scandesc(int desc, const char **virname, unsigned long int *scanned, const struct cl_node *root, const struct cl_limits *limits, int options);

 extern int cl_scanfile(const char *filename, const char **virname, unsigned long int *scanned, const struct cl_node *root, const struct cl_limits *limits, int options);

 
 /* database loading */
 extern int cl_loaddb(const char *filename, struct cl_node **root, int *virnum);
 extern int cl_loaddbdir(const char *dirname, struct cl_node **root, int *virnum);

 extern const char *cl_retdbdir(void);

 extern int cl_retflevel(void);

 extern struct cl_cvd *cl_cvdhead(const char *file);

 extern struct cl_cvd *cl_cvdparse(const char *head);
 extern int cl_cvdverify(const char *file);

 extern void cl_cvdfree(struct cl_cvd *cvd);
 

 /* data dir stat functions */

 extern int cl_statinidir(const char *dirname, struct cl_stat *dbstat);
 extern int cl_statchkdir(const struct cl_stat *dbstat);
 extern int cl_statfree(struct cl_stat *dbstat);
 

 /* enable debug information */
 extern void cl_debug(void);

 extern void cl_settempdir(const char *dir, short leavetemps);
 

 /* build a trie */

 extern int cl_buildtrie(struct cl_node *root);

 
 extern void cl_freetrie(struct cl_node *root);
 

 extern const char *cl_strerror(int clerror);
 extern const char *cl_perror(int clerror); /* deprecated */

 
 extern char *cl_md5buff(const char *buffer, unsigned int length);
 
 extern int cl_mbox(const char *dir, int desc);
 

 /* compute MD5 message digest from file (compatible with md5sum(1)) */

 extern char *cl_md5file(const char *filename);
 
 /* decode hexadecimal string */
 extern short int *cl_hex2str(const char *hex);
 
 /* encode a buffer 'string' length of 'len' to a hexadecimal string */
 extern char *cl_str2hex(const char *string, unsigned int len);
 

 /* generate a pseudo-random number */
 extern unsigned int cl_rndnum(unsigned int max);
 

 /* generate unique file name in temporary directory */
 char *cl_gentemp(const char *dir);
 

 #ifdef __cplusplus
 };
 #endif
  
 #endif