/*****************************************************************************
 *
 * $Id$
 * Purpose ...............: Setup Virus structure.
 *
 *****************************************************************************
 * Copyright (C) 1997-2001
 *   
 * Michiel Broek		FIDO:		2:280/2802
 * Beekmansbos 10
 * 1971 BV IJmuiden
 * the Netherlands
 *
 * This file is part of MBSE BBS.
 *
 * This BBS 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, or (at your option) any
 * later version.
 *
 * MB BBS 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 MB BBS; see the file COPYING.  If not, write to the Free
 * Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.
 *****************************************************************************/

#include "../lib/libs.h"
#include "../lib/structs.h"
#include "../lib/records.h"
#include "../lib/clcomm.h"
#include "../lib/common.h"
#include "screen.h"
#include "mutil.h"
#include "ledit.h"
#include "stlist.h"
#include "m_global.h"
#include "m_virus.h"



int	VirUpdated = 0;


/*
 * Count nr of virscan records in the database.
 * Creates the database if it doesn't exist.
 */
int CountVirus(void)
{
	FILE	*fil;
	char	ffile[PATH_MAX];
	int	count;

	sprintf(ffile, "%s/etc/virscan.data", getenv("MBSE_ROOT"));
	if ((fil = fopen(ffile, "r")) == NULL) {
		if ((fil = fopen(ffile, "a+")) != NULL) {
			Syslog('+', "Created new %s", ffile);
			virscanhdr.hdrsize = sizeof(virscanhdr);
			virscanhdr.recsize = sizeof(virscan);
			fwrite(&virscanhdr, sizeof(virscanhdr), 1, fil);

			/*
			 *  Create some default records but don't enable them.
			 */
			memset(&virscan, 0, sizeof(virscan));
			sprintf(virscan.comment, "AntiVir/Linux Scanner");
			sprintf(virscan.scanner, "/usr/bin/antivir");
			sprintf(virscan.options, "-allfiles -s -q");
			fwrite(&virscan, sizeof(virscan), 1, fil);

			memset(&virscan, 0, sizeof(virscan));
			sprintf(virscan.comment, "F-Prot scanner");
			sprintf(virscan.scanner, "/usr/local/bin/f-prot .");
			sprintf(virscan.options, "-archive -silent");
			fwrite(&virscan, sizeof(virscan), 1, fil);

                        memset(&virscan, 0, sizeof(virscan));
                        sprintf(virscan.comment, "McAfee VirusScan for Linux");
                        sprintf(virscan.scanner, "/usr/local/bin/uvscan");
                        sprintf(virscan.options, "--noboot --noexpire -r --secure -");
                        fwrite(&virscan, sizeof(virscan), 1, fil);

			fclose(fil);
			chmod(ffile, 0640);
			return 3;
		} else
			return -1;
	}

	fread(&virscanhdr, sizeof(virscanhdr), 1, fil);
	fseek(fil, 0, SEEK_END);
	count = (ftell(fil) - virscanhdr.hdrsize) / virscanhdr.recsize;
	fclose(fil);

	return count;
}



/*
 * Open database for editing. The datafile is copied, if the format
 * is changed it will be converted on the fly. All editing must be 
 * done on the copied file.
 */
int OpenVirus(void);
int OpenVirus(void)
{
	FILE	*fin, *fout;
	char	fnin[PATH_MAX], fnout[PATH_MAX];
	long	oldsize;

	sprintf(fnin,  "%s/etc/virscan.data", getenv("MBSE_ROOT"));
	sprintf(fnout, "%s/etc/virscan.temp", getenv("MBSE_ROOT"));
	if ((fin = fopen(fnin, "r")) != NULL) {
		if ((fout = fopen(fnout, "w")) != NULL) {
			fread(&virscanhdr, sizeof(virscanhdr), 1, fin);
			/*
			 * In case we are automatic upgrading the data format
			 * we save the old format. If it is changed, the
			 * database must always be updated.
			 */
			oldsize = virscanhdr.recsize;
			if (oldsize != sizeof(virscan)) {
				VirUpdated = 1;
				Syslog('+', "Upgraded %s, format changed", fnin);
			} else
				VirUpdated = 0;
			virscanhdr.hdrsize = sizeof(virscanhdr);
			virscanhdr.recsize = sizeof(virscan);
			fwrite(&virscanhdr, sizeof(virscanhdr), 1, fout);

			/*
			 * The datarecord is filled with zero's before each
			 * read, so if the format changed, the new fields
			 * will be empty.
			 */
			memset(&virscan, 0, sizeof(virscan));
			while (fread(&virscan, oldsize, 1, fin) == 1) {
				fwrite(&virscan, sizeof(virscan), 1, fout);
				memset(&virscan, 0, sizeof(virscan));
			}

			fclose(fin);
			fclose(fout);
			return 0;
		} else
			return -1;
	}
	return -1;
}



void CloseVirus(int);
void CloseVirus(int force)
{
	char	fin[PATH_MAX], fout[PATH_MAX];
	FILE	*fi, *fo;
	st_list	*vir = NULL, *tmp;

	sprintf(fin, "%s/etc/virscan.data", getenv("MBSE_ROOT"));
	sprintf(fout,"%s/etc/virscan.temp", getenv("MBSE_ROOT"));

	if (VirUpdated == 1) {
		if (force || (yes_no((char *)"Database is changed, save changes") == 1)) {
			working(1, 0, 0);
			fi = fopen(fout, "r");
			fo = fopen(fin,  "w");
			fread(&virscanhdr, virscanhdr.hdrsize, 1, fi);
			fwrite(&virscanhdr, virscanhdr.hdrsize, 1, fo);

			while (fread(&virscan, virscanhdr.recsize, 1, fi) == 1)
				if (!virscan.deleted)
					fill_stlist(&vir, virscan.comment, ftell(fi) - virscanhdr.recsize);
			sort_stlist(&vir);

			for (tmp = vir; tmp; tmp = tmp->next) {
				fseek(fi, tmp->pos, SEEK_SET);
				fread(&virscan, virscanhdr.recsize, 1, fi);
				fwrite(&virscan, virscanhdr.recsize, 1, fo);
			}

			tidy_stlist(&vir);
			fclose(fi);
			fclose(fo);
			unlink(fout);
			chmod(fin, 0640);
			Syslog('+', "Updated \"virscan.data\"");
			return;
		}
	}
	working(1, 0, 0);
	unlink(fout); 
}



int AppendVirus(void)
{
	FILE	*fil;
	char	ffile[PATH_MAX];

	sprintf(ffile, "%s/etc/virscan.temp", getenv("MBSE_ROOT"));
	if ((fil = fopen(ffile, "a")) != NULL) {
		memset(&virscan, 0, sizeof(virscan));
		fwrite(&virscan, sizeof(virscan), 1, fil);
		fclose(fil);
		VirUpdated = 1;
		return 0;
	} else
		return -1;
}



/*
 * Edit one record, return -1 if there are errors, 0 if ok.
 */
int EditVirRec(int Area)
{
	FILE	*fil;
	char	mfile[PATH_MAX];
	long	offset;
	int	j;
	unsigned long crc, crc1;

	clr_index();
	working(1, 0, 0);
	IsDoing("Edit VirScan");

	sprintf(mfile, "%s/etc/virscan.temp", getenv("MBSE_ROOT"));
	if ((fil = fopen(mfile, "r")) == NULL) {
		working(2, 0, 0);
		return -1;
	}

	offset = sizeof(virscanhdr) + ((Area -1) * sizeof(virscan));
	if (fseek(fil, offset, 0) != 0) {
		working(2, 0, 0);
		return -1;
	}

	fread(&virscan, sizeof(virscan), 1, fil);
	fclose(fil);
	crc = 0xffffffff;
	crc = upd_crc32((char *)&virscan, crc, sizeof(virscan));
	working(0, 0, 0);

	set_color(WHITE, BLACK);
	mvprintw( 5, 2, "4.  EDIT VIRUS SCANNER");
	set_color(CYAN, BLACK);
	mvprintw( 7, 2, "1.  Comment");
	mvprintw( 8, 2, "2.  Command");
	mvprintw( 9, 2, "3.  Options");
	mvprintw(10, 2, "4.  Available");
	mvprintw(11, 2, "5.  Deleted");
	mvprintw(12, 2, "6.  Error lvl");

	for (;;) {
		set_color(WHITE, BLACK);
		show_str(  7,16,40, virscan.comment);
		show_str(  8,16,64, virscan.scanner);
		show_str(  9,16,64, virscan.options);
		show_bool(10,16,    virscan.available);
		show_bool(11,16,    virscan.deleted);
		show_int( 12,16,    virscan.error);

		j = select_menu(6);
		switch(j) {
		case 0:	crc1 = 0xffffffff;
			crc1 = upd_crc32((char *)&virscan, crc1, sizeof(virscan));
			if (crc != crc1) {
				if (yes_no((char *)"Record is changed, save") == 1) {
					working(1, 0, 0);
					if ((fil = fopen(mfile, "r+")) == NULL) {
						working(2, 0, 0);
						return -1;
					}
					fseek(fil, offset, 0);
					fwrite(&virscan, sizeof(virscan), 1, fil);
					fclose(fil);
					VirUpdated = 1;
					working(0, 0, 0);
				}
			}
			IsDoing("Browsing Menu");
			return 0;
		case 1:	E_STR(  7,16,40,virscan.comment,  "The ^Comment^ for this record")
		case 2:	E_STR(  8,16,64,virscan.scanner,  "The full ^name and path^ to the binary of this scanner")
		case 3:	E_STR(  9,16,64,virscan.options,  "The ^commandline options^ for this scanner")
		case 4:	E_BOOL(10,16,   virscan.available,"Switch if this virus scanner is ^Available^ for use.")
		case 5:	E_BOOL(11,16,   virscan.deleted,  "Is this scanner ^deleted^")
		case 6:	E_INT( 12,16,   virscan.error,    "The ^Error Level^ the scanner returns when no virus is found")
		}
	}

	return 0;
}



void EditVirus(void)
{
	int	records, i, x, y;
	char	pick[12];
	FILE	*fil;
	char	temp[PATH_MAX];
	long	offset;

	clr_index();
	working(1, 0, 0);
	IsDoing("Browsing Menu");
	if (config_read() == -1) {
		working(2, 0, 0);
		return;
	}

	records = CountVirus();
	if (records == -1) {
		working(2, 0, 0);
		return;
	}

	if (OpenVirus() == -1) {
		working(2, 0, 0);
		return;
	}
	working(0, 0, 0);

	for (;;) {
		clr_index();
		set_color(WHITE, BLACK);
		mvprintw( 5, 4, "4.  VIRUS SCANNERS SETUP");
		set_color(CYAN, BLACK);
		if (records != 0) {
			sprintf(temp, "%s/etc/virscan.temp", getenv("MBSE_ROOT"));
			if ((fil = fopen(temp, "r")) != NULL) {
				fread(&virscanhdr, sizeof(virscanhdr), 1, fil);
				x = 2;
				y = 7;
				set_color(CYAN, BLACK);
				for (i = 1; i <= records; i++) {
					offset = sizeof(virscanhdr) + ((i - 1) * virscanhdr.recsize);
					fseek(fil, offset, 0);
					fread(&virscan, virscanhdr.recsize, 1, fil);
					if (i == 11) {
						x = 42;
						y = 7;
					}
					if (virscan.available)
						set_color(CYAN, BLACK);
					else
						set_color(LIGHTBLUE, BLACK);
					sprintf(temp, "%3d.  %-32s", i, virscan.comment);
					temp[37] = 0;
					mvprintw(y, x, temp);
					y++;
				}
				fclose(fil);
			}
		}
		strcpy(pick, select_record(records, 20));
		
		if (strncmp(pick, "-", 1) == 0) {
			CloseVirus(FALSE);
			return;
		}

		if (strncmp(pick, "A", 1) == 0) {
			working(1, 0, 0);
			if (AppendVirus() == 0) {
				records++;
				working(3, 0, 0);
			} else
				working(2, 0, 0);
			working(0, 0, 0);
		}

		if ((atoi(pick) >= 1) && (atoi(pick) <= records))
			EditVirRec(atoi(pick));
	}
}



void InitVirus(void)
{
    CountVirus();
    OpenVirus();
    CloseVirus(TRUE);
}



int virus_doc(FILE *fp, FILE *toc, int page)
{
	char	temp[PATH_MAX];
	FILE	*vir;
	int	j;

	sprintf(temp, "%s/etc/virscan.data", getenv("MBSE_ROOT"));
	if ((vir = fopen(temp, "r")) == NULL)
		return page;

	page = newpage(fp, page);
	addtoc(fp, toc, 4, 0, page, (char *)"Virus scanners");
	j = 0;
	fprintf(fp, "\n\n");
	fread(&virscanhdr, sizeof(virscanhdr), 1, vir);

	while ((fread(&virscan, virscanhdr.recsize, 1, vir)) == 1) {

		if (j == 5) {
			page = newpage(fp, page);
			fprintf(fp, "\n");
			j = 0;
		}

		fprintf(fp, "      Scanner name   %s\n", virscan.comment);
		fprintf(fp, "      Command line   %s\n", virscan.scanner);
		fprintf(fp, "      Options        %s\n", virscan.options);
		fprintf(fp, "      Available      %s\n", getboolean(virscan.available));
		fprintf(fp, "      Errorlevel OK  %d\n", virscan.error); 
		fprintf(fp, "\n\n\n");
		j++;
	}

	fclose(vir);
	return page;
}