Add dynamic access key #7

Merged
redbinder0526 merged 4 commits from v0 into v0 2026-04-25 23:07:52 +02:00
12 changed files with 406 additions and 153 deletions
Showing only changes of commit b3c351100b - Show all commits

Merge branch rust-nex:v0 into v0

red binder 2026-04-25 21:05:47 +00:00

View file

@ -10,8 +10,8 @@ use syn::parse::{Parse, ParseStream};
use syn::punctuated::Punctuated; use syn::punctuated::Punctuated;
use syn::spanned::Spanned; use syn::spanned::Spanned;
use syn::{ use syn::{
parse_macro_input, Attribute, Data, DataStruct, DeriveInput, Fields, FnArg, LitInt, Pat, Token, parse_macro_input, Attribute, Data, DataStruct, DeriveInput, Fields, FnArg, Lit, LitInt,
TraitItem, LitStr, Pat, Token, TraitItem,
}; };
struct ProtoInputParams { struct ProtoInputParams {
@ -400,6 +400,10 @@ pub fn rmc_serialize(input: TokenStream) -> TokenStream {
// generate base data // generate base data
let str_name = Lit::Str(LitStr::new(
&derive_input.ident.to_string(),
derive_input.ident.span(),
));
let ident = derive_input.ident; let ident = derive_input.ident;
let tokens = quote! { let tokens = quote! {
@ -414,6 +418,10 @@ pub fn rmc_serialize(input: TokenStream) -> TokenStream {
} }
#write_size #write_size
fn name() -> &'static str{
#str_name
}
} }
}; };

View file

@ -106,17 +106,21 @@ impl<C: Crypto> Server<C> {
.expect("packet malformed in creation"), .expect("packet malformed in creation"),
);*/ );*/
let mut inner = conn.inner.lock().await; let mut inner = conn.inner.lock().await;
let pieces = data.chunks(700);
let max_piece = pieces.len() - 1;
let mut frag_num = 1;
for (i, piece) in pieces.enumerate() {
let seq = inner.server_packet_counter; let seq = inner.server_packet_counter;
let packet = new_data_packet( let packet = new_data_packet(
NEED_ACK | RELIABLE, NEED_ACK | RELIABLE,
self.param.virtual_port, (&self).param.virtual_port,
conn.addr.virtual_port, conn.addr.virtual_port,
data, piece,
inner.server_packet_counter, inner.server_packet_counter,
conn.session_id, conn.session_id,
0, if i == max_piece { 0 } else { frag_num },
&mut inner.crypto_instance, &mut inner.crypto_instance,
&self.crypto, &(&self).crypto,
); );
inner.server_packet_counter += 1; inner.server_packet_counter += 1;
@ -125,12 +129,11 @@ impl<C: Crypto> Server<C> {
inner.unacknowledged_packets.insert(seq, packet); inner.unacknowledged_packets.insert(seq, packet);
drop(inner);
let conn = Arc::downgrade(&conn); let conn = Arc::downgrade(&conn);
let this = Arc::downgrade(&self); let this = Arc::downgrade(&self);
spawn(async move { spawn(async move {
sleep(Duration::from_millis(i as u64 * 16)).await;
for n in 0..5 { for n in 0..5 {
let Some(data) = packet_ref.upgrade() else { let Some(data) = packet_ref.upgrade() else {
return; return;
@ -143,13 +146,16 @@ impl<C: Crypto> Server<C> {
}; };
info!("send attempt {}", n); info!("send attempt {}", n);
self.socket this.socket
.send_to(&data, conn.addr.regular_socket_addr) .send_to(&data, conn.addr.regular_socket_addr)
.await; .await;
break; break;
} }
}); });
frag_num += 1;
}
drop(inner);
} }
async fn connection_thread( async fn connection_thread(
self: Arc<Self>, self: Arc<Self>,

View file

@ -9,10 +9,14 @@ use tokio::net::TcpListener;
use crate::{ use crate::{
executables::common::{OWN_IP_PRIVATE, SERVER_PORT, new_simple_backend}, executables::common::{OWN_IP_PRIVATE, SERVER_PORT, new_simple_backend},
nex::friends_handler::{FriendsGuest, FriendsManager, FriendsUser}, nex::friends_handler::{
FriendsGuest, FriendsManager, FriendsUser, RemoteFriendRemote, RemoteFriendsUser,
},
reggie::UnitPacketRead, reggie::UnitPacketRead,
rmc::{ rmc::{
protocols::{RmcCallable, new_rmc_gateway_connection}, protocols::{
RmcCallable, RmcPureRemoteObject, friends::RemoteFriends, new_rmc_gateway_connection,
},
structures::RmcSerialize, structures::RmcSerialize,
}, },
rnex_proxy_common::ConnectionInitData, rnex_proxy_common::ConnectionInitData,
@ -21,6 +25,7 @@ use crate::{
pub async fn start_friends_backend() { pub async fn start_friends_backend() {
let fm = Arc::new(FriendsManager { let fm = Arc::new(FriendsManager {
cid_counter: AtomicU32::new(1), cid_counter: AtomicU32::new(1),
users: Default::default(),
}); });
let listen = TcpListener::bind(SocketAddrV4::new(*OWN_IP_PRIVATE, *SERVER_PORT)) let listen = TcpListener::bind(SocketAddrV4::new(*OWN_IP_PRIVATE, *SERVER_PORT))
.await .await
@ -53,6 +58,10 @@ pub async fn start_friends_backend() {
fm, fm,
addr: c.prudpsock_addr, addr: c.prudpsock_addr,
pid: c.pid, pid: c.pid,
data: Default::default(),
current_friends: Default::default(),
this: this.clone(),
remote: RemoteFriendRemote::new(r),
}) })
}); });
} else { } else {

View file

@ -1,4 +1,6 @@
use std::io::{Cursor, Write}; use std::io::{Cursor, Write};
use std::ops::Deref;
use std::sync::Weak;
use std::sync::{Arc, atomic::AtomicU32}; use std::sync::{Arc, atomic::AtomicU32};
use bytemuck::bytes_of; use bytemuck::bytes_of;
@ -9,6 +11,10 @@ use rnex_core::rmc::protocols::account_management::{
AccountManagement, RawAccountManagement, RawAccountManagementInfo, RemoteAccountManagement, AccountManagement, RawAccountManagement, RawAccountManagementInfo, RemoteAccountManagement,
}; };
use rnex_core::rmc::protocols::friends::{Friends, RawFriends, RawFriendsInfo, RemoteFriends}; use rnex_core::rmc::protocols::friends::{Friends, RawFriends, RawFriendsInfo, RemoteFriends};
use rnex_core::rmc::protocols::nintendo_notification::{
NintendoNotification, RawNintendoNotification, RawNintendoNotificationInfo,
RemoteNintendoNotification,
};
use rnex_core::rmc::protocols::secure::{RawSecure, RawSecureInfo, RemoteSecure, Secure}; use rnex_core::rmc::protocols::secure::{RawSecure, RawSecureInfo, RemoteSecure, Secure};
use rnex_core::{ use rnex_core::{
define_rmc_proto, define_rmc_proto,
@ -25,31 +31,51 @@ use rnex_core::{
}, },
}; };
use std::sync::atomic::Ordering::Relaxed; use std::sync::atomic::Ordering::Relaxed;
use tokio::spawn;
use tokio::sync::RwLock;
use rnex_core::rmc::protocols::friends::{GameKey, MiiV2, PrincipalBasicInfo}; use rnex_core::rmc::protocols::friends::{GameKey, MiiV2, PrincipalBasicInfo};
use rnex_core::PID; use rnex_core::PID;
use crate::rmc::protocols::account_management::NintendoCreateAccountData; use rnex_core::rmc::protocols::account_management::NintendoCreateAccountData;
use rnex_core::rmc::protocols::nintendo_notification::NintendoNotificationEvent;
use rnex_core::rmc::structures::RmcSerialize; use rnex_core::rmc::structures::RmcSerialize;
use rnex_core::rmc::structures::data::Data;
define_rmc_proto!( define_rmc_proto!(
proto FriendsUser{ proto FriendsUser{
Secure, Secure,
Friends Friends
} }
); );
define_rmc_proto!(
proto FriendRemote{
NintendoNotification
}
);
define_rmc_proto!( define_rmc_proto!(
proto FriendsGuest{ proto FriendsGuest{
Secure, Secure,
AccountManagement AccountManagement
} }
); );
pub struct UserData {
info: NNAInfo,
presence: NintendoPresenceV2,
}
#[rmc_struct(FriendsUser)] #[rmc_struct(FriendsUser)]
pub struct FriendsUser { pub struct FriendsUser {
pub fm: Arc<FriendsManager>, pub fm: Arc<FriendsManager>,
pub addr: PRUDPSockAddr, pub addr: PRUDPSockAddr,
pub pid: PID, pub pid: PID,
pub data: RwLock<Option<UserData>>,
pub current_friends: RwLock<Vec<PID>>,
pub this: Weak<FriendsUser>,
pub remote: RemoteFriendRemote,
} }
#[rmc_struct(FriendsGuest)] #[rmc_struct(FriendsGuest)]
@ -60,6 +86,7 @@ pub struct FriendsGuest {
pub struct FriendsManager { pub struct FriendsManager {
pub cid_counter: AtomicU32, pub cid_counter: AtomicU32,
pub users: RwLock<Vec<Weak<FriendsUser>>>,
} }
impl FriendsManager { impl FriendsManager {
@ -68,11 +95,28 @@ impl FriendsManager {
} }
} }
pub fn friend_info_from_user(data: &UserData) -> FriendInfo {
FriendInfo {
data: Data {},
nna_info: data.info.clone(),
presence: data.presence.clone(),
comment: Comment {
data: Data {},
unk: 0,
message: "haii =w=".to_string(),
last_changed: KerberosDateTime::now(),
},
became_friends: KerberosDateTime::now(),
last_online: KerberosDateTime::now(),
unk: 0,
}
}
impl Friends for FriendsUser { impl Friends for FriendsUser {
async fn update_and_get_all_information( async fn update_and_get_all_information(
&self, &self,
_info: NNAInfo, info: NNAInfo,
_presence: NintendoPresenceV2, presence: NintendoPresenceV2,
_date_time: KerberosDateTime, _date_time: KerberosDateTime,
) -> Result< ) -> Result<
( (
@ -88,30 +132,33 @@ impl Friends for FriendsUser {
), ),
ErrorCode, ErrorCode,
> { > {
Ok(( println!("updating own data");
PrincipalPreference { let mut data = self.data.write().await;
block_friend_request: false, *data = Some(UserData { info, presence });
show_online: false, let self_fr_info = friend_info_from_user(data.as_ref().unwrap());
show_playing_title: false, let Ok(any_self_fr_info) = Any::new(&self_fr_info) else {
}, return Err(ErrorCode::RendezVous_ControlScriptFailure);
Comment { };
last_changed: KerberosDateTime::now(), drop(data);
message: "".to_string(),
unk: 0, let mut fr_list = vec![FriendInfo {
}, data: Data{},
vec![FriendInfo {
became_friends: KerberosDateTime::now(), became_friends: KerberosDateTime::now(),
comment: Comment { comment: Comment {
data: Data{},
last_changed: KerberosDateTime::now(), last_changed: KerberosDateTime::now(),
message: "I'm just a dummy account :3".to_string(), message: "I'm just a dummy account :3".to_string(),
unk: 0, unk: 0,
}, },
last_online: KerberosDateTime::now(), last_online: KerberosDateTime::now(),
nna_info: NNAInfo { nna_info: NNAInfo {
data: Data{},
principal_basic_info: PrincipalBasicInfo { principal_basic_info: PrincipalBasicInfo {
data: Data{},
pid: 101, pid: 101,
nnid: "dummy:3".to_string(), nnid: "dummy:3".to_string(),
mii: MiiV2{ mii: MiiV2{
data: Data{},
date_time: KerberosDateTime::now(), date_time: KerberosDateTime::now(),
name: "TheDummy".to_string(), name: "TheDummy".to_string(),
mii_data: hex::decode("030000402bd7c32986a771f2dc6b35e31da15e37ff7c0000391e6f006f006d0069000000000000000000000000004040001065033568641e2013661a611821640f0000290052485000000000000000000000000000000000000000000000e838").unwrap(), mii_data: hex::decode("030000402bd7c32986a771f2dc6b35e31da15e37ff7c0000391e6f006f006d0069000000000000000000000000004040001065033568641e2013661a611821640f0000290052485000000000000000000000000000000000000000000000e838").unwrap(),
@ -124,10 +171,12 @@ impl Friends for FriendsUser {
unk2: 0 unk2: 0
}, },
presence: NintendoPresenceV2{ presence: NintendoPresenceV2{
data: Data{},
changed_flags: 0, changed_flags: 0,
message: "".to_string(), message: "".to_string(),
app_data: vec![], app_data: vec![],
game_key: GameKey{ game_key: GameKey{
data: Data{},
tid: 0x00050002101ce400, tid: 0x00050002101ce400,
version: 0x0 version: 0x0
}, },
@ -144,7 +193,66 @@ impl Friends for FriendsUser {
unk7: 0 unk7: 0
}, },
unk: 0 unk: 0
}], }];
println!("acquiring user and current friends locks");
let users = self.fm.users.read().await;
println!("started summing users");
for u in users.deref().iter().filter_map(|u| u.upgrade()) {
let data = u.data.read().await;
let Some(inner_data) = data.as_ref() else {
continue;
};
fr_list.push(friend_info_from_user(&inner_data));
drop(data);
let mut curr_friends = self.current_friends.write().await;
curr_friends.push(u.pid);
drop(curr_friends);
let mut fr = u.current_friends.write().await;
if !fr.contains(&self.pid) {
fr.push(self.pid);
drop(fr);
let data = any_self_fr_info.clone();
let u = u.clone();
let sender = self.pid;
spawn(async move {
u.remote
.process_nintendo_notification_event_1(NintendoNotificationEvent {
event_type: 30,
sender,
data,
})
.await;
});
} else {
drop(fr);
}
}
println!("finished summing users");
drop(users);
println!("adding self to users");
let mut users = self.fm.users.write().await;
users.push(self.this.clone());
drop(users);
println!("done...");
Ok((
PrincipalPreference {
data: Data {},
block_friend_request: false,
show_online: false,
show_playing_title: false,
},
Comment {
data: Data {},
last_changed: KerberosDateTime::now(),
message: "".to_string(),
unk: 0,
},
fr_list,
vec![], vec![],
vec![], vec![],
vec![], vec![],
@ -154,6 +262,39 @@ impl Friends for FriendsUser {
)) ))
} }
async fn update_presence(&self, presence: NintendoPresenceV2) -> Result<(), ErrorCode> {
let mut data = self.data.write().await;
let Some(inner_data) = data.as_mut() else {
return Err(ErrorCode::RendezVous_PermissionDenied);
};
inner_data.presence = presence;
let Ok(any_self_fr_info) = Any::new(&inner_data.presence) else {
return Err(ErrorCode::RendezVous_ControlScriptFailure);
};
drop(data);
let users = self.fm.users.read().await;
for u in users.deref().iter().filter_map(|u| u.upgrade()) {
u.remote
.process_nintendo_notification_event_2(NintendoNotificationEvent {
event_type: 24,
sender: self.pid,
data: any_self_fr_info.clone(),
})
.await;
}
drop(users);
Ok(())
}
async fn delete_persistent_notification(
&self,
notifs: Vec<PersistentNotification>,
) -> Result<(), ErrorCode> {
Ok(())
}
async fn check_setting_status(&self) -> Result<u8, ErrorCode> { async fn check_setting_status(&self) -> Result<u8, ErrorCode> {
Ok(0xFF) Ok(0xFF)
} }

View file

@ -254,10 +254,10 @@ impl ExtendedMatchmakeSession {
let Some(old_conns) = old_conns.upgrade() else { let Some(old_conns) = old_conns.upgrade() else {
continue; continue;
}; };
if old_conns.pid != self.session.gathering.host_pid{ /*if old_conns.pid != self.session.gathering.host_pid {
continue; continue;
} }*/
for new_conn_pid in conns.iter().filter_map(Weak::upgrade).map(|c| c.pid){ for new_conn_pid in conns.iter().filter_map(Weak::upgrade).map(|c| c.pid) {
old_conns old_conns
.remote .remote
.process_notification_event(NotificationEvent { .process_notification_event(NotificationEvent {
@ -272,6 +272,7 @@ impl ExtendedMatchmakeSession {
} }
} }
} }
pub fn has_active_players(&self) -> bool { pub fn has_active_players(&self) -> bool {
self.connected_players self.connected_players
.iter() .iter()

View file

@ -5,15 +5,12 @@ use crate::nex::remote_console::RemoteConsole;
use crate::rmc::protocols::matchmake::{ use crate::rmc::protocols::matchmake::{
Matchmake, RawMatchmake, RawMatchmakeInfo, RemoteMatchmake, Matchmake, RawMatchmake, RawMatchmakeInfo, RemoteMatchmake,
}; };
use serde::{Serialize, Deserialize};
use std::env;
use std::str::FromStr;
use crate::rmc::protocols::nat_traversal::{ use crate::rmc::protocols::nat_traversal::{
NatTraversal, RawNatTraversal, RawNatTraversalInfo, RemoteNatTraversal, NatTraversal, RawNatTraversal, RawNatTraversalInfo, RemoteNatTraversal,
RemoteNatTraversalConsole, RemoteNatTraversalConsole,
}; };
use rnex_core::kerberos::KerberosDateTime;
use rnex_core::PID; use rnex_core::PID;
use rnex_core::kerberos::KerberosDateTime;
use rnex_core::prudp::station_url::StationUrl; use rnex_core::prudp::station_url::StationUrl;
use rnex_core::prudp::station_url::UrlOptions::{ use rnex_core::prudp::station_url::UrlOptions::{
Address, NatFiltering, NatMapping, Port, RVConnectionID, Address, NatFiltering, NatMapping, Port, RVConnectionID,
@ -31,17 +28,22 @@ use rnex_core::rmc::structures::any::Any;
use rnex_core::rmc::structures::matchmake::{ use rnex_core::rmc::structures::matchmake::{
AutoMatchmakeParam, CreateMatchmakeSessionParam, JoinMatchmakeSessionParam, MatchmakeSession, AutoMatchmakeParam, CreateMatchmakeSessionParam, JoinMatchmakeSessionParam, MatchmakeSession,
}; };
use serde::{Deserialize, Serialize};
use std::env;
use std::str::FromStr;
use crate::rmc::protocols::notifications::{NotificationEvent, RemoteNotification}; use crate::rmc::protocols::notifications::{NotificationEvent, RemoteNotification};
use log::{info, error}; use log::{error, info};
use macros::rmc_struct; use macros::rmc_struct;
use rnex_core::rmc::structures::qbuffer::QBuffer;
use rnex_core::prudp::socket_addr::PRUDPSockAddr; use rnex_core::prudp::socket_addr::PRUDPSockAddr;
use rnex_core::rmc::protocols::ranking::{
CompetitionRankingGetParam, CompetitionRankingScoreData, CompetitionRankingScoreInfo,
};
use rnex_core::rmc::response::ErrorCode::{Core_InvalidArgument, RendezVous_AccountExpired}; use rnex_core::rmc::response::ErrorCode::{Core_InvalidArgument, RendezVous_AccountExpired};
use rnex_core::rmc::structures::qbuffer::QBuffer;
use rnex_core::rmc::structures::qresult::QResult; use rnex_core::rmc::structures::qresult::QResult;
use rnex_core::rmc::structures::ranking::UploadCompetitionData;
use std::sync::{Arc, Weak}; use std::sync::{Arc, Weak};
use rnex_core::rmc::protocols::ranking::{CompetitionRankingScoreData, CompetitionRankingGetParam, CompetitionRankingScoreInfo};
use rnex_core::rmc::structures::ranking::{UploadCompetitionData};
use tokio::sync::{Mutex, RwLock}; use tokio::sync::{Mutex, RwLock};
define_rmc_proto!( define_rmc_proto!(
@ -617,10 +619,7 @@ fn fetch_team_votes(fest_id: u32) -> Result<Vec<u32>, ErrorCode> {
})?; })?;
let body = body.trim().trim_start_matches('[').trim_end_matches(']'); let body = body.trim().trim_start_matches('[').trim_end_matches(']');
let votes: Result<Vec<u32>, _> = body let votes: Result<Vec<u32>, _> = body.split(',').map(|s| u32::from_str(s.trim())).collect();
.split(',')
.map(|s| u32::from_str(s.trim()))
.collect();
votes.map_err(|e| { votes.map_err(|e| {
error!("failed to parse votes: {:?}", e); error!("failed to parse votes: {:?}", e);
@ -635,8 +634,7 @@ impl Ranking for User {
) -> Result<Vec<CompetitionRankingScoreInfo>, ErrorCode> { ) -> Result<Vec<CompetitionRankingScoreInfo>, ErrorCode> {
let fest_id = param.festival_ids.get(0).copied().unwrap_or(0); let fest_id = param.festival_ids.get(0).copied().unwrap_or(0);
let endpoint_results = env::var("RNEX_SPLATOON_RESULTS_GET") let endpoint_results = env::var("RNEX_SPLATOON_RESULTS_GET").map_err(|_| {
.map_err(|_| {
error!("RNEX_SPLATOON_RESULTS_GET not set"); error!("RNEX_SPLATOON_RESULTS_GET not set");
ErrorCode::RendezVous_InvalidConfiguration ErrorCode::RendezVous_InvalidConfiguration
})?; })?;
@ -645,10 +643,7 @@ impl Ranking for User {
let response_results = ureq::get(&url_results).call(); let response_results = ureq::get(&url_results).call();
let results: Vec<CompetitionPostResults> = match response_results { let results: Vec<CompetitionPostResults> = match response_results {
Ok(mut res) => res Ok(mut res) => res.body_mut().read_json().map_err(|e| {
.body_mut()
.read_json()
.map_err(|e| {
error!("failed to parse JSON: {:?}", e); error!("failed to parse JSON: {:?}", e);
ErrorCode::RendezVous_InvalidConfiguration ErrorCode::RendezVous_InvalidConfiguration
})?, })?,
@ -661,9 +656,10 @@ impl Ranking for User {
let team_votes = fetch_team_votes(fest_id)?; let team_votes = fetch_team_votes(fest_id)?;
let mut wins = vec![0u32, 0u32]; let mut wins = vec![0u32, 0u32];
for r in &results { for r in &results {
if r.team_win == 1 && (r.team_id == 0 || r.team_id == 1) { let won_team = r.team_id ^ (!r.team_win);
wins[r.team_id as usize] += 1; if let Some(team) = wins.get_mut(won_team as usize) {
} *team += 1
};
} }
let score_data: Vec<CompetitionRankingScoreData> = results let score_data: Vec<CompetitionRankingScoreData> = results
@ -689,7 +685,10 @@ impl Ranking for User {
Ok(vec![info]) Ok(vec![info])
} }
async fn upload_competition_ranking_score(&self, param: UploadCompetitionData) -> Result<bool, ErrorCode> { async fn upload_competition_ranking_score(
&self,
param: UploadCompetitionData,
) -> Result<bool, ErrorCode> {
info!("fest results for user {:?}:", self.pid); info!("fest results for user {:?}:", self.pid);
info!("fest id: {:?}", param.splatfest_id); info!("fest id: {:?}", param.splatfest_id);
info!("score: {:?}", param.score); info!("score: {:?}", param.score);
@ -736,4 +735,3 @@ impl Ranking for User {
Ok(true) Ok(true)
} }
} }

View file

@ -2,9 +2,13 @@ use macros::{RmcSerialize, method_id, rmc_proto};
use rnex_core::{kerberos::KerberosDateTime, rmc::response::ErrorCode}; use rnex_core::{kerberos::KerberosDateTime, rmc::response::ErrorCode};
use rnex_core::rmc::structures::{data::Data, rmc_struct};
#[derive(RmcSerialize, Debug, Clone)] #[derive(RmcSerialize, Debug, Clone)]
#[rmc_struct(0)] #[rmc_struct(0)]
pub struct MiiV2 { pub struct MiiV2 {
#[extends]
pub data: Data,
pub name: String, pub name: String,
pub unk: u8, pub unk: u8,
pub unk2: u8, pub unk2: u8,
@ -15,6 +19,8 @@ pub struct MiiV2 {
#[derive(RmcSerialize, Debug, Clone)] #[derive(RmcSerialize, Debug, Clone)]
#[rmc_struct(0)] #[rmc_struct(0)]
pub struct PrincipalBasicInfo { pub struct PrincipalBasicInfo {
#[extends]
pub data: Data,
pub pid: u32, pub pid: u32,
pub nnid: String, pub nnid: String,
pub mii: MiiV2, pub mii: MiiV2,
@ -24,21 +30,27 @@ pub struct PrincipalBasicInfo {
#[derive(RmcSerialize, Debug, Clone)] #[derive(RmcSerialize, Debug, Clone)]
#[rmc_struct(0)] #[rmc_struct(0)]
pub struct NNAInfo { pub struct NNAInfo {
#[extends]
pub data: Data,
pub principal_basic_info: PrincipalBasicInfo, pub principal_basic_info: PrincipalBasicInfo,
pub unk: u8, pub unk: u8,
pub unk2: u8, pub unk2: u8,
} }
#[derive(RmcSerialize)] #[derive(RmcSerialize, Clone, Copy, Debug)]
#[rmc_struct(0)] #[rmc_struct(0)]
pub struct GameKey { pub struct GameKey {
#[extends]
pub data: Data,
pub tid: u64, pub tid: u64,
pub version: u16, pub version: u16,
} }
#[derive(RmcSerialize)] #[derive(RmcSerialize, Clone, Debug)]
#[rmc_struct(0)] #[rmc_struct(0)]
pub struct NintendoPresenceV2 { pub struct NintendoPresenceV2 {
#[extends]
pub data: Data,
pub changed_flags: u32, pub changed_flags: u32,
pub is_online: bool, pub is_online: bool,
pub game_key: GameKey, pub game_key: GameKey,
@ -58,6 +70,8 @@ pub struct NintendoPresenceV2 {
#[derive(RmcSerialize)] #[derive(RmcSerialize)]
#[rmc_struct(0)] #[rmc_struct(0)]
pub struct PrincipalPreference { pub struct PrincipalPreference {
#[extends]
pub data: Data,
pub show_online: bool, pub show_online: bool,
pub show_playing_title: bool, pub show_playing_title: bool,
pub block_friend_request: bool, pub block_friend_request: bool,
@ -66,6 +80,8 @@ pub struct PrincipalPreference {
#[derive(RmcSerialize)] #[derive(RmcSerialize)]
#[rmc_struct(0)] #[rmc_struct(0)]
pub struct Comment { pub struct Comment {
#[extends]
pub data: Data,
pub unk: u8, pub unk: u8,
pub message: String, pub message: String,
pub last_changed: KerberosDateTime, pub last_changed: KerberosDateTime,
@ -74,6 +90,8 @@ pub struct Comment {
#[derive(RmcSerialize)] #[derive(RmcSerialize)]
#[rmc_struct(0)] #[rmc_struct(0)]
pub struct FriendInfo { pub struct FriendInfo {
#[extends]
pub data: Data,
pub nna_info: NNAInfo, pub nna_info: NNAInfo,
pub presence: NintendoPresenceV2, pub presence: NintendoPresenceV2,
pub comment: Comment, pub comment: Comment,
@ -85,6 +103,8 @@ pub struct FriendInfo {
#[derive(RmcSerialize)] #[derive(RmcSerialize)]
#[rmc_struct(0)] #[rmc_struct(0)]
pub struct FriendRequestMessage { pub struct FriendRequestMessage {
#[extends]
pub data: Data,
pub friend_request_id: u64, pub friend_request_id: u64,
pub is_recieved: u8, pub is_recieved: u8,
pub unk: u8, pub unk: u8,
@ -99,6 +119,8 @@ pub struct FriendRequestMessage {
#[derive(RmcSerialize)] #[derive(RmcSerialize)]
#[rmc_struct(0)] #[rmc_struct(0)]
pub struct FriendRequest { pub struct FriendRequest {
#[extends]
pub data: Data,
pub basic_info: PrincipalBasicInfo, pub basic_info: PrincipalBasicInfo,
pub request_message: FriendRequestMessage, pub request_message: FriendRequestMessage,
pub sent_on: KerberosDateTime, pub sent_on: KerberosDateTime,
@ -107,6 +129,8 @@ pub struct FriendRequest {
#[derive(RmcSerialize)] #[derive(RmcSerialize)]
#[rmc_struct(0)] #[rmc_struct(0)]
pub struct BlacklistedPrincipal { pub struct BlacklistedPrincipal {
#[extends]
pub data: Data,
pub basic_info: PrincipalBasicInfo, pub basic_info: PrincipalBasicInfo,
pub game_key: GameKey, pub game_key: GameKey,
pub since: KerberosDateTime, pub since: KerberosDateTime,
@ -114,6 +138,8 @@ pub struct BlacklistedPrincipal {
#[derive(RmcSerialize)] #[derive(RmcSerialize)]
#[rmc_struct(0)] #[rmc_struct(0)]
pub struct PersistentNotification { pub struct PersistentNotification {
#[extends]
pub data: Data,
pub unk1: u64, pub unk1: u64,
pub unk2: u32, pub unk2: u32,
pub unk3: u32, pub unk3: u32,
@ -143,6 +169,13 @@ pub trait Friends {
), ),
ErrorCode, ErrorCode,
>; >;
#[method_id(13)]
async fn update_presence(&self, presence: NintendoPresenceV2) -> Result<(), ErrorCode>;
#[method_id(18)]
async fn delete_persistent_notification(
&self,
notifs: Vec<PersistentNotification>,
) -> Result<(), ErrorCode>;
#[method_id(19)] #[method_id(19)]
async fn check_setting_status(&self) -> Result<u8, ErrorCode>; async fn check_setting_status(&self) -> Result<u8, ErrorCode>;
} }

View file

@ -7,6 +7,7 @@ pub mod matchmake;
pub mod matchmake_ext; pub mod matchmake_ext;
pub mod matchmake_extension; pub mod matchmake_extension;
pub mod nat_traversal; pub mod nat_traversal;
pub mod nintendo_notification;
pub mod notifications; pub mod notifications;
pub mod ranking; pub mod ranking;
pub mod secure; pub mod secure;

View file

@ -0,0 +1,39 @@
use macros::{RmcSerialize, method_id, rmc_proto};
use rnex_core::PID;
use rnex_core::rmc::structures::any::Any;
#[derive(RmcSerialize)]
#[rmc_struct(0)]
pub struct NintendoNotificationEvent {
pub event_type: u32,
pub sender: PID,
pub data: Any,
}
#[derive(RmcSerialize)]
#[rmc_struct(0)]
pub struct NintendoNotificationEventGeneral {
pub param1: u32,
pub param2: u64,
pub param3: u64,
pub str_param: String,
}
#[derive(RmcSerialize)]
#[rmc_struct(0)]
pub struct NintendoNotificationEventProfile {
pub region: u8,
pub country: u8,
pub area: u8,
pub language: u8,
pub platform: u8,
}
#[rmc_proto(100, NoReturn)]
pub trait NintendoNotification {
#[method_id(1)]
async fn process_nintendo_notification_event_1(&self, notif: NintendoNotificationEvent);
#[method_id(2)]
async fn process_nintendo_notification_event_2(&self, notif: NintendoNotificationEvent);
}

View file

@ -1,8 +1,8 @@
use rnex_core::rmc::structures::{Result, RmcSerialize}; use rnex_core::rmc::structures::{Result, RmcSerialize};
use std::io::{Read, Write}; use std::io::{Cursor, Read, Write};
use v_byte_helpers::{IS_BIG_ENDIAN, ReadExtensions}; use v_byte_helpers::{IS_BIG_ENDIAN, ReadExtensions};
#[derive(Debug, Default)] #[derive(Debug, Default, Clone)]
pub struct Any { pub struct Any {
pub name: String, pub name: String,
pub data: Vec<u8>, pub data: Vec<u8>,
@ -13,10 +13,7 @@ impl RmcSerialize for Any {
self.name.serialize(writer)?; self.name.serialize(writer)?;
let u32_len = self.data.len() as u32; let u32_len = self.data.len() as u32;
(u32_len + 4).serialize(writer)?;
u32_len.serialize(writer)?;
u32_len.serialize(writer)?;
self.data.serialize(writer)?; self.data.serialize(writer)?;
Ok(()) Ok(())
@ -26,12 +23,23 @@ impl RmcSerialize for Any {
// also length ? // also length ?
let _len2: u32 = reader.read_struct(IS_BIG_ENDIAN)?; let _len2: u32 = reader.read_struct(IS_BIG_ENDIAN)?;
let length: u32 = reader.read_struct(IS_BIG_ENDIAN)?; let data = Vec::deserialize(reader)?;
let mut data = vec![0; length as usize];
reader.read_exact(&mut data)?;
Ok(Any { name, data }) Ok(Any { name, data })
} }
} }
impl Any {
pub fn try_get<T: RmcSerialize>(&self) -> Option<Result<T>> {
if self.name != T::name() {
return None;
}
return Some(T::deserialize(&mut Cursor::new(&self.data[..])));
}
pub fn new<T: RmcSerialize>(val: &T) -> Result<Self> {
return Ok(Self {
name: T::name().to_owned(),
data: val.to_data()?,
});
}
}

View file

@ -0,0 +1,5 @@
use macros::RmcSerialize;
#[derive(RmcSerialize, Debug, Clone, Copy)]
#[rmc_struct(0)]
pub struct Data {}

View file

@ -26,6 +26,7 @@ pub type Result<T> = std::result::Result<T, Error>;
pub mod any; pub mod any;
pub mod buffer; pub mod buffer;
pub mod connection_data; pub mod connection_data;
pub mod data;
pub mod helpers; pub mod helpers;
pub mod list; pub mod list;
pub mod matchmake; pub mod matchmake;
@ -61,6 +62,9 @@ pub trait RmcSerialize {
Ok(data) Ok(data)
} }
fn name() -> &'static str {
"NoNameSpecified"
}
} }
impl RmcSerialize for () { impl RmcSerialize for () {