tron@2186: /* $Id$ */
tron@2186: 
truelight@543: #include "stdafx.h"
tron@1299: #include "debug.h"
tron@1317: #include "string.h"
truelight@543: #include "network_data.h"
truelight@543: 
truelight@543: #ifdef ENABLE_NETWORK
truelight@543: 
tron@850: #include "map.h"
truelight@543: #include "network_gamelist.h"
truelight@764: #include "network_udp.h"
tron@2153: #include "variables.h"
truelight@543: 
truelight@543: extern void UpdateNetworkGameWindow(bool unselect);
truelight@638: extern void NetworkPopulateCompanyInfo(void);
truelight@543: 
truelight@543: //
truelight@543: // This file handles all the LAN-stuff
truelight@543: // Stuff like:
truelight@543: //   - UDP search over the network
truelight@543: //
truelight@543: 
truelight@543: typedef enum {
truelight@638: 	PACKET_UDP_CLIENT_FIND_SERVER,
truelight@543: 	PACKET_UDP_SERVER_RESPONSE,
truelight@638: 	PACKET_UDP_CLIENT_DETAIL_INFO,
truelight@638: 	PACKET_UDP_SERVER_DETAIL_INFO, // Is not used in OpenTTD itself, only for external querying
truelight@668: 	PACKET_UDP_SERVER_REGISTER, // Packet to register itself to the master server
truelight@764: 	PACKET_UDP_MASTER_ACK_REGISTER, // Packet indicating registration has succedeed
truelight@764: 	PACKET_UDP_CLIENT_GET_LIST, // Request for serverlist from master server
truelight@764: 	PACKET_UDP_MASTER_RESPONSE_LIST, // Response from master server with server ip's + port's
truelight@765: 	PACKET_UDP_SERVER_UNREGISTER, // Request to be removed from the server-list
truelight@543: 	PACKET_UDP_END
truelight@543: } PacketUDPType;
truelight@543: 
truelight@764: enum {
peter1138@2861: 	ADVERTISE_NORMAL_INTERVAL = 30000, // interval between advertising in ticks (15 minutes)
peter1138@2861: 	ADVERTISE_RETRY_INTERVAL = 300,    // readvertise when no response after this many ticks (9 seconds)
peter1138@2861: 	ADVERTISE_RETRY_TIMES = 3          // give up readvertising after this much failed retries
truelight@764: };
truelight@543: 
truelight@543: #define DEF_UDP_RECEIVE_COMMAND(type) void NetworkPacketReceive_ ## type ## _command(Packet *p, struct sockaddr_in *client_addr)
tron@2817: static void NetworkSendUDP_Packet(SOCKET udp, Packet* p, struct sockaddr_in* recv);
truelight@543: 
tron@2789: static NetworkClientState _udp_cs;
truelight@903: 
truelight@638: DEF_UDP_RECEIVE_COMMAND(PACKET_UDP_CLIENT_FIND_SERVER)
truelight@543: {
truelight@543: 	Packet *packet;
truelight@543: 	// Just a fail-safe.. should never happen
truelight@543: 	if (!_network_udp_server)
truelight@543: 		return;
truelight@543: 
truelight@543: 	packet = NetworkSend_Init(PACKET_UDP_SERVER_RESPONSE);
truelight@543: 
truelight@543: 	// Update some game_info
truelight@543: 	_network_game_info.game_date = _date;
tron@850: 	_network_game_info.map_width = MapSizeX();
tron@850: 	_network_game_info.map_height = MapSizeY();
truelight@543: 	_network_game_info.map_set = _opt.landscape;
truelight@543: 
truelight@543: 	NetworkSend_uint8 (packet, NETWORK_GAME_INFO_VERSION);
Darkvater@2879: 
Darkvater@2879: 	/* NETWORK_GAME_INFO_VERSION = 2 */
Darkvater@2879: 	NetworkSend_uint8 (packet, _network_game_info.companies_max);
Darkvater@2944: 	NetworkSend_uint8 (packet, ActivePlayerCount());
Darkvater@2879: 	NetworkSend_uint8 (packet, _network_game_info.spectators_max);
Darkvater@2879: 
Darkvater@2879: 	/* NETWORK_GAME_INFO_VERSION = 1 */
truelight@543: 	NetworkSend_string(packet, _network_game_info.server_name);
truelight@543: 	NetworkSend_string(packet, _network_game_info.server_revision);
truelight@543: 	NetworkSend_uint8 (packet, _network_game_info.server_lang);
truelight@543: 	NetworkSend_uint8 (packet, _network_game_info.use_password);
truelight@543: 	NetworkSend_uint8 (packet, _network_game_info.clients_max);
truelight@543: 	NetworkSend_uint8 (packet, _network_game_info.clients_on);
Darkvater@2944: 	NetworkSend_uint8 (packet, NetworkSpectatorCount());
truelight@543: 	NetworkSend_uint16(packet, _network_game_info.game_date);
truelight@543: 	NetworkSend_uint16(packet, _network_game_info.start_date);
truelight@543: 	NetworkSend_string(packet, _network_game_info.map_name);
truelight@543: 	NetworkSend_uint16(packet, _network_game_info.map_width);
truelight@543: 	NetworkSend_uint16(packet, _network_game_info.map_height);
truelight@543: 	NetworkSend_uint8 (packet, _network_game_info.map_set);
truelight@543: 	NetworkSend_uint8 (packet, _network_game_info.dedicated);
truelight@543: 
truelight@543: 	// Let the client know that we are here
truelight@764: 	NetworkSendUDP_Packet(_udp_server_socket, packet, client_addr);
truelight@543: 
truelight@543: 	free(packet);
truelight@543: 
truelight@543: 	DEBUG(net, 2)("[NET][UDP] Queried from %s", inet_ntoa(client_addr->sin_addr));
truelight@543: }
truelight@543: 
truelight@543: DEF_UDP_RECEIVE_COMMAND(PACKET_UDP_SERVER_RESPONSE)
truelight@543: {
Darkvater@2881: 	extern const char _openttd_revision[];
truelight@543: 	NetworkGameList *item;
truelight@543: 	byte game_info_version;
truelight@543: 
truelight@543: 	// Just a fail-safe.. should never happen
truelight@543: 	if (_network_udp_server)
truelight@543: 		return;
truelight@543: 
truelight@903: 	game_info_version = NetworkRecv_uint8(&_udp_cs, p);
truelight@543: 
Darkvater@2881: 	if (_udp_cs.quited) return;
truelight@764: 
truelight@764: 	DEBUG(net, 6)("[NET][UDP] Server response from %s:%d", inet_ntoa(client_addr->sin_addr),ntohs(client_addr->sin_port));
truelight@764: 
truelight@543: 	// Find next item
truelight@543: 	item = NetworkGameListAddItem(inet_addr(inet_ntoa(client_addr->sin_addr)), ntohs(client_addr->sin_port));
truelight@543: 
Darkvater@2881: 	/* Please observer the order. In the order in which packets are sent
Darkvater@2879: 	 * they are to be received */
Darkvater@2879: 	switch (game_info_version) {
Darkvater@2879: 		case 2:
Darkvater@2879: 			item->info.companies_max = NetworkRecv_uint8(&_udp_cs, p);
Darkvater@2879: 			item->info.companies_on = NetworkRecv_uint8(&_udp_cs, p);
Darkvater@2879: 			item->info.spectators_max = NetworkRecv_uint8(&_udp_cs, p);
Darkvater@2879: 			/* Fallthrough */
Darkvater@2879: 		case 1:
Darkvater@2879: 			NetworkRecv_string(&_udp_cs, p, item->info.server_name, sizeof(item->info.server_name));
Darkvater@2879: 			NetworkRecv_string(&_udp_cs, p, item->info.server_revision, sizeof(item->info.server_revision));
Darkvater@2879: 			item->info.server_lang   = NetworkRecv_uint8(&_udp_cs, p);
Darkvater@2879: 			item->info.use_password  = NetworkRecv_uint8(&_udp_cs, p);
Darkvater@2879: 			item->info.clients_max   = NetworkRecv_uint8(&_udp_cs, p);
Darkvater@2879: 			item->info.clients_on    = NetworkRecv_uint8(&_udp_cs, p);
Darkvater@2879: 			item->info.spectators_on = NetworkRecv_uint8(&_udp_cs, p);
Darkvater@2879: 			item->info.game_date     = NetworkRecv_uint16(&_udp_cs, p);
Darkvater@2879: 			item->info.start_date    = NetworkRecv_uint16(&_udp_cs, p);
Darkvater@2879: 			NetworkRecv_string(&_udp_cs, p, item->info.map_name, sizeof(item->info.map_name));
Darkvater@2879: 			item->info.map_width     = NetworkRecv_uint16(&_udp_cs, p);
Darkvater@2879: 			item->info.map_height    = NetworkRecv_uint16(&_udp_cs, p);
Darkvater@2879: 			item->info.map_set       = NetworkRecv_uint8(&_udp_cs, p);
Darkvater@2879: 			item->info.dedicated     = NetworkRecv_uint8(&_udp_cs, p);
truelight@543: 
Darkvater@2879: 			str_validate(item->info.server_name);
Darkvater@2879: 			str_validate(item->info.server_revision);
Darkvater@2879: 			str_validate(item->info.map_name);
Darkvater@2879: 			if (item->info.server_lang >= NETWORK_NUM_LANGUAGES) item->info.server_lang = 0;
Darkvater@2879: 			if (item->info.map_set >= NUM_LANDSCAPE ) item->info.map_set = 0;
Darkvater@2775: 
Darkvater@2879: 			if (item->info.hostname[0] == '\0')
Darkvater@2879: 				snprintf(item->info.hostname, sizeof(item->info.hostname), "%s", inet_ntoa(client_addr->sin_addr));
Darkvater@2881: 
Darkvater@2881: 			/* Check if we are allowed on this server based on the revision-match */
Darkvater@2881: 			item->info.compatible = (
Darkvater@2881: 			strncmp(item->info.server_revision, _openttd_revision, NETWORK_REVISION_LENGTH) == 0 ||
Darkvater@2881: 			strncmp(item->info.server_revision, NOREV_STRING, NETWORK_REVISION_LENGTH) == 0) ? true : false;
Darkvater@2881: 
Darkvater@2879: 			break;
truelight@543: 	}
truelight@543: 
truelight@543: 	item->online = true;
truelight@543: 
truelight@543: 	UpdateNetworkGameWindow(false);
truelight@543: }
truelight@543: 
truelight@638: DEF_UDP_RECEIVE_COMMAND(PACKET_UDP_CLIENT_DETAIL_INFO)
truelight@638: {
truelight@716: 	NetworkClientState *cs;
truelight@638: 	NetworkClientInfo *ci;
truelight@638: 	Packet *packet;
truelight@638: 	Player *player;
truelight@638: 	byte current = 0;
truelight@638: 	int i;
truelight@638: 
truelight@638: 	// Just a fail-safe.. should never happen
truelight@638: 	if (!_network_udp_server)
truelight@638: 		return;
truelight@638: 
truelight@638: 	packet = NetworkSend_Init(PACKET_UDP_SERVER_DETAIL_INFO);
truelight@638: 
truelight@638: 	/* Send the amount of active companies */
truelight@638: 	NetworkSend_uint8 (packet, NETWORK_COMPANY_INFO_VERSION);
Darkvater@2944: 	NetworkSend_uint8 (packet, ActivePlayerCount());
truelight@638: 
truelight@638: 	/* Fetch the latest version of everything */
truelight@638: 	NetworkPopulateCompanyInfo();
truelight@638: 
truelight@638: 	/* Go through all the players */
truelight@638: 	FOR_ALL_PLAYERS(player) {
truelight@638: 		/* Skip non-active players */
truelight@638: 		if (!player->is_active)
truelight@638: 			continue;
truelight@638: 
truelight@638: 		current++;
truelight@638: 
truelight@638: 		/* Send the information */
truelight@638: 		NetworkSend_uint8 (packet, current);
truelight@638: 
truelight@638: 		NetworkSend_string(packet, _network_player_info[player->index].company_name);
truelight@638: 		NetworkSend_uint8 (packet, _network_player_info[player->index].inaugurated_year);
truelight@638: 		NetworkSend_uint64(packet, _network_player_info[player->index].company_value);
truelight@638: 		NetworkSend_uint64(packet, _network_player_info[player->index].money);
truelight@638: 		NetworkSend_uint64(packet, _network_player_info[player->index].income);
truelight@638: 		NetworkSend_uint16(packet, _network_player_info[player->index].performance);
truelight@638: 
truelight@1011:                 /* Send 1 if there is a passord for the company else send 0 */
truelight@1011: 		if (_network_player_info[player->index].password[0] != '\0') {
truelight@1186: 			NetworkSend_uint8 (packet, 1);
truelight@1011: 		} else {
truelight@1186: 			NetworkSend_uint8 (packet, 0);
tron@1019: 		}
tron@1019: 
truelight@638: 		for (i = 0; i < NETWORK_VEHICLE_TYPES; i++)
truelight@638: 			NetworkSend_uint16(packet, _network_player_info[player->index].num_vehicle[i]);
truelight@638: 
truelight@638: 		for (i = 0; i < NETWORK_STATION_TYPES; i++)
truelight@638: 			NetworkSend_uint16(packet, _network_player_info[player->index].num_station[i]);
truelight@638: 
truelight@638: 		/* Find the clients that are connected to this player */
truelight@638: 		FOR_ALL_CLIENTS(cs) {
truelight@638: 			ci = DEREF_CLIENT_INFO(cs);
truelight@638: 			if ((ci->client_playas - 1) == player->index) {
truelight@638: 				/* The uint8 == 1 indicates that a client is following */
truelight@638: 				NetworkSend_uint8(packet, 1);
truelight@638: 				NetworkSend_string(packet, ci->client_name);
truelight@638: 				NetworkSend_string(packet, ci->unique_id);
truelight@638: 				NetworkSend_uint16(packet, ci->join_date);
truelight@638: 			}
truelight@638: 		}
truelight@638: 		/* Also check for the server itself */
truelight@638: 		ci = NetworkFindClientInfoFromIndex(NETWORK_SERVER_INDEX);
truelight@638: 		if ((ci->client_playas - 1) == player->index) {
truelight@638: 			/* The uint8 == 1 indicates that a client is following */
truelight@638: 			NetworkSend_uint8(packet, 1);
truelight@638: 			NetworkSend_string(packet, ci->client_name);
truelight@638: 			NetworkSend_string(packet, ci->unique_id);
truelight@638: 			NetworkSend_uint16(packet, ci->join_date);
truelight@638: 		}
truelight@638: 
truelight@638: 		/* Indicates end of client list */
truelight@638: 		NetworkSend_uint8(packet, 0);
truelight@638: 	}
truelight@638: 
truelight@638: 	/* And check if we have any spectators */
truelight@638: 	FOR_ALL_CLIENTS(cs) {
truelight@638: 		ci = DEREF_CLIENT_INFO(cs);
truelight@638: 		if ((ci->client_playas - 1) > MAX_PLAYERS) {
truelight@638: 			/* The uint8 == 1 indicates that a client is following */
truelight@638: 			NetworkSend_uint8(packet, 1);
truelight@638: 			NetworkSend_string(packet, ci->client_name);
truelight@638: 			NetworkSend_string(packet, ci->unique_id);
truelight@638: 			NetworkSend_uint16(packet, ci->join_date);
truelight@638: 		}
truelight@638: 	}
truelight@638: 	/* Also check for the server itself */
truelight@638: 	ci = NetworkFindClientInfoFromIndex(NETWORK_SERVER_INDEX);
truelight@638: 	if ((ci->client_playas - 1) > MAX_PLAYERS) {
truelight@638: 		/* The uint8 == 1 indicates that a client is following */
truelight@638: 		NetworkSend_uint8(packet, 1);
truelight@638: 		NetworkSend_string(packet, ci->client_name);
truelight@638: 		NetworkSend_string(packet, ci->unique_id);
truelight@638: 		NetworkSend_uint16(packet, ci->join_date);
truelight@638: 	}
truelight@638: 
truelight@638: 	/* Indicates end of client list */
truelight@638: 	NetworkSend_uint8(packet, 0);
truelight@638: 
truelight@764: 	NetworkSendUDP_Packet(_udp_server_socket, packet, client_addr);
truelight@638: 
truelight@638: 	free(packet);
truelight@638: }
truelight@638: 
truelight@764: DEF_UDP_RECEIVE_COMMAND(PACKET_UDP_MASTER_RESPONSE_LIST) {
truelight@764: 	int i;
truelight@764: 	struct in_addr ip;
truelight@764: 	uint16 port;
truelight@764: 	uint8 ver;
truelight@764: 
truelight@764: 	/* packet begins with the protocol version (uint8)
truelight@764: 	 * then an uint16 which indicates how many
truelight@764: 	 * ip:port pairs are in this packet, after that
truelight@764: 	 * an uint32 (ip) and an uint16 (port) for each pair
truelight@764: 	 */
truelight@764: 
truelight@903: 	ver = NetworkRecv_uint8(&_udp_cs, p);
truelight@903: 
truelight@903: 	if (_udp_cs.quited)
truelight@903: 		return;
truelight@764: 
truelight@764: 	if (ver == 1) {
truelight@903: 		for (i = NetworkRecv_uint16(&_udp_cs, p); i != 0 ; i--) {
truelight@903: 			ip.s_addr = TO_LE32(NetworkRecv_uint32(&_udp_cs, p));
truelight@903: 			port = NetworkRecv_uint16(&_udp_cs, p);
truelight@764: 			NetworkUDPQueryServer(inet_ntoa(ip), port);
truelight@764: 		}
truelight@764: 	}
truelight@764: }
truelight@764: 
truelight@764: DEF_UDP_RECEIVE_COMMAND(PACKET_UDP_MASTER_ACK_REGISTER) {
truelight@764: 	_network_advertise_retries = 0;
truelight@765: 	DEBUG(net, 2)("[NET][UDP] We are advertised on the master-server!");
truelight@765: 
truelight@765: 	if (!_network_advertise)
truelight@765: 		/* We are advertised, but we don't want to! */
truelight@765: 		NetworkUDPRemoveAdvertise();
truelight@764: }
truelight@764: 
truelight@543: 
truelight@543: // The layout for the receive-functions by UDP
truelight@543: typedef void NetworkUDPPacket(Packet *p, struct sockaddr_in *client_addr);
truelight@543: 
truelight@543: static NetworkUDPPacket* const _network_udp_packet[] = {
truelight@638: 	RECEIVE_COMMAND(PACKET_UDP_CLIENT_FIND_SERVER),
truelight@543: 	RECEIVE_COMMAND(PACKET_UDP_SERVER_RESPONSE),
truelight@638: 	RECEIVE_COMMAND(PACKET_UDP_CLIENT_DETAIL_INFO),
truelight@638: 	NULL,
truelight@668: 	NULL,
truelight@764: 	RECEIVE_COMMAND(PACKET_UDP_MASTER_ACK_REGISTER),
truelight@764: 	NULL,
truelight@765: 	RECEIVE_COMMAND(PACKET_UDP_MASTER_RESPONSE_LIST),
truelight@765: 	NULL
truelight@543: };
truelight@543: 
truelight@764: 
truelight@543: // If this fails, check the array above with network_data.h
truelight@543: assert_compile(lengthof(_network_udp_packet) == PACKET_UDP_END);
truelight@543: 
truelight@543: 
tron@2817: static void NetworkHandleUDPPacket(Packet* p, struct sockaddr_in* client_addr)
truelight@543: {
truelight@543: 	byte type;
truelight@543: 
truelight@903: 	/* Fake a client, so we can see when there is an illegal packet */
truelight@903: 	_udp_cs.socket = INVALID_SOCKET;
truelight@903: 	_udp_cs.quited = false;
truelight@543: 
truelight@903: 	type = NetworkRecv_uint8(&_udp_cs, p);
truelight@903: 
truelight@903: 	if (type < PACKET_UDP_END && _network_udp_packet[type] != NULL && !_udp_cs.quited) {
truelight@543: 		_network_udp_packet[type](p, client_addr);
truelight@543: 	}	else {
truelight@543: 		DEBUG(net, 0)("[NET][UDP] Received invalid packet type %d", type);
truelight@543: 	}
truelight@543: }
truelight@543: 
truelight@543: 
truelight@543: // Send a packet over UDP
tron@2817: static void NetworkSendUDP_Packet(SOCKET udp, Packet* p, struct sockaddr_in* recv)
truelight@543: {
truelight@543: 	int res;
truelight@543: 
truelight@543: 	// Put the length in the buffer
truelight@543: 	p->buffer[0] = p->size & 0xFF;
truelight@543: 	p->buffer[1] = p->size >> 8;
truelight@543: 
truelight@543: 	// Send the buffer
truelight@543: 	res = sendto(udp, p->buffer, p->size, 0, (struct sockaddr *)recv, sizeof(*recv));
truelight@543: 
truelight@543: 	// Check for any errors, but ignore it for the rest
truelight@543: 	if (res == -1) {
truelight@543: 		DEBUG(net, 1)("[NET][UDP] Send error: %i", GET_LAST_ERROR());
truelight@543: 	}
truelight@543: }
truelight@543: 
truelight@543: // Start UDP listener
truelight@764: bool NetworkUDPListen(SOCKET *udp, uint32 host, uint16 port, bool broadcast)
truelight@543: {
truelight@543: 	struct sockaddr_in sin;
truelight@543: 
truelight@764: 	// Make sure socket is closed
truelight@764: 	closesocket(*udp);
truelight@543: 
truelight@764: 	*udp = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
truelight@764: 	if (*udp == INVALID_SOCKET) {
truelight@543: 		DEBUG(net, 1)("[NET][UDP] Failed to start UDP support");
truelight@543: 		return false;
truelight@543: 	}
truelight@543: 
truelight@543: 	// set nonblocking mode for socket
truelight@543: 	{
truelight@543: 		unsigned long blocking = 1;
bjarni@2497: #ifndef BEOS_NET_SERVER
truelight@764: 		ioctlsocket(*udp, FIONBIO, &blocking);
bjarni@2497: #else
truelight@2524: 		setsockopt(*udp, SOL_SOCKET, SO_NONBLOCK, &blocking, NULL);
bjarni@2497: #endif
truelight@543: 	}
truelight@543: 
truelight@543: 	sin.sin_family = AF_INET;
truelight@543: 	// Listen on all IPs
truelight@543: 	sin.sin_addr.s_addr = host;
truelight@543: 	sin.sin_port = htons(port);
truelight@543: 
truelight@764: 	if (bind(*udp, (struct sockaddr*)&sin, sizeof(sin)) != 0) {
truelight@629: 		DEBUG(net, 1) ("[NET][UDP] error: bind failed on %s:%i", inet_ntoa(*(struct in_addr *)&host), port);
truelight@543: 		return false;
truelight@543: 	}
truelight@543: 
truelight@764: 	if (broadcast) {
truelight@764: 		/* Enable broadcast */
truelight@543: 		unsigned long val = 1;
bjarni@2497: #ifndef BEOS_NET_SERVER // will work around this, some day; maybe.
truelight@764: 		setsockopt(*udp, SOL_SOCKET, SO_BROADCAST, (char *) &val , sizeof(val));
bjarni@2497: #endif
truelight@543: 	}
truelight@543: 
truelight@629: 	DEBUG(net, 1)("[NET][UDP] Listening on port %s:%d", inet_ntoa(*(struct in_addr *)&host), port);
truelight@543: 
truelight@543: 	return true;
truelight@543: }
truelight@543: 
truelight@543: // Close UDP connection
truelight@543: void NetworkUDPClose(void)
truelight@543: {
truelight@764: 	DEBUG(net, 1) ("[NET][UDP] Closed listeners");
truelight@543: 
truelight@543: 	if (_network_udp_server) {
truelight@985: 		if (_udp_server_socket != INVALID_SOCKET) {
truelight@985: 			closesocket(_udp_server_socket);
truelight@985: 			_udp_server_socket = INVALID_SOCKET;
truelight@985: 		}
truelight@764: 
truelight@985: 		if (_udp_master_socket != INVALID_SOCKET) {
truelight@985: 			closesocket(_udp_master_socket);
truelight@985: 			_udp_master_socket = INVALID_SOCKET;
truelight@985: 		}
truelight@764: 
truelight@543: 		_network_udp_server = false;
truelight@543: 		_network_udp_broadcast = 0;
truelight@543: 	} else {
truelight@985: 		if (_udp_client_socket != INVALID_SOCKET) {
truelight@985: 			closesocket(_udp_client_socket);
truelight@985: 			_udp_client_socket = INVALID_SOCKET;
truelight@985: 		}
truelight@543: 		_network_udp_broadcast = 0;
truelight@543: 	}
truelight@543: }
truelight@543: 
truelight@543: // Receive something on UDP level
truelight@764: void NetworkUDPReceive(SOCKET udp)
truelight@543: {
truelight@543: 	struct sockaddr_in client_addr;
tron@1343: 	socklen_t client_len;
truelight@543: 	int nbytes;
truelight@543: 	static Packet *p = NULL;
truelight@543: 	int packet_len;
truelight@543: 
truelight@543: 	// If p is NULL, malloc him.. this prevents unneeded mallocs
truelight@543: 	if (p == NULL)
truelight@543: 		p = malloc(sizeof(Packet));
truelight@543: 
truelight@543: 	packet_len = sizeof(p->buffer);
truelight@543: 	client_len = sizeof(client_addr);
truelight@543: 
truelight@543: 	// Try to receive anything
truelight@543: 	nbytes = recvfrom(udp, p->buffer, packet_len, 0, (struct sockaddr *)&client_addr, &client_len);
truelight@543: 
truelight@543: 	// We got some bytes.. just asume we receive the whole packet
truelight@543: 	if (nbytes > 0) {
truelight@543: 		// Get the size of the buffer
truelight@543: 		p->size = (uint16)p->buffer[0];
truelight@543: 		p->size += (uint16)p->buffer[1] << 8;
truelight@543: 		// Put the position on the right place
truelight@543: 		p->pos = 2;
truelight@543: 		p->next = NULL;
truelight@543: 
truelight@543: 		// Handle the packet
truelight@543: 		NetworkHandleUDPPacket(p, &client_addr);
truelight@543: 
truelight@543: 		// Free the packet
truelight@543: 		free(p);
truelight@543: 		p = NULL;
truelight@543: 	}
truelight@543: }
truelight@543: 
truelight@543: // Broadcast to all ips
tron@2817: static void NetworkUDPBroadCast(SOCKET udp)
truelight@543: {
truelight@543: 	int i;
truelight@543: 	struct sockaddr_in out_addr;
truelight@543: 	byte *bcptr;
truelight@543: 	uint32 bcaddr;
truelight@543: 	Packet *p;
truelight@543: 
truelight@543: 	// Init the packet
truelight@638: 	p = NetworkSend_Init(PACKET_UDP_CLIENT_FIND_SERVER);
truelight@543: 
truelight@543: 	// Go through all the ips on this pc
truelight@543: 	i = 0;
truelight@543: 	while (_network_ip_list[i] != 0) {
truelight@543: 		bcaddr = _network_ip_list[i];
truelight@543: 		bcptr = (byte *)&bcaddr;
truelight@543: 		// Make the address a broadcast address
truelight@543: 		bcptr[3] = 255;
truelight@543: 
truelight@543: 		DEBUG(net, 6)("[NET][UDP] Broadcasting to %s", inet_ntoa(*(struct in_addr *)&bcaddr));
truelight@543: 
truelight@543: 		out_addr.sin_family = AF_INET;
truelight@543: 		out_addr.sin_port = htons(_network_server_port);
truelight@543: 		out_addr.sin_addr.s_addr = bcaddr;
truelight@543: 
truelight@764: 		NetworkSendUDP_Packet(udp, p, &out_addr);
truelight@543: 
truelight@543: 		i++;
truelight@543: 	}
truelight@764: }
truelight@764: 
truelight@764: 
truelight@764: // Request the the server-list from the master server
truelight@764: void NetworkUDPQueryMasterServer(void)
truelight@764: {
truelight@764: 	struct sockaddr_in out_addr;
truelight@764: 	Packet *p;
truelight@764: 
truelight@764: 	if (_udp_client_socket == INVALID_SOCKET)
truelight@764: 		if (!NetworkUDPListen(&_udp_client_socket, 0, 0, true))
truelight@764: 			return;
truelight@764: 
truelight@764: 	p = NetworkSend_Init(PACKET_UDP_CLIENT_GET_LIST);
truelight@764: 
truelight@764: 	out_addr.sin_family = AF_INET;
truelight@764: 	out_addr.sin_port = htons(NETWORK_MASTER_SERVER_PORT);
truelight@764: 	out_addr.sin_addr.s_addr = NetworkResolveHost(NETWORK_MASTER_SERVER_HOST);
truelight@764: 
truelight@764: 	// packet only contains protocol version
truelight@764: 	NetworkSend_uint8(p, NETWORK_MASTER_SERVER_VERSION);
truelight@764: 
truelight@764: 	NetworkSendUDP_Packet(_udp_client_socket, p, &out_addr);
truelight@764: 
truelight@764: 	DEBUG(net, 2)("[NET][UDP] Queried Master Server at %s:%d", inet_ntoa(out_addr.sin_addr),ntohs(out_addr.sin_port));
truelight@543: 
truelight@543: 	free(p);
truelight@543: }
truelight@543: 
truelight@543: // Find all servers
truelight@543: void NetworkUDPSearchGame(void)
truelight@543: {
truelight@543: 	// We are still searching..
truelight@543: 	if (_network_udp_broadcast > 0)
truelight@543: 		return;
truelight@543: 
truelight@543: 	// No UDP-socket yet..
truelight@543: 	if (_udp_client_socket == INVALID_SOCKET)
truelight@764: 		if (!NetworkUDPListen(&_udp_client_socket, 0, 0, true))
truelight@543: 			return;
truelight@543: 
truelight@543: 	DEBUG(net, 0)("[NET][UDP] Searching server");
truelight@543: 
truelight@764: 	NetworkUDPBroadCast(_udp_client_socket);
truelight@543: 	_network_udp_broadcast = 300; // Stay searching for 300 ticks
truelight@543: }
truelight@543: 
tron@1329: NetworkGameList *NetworkUDPQueryServer(const char* host, unsigned short port)
truelight@543: {
truelight@543: 	struct sockaddr_in out_addr;
truelight@543: 	Packet *p;
truelight@543: 	NetworkGameList *item;
truelight@543: 	char hostname[NETWORK_HOSTNAME_LENGTH];
truelight@543: 
truelight@543: 	// No UDP-socket yet..
truelight@543: 	if (_udp_client_socket == INVALID_SOCKET)
truelight@764: 		if (!NetworkUDPListen(&_udp_client_socket, 0, 0, true))
dominik@738: 			return NULL;
truelight@543: 
truelight@543: 	ttd_strlcpy(hostname, host, sizeof(hostname));
truelight@543: 
truelight@543: 	out_addr.sin_family = AF_INET;
truelight@543: 	out_addr.sin_port = htons(port);
truelight@543: 	out_addr.sin_addr.s_addr = NetworkResolveHost(host);
truelight@543: 
truelight@543: 	// Clear item in gamelist
truelight@543: 	item = NetworkGameListAddItem(inet_addr(inet_ntoa(out_addr.sin_addr)), ntohs(out_addr.sin_port));
truelight@543: 	memset(&item->info, 0, sizeof(item->info));
truelight@543: 	snprintf(item->info.server_name, sizeof(item->info.server_name), "%s", hostname);
truelight@543: 	snprintf(item->info.hostname, sizeof(item->info.hostname), "%s", hostname);
truelight@543: 	item->online = false;
truelight@543: 
truelight@543: 	// Init the packet
truelight@638: 	p = NetworkSend_Init(PACKET_UDP_CLIENT_FIND_SERVER);
truelight@543: 
truelight@764: 	NetworkSendUDP_Packet(_udp_client_socket, p, &out_addr);
truelight@543: 
truelight@543: 	free(p);
truelight@543: 
truelight@543: 	UpdateNetworkGameWindow(false);
dominik@738: 	return item;
truelight@543: }
truelight@543: 
truelight@765: /* Remove our advertise from the master-server */
truelight@765: void NetworkUDPRemoveAdvertise(void)
truelight@765: {
truelight@765: 	struct sockaddr_in out_addr;
truelight@765: 	Packet *p;
truelight@765: 
truelight@765: 	/* Check if we are advertising */
truelight@1832: 	if (!_networking || !_network_server || !_network_udp_server)
truelight@765: 		return;
truelight@765: 
truelight@765: 	/* check for socket */
truelight@765: 	if (_udp_master_socket == INVALID_SOCKET)
truelight@805: 		if (!NetworkUDPListen(&_udp_master_socket, _network_server_bind_ip, 0, false))
truelight@765: 			return;
truelight@765: 
truelight@765: 	DEBUG(net, 2)("[NET][UDP] Removing advertise..");
truelight@765: 
truelight@765: 	/* Find somewhere to send */
truelight@765: 	out_addr.sin_family = AF_INET;
truelight@765: 	out_addr.sin_port = htons(NETWORK_MASTER_SERVER_PORT);
truelight@765: 	out_addr.sin_addr.s_addr = NetworkResolveHost(NETWORK_MASTER_SERVER_HOST);
truelight@765: 
truelight@765: 	/* Send the packet */
truelight@765: 	p = NetworkSend_Init(PACKET_UDP_SERVER_UNREGISTER);
truelight@765: 	/* Packet is: Version, server_port */
truelight@765: 	NetworkSend_uint8(p, NETWORK_MASTER_SERVER_VERSION);
truelight@765: 	NetworkSend_uint16(p, _network_server_port);
truelight@765: 	NetworkSendUDP_Packet(_udp_master_socket, p, &out_addr);
truelight@765: 
truelight@765: 	free(p);
truelight@765: }
truelight@765: 
truelight@668: /* Register us to the master server
truelight@668:      This function checks if it needs to send an advertise */
truelight@716: void NetworkUDPAdvertise(void)
truelight@668: {
truelight@668: 	struct sockaddr_in out_addr;
truelight@668: 	Packet *p;
truelight@668: 
truelight@668: 	/* Check if we should send an advertise */
truelight@668: 	if (!_networking || !_network_server || !_network_udp_server || !_network_advertise)
truelight@668: 		return;
truelight@668: 
truelight@764: 	/* check for socket */
truelight@764: 	if (_udp_master_socket == INVALID_SOCKET)
truelight@805: 		if (!NetworkUDPListen(&_udp_master_socket, _network_server_bind_ip, 0, false))
truelight@764: 			return;
truelight@764: 
peter1138@2861: 	if (_network_need_advertise) {
peter1138@2861: 		_network_need_advertise = false;
peter1138@2861: 		_network_advertise_retries = ADVERTISE_RETRY_TIMES;
peter1138@2861: 	} else {
peter1138@2861: 		/* Only send once every ADVERTISE_NORMAL_INTERVAL ticks */
peter1138@2861: 		if (_network_advertise_retries == 0) {
peter1138@2861: 			if ((_network_last_advertise_frame + ADVERTISE_NORMAL_INTERVAL) > _frame_counter)
peter1138@2861: 				return;
peter1138@2861: 			_network_advertise_retries = ADVERTISE_RETRY_TIMES;
peter1138@2861: 		}
peter1138@2861: 
peter1138@2861: 		if ((_network_last_advertise_frame + ADVERTISE_RETRY_INTERVAL) > _frame_counter)
truelight@764: 			return;
truelight@764: 	}
truelight@764: 
truelight@764: 	_network_advertise_retries--;
peter1138@2861: 	_network_last_advertise_frame = _frame_counter;
truelight@668: 
truelight@668: 	/* Find somewhere to send */
truelight@668: 	out_addr.sin_family = AF_INET;
truelight@668: 	out_addr.sin_port = htons(NETWORK_MASTER_SERVER_PORT);
truelight@668: 	out_addr.sin_addr.s_addr = NetworkResolveHost(NETWORK_MASTER_SERVER_HOST);
truelight@668: 
truelight@668: 	DEBUG(net, 1)("[NET][UDP] Advertising to master server");
truelight@668: 
truelight@668: 	/* Send the packet */
truelight@668: 	p = NetworkSend_Init(PACKET_UDP_SERVER_REGISTER);
truelight@668: 	/* Packet is: WELCOME_MESSAGE, Version, server_port */
truelight@668: 	NetworkSend_string(p, NETWORK_MASTER_SERVER_WELCOME_MESSAGE);
truelight@668: 	NetworkSend_uint8(p, NETWORK_MASTER_SERVER_VERSION);
truelight@668: 	NetworkSend_uint16(p, _network_server_port);
truelight@764: 	NetworkSendUDP_Packet(_udp_master_socket, p, &out_addr);
truelight@764: 
truelight@668: 	free(p);
truelight@668: }
truelight@668: 
truelight@543: void NetworkUDPInitialize(void)
truelight@543: {
truelight@543: 	_udp_client_socket = INVALID_SOCKET;
truelight@543: 	_udp_server_socket = INVALID_SOCKET;
truelight@764: 	_udp_master_socket = INVALID_SOCKET;
truelight@543: 
truelight@543: 	_network_udp_server = false;
truelight@543: 	_network_udp_broadcast = 0;
truelight@543: }
truelight@543: 
truelight@543: #endif /* ENABLE_NETWORK */