Fix timestamps in signaling_handler

This commit is contained in:
RipleyTom 2023-03-14 21:11:40 +01:00 committed by Megamouse
parent 5fa9dfac6a
commit 8cd6b9275a
2 changed files with 16 additions and 8 deletions

View file

@ -184,6 +184,11 @@ bool signaling_handler::validate_signaling_packet(const signaling_packet* sp)
return true; return true;
} }
u64 signaling_handler::get_micro_timestamp(const std::chrono::steady_clock::time_point& time_point)
{
return std::chrono::duration_cast<std::chrono::microseconds>(time_point.time_since_epoch()).count();
}
void signaling_handler::process_incoming_messages() void signaling_handler::process_incoming_messages()
{ {
auto msgs = get_sign_msgs(); auto msgs = get_sign_msgs();
@ -249,14 +254,15 @@ void signaling_handler::process_incoming_messages()
} }
sent_packet.command = signal_ping; sent_packet.command = signal_ping;
sent_packet.timestamp_sender = now.time_since_epoch().count(); sent_packet.timestamp_sender = get_micro_timestamp(now);
send_signaling_packet(sent_packet, si->addr, si->port); send_signaling_packet(sent_packet, si->addr, si->port);
queue_signaling_packet(sent_packet, si, now + REPEAT_PING_DELAY); queue_signaling_packet(sent_packet, si, now + REPEAT_PING_DELAY);
}; };
const auto update_rtt = [&](u64 rtt_timestamp) const auto update_rtt = [&](u64 rtt_timestamp)
{ {
u32 rtt = now.time_since_epoch().count() - rtt_timestamp; u64 timestamp_now = get_micro_timestamp(now);
u64 rtt = timestamp_now - rtt_timestamp;
si->last_rtts[(si->rtt_counters % 6)] = rtt; si->last_rtts[(si->rtt_counters % 6)] = rtt;
si->rtt_counters++; si->rtt_counters++;
@ -267,7 +273,7 @@ void signaling_handler::process_incoming_messages()
sum += si->last_rtts[index]; sum += si->last_rtts[index];
} }
si->rtt = (sum / num_rtts) / 1000; si->rtt = (sum / num_rtts);
}; };
switch (sp->command) switch (sp->command)
@ -289,7 +295,7 @@ void signaling_handler::process_incoming_messages()
schedule_repeat = true; schedule_repeat = true;
sent_packet.command = signal_connect_ack; sent_packet.command = signal_connect_ack;
sent_packet.timestamp_sender = sp->timestamp_sender; sent_packet.timestamp_sender = sp->timestamp_sender;
sent_packet.timestamp_receiver = now.time_since_epoch().count(); sent_packet.timestamp_receiver = get_micro_timestamp(now);
update_si_addr(si, op_addr, op_port); update_si_addr(si, op_addr, op_port);
break; break;
case signal_connect_ack: case signal_connect_ack:
@ -298,7 +304,7 @@ void signaling_handler::process_incoming_messages()
schedule_repeat = false; schedule_repeat = false;
setup_ping(); setup_ping();
sent_packet.command = signal_confirm; sent_packet.command = signal_confirm;
sent_packet.timestamp_receiver = now.time_since_epoch().count(); sent_packet.timestamp_receiver = sp->timestamp_receiver;
retire_packet(si, signal_connect); retire_packet(si, signal_connect);
update_si_addr(si, op_addr, op_port); update_si_addr(si, op_addr, op_port);
update_si_mapped_addr(si, sp->sent_addr, sp->sent_port); update_si_mapped_addr(si, sp->sent_addr, sp->sent_port);
@ -375,10 +381,10 @@ void signaling_handler::operator()()
{ {
case signal_connect: case signal_connect:
case signal_ping: case signal_ping:
sig.packet.timestamp_sender = now.time_since_epoch().count(); sig.packet.timestamp_sender = get_micro_timestamp(now);
break; break;
case signal_connect_ack: case signal_connect_ack:
sig.packet.timestamp_receiver = now.time_since_epoch().count(); sig.packet.timestamp_receiver = get_micro_timestamp(now);
break; break;
default: default:
break; break;
@ -586,7 +592,7 @@ void signaling_handler::start_sig(u32 conn_id, u32 addr, u16 port)
std::lock_guard lock(data_mutex); std::lock_guard lock(data_mutex);
auto& sent_packet = sig_packet; auto& sent_packet = sig_packet;
sent_packet.command = signal_connect; sent_packet.command = signal_connect;
sent_packet.timestamp_sender = steady_clock::now().time_since_epoch().count(); sent_packet.timestamp_sender = get_micro_timestamp(steady_clock::now());
ensure(sig_peers.contains(conn_id)); ensure(sig_peers.contains(conn_id));
std::shared_ptr<signaling_info> si = ::at32(sig_peers, conn_id); std::shared_ptr<signaling_info> si = ::at32(sig_peers, conn_id);

View file

@ -116,6 +116,8 @@ private:
vm::ptr<SceNpMatching2SignalingCallback> sig2_cb{}; vm::ptr<SceNpMatching2SignalingCallback> sig2_cb{};
vm::ptr<void> sig2_cb_arg{}; vm::ptr<void> sig2_cb_arg{};
static u64 get_micro_timestamp(const std::chrono::steady_clock::time_point& time_point);
u32 get_always_conn_id(const SceNpId& npid); u32 get_always_conn_id(const SceNpId& npid);
static void update_si_addr(std::shared_ptr<signaling_info>& si, u32 new_addr, u16 new_port); static void update_si_addr(std::shared_ptr<signaling_info>& si, u32 new_addr, u16 new_port);
static void update_si_mapped_addr(std::shared_ptr<signaling_info>& si, u32 new_addr, u16 new_port); static void update_si_mapped_addr(std::shared_ptr<signaling_info>& si, u32 new_addr, u16 new_port);