/***************************************************************************
 *   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 <stdio.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/ip.h>
#include <arpa/inet.h>
#include <inttypes.h>

#include "plugin_helper.h"

struct hlsw_entry {
	uint16_t gameid;
	uint32_t ip;
	uint16_t port1;
	uint16_t port2;
} __attribute__ ((packed));

static struct in_addr *ip_arr = NULL;
static int ip_arr_size = 0;

static char scanmsg[] = "\xff\xff\xff\xffHLSWLANSEARCH";

static int scan(void)
{
	int i;
	for (i = 0; i < ip_arr_size; i++)
		pkt_send(&ip_arr[i], 7140, scanmsg, sizeof(scanmsg));
	return 1;
}

static int parse(struct net_pkt *pkt)
{
	struct hlsw_entry *server;
	if (pkt_getport(pkt) != 7140)
		return PARSE_REJECT;

	server = (void *)pkt->buf + sizeof(scanmsg);

	while ((void *)server < ((void *)pkt->buf + pkt->size)) {
		server_add(server->gameid, server->ip, server->port1, server->port2);
		server++;
	}
	
	return PARSE_ACCEPT;
}

static int init(struct conf_section *section)
{
	struct conf_tupel *tupel;
	int i = 0;
	list_for_each_entry(tupel, &section->tupel, list)
		if (!strcmp(tupel->option, "scan_ip"))
			i++;
			
	if (i == 0 || !(ip_arr = malloc(sizeof(struct in_addr) * i)))
		return 0;

	i = 0;
	list_for_each_entry(tupel, &section->tupel, list) {
		if (!strcmp(tupel->option, "scan_ip")) {
			if (inet_pton(AF_INET, tupel->parameter, &ip_arr[i]) <= 0) {
				log_print(" invalid ip: %s", tupel->parameter);
			} else {
				log_print(" adding %s", tupel->parameter);
				i++;
			}
		}
	}
	ip_arr_size = i;
	return 1;
}

static int fini()
{
	if (ip_arr_size > 0 && ip_arr)
		free(ip_arr);
}

struct hlswmaster_plugin plugin = {
	.name		= "hlswproxy",
	.scan		= &scan,
	.parse		= &parse,
	.init		= &init,
	.fini		= &fini,
};