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
122
123
124
125
|
class_name Server
signal received_packet(packet: Packet)
var received_packets: Dictionary # [int, Array[Packet]]
var host: String
var port: int
var peer: StreamPeerTCP = StreamPeerTCP.new()
@warning_ignore("shadowed_variable")
func _init(host: String, port: int) -> void:
self.host = host
self.port = port
func establish_connection() -> Error:
var connection_result := peer.connect_to_host(host, port)
if connection_result == Error.OK:
peer.poll()
return connection_result
func get_status() -> Error:
if peer.get_status() in [StreamPeerTCP.STATUS_CONNECTING, StreamPeerTCP.STATUS_CONNECTED]:
return Error.OK
if peer.get_status() in [StreamPeerTCP.STATUS_NONE, StreamPeerTCP.STATUS_ERROR]:
return Error.FAILED
return Error.FAILED
## Emits [signal received_packet].
func listen() -> void:
if peer.get_status() == StreamPeerTCP.Status.STATUS_CONNECTED:
var raw_packet: PackedByteArray = peer.get_partial_data(2)[1]
if raw_packet.size() > 0:
var header = raw_packet.decode_u16(0)
if Constants.PacketDB.has(header):
var packet_type = Constants.PacketDB[header]
if packet_type.BYTE_LENGTH > 0:
raw_packet += peer.get_partial_data(packet_type.BYTE_LENGTH - 2)[1]
else:
var packet_length: PackedByteArray = peer.get_data(2)[1]
raw_packet += packet_length
raw_packet += peer.get_data(packet_length.decode_u16(0) - 4)[1]
prints("Upcoming Length:", packet_length.decode_u16(0), "for =>")
var packet = packet_type.from_bytes(raw_packet)
if not received_packets.has(header):
received_packets[header] = []
received_packets[header].append(packet)
received_packet.emit(packet)
var display_header = raw_packet.slice(0, 2)
display_header.reverse()
print("Received known packet with header ", display_header.hex_encode(), " = ", packet_type.get_global_name())
else:
raw_packet.reverse()
print("Received unknown packet with header ", raw_packet.hex_encode())
func send(packet: Packet) -> Error:
var numerical_header := packet.get_header().decode_u16(0)
var display_header := packet.get_header()
display_header.reverse()
print("Sent packet %s with header %s" % [
Constants.PacketDB[numerical_header].get_global_name(),
display_header.hex_encode(),
])
return send_raw(packet.to_bytes())
func send_raw(bytes: PackedByteArray) -> Error:
return peer.put_data(bytes)
func wait_for_packets(packet_types: Array) -> Packet:
var packet: Packet
while true:
packet = await received_packet
for type in packet_types:
if is_instance_of(packet, type):
return packet
# code path return satisfaction
return packet
func wait_for_all_packets(packet_types: Array, error_packet_types: Array = []) -> Dictionary[int, Packet]:
var packets: Dictionary[int, Packet] = {}
var packet: Packet
var has_error := false
while true:
packet = await received_packet
# check packets
for type in packet_types:
if is_instance_of(packet, type):
packets.set(type.get("HEADER"), packet)
break
# check for errors
for type in error_packet_types:
if is_instance_of(packet, type):
packets.set(type.get("HEADER"), packet)
has_error = true
break
# finish if all packages received or one package errors
if packets.size() == packet_types.size() or has_error:
break
return packets
|
