/* For terms of usage/redistribution/modification see the LICENSE file */
/* For authors and contributors see the AUTHORS file */

/***

fltselect.c - a menu-based module that allows selection of
	other protocols to display

 ***/

#include "iptraf-ng-compat.h"

#include "tui/menurt.h"
#include "tui/msgboxes.h"
#include "tui/winops.h"

#include "addproto.h"
#include "dirs.h"
#include "fltdefs.h"
#include "fltselect.h"
#include "fltedit.h"
#include "fltmgr.h"
#include "ipfilter.h"
#include "deskman.h"
#include "attrs.h"

struct filterstate ofilter;

void makemainfiltermenu(struct MENU *menu)
{
	tx_initmenu(menu, 8, 18, (LINES - 8) / 2, (COLS - 31) / 2, BOXATTR,
		    STDATTR, HIGHATTR, BARSTDATTR, BARHIGHATTR, DESCATTR);
	tx_additem(menu, " ^I^P...", "Manages IP packet filters");
	tx_additem(menu, " ^A^RP",
		   "Toggles Address Resolution Protocol filter");
	tx_additem(menu, " ^R^ARP", "Toggles Reverse ARP filter");
	tx_additem(menu, " ^N^on-IP",
		   "Toggles filter for all other non-IP packets");
	tx_additem(menu, NULL, NULL);
	tx_additem(menu, " E^x^it menu",
		   "Returns to the filter management menu");
}

void setfilters(unsigned int row)
{
	int aborted;

	switch (row) {
	case 1:
		ipfilterselect(&aborted);
		break;
	case 2:
		ofilter.arp = ~ofilter.arp;
		break;
	case 3:
		ofilter.rarp = ~ofilter.rarp;
		break;
	case 4:
		ofilter.nonip = ~ofilter.nonip;
		break;
	}
}

void toggleprotodisplay(WINDOW *win, unsigned int row)
{
	wmove(win, row, 2);
	switch (row) {
	case 1:
		if (ofilter.filtercode == 0)
			wprintw(win, "No IP filter active");
		else
			wprintw(win, "IP filter active   ");
		break;
	case 2:
		if (ofilter.arp)
			wprintw(win, "ARP visible    ");
		else
			wprintw(win, "ARP not visible");

		break;
	case 3:
		if (ofilter.rarp)
			wprintw(win, "RARP visible    ");
		else
			wprintw(win, "RARP not visible");

		break;
	case 4:
		if (ofilter.nonip)
			wprintw(win, "Non-IP visible    ");
		else
			wprintw(win, "Non-IP not visible");

		break;
	}
}

/*
 * Filter for non-IP packets
 */
int nonipfilter(unsigned int protocol)
{
	int result = 0;

	switch (protocol) {
	case ETH_P_ARP:
		result = ofilter.arp;
		break;
	case ETH_P_RARP:
		result = ofilter.rarp;
		break;
	default:
		result = ofilter.nonip;
		break;
	}

	return result;
}

void config_filters(void)
{
	struct MENU menu;
	WINDOW *statwin;
	PANEL *statpanel;
	int row;
	int aborted;

	statwin = newwin(6, 30, (LINES - 8) / 2, (COLS - 15) / 2 + 10);
	statpanel = new_panel(statwin);
	wattrset(statwin, BOXATTR);
	tx_colorwin(statwin);
	tx_box(statwin, ACS_VLINE, ACS_HLINE);
	tx_stdwinset(statwin);
	wmove(statwin, 0, 1);
	wprintw(statwin, " Filter Status ");
	wattrset(statwin, STDATTR);

	for (row = 1; row <= 4; row++)
		toggleprotodisplay(statwin, row);

	makemainfiltermenu(&menu);

	row = 1;
	do {
		tx_showmenu(&menu);
		tx_operatemenu(&menu, &row, &aborted);
		setfilters(row);
		toggleprotodisplay(statwin, row);
	} while (row != 6);

	tx_destroymenu(&menu);
	del_panel(statpanel);
	delwin(statwin);
	update_panels();
	doupdate();
}

void setodefaults(void)
{
	memset(&ofilter, 0, sizeof(struct filterstate));
	ofilter.filtercode = 0;
}

void loadfilters(void)
{
	int pfd;
	int br;

	pfd = open(FLTSTATEFILE, O_RDONLY);	/* open filter state file */

	if (pfd < 0) {
		setodefaults();
		return;
	}
	br = read(pfd, &ofilter, sizeof(struct filterstate));
	if (br < 0)
		setodefaults();

	close(pfd);

	/*
	 * Reload IP filter if one was previously applied
	 */

	if (ofilter.filtercode != 0)
		loadfilter(ofilter.filename, &ofilter.fl, FLT_RESOLVE);
}

void savefilters(void)
{
	int pfd;
	int bw;

	pfd =
	    open(FLTSTATEFILE, O_CREAT | O_TRUNC | O_WRONLY, S_IRUSR | S_IWUSR);
	bw = write(pfd, &ofilter, sizeof(struct filterstate));
	if (bw < 1)
		tui_error(ANYKEY_MSG,
			  "Unable to write filter state information");

	close(pfd);

}