diff --git a/dll/steam_networking_sockets.h b/dll/steam_networking_sockets.h index 59d36a66..509f1c9b 100644 --- a/dll/steam_networking_sockets.h +++ b/dll/steam_networking_sockets.h @@ -44,6 +44,9 @@ struct Connect_Socket { int64 user_data; std::queue data; + HSteamNetPollGroup poll_group; + + unsigned long long packet_receive_counter; }; class Steam_Networking_Sockets : @@ -62,6 +65,9 @@ public ISteamNetworkingSockets std::vector listen_sockets; std::map connect_sockets; + std::map> poll_groups; + std::chrono::steady_clock::time_point created; + public: static void steam_callback(void *object, Common_Message *msg) { @@ -90,6 +96,8 @@ Steam_Networking_Sockets(class Settings *settings, class Networking *network, cl this->callback_results = callback_results; this->callbacks = callbacks; + + this->created = std::chrono::steady_clock::now(); } ~Steam_Networking_Sockets() @@ -154,6 +162,7 @@ HSteamNetConnection new_connect_socket(SteamNetworkingIdentity remote_identity, socket.remote_id = remote_id; socket.status = status; socket.user_data = -1; + socket.poll_group = k_HSteamNetPollGroup_Invalid; static HSteamNetConnection socket_id; ++socket_id; @@ -725,13 +734,15 @@ SteamNetworkingMessage_t *get_steam_message_connection(HSteamNetConnection hConn pMsg->m_conn = hConn; pMsg->m_identityPeer = connect_socket->second.remote_identity; pMsg->m_nConnUserData = connect_socket->second.user_data; - //TODO - //pMsg->m_usecTimeReceived = - //pMsg->m_nMessageNumber = + pMsg->m_usecTimeReceived = std::chrono::duration_cast(std::chrono::steady_clock::now() - created).count(); + pMsg->m_nMessageNumber = connect_socket->second.packet_receive_counter; + ++connect_socket->second.packet_receive_counter; + pMsg->m_pfnFreeData = &free_steam_message_data; pMsg->m_pfnRelease = &delete_steam_message; pMsg->m_nChannel = 0; connect_socket->second.data.pop(); + PRINT_DEBUG("get_steam_message_connection %u %u\n", hConn, size); return pMsg; } @@ -1023,7 +1034,13 @@ ESteamNetworkingAvailability GetAuthenticationStatus( SteamNetAuthenticationStat HSteamNetPollGroup CreatePollGroup() { PRINT_DEBUG("Steam_Networking_Sockets::CreatePollGroup\n"); - return k_HSteamNetPollGroup_Invalid; + std::lock_guard lock(global_mutex); + static HSteamNetPollGroup poll_group_counter; + ++poll_group_counter; + + HSteamNetPollGroup poll_group_number = poll_group_counter; + poll_groups[poll_group_number] = std::list(); + return poll_group_number; } /// Destroy a poll group created with CreatePollGroup(). @@ -1034,7 +1051,21 @@ HSteamNetPollGroup CreatePollGroup() bool DestroyPollGroup( HSteamNetPollGroup hPollGroup ) { PRINT_DEBUG("Steam_Networking_Sockets::DestroyPollGroup\n"); - return false; + std::lock_guard lock(global_mutex); + auto group = poll_groups.find(hPollGroup); + if (group == poll_groups.end()) { + return false; + } + + for (auto c : group->second) { + auto connect_socket = connect_sockets.find(c); + if (connect_socket != connect_sockets.end()) { + connect_socket->second.poll_group = k_HSteamNetPollGroup_Invalid; + } + } + + poll_groups.erase(group); + return true; } /// Assign a connection to a poll group. Note that a connection may only belong to a @@ -1053,8 +1084,33 @@ bool DestroyPollGroup( HSteamNetPollGroup hPollGroup ) /// is invalid (and not k_HSteamNetPollGroup_Invalid). bool SetConnectionPollGroup( HSteamNetConnection hConn, HSteamNetPollGroup hPollGroup ) { - PRINT_DEBUG("Steam_Networking_Sockets::SetConnectionPollGroup\n"); - return false; + PRINT_DEBUG("Steam_Networking_Sockets::SetConnectionPollGroup %u %u\n", hConn, hPollGroup); + std::lock_guard lock(global_mutex); + auto connect_socket = connect_sockets.find(hConn); + if (connect_socket == connect_sockets.end()) { + return false; + } + + auto group = poll_groups.find(hPollGroup); + if (group == poll_groups.end() && hPollGroup != k_HSteamNetPollGroup_Invalid) { + return false; + } + + HSteamNetPollGroup old_poll_group = connect_socket->second.poll_group; + if (old_poll_group != k_HSteamNetPollGroup_Invalid) { + auto group = poll_groups.find(hPollGroup); + if (group != poll_groups.end()) { + group->second.remove(hConn); + } + } + + connect_socket->second.poll_group = hPollGroup; + if (hPollGroup == k_HSteamNetPollGroup_Invalid) { + return true; + } + + group->second.push_back(hConn); + return true; } /// Same as ReceiveMessagesOnConnection, but will return the next messages available @@ -1074,8 +1130,24 @@ bool SetConnectionPollGroup( HSteamNetConnection hConn, HSteamNetPollGroup hPoll /// other connections.) int ReceiveMessagesOnPollGroup( HSteamNetPollGroup hPollGroup, SteamNetworkingMessage_t **ppOutMessages, int nMaxMessages ) { - PRINT_DEBUG("Steam_Networking_Sockets::ReceiveMessagesOnPollGroup\n"); - return -1; + PRINT_DEBUG("Steam_Networking_Sockets::ReceiveMessagesOnPollGroup %u %i\n", hPollGroup, nMaxMessages); + std::lock_guard lock(global_mutex); + auto group = poll_groups.find(hPollGroup); + if (group == poll_groups.end()) { + return 0; + } + + SteamNetworkingMessage_t *msg = NULL; + int messages = 0; + + for (auto c : group->second) { + while ((msg = get_steam_message_connection(c)) && messages < nMaxMessages) { + ppOutMessages[messages] = msg; + ++messages; + } + } + + return messages; }