Compare commits
1 Commits
main
...
testing/vu
| Author | SHA1 | Date |
|---|---|---|
Ian Chamberlain
|
892a29b039 | 10 months ago |
22 changed files with 193 additions and 2095 deletions
@ -1,301 +0,0 @@
@@ -1,301 +0,0 @@
|
||||
//! Local networking example.
|
||||
//!
|
||||
//! This example showcases local networking using the UDS module.
|
||||
|
||||
use ctru::prelude::*; |
||||
use ctru::services::uds::*; |
||||
|
||||
fn handle_status_event(uds: &Uds, prev_node_mask: u16) -> ctru::Result<u16> { |
||||
println!("Connection status event signalled"); |
||||
let status = uds.connection_status()?; |
||||
println!("Status: {status:#02X?}"); |
||||
let left = prev_node_mask & (status.node_bitmask() ^ prev_node_mask); |
||||
let joined = status.node_bitmask() & (status.node_bitmask() ^ prev_node_mask); |
||||
for i in 0..16 { |
||||
if left & (1 << i) != 0 { |
||||
println!("Node {} disconnected", i + 1); |
||||
} |
||||
} |
||||
for i in 0..16 { |
||||
if joined & (1 << i) != 0 { |
||||
println!( |
||||
"Node {} connected: {:?}", |
||||
i + 1, |
||||
uds.node_info(NodeID::Node(i + 1)) |
||||
); |
||||
} |
||||
} |
||||
Ok(status.node_bitmask()) |
||||
} |
||||
|
||||
fn main() -> Result<(), Error> { |
||||
let apt = Apt::new().unwrap(); |
||||
let mut hid = Hid::new().unwrap(); |
||||
let gfx = Gfx::new().unwrap(); |
||||
let console = Console::new(gfx.top_screen.borrow_mut()); |
||||
|
||||
println!("Local networking demo"); |
||||
|
||||
let mut uds = Uds::new(None).unwrap(); |
||||
|
||||
println!("UDS initialised"); |
||||
|
||||
enum State { |
||||
Initialised, |
||||
Scanning, |
||||
DrawList, |
||||
List, |
||||
Connect, |
||||
Connected, |
||||
Create, |
||||
Created, |
||||
} |
||||
|
||||
let mut state = State::Initialised; |
||||
|
||||
println!("Press A to start scanning or B to create a new network"); |
||||
|
||||
let mut networks = vec![]; |
||||
let mut selected_network = 0; |
||||
|
||||
let mut mode = ConnectionType::Client; |
||||
|
||||
let mut channel = 0; |
||||
let data_channel = 1; |
||||
|
||||
let mut prev_node_mask = 0; |
||||
|
||||
while apt.main_loop() { |
||||
gfx.wait_for_vblank(); |
||||
|
||||
hid.scan_input(); |
||||
if hid.keys_down().contains(KeyPad::START) { |
||||
break; |
||||
} |
||||
|
||||
match state { |
||||
State::Initialised => { |
||||
if hid.keys_down().contains(KeyPad::A) { |
||||
state = State::Scanning; |
||||
console.clear(); |
||||
prev_node_mask = 0; |
||||
} else if hid.keys_down().contains(KeyPad::B) { |
||||
state = State::Create; |
||||
console.clear(); |
||||
prev_node_mask = 0; |
||||
} |
||||
} |
||||
State::Scanning => { |
||||
println!("Scanning..."); |
||||
|
||||
let nwks = uds.scan(b"HBW\x10", None, None); |
||||
|
||||
match nwks { |
||||
Ok(n) => { |
||||
if n.is_empty() { |
||||
state = State::Initialised; |
||||
console.clear(); |
||||
println!("Scanned successfully; no networks found"); |
||||
println!("Press A to start scanning or B to create a new network"); |
||||
} else { |
||||
networks = n; |
||||
selected_network = 0; |
||||
state = State::DrawList; |
||||
} |
||||
} |
||||
Err(e) => { |
||||
state = State::Initialised; |
||||
console.clear(); |
||||
eprintln!("Error while scanning: {e}"); |
||||
println!("Press A to start scanning or B to create a new network"); |
||||
} |
||||
} |
||||
} |
||||
State::DrawList => { |
||||
console.clear(); |
||||
|
||||
println!( |
||||
"Scanned successfully; {} network{} found", |
||||
networks.len(), |
||||
if networks.len() == 1 { "" } else { "s" } |
||||
); |
||||
|
||||
println!("D-Pad to select, A to connect as client, R + A to connect as spectator, B to create a new network"); |
||||
|
||||
for (index, n) in networks.iter().enumerate() { |
||||
println!( |
||||
"{} Username: {}", |
||||
if index == selected_network { ">" } else { " " }, |
||||
n.nodes()[0].unwrap().username() |
||||
); |
||||
} |
||||
|
||||
state = State::List; |
||||
} |
||||
State::List => { |
||||
if hid.keys_down().contains(KeyPad::UP) && selected_network > 0 { |
||||
selected_network -= 1; |
||||
state = State::DrawList; |
||||
} else if hid.keys_down().contains(KeyPad::DOWN) |
||||
&& selected_network < networks.len() - 1 |
||||
{ |
||||
selected_network += 1; |
||||
state = State::DrawList; |
||||
} else if hid.keys_down().contains(KeyPad::A) { |
||||
state = State::Connect; |
||||
mode = if hid.keys_held().contains(KeyPad::R) { |
||||
ConnectionType::Spectator |
||||
} else { |
||||
ConnectionType::Client |
||||
}; |
||||
} else if hid.keys_down().contains(KeyPad::B) { |
||||
state = State::Create; |
||||
} |
||||
} |
||||
State::Connect => { |
||||
let appdata = uds.network_appdata(&networks[selected_network], None)?; |
||||
println!("App data: {:02X?}", appdata); |
||||
|
||||
if let Err(e) = uds.connect_network( |
||||
&networks[selected_network], |
||||
b"udsdemo passphrase c186093cd2652741\0", |
||||
mode, |
||||
data_channel, |
||||
) { |
||||
console.clear(); |
||||
eprintln!("Error while connecting to network: {e}"); |
||||
state = State::Initialised; |
||||
println!("Press A to start scanning or B to create a new network"); |
||||
} else { |
||||
channel = uds.channel()?; |
||||
println!("Connected using channel {}", channel); |
||||
|
||||
let appdata = uds.appdata(None)?; |
||||
println!("App data: {:02X?}", appdata); |
||||
|
||||
if uds.wait_status_event(false, false)? { |
||||
prev_node_mask = handle_status_event(&uds, prev_node_mask)?; |
||||
} |
||||
|
||||
println!("Press A to stop data transfer"); |
||||
state = State::Connected; |
||||
} |
||||
} |
||||
State::Connected => { |
||||
let packet = uds.pull_packet(); |
||||
|
||||
match packet { |
||||
Ok(p) => { |
||||
if let Some((pkt, node)) = p { |
||||
println!( |
||||
"{:02X}{:02X}{:02X}{:02X} from {:?}", |
||||
pkt[0], pkt[1], pkt[2], pkt[3], node |
||||
); |
||||
} |
||||
|
||||
if uds.wait_status_event(false, false)? { |
||||
prev_node_mask = handle_status_event(&uds, prev_node_mask)?; |
||||
} |
||||
|
||||
if hid.keys_down().contains(KeyPad::A) { |
||||
uds.disconnect_network()?; |
||||
state = State::Initialised; |
||||
console.clear(); |
||||
println!("Press A to start scanning or B to create a new network"); |
||||
} else if !hid.keys_down().is_empty() || !hid.keys_up().is_empty() { |
||||
let transfer_data = hid.keys_held().bits(); |
||||
if mode != ConnectionType::Spectator { |
||||
uds.send_packet( |
||||
&transfer_data.to_le_bytes(), |
||||
NodeID::Broadcast, |
||||
data_channel, |
||||
SendFlags::Default, |
||||
)?; |
||||
} |
||||
} |
||||
} |
||||
Err(e) => { |
||||
uds.disconnect_network()?; |
||||
console.clear(); |
||||
eprintln!("Error while grabbing packet from network: {e}"); |
||||
state = State::Initialised; |
||||
println!("Press A to start scanning or B to create a new network"); |
||||
} |
||||
} |
||||
} |
||||
State::Create => { |
||||
console.clear(); |
||||
println!("Creating network..."); |
||||
|
||||
match uds.create_network( |
||||
b"HBW\x10", |
||||
None, |
||||
None, |
||||
b"udsdemo passphrase c186093cd2652741\0", |
||||
data_channel, |
||||
) { |
||||
Ok(_) => { |
||||
let appdata = [0x69u8, 0x8a, 0x05, 0x5c] |
||||
.into_iter() |
||||
.chain((*b"Test appdata.").into_iter()) |
||||
.chain(std::iter::repeat(0).take(3)) |
||||
.collect::<Vec<_>>(); |
||||
|
||||
uds.set_appdata(&appdata)?; |
||||
|
||||
println!("Press A to stop data transfer"); |
||||
state = State::Created; |
||||
} |
||||
Err(e) => { |
||||
console.clear(); |
||||
eprintln!("Error while creating network: {e}"); |
||||
state = State::Initialised; |
||||
println!("Press A to start scanning or B to create a new network"); |
||||
} |
||||
} |
||||
} |
||||
State::Created => { |
||||
let packet = uds.pull_packet(); |
||||
|
||||
match packet { |
||||
Ok(p) => { |
||||
if let Some((pkt, node)) = p { |
||||
println!( |
||||
"{:02X}{:02X}{:02X}{:02X} from {:?}", |
||||
pkt[0], pkt[1], pkt[2], pkt[3], node |
||||
); |
||||
} |
||||
|
||||
if uds.wait_status_event(false, false)? { |
||||
prev_node_mask = handle_status_event(&uds, prev_node_mask)?; |
||||
} |
||||
|
||||
if hid.keys_down().contains(KeyPad::A) { |
||||
uds.destroy_network()?; |
||||
state = State::Initialised; |
||||
console.clear(); |
||||
println!("Press A to start scanning or B to create a new network"); |
||||
} else if !hid.keys_down().is_empty() || !hid.keys_up().is_empty() { |
||||
let transfer_data = hid.keys_held().bits(); |
||||
uds.send_packet( |
||||
&transfer_data.to_le_bytes(), |
||||
NodeID::Broadcast, |
||||
data_channel, |
||||
SendFlags::Default, |
||||
)?; |
||||
} |
||||
} |
||||
Err(e) => { |
||||
uds.destroy_network()?; |
||||
console.clear(); |
||||
eprintln!("Error while grabbing packet from network: {e}"); |
||||
state = State::Initialised; |
||||
println!("Press A to start scanning or B to create a new network"); |
||||
} |
||||
} |
||||
} |
||||
} |
||||
} |
||||
|
||||
Ok(()) |
||||
} |
||||
@ -1,153 +0,0 @@
@@ -1,153 +0,0 @@
|
||||
//! Error applet.
|
||||
//!
|
||||
//! This applet displays error text as a pop-up message on the lower screen.
|
||||
|
||||
use crate::services::{apt::Apt, gfx::Gfx}; |
||||
|
||||
use ctru_sys::errorConf; |
||||
|
||||
/// Configuration struct to set up the Error applet.
|
||||
#[doc(alias = "errorConf")] |
||||
pub struct PopUp { |
||||
state: Box<errorConf>, |
||||
} |
||||
|
||||
/// Determines whether the Error applet will use word wrapping when displaying a message.
|
||||
#[doc(alias = "errorType")] |
||||
#[derive(Copy, Clone, Debug, PartialEq, Eq)] |
||||
#[repr(u32)] |
||||
pub enum WordWrap { |
||||
/// Error text is centered in the error applet window and does not use word wrapping.
|
||||
Disabled = ctru_sys::ERROR_TEXT, |
||||
/// Error text starts at the top of the error applet window and uses word wrapping.
|
||||
Enabled = ctru_sys::ERROR_TEXT_WORD_WRAP, |
||||
} |
||||
|
||||
/// Error returned by an unsuccessful [`PopUp::launch()`].
|
||||
#[doc(alias = "errorReturnCode")] |
||||
#[derive(Copy, Clone, Debug, PartialEq, Eq)] |
||||
#[repr(i8)] |
||||
pub enum Error { |
||||
/// Unknown error occurred.
|
||||
Unknown = ctru_sys::ERROR_UNKNOWN, |
||||
/// Operation not supported.
|
||||
NotSupported = ctru_sys::ERROR_NOT_SUPPORTED, |
||||
/// Home button pressed while [`PopUp`] was running.
|
||||
HomePressed = ctru_sys::ERROR_HOME_BUTTON, |
||||
/// Power button pressed while [`PopUp`] was running.
|
||||
PowerPressed = ctru_sys::ERROR_POWER_BUTTON, |
||||
/// Reset button pressed while [`PopUp`] was running.
|
||||
ResetPressed = ctru_sys::ERROR_SOFTWARE_RESET, |
||||
} |
||||
|
||||
impl PopUp { |
||||
/// Initializes the error applet with the provided word wrap setting.
|
||||
#[doc(alias = "errorInit")] |
||||
pub fn new(word_wrap: WordWrap) -> Self { |
||||
let mut state = Box::<errorConf>::default(); |
||||
|
||||
unsafe { ctru_sys::errorInit(state.as_mut(), word_wrap as _, 0) }; |
||||
|
||||
Self { state } |
||||
} |
||||
|
||||
/// Sets the error text to display.
|
||||
///
|
||||
/// # Notes
|
||||
///
|
||||
/// The text will be converted to UTF-16 for display with the applet, and the message will be truncated if it exceeds
|
||||
/// 1900 UTF-16 code units in length after conversion.
|
||||
#[doc(alias = "errorText")] |
||||
pub fn set_text(&mut self, text: &str) { |
||||
for (idx, code_unit) in text |
||||
.encode_utf16() |
||||
.take(self.state.Text.len() - 1) |
||||
.chain(std::iter::once(0)) |
||||
.enumerate() |
||||
{ |
||||
self.state.Text[idx] = code_unit; |
||||
} |
||||
} |
||||
|
||||
/// Launches the error applet.
|
||||
#[doc(alias = "errorDisp")] |
||||
pub fn launch(&mut self, _apt: &Apt, _gfx: &Gfx) -> Result<(), Error> { |
||||
unsafe { self.launch_unchecked() } |
||||
} |
||||
|
||||
/// Launches the error applet without requiring an [`Apt`] or [`Gfx`] handle.
|
||||
///
|
||||
/// # Safety
|
||||
///
|
||||
/// Potentially leads to undefined behavior if the aforementioned services are not actually active when the applet launches.
|
||||
unsafe fn launch_unchecked(&mut self) -> Result<(), Error> { |
||||
unsafe { ctru_sys::errorDisp(self.state.as_mut()) }; |
||||
|
||||
match self.state.returnCode { |
||||
ctru_sys::ERROR_NONE | ctru_sys::ERROR_SUCCESS => Ok(()), |
||||
ctru_sys::ERROR_NOT_SUPPORTED => Err(Error::NotSupported), |
||||
ctru_sys::ERROR_HOME_BUTTON => Err(Error::HomePressed), |
||||
ctru_sys::ERROR_POWER_BUTTON => Err(Error::PowerPressed), |
||||
ctru_sys::ERROR_SOFTWARE_RESET => Err(Error::ResetPressed), |
||||
_ => Err(Error::Unknown), |
||||
} |
||||
} |
||||
} |
||||
|
||||
/// Sets a custom [panic hook](https://doc.rust-lang.org/std/panic/fn.set_hook.html) that uses the error applet to display panic messages.
|
||||
///
|
||||
/// You can also choose to have the previously registered panic hook called along with the error applet popup, which can be useful
|
||||
/// if you want to use output redirection to display panic messages over `3dslink` or `GDB`.
|
||||
///
|
||||
/// You can use [`std::panic::take_hook`](https://doc.rust-lang.org/std/panic/fn.take_hook.html) to unregister the panic hook
|
||||
/// set by this function.
|
||||
///
|
||||
/// # Notes
|
||||
///
|
||||
/// * If the [`Gfx`] service is not initialized during a panic, the error applet will not be displayed and the old panic hook will be called.
|
||||
pub fn set_panic_hook(call_old_hook: bool) { |
||||
use crate::services::gfx::GFX_ACTIVE; |
||||
use std::sync::TryLockError; |
||||
|
||||
let old_hook = std::panic::take_hook(); |
||||
|
||||
std::panic::set_hook(Box::new(move |panic_info| { |
||||
// If we get a `WouldBlock` error, we know that the `Gfx` service has been initialized.
|
||||
// Otherwise fallback to using the old panic hook.
|
||||
if let (Err(TryLockError::WouldBlock), Ok(_apt)) = (GFX_ACTIVE.try_lock(), Apt::new()) { |
||||
if call_old_hook { |
||||
old_hook(panic_info); |
||||
} |
||||
|
||||
let thread = std::thread::current(); |
||||
|
||||
let name = thread.name().unwrap_or("<unnamed>"); |
||||
|
||||
let message = format!("thread '{name}' {panic_info}"); |
||||
|
||||
let mut popup = PopUp::new(WordWrap::Enabled); |
||||
|
||||
popup.set_text(&message); |
||||
|
||||
unsafe { |
||||
let _ = popup.launch_unchecked(); |
||||
} |
||||
} else { |
||||
old_hook(panic_info); |
||||
} |
||||
})); |
||||
} |
||||
|
||||
impl std::fmt::Display for Error { |
||||
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result { |
||||
match self { |
||||
Self::NotSupported => write!(f, "operation not supported"), |
||||
Self::HomePressed => write!(f, "home button pressed while error applet was running"), |
||||
Self::PowerPressed => write!(f, "power button pressed while error applet was running"), |
||||
Self::ResetPressed => write!(f, "reset button pressed while error applet was running"), |
||||
Self::Unknown => write!(f, "an unknown error occurred"), |
||||
} |
||||
} |
||||
} |
||||
|
||||
impl std::error::Error for Error {} |
||||
Loading…
Reference in new issue