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.
127 lines
4.0 KiB
127 lines
4.0 KiB
13 years ago
|
|
||
|
--- python-clamav-0.4.1.orig/pyclamav.c
|
||
|
+++ python-clamav-0.4.1/pyclamav.c
|
||
|
@@ -33,25 +33,6 @@
|
||
|
|
||
|
/* ********************************************************* */
|
||
|
|
||
|
-/* To be able to compile with
|
||
|
- releases 0.75 of libclamav
|
||
|
-
|
||
|
- Where cl_free was cl_freetrie
|
||
|
- and cl_build was cl_buildtrie
|
||
|
- CL_SCAN_STDOPT did not exist
|
||
|
-*/
|
||
|
-#ifndef CL_SCAN_STDOPT
|
||
|
-#define CL_SCAN_STDOPT CL_RAW | CL_ARCHIVE | CL_MAIL | CL_DISABLERAR | CL_OLE2 | CL_ENCRYPTED
|
||
|
-void cl_free(struct cl_node *rootnode) {
|
||
|
- cl_freetrie(rootnode);
|
||
|
- return;
|
||
|
-}
|
||
|
-
|
||
|
-int cl_build(struct cl_node *rootnode) {
|
||
|
- return cl_buildtrie(rootnode);
|
||
|
-}
|
||
|
-#endif
|
||
|
-
|
||
|
/* For python prior to 2.3 */
|
||
|
#ifndef PyMODINIT_FUNC
|
||
|
#define PyMODINIT_FUNC void
|
||
|
@@ -68,11 +49,9 @@
|
||
|
unsigned int signumber = 0;
|
||
|
|
||
|
/* Structures for clamav */
|
||
|
-struct cl_node *root = NULL;
|
||
|
-struct cl_limits limits;
|
||
|
+struct cl_engine *engine = NULL;
|
||
|
struct cl_stat dbstat;
|
||
|
|
||
|
-
|
||
|
/*
|
||
|
* If the virus database has been changed, then
|
||
|
* free the current tree and reload the new one
|
||
|
@@ -85,22 +64,22 @@
|
||
|
/* If yes : reload DB */
|
||
|
if (cl_statchkdir(&dbstat) == 1)
|
||
|
{
|
||
|
- /* free the tree */
|
||
|
- cl_free(root);
|
||
|
+ /* free the engine */
|
||
|
+ cl_engine_free(engine);
|
||
|
signumber=0;
|
||
|
- root=NULL;
|
||
|
+ engine=NULL;
|
||
|
|
||
|
/* Load DB */
|
||
|
- if((ret = cl_load(cl_retdbdir(), &root, &signumber, CL_DB_STDOPT))) {
|
||
|
+ if((ret = cl_load(cl_retdbdir(), engine, &signumber, CL_DB_STDOPT)) != CL_SUCCESS) {
|
||
|
/* Raise exception with error message */
|
||
|
PyErr_SetString(PyclamavError, cl_strerror(ret));
|
||
|
return -2;
|
||
|
}
|
||
|
|
||
|
- /* build the final tree */
|
||
|
- if((ret = cl_build(root))) {
|
||
|
- /* free the partial tree */
|
||
|
- cl_free(root);
|
||
|
+ /* prepare the engine */
|
||
|
+ if((ret = cl_engine_compile(engine))) {
|
||
|
+ /* free the engine */
|
||
|
+ cl_engine_free(engine);
|
||
|
/* Raise exception with error message */
|
||
|
PyErr_SetString(PyclamavError, cl_strerror(ret));
|
||
|
return -2;
|
||
|
@@ -231,7 +210,7 @@
|
||
|
return NULL;
|
||
|
}
|
||
|
|
||
|
- ret = cl_scanfile(file_to_scan, &virname, &size, root, &limits, CL_SCAN_STDOPT);
|
||
|
+ ret = cl_scanfile(file_to_scan, &virname, &size, engine, CL_SCAN_STDOPT);
|
||
|
|
||
|
/* Test return code */
|
||
|
switch (ret) {
|
||
|
@@ -278,17 +257,21 @@
|
||
|
/* Set documentation string for the module */
|
||
|
PyDict_SetItemString(dict, "__doc__", PyString_FromString("pyclamav :\n\n This is a python binding to the C libclamav library\n (from the Clamav project - http://www.clamav.net).\n It can be used to easily allow a Python script to scan\n a file or a buffer against known viruses.\n\nAuthor : Alexandre Norman [norman@xael.org]\n\nFunctions :\n - scanfile(string filename) : Scan a file for virus.\n - get_numsig() : Return the number of known signatures.\n - get_version() : Return the version of Clamav.\n - version() : Return the version of pyclamav.\n"));
|
||
|
|
||
|
+ /* initialize libclamav */
|
||
|
+ cl_init(CL_INIT_DEFAULT);
|
||
|
+ engine = cl_engine_new();
|
||
|
|
||
|
- if((ret = cl_load(cl_retdbdir(), &root, &signumber, CL_DB_STDOPT))) {
|
||
|
+ if((ret = cl_load(cl_retdbdir(), engine, &signumber, CL_DB_STDOPT))) {
|
||
|
/* Raise exception with error message */
|
||
|
PyErr_SetString(PyclamavError, cl_strerror(ret));
|
||
|
+ cl_engine_free(engine);
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
- /* build the final tree */
|
||
|
- if((ret = cl_build(root))) {
|
||
|
- /* free the partial tree */
|
||
|
- cl_free(root);
|
||
|
+ /* prepare the engine */
|
||
|
+ if((ret = cl_engine_compile(engine)) != CL_SUCCESS) {
|
||
|
+ /* free up the engine resources */
|
||
|
+ cl_engine_free(engine);
|
||
|
/* Raise exception with error message */
|
||
|
PyErr_SetString(PyclamavError, cl_strerror(ret));
|
||
|
return;
|
||
|
@@ -299,15 +282,6 @@
|
||
|
memset(&dbstat, 0, sizeof(struct cl_stat));
|
||
|
cl_statinidir(cl_retdbdir(), &dbstat);
|
||
|
|
||
|
-
|
||
|
-
|
||
|
- /* set up archive limits */
|
||
|
- memset(&limits, 0, sizeof(struct cl_limits));
|
||
|
- limits.maxfiles = 1000; /* max files */
|
||
|
- limits.maxfilesize = 10 * 1048576; /* maximal archived file size == 10 Mb */
|
||
|
- limits.maxreclevel = 5; /* maximal recursion level */
|
||
|
- limits.archivememlim = 0; /* disable memory limit for bzip2 scanner */
|
||
|
-
|
||
|
return ;
|
||
|
}
|
||
|
|
||
|
|