4ff0a5efcc
Some checks failed
Build and Test / super-mario-maker (push) Failing after 10m30s
Build and Test / fast-racing-neo (push) Failing after 11m40s
Build and Test / friends (push) Failing after 11m56s
Build and Test / wii-sports-club (push) Failing after 12m9s
Build and Test / splatoon (push) Failing after 12m14s
Build and Test / puyopuyo (push) Failing after 12m23s
Build and Test / wii-u-chat (push) Failing after 12m23s
Build and Test / mario-tennis (push) Failing after 12m23s
Build and Test / sonic-transformed (push) Failing after 12m23s
Build and Test / splatoon-testfire (push) Failing after 12m30s
Build and Test / minecraft-wiiu (push) Failing after 12m37s
103 lines
2.9 KiB
Rust
103 lines
2.9 KiB
Rust
use rnex_core::prudp::station_url::StationUrl;
|
|
use rnex_core::prudp::station_url::UrlOptions::{
|
|
Address, NatFiltering, NatMapping, NatType, Port, PrincipalID, RVConnectionID,
|
|
};
|
|
use rnex_core::prudp::station_url::nat_types::PUBLIC;
|
|
use rnex_core::rmc::response::ErrorCode::Core_Exception;
|
|
|
|
use rnex_core::prudp::socket_addr::PRUDPSockAddr;
|
|
use rnex_core::rmc::response::ErrorCode;
|
|
|
|
use rnex_core::PID;
|
|
|
|
use crate::prudp::station_url::UrlOptions::ConnectionID;
|
|
|
|
pub async fn get_station_urls(
|
|
station_urls: &[StationUrl],
|
|
addr: PRUDPSockAddr,
|
|
pid: PID,
|
|
cid: u32,
|
|
) -> Result<Vec<StationUrl>, ErrorCode> {
|
|
let mut public_station: Option<StationUrl> = None;
|
|
let mut private_station: Option<StationUrl> = None;
|
|
|
|
for station in station_urls {
|
|
let is_public = station.options.iter().any(|v| {
|
|
if let NatType(v) = v
|
|
&& *v & PUBLIC != 0
|
|
{
|
|
return true;
|
|
}
|
|
false
|
|
});
|
|
|
|
let Some(nat_filtering) = station.options.iter().find_map(|v| match v {
|
|
NatFiltering(v) => Some(v),
|
|
_ => None,
|
|
}) else {
|
|
return Err(Core_Exception);
|
|
};
|
|
|
|
let Some(nat_mapping) = station.options.iter().find_map(|v| match v {
|
|
NatMapping(v) => Some(v),
|
|
_ => None,
|
|
}) else {
|
|
return Err(Core_Exception);
|
|
};
|
|
|
|
if !is_public || (*nat_filtering == 0 && *nat_mapping == 0) {
|
|
private_station = Some(station.clone());
|
|
}
|
|
|
|
if is_public {
|
|
public_station = Some(station.clone());
|
|
}
|
|
}
|
|
|
|
let Some(mut private_station) = private_station else {
|
|
return Err(Core_Exception);
|
|
};
|
|
|
|
let mut public_station = if let Some(public_station) = public_station {
|
|
public_station
|
|
} else {
|
|
let mut public_station = private_station.clone();
|
|
|
|
public_station.options.retain(|v| {
|
|
!matches!(
|
|
v,
|
|
Address(_) | Port(_) | NatFiltering(_) | NatMapping(_) | NatType(_)
|
|
)
|
|
});
|
|
|
|
public_station
|
|
.options
|
|
.push(Address(addr.regular_socket_addr.ip()));
|
|
public_station
|
|
.options
|
|
.push(Port(addr.regular_socket_addr.port()));
|
|
public_station.options.push(NatFiltering(0));
|
|
public_station.options.push(NatMapping(0));
|
|
public_station.options.push(NatType(3));
|
|
|
|
public_station
|
|
};
|
|
|
|
let both = [&mut public_station, &mut private_station];
|
|
|
|
for station in both {
|
|
station
|
|
.options
|
|
.retain(|v| !matches!(v, PrincipalID(_) | RVConnectionID(_)));
|
|
|
|
station.options.push(PrincipalID(pid));
|
|
station.options.push(RVConnectionID(cid));
|
|
station.options.push(ConnectionID(cid));
|
|
}
|
|
|
|
Ok(vec![public_station])
|
|
}
|
|
|
|
#[track_caller]
|
|
#[cfg(feature = "database-support")]
|
|
fn test() {}
|