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@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@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@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; 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@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 active = 0; 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: FOR_ALL_PLAYERS(player) { truelight@638: if (player->is_active) truelight@638: active++; truelight@638: } truelight@638: truelight@638: /* Send the amount of active companies */ truelight@638: NetworkSend_uint8 (packet, NETWORK_COMPANY_INFO_VERSION); truelight@638: NetworkSend_uint8 (packet, active); 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@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@638: NetworkSendUDP_Packet(packet, client_addr); truelight@638: truelight@638: free(packet); truelight@638: } truelight@638: 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@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@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@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@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@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@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@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@638: p = NetworkSend_Init(PACKET_UDP_CLIENT_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@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@668: /* Only send once in the 450 game-days (about 15 minutes) */ truelight@668: if (_network_last_advertise_date + 450 > _date) truelight@668: return; truelight@668: _network_last_advertise_date = _date; 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@668: NetworkSendUDP_Packet(p, &out_addr); 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@543: truelight@543: _network_udp_server = false; truelight@543: _network_udp_broadcast = 0; truelight@543: } truelight@543: truelight@543: #endif /* ENABLE_NETWORK */