truelight@543: #include "stdafx.h"
truelight@543: #include "network_data.h"
truelight@543: 
truelight@543: #ifdef ENABLE_NETWORK
truelight@543: 
truelight@543: #include "network_gamelist.h"
truelight@543: 
truelight@543: extern void UpdateNetworkGameWindow(bool unselect);
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@543: 	PACKET_UDP_FIND_SERVER,
truelight@543: 	PACKET_UDP_SERVER_RESPONSE,
truelight@543: 	PACKET_UDP_END
truelight@543: } PacketUDPType;
truelight@543: 
truelight@543: static SOCKET _udp_server_socket; // udp server socket
truelight@543: 
truelight@543: #define DEF_UDP_RECEIVE_COMMAND(type) void NetworkPacketReceive_ ## type ## _command(Packet *p, struct sockaddr_in *client_addr)
truelight@543: void NetworkSendUDP_Packet(Packet *p, struct sockaddr_in *recv);
truelight@543: 
truelight@543: DEF_UDP_RECEIVE_COMMAND(PACKET_UDP_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;
truelight@543: 	_network_game_info.map_set = _opt.landscape;
truelight@543: 
truelight@543: 	NetworkSend_uint8 (packet, NETWORK_GAME_INFO_VERSION);
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);
truelight@543: 	NetworkSend_uint8 (packet, _network_game_info.spectators_on);
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@543: 	NetworkSendUDP_Packet(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: {
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@543: 	game_info_version = NetworkRecv_uint8(p);
truelight@543: 
truelight@543: 	// Find next item
truelight@543: 	item = NetworkGameListAddItem(inet_addr(inet_ntoa(client_addr->sin_addr)), ntohs(client_addr->sin_port));
truelight@543: 
truelight@543: 	if (game_info_version == 1) {
truelight@543: 		NetworkRecv_string(p, item->info.server_name, sizeof(item->info.server_name));
truelight@543: 		NetworkRecv_string(p, item->info.server_revision, sizeof(item->info.server_revision));
truelight@543: 		item->info.server_lang = NetworkRecv_uint8(p);
truelight@543: 		item->info.use_password = NetworkRecv_uint8(p);
truelight@543: 		item->info.clients_max = NetworkRecv_uint8(p);
truelight@543: 		item->info.clients_on = NetworkRecv_uint8(p);
truelight@543: 		item->info.spectators_on = NetworkRecv_uint8(p);
truelight@543: 		item->info.game_date = NetworkRecv_uint16(p);
truelight@543: 		item->info.start_date = NetworkRecv_uint16(p);
truelight@543: 		NetworkRecv_string(p, item->info.map_name, sizeof(item->info.map_name));
truelight@543: 		item->info.map_width = NetworkRecv_uint16(p);
truelight@543: 		item->info.map_height = NetworkRecv_uint16(p);
truelight@543: 		item->info.map_set = NetworkRecv_uint8(p);
truelight@543: 		item->info.dedicated = NetworkRecv_uint8(p);
truelight@543: 
truelight@543: 		if (item->info.hostname[0] == '\0')
truelight@543: 			snprintf(item->info.hostname, sizeof(item->info.hostname), "%s", inet_ntoa(client_addr->sin_addr));
truelight@543: 	}
truelight@543: 
truelight@543: 	item->online = true;
truelight@543: 
truelight@543: 	UpdateNetworkGameWindow(false);
truelight@543: }
truelight@543: 
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@543: 	RECEIVE_COMMAND(PACKET_UDP_FIND_SERVER),
truelight@543: 	RECEIVE_COMMAND(PACKET_UDP_SERVER_RESPONSE),
truelight@543: };
truelight@543: 
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: 
truelight@543: void NetworkHandleUDPPacket(Packet *p, struct sockaddr_in *client_addr)
truelight@543: {
truelight@543: 	byte type;
truelight@543: 
truelight@543: 	type = NetworkRecv_uint8(p);
truelight@543: 
truelight@543: 	if (type < PACKET_UDP_END && _network_udp_packet[type] != NULL) {
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
truelight@543: void NetworkSendUDP_Packet(Packet *p, struct sockaddr_in *recv)
truelight@543: {
truelight@543: 	SOCKET udp;
truelight@543: 	int res;
truelight@543: 
truelight@543: 	// Find the correct socket
truelight@543: 	if (_network_udp_server)
truelight@543: 		udp = _udp_server_socket;
truelight@543: 	else
truelight@543: 		udp = _udp_client_socket;
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@543: bool NetworkUDPListen(uint32 host, uint16 port)
truelight@543: {
truelight@543: 	struct sockaddr_in sin;
truelight@543: 	SOCKET udp;
truelight@543: 
truelight@543: 	// Make sure sockets are closed
truelight@543: 	if (_network_udp_server)
truelight@543: 		closesocket(_udp_server_socket);
truelight@543: 	else
truelight@543: 		closesocket(_udp_client_socket);
truelight@543: 
truelight@543: 	udp = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
truelight@543: 	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;
truelight@543: 		ioctlsocket(udp, FIONBIO, &blocking);
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@543: 	if (bind(udp, (struct sockaddr*)&sin, sizeof(sin)) != 0) {
truelight@543: 		DEBUG(net, 1) ("[NET][UDP] error: bind failed on port %i", port);
truelight@543: 		return false;
truelight@543: 	}
truelight@543: 
truelight@543: 	// enable broadcasting
truelight@543: 	// allow reusing
truelight@543: 	{
truelight@543: 		unsigned long val = 1;
truelight@543: 		setsockopt(udp, SOL_SOCKET, SO_BROADCAST, (char *) &val , sizeof(val));
truelight@543: 		val = 1;
truelight@543: 		setsockopt(udp, SOL_SOCKET, SO_REUSEADDR, (char *) &val , sizeof(val));
truelight@543: 	}
truelight@543: 
truelight@543: 	if (_network_udp_server)
truelight@543: 		_udp_server_socket = udp;
truelight@543: 	else
truelight@543: 		_udp_client_socket = udp;
truelight@543: 
truelight@543: 	DEBUG(net, 1)("[NET][UDP] Listening on port %d", port);
truelight@543: 
truelight@543: 	return true;
truelight@543: }
truelight@543: 
truelight@543: // Close UDP connection
truelight@543: void NetworkUDPClose(void)
truelight@543: {
truelight@543: 	DEBUG(net, 1) ("[NET][UDP] Closed listener");
truelight@543: 
truelight@543: 	if (_network_udp_server) {
truelight@543: 		closesocket(_udp_server_socket);
truelight@543: 		_udp_server_socket = INVALID_SOCKET;
truelight@543: 		_network_udp_server = false;
truelight@543: 		_network_udp_broadcast = 0;
truelight@543: 	} else {
truelight@543: 		closesocket(_udp_client_socket);
truelight@543: 		_udp_client_socket = INVALID_SOCKET;
truelight@543: 		_network_udp_broadcast = 0;
truelight@543: 	}
truelight@543: }
truelight@543: 
truelight@543: // Receive something on UDP level
truelight@543: void NetworkUDPReceive(void)
truelight@543: {
truelight@543: 	struct sockaddr_in client_addr;
truelight@543: #ifndef __MORPHOS__
truelight@543: 	int client_len;
truelight@543: #else
truelight@543: 	LONG client_len; // for some reason we need a 'LONG' under MorphOS
truelight@543: #endif
truelight@543: 	int nbytes;
truelight@543: 	static Packet *p = NULL;
truelight@543: 	int packet_len;
truelight@543: 	SOCKET udp;
truelight@543: 
truelight@543: 	if (_network_udp_server)
truelight@543: 		udp = _udp_server_socket;
truelight@543: 	else
truelight@543: 		udp = _udp_client_socket;
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
truelight@543: void NetworkUDPBroadCast(void)
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@543: 	p = NetworkSend_Init(PACKET_UDP_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@543: 		NetworkSendUDP_Packet(p, &out_addr);
truelight@543: 
truelight@543: 		i++;
truelight@543: 	}
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@543: 		if (!NetworkUDPListen(0, 0))
truelight@543: 			return;
truelight@543: 
truelight@543: 	DEBUG(net, 0)("[NET][UDP] Searching server");
truelight@543: 
truelight@543: 	NetworkUDPBroadCast();
truelight@543: 	_network_udp_broadcast = 300; // Stay searching for 300 ticks
truelight@543: }
truelight@543: 
truelight@543: void NetworkUDPQueryServer(const byte* 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@543: 		if (!NetworkUDPListen(0, 0))
truelight@543: 			return;
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@543: 	p = NetworkSend_Init(PACKET_UDP_FIND_SERVER);
truelight@543: 
truelight@543: 	NetworkSendUDP_Packet(p, &out_addr);
truelight@543: 
truelight@543: 	free(p);
truelight@543: 
truelight@543: 	UpdateNetworkGameWindow(false);
truelight@543: }
truelight@543: 
truelight@543: void NetworkUDPInitialize(void)
truelight@543: {
truelight@543: 	_udp_client_socket = INVALID_SOCKET;
truelight@543: 	_udp_server_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 */