clamav/contrib/test/libclamav/libclamav.c

/*
 *  Copyright (C) 2006 Nigel Horne <njh@bandsman.co.uk>
 *
 *  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., 51 Franklin Street, Fifth Floor, Boston,
 *  MA 02110-1301, USA.
 *
 * $CC $CFLAGS -I../.. -I../../libclamav debugm.c -lclamav
 * Now try a.out /sbin/* or a.out /usr/bin/*
 *
 * njh@bandsman.co.uk
 */
#include <stdio.h>
#include <errno.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <fcntl.h>
#include <malloc.h>
#include <clamav.h>
#include <sys/resource.h>
#include <features.h>
#include <unistd.h>
#include <memory.h>
#include <pthread.h>
#include "clamav-config.h"
#include "others.h"
#include "mbox.h"
#include "pdf.h"
#include "binhex.h"
#include "untar.h"
#include "special.h"
#include "tnef.h"
#include "pst.h"
#include "pe.h"

struct args {
	cli_ctx	ctx;
	const	char	*filename;
};

static void *
mbox(void *v)
{
	struct args *args = (struct args *)v;
	int fd = open(args->filename, O_RDONLY);

	if(fd < 0) {
		perror(args->filename);
		return NULL;
	}
	printf("cl_mbox(%s) returns %d\n",
		args->filename, cli_mbox("/tmp/libclamav", fd, &args->ctx));
	close(fd);

	return NULL;
}

static void *
pdf(void *v)
{
	struct args *args = (struct args *)v;
	int fd = open(args->filename, O_RDONLY);

	if(fd < 0) {
		perror(args->filename);
		return NULL;
	}
	printf("cl_pdf(%s) returns %d\n",
		args->filename, cli_pdf("/tmp/libclamav", fd, &args->ctx));
	close(fd);

	return NULL;
}

static void *
scandir(void *v)
{
	struct args *args = (struct args *)v;
	int fd = open(args->filename, O_RDONLY);

	if(fd < 0) {
		perror(args->filename);
		return NULL;
	}
	printf("cl_scandir(%s) returns %d\n",
		args->filename, cli_scandir("/tmp/libclamav", &args->ctx));
	close(fd);

	return NULL;
}

static void *
untar(void *v)
{
	struct args *args = (struct args *)v;
	int fd = open(args->filename, O_RDONLY);

	if(fd < 0) {
		perror(args->filename);
		return NULL;
	}
	printf("cl_untar(%s) returns %d\n",
		args->filename, cli_untar("/tmp/libclamav", fd, 1));
	close(fd);

	return NULL;
}

static void *
binhex(void *v)
{
	struct args *args = (struct args *)v;
	int fd = open(args->filename, O_RDONLY);

	if(fd < 0) {
		perror(args->filename);
		return NULL;
	}
	printf("cl_binhex(%s) returns %d\n",
		args->filename, cli_binhex("/tmp/libclamav", fd));
	close(fd);

	return NULL;
}

static void *
jpeg(void *v)
{
	struct args *args = (struct args *)v;
	int fd = open(args->filename, O_RDONLY);

	if(fd < 0) {
		perror(args->filename);
		return NULL;
	}
	printf("cli_check_jpeg_exploit(%s) returns %d\n",
		args->filename, cli_check_jpeg_exploit(fd));
	close(fd);

	return NULL;
}

static void *
tnef(void *v)
{
	struct args *args = (struct args *)v;
	int fd = open(args->filename, O_RDONLY);

	if(fd < 0) {
		perror(args->filename);
		return NULL;
	}
	printf("cli_tnef(%s) returns %d\n",
		args->filename, cli_tnef("/tmp/libclamav", fd));
	close(fd);

	return NULL;
}

static void *
uuencode(void *v)
{
	struct args *args = (struct args *)v;
	int fd = open(args->filename, O_RDONLY);

	if(fd < 0) {
		perror(args->filename);
		return NULL;
	}
	printf("cli_uuencode(%s) returns %d\n",
		args->filename, cli_uuencode("/tmp/libclamav", fd));
	close(fd);

	return NULL;
}

static void *
pst(void *v)
{
	struct args *args = (struct args *)v;
	int fd = open(args->filename, O_RDONLY);

	if(fd < 0) {
		perror(args->filename);
		return NULL;
	}
	printf("cli_pst(%s) returns %d\n",
		args->filename, cli_pst("/tmp/libclamav", fd));
	close(fd);

	return NULL;
}

static void *
pe(void *v)
{
	struct args *args = (struct args *)v;
	int fd = open(args->filename, O_RDONLY);

	if(fd < 0) {
		perror(args->filename);
		return NULL;
	}
	printf("cli_scanpe(%s) returns %d\n",
		args->filename, cli_scanpe(fd, &args->ctx));
	close(fd);

	return NULL;
}

int
main(int argc, char **argv)
{
	struct rlimit rlim;
	const char *virname;
	struct cl_engine engine;
	struct cl_limits limits;
	unsigned long scanned;
	struct args args;

	if(argc == 1) {
		fprintf(stderr, "Usage: %s files...\n", argv[0]);
		return 1;
	}
	rlim.rlim_cur = rlim.rlim_max = RLIM_INFINITY;
	if(setrlimit(RLIMIT_CORE, &rlim) < 0)
		perror("setrlimit");

	if((mkdir("/tmp/libclamav", 0750) < 0) && (errno != EEXIST)) {
		perror("/tmp/libclamav");
		return errno;
	}

	memset(&args.ctx, '\0', sizeof(cli_ctx));
	args.ctx.engine = &engine;
	args.ctx.virname = &virname;
	args.ctx.limits = &limits;
	args.ctx.scanned = &scanned;
	args.ctx.options = 0;

	while(*++argv) {
		pthread_t t;

		args.filename = *argv;

		if(pthread_create(&t, NULL, mbox, &args) != 0)
			perror("pthread_create");
		if(pthread_detach(t) != 0)
			perror("pthread_detach");

		if(pthread_create(&t, NULL, pdf, &args) != 0)
			perror("pthread_create");
		if(pthread_detach(t) != 0)
			perror("pthread_detach");

		if(pthread_create(&t, NULL, untar, &args) != 0)
			perror("pthread_create");
		if(pthread_detach(t) != 0)
			perror("pthread_detach");

		if(pthread_create(&t, NULL, binhex, &args) != 0)
			perror("pthread_create");
		if(pthread_detach(t) != 0)
			perror("pthread_detach");

		if(pthread_create(&t, NULL, jpeg, &args) != 0)
			perror("pthread_create");
		if(pthread_detach(t) != 0)
			perror("pthread_detach");

		if(pthread_create(&t, NULL, tnef, &args) != 0)
			perror("pthread_create");
		if(pthread_detach(t) != 0)
			perror("pthread_detach");

		if(pthread_create(&t, NULL, uuencode, &args) != 0)
			perror("pthread_create");
		if(pthread_detach(t) != 0)
			perror("pthread_detach");

		if(pthread_create(&t, NULL, pst, &args) != 0)
			perror("pthread_create");
		if(pthread_detach(t) != 0)
			perror("pthread_detach");

		if(pthread_create(&t, NULL, pe, &args) != 0)
			perror("pthread_create");
		if(pthread_detach(t) != 0)
			perror("pthread_detach");

		/* TODO: pass through all in cli_magic_scandesc */
		if(pthread_create(&t, NULL, scandir, &args) != 0)
			perror("pthread_create");
		if(pthread_detach(t) != 0)
			perror("pthread_detach");
	}
	puts("Hit SIGINT when all is finished");
	pause();
	puts("Finished - don't forget to rm -rf /tmp/libclamav");

	return 0;
}