summaryrefslogtreecommitdiff
path: root/src/WebRTCPeer.hpp
blob: 45a90ec0767cfcec9da56de85a7b27adfc553790 (plain)
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
118
119
120
121
#ifndef WEBRTC_PEER_H
#define WEBRTC_PEER_H

#include "api/peerconnectioninterface.h" // interface for all things needed from WebRTC
#include "media/base/mediaengine.h" // needed for CreateModularPeerConnectionFactory
#ifdef interface
#pragma message("\n 'interface' is defined \n")
#undef interface
#endif // interface

#include <functional> // std::function
#include <mutex> // mutex @TODO replace std::mutex with Godot mutex

#include "net/WebRTCPeerNative.hpp"
#include <Godot.hpp>

namespace godot_webrtc {

class WebRTCPeer : public WebRTCPeerNative {
	GODOT_CLASS(WebRTCPeer, WebRTCPeerNative);

public:
	static void _register_methods();

	void _init();

	void set_write_mode(godot_int mode);
	godot_int get_write_mode() const;
	bool was_string_packet() const;
	godot_int get_connection_state() const;

	godot_error create_offer();
	godot_error set_remote_description(const char *type, const char *sdp);
	godot_error set_local_description(const char *type, const char *sdp);
	godot_error add_ice_candidate(const char *sdpMidName, int sdpMlineIndexName, const char *sdpName);
	godot_error poll();

	/* WebRTCPeer */
	virtual godot_error get_packet(const uint8_t **r_buffer, int &r_len);
	virtual godot_error put_packet(const uint8_t *p_buffer, int p_len);
	virtual godot_int get_available_packet_count() const;
	virtual godot_int get_max_packet_size() const;

	WebRTCPeer();
	~WebRTCPeer();

	/* helper functions */

	void queue_signal(godot::String p_name, int p_argc, const godot::Variant &p_arg1 = godot::Variant(), const godot::Variant &p_arg2 = godot::Variant(), const godot::Variant &p_arg3 = godot::Variant());
	// void queue_signal(godot::StringName p_name, Variant_ARG_LIST);
	void queue_packet(uint8_t *, int);
	godot_error set_description(const char *type, const char *sdp, bool isLocal);

	/** DataChannelObserver callback functions **/
	class GodotDCO : public webrtc::DataChannelObserver {
	public:
		WebRTCPeer *parent;

		GodotDCO(WebRTCPeer *parent);
		void OnMessage(const webrtc::DataBuffer &buffer) override;
		void OnStateChange() override; // UNUSED
		void OnBufferedAmountChange(uint64_t previous_amount) override; // UNUSED
	};

	/** PeerConnectionObserver callback functions **/
	class GodotPCO : public webrtc::PeerConnectionObserver {
	public:
		WebRTCPeer *parent;

		GodotPCO(WebRTCPeer *parent);
		void OnSignalingChange(webrtc::PeerConnectionInterface::SignalingState new_state) override;
		void OnAddStream(rtc::scoped_refptr<webrtc::MediaStreamInterface> stream) override;
		void OnRemoveStream(rtc::scoped_refptr<webrtc::MediaStreamInterface> stream) override;
		void OnDataChannel(rtc::scoped_refptr<webrtc::DataChannelInterface> data_channel) override;
		void OnRenegotiationNeeded() override;
		void OnIceConnectionChange(webrtc::PeerConnectionInterface::IceConnectionState new_state) override;
		void OnIceGatheringChange(webrtc::PeerConnectionInterface::IceGatheringState new_state) override;
		void OnIceCandidate(const webrtc::IceCandidateInterface *candidate) override;
	};

	/** CreateSessionDescriptionObserver callback functions **/
	class GodotCSDO : public webrtc::CreateSessionDescriptionObserver {
	public:
		WebRTCPeer *parent;

		GodotCSDO(WebRTCPeer *parent);
		void OnSuccess(webrtc::SessionDescriptionInterface *desc) override;
		void OnFailure(const std::string &error) override;
	};

	/** SetSessionDescriptionObserver callback functions **/
	class GodotSSDO : public webrtc::SetSessionDescriptionObserver {
	public:
		WebRTCPeer *parent;

		GodotSSDO(WebRTCPeer *parent);
		void OnSuccess() override;
		void OnFailure(const std::string &error) override;
	};

	GodotDCO dco;
	GodotPCO pco;
	rtc::scoped_refptr<GodotSSDO> ptr_ssdo;
	rtc::scoped_refptr<GodotCSDO> ptr_csdo;

	std::mutex *mutex_signal_queue;
	std::mutex *mutex_packet_queue;
	int packet_queue_size;
	std::queue<uint8_t *> packet_queue;
	std::queue<int> packet_sizes_queue;
	std::queue<std::function<void()> > signal_queue;

	rtc::Thread *signaling_thread;
	rtc::scoped_refptr<webrtc::PeerConnectionFactoryInterface> pc_factory;
	rtc::scoped_refptr<webrtc::PeerConnectionInterface> peer_connection;
	rtc::scoped_refptr<webrtc::DataChannelInterface> data_channel;
};

} // namespace godot_webrtc

#endif // WEBRTC_PEER_H