celestar@5642: /* $Id$ */
celestar@5642: 
celestar@5642: #ifdef ENABLE_NETWORK
celestar@5642: 
celestar@5642: #include "../stdafx.h"
celestar@5642: #include "../debug.h"
celestar@5642: #include "../string.h"
celestar@5642: #include "network_data.h"
celestar@5642: #include "../date.h"
celestar@5642: #include "../map.h"
celestar@5642: #include "network_gamelist.h"
celestar@5642: #include "network_udp.h"
celestar@5642: #include "../variables.h"
celestar@5642: #include "../newgrf_config.h"
celestar@5642: 
celestar@5642: #include "core/udp.h"
celestar@5642: 
celestar@5642: /**
celestar@5642:  * @file network_udp.c This file handles the UDP related communication.
celestar@5642:  *
celestar@5642:  * This is the GameServer <-> MasterServer and GameServer <-> GameClient
celestar@5642:  * communication before the game is being joined.
celestar@5642:  */
celestar@5642: 
celestar@5642: enum {
celestar@5642: 	ADVERTISE_NORMAL_INTERVAL = 30000, // interval between advertising in ticks (15 minutes)
celestar@5642: 	ADVERTISE_RETRY_INTERVAL  =   300, // readvertise when no response after this many ticks (9 seconds)
celestar@5642: 	ADVERTISE_RETRY_TIMES     =     3  // give up readvertising after this much failed retries
celestar@5642: };
celestar@5642: 
celestar@5648: #define DEF_UDP_RECEIVE_COMMAND(type) void NetworkPacketReceive_ ## type ## _command(Packet *p, const struct sockaddr_in *client_addr)
celestar@5642: 
celestar@5642: static NetworkClientState _udp_cs;
celestar@5642: 
celestar@5642: DEF_UDP_RECEIVE_COMMAND(PACKET_UDP_CLIENT_FIND_SERVER)
celestar@5642: {
celestar@5642: 	Packet *packet;
celestar@5642: 	// Just a fail-safe.. should never happen
celestar@5642: 	if (!_network_udp_server)
celestar@5642: 		return;
celestar@5642: 
celestar@5642: 	packet = NetworkSend_Init(PACKET_UDP_SERVER_RESPONSE);
celestar@5642: 
celestar@5642: 	// Update some game_info
celestar@5642: 	_network_game_info.game_date     = _date;
celestar@5642: 	_network_game_info.map_width     = MapSizeX();
celestar@5642: 	_network_game_info.map_height    = MapSizeY();
celestar@5642: 	_network_game_info.map_set       = _opt.landscape;
celestar@5642: 	_network_game_info.companies_on  = ActivePlayerCount();
celestar@5642: 	_network_game_info.spectators_on = NetworkSpectatorCount();
celestar@5642: 	_network_game_info.grfconfig     = _grfconfig;
celestar@5642: 
celestar@5648: 	NetworkSend_NetworkGameInfo(packet, &_network_game_info);
celestar@5642: 
celestar@5642: 	// Let the client know that we are here
celestar@5642: 	NetworkSendUDP_Packet(_udp_server_socket, packet, client_addr);
celestar@5642: 
celestar@5642: 	free(packet);
celestar@5642: 
celestar@5642: 	DEBUG(net, 2, "[udp] queried from '%s'", inet_ntoa(client_addr->sin_addr));
celestar@5642: }
celestar@5642: 
celestar@5642: void HandleIncomingNetworkGameInfoGRFConfig(GRFConfig *config)
celestar@5642: {
celestar@5642: 	/* Find the matching GRF file */
celestar@5642: 	const GRFConfig *f = FindGRFConfig(config->grfid, config->md5sum);
celestar@5642: 	if (f == NULL) {
celestar@5642: 		/* Don't know the GRF, so mark game incompatible and the (possibly)
celestar@5642: 		 * already resolved name for this GRF (another server has sent the
celestar@5642: 		 * name of the GRF already */
celestar@5642: 		config->name     = FindUnknownGRFName(config->grfid, config->md5sum, true);
celestar@5642: 		SETBIT(config->flags, GCF_NOT_FOUND);
celestar@5642: 	} else {
celestar@5642: 		config->filename = f->filename;
celestar@5642: 		config->name     = f->name;
celestar@5642: 		config->info     = f->info;
celestar@5642: 	}
celestar@5642: 	SETBIT(config->flags, GCF_COPY);
celestar@5642: }
celestar@5642: 
celestar@5642: DEF_UDP_RECEIVE_COMMAND(PACKET_UDP_SERVER_RESPONSE)
celestar@5642: {
celestar@5642: 	extern const char _openttd_revision[];
celestar@5642: 	NetworkGameList *item;
celestar@5642: 
celestar@5642: 	// Just a fail-safe.. should never happen
celestar@5642: 	if (_network_udp_server || _udp_cs.has_quit) return;
celestar@5642: 
celestar@5642: 	DEBUG(net, 4, "[udp] server response from %s:%d", inet_ntoa(client_addr->sin_addr),ntohs(client_addr->sin_port));
celestar@5642: 
celestar@5642: 	// Find next item
celestar@5642: 	item = NetworkGameListAddItem(inet_addr(inet_ntoa(client_addr->sin_addr)), ntohs(client_addr->sin_port));
celestar@5642: 
celestar@5642: 	NetworkRecv_NetworkGameInfo(&_udp_cs, p, &item->info);
celestar@5642: 
celestar@5642: 	item->info.compatible = true;
celestar@5642: 	{
celestar@5642: 		/* Checks whether there needs to be a request for names of GRFs and makes
celestar@5642: 		 * the request if necessary. GRFs that need to be requested are the GRFs
celestar@5642: 		 * that do not exist on the clients system and we do not have the name
celestar@5642: 		 * resolved of, i.e. the name is still UNKNOWN_GRF_NAME_PLACEHOLDER.
celestar@5642: 		 * The in_request array and in_request_count are used so there is no need
celestar@5642: 		 * to do a second loop over the GRF list, which can be relatively expensive
celestar@5642: 		 * due to the string comparisons. */
celestar@5642: 		const GRFConfig *in_request[NETWORK_MAX_GRF_COUNT];
celestar@5642: 		const GRFConfig *c;
celestar@5642: 		uint in_request_count = 0;
celestar@5642: 		struct sockaddr_in out_addr;
celestar@5642: 
celestar@5642: 		for (c = item->info.grfconfig; c != NULL; c = c->next) {
celestar@5642: 			if (HASBIT(c->flags, GCF_NOT_FOUND)) item->info.compatible = false;
celestar@5642: 			if (!HASBIT(c->flags, GCF_NOT_FOUND) || strcmp(c->name, UNKNOWN_GRF_NAME_PLACEHOLDER) != 0) continue;
celestar@5642: 			in_request[in_request_count] = c;
celestar@5642: 			in_request_count++;
celestar@5642: 		}
celestar@5642: 
celestar@5642: 		if (in_request_count > 0) {
celestar@5642: 			/* There are 'unknown' GRFs, now send a request for them */
celestar@5642: 			uint i;
celestar@5642: 			Packet *packet = NetworkSend_Init(PACKET_UDP_CLIENT_GET_NEWGRFS);
celestar@5642: 
celestar@5642: 			NetworkSend_uint8 (packet, in_request_count);
celestar@5642: 			for (i = 0; i < in_request_count; i++) {
celestar@5642: 				NetworkSend_GRFIdentifier(packet, in_request[i]);
celestar@5642: 			}
celestar@5642: 
celestar@5642: 			out_addr.sin_family      = AF_INET;
celestar@5642: 			out_addr.sin_port        = htons(item->port);
celestar@5642: 			out_addr.sin_addr.s_addr = item->ip;
celestar@5642: 			NetworkSendUDP_Packet(_udp_client_socket, packet, &out_addr);
celestar@5642: 			free(packet);
celestar@5642: 		}
celestar@5642: 	}
celestar@5642: 
celestar@5642: 	if (item->info.server_lang >= NETWORK_NUM_LANGUAGES) item->info.server_lang = 0;
celestar@5642: 	if (item->info.map_set >= NUM_LANDSCAPE ) item->info.map_set = 0;
celestar@5642: 
celestar@5642: 	if (item->info.hostname[0] == '\0')
celestar@5642: 		snprintf(item->info.hostname, sizeof(item->info.hostname), "%s", inet_ntoa(client_addr->sin_addr));
celestar@5642: 
celestar@5642: 	/* Check if we are allowed on this server based on the revision-match */
celestar@5642: 	item->info.version_compatible =
celestar@5642: 		strcmp(item->info.server_revision, _openttd_revision) == 0 ||
celestar@5642: 		strcmp(item->info.server_revision, NOREV_STRING) == 0;
celestar@5642: 	item->info.compatible &= item->info.version_compatible; // Already contains match for GRFs
celestar@5642: 
celestar@5642: 	item->online = true;
celestar@5642: 
celestar@5642: 	UpdateNetworkGameWindow(false);
celestar@5642: }
celestar@5642: 
celestar@5642: DEF_UDP_RECEIVE_COMMAND(PACKET_UDP_CLIENT_DETAIL_INFO)
celestar@5642: {
celestar@5642: 	NetworkClientState *cs;
celestar@5642: 	NetworkClientInfo *ci;
celestar@5642: 	Packet *packet;
celestar@5642: 	Player *player;
celestar@5642: 	byte current = 0;
celestar@5642: 	int i;
celestar@5642: 
celestar@5642: 	// Just a fail-safe.. should never happen
celestar@5642: 	if (!_network_udp_server) return;
celestar@5642: 
celestar@5642: 	packet = NetworkSend_Init(PACKET_UDP_SERVER_DETAIL_INFO);
celestar@5642: 
celestar@5642: 	/* Send the amount of active companies */
celestar@5642: 	NetworkSend_uint8 (packet, NETWORK_COMPANY_INFO_VERSION);
celestar@5642: 	NetworkSend_uint8 (packet, ActivePlayerCount());
celestar@5642: 
celestar@5642: 	/* Fetch the latest version of everything */
celestar@5642: 	NetworkPopulateCompanyInfo();
celestar@5642: 
celestar@5642: 	/* Go through all the players */
celestar@5642: 	FOR_ALL_PLAYERS(player) {
celestar@5642: 		/* Skip non-active players */
celestar@5642: 		if (!player->is_active) continue;
celestar@5642: 
celestar@5642: 		current++;
celestar@5642: 
celestar@5642: 		/* Send the information */
celestar@5642: 		NetworkSend_uint8(packet, current);
celestar@5642: 
celestar@5642: 		NetworkSend_string(packet, _network_player_info[player->index].company_name);
celestar@5642: 		NetworkSend_uint32(packet, _network_player_info[player->index].inaugurated_year);
celestar@5642: 		NetworkSend_uint64(packet, _network_player_info[player->index].company_value);
celestar@5642: 		NetworkSend_uint64(packet, _network_player_info[player->index].money);
celestar@5642: 		NetworkSend_uint64(packet, _network_player_info[player->index].income);
celestar@5642: 		NetworkSend_uint16(packet, _network_player_info[player->index].performance);
celestar@5642: 
celestar@5642: 		/* Send 1 if there is a passord for the company else send 0 */
celestar@5642: 		if (_network_player_info[player->index].password[0] != '\0') {
celestar@5642: 			NetworkSend_uint8(packet, 1);
celestar@5642: 		} else {
celestar@5642: 			NetworkSend_uint8(packet, 0);
celestar@5642: 		}
celestar@5642: 
celestar@5642: 		for (i = 0; i < NETWORK_VEHICLE_TYPES; i++)
celestar@5642: 			NetworkSend_uint16(packet, _network_player_info[player->index].num_vehicle[i]);
celestar@5642: 
celestar@5642: 		for (i = 0; i < NETWORK_STATION_TYPES; i++)
celestar@5642: 			NetworkSend_uint16(packet, _network_player_info[player->index].num_station[i]);
celestar@5642: 
celestar@5642: 		/* Find the clients that are connected to this player */
celestar@5642: 		FOR_ALL_CLIENTS(cs) {
celestar@5642: 			ci = DEREF_CLIENT_INFO(cs);
celestar@5642: 			if (ci->client_playas == player->index) {
celestar@5642: 				/* The uint8 == 1 indicates that a client is following */
celestar@5642: 				NetworkSend_uint8(packet, 1);
celestar@5642: 				NetworkSend_string(packet, ci->client_name);
celestar@5642: 				NetworkSend_string(packet, ci->unique_id);
celestar@5642: 				NetworkSend_uint32(packet, ci->join_date);
celestar@5642: 			}
celestar@5642: 		}
celestar@5642: 		/* Also check for the server itself */
celestar@5642: 		ci = NetworkFindClientInfoFromIndex(NETWORK_SERVER_INDEX);
celestar@5642: 		if (ci->client_playas == player->index) {
celestar@5642: 			/* The uint8 == 1 indicates that a client is following */
celestar@5642: 			NetworkSend_uint8(packet, 1);
celestar@5642: 			NetworkSend_string(packet, ci->client_name);
celestar@5642: 			NetworkSend_string(packet, ci->unique_id);
celestar@5642: 			NetworkSend_uint32(packet, ci->join_date);
celestar@5642: 		}
celestar@5642: 
celestar@5642: 		/* Indicates end of client list */
celestar@5642: 		NetworkSend_uint8(packet, 0);
celestar@5642: 	}
celestar@5642: 
celestar@5642: 	/* And check if we have any spectators */
celestar@5642: 	FOR_ALL_CLIENTS(cs) {
celestar@5642: 		ci = DEREF_CLIENT_INFO(cs);
celestar@5642: 		if (!IsValidPlayer(ci->client_playas)) {
celestar@5642: 			/* The uint8 == 1 indicates that a client is following */
celestar@5642: 			NetworkSend_uint8(packet, 1);
celestar@5642: 			NetworkSend_string(packet, ci->client_name);
celestar@5642: 			NetworkSend_string(packet, ci->unique_id);
celestar@5642: 			NetworkSend_uint32(packet, ci->join_date);
celestar@5642: 		}
celestar@5642: 	}
celestar@5642: 
celestar@5642: 	/* Also check for the server itself */
celestar@5642: 	ci = NetworkFindClientInfoFromIndex(NETWORK_SERVER_INDEX);
celestar@5642: 	if (!IsValidPlayer(ci->client_playas)) {
celestar@5642: 		/* The uint8 == 1 indicates that a client is following */
celestar@5642: 		NetworkSend_uint8(packet, 1);
celestar@5642: 		NetworkSend_string(packet, ci->client_name);
celestar@5642: 		NetworkSend_string(packet, ci->unique_id);
celestar@5642: 		NetworkSend_uint32(packet, ci->join_date);
celestar@5642: 	}
celestar@5642: 
celestar@5642: 	/* Indicates end of client list */
celestar@5642: 	NetworkSend_uint8(packet, 0);
celestar@5642: 
celestar@5642: 	NetworkSendUDP_Packet(_udp_server_socket, packet, client_addr);
celestar@5642: 
celestar@5642: 	free(packet);
celestar@5642: }
celestar@5642: 
celestar@5642: DEF_UDP_RECEIVE_COMMAND(PACKET_UDP_MASTER_RESPONSE_LIST)
celestar@5642: {
celestar@5642: 	int i;
celestar@5642: 	struct in_addr ip;
celestar@5642: 	uint16 port;
celestar@5642: 	uint8 ver;
celestar@5642: 
celestar@5642: 	/* packet begins with the protocol version (uint8)
celestar@5642: 	 * then an uint16 which indicates how many
celestar@5642: 	 * ip:port pairs are in this packet, after that
celestar@5642: 	 * an uint32 (ip) and an uint16 (port) for each pair
celestar@5642: 	 */
celestar@5642: 
celestar@5642: 	ver = NetworkRecv_uint8(&_udp_cs, p);
celestar@5642: 
celestar@5642: 	if (_udp_cs.has_quit) return;
celestar@5642: 
celestar@5642: 	if (ver == 1) {
celestar@5642: 		for (i = NetworkRecv_uint16(&_udp_cs, p); i != 0 ; i--) {
celestar@5642: 			ip.s_addr = TO_LE32(NetworkRecv_uint32(&_udp_cs, p));
celestar@5642: 			port = NetworkRecv_uint16(&_udp_cs, p);
celestar@5642: 			NetworkUDPQueryServer(inet_ntoa(ip), port);
celestar@5642: 		}
celestar@5642: 	}
celestar@5642: }
celestar@5642: 
celestar@5642: DEF_UDP_RECEIVE_COMMAND(PACKET_UDP_MASTER_ACK_REGISTER)
celestar@5642: {
celestar@5642: 	_network_advertise_retries = 0;
celestar@5642: 	DEBUG(net, 2, "[udp] advertising on master server successfull");
celestar@5642: 
celestar@5642: 	/* We are advertised, but we don't want to! */
celestar@5642: 	if (!_network_advertise) NetworkUDPRemoveAdvertise();
celestar@5642: }
celestar@5642: 
celestar@5642: /**
celestar@5642:  * A client has requested the names of some NewGRFs.
celestar@5642:  *
celestar@5642:  * Replying this can be tricky as we have a limit of SEND_MTU bytes
celestar@5642:  * in the reply packet and we can send up to 100 bytes per NewGRF
celestar@5642:  * (GRF ID, MD5sum and NETWORK_GRF_NAME_LENGTH bytes for the name).
celestar@5642:  * As SEND_MTU is _much_ less than 100 * NETWORK_MAX_GRF_COUNT, it
celestar@5642:  * could be that a packet overflows. To stop this we only reply
celestar@5642:  * with the first N NewGRFs so that if the first N + 1 NewGRFs
celestar@5642:  * would be sent, the packet overflows.
celestar@5642:  * in_reply and in_reply_count are used to keep a list of GRFs to
celestar@5642:  * send in the reply.
celestar@5642:  */
celestar@5642: DEF_UDP_RECEIVE_COMMAND(PACKET_UDP_CLIENT_GET_NEWGRFS)
celestar@5642: {
celestar@5642: 	uint8 num_grfs;
celestar@5642: 	uint i;
celestar@5642: 
celestar@5642: 	const GRFConfig *in_reply[NETWORK_MAX_GRF_COUNT];
celestar@5642: 	Packet *packet;
celestar@5642: 	uint8 in_reply_count = 0;
celestar@5642: 	uint packet_len = 0;
celestar@5642: 
celestar@5642: 	/* Just a fail-safe.. should never happen */
celestar@5642: 	if (_udp_cs.has_quit) return;
celestar@5642: 
celestar@5642: 	DEBUG(net, 6, "[udp] newgrf data request from %s:%d", inet_ntoa(client_addr->sin_addr), ntohs(client_addr->sin_port));
celestar@5642: 
celestar@5642: 	num_grfs = NetworkRecv_uint8 (&_udp_cs, p);
celestar@5642: 	if (num_grfs > NETWORK_MAX_GRF_COUNT) return;
celestar@5642: 
celestar@5642: 	for (i = 0; i < num_grfs; i++) {
celestar@5642: 		GRFConfig c;
celestar@5642: 		const GRFConfig *f;
celestar@5642: 
celestar@5642: 		NetworkRecv_GRFIdentifier(&_udp_cs, p, &c);
celestar@5642: 
celestar@5642: 		/* Find the matching GRF file */
celestar@5642: 		f = FindGRFConfig(c.grfid, c.md5sum);
celestar@5642: 		if (f == NULL) continue; // The GRF is unknown to this server
celestar@5642: 
celestar@5642: 		/* If the reply might exceed the size of the packet, only reply
celestar@5642: 		 * the current list and do not send the other data.
celestar@5642: 		 * The name could be an empty string, if so take the filename. */
celestar@5642: 		packet_len += sizeof(c.grfid) + sizeof(c.md5sum) +
celestar@5650: 				min(strlen((f->name != NULL && strlen(f->name) > 0) ? f->name : f->filename) + 1, (size_t)NETWORK_GRF_NAME_LENGTH);
celestar@5642: 		if (packet_len > SEND_MTU - 4) { // 4 is 3 byte header + grf count in reply
celestar@5642: 			break;
celestar@5642: 		}
celestar@5642: 		in_reply[in_reply_count] = f;
celestar@5642: 		in_reply_count++;
celestar@5642: 	}
celestar@5642: 
celestar@5642: 	if (in_reply_count == 0) return;
celestar@5642: 
celestar@5642: 	packet = NetworkSend_Init(PACKET_UDP_SERVER_NEWGRFS);
celestar@5642: 	NetworkSend_uint8 (packet, in_reply_count);
celestar@5642: 	for (i = 0; i < in_reply_count; i++) {
celestar@5642: 		char name[NETWORK_GRF_NAME_LENGTH];
celestar@5642: 
celestar@5642: 		/* The name could be an empty string, if so take the filename */
celestar@5642: 		ttd_strlcpy(name, (in_reply[i]->name != NULL && strlen(in_reply[i]->name) > 0) ?
celestar@5642: 				in_reply[i]->name : in_reply[i]->filename, sizeof(name));
celestar@5642: 	 	NetworkSend_GRFIdentifier(packet, in_reply[i]);
celestar@5642: 		NetworkSend_string(packet, name);
celestar@5642: 	}
celestar@5642: 
celestar@5642: 	NetworkSendUDP_Packet(_udp_server_socket, packet, client_addr);
celestar@5642: 	free(packet);
celestar@5642: }
celestar@5642: 
celestar@5642: /** The return of the client's request of the names of some NewGRFs */
celestar@5642: DEF_UDP_RECEIVE_COMMAND(PACKET_UDP_SERVER_NEWGRFS)
celestar@5642: {
celestar@5642: 	uint8 num_grfs;
celestar@5642: 	uint i;
celestar@5642: 
celestar@5642: 	/* Just a fail-safe.. should never happen */
celestar@5642: 	if (_udp_cs.has_quit) return;
celestar@5642: 
celestar@5642: 	DEBUG(net, 6, "[udp] newgrf data reply from %s:%d", inet_ntoa(client_addr->sin_addr),ntohs(client_addr->sin_port));
celestar@5642: 
celestar@5642: 	num_grfs = NetworkRecv_uint8 (&_udp_cs, p);
celestar@5642: 	if (num_grfs > NETWORK_MAX_GRF_COUNT) return;
celestar@5642: 
celestar@5642: 	for (i = 0; i < num_grfs; i++) {
celestar@5642: 		char *unknown_name;
celestar@5642: 		char name[NETWORK_GRF_NAME_LENGTH];
celestar@5642: 		GRFConfig c;
celestar@5642: 
celestar@5642: 		NetworkRecv_GRFIdentifier(&_udp_cs, p, &c);
celestar@5642: 		NetworkRecv_string(&_udp_cs, p, name, sizeof(name));
celestar@5642: 
celestar@5642: 		/* An empty name is not possible under normal circumstances
celestar@5642: 		 * and causes problems when showing the NewGRF list. */
celestar@5642: 		if (strlen(name) == 0) continue;
celestar@5642: 
celestar@5642: 		/* Finds the fake GRFConfig for the just read GRF ID and MD5sum tuple.
celestar@5642: 		 * If it exists and not resolved yet, then name of the fake GRF is
celestar@5642: 		 * overwritten with the name from the reply. */
celestar@5642: 		unknown_name = FindUnknownGRFName(c.grfid, c.md5sum, false);
celestar@5642: 		if (unknown_name != NULL && strcmp(unknown_name, UNKNOWN_GRF_NAME_PLACEHOLDER) == 0) {
celestar@5642: 			ttd_strlcpy(unknown_name, name, NETWORK_GRF_NAME_LENGTH);
celestar@5642: 		}
celestar@5642: 	}
celestar@5642: }
celestar@5642: 
celestar@5648: /**
celestar@5648:  * Every type of UDP packet should only be received by a single socket;
celestar@5648:  * The socket communicating with the masterserver should receive the
celestar@5648:  * game information of some 'random' host.
celestar@5648:  */
celestar@5648: typedef struct NetworkUDPPacketAndSocket {
celestar@5648: 	void (*callback)(Packet *p, const struct sockaddr_in *client_addr);
celestar@5648: 	SOCKET *incoming_socket;
celestar@5648: } NetworkUPDPacketAndSocket;
celestar@5642: 
celestar@5648: static const NetworkUPDPacketAndSocket _network_udp_packet[PACKET_UDP_END] = {
celestar@5648: 	{ RECEIVE_COMMAND(PACKET_UDP_CLIENT_FIND_SERVER),   &_udp_server_socket },
celestar@5648: 	{ RECEIVE_COMMAND(PACKET_UDP_SERVER_RESPONSE),      &_udp_client_socket },
celestar@5648: 	{ RECEIVE_COMMAND(PACKET_UDP_CLIENT_DETAIL_INFO),   &_udp_server_socket },
celestar@5648: 	{ NULL,                                             NULL                },
celestar@5648: 	{ NULL,                                             NULL                },
celestar@5648: 	{ RECEIVE_COMMAND(PACKET_UDP_MASTER_ACK_REGISTER),  &_udp_master_socket },
celestar@5648: 	{ NULL,                                             NULL                },
celestar@5648: 	{ RECEIVE_COMMAND(PACKET_UDP_MASTER_RESPONSE_LIST), &_udp_client_socket },
celestar@5648: 	{ NULL,                                             NULL                },
celestar@5648: 	{ RECEIVE_COMMAND(PACKET_UDP_CLIENT_GET_NEWGRFS),   &_udp_server_socket },
celestar@5648: 	{ RECEIVE_COMMAND(PACKET_UDP_SERVER_NEWGRFS),       &_udp_client_socket },
celestar@5642: };
celestar@5642: 
celestar@5648: void NetworkHandleUDPPacket(const SOCKET udp, Packet *p, const struct sockaddr_in *client_addr)
celestar@5642: {
celestar@5642: 	byte type;
celestar@5642: 
celestar@5642: 	/* Fake a client, so we can see when there is an illegal packet */
celestar@5642: 	_udp_cs.socket = INVALID_SOCKET;
celestar@5642: 	_udp_cs.has_quit = false;
celestar@5642: 
celestar@5642: 	type = NetworkRecv_uint8(&_udp_cs, p);
celestar@5642: 
celestar@5648: 	if (type < PACKET_UDP_END && *_network_udp_packet[type].incoming_socket == udp && !_udp_cs.has_quit) {
celestar@5648: 		_network_udp_packet[type].callback(p, client_addr);
celestar@5642: 	} else {
celestar@5648: 		if (*_network_udp_packet[type].incoming_socket != udp) {
celestar@5648: 			DEBUG(net, 0, "[udp] received packet on wrong port from %s:%d", inet_ntoa(client_addr->sin_addr),ntohs(client_addr->sin_port));
celestar@5648: 		} else if (!_udp_cs.has_quit) {
celestar@5648: 			DEBUG(net, 0, "[udp] received invalid packet type %d from %s:%d", type,  inet_ntoa(client_addr->sin_addr),ntohs(client_addr->sin_port));
celestar@5642: 		} else {
celestar@5648: 			DEBUG(net, 0, "[udp] received illegal packet from %s:%d", inet_ntoa(client_addr->sin_addr),ntohs(client_addr->sin_port));
celestar@5642: 		}
celestar@5642: 	}
celestar@5642: }
celestar@5642: 
celestar@5642: 
celestar@5642: // Close UDP connection
celestar@5648: void NetworkUDPStop(void)
celestar@5642: {
celestar@5642: 	DEBUG(net, 1, "[udp] closed listeners");
celestar@5642: 
celestar@5642: 	if (_network_udp_server) {
celestar@5648: 		NetworkUDPClose(&_udp_server_socket);
celestar@5648: 		NetworkUDPClose(&_udp_master_socket);
celestar@5648: 	} else {
celestar@5648: 		NetworkUDPClose(&_udp_client_socket);
celestar@5648: 	}
celestar@5642: 
celestar@5648: 	_network_udp_server = false;
celestar@5648: 	_network_udp_broadcast = 0;
celestar@5642: }
celestar@5642: 
celestar@5642: // Broadcast to all ips
celestar@5642: static void NetworkUDPBroadCast(SOCKET udp)
celestar@5642: {
celestar@5642: 	Packet* p = NetworkSend_Init(PACKET_UDP_CLIENT_FIND_SERVER);
celestar@5642: 	uint i;
celestar@5642: 
celestar@5642: 	for (i = 0; _broadcast_list[i] != 0; i++) {
celestar@5642: 		struct sockaddr_in out_addr;
celestar@5642: 
celestar@5642: 		out_addr.sin_family = AF_INET;
celestar@5642: 		out_addr.sin_port = htons(_network_server_port);
celestar@5642: 		out_addr.sin_addr.s_addr = _broadcast_list[i];
celestar@5642: 
celestar@5642: 		DEBUG(net, 4, "[udp] broadcasting to %s", inet_ntoa(out_addr.sin_addr));
celestar@5642: 
celestar@5642: 		NetworkSendUDP_Packet(udp, p, &out_addr);
celestar@5642: 	}
celestar@5642: 
celestar@5642: 	free(p);
celestar@5642: }
celestar@5642: 
celestar@5642: 
celestar@5642: // Request the the server-list from the master server
celestar@5642: void NetworkUDPQueryMasterServer(void)
celestar@5642: {
celestar@5642: 	struct sockaddr_in out_addr;
celestar@5642: 	Packet *p;
celestar@5642: 
celestar@5642: 	if (_udp_client_socket == INVALID_SOCKET)
celestar@5642: 		if (!NetworkUDPListen(&_udp_client_socket, 0, 0, true))
celestar@5642: 			return;
celestar@5642: 
celestar@5642: 	p = NetworkSend_Init(PACKET_UDP_CLIENT_GET_LIST);
celestar@5642: 
celestar@5642: 	out_addr.sin_family = AF_INET;
celestar@5642: 	out_addr.sin_port = htons(NETWORK_MASTER_SERVER_PORT);
celestar@5642: 	out_addr.sin_addr.s_addr = NetworkResolveHost(NETWORK_MASTER_SERVER_HOST);
celestar@5642: 
celestar@5642: 	// packet only contains protocol version
celestar@5642: 	NetworkSend_uint8(p, NETWORK_MASTER_SERVER_VERSION);
celestar@5642: 
celestar@5642: 	NetworkSendUDP_Packet(_udp_client_socket, p, &out_addr);
celestar@5642: 
celestar@5642: 	DEBUG(net, 2, "[udp] master server queried at %s:%d", inet_ntoa(out_addr.sin_addr),ntohs(out_addr.sin_port));
celestar@5642: 
celestar@5642: 	free(p);
celestar@5642: }
celestar@5642: 
celestar@5642: // Find all servers
celestar@5642: void NetworkUDPSearchGame(void)
celestar@5642: {
celestar@5642: 	// We are still searching..
celestar@5642: 	if (_network_udp_broadcast > 0) return;
celestar@5642: 
celestar@5642: 	// No UDP-socket yet..
celestar@5642: 	if (_udp_client_socket == INVALID_SOCKET)
celestar@5642: 		if (!NetworkUDPListen(&_udp_client_socket, 0, 0, true))
celestar@5642: 			return;
celestar@5642: 
celestar@5642: 	DEBUG(net, 0, "[udp] searching server");
celestar@5642: 
celestar@5642: 	NetworkUDPBroadCast(_udp_client_socket);
celestar@5642: 	_network_udp_broadcast = 300; // Stay searching for 300 ticks
celestar@5642: }
celestar@5642: 
celestar@5642: NetworkGameList *NetworkUDPQueryServer(const char* host, unsigned short port)
celestar@5642: {
celestar@5642: 	struct sockaddr_in out_addr;
celestar@5642: 	Packet *p;
celestar@5642: 	NetworkGameList *item;
celestar@5642: 
celestar@5642: 	// No UDP-socket yet..
celestar@5642: 	if (_udp_client_socket == INVALID_SOCKET)
celestar@5642: 		if (!NetworkUDPListen(&_udp_client_socket, 0, 0, true))
celestar@5642: 			return NULL;
celestar@5642: 
celestar@5642: 	out_addr.sin_family = AF_INET;
celestar@5642: 	out_addr.sin_port = htons(port);
celestar@5642: 	out_addr.sin_addr.s_addr = NetworkResolveHost(host);
celestar@5642: 
celestar@5642: 	// Clear item in gamelist
celestar@5642: 	item = NetworkGameListAddItem(inet_addr(inet_ntoa(out_addr.sin_addr)), ntohs(out_addr.sin_port));
celestar@5642: 	memset(&item->info, 0, sizeof(item->info));
celestar@5642: 	ttd_strlcpy(item->info.server_name, host, lengthof(item->info.server_name));
celestar@5642: 	ttd_strlcpy(item->info.hostname, host, lengthof(item->info.hostname));
celestar@5642: 	item->online = false;
celestar@5642: 
celestar@5642: 	// Init the packet
celestar@5642: 	p = NetworkSend_Init(PACKET_UDP_CLIENT_FIND_SERVER);
celestar@5642: 
celestar@5642: 	NetworkSendUDP_Packet(_udp_client_socket, p, &out_addr);
celestar@5642: 
celestar@5642: 	free(p);
celestar@5642: 
celestar@5642: 	UpdateNetworkGameWindow(false);
celestar@5642: 	return item;
celestar@5642: }
celestar@5642: 
celestar@5642: /* Remove our advertise from the master-server */
celestar@5642: void NetworkUDPRemoveAdvertise(void)
celestar@5642: {
celestar@5642: 	struct sockaddr_in out_addr;
celestar@5642: 	Packet *p;
celestar@5642: 
celestar@5642: 	/* Check if we are advertising */
celestar@5642: 	if (!_networking || !_network_server || !_network_udp_server) return;
celestar@5642: 
celestar@5642: 	/* check for socket */
celestar@5642: 	if (_udp_master_socket == INVALID_SOCKET)
celestar@5642: 		if (!NetworkUDPListen(&_udp_master_socket, _network_server_bind_ip, 0, false))
celestar@5642: 			return;
celestar@5642: 
celestar@5642: 	DEBUG(net, 1, "[udp] removing advertise from master server");
celestar@5642: 
celestar@5642: 	/* Find somewhere to send */
celestar@5642: 	out_addr.sin_family = AF_INET;
celestar@5642: 	out_addr.sin_port = htons(NETWORK_MASTER_SERVER_PORT);
celestar@5642: 	out_addr.sin_addr.s_addr = NetworkResolveHost(NETWORK_MASTER_SERVER_HOST);
celestar@5642: 
celestar@5642: 	/* Send the packet */
celestar@5642: 	p = NetworkSend_Init(PACKET_UDP_SERVER_UNREGISTER);
celestar@5642: 	/* Packet is: Version, server_port */
celestar@5642: 	NetworkSend_uint8(p, NETWORK_MASTER_SERVER_VERSION);
celestar@5642: 	NetworkSend_uint16(p, _network_server_port);
celestar@5642: 	NetworkSendUDP_Packet(_udp_master_socket, p, &out_addr);
celestar@5642: 
celestar@5642: 	free(p);
celestar@5642: }
celestar@5642: 
celestar@5642: /* Register us to the master server
celestar@5642:      This function checks if it needs to send an advertise */
celestar@5642: void NetworkUDPAdvertise(void)
celestar@5642: {
celestar@5642: 	struct sockaddr_in out_addr;
celestar@5642: 	Packet *p;
celestar@5642: 
celestar@5642: 	/* Check if we should send an advertise */
celestar@5642: 	if (!_networking || !_network_server || !_network_udp_server || !_network_advertise)
celestar@5642: 		return;
celestar@5642: 
celestar@5642: 	/* check for socket */
celestar@5642: 	if (_udp_master_socket == INVALID_SOCKET)
celestar@5642: 		if (!NetworkUDPListen(&_udp_master_socket, _network_server_bind_ip, 0, false))
celestar@5642: 			return;
celestar@5642: 
celestar@5642: 	if (_network_need_advertise) {
celestar@5642: 		_network_need_advertise = false;
celestar@5642: 		_network_advertise_retries = ADVERTISE_RETRY_TIMES;
celestar@5642: 	} else {
celestar@5642: 		/* Only send once every ADVERTISE_NORMAL_INTERVAL ticks */
celestar@5642: 		if (_network_advertise_retries == 0) {
celestar@5642: 			if ((_network_last_advertise_frame + ADVERTISE_NORMAL_INTERVAL) > _frame_counter)
celestar@5642: 				return;
celestar@5642: 			_network_advertise_retries = ADVERTISE_RETRY_TIMES;
celestar@5642: 		}
celestar@5642: 
celestar@5642: 		if ((_network_last_advertise_frame + ADVERTISE_RETRY_INTERVAL) > _frame_counter)
celestar@5642: 			return;
celestar@5642: 	}
celestar@5642: 
celestar@5642: 	_network_advertise_retries--;
celestar@5642: 	_network_last_advertise_frame = _frame_counter;
celestar@5642: 
celestar@5642: 	/* Find somewhere to send */
celestar@5642: 	out_addr.sin_family = AF_INET;
celestar@5642: 	out_addr.sin_port = htons(NETWORK_MASTER_SERVER_PORT);
celestar@5642: 	out_addr.sin_addr.s_addr = NetworkResolveHost(NETWORK_MASTER_SERVER_HOST);
celestar@5642: 
celestar@5642: 	DEBUG(net, 1, "[udp] advertising to master server");
celestar@5642: 
celestar@5642: 	/* Send the packet */
celestar@5642: 	p = NetworkSend_Init(PACKET_UDP_SERVER_REGISTER);
celestar@5642: 	/* Packet is: WELCOME_MESSAGE, Version, server_port */
celestar@5642: 	NetworkSend_string(p, NETWORK_MASTER_SERVER_WELCOME_MESSAGE);
celestar@5642: 	NetworkSend_uint8(p, NETWORK_MASTER_SERVER_VERSION);
celestar@5642: 	NetworkSend_uint16(p, _network_server_port);
celestar@5642: 	NetworkSendUDP_Packet(_udp_master_socket, p, &out_addr);
celestar@5642: 
celestar@5642: 	free(p);
celestar@5642: }
celestar@5642: 
celestar@5642: void NetworkUDPInitialize(void)
celestar@5642: {
celestar@5642: 	_udp_client_socket = INVALID_SOCKET;
celestar@5642: 	_udp_server_socket = INVALID_SOCKET;
celestar@5642: 	_udp_master_socket = INVALID_SOCKET;
celestar@5642: 
celestar@5642: 	_network_udp_server = false;
celestar@5642: 	_network_udp_broadcast = 0;
celestar@5642: }
celestar@5642: 
celestar@5642: #endif /* ENABLE_NETWORK */