Skip to content

Commit

Permalink
Refactor: peer_input_handle (coturn#1325)
Browse files Browse the repository at this point in the history 
see: coturn#1327
  • Loading branch information
@KangLin
KangLin committed Apr 10, 2024
1 parent 294f897 commit 243b2d7
Showing 1 changed file with 58 additions and 57 deletions.
115 changes: 58 additions & 57 deletions src/server/ns_turn_server.c
View file
Original file line number Diff line number Diff line change
Expand Up @@ -4669,8 +4669,9 @@ static void peer_input_handler(ioa_socket_handle s, int event_type, ioa_net_data

turn_turnserver *server = (turn_turnserver *)(ss->server);

if (!server)
if (!server) {
return;
}

relay_endpoint_session *elem = get_relay_session_ss(ss, get_ioa_socket_address_family(s));
if (elem->s == NULL) {
Expand All @@ -4681,80 +4682,80 @@ static void peer_input_handler(ioa_socket_handle s, int event_type, ioa_net_data

int ilen =
min((int)ioa_network_buffer_get_size(in_buffer->nbh), (int)(ioa_network_buffer_get_capacity_udp() - offset));
if (ilen < 0) {
return;
}

if (ilen >= 0) {
++(ss->peer_received_packets);
ss->peer_received_bytes += ilen;
turn_report_session_usage(ss, 0);
++(ss->peer_received_packets);
ss->peer_received_bytes += ilen;
turn_report_session_usage(ss, 0);

allocation *a = get_allocation_ss(ss);
if (is_allocation_valid(a)) {
allocation *a = get_allocation_ss(ss);
if (!is_allocation_valid(a))
return;

uint16_t chnum = 0;
uint16_t chnum = 0;

ioa_network_buffer_handle nbh = NULL;
ioa_network_buffer_handle nbh = NULL;

turn_permission_info *tinfo = allocation_get_permission(a, &(in_buffer->src_addr));
if (tinfo) {
chnum = get_turn_channel_number(tinfo, &(in_buffer->src_addr));
} else if (!(server->server_relay)) {
return;
}

if (chnum) {
turn_permission_info *tinfo = allocation_get_permission(a, &(in_buffer->src_addr));
if (tinfo) {
chnum = get_turn_channel_number(tinfo, &(in_buffer->src_addr));
} else if (!(server->server_relay)) {
return;
}

size_t len = (size_t)(ilen);
if (chnum) {

nbh = in_buffer->nbh;
size_t len = (size_t)(ilen);

ioa_network_buffer_add_offset_size(nbh, 0, STUN_CHANNEL_HEADER_LENGTH,
ioa_network_buffer_get_size(nbh) + STUN_CHANNEL_HEADER_LENGTH);
nbh = in_buffer->nbh;

ioa_network_buffer_header_init(nbh);
ioa_network_buffer_add_offset_size(nbh, 0, STUN_CHANNEL_HEADER_LENGTH,
ioa_network_buffer_get_size(nbh) + STUN_CHANNEL_HEADER_LENGTH);

SOCKET_TYPE st = get_ioa_socket_type(ss->client_socket);
int do_padding = is_stream_socket(st);
ioa_network_buffer_header_init(nbh);

stun_init_channel_message_str(chnum, ioa_network_buffer_data(nbh), &len, len, do_padding);
ioa_network_buffer_set_size(nbh, len);
in_buffer->nbh = NULL;
if (eve(server->verbose)) {
TURN_LOG_FUNC(TURN_LOG_LEVEL_INFO, "%s: send channel 0x%x\n", __FUNCTION__, (int)(chnum));
}
} else {
SOCKET_TYPE st = get_ioa_socket_type(ss->client_socket);
int do_padding = is_stream_socket(st);

size_t len = 0;
stun_init_channel_message_str(chnum, ioa_network_buffer_data(nbh), &len, len, do_padding);
ioa_network_buffer_set_size(nbh, len);
in_buffer->nbh = NULL;
if (eve(server->verbose)) {
TURN_LOG_FUNC(TURN_LOG_LEVEL_INFO, "%s: send channel 0x%x\n", __FUNCTION__, (int)(chnum));
}
} else {

nbh = ioa_network_buffer_allocate(server->e);
stun_init_indication_str(STUN_METHOD_DATA, ioa_network_buffer_data(nbh), &len);
stun_attr_add_str(ioa_network_buffer_data(nbh), &len, STUN_ATTRIBUTE_DATA,
ioa_network_buffer_data(in_buffer->nbh), (size_t)ilen);
stun_attr_add_addr_str(ioa_network_buffer_data(nbh), &len, STUN_ATTRIBUTE_XOR_PEER_ADDRESS,
&(in_buffer->src_addr));
ioa_network_buffer_set_size(nbh, len);
size_t len = 0;

if (!(*server->no_software_attribute)) {
const uint8_t *field = (const uint8_t *)get_version(server);
size_t fsz = strlen(get_version(server));
size_t len = ioa_network_buffer_get_size(nbh);
stun_attr_add_str(ioa_network_buffer_data(nbh), &len, STUN_ATTRIBUTE_SOFTWARE, field, fsz);
ioa_network_buffer_set_size(nbh, len);
}
nbh = ioa_network_buffer_allocate(server->e);
stun_init_indication_str(STUN_METHOD_DATA, ioa_network_buffer_data(nbh), &len);
stun_attr_add_str(ioa_network_buffer_data(nbh), &len, STUN_ATTRIBUTE_DATA, ioa_network_buffer_data(in_buffer->nbh),
(size_t)ilen);
stun_attr_add_addr_str(ioa_network_buffer_data(nbh), &len, STUN_ATTRIBUTE_XOR_PEER_ADDRESS, &(in_buffer->src_addr));
ioa_network_buffer_set_size(nbh, len);

if ((server->fingerprint) || ss->enforce_fingerprints) {
size_t len = ioa_network_buffer_get_size(nbh);
stun_attr_add_fingerprint_str(ioa_network_buffer_data(nbh), &len);
ioa_network_buffer_set_size(nbh, len);
}
}
if (eve(server->verbose)) {
uint16_t *t = (uint16_t *)ioa_network_buffer_data(nbh);
TURN_LOG_FUNC(TURN_LOG_LEVEL_INFO, "Send data: 0x%x\n", (int)(nswap16(t[0])));
}
if (!(*server->no_software_attribute)) {
const uint8_t *field = (const uint8_t *)get_version(server);
size_t fsz = strlen(get_version(server));
size_t len = ioa_network_buffer_get_size(nbh);
stun_attr_add_str(ioa_network_buffer_data(nbh), &len, STUN_ATTRIBUTE_SOFTWARE, field, fsz);
ioa_network_buffer_set_size(nbh, len);
}

write_client_connection(server, ss, nbh, in_buffer->recv_ttl - 1, in_buffer->recv_tos);
if ((server->fingerprint) || ss->enforce_fingerprints) {
size_t len = ioa_network_buffer_get_size(nbh);
stun_attr_add_fingerprint_str(ioa_network_buffer_data(nbh), &len);
ioa_network_buffer_set_size(nbh, len);
}
}
if (eve(server->verbose)) {
uint16_t *t = (uint16_t *)ioa_network_buffer_data(nbh);
TURN_LOG_FUNC(TURN_LOG_LEVEL_INFO, "Send data: 0x%x\n", (int)(nswap16(t[0])));
}

write_client_connection(server, ss, nbh, in_buffer->recv_ttl - 1, in_buffer->recv_tos);
}

static void client_input_handler(ioa_socket_handle s, int event_type, ioa_net_data *data, void *arg, int can_resume) {
Expand Down

0 comments on commit 243b2d7

Please sign in to comment.