/***************************************************************************
 *   Copyright (C) 03/2005 by Olaf Rempel                                  *
 *   razzor@kopf-tisch.de                                                  *
 *                                                                         *
 *   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.,                                       *
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 ***************************************************************************/
#include <string.h>
#include "plugin.h"

struct scan_ports port_arr[] = {
	{ 27960, 27969,  6 }, /* Quake3(6), Elite Force(7), Enemy Territory(25) */
	{ 28960, 28963, 31 }, /* Call of Duty(31), Call of Duty: United Offensive (42) */
	{ 0,0,0 }
};

static char scanmsg[] = "\xff\xff\xff\xffgetStatus";
static char replymsg[] = "\xff\xff\xff\xffstatusResponse";

static char q3_reply[] = "\\version\\Q3 ";
static char ef_reply[] = "\\version\\ST:V ";
static char et_reply[] = "\\version\\ET ";
static char cod_reply[] = "\\gamename\\Call of Duty\\";
static char coduo_reply[] = "\\gamename\\CoD:United Offensive\\";

int scan(void) {
	pkt_send_portarr(NULL, port_arr, scanmsg, strlen(scanmsg));
	return 1;
}

int parse(struct net_pkt *pkt) {
	int gameid;

	if (!(gameid = pkt_check_portarr(pkt, port_arr)))
		return 0;
	
	if (pkt_memcmp(pkt, 0, replymsg, strlen(replymsg)))
		return 0;

	switch (gameid) {
	case  6: /* q3, ef, et */
		if (pkt_memmem(pkt, 0, q3_reply, strlen(q3_reply))) {
			gameid = 6;

		} else if (pkt_memmem(pkt, 0, ef_reply, strlen(ef_reply))) {
			gameid = 7;
		
		} else if (pkt_memmem(pkt, 0, et_reply, strlen(et_reply))) {
			gameid = 25;

		} else {
			return 0;
		}
		break;
		
	case 31: /* cod, cod:uo */
		if (pkt_memmem(pkt, 0, cod_reply, strlen(cod_reply))) {
			gameid = 31;

		} else if (pkt_memmem(pkt, 0, coduo_reply, strlen(coduo_reply))) {
			gameid = 42;

		} else {
			return 0;
		}
		break;		
	}

	server_add_pkt(gameid, pkt);
	return 1;
}

static struct hlswmaster_plugin plugin = {
	.name		= "q3engine",
	.scan		= &scan,
	.parse		= &parse,
};

void _init(void) {
	register_plugin(&plugin);
}

void _fini(void) {
	unregister_plugin(&plugin);
}