#include "stdafx.h" #include "Emu/Cell/PPUModule.h" #include "Emu/system_config.h" #include "Emu/Cell/Modules/cellSysutil.h" #include "Emu/Memory/vm_ptr.h" #include "Emu/IdManager.h" #include "np_handler.h" #include "np_contexts.h" #include "np_structs_extra.h" LOG_CHANNEL(rpcn_log, "rpcn"); namespace np { std::vector np_handler::get_match2_server_list(SceNpMatching2ContextId ctx_id) { std::vector server_list{}; if (g_cfg.net.psn_status != np_psn_status::psn_rpcn) { return server_list; } if (!rpcn->get_server_list(get_req_id(0), get_match2_context(ctx_id)->communicationId, server_list)) { rpcn_log.error("Disconnecting from RPCN!"); is_psn_active = false; } return server_list; } u32 np_handler::get_server_status(SceNpMatching2ContextId ctx_id, vm::cptr optParam, u16 server_id) { // TODO: actually implement interaction with server for this? u32 req_id = generate_callback_info(ctx_id, optParam); u32 event_key = get_event_key(); auto& edata = allocate_req_result(event_key, SCE_NP_MATCHING2_EVENT_DATA_MAX_SIZE_GetServerInfo, sizeof(SceNpMatching2GetServerInfoResponse)); SceNpMatching2GetServerInfoResponse* serv_info = reinterpret_cast(edata.data()); serv_info->server.serverId = server_id; serv_info->server.status = SCE_NP_MATCHING2_SERVER_STATUS_AVAILABLE; np_memory.shrink_allocation(edata.addr(), edata.size()); const auto cb_info = take_pending_request(req_id); if (cb_info.cb) { sysutil_register_cb([=, size = edata.size()](ppu_thread& cb_ppu) -> s32 { cb_info.cb(cb_ppu, cb_info.ctx_id, req_id, SCE_NP_MATCHING2_REQUEST_EVENT_GetServerInfo, event_key, 0, size, cb_info.cb_arg); return 0; }); } return req_id; } u32 np_handler::create_server_context(SceNpMatching2ContextId ctx_id, vm::cptr optParam, u16 /*server_id*/) { u32 req_id = generate_callback_info(ctx_id, optParam); u32 event_key = get_event_key(); const auto cb_info = take_pending_request(req_id); if (cb_info.cb) { sysutil_register_cb([=](ppu_thread& cb_ppu) -> s32 { cb_info.cb(cb_ppu, cb_info.ctx_id, req_id, SCE_NP_MATCHING2_REQUEST_EVENT_CreateServerContext, event_key, 0, 0, cb_info.cb_arg); return 0; }); } return req_id; } u32 np_handler::delete_server_context(SceNpMatching2ContextId ctx_id, vm::cptr optParam, u16 /*server_id*/) { u32 req_id = generate_callback_info(ctx_id, optParam); u32 event_key = get_event_key(); const auto cb_info = take_pending_request(req_id); if (cb_info.cb) { sysutil_register_cb([=](ppu_thread& cb_ppu) -> s32 { cb_info.cb(cb_ppu, cb_info.ctx_id, req_id, SCE_NP_MATCHING2_REQUEST_EVENT_DeleteServerContext, event_key, 0, 0, cb_info.cb_arg); return 0; }); } return req_id; } u32 np_handler::get_world_list(SceNpMatching2ContextId ctx_id, vm::cptr optParam, u16 server_id) { u32 req_id = generate_callback_info(ctx_id, optParam); if (!rpcn->get_world_list(req_id, get_match2_context(ctx_id)->communicationId, server_id)) { rpcn_log.error("Disconnecting from RPCN!"); is_psn_active = false; } return req_id; } bool np_handler::reply_get_world_list(u32 req_id, std::vector& reply_data) { const auto cb_info = take_pending_request(req_id); vec_stream reply(reply_data, 1); std::vector world_list; u32 num_worlds = reply.get(); for (u32 i = 0; i < num_worlds; i++) { world_list.push_back(reply.get()); } if (reply.is_error()) { return error_and_disconnect("Malformed reply to GetWorldList command"); } u32 event_key = get_event_key(); auto& edata = allocate_req_result(event_key, SCE_NP_MATCHING2_EVENT_DATA_MAX_SIZE_GetWorldInfoList, sizeof(SceNpMatching2GetWorldInfoListResponse)); auto* world_info = reinterpret_cast(edata.data()); world_info->worldNum = world_list.size(); if (!world_list.empty()) { auto* worlds = edata.allocate(sizeof(SceNpMatching2World) * world_list.size(), world_info->world); for (usz i = 0; i < world_list.size(); i++) { worlds[i].worldId = world_list[i]; worlds[i].numOfLobby = 1; // TODO worlds[i].maxNumOfTotalLobbyMember = 10000; worlds[i].curNumOfTotalLobbyMember = 1; worlds[i].curNumOfRoom = 1; worlds[i].curNumOfTotalRoomMember = 1; } } np_memory.shrink_allocation(edata.addr(), edata.size()); if (cb_info.cb) { sysutil_register_cb([=, size = edata.size()](ppu_thread& cb_ppu) -> s32 { cb_info.cb(cb_ppu, cb_info.ctx_id, req_id, SCE_NP_MATCHING2_REQUEST_EVENT_GetWorldInfoList, event_key, 0, size, cb_info.cb_arg); return 0; }); } return true; } u32 np_handler::create_join_room(SceNpMatching2ContextId ctx_id, vm::cptr optParam, const SceNpMatching2CreateJoinRoomRequest* req) { u32 req_id = generate_callback_info(ctx_id, optParam); if (!rpcn->createjoin_room(req_id, get_match2_context(ctx_id)->communicationId, req)) { rpcn_log.error("Disconnecting from RPCN!"); is_psn_active = false; } // More elegant solution would be to send back password with creation reply cached_cj_password = req->roomPassword ? std::optional{*req->roomPassword} : std::nullopt; return req_id; } bool np_handler::reply_create_join_room(u32 req_id, std::vector& reply_data) { const auto cb_info = take_pending_request(req_id); vec_stream reply(reply_data, 1); auto create_room_resp = reply.get_rawdata(); if (reply.is_error()) return error_and_disconnect("Malformed reply to CreateRoom command"); u32 event_key = get_event_key(); auto* resp = flatbuffers::GetRoot(create_room_resp.data()); auto& edata = allocate_req_result(event_key, SCE_NP_MATCHING2_EVENT_DATA_MAX_SIZE_CreateJoinRoom, sizeof(SceNpMatching2CreateJoinRoomResponse)); auto* room_resp = reinterpret_cast(edata.data()); auto* room_info = edata.allocate(sizeof(SceNpMatching2RoomDataInternal), room_resp->roomDataInternal); RoomDataInternal_to_SceNpMatching2RoomDataInternal(edata, resp, room_info, npid); np_memory.shrink_allocation(edata.addr(), edata.size()); np_cache.insert_room(room_info); np_cache.update_password(room_resp->roomDataInternal->roomId, cached_cj_password); // Establish Matching2 self signaling info auto& sigh = g_fxo->get>(); sigh.set_self_sig2_info(room_info->roomId, 1); sigh.set_sig2_infos(room_info->roomId, 1, SCE_NP_SIGNALING_CONN_STATUS_ACTIVE, rpcn->get_addr_sig(), rpcn->get_port_sig(), true); // TODO? Should this send a message to Signaling CB? Is this even necessary? extra_nps::print_create_room_resp(room_resp); if (cb_info.cb) { sysutil_register_cb([=, size = edata.size()](ppu_thread& cb_ppu) -> s32 { cb_info.cb(cb_ppu, cb_info.ctx_id, req_id, SCE_NP_MATCHING2_REQUEST_EVENT_CreateJoinRoom, event_key, 0, size, cb_info.cb_arg); return 0; }); } return true; } u32 np_handler::join_room(SceNpMatching2ContextId ctx_id, vm::cptr optParam, const SceNpMatching2JoinRoomRequest* req) { u32 req_id = generate_callback_info(ctx_id, optParam); if (!rpcn->join_room(req_id, get_match2_context(ctx_id)->communicationId, req)) { rpcn_log.error("Disconnecting from RPCN!"); is_psn_active = false; } return req_id; } bool np_handler::reply_join_room(u32 req_id, std::vector& reply_data) { const auto cb_info = take_pending_request(req_id); vec_stream reply(reply_data, 1); auto join_room_resp = reply.get_rawdata(); if (reply.is_error()) return error_and_disconnect("Malformed reply to JoinRoom command"); u32 event_key = get_event_key(); auto resp = flatbuffers::GetRoot(join_room_resp.data()); auto& edata = allocate_req_result(event_key, SCE_NP_MATCHING2_EVENT_DATA_MAX_SIZE_JoinRoom, sizeof(SceNpMatching2JoinRoomResponse)); auto* room_resp = reinterpret_cast(edata.data()); auto* room_info = edata.allocate(sizeof(SceNpMatching2RoomDataInternal), room_resp->roomDataInternal); u16 member_id = RoomDataInternal_to_SceNpMatching2RoomDataInternal(edata, resp, room_info, npid); np_memory.shrink_allocation(edata.addr(), edata.size()); np_cache.insert_room(room_info); extra_nps::print_room_data_internal(room_info); // Establish Matching2 self signaling info auto& sigh = g_fxo->get>(); sigh.set_self_sig2_info(room_info->roomId, member_id); sigh.set_sig2_infos(room_info->roomId, member_id, SCE_NP_SIGNALING_CONN_STATUS_ACTIVE, rpcn->get_addr_sig(), rpcn->get_port_sig(), true); // TODO? Should this send a message to Signaling CB? Is this even necessary? if (cb_info.cb) { sysutil_register_cb([=, size = edata.size()](ppu_thread& cb_ppu) -> s32 { cb_info.cb(cb_ppu, cb_info.ctx_id, req_id, SCE_NP_MATCHING2_REQUEST_EVENT_JoinRoom, event_key, 0, size, cb_info.cb_arg); return 0; }); } return true; } u32 np_handler::leave_room(SceNpMatching2ContextId ctx_id, vm::cptr optParam, const SceNpMatching2LeaveRoomRequest* req) { u32 req_id = generate_callback_info(ctx_id, optParam); if (!rpcn->leave_room(req_id, get_match2_context(ctx_id)->communicationId, req)) { rpcn_log.error("Disconnecting from RPCN!"); is_psn_active = false; } return req_id; } bool np_handler::reply_leave_room(u32 req_id, std::vector& reply_data) { const auto cb_info = take_pending_request(req_id); vec_stream reply(reply_data, 1); u64 room_id = reply.get(); if (reply.is_error()) return error_and_disconnect("Malformed reply to LeaveRoom command"); u32 event_key = get_event_key(); // Unsure if necessary if there is no data // Disconnect all users from that room auto& sigh = g_fxo->get>(); sigh.disconnect_sig2_users(room_id); if (cb_info.cb) { sysutil_register_cb([=](ppu_thread& cb_ppu) -> s32 { cb_info.cb(cb_ppu, cb_info.ctx_id, req_id, SCE_NP_MATCHING2_REQUEST_EVENT_LeaveRoom, event_key, 0, 0, cb_info.cb_arg); return 0; }); } return true; } u32 np_handler::search_room(SceNpMatching2ContextId ctx_id, vm::cptr optParam, const SceNpMatching2SearchRoomRequest* req) { u32 req_id = generate_callback_info(ctx_id, optParam); if (!rpcn->search_room(req_id, get_match2_context(ctx_id)->communicationId, req)) { rpcn_log.error("Disconnecting from RPCN!"); is_psn_active = false; } return req_id; } bool np_handler::reply_search_room(u32 req_id, std::vector& reply_data) { const auto cb_info = take_pending_request(req_id); vec_stream reply(reply_data, 1); auto search_room_resp = reply.get_rawdata(); if (reply.is_error()) return error_and_disconnect("Malformed reply to SearchRoom command"); u32 event_key = get_event_key(); auto* resp = flatbuffers::GetRoot(search_room_resp.data()); auto& edata = allocate_req_result(event_key, SCE_NP_MATCHING2_EVENT_DATA_MAX_SIZE_SearchRoom, sizeof(SceNpMatching2SearchRoomResponse)); auto* search_resp = reinterpret_cast(edata.data()); SearchRoomResponse_to_SceNpMatching2SearchRoomResponse(edata, resp, search_resp); np_memory.shrink_allocation(edata.addr(), edata.size()); if (cb_info.cb) { sysutil_register_cb([=, size = edata.size()](ppu_thread& cb_ppu) -> s32 { cb_info.cb(cb_ppu, cb_info.ctx_id, req_id, SCE_NP_MATCHING2_REQUEST_EVENT_SearchRoom, event_key, 0, size, cb_info.cb_arg); return 0; }); } return true; } u32 np_handler::get_roomdata_external_list(SceNpMatching2ContextId ctx_id, vm::cptr optParam, const SceNpMatching2GetRoomDataExternalListRequest* req) { u32 req_id = generate_callback_info(ctx_id, optParam); if (!rpcn->get_roomdata_external_list(req_id, get_match2_context(ctx_id)->communicationId, req)) { rpcn_log.error("Disconnecting from RPCN!"); is_psn_active = false; } return req_id; } bool np_handler::reply_get_roomdata_external_list(u32 req_id, std::vector& reply_data) { const auto cb_info = take_pending_request(req_id); vec_stream reply(reply_data, 1); auto get_room_ext_resp = reply.get_rawdata(); if (reply.is_error()) return error_and_disconnect("Malformed reply to GetRoomDataExternalList command"); u32 event_key = get_event_key(); auto* resp = flatbuffers::GetRoot(get_room_ext_resp.data()); auto& edata = allocate_req_result(event_key, SCE_NP_MATCHING2_EVENT_DATA_MAX_SIZE_GetRoomDataExternalList, sizeof(SceNpMatching2GetRoomDataExternalListResponse)); auto* sce_get_room_ext_resp = reinterpret_cast(edata.data()); GetRoomDataExternalListResponse_to_SceNpMatching2GetRoomDataExternalListResponse(edata, resp, sce_get_room_ext_resp); np_memory.shrink_allocation(edata.addr(), edata.size()); if (cb_info.cb) { sysutil_register_cb([=, size = edata.size()](ppu_thread& cb_ppu) -> s32 { cb_info.cb(cb_ppu, cb_info.ctx_id, req_id, SCE_NP_MATCHING2_REQUEST_EVENT_GetRoomDataExternalList, event_key, 0, size, cb_info.cb_arg); return 0; }); } return true; } u32 np_handler::set_roomdata_external(SceNpMatching2ContextId ctx_id, vm::cptr optParam, const SceNpMatching2SetRoomDataExternalRequest* req) { u32 req_id = generate_callback_info(ctx_id, optParam); extra_nps::print_set_roomdata_ext_req(req); if (!rpcn->set_roomdata_external(req_id, get_match2_context(ctx_id)->communicationId, req)) { rpcn_log.error("Disconnecting from RPCN!"); is_psn_active = false; } return req_id; } bool np_handler::reply_set_roomdata_external(u32 req_id, std::vector& /*reply_data*/) { const auto cb_info = take_pending_request(req_id); u32 event_key = get_event_key(); // Unsure if necessary if there is no data if (cb_info.cb) { sysutil_register_cb([=](ppu_thread& cb_ppu) -> s32 { cb_info.cb(cb_ppu, cb_info.ctx_id, req_id, SCE_NP_MATCHING2_REQUEST_EVENT_SetRoomDataExternal, event_key, 0, 0, cb_info.cb_arg); return 0; }); } return true; } u32 np_handler::get_roomdata_internal(SceNpMatching2ContextId ctx_id, vm::cptr optParam, const SceNpMatching2GetRoomDataInternalRequest* req) { u32 req_id = generate_callback_info(ctx_id, optParam); if (!rpcn->get_roomdata_internal(req_id, get_match2_context(ctx_id)->communicationId, req)) { rpcn_log.error("Disconnecting from RPCN!"); is_psn_active = false; } return req_id; } bool np_handler::reply_get_roomdata_internal(u32 req_id, std::vector& reply_data) { const auto cb_info = take_pending_request(req_id); vec_stream reply(reply_data, 1); auto internal_data = reply.get_rawdata(); if (reply.is_error()) return error_and_disconnect("Malformed reply to GetRoomDataInternal command"); u32 event_key = get_event_key(); auto* resp = flatbuffers::GetRoot(internal_data.data()); auto& edata = allocate_req_result(event_key, SCE_NP_MATCHING2_EVENT_DATA_MAX_SIZE_GetRoomDataInternal, sizeof(SceNpMatching2GetRoomDataInternalResponse)); auto* room_resp = reinterpret_cast(edata.data()); auto* room_info = edata.allocate(sizeof(SceNpMatching2RoomDataInternal), room_resp->roomDataInternal); RoomDataInternal_to_SceNpMatching2RoomDataInternal(edata, resp, room_info, npid); np_memory.shrink_allocation(edata.addr(), edata.size()); np_cache.insert_room(room_info); extra_nps::print_room_data_internal(room_info); if (cb_info.cb) { sysutil_register_cb([=, size = edata.size()](ppu_thread& cb_ppu) -> s32 { cb_info.cb(cb_ppu, cb_info.ctx_id, req_id, SCE_NP_MATCHING2_REQUEST_EVENT_GetRoomDataInternal, event_key, 0, size, cb_info.cb_arg); return 0; }); } return true; } u32 np_handler::set_roomdata_internal(SceNpMatching2ContextId ctx_id, vm::cptr optParam, const SceNpMatching2SetRoomDataInternalRequest* req) { u32 req_id = generate_callback_info(ctx_id, optParam); extra_nps::print_set_roomdata_int_req(req); if (!rpcn->set_roomdata_internal(req_id, get_match2_context(ctx_id)->communicationId, req)) { rpcn_log.error("Disconnecting from RPCN!"); is_psn_active = false; } return req_id; } bool np_handler::reply_set_roomdata_internal(u32 req_id, std::vector& /*reply_data*/) { const auto cb_info = take_pending_request(req_id); u32 event_key = get_event_key(); // Unsure if necessary if there is no data if (cb_info.cb) { sysutil_register_cb([=](ppu_thread& cb_ppu) -> s32 { cb_info.cb(cb_ppu, cb_info.ctx_id, req_id, SCE_NP_MATCHING2_REQUEST_EVENT_SetRoomDataInternal, event_key, 0, 0, cb_info.cb_arg); return 0; }); } return true; } u32 np_handler::set_roommemberdata_internal(SceNpMatching2ContextId ctx_id, vm::cptr optParam, const SceNpMatching2SetRoomMemberDataInternalRequest* req) { u32 req_id = generate_callback_info(ctx_id, optParam); if (!rpcn->set_roommemberdata_internal(req_id, get_match2_context(ctx_id)->communicationId, req)) { rpcn_log.error("Disconnecting from RPCN!"); is_psn_active = false; } return req_id; } bool np_handler::reply_set_roommemberdata_internal(u32 req_id, std::vector& /*reply_data*/) { const auto cb_info = take_pending_request(req_id); u32 event_key = get_event_key(); // Unsure if necessary if there is no data if (cb_info.cb) { sysutil_register_cb([=](ppu_thread& cb_ppu) -> s32 { cb_info.cb(cb_ppu, cb_info.ctx_id, req_id, SCE_NP_MATCHING2_REQUEST_EVENT_SetRoomMemberDataInternal, event_key, 0, 0, cb_info.cb_arg); return 0; }); } return true; } u32 np_handler::get_ping_info(SceNpMatching2ContextId ctx_id, vm::cptr optParam, const SceNpMatching2SignalingGetPingInfoRequest* req) { u32 req_id = generate_callback_info(ctx_id, optParam); if (!rpcn->ping_room_owner(req_id, get_match2_context(ctx_id)->communicationId, req->roomId)) { rpcn_log.error("Disconnecting from RPCN!"); is_psn_active = false; } return req_id; } bool np_handler::reply_get_ping_info(u32 req_id, std::vector& reply_data) { const auto cb_info = take_pending_request(req_id); vec_stream reply(reply_data, 1); auto ping_resp = reply.get_rawdata(); if (reply.is_error()) return error_and_disconnect("Malformed reply to PingRoomOwner command"); u32 event_key = get_event_key(); auto* resp = flatbuffers::GetRoot(ping_resp.data()); auto& edata = allocate_req_result(event_key, SCE_NP_MATCHING2_EVENT_DATA_MAX_SIZE_SignalingGetPingInfo, sizeof(SceNpMatching2SignalingGetPingInfoResponse)); auto* final_ping_resp = reinterpret_cast(edata.data()); GetPingInfoResponse_to_SceNpMatching2SignalingGetPingInfoResponse(resp, final_ping_resp); np_memory.shrink_allocation(edata.addr(), edata.size()); if (cb_info.cb) { sysutil_register_cb([=, size = edata.size()](ppu_thread& cb_ppu) -> s32 { cb_info.cb(cb_ppu, cb_info.ctx_id, req_id, SCE_NP_MATCHING2_REQUEST_EVENT_SignalingGetPingInfo, event_key, 0, size, cb_info.cb_arg); return 0; }); } return true; } u32 np_handler::send_room_message(SceNpMatching2ContextId ctx_id, vm::cptr optParam, const SceNpMatching2SendRoomMessageRequest* req) { u32 req_id = generate_callback_info(ctx_id, optParam); if (!rpcn->send_room_message(req_id, get_match2_context(ctx_id)->communicationId, req)) { rpcn_log.error("Disconnecting from RPCN!"); is_psn_active = false; } return req_id; } bool np_handler::reply_send_room_message(u32 req_id, std::vector& /*reply_data*/) { const auto cb_info = take_pending_request(req_id); if (cb_info.cb) { sysutil_register_cb([=](ppu_thread& cb_ppu) -> s32 { cb_info.cb(cb_ppu, cb_info.ctx_id, req_id, SCE_NP_MATCHING2_REQUEST_EVENT_SendRoomMessage, 0, 0, 0, cb_info.cb_arg); return 0; }); } return true; } void np_handler::req_sign_infos(const std::string& npid, u32 conn_id) { u32 req_id = get_req_id(0x3333); { std::lock_guard lock(mutex_pending_sign_infos_requests); pending_sign_infos_requests[req_id] = conn_id; } if (!rpcn->req_sign_infos(req_id, npid)) { rpcn_log.error("Disconnecting from RPCN!"); is_psn_active = false; } return; } bool np_handler::reply_req_sign_infos(u32 req_id, std::vector& reply_data) { u32 conn_id; { std::lock_guard lock(mutex_pending_sign_infos_requests); conn_id = pending_sign_infos_requests.at(req_id); pending_sign_infos_requests.erase(req_id); } vec_stream reply(reply_data, 1); u32 addr = reply.get(); u16 port = reply.get(); if (reply.is_error()) return error_and_disconnect("Malformed reply to RequestSignalingInfos command"); auto& sigh = g_fxo->get>(); sigh.start_sig(conn_id, addr, port); return true; } void np_handler::req_ticket([[maybe_unused]] u32 version, [[maybe_unused]] const SceNpId* npid, const char* service_id, const u8* cookie, u32 cookie_size, [[maybe_unused]] const char* entitlement_id, [[maybe_unused]] u32 consumed_count) { u32 req_id = get_req_id(0x3333); std::string service_id_str(service_id); std::vector cookie_vec; if (cookie && cookie_size) { cookie_vec.assign(cookie, cookie + cookie_size); } if (!rpcn->req_ticket(req_id, service_id_str, cookie_vec)) { rpcn_log.error("Disconnecting from RPCN!"); is_psn_active = false; } return; } const ticket& np_handler::get_ticket() const { return current_ticket; } bool np_handler::reply_req_ticket([[maybe_unused]] u32 req_id, std::vector& reply_data) { vec_stream reply(reply_data, 1); auto ticket_raw = reply.get_rawdata(); if (reply.is_error()) return error_and_disconnect("Malformed reply to RequestTicket command"); current_ticket = ticket(std::move(ticket_raw)); auto ticket_size = static_cast(current_ticket.size()); if (manager_cb) { sysutil_register_cb([manager_cb = this->manager_cb, ticket_size, manager_cb_arg = this->manager_cb_arg](ppu_thread& cb_ppu) -> s32 { manager_cb(cb_ppu, SCE_NP_MANAGER_EVENT_GOT_TICKET, ticket_size, manager_cb_arg); return 0; }); } return true; } } // namespace np