Fix interface binding

2 files changed, 11 insertions, 2 deletions

diff --git a/src/net/WebRTCPeerNative.cpp b/src/net/WebRTCPeerNative.cpp
index 0e32d61..be9040c 100644
--- a/src/net/WebRTCPeerNative.cpp
+++ b/src/net/WebRTCPeerNative.cpp
@@ -1,16 +1,23 @@
 #include "WebRTCPeerNative.hpp"
 
+const godot_gdnative_ext_net_api_struct *WebRTCPeerNative::_net_api = NULL;
+
+void WebRTCPeerNative::register_interface(const godot_net_webrtc_peer *p_interface) {
+	ERR_FAIL_COND(!_net_api);
+	_net_api->godot_net_bind_webrtc_peer(_owner, p_interface);
+}
+
 void WebRTCPeerNative::_register_methods() { }
 
 void WebRTCPeerNative::_init() {
 	printf("Binding PacketPeer interface");
-	godot_net_bind_webrtc_peer(_owner, &interface);
+	register_interface(&interface);
 }
 
 WebRTCPeerNative::~WebRTCPeerNative() {
 	if (_owner) {
 		printf("Unbinding PacketPeer interface");
-		godot_net_bind_webrtc_peer(_owner, NULL);
+		register_interface(NULL);
 	}
 }
 
diff --git a/src/net/WebRTCPeerNative.hpp b/src/net/WebRTCPeerNative.hpp
index 6387cb1..603ba34 100644
--- a/src/net/WebRTCPeerNative.hpp
+++ b/src/net/WebRTCPeerNative.hpp
@@ -52,8 +52,10 @@ protected:
 
 public:
 	static void _register_methods();
+	static const godot_gdnative_ext_net_api_struct *_net_api;
 
 	void _init();
+	void register_interface(const godot_net_webrtc_peer *interface);
 
 	virtual void set_write_mode(godot_int mode) = 0;
 	virtual godot_int get_write_mode() const = 0;