e3aaff8e |
/* |
f893c0f3 |
* Copyright (C) 2002 - 2004 Tomasz Kojm <tkojm@clamav.net> |
e3aaff8e |
*
* 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 <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <dirent.h>
#ifdef CL_THREAD_SAFE
# include <pthread.h>
pthread_mutex_t cli_scanrar_mutex = PTHREAD_MUTEX_INITIALIZER;
int cli_scanrar_inuse = 0;
#endif
#include "clamav.h"
#include "others.h"
#include "matcher.h"
#include "unrarlib.h" |
47bbbc56 |
#include "ole2_extract.h"
#include "vba_extract.h" |
e3aaff8e |
#ifdef HAVE_ZLIB_H
#include <zlib.h>
#include <zzip.h>
#endif
#ifdef HAVE_BZLIB_H
#include <bzlib.h>
#endif
#define SCAN_ARCHIVE (options & CL_ARCHIVE)
#define SCAN_MAIL (options & CL_MAIL) |
47bbbc56 |
#define SCAN_OLE2 (options & CL_OLE2) |
8139fd99 |
#define DISABLE_RAR (options & CL_DISABLERAR) |
e3aaff8e |
|
ed012c00 |
#define MAGIC_BUFFER_SIZE 14 |
e3aaff8e |
#define RAR_MAGIC_STR "Rar!"
#define ZIP_MAGIC_STR "PK\003\004"
#define GZIP_MAGIC_STR "\037\213"
#define MAIL_MAGIC_STR "From " |
d4d14218 |
#define RAWMAIL_MAGIC_STR "Received: " |
049a18b9 |
#define MAILDIR_MAGIC_STR "Return-Path: " |
6b997684 |
#define DELIVERED_MAGIC_STR "Delivered-To: " |
e3aaff8e |
#define BZIP_MAGIC_STR "BZh" |
47bbbc56 |
#define OLE2_MAGIC_STR "\xD0\xCF\x11\xE0\xA1\xB1\x1A\xE1" |
e3aaff8e |
int cli_magic_scandesc(int desc, char **virname, long int *scanned, const struct cl_node *root, const struct cl_limits *limits, int options, int *reclev);
int cli_scandesc(int desc, char **virname, long int *scanned, const struct
cl_node *root)
{
char *buffer, *buff, *endbl, *pt;
int bytes, buffsize, length;
/* prepare the buffer */ |
f7148839 |
buffsize = root->maxpatlen + SCANBUFF; |
9e431a95 |
if(!(buffer = (char *) cli_calloc(buffsize, sizeof(char)))) {
cli_dbgmsg("cli_scandesc(): unable to malloc(%d)\n", buffsize); |
e3aaff8e |
return CL_EMEM; |
9e431a95 |
} |
e3aaff8e |
buff = buffer;
buff += root->maxpatlen; /* pointer to read data block */ |
f7148839 |
endbl = buff + SCANBUFF - root->maxpatlen; /* pointer to the last block |
e3aaff8e |
* length of root->maxpatlen
*/
pt= buff; |
f7148839 |
length = SCANBUFF; |
e3aaff8e |
|
f7148839 |
while((bytes = read(desc, buff, SCANBUFF)) > 0) { |
e3aaff8e |
if(scanned != NULL)
*scanned += bytes / CL_COUNT_PRECISION;
|
f7148839 |
if(bytes < SCANBUFF)
length -= SCANBUFF - bytes; |
e3aaff8e |
if(cl_scanbuff(pt, length, virname, root) == CL_VIRUS) {
free(buffer);
return CL_VIRUS;
}
|
f7148839 |
if(bytes == SCANBUFF) |
e3aaff8e |
memmove(buffer, endbl, root->maxpatlen);
pt = buffer;
length=buffsize;
}
free(buffer);
return CL_CLEAN;
}
#ifdef CL_THREAD_SAFE
void cli_unlock_mutex(void *mtx)
{
cli_dbgmsg("Pthread cancelled. Unlocking mutex.\n");
pthread_mutex_unlock(mtx);
}
#endif
int cli_scanrar(int desc, char **virname, long int *scanned, const struct cl_node *root, const struct cl_limits *limits, int options, int *reclev)
{ |
fc56deed |
FILE *tmp = NULL; |
e3aaff8e |
int files = 0, fd, ret = CL_CLEAN;
ArchiveList_struct *rarlist = NULL;
char *rar_data_ptr;
unsigned long rar_data_size;
cli_dbgmsg("Starting scanrar()\n");
#ifdef CL_THREAD_SAFE
pthread_cleanup_push(cli_unlock_mutex, &cli_scanrar_mutex);
pthread_mutex_lock(&cli_scanrar_mutex);
cli_scanrar_inuse = 1;
#endif
if(!urarlib_list(desc, (ArchiveList_struct *) &rarlist)) {
#ifdef CL_THREAD_SAFE
pthread_mutex_unlock(&cli_scanrar_mutex);
cli_scanrar_inuse = 0;
#endif
return CL_ERAR;
}
while(rarlist) {
if(limits) {
if(limits->maxfilesize && (rarlist->item.UnpSize > limits->maxfilesize)) {
cli_dbgmsg("RAR->%s: Size exceeded (%d, max: %d)\n", rarlist->item.Name, rarlist->item.UnpSize, limits->maxfilesize);
rarlist = rarlist->next;
files++;
ret = CL_EMAXSIZE;
continue;
}
if(limits->maxfiles && (files > limits->maxfiles)) {
cli_dbgmsg("RAR: Files limit reached (max: %d)\n", limits->maxfiles);
ret = CL_EMAXFILES;
break;
}
}
if((tmp = tmpfile()) == NULL) {
cli_dbgmsg("RAR -> Can't generate tmpfile().\n");
#ifdef CL_THREAD_SAFE
pthread_mutex_unlock(&cli_scanrar_mutex);
cli_scanrar_inuse = 0;
#endif
return CL_ETMPFILE;
}
fd = fileno(tmp);
if(urarlib_get(&rar_data_ptr, &rar_data_size, rarlist->item.Name, desc, "clam")) {
cli_dbgmsg("RAR -> Extracted: %s, size: %d\n", rarlist->item.Name, rar_data_size); |
cdb0ae9c |
if(fwrite(rar_data_ptr, rar_data_size, 1, tmp) != 1) { |
e3aaff8e |
cli_dbgmsg("RAR -> Can't write() file.\n"); |
cdb0ae9c |
fclose(tmp); |
e3aaff8e |
tmp = NULL;
ret = CL_ERAR; |
cdb0ae9c |
if(rar_data_ptr) { |
e3aaff8e |
free(rar_data_ptr); |
cdb0ae9c |
rar_data_ptr = NULL;
} |
27a3f44a |
break; |
e3aaff8e |
}
|
cdb0ae9c |
if(rar_data_ptr) { |
e3aaff8e |
free(rar_data_ptr); |
cdb0ae9c |
rar_data_ptr = NULL;
}
if(fflush(tmp) != 0) {
cli_dbgmsg("fflush() failed: %s\n", strerror(errno));
fclose(tmp); |
e3aaff8e |
urarlib_freelist(rarlist);
#ifdef CL_THREAD_SAFE
pthread_mutex_unlock(&cli_scanrar_mutex);
cli_scanrar_inuse = 0;
#endif
return CL_EFSYNC;
}
lseek(fd, 0, SEEK_SET);
if((ret = cli_magic_scandesc(fd, virname, scanned, root, limits, options, reclev)) == CL_VIRUS ) {
cli_dbgmsg("RAR -> Found %s virus.\n", *virname); |
cdb0ae9c |
fclose(tmp); |
e3aaff8e |
urarlib_freelist(rarlist);
#ifdef CL_THREAD_SAFE
pthread_mutex_unlock(&cli_scanrar_mutex);
cli_scanrar_inuse = 0;
#endif
return CL_VIRUS;
}
} else {
cli_dbgmsg("RAR -> Can't decompress file %s\n", rarlist->item.Name); |
cdb0ae9c |
fclose(tmp);
ret = CL_ERAR; /* WinRAR 3.0 ? */
break; |
e3aaff8e |
}
|
cdb0ae9c |
fclose(tmp); |
e3aaff8e |
tmp = NULL;
rarlist = rarlist->next;
files++;
}
urarlib_freelist(rarlist);
#ifdef CL_THREAD_SAFE
pthread_mutex_unlock(&cli_scanrar_mutex);
cli_scanrar_inuse = 0;
pthread_cleanup_pop(0);
#endif
return ret;
}
#ifdef HAVE_ZLIB_H
int cli_scanzip(int desc, char **virname, long int *scanned, const struct cl_node *root, const struct cl_limits *limits, int options, int *reclev)
{
ZZIP_DIR *zdir;
ZZIP_DIRENT zdirent;
ZZIP_FILE *zfp; |
fc56deed |
FILE *tmp = NULL; |
f7148839 |
char *buff; |
ee039e40 |
int fd, bytes, files = 0, ret = CL_CLEAN; |
fc56deed |
struct stat source; |
ee039e40 |
zzip_error_t err; |
e3aaff8e |
cli_dbgmsg("Starting scanzip()\n");
|
049a18b9 |
if((zdir = zzip_dir_fdopen(dup(desc), &err)) == NULL) {
cli_dbgmsg("Zip -> Not supported file format ?.\n");
cli_dbgmsg("zzip_dir_fdopen() return code: %d\n", err); |
77be7ea9 |
/* no return with CL_EZIP due to password protected zips */
return CL_CLEAN; |
e3aaff8e |
}
|
fc56deed |
fstat(desc, &source);
|
33f40ee5 |
if(!(buff = (char *) cli_malloc(FILEBUFF))) { |
9e431a95 |
cli_dbgmsg("cli_scanzip(): unable to malloc(%d)\n", FILEBUFF); |
33f40ee5 |
zzip_dir_close(zdir);
return CL_EMEM;
}
|
e3aaff8e |
while(zzip_dir_read(zdir, &zdirent)) { |
fc56deed |
if(!zdirent.d_name || !strlen(zdirent.d_name)) { /* Mimail fix */
cli_dbgmsg("strlen(zdirent.d_name) == %d\n", strlen(zdirent.d_name)); |
a0977bfe |
*virname = "Suspected.Zip"; |
fc56deed |
ret = CL_VIRUS;
break;
}
|
e3aaff8e |
cli_dbgmsg("Zip -> %s, compressed: %d, normal: %d.\n", zdirent.d_name, zdirent.d_csize, zdirent.st_size);
|
a6945b5d |
if(limits && limits->maxratio > 0 && source.st_size && (zdirent.st_size / source.st_size) >= limits->maxratio) { |
a0977bfe |
*virname = "Oversized.Zip"; |
fc56deed |
ret = CL_VIRUS;
break;
}
|
e3aaff8e |
if(!zdirent.st_size) { /* omit directories and null files */
files++;
continue;
}
/* work-around for problematic zips (zziplib crashes with them) */
if(zdirent.d_csize < 0 || zdirent.st_size < 0) {
files++;
cli_dbgmsg("Zip -> Malformed archive detected.\n"); |
fc56deed |
/* ret = CL_EMALFZIP; */
/* report it as a virus */ |
a0977bfe |
*virname = "Suspected.Zip"; |
fc56deed |
ret = CL_VIRUS; |
e3aaff8e |
break;
}
if(limits) {
if(limits->maxfilesize && (zdirent.st_size > limits->maxfilesize)) {
cli_dbgmsg("Zip -> %s: Size exceeded (%d, max: %d)\n", zdirent.d_name, zdirent.st_size, limits->maxfilesize);
files++;
ret = CL_EMAXSIZE; |
fc56deed |
continue; /* this is not a bug */ |
e3aaff8e |
}
if(limits->maxfiles && (files > limits->maxfiles)) {
cli_dbgmsg("Zip: Files limit reached (max: %d)\n", limits->maxfiles);
ret = CL_EMAXFILES;
break;
}
}
/* generate temporary file and get its descriptor */
if((tmp = tmpfile()) == NULL) {
cli_dbgmsg("Zip -> Can't generate tmpfile().\n"); |
fc56deed |
ret = CL_ETMPFILE;
break; |
e3aaff8e |
}
if((zfp = zzip_file_open(zdir, zdirent.d_name, 0)) == NULL) {
cli_dbgmsg("Zip -> %s: Can't open file.\n", zdirent.d_name);
ret = CL_EZIP; |
fc56deed |
break; |
e3aaff8e |
}
|
f7148839 |
while((bytes = zzip_file_read(zfp, buff, FILEBUFF)) > 0) { |
cdb0ae9c |
if(fwrite(buff, bytes, 1, tmp)*bytes != bytes) {
cli_dbgmsg("Zip -> Can't fwrite() file: %s\n", strerror(errno)); |
e3aaff8e |
zzip_file_close(zfp); |
33f40ee5 |
zzip_dir_close(zdir);
fclose(tmp); |
f7148839 |
free(buff); |
33f40ee5 |
return CL_EZIP; |
e3aaff8e |
}
}
zzip_file_close(zfp);
|
cdb0ae9c |
if(fflush(tmp) != 0) {
cli_errmsg("fflush() failed: %s\n", strerror(errno)); |
fc56deed |
ret = CL_EFSYNC;
break; |
e3aaff8e |
}
|
cdb0ae9c |
fd = fileno(tmp);
|
e3aaff8e |
lseek(fd, 0, SEEK_SET);
if((ret = cli_magic_scandesc(fd, virname, scanned, root, limits, options, reclev)) == CL_VIRUS ) {
cli_dbgmsg("Zip -> Found %s virus.\n", *virname);
ret = CL_VIRUS;
break;
} else if(ret == CL_EMALFZIP) {
/* |
fc56deed |
* The trick with detection of ZoD only works with higher (>= 5) |
e3aaff8e |
* recursion limit level.
*/
cli_dbgmsg("Zip -> Malformed Zip, scanning stopped.\n"); |
a0977bfe |
*virname = "Suspected.Zip"; |
e3aaff8e |
ret = CL_VIRUS;
break;
}
|
cdb0ae9c |
if (tmp) {
fclose(tmp);
tmp = NULL;
} |
e3aaff8e |
files++;
}
zzip_dir_close(zdir); |
cdb0ae9c |
if (tmp) {
fclose(tmp);
tmp = NULL;
} |
33f40ee5 |
free(buff); |
e3aaff8e |
return ret;
}
int cli_scangzip(int desc, char **virname, long int *scanned, const struct cl_node *root, const struct cl_limits *limits, int options, int *reclev)
{
int fd, bytes, ret = CL_CLEAN;
long int size = 0; |
f7148839 |
char *buff; |
fc56deed |
FILE *tmp = NULL; |
e3aaff8e |
gzFile gd;
|
fc56deed |
cli_dbgmsg("in cli_scangzip()\n");
|
e3aaff8e |
if((gd = gzdopen(dup(desc), "rb")) == NULL) {
cli_dbgmsg("Can't gzdopen() descriptor %d.\n", desc);
return CL_EGZIP;
}
if((tmp = tmpfile()) == NULL) {
cli_dbgmsg("Can't generate tmpfile().\n");
gzclose(gd);
return CL_ETMPFILE;
}
fd = fileno(tmp);
|
fc56deed |
if(!(buff = (char *) cli_malloc(FILEBUFF))) { |
9e431a95 |
cli_dbgmsg("cli_scangzip(): unable to malloc(%d)\n", FILEBUFF); |
fc56deed |
gzclose(gd); |
f7148839 |
return CL_EMEM; |
fc56deed |
} |
f7148839 |
|
fc56deed |
while((bytes = gzread(gd, buff, FILEBUFF)) > 0) { |
e3aaff8e |
size += bytes;
if(limits) |
fc56deed |
if(limits->maxfilesize && (size + FILEBUFF > limits->maxfilesize)) { |
e3aaff8e |
cli_dbgmsg("Gzip->desc(%d): Size exceeded (stopped at %d, max: %d)\n", desc, size, limits->maxfilesize);
ret = CL_EMAXSIZE;
break;
}
if(write(fd, buff, bytes) != bytes) {
cli_dbgmsg("Gzip -> Can't write() file.\n"); |
cdb0ae9c |
fclose(tmp); |
e3aaff8e |
gzclose(gd); |
f7148839 |
free(buff); |
e3aaff8e |
return CL_EGZIP;
}
}
|
f7148839 |
free(buff); |
e3aaff8e |
gzclose(gd);
if(fsync(fd) == -1) {
cli_dbgmsg("fsync() failed for descriptor %d\n", fd); |
cdb0ae9c |
fclose(tmp); |
e3aaff8e |
return CL_EFSYNC;
}
lseek(fd, 0, SEEK_SET);
if((ret = cli_magic_scandesc(fd, virname, scanned, root, limits, options, reclev)) == CL_VIRUS ) {
cli_dbgmsg("Gzip -> Found %s virus.\n", *virname); |
cdb0ae9c |
fclose(tmp); |
e3aaff8e |
return CL_VIRUS;
} |
cdb0ae9c |
fclose(tmp); |
e3aaff8e |
return ret;
}
#endif
#ifdef HAVE_BZLIB_H
#ifdef NOBZ2PREFIX
#define BZ2_bzReadOpen bzReadOpen
#define BZ2_bzReadClose bzReadClose
#define BZ2_bzRead bzRead
#endif
int cli_scanbzip(int desc, char **virname, long int *scanned, const struct cl_node *root, const struct cl_limits *limits, int options, int *reclev)
{
int fd, bytes, ret = CL_CLEAN, bzerror = 0;
short memlim = 0;
long int size = 0; |
f7148839 |
char *buff; |
fc56deed |
FILE *fs, *tmp = NULL; |
e3aaff8e |
BZFILE *bfd;
|
9e431a95 |
if((fs = fdopen(dup(desc), "rb")) == NULL) { |
e3aaff8e |
cli_errmsg("Can't fdopen() descriptor %d.\n", desc);
return CL_EBZIP;
}
if(limits)
if(limits->archivememlim)
memlim = 1;
|
68b96877 |
if((bfd = BZ2_bzReadOpen(&bzerror, fs, 0, memlim, NULL, 0)) == NULL) { |
e3aaff8e |
cli_dbgmsg("Can't initialize bzip2 library (descriptor %d).\n", desc); |
9e431a95 |
fclose(fs); |
e3aaff8e |
return CL_EBZIP;
}
if((tmp = tmpfile()) == NULL) {
cli_dbgmsg("Can't generate tmpfile().\n");
BZ2_bzReadClose(&bzerror, bfd); |
9e431a95 |
fclose(fs); |
e3aaff8e |
return CL_ETMPFILE;
}
fd = fileno(tmp);
|
9e431a95 |
if(!(buff = (char *) malloc(FILEBUFF))) {
cli_dbgmsg("cli_scanbzip(): unable to malloc(%d)\n", FILEBUFF);
fclose(tmp);
fclose(fs);
BZ2_bzReadClose(&bzerror, bfd); |
f7148839 |
return CL_EMEM; |
9e431a95 |
} |
f7148839 |
|
fc56deed |
while((bytes = BZ2_bzRead(&bzerror, bfd, buff, FILEBUFF)) > 0) { |
e3aaff8e |
size += bytes;
if(limits) |
fc56deed |
if(limits->maxfilesize && (size + FILEBUFF > limits->maxfilesize)) { |
e3aaff8e |
cli_dbgmsg("Bzip2->desc(%d): Size exceeded (stopped at %d, max: %d)\n", desc, size, limits->maxfilesize);
ret = CL_EMAXSIZE;
break;
}
if(write(fd, buff, bytes) != bytes) {
cli_dbgmsg("Bzip2 -> Can't write() file.\n");
BZ2_bzReadClose(&bzerror, bfd); |
cdb0ae9c |
fclose(tmp); |
f7148839 |
free(buff); |
9e431a95 |
fclose(fs); |
e3aaff8e |
return CL_EGZIP;
}
}
|
f7148839 |
free(buff); |
e3aaff8e |
BZ2_bzReadClose(&bzerror, bfd);
if(fsync(fd) == -1) {
cli_dbgmsg("fsync() failed for descriptor %d\n", fd); |
cdb0ae9c |
fclose(tmp); |
9e431a95 |
fclose(fs); |
e3aaff8e |
return CL_EFSYNC;
}
lseek(fd, 0, SEEK_SET);
if((ret = cli_magic_scandesc(fd, virname, scanned, root, limits, options, reclev)) == CL_VIRUS ) {
cli_dbgmsg("Bzip2 -> Found %s virus.\n", *virname);
} |
cdb0ae9c |
fclose(tmp); |
9e431a95 |
fclose(fs); |
e3aaff8e |
return ret;
}
#endif
|
47bbbc56 |
int cli_scanole2(int desc, char **virname, long int *scanned, const struct cl_node *root, const struct cl_limits *limits, int options, int *reclev)
{
const char *tmpdir;
char *dir, *fullname;
unsigned char *data;
int ret = CL_CLEAN, fd, i;
vba_project_t *vba_project;
cli_dbgmsg("in cli_scanole2()\n");
tmpdir = getenv("TMPDIR");
if(tmpdir == NULL)
#ifdef P_tmpdir
tmpdir = P_tmpdir;
#else
tmpdir = "/tmp";
#endif
/* generate the temporary directory */
dir = cl_gentemp(tmpdir);
if(mkdir(dir, 0700)) {
cli_errmsg("ScanOLE2 -> Can't create temporary directory %s\n", dir);
return CL_ETMPDIR;
}
if((ret = cli_ole2_extract(desc, dir))) {
cli_errmsg("ScanOLE2 -> %s\n", cl_strerror(ret));
cli_rmdirs(dir);
free(dir);
return ret;
}
if((vba_project = (vba_project_t *) vba56_dir_read(dir))) {
for(i = 0; i < vba_project->count; i++) {
fullname = (char *) malloc(strlen(vba_project->dir) + strlen(vba_project->name[i]) + 2);
sprintf(fullname, "%s/%s", vba_project->dir, vba_project->name[i]);
fd = open(fullname, O_RDONLY);
if(fd == -1) {
cli_errmsg("Scan->OLE2 -> Can't open file %s\n", fullname);
free(fullname);
ret = CL_EOPEN;
break;
}
free(fullname);
data = (unsigned char *) vba_decompress(fd, vba_project->offset[i]); |
9d0ee50b |
close(fd); |
47bbbc56 |
if(cl_scanbuff(data, strlen(data), virname, root) == CL_VIRUS) {
free(data);
ret = CL_VIRUS;
break;
}
free(data);
}
|
f893c0f3 |
for(i = 0; i < vba_project->count; i++)
free(vba_project->name[i]);
free(vba_project->name);
free(vba_project->dir);
free(vba_project->offset); |
9d0ee50b |
free(vba_project); |
47bbbc56 |
}
cli_rmdirs(dir);
free(dir);
return ret;
} |
e3aaff8e |
int cli_scandir(char *dirname, char **virname, long int *scanned, const struct cl_node *root, const struct cl_limits *limits, int options, int *reclev)
{
DIR *dd;
struct dirent *dent;
struct stat statbuf;
char *fname;
if((dd = opendir(dirname)) != NULL) {
while((dent = readdir(dd))) {
if(dent->d_ino) {
if(strcmp(dent->d_name, ".") && strcmp(dent->d_name, "..")) {
/* build the full name */
fname = cli_calloc(strlen(dirname) + strlen(dent->d_name) + 2, sizeof(char));
sprintf(fname, "%s/%s", dirname, dent->d_name);
/* stat the file */
if(lstat(fname, &statbuf) != -1) {
if(S_ISDIR(statbuf.st_mode) && !S_ISLNK(statbuf.st_mode)) |
f0d72b10 |
cli_scandir(fname, virname, scanned, root, limits, options, reclev); |
e3aaff8e |
else
if(S_ISREG(statbuf.st_mode))
if(cl_scanfile(fname, virname, scanned, root, limits, options) == CL_VIRUS) {
free(fname);
closedir(dd);
return CL_VIRUS;
}
}
free(fname);
}
}
}
} else {
cli_errmsg("ScanDir -> Can't open directory %s.\n", dirname);
return CL_EOPEN;
}
closedir(dd);
return 0;
}
int cli_scanmail(int desc, char **virname, long int *scanned, const struct cl_node *root, const struct cl_limits *limits, int options, int *reclev)
{
const char *tmpdir;
char *dir;
int ret;
|
5920e217 |
cli_dbgmsg("Starting cli_scanmail()\n");
|
e1e5b905 |
if(*reclev > 5) /* FIXME: a temporary workaround */ |
5920e217 |
return CL_CLEAN; |
e3aaff8e |
tmpdir = getenv("TMPDIR");
if(tmpdir == NULL)
#ifdef P_tmpdir
tmpdir = P_tmpdir;
#else
tmpdir = "/tmp";
#endif
/* generate the temporary directory */ |
8139fd99 |
dir = cl_gentemp(tmpdir); |
e3aaff8e |
if(mkdir(dir, 0700)) {
cli_errmsg("ScanMail -> Can't create temporary directory %s\n", dir);
return CL_ETMPDIR;
}
/*
* Extract the attachments into the temporary directory
*/
ret = cl_mbox(dir, desc);
/* FIXME: check mbox return code */
ret = cli_scandir(dir, virname, scanned, root, limits, options, reclev);
cli_rmdirs(dir);
free(dir);
return ret;
}
int cli_magic_scandesc(int desc, char **virname, long int *scanned, const struct cl_node *root, const struct cl_limits *limits, int options, int *reclev)
{ |
cdb0ae9c |
char magic[MAGIC_BUFFER_SIZE+1]; |
e3aaff8e |
int ret = CL_CLEAN; |
cdb0ae9c |
int bread = 0; |
e3aaff8e |
if(!root) {
cli_errmsg("root == NULL\n");
return -1;
}
|
cdb0ae9c |
if(SCAN_ARCHIVE || SCAN_MAIL) {
/* Need to examine file type */ |
e3aaff8e |
|
8adf0608 |
if(SCAN_ARCHIVE && limits && limits->maxreclevel) |
e3aaff8e |
if(*reclev > limits->maxreclevel)
return CL_EMAXREC;
(*reclev)++;
|
cdb0ae9c |
lseek(desc, 0, SEEK_SET);
bread = read(desc, magic, MAGIC_BUFFER_SIZE);
magic[MAGIC_BUFFER_SIZE] = '\0'; /* terminate magic string properly */ |
e3aaff8e |
lseek(desc, 0, SEEK_SET);
|
67940173 |
if (bread != MAGIC_BUFFER_SIZE) {
/* short read: No need to do magic */ |
8139fd99 |
(*reclev)--; |
67940173 |
return ret;
} |
e3aaff8e |
#ifdef CL_THREAD_SAFE
/* this check protects against recursive deadlock */ |
8139fd99 |
if(!DISABLE_RAR && SCAN_ARCHIVE && !cli_scanrar_inuse && !strncmp(magic, RAR_MAGIC_STR, strlen(RAR_MAGIC_STR))) { |
9e431a95 |
cli_dbgmsg("Recognized rar file.\n"); |
e3aaff8e |
ret = cli_scanrar(desc, virname, scanned, root, limits, options, reclev);
}
#else |
8139fd99 |
if(!DISABLE_RAR && SCAN_ARCHIVE && !strncmp(magic, RAR_MAGIC_STR, strlen(RAR_MAGIC_STR))) { |
9e431a95 |
cli_dbgmsg("Recognized rar file.\n"); |
e3aaff8e |
ret = cli_scanrar(desc, virname, scanned, root, limits, options, reclev);
}
#endif
#ifdef HAVE_ZLIB_H |
cdb0ae9c |
else if(SCAN_ARCHIVE && !strncmp(magic, ZIP_MAGIC_STR, strlen(ZIP_MAGIC_STR))) { |
9e431a95 |
cli_dbgmsg("Recognized zip file.\n"); |
e3aaff8e |
ret = cli_scanzip(desc, virname, scanned, root, limits, options, reclev); |
8139fd99 |
} else if(SCAN_ARCHIVE && !strncmp(magic, GZIP_MAGIC_STR, strlen(GZIP_MAGIC_STR))) { |
9e431a95 |
cli_dbgmsg("Recognized gzip file.\n"); |
e3aaff8e |
ret = cli_scangzip(desc, virname, scanned, root, limits, options, reclev);
}
#endif
#ifdef HAVE_BZLIB_H |
cdb0ae9c |
else if(SCAN_ARCHIVE && !strncmp(magic, BZIP_MAGIC_STR, strlen(BZIP_MAGIC_STR))) { |
9e431a95 |
cli_dbgmsg("Recognized bzip file.\n"); |
e3aaff8e |
ret = cli_scanbzip(desc, virname, scanned, root, limits, options, reclev);
}
#endif |
47bbbc56 |
else if(SCAN_OLE2 && !strncmp(magic, OLE2_MAGIC_STR, 8)) {
cli_dbgmsg("Recognized OLE2 file.\n");
ret = cli_scanole2(desc, virname, scanned, root, limits, options, reclev);
} |
0de3db22 |
else if(SCAN_MAIL && !strncmp(magic, MAIL_MAGIC_STR, strlen(MAIL_MAGIC_STR))) { |
9e431a95 |
cli_dbgmsg("Recognized mail file.\n"); |
e3aaff8e |
ret = cli_scanmail(desc, virname, scanned, root, limits, options, reclev);
} |
cdb0ae9c |
else if(SCAN_MAIL && !strncmp(magic, RAWMAIL_MAGIC_STR, strlen(RAWMAIL_MAGIC_STR))) { |
9e431a95 |
cli_dbgmsg("Recognized raw mail file.\n"); |
e3aaff8e |
ret = cli_scanmail(desc, virname, scanned, root, limits, options, reclev); |
ee039e40 |
} else if(SCAN_MAIL && !strncasecmp(magic, MAILDIR_MAGIC_STR, strlen(MAILDIR_MAGIC_STR))) { |
049a18b9 |
cli_dbgmsg("Recognized Maildir mail file.\n");
ret = cli_scanmail(desc, virname, scanned, root, limits, options, reclev); |
8139fd99 |
} else if(SCAN_MAIL && !strncmp(magic, DELIVERED_MAGIC_STR, strlen(DELIVERED_MAGIC_STR))) { |
6b997684 |
cli_dbgmsg("Recognized (Delivered-To) mail file.\n");
ret = cli_scanmail(desc, virname, scanned, root, limits, options, reclev); |
e3aaff8e |
} |
cdb0ae9c |
(*reclev)--; |
e3aaff8e |
}
|
46c2e927 |
if(ret != CL_VIRUS) { /* scan the raw file */ |
cdb0ae9c |
lseek(desc, 0, SEEK_SET); /* If archive scan didn't rewind desc */ |
e3aaff8e |
if(cli_scandesc(desc, virname, scanned, root) == CL_VIRUS) {
cli_dbgmsg("%s virus found in descriptor %d.\n", *virname, desc);
return CL_VIRUS;
} |
46c2e927 |
} |
e3aaff8e |
return ret;
}
int cl_scandesc(int desc, char **virname, unsigned long int *scanned, const struct cl_node *root, const struct cl_limits *limits, int options)
{
int reclev = 0;
return cli_magic_scandesc(desc, virname, scanned, root, limits, options, &reclev);
}
int cl_scanfile(const char *filename, char **virname, unsigned long int *scanned, const struct cl_node *root, const struct cl_limits *limits, int options)
{
int fd, ret;
if((fd = open(filename, O_RDONLY)) == -1)
return CL_EOPEN;
cli_dbgmsg("Scanning %s\n", filename);
ret = cl_scandesc(fd, virname, scanned, root, limits, options);
close(fd);
return ret;
} |