steamworks 0.13.0

Provides rust friendly bindings to the steamworks sdk
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117

use crate::networking_sockets::NetConnection;
use crate::networking_types::{
    AppNetConnectionEnd, NetConnectionEnd, NetConnectionEvent, NetConnectionStatusChanged,
    NetworkingConnectionState,
};
use crate::{register_callback, CallbackHandle, Inner};
use std::sync::{Arc, Weak};
use steamworks_sys as sys;
use sys::ISteamNetworkingSockets;

/// All independent connections (to a remote host) and listening sockets share the same Callback for
/// `NetConnectionStatusChangedCallback`. This function either returns the existing handle, or creates a new
/// handler.
pub(crate) fn get_or_create_connection_callback(
    inner: Arc<Inner>,
    sockets: *mut ISteamNetworkingSockets,
) -> Arc<CallbackHandle> {
    let mut network_socket_data = inner.networking_sockets_data.lock().unwrap();
    if let Some(callback) = network_socket_data.connection_callback.upgrade() {
        callback
    } else {
        let handler = ConnectionCallbackHandler {
            inner: Arc::downgrade(&inner),
            sockets,
        };
        let callback = unsafe {
            register_callback(&inner, move |event: NetConnectionStatusChanged| {
                handler.callback(event);
            })
        };

        let callback = Arc::new(callback);
        network_socket_data.connection_callback = Arc::downgrade(&callback);
        callback
    }
}

pub(crate) struct ConnectionCallbackHandler {
    inner: Weak<Inner>,
    sockets: *mut ISteamNetworkingSockets,
}

unsafe impl Send for ConnectionCallbackHandler {}
unsafe impl Sync for ConnectionCallbackHandler {}

impl ConnectionCallbackHandler {
    pub(crate) fn callback(&self, event: NetConnectionStatusChanged) {
        if let Some(socket) = event.connection_info.listen_socket() {
            self.listen_socket_callback(socket, event);
        } else {
            self.independent_connection_callback(event);
        }
    }

    fn listen_socket_callback(
        &self,
        socket_handle: sys::HSteamListenSocket,
        event: NetConnectionStatusChanged,
    ) {
        if let Some(inner) = self.inner.upgrade() {
            let data = inner.networking_sockets_data.lock().unwrap();
            if let Some((socket, sender)) = data
                .sockets
                .get(&socket_handle)
                .and_then(|(socket, sender)| socket.upgrade().map(|socket| (socket, sender)))
            {
                let connection_handle = event.connection;
                let state = event.connection_info.state().expect("invalid state");
                if let Ok(event) = event.into_listen_socket_event(socket) {
                    if let Err(_err) = sender.send(event) {
                        // If the main socket was dropped, but the inner socket still exists, reject all new connections,
                        // as there's no way to accept them.
                        if let NetworkingConnectionState::Connecting = state {
                            self.reject_connection(connection_handle);
                        }
                    }
                } else {
                    // Ignore events that couldn't be converted
                }
            }
        }
    }

    fn reject_connection(&self, connection_handle: sys::HSteamNetConnection) {
        if let Some(inner) = self.inner.upgrade() {
            NetConnection::new_internal(connection_handle, self.sockets, inner.clone()).close(
                NetConnectionEnd::App(AppNetConnectionEnd::generic_normal()).into(),
                Some("no new connections will be accepted"),
                false,
            );
        }
    }

    fn independent_connection_callback(&self, status_changed: NetConnectionStatusChanged) {
        let Some(inner) = self.inner.upgrade() else {
            return;
        };
        let data = inner.networking_sockets_data.lock().unwrap();

        let Some(sender) = data.independent_connections.get(&status_changed.connection) else {
            // received an event of a connection we did not track... this should basically never happen
            // println!("wrong connection {}; avail = {:?}",
            //    event.connection, data.independent_connections.keys().collect::<Vec<_>>());
            return;
        };
        let Ok(new_state) = status_changed.connection_info.state() else {
            return;
        };
        let event = NetConnectionEvent {
            new_state,
            old_state: status_changed.old_state,
        };
        if let Err(_r) = sender.send(event) {
            // something definitely went wrong... but realistically what can we do but simply drop the event?
        }
    }
}