#include "inspircd.h"
#include "inspircd_io.h"
#include "inspircd_util.h"
#include "inspircd_config.h"
#include <unistd.h>
#include <fcntl.h>
#include <sys/errno.h>
#include <sys/ioctl.h>
#include <sys/utsname.h>
#include <cstdio>
#include <time.h>
#include <string>
#ifdef GCC3
#include <ext/hash_map>
#else
#include <hash_map>
#endif
#include <map>
#include <sstream>
#include <vector>
#include <errno.h>
#include <deque>
#include <errno.h>
#include <unistd.h>
#include <sched.h>
#include "connection.h"
#include "users.h"
#include "servers.h"
#include "ctables.h"
#include "globals.h"
#include "modules.h"
#include "dynamic.h"
#include "wildcard.h"
#include "message.h"
#include "commands.h"
#include "xline.h"

#ifdef GCC3
#define nspace __gnu_cxx
#else
#define nspace std
#endif


using namespace std;

extern int MODCOUNT;
extern vector<Module*> modules;
extern vector<ircd_module*> factory;

extern int LogLevel;
extern char ServerName[MAXBUF];
extern char Network[MAXBUF];
extern char ServerDesc[MAXBUF];
extern char AdminName[MAXBUF];
extern char AdminEmail[MAXBUF];
extern char AdminNick[MAXBUF];
extern char diepass[MAXBUF];
extern char restartpass[MAXBUF];
extern char motd[MAXBUF];
extern char rules[MAXBUF];
extern char list[MAXBUF];
extern char PrefixQuit[MAXBUF];
extern char DieValue[MAXBUF];

extern int debugging;
extern int WHOWAS_STALE;
extern int WHOWAS_MAX;
extern int DieDelay;
extern time_t startup_time;
extern int NetBufferSize;
extern time_t nb_start;

extern std::vector<int> fd_reap;
extern std::vector<std::string> module_names;

extern char bannerBuffer[MAXBUF];
extern int boundPortCount;
extern int portCount;
extern int UDPportCount;
extern int ports[MAXSOCKS];
extern int defaultRoute;

extern std::vector<long> auth_cookies;
extern std::stringstream config_f;

extern serverrec* me[32];

extern FILE *log_file;

namespace nspace
{
	template<> struct nspace::hash<in_addr>
	{
		size_t operator()(const struct in_addr &a) const
		{
			size_t q;
			memcpy(&q,&a,sizeof(size_t));
			return q;
		}
	};

	template<> struct nspace::hash<string>
	{
		size_t operator()(const string &s) const
		{
			char a[MAXBUF];
			static struct hash<const char *> strhash;
			strcpy(a,s.c_str());
			strlower(a);
			return strhash(a);
		}
	};
}	


struct StrHashComp
{

	bool operator()(const string& s1, const string& s2) const
	{
		char a[MAXBUF],b[MAXBUF];
		strcpy(a,s1.c_str());
		strcpy(b,s2.c_str());
		return (strcasecmp(a,b) == 0);
	}

};

struct InAddr_HashComp
{

	bool operator()(const in_addr &s1, const in_addr &s2) const
	{
		size_t q;
		size_t p;
		
		memcpy(&q,&s1,sizeof(size_t));
		memcpy(&p,&s2,sizeof(size_t));
		
		return (q == p);
	}

};


typedef nspace::hash_map<std::string, userrec*, nspace::hash<string>, StrHashComp> user_hash;
typedef nspace::hash_map<std::string, chanrec*, nspace::hash<string>, StrHashComp> chan_hash;
typedef nspace::hash_map<in_addr,string*, nspace::hash<in_addr>, InAddr_HashComp> address_cache;
typedef std::deque<command_t> command_table;


extern user_hash clientlist;
extern chan_hash chanlist;
extern user_hash whowas;
extern command_table cmdlist;
extern file_cache MOTD;
extern file_cache RULES;
extern address_cache IP;


std::vector<KLine> klines;
std::vector<GLine> glines;
std::vector<ZLine> zlines;
std::vector<QLine> qlines;

// Reads the default bans from the config file.
// only a very small number of bans are defined
// this way these days, such as qlines against 
// services nicks, etc.

void read_xline_defaults()
{
	char ipmask[MAXBUF];
	char nick[MAXBUF];
	char host[MAXBUF];
	char reason[MAXBUF];

	for (int i = 0; i < ConfValueEnum("badip",&config_f); i++)
	{
		ConfValue("badip","ipmask",i,ipmask,&config_f);
		ConfValue("badip","reason",i,reason,&config_f);
		add_zline(0,"<Config>",reason,ipmask);
		log(DEBUG,"Read Z line (badip tag): ipmask=%s reason=%s",ipmask,reason);
	}
	
	for (int i = 0; i < ConfValueEnum("badnick",&config_f); i++)
	{
		ConfValue("badnick","nick",i,nick,&config_f);
		ConfValue("badnick","reason",i,reason,&config_f);
		add_qline(0,"<Config>",reason,nick);
		log(DEBUG,"Read Q line (badnick tag): nick=%s reason=%s",nick,reason);
	}
	
	for (int i = 0; i < ConfValueEnum("badhost",&config_f); i++)
	{
		ConfValue("badhost","host",i,host,&config_f);
		ConfValue("badhost","reason",i,reason,&config_f);
		add_kline(0,"<Config>",reason,host);
		log(DEBUG,"Read K line (badhost tag): host=%s reason=%s",host,reason);
	}
}

// adds a g:line

void add_gline(long duration, char* source, char* reason, char* hostmask)
{
	del_gline(hostmask);
	GLine item;
	item.duration = duration;
	strncpy(item.hostmask,hostmask,MAXBUF);
	strncpy(item.reason,reason,MAXBUF);
	strncpy(item.source,source,MAXBUF);
	item.n_matches = 0;
	item.set_time = time(NULL);
	glines.push_back(item);
}

// adds a q:line

void add_qline(long duration, char* source, char* reason, char* nickname)
{
	del_qline(nickname);
	QLine item;
	item.duration = duration;
	strncpy(item.nick,nickname,MAXBUF);
	strncpy(item.reason,reason,MAXBUF);
	strncpy(item.source,source,MAXBUF);
	item.n_matches = 0;
	item.is_global = false;
	item.set_time = time(NULL);
	qlines.push_back(item);
}

// adds a z:line

void add_zline(long duration, char* source, char* reason, char* ipaddr)
{
	del_zline(ipaddr);
	ZLine item;
	item.duration = duration;
	strncpy(item.ipaddr,ipaddr,MAXBUF);
	strncpy(item.reason,reason,MAXBUF);
	strncpy(item.source,source,MAXBUF);
	item.n_matches = 0;
	item.is_global = false;
	item.set_time = time(NULL);
	zlines.push_back(item);
}

// adds a k:line

void add_kline(long duration, char* source, char* reason, char* hostmask)
{
	del_kline(hostmask);
	KLine item;
	item.duration = duration;
	strncpy(item.hostmask,hostmask,MAXBUF);
	strncpy(item.reason,reason,MAXBUF);
	strncpy(item.source,source,MAXBUF);
	item.n_matches = 0;
	item.set_time = time(NULL);
	klines.push_back(item);
}

// deletes a g:line, returns true if the line existed and was removed

bool del_gline(char* hostmask)
{
	for (std::vector<GLine>::iterator i = glines.begin(); i != glines.end(); i++)
	{
		if (!strcasecmp(hostmask,i->hostmask))
		{
			glines.erase(i);
			return true;
		}
	}
	return false;
}

// deletes a q:line, returns true if the line existed and was removed

bool del_qline(char* nickname)
{
	for (std::vector<QLine>::iterator i = qlines.begin(); i != qlines.end(); i++)
	{
		if (!strcasecmp(nickname,i->nick))
		{
			qlines.erase(i);
			return true;
		}
	}
	return false;
}

bool qline_make_global(char* nickname)
{
	for (std::vector<QLine>::iterator i = qlines.begin(); i != qlines.end(); i++)
	{
		if (!strcasecmp(nickname,i->nick))
		{
			i->is_global = true;
			return true;
		}
	}
	return false;
}

bool zline_make_global(char* ipaddr)
{
	for (std::vector<ZLine>::iterator i = zlines.begin(); i != zlines.end(); i++)
	{
		if (!strcasecmp(ipaddr,i->ipaddr))
		{
			i->is_global = true;
			return true;
		}
	}
	return false;
}

void sync_xlines(serverrec* serv, char* tcp_host)
{
	char data[MAXBUF];
	
	// for zlines and qlines, we should first check if theyre global...
	for (std::vector<ZLine>::iterator i = zlines.begin(); i != zlines.end(); i++)
	{
		if (i->is_global)
		{
			snprintf(data,MAXBUF,"} %s %s %ld %ld :%s",i->ipaddr,i->source,i->set_time,i->duration,i->reason);
			serv->SendPacket(data,tcp_host);
		}
	}
	for (std::vector<QLine>::iterator i = qlines.begin(); i != qlines.end(); i++)
	{
		if (i->is_global)
		{
			snprintf(data,MAXBUF,"{ %s %s %ld %ld :%s",i->nick,i->source,i->set_time,i->duration,i->reason);
			serv->SendPacket(data,tcp_host);
		}
	}
	// glines are always global, so no need to check
	for (std::vector<GLine>::iterator i = glines.begin(); i != glines.end(); i++)
	{
		snprintf(data,MAXBUF,"# %s %s %ld %ld :%s",i->hostmask,i->source,i->set_time,i->duration,i->reason);
		serv->SendPacket(data,tcp_host);
	}
}


// deletes a z:line, returns true if the line existed and was removed

bool del_zline(char* ipaddr)
{
	for (std::vector<ZLine>::iterator i = zlines.begin(); i != zlines.end(); i++)
	{
		if (!strcasecmp(ipaddr,i->ipaddr))
		{
			zlines.erase(i);
			return true;
		}
	}
	return false;
}

// deletes a k:line, returns true if the line existed and was removed

bool del_kline(char* hostmask)
{
	for (std::vector<KLine>::iterator i = klines.begin(); i != klines.end(); i++)
	{
		if (!strcasecmp(hostmask,i->hostmask))
		{
			klines.erase(i);
			return true;
		}
	}
	return false;
}

// returns a pointer to the reason if a nickname matches a qline, NULL if it didnt match

char* matches_qline(const char* nick)
{
	for (std::vector<QLine>::iterator i = qlines.begin(); i != qlines.end(); i++)
	{
		if (match(nick,i->nick))
		{
			return i->reason;
		}
	}
	return NULL;
}

// returns a pointer to the reason if a host matches a gline, NULL if it didnt match

char* matches_gline(const char* host)
{
	for (std::vector<GLine>::iterator i = glines.begin(); i != glines.end(); i++)
	{
		if (match(host,i->hostmask))
		{
			return i->reason;
		}
	}
	return NULL;
}

void gline_set_creation_time(char* host, time_t create_time)
{
	for (std::vector<GLine>::iterator i = glines.begin(); i != glines.end(); i++)
	{
		if (!strcasecmp(host,i->hostmask))
		{
			i->set_time = create_time;
			return;
		}
	}
	return ;	
}

void qline_set_creation_time(char* nick, time_t create_time)
{
	for (std::vector<QLine>::iterator i = qlines.begin(); i != qlines.end(); i++)
	{
		if (!strcasecmp(nick,i->nick))
		{
			i->set_time = create_time;
			return;
		}
	}
	return ;	
}

void zline_set_creation_time(char* ip, time_t create_time)
{
	for (std::vector<ZLine>::iterator i = zlines.begin(); i != zlines.end(); i++)
	{
		if (!strcasecmp(ip,i->ipaddr))
		{
			i->set_time = create_time;
			return;
		}
	}
	return ;	
}

// returns a pointer to the reason if an ip address matches a zline, NULL if it didnt match

char* matches_zline(const char* ipaddr)
{
	for (std::vector<ZLine>::iterator i = zlines.begin(); i != zlines.end(); i++)
	{
		if (match(ipaddr,i->ipaddr))
		{
			return i->reason;
		}
	}
	return NULL;
}

// returns a pointer to the reason if a host matches a kline, NULL if it didnt match

char* matches_kline(const char* host)
{
	for (std::vector<KLine>::iterator i = klines.begin(); i != klines.end(); i++)
	{
		if (match(host,i->hostmask))
		{
			return i->reason;
		}
	}
	return NULL;
}

// removes lines that have expired

void expire_lines()
{
	bool go_again = true;
	time_t current = time(NULL);
	
	// because we mess up an iterator when we remove from the vector, we must bail from
	// the loop early if we delete an item, therefore this outer while loop is required.
	while (go_again)
	{
		go_again = false;

		for (std::vector<KLine>::iterator i = klines.begin(); i != klines.end(); i++)
		{
			if ((current > (i->duration + i->set_time)) && (i->duration > 0))
			{
				WriteOpers("Expiring timed K-Line %s (set by %s %d seconds ago)",i->hostmask,i->source,i->duration);
				klines.erase(i);
				go_again = true;
				break;
			}
		}

		for (std::vector<GLine>::iterator i = glines.begin(); i != glines.end(); i++)
		{
			if ((current > (i->duration + i->set_time)) && (i->duration > 0))
			{
				WriteOpers("Expiring timed G-Line %s (set by %s %d seconds ago)",i->hostmask,i->source,i->duration);
				glines.erase(i);
				go_again = true;
				break;
			}
		}

		for (std::vector<ZLine>::iterator i = zlines.begin(); i != zlines.end(); i++)
		{
			if ((current > (i->duration + i->set_time)) && (i->duration > 0))
			{
				WriteOpers("Expiring timed Z-Line %s (set by %s %d seconds ago)",i->ipaddr,i->source,i->duration);
				zlines.erase(i);
				go_again = true;
				break;
			}
		}

		for (std::vector<QLine>::iterator i = qlines.begin(); i != qlines.end(); i++)
		{
			if ((current > (i->duration + i->set_time)) && (i->duration > 0))
			{
				WriteOpers("Expiring timed Q-Line %s (set by %s %d seconds ago)",i->nick,i->source,i->duration);
				qlines.erase(i);
				go_again = true;
				break;
			}
		}
	}
}

// applies lines, removing clients and changing nicks etc as applicable

void apply_lines()
{
	bool go_again = true;
	char reason[MAXBUF];
	char host[MAXBUF];
	
	while (go_again)
	{
		go_again = false;
		for (user_hash::const_iterator u = clientlist.begin(); u != clientlist.end(); u++)
		{
			if (!strcasecmp(u->second->server,ServerName))
			{
				snprintf(host,MAXBUF,"%s@%s",u->second->ident,u->second->host);
				char* check = matches_gline(host);
				if (check)
				{
					WriteOpers("*** User %s matches G-Line: %s",u->second->nick,check);
					snprintf(reason,MAXBUF,"G-Lined: %s",check);
					kill_link(u->second,reason);
					go_again = true;
					break;
				}
			}
		}

		for (user_hash::const_iterator u = clientlist.begin(); u != clientlist.end(); u++)
		{
			if (!strcasecmp(u->second->server,ServerName))
			{
				snprintf(host,MAXBUF,"%s@%s",u->second->ident,u->second->host);
				char* check = matches_kline(host);
				if (check)
				{
					WriteOpers("*** User %s matches K-Line: %s",u->second->nick,check);
					snprintf(reason,MAXBUF,"K-Lined: %s",check);
					kill_link(u->second,reason);
					go_again = true;
					break;
				}
			}
		}

		for (user_hash::const_iterator u = clientlist.begin(); u != clientlist.end(); u++)
		{
			if (!strcasecmp(u->second->server,ServerName))
			{
				char* check = matches_qline(u->second->nick);
				if (check)
				{
					snprintf(reason,MAXBUF,"Matched Q-Lined nick: %s",check);
					WriteOpers("*** Q-Lined nickname %s from %s: %s",u->second->nick,u->second->host,check);
					WriteServ(u->second->fd,"432 %s %s :Invalid nickname: %s",u->second->nick,u->second->nick,check);
					kill_link(u->second,reason);
					go_again = true;
					break;
				}
			}
		}

		for (user_hash::const_iterator u = clientlist.begin(); u != clientlist.end(); u++)
		{
			if (!strcasecmp(u->second->server,ServerName))
			{
				char* check = matches_zline(u->second->ip);
				if (check)
				{
					snprintf(reason,MAXBUF,"Z-Lined: %s",check);
					WriteOpers("*** User %s matches Z-Line: %s",u->second->nick,u->second->host,check);
					kill_link(u->second,reason);
					go_again = true;
					break;
				}
			}
		}

	}
}

void stats_k(userrec* user)
{
	for (std::vector<KLine>::iterator i = klines.begin(); i != klines.end(); i++)
	{
		WriteServ(user->fd,"216 %s :%s %d %d %s %s",user->nick,i->hostmask,i->set_time,i->duration,i->source,i->reason);
	}
}

void stats_g(userrec* user)
{
	for (std::vector<GLine>::iterator i = glines.begin(); i != glines.end(); i++)
	{
		WriteServ(user->fd,"223 %s :%s %d %d %s %s",user->nick,i->hostmask,i->set_time,i->duration,i->source,i->reason);
	}
}

void stats_q(userrec* user)
{
	for (std::vector<QLine>::iterator i = qlines.begin(); i != qlines.end(); i++)
	{
		WriteServ(user->fd,"217 %s :%s %d %d %s %s",user->nick,i->nick,i->set_time,i->duration,i->source,i->reason);
	}
}

void stats_z(userrec* user)
{
	for (std::vector<ZLine>::iterator i = zlines.begin(); i != zlines.end(); i++)
	{
		WriteServ(user->fd,"223 %s :%s %d %d %s %s",user->nick,i->ipaddr,i->set_time,i->duration,i->source,i->reason);
	}
}