/* IPXWrapper - Router header
 * Copyright (C) 2011 Daniel Collins <solemnwarning@solemnwarning.net>
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License version 2 as published by
 * the Free Software Foundation.
 *
 * 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., 51
 * Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
*/

#ifndef IPXWRAPPER_ROUTER_H
#define IPXWRAPPER_ROUTER_H

#include <windows.h>
#include <winsock2.h>
#include <wsipx.h>
#include <stdint.h>

#define MAX_ROUTER_CLIENTS 128

struct router_call {
	enum {
		rc_bind,
		rc_unbind,
		rc_port,
		rc_filter,
		rc_flags,
		rc_remote
	} call;
	
	SOCKET sock;
	
	struct sockaddr_ipx arg_addr;
	int arg_int;
};

struct router_ret {
	int err_code;	/* ERROR_SUCCESS on success */
	
	struct sockaddr_ipx ret_addr;
	uint32_t ret_u32;
};

/* Represents a bound IPX address */
struct router_addr {
	struct sockaddr_ipx addr;
	
	uint16_t local_port;	/* Local UDP port (NETWORK BYTE ORDER) */
	SOCKET ws_socket;	/* Application socket */
	SOCKET control_socket;	/* Control socket */
	int filter_ptype;	/* Packet type filter, negative to disable */
	int flags;
	
	/* Address of IP interface */
	uint32_t ipaddr;
	uint32_t netmask;
	
	/* Only accept packets from this address (any if AF_UNSPEC) */
	struct sockaddr_ipx remote_addr;
	
	struct router_addr *next;
};

struct router_client {
	SOCKET sock;
	
	struct router_call recvbuf;
	int recvbuf_len;
};

struct router_vars {
	BOOL running;
	
	SOCKET udp_sock;
	SOCKET listener;
	
	struct router_client clients[MAX_ROUTER_CLIENTS];
	int client_count;
	
	WSAEVENT wsa_event;
	
	CRITICAL_SECTION crit_sec;
	BOOL crit_sec_init;
	
	struct router_addr *addrs;
	
	char *recvbuf;
};

struct rclient {
	CRITICAL_SECTION cs;
	BOOL cs_init;
	
	SOCKET sock;
	
	struct router_vars *router;
	HANDLE thread;
};

struct rpacket_header {
	uint32_t src_ipaddr;
	char spare[20];
} __attribute__((__packed__));

struct router_vars *router_init(BOOL global);
void router_destroy(struct router_vars *router);

DWORD router_main(void *arg);

BOOL rclient_init(struct rclient *rclient);
BOOL rclient_start(struct rclient *rclient);
void rclient_stop(struct rclient *rclient);

BOOL rclient_bind(struct rclient *rclient, SOCKET sock, struct sockaddr_ipx *addr, uint32_t *nic_bcast, int flags);
BOOL rclient_unbind(struct rclient *rclient, SOCKET sock);
BOOL rclient_set_port(struct rclient *rclient, SOCKET sock, uint16_t port);
BOOL rclient_set_filter(struct rclient *rclient, SOCKET sock, int ptype);
BOOL rclient_set_flags(struct rclient *rclient, SOCKET sock, int flags);
BOOL rclient_set_remote(struct rclient *rclient, SOCKET sock, const struct sockaddr_ipx *addr);

#endif /* !IPXWRAPPER_ROUTER_H */