From 1a5b8c19989c4ab4e1eee7f712913f8f3e459e71 Mon Sep 17 00:00:00 2001 From: Benny Prijono Date: Wed, 21 Feb 2007 00:40:05 +0000 Subject: Merged the ICE branch into the trunk git-svn-id: http://svn.pjsip.org/repos/pjproject/trunk@992 74dad513-b988-da41-8d7b-12977e46ad98 --- pjlib-util/src/pjlib-util/stun.c | 125 -- pjlib-util/src/pjlib-util/stun_client.c | 241 ---- pjlib-util/src/pjlib-util/stun_endpoint.c | 68 ++ pjlib-util/src/pjlib-util/stun_msg.c | 1562 ++++++++++++++++++++++++ pjlib-util/src/pjlib-util/stun_server.c | 129 ++ pjlib-util/src/pjlib-util/stun_simple.c | 125 ++ pjlib-util/src/pjlib-util/stun_simple_client.c | 241 ++++ pjlib-util/src/pjlib-util/stun_transaction.c | 333 +++++ pjlib-util/src/pjstun-srv/server.h | 61 + pjlib-util/src/pjstun-srv/server_main.c | 452 +++++++ 10 files changed, 2971 insertions(+), 366 deletions(-) delete mode 100644 pjlib-util/src/pjlib-util/stun.c delete mode 100644 pjlib-util/src/pjlib-util/stun_client.c create mode 100644 pjlib-util/src/pjlib-util/stun_endpoint.c create mode 100644 pjlib-util/src/pjlib-util/stun_msg.c create mode 100644 pjlib-util/src/pjlib-util/stun_server.c create mode 100644 pjlib-util/src/pjlib-util/stun_simple.c create mode 100644 pjlib-util/src/pjlib-util/stun_simple_client.c create mode 100644 pjlib-util/src/pjlib-util/stun_transaction.c create mode 100644 pjlib-util/src/pjstun-srv/server.h create mode 100644 pjlib-util/src/pjstun-srv/server_main.c (limited to 'pjlib-util/src') diff --git a/pjlib-util/src/pjlib-util/stun.c b/pjlib-util/src/pjlib-util/stun.c deleted file mode 100644 index c06479ac..00000000 --- a/pjlib-util/src/pjlib-util/stun.c +++ /dev/null @@ -1,125 +0,0 @@ -/* $Id$ */ -/* - * Copyright (C) 2003-2007 Benny Prijono - * - * This program is free software; you can redistribute it and/or modify - * it under the terms of the GNU General Public License as published by - * the Free Software Foundation; either version 2 of the License, or - * (at your option) any later version. - * - * This program is distributed in the hope that it will be useful, - * but WITHOUT ANY WARRANTY; without even the implied warranty of - * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the - * GNU General Public License for more details. - * - * You should have received a copy of the GNU General Public License - * along with this program; if not, write to the Free Software - * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA - */ -#include -#include -#include -#include -#include -#include - -#define THIS_FILE "stun.c" - -PJ_DEF(pj_status_t) pj_stun_create_bind_req( pj_pool_t *pool, - void **msg, pj_size_t *len, - pj_uint32_t id_hi, - pj_uint32_t id_lo) -{ - pj_stun_msg_hdr *hdr; - - PJ_CHECK_STACK(); - - - hdr = pj_pool_calloc(pool, 1, sizeof(pj_stun_msg_hdr)); - if (!hdr) - return PJ_ENOMEM; - - hdr->type = pj_htons(PJ_STUN_BINDING_REQUEST); - hdr->tsx[2] = pj_htonl(id_hi); - hdr->tsx[3] = pj_htonl(id_lo); - *msg = hdr; - *len = sizeof(pj_stun_msg_hdr); - - return PJ_SUCCESS; -} - -PJ_DEF(pj_status_t) pj_stun_parse_msg( void *buf, pj_size_t len, - pj_stun_msg *msg) -{ - pj_uint16_t msg_type, msg_len; - char *p_attr; - - PJ_CHECK_STACK(); - - msg->hdr = (pj_stun_msg_hdr*)buf; - msg_type = pj_ntohs(msg->hdr->type); - - switch (msg_type) { - case PJ_STUN_BINDING_REQUEST: - case PJ_STUN_BINDING_RESPONSE: - case PJ_STUN_BINDING_ERROR_RESPONSE: - case PJ_STUN_SHARED_SECRET_REQUEST: - case PJ_STUN_SHARED_SECRET_RESPONSE: - case PJ_STUN_SHARED_SECRET_ERROR_RESPONSE: - break; - default: - PJ_LOG(4,(THIS_FILE, "Error: unknown msg type %d", msg_type)); - return PJLIB_UTIL_ESTUNINMSGTYPE; - } - - msg_len = pj_ntohs(msg->hdr->length); - if (msg_len != len - sizeof(pj_stun_msg_hdr)) { - PJ_LOG(4,(THIS_FILE, "Error: invalid msg_len %d (expecting %d)", - msg_len, len - sizeof(pj_stun_msg_hdr))); - return PJLIB_UTIL_ESTUNINMSGLEN; - } - - msg->attr_count = 0; - p_attr = (char*)buf + sizeof(pj_stun_msg_hdr); - - while (msg_len > 0) { - pj_stun_attr_hdr **attr = &msg->attr[msg->attr_count]; - pj_uint32_t len; - - *attr = (pj_stun_attr_hdr*)p_attr; - len = pj_ntohs((pj_uint16_t) ((*attr)->length)) + sizeof(pj_stun_attr_hdr); - - if (msg_len < len) { - PJ_LOG(4,(THIS_FILE, "Error: length mismatch in attr %d", - msg->attr_count)); - return PJLIB_UTIL_ESTUNINATTRLEN; - } - - if (pj_ntohs((*attr)->type) > PJ_STUN_ATTR_REFLECTED_FORM) { - PJ_LOG(5,(THIS_FILE, "Warning: unknown attr type %x in attr %d. " - "Attribute was ignored.", - pj_ntohs((*attr)->type), msg->attr_count)); - } - - msg_len = (pj_uint16_t)(msg_len - len); - p_attr += len; - ++msg->attr_count; - } - - return PJ_SUCCESS; -} - -PJ_DEF(void*) pj_stun_msg_find_attr( pj_stun_msg *msg, pj_stun_attr_type t) -{ - int i; - - PJ_CHECK_STACK(); - - for (i=0; iattr_count; ++i) { - pj_stun_attr_hdr *attr = msg->attr[i]; - if (pj_ntohs(attr->type) == t) - return attr; - } - - return 0; -} diff --git a/pjlib-util/src/pjlib-util/stun_client.c b/pjlib-util/src/pjlib-util/stun_client.c deleted file mode 100644 index 6005694c..00000000 --- a/pjlib-util/src/pjlib-util/stun_client.c +++ /dev/null @@ -1,241 +0,0 @@ -/* $Id$ */ -/* - * Copyright (C) 2003-2007 Benny Prijono - * - * This program is free software; you can redistribute it and/or modify - * it under the terms of the GNU General Public License as published by - * the Free Software Foundation; either version 2 of the License, or - * (at your option) any later version. - * - * This program is distributed in the hope that it will be useful, - * but WITHOUT ANY WARRANTY; without even the implied warranty of - * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the - * GNU General Public License for more details. - * - * You should have received a copy of the GNU General Public License - * along with this program; if not, write to the Free Software - * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA - */ -#include -#include -#include -#include -#include -#include -#include - - -enum { MAX_REQUEST = 3 }; -static int stun_timer[] = {1600, 1600, 1600 }; - -#define THIS_FILE "stun_client.c" -#define LOG_ADDR(addr) pj_inet_ntoa(addr.sin_addr), pj_ntohs(addr.sin_port) - - -PJ_DECL(pj_status_t) pj_stun_get_mapped_addr( pj_pool_factory *pf, - int sock_cnt, pj_sock_t sock[], - const pj_str_t *srv1, int port1, - const pj_str_t *srv2, int port2, - pj_sockaddr_in mapped_addr[]) -{ - pj_sockaddr_in srv_addr[2]; - int i, j, send_cnt = 0; - pj_pool_t *pool; - struct { - struct { - pj_uint32_t mapped_addr; - pj_uint32_t mapped_port; - } srv[2]; - } *rec; - void *out_msg; - pj_size_t out_msg_len; - int wait_resp = 0; - pj_status_t status; - - PJ_CHECK_STACK(); - - /* Create pool. */ - pool = pj_pool_create(pf, "stun%p", 1024, 1024, NULL); - if (!pool) - return PJ_ENOMEM; - - - /* Allocate client records */ - rec = pj_pool_calloc(pool, sock_cnt, sizeof(*rec)); - if (!rec) { - status = PJ_ENOMEM; - goto on_error; - } - - - /* Create the outgoing BIND REQUEST message template */ - status = pj_stun_create_bind_req( pool, &out_msg, &out_msg_len, - pj_rand(), pj_rand()); - if (status != PJ_SUCCESS) - goto on_error; - - /* Resolve servers. */ - status = pj_sockaddr_in_init(&srv_addr[0], srv1, (pj_uint16_t)port1); - if (status != PJ_SUCCESS) - goto on_error; - - status = pj_sockaddr_in_init(&srv_addr[1], srv2, (pj_uint16_t)port2); - if (status != PJ_SUCCESS) - goto on_error; - - /* Init mapped addresses to zero */ - pj_memset(mapped_addr, 0, sock_cnt * sizeof(pj_sockaddr_in)); - - /* Main retransmission loop. */ - for (send_cnt=0; send_cnttsx[2] = pj_htonl(i); - msg_hdr->tsx[3] = pj_htonl(j); - - /* Send! */ - sent_len = out_msg_len; - status = pj_sock_sendto(sock[i], out_msg, &sent_len, 0, - (pj_sockaddr_t*)&srv_addr[j], - sizeof(pj_sockaddr_in)); - if (status == PJ_SUCCESS) - ++wait_resp; - } - } - - /* All requests sent. - * The loop below will wait for responses until all responses have - * been received (i.e. wait_resp==0) or timeout occurs, which then - * we'll go to the next retransmission iteration. - */ - - /* Calculate time of next retransmission. */ - pj_gettimeofday(&next_tx); - next_tx.sec += (stun_timer[send_cnt]/1000); - next_tx.msec += (stun_timer[send_cnt]%1000); - pj_time_val_normalize(&next_tx); - - for (pj_gettimeofday(&now), select_rc=1; - status==PJ_SUCCESS && select_rc==1 && wait_resp>0 - && PJ_TIME_VAL_LT(now, next_tx); - pj_gettimeofday(&now)) - { - pj_time_val timeout; - - timeout = next_tx; - PJ_TIME_VAL_SUB(timeout, now); - - for (i=0; itsx[2]); - srv_idx = pj_ntohl(msg.hdr->tsx[3]); - - if (sock_idx<0 || sock_idx>=sock_cnt || srv_idx<0 || srv_idx>=2) { - status = PJLIB_UTIL_ESTUNININDEX; - continue; - } - - if (pj_ntohs(msg.hdr->type) != PJ_STUN_BINDING_RESPONSE) { - status = PJLIB_UTIL_ESTUNNOBINDRES; - continue; - } - - if (pj_stun_msg_find_attr(&msg, PJ_STUN_ATTR_ERROR_CODE) != NULL) { - status = PJLIB_UTIL_ESTUNRECVERRATTR; - continue; - } - - attr = (void*)pj_stun_msg_find_attr(&msg, PJ_STUN_ATTR_MAPPED_ADDR); - if (!attr) { - status = PJLIB_UTIL_ESTUNNOMAP; - continue; - } - - rec[sock_idx].srv[srv_idx].mapped_addr = attr->addr; - rec[sock_idx].srv[srv_idx].mapped_port = attr->port; - } - } - - /* The best scenario is if all requests have been replied. - * Then we don't need to go to the next retransmission iteration. - */ - if (wait_resp <= 0) - break; - } - - for (i=0; i + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + */ +#include +#include +#include +#include + + +/* + * Create a STUN endpoint instance. + */ +PJ_DEF(pj_status_t) pj_stun_endpt_create( pj_pool_factory *factory, + unsigned options, + pj_ioqueue_t *ioqueue, + pj_timer_heap_t *timer_heap, + pj_stun_endpoint **p_endpt) +{ + pj_pool_t *pool; + pj_stun_endpoint *endpt; + + PJ_ASSERT_RETURN(factory && p_endpt, PJ_EINVAL); + + pool = pj_pool_create(factory, "stunendpt", 1000, 1000, NULL); + if (!pool) + return PJ_ENOMEM; + + endpt = PJ_POOL_ZALLOC_TYPE(pool, pj_stun_endpoint); + endpt->pool = pool; + endpt->pf = factory; + endpt->options = options; + endpt->ioqueue = ioqueue; + endpt->timer_heap = timer_heap; + endpt->rto_msec = PJ_STUN_RTO_VALUE; + + *p_endpt = endpt; + + return PJ_SUCCESS; +} + + +/* + * Destroy STUN endpoint instance. + */ +PJ_DEF(pj_status_t) pj_stun_endpt_destroy(pj_stun_endpoint *endpt) +{ + PJ_ASSERT_RETURN(endpt, PJ_EINVAL); + + pj_pool_release(endpt->pool); + + return PJ_SUCCESS; +} + diff --git a/pjlib-util/src/pjlib-util/stun_msg.c b/pjlib-util/src/pjlib-util/stun_msg.c new file mode 100644 index 00000000..9abb2a4c --- /dev/null +++ b/pjlib-util/src/pjlib-util/stun_msg.c @@ -0,0 +1,1562 @@ +/* $Id$ */ +/* + * Copyright (C) 2003-2005 Benny Prijono + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + */ +#include +#include +#include +#include +#include +#include +#include +#include + +#define THIS_FILE "stun_msg.c" + + +static const char *stun_method_names[] = +{ + "Unknown", /* 0 */ + "Binding", /* 1 */ + "Shared Secret", /* 2 */ + "Allocate", /* 3 */ + "Send", /* 4 */ + "Data", /* 5 */ + "Set Active Destination", /* 6 */ + "Connect", /* 7 */ + "Connect Status" /* 8 */ +}; + +static struct +{ + int err_code; + const char *err_msg; +} stun_err_msg_map[] = +{ + { PJ_STUN_STATUS_TRY_ALTERNATE, "Try Alternate"}, + { PJ_STUN_STATUS_BAD_REQUEST, "Bad Request"}, + { PJ_STUN_STATUS_UNAUTHORIZED, "Unauthorized"}, + { PJ_STUN_STATUS_UNKNOWN_ATTRIBUTE, "Unknown Attribute"}, + { PJ_STUN_STATUS_STALE_CREDENTIALS, "Stale Credentials"}, + { PJ_STUN_STATUS_INTEGRITY_CHECK_FAILURE,"Integrity Check Failure"}, + { PJ_STUN_STATUS_MISSING_USERNAME, "Missing Username"}, + { PJ_STUN_STATUS_USE_TLS, "Use TLS"}, + { PJ_STUN_STATUS_MISSING_REALM, "Missing Realm"}, + { PJ_STUN_STATUS_MISSING_NONCE, "Missing Nonce"}, + { PJ_STUN_STATUS_UNKNOWN_USERNAME, "Unknown Username"}, + { PJ_STUN_STATUS_NO_BINDING, "No Binding"}, + { PJ_STUN_STATUS_STALE_NONCE, "Stale Nonce"}, + { PJ_STUN_STATUS_TRANSITIONING, "Transitioning"}, + { PJ_STUN_STATUS_WRONG_USERNAME, "Wrong Username"}, + { PJ_STUN_STATUS_UNSUPP_TRANSPORT_PROTO,"Unsupported Transport Protocol"}, + { PJ_STUN_STATUS_INVALID_IP_ADDR, "Invalid IP Address"}, + { PJ_STUN_STATUS_INVALID_PORT, "Invalid Port"}, + { PJ_STUN_STATUS_OPER_TCP_ONLY, "Operation for TCP Only"}, + { PJ_STUN_STATUS_CONNECTION_FAILURE, "Connection Failure"}, + { PJ_STUN_STATUS_CONNECTION_TIMEOUT, "Connection Timeout"}, + { PJ_STUN_STATUS_SERVER_ERROR, "Server Error"}, + { PJ_STUN_STATUS_GLOBAL_FAILURE, "Global Failure"} +}; + + + +struct attr_desc +{ + const char *name; + pj_status_t (*decode_attr)(pj_pool_t *pool, const pj_uint8_t *buf, + void **p_attr); + pj_status_t (*encode_attr)(const void *a, pj_uint8_t *buf, + unsigned len, unsigned *printed); + +}; + +static pj_status_t decode_generic_ip_addr_attr(pj_pool_t *pool, + const pj_uint8_t *buf, + void **p_attr); +static pj_status_t encode_generic_ip_addr_attr(const void *a, pj_uint8_t *buf, + unsigned len, + unsigned *printed); +static pj_status_t decode_generic_string_attr(pj_pool_t *pool, + const pj_uint8_t *buf, + void **p_attr); +static pj_status_t encode_generic_string_attr(const void *a, pj_uint8_t *buf, + unsigned len, unsigned *printed); +static pj_status_t decode_msg_integrity_attr(pj_pool_t *pool, + const pj_uint8_t *buf, + void **p_attr); +static pj_status_t encode_msg_integrity_attr(const void *a, pj_uint8_t *buf, + unsigned len, unsigned *printed); +static pj_status_t decode_error_code_attr(pj_pool_t *pool, + const pj_uint8_t *buf, + void **p_attr); +static pj_status_t encode_error_code_attr(const void *a, pj_uint8_t *buf, + unsigned len, unsigned *printed); +static pj_status_t decode_unknown_attr(pj_pool_t *pool, + const pj_uint8_t *buf, + void **p_attr); +static pj_status_t encode_unknown_attr(const void *a, pj_uint8_t *buf, + unsigned len, unsigned *printed); +static pj_status_t decode_generic_uint_attr(pj_pool_t *pool, + const pj_uint8_t *buf, + void **p_attr); +static pj_status_t encode_generic_uint_attr(const void *a, pj_uint8_t *buf, + unsigned len, unsigned *printed); +static pj_status_t decode_binary_attr(pj_pool_t *pool, + const pj_uint8_t *buf, + void **p_attr); +static pj_status_t encode_binary_attr(const void *a, pj_uint8_t *buf, + unsigned len, unsigned *printed); +static pj_status_t decode_empty_attr(pj_pool_t *pool, + const pj_uint8_t *buf, + void **p_attr); +static pj_status_t encode_empty_attr(const void *a, pj_uint8_t *buf, + unsigned len, unsigned *printed); + + +struct attr_desc mandatory_attr_desc[] = +{ + { + /* type zero */ + NULL, + NULL, + NULL + }, + { + /* PJ_STUN_ATTR_MAPPED_ADDR, */ + "MAPPED-ADDRESS", + &decode_generic_ip_addr_attr, + &encode_generic_ip_addr_attr + }, + { + /* PJ_STUN_ATTR_RESPONSE_ADDR, */ + "RESPONSE-ADDRESS", + &decode_generic_ip_addr_attr, + &encode_generic_ip_addr_attr + }, + { + /* PJ_STUN_ATTR_CHANGE_REQUEST, */ + "CHANGE-REQUEST", + &decode_generic_uint_attr, + &encode_generic_uint_attr + }, + { + /* PJ_STUN_ATTR_SOURCE_ADDR, */ + "SOURCE-ADDRESS", + &decode_generic_ip_addr_attr, + &encode_generic_ip_addr_attr + }, + { + /* PJ_STUN_ATTR_CHANGED_ADDR, */ + "CHANGED-ADDRESS", + &decode_generic_ip_addr_attr, + &encode_generic_ip_addr_attr + }, + { + /* PJ_STUN_ATTR_USERNAME, */ + "USERNAME", + &decode_generic_string_attr, + &encode_generic_string_attr + }, + { + /* PJ_STUN_ATTR_PASSWORD, */ + "PASSWORD", + &decode_generic_string_attr, + &encode_generic_string_attr + }, + { + /* PJ_STUN_ATTR_MESSAGE_INTEGRITY, */ + "MESSAGE-INTEGRITY", + &decode_msg_integrity_attr, + &encode_msg_integrity_attr + }, + { + /* PJ_STUN_ATTR_ERROR_CODE, */ + "ERROR-CODE", + &decode_error_code_attr, + &encode_error_code_attr + }, + { + /* PJ_STUN_ATTR_UNKNOWN_ATTRIBUTES, */ + "UNKNOWN-ATTRIBUTES", + &decode_unknown_attr, + &encode_unknown_attr + }, + { + /* PJ_STUN_ATTR_REFLECTED_FROM, */ + "REFLECTED-FROM", + &decode_generic_ip_addr_attr, + &encode_generic_ip_addr_attr + }, + { + /* ID 0x000C is not assigned */ + NULL, + NULL, + NULL + }, + { + /* PJ_STUN_ATTR_LIFETIME, */ + "LIFETIME", + &decode_generic_uint_attr, + &encode_generic_uint_attr + }, + { + /* ID 0x000E is not assigned */ + NULL, + NULL, + NULL + }, + { + /* ID 0x000F is not assigned */ + NULL, + NULL, + NULL + }, + { + /* PJ_STUN_ATTR_BANDWIDTH, */ + "BANDWIDTH", + &decode_generic_uint_attr, + &encode_generic_uint_attr + }, + { + /* ID 0x0011 is not assigned */ + NULL, + NULL, + NULL + }, + { + /* PJ_STUN_ATTR_REMOTE_ADDRESS, */ + "REMOTE-ADDRESS", + &decode_generic_ip_addr_attr, + &encode_generic_ip_addr_attr + }, + { + /* PJ_STUN_ATTR_DATA, */ + "DATA", + &decode_binary_attr, + &encode_binary_attr + }, + { + /* PJ_STUN_ATTR_REALM, */ + "REALM", + &decode_generic_string_attr, + &encode_generic_string_attr + }, + { + /* PJ_STUN_ATTR_NONCE, */ + "NONCE", + &decode_generic_string_attr, + &encode_generic_string_attr + }, + { + /* PJ_STUN_ATTR_RELAY_ADDRESS, */ + "RELAY-ADDRESS", + &decode_generic_ip_addr_attr, + &encode_generic_ip_addr_attr + }, + { + /* PJ_STUN_ATTR_REQUESTED_ADDR_TYPE, */ + "REQUESTED-ADDRESS-TYPE", + &decode_generic_uint_attr, + &encode_generic_uint_attr + }, + { + /* PJ_STUN_ATTR_REQUESTED_PORT_PROPS, */ + "REQUESTED-PORT-PROPS", + &decode_generic_uint_attr, + &encode_generic_uint_attr + }, + { + /* PJ_STUN_ATTR_REQUESTED_TRANSPORT, */ + "REQUESTED-TRANSPORT", + &decode_generic_uint_attr, + &encode_generic_uint_attr + }, + { + /* ID 0x001A is not assigned */ + NULL, + NULL, + NULL + }, + { + /* ID 0x001B is not assigned */ + NULL, + NULL, + NULL + }, + { + /* ID 0x001C is not assigned */ + NULL, + NULL, + NULL + }, + { + /* ID 0x001D is not assigned */ + NULL, + NULL, + NULL + }, + { + /* ID 0x001E is not assigned */ + NULL, + NULL, + NULL + }, + { + /* ID 0x001F is not assigned */ + NULL, + NULL, + NULL + }, + { + /* PJ_STUN_ATTR_XOR_MAPPED_ADDRESS, */ + "XOR-MAPPED-ADDRESS", + &decode_generic_ip_addr_attr, + &encode_generic_ip_addr_attr + }, + { + /* PJ_STUN_ATTR_TIMER_VAL, */ + "TIMER-VAL", + &decode_generic_uint_attr, + &encode_generic_uint_attr + }, + { + /* PJ_STUN_ATTR_REQUESTED_IP, */ + "REQUESTED-IP", + &decode_generic_ip_addr_attr, + &encode_generic_ip_addr_attr + }, + { + /* PJ_STUN_ATTR_XOR_REFLECTED_FROM, */ + "XOR-REFLECTED-FROM", + &decode_generic_ip_addr_attr, + &encode_generic_ip_addr_attr + }, + { + /* PJ_STUN_ATTR_PRIORITY, */ + "PRIORITY", + &decode_generic_uint_attr, + &encode_generic_uint_attr + }, + { + /* PJ_STUN_ATTR_USE_CANDIDATE, */ + "USE-CANDIDATE", + &decode_empty_attr, + &encode_empty_attr + }, + { + /* PJ_STUN_ATTR_XOR_INTERNAL_ADDR, */ + "XOR-INTERNAL-ADDRESS", + &decode_generic_ip_addr_attr, + &encode_generic_ip_addr_attr + }, + + /* Sentinel */ + { + /* PJ_STUN_ATTR_END_MANDATORY_ATTR */ + NULL, + NULL, + NULL + } +}; + +static struct attr_desc extended_attr_desc[] = +{ + { + /* PJ_STUN_ATTR_FINGERPRINT, */ + "FINGERPRINT", + &decode_generic_uint_attr, + &encode_generic_uint_attr + }, + { + /* PJ_STUN_ATTR_SERVER, */ + "SERVER", + &decode_generic_string_attr, + &encode_generic_string_attr + }, + { + /* PJ_STUN_ATTR_ALTERNATE_SERVER, */ + "ALTERNATE-SERVER", + &decode_generic_ip_addr_attr, + &encode_generic_ip_addr_attr + }, + { + /* PJ_STUN_ATTR_REFRESH_INTERVAL, */ + "REFRESH-INTERVAL", + &decode_generic_uint_attr, + &encode_generic_uint_attr + }, +}; + + + +/* + * Get STUN message type name. + */ +PJ_DEF(const char*) pj_stun_get_method_name(unsigned msg_type) +{ + unsigned method = PJ_STUN_GET_METHOD(msg_type); + + if (method >= PJ_ARRAY_SIZE(stun_method_names)) + return "???"; + + return stun_method_names[method]; +} + + +/* + * Get STUN message class name. + */ +PJ_DEF(const char*) pj_stun_get_class_name(unsigned msg_type) +{ + if (PJ_STUN_IS_REQUEST(msg_type)) + return "request"; + else if (PJ_STUN_IS_RESPONSE(msg_type)) + return "success response"; + else if (PJ_STUN_IS_ERROR_RESPONSE(msg_type)) + return "error response"; + else if (PJ_STUN_IS_INDICATION(msg_type)) + return "indication"; + else + return "???"; +} + + +static const struct attr_desc *find_attr_desc(unsigned attr_type) +{ + struct attr_desc *desc; + + /* Check that attr_desc array is valid */ + pj_assert(PJ_ARRAY_SIZE(mandatory_attr_desc)== + PJ_STUN_ATTR_END_MANDATORY_ATTR+1); + pj_assert(mandatory_attr_desc[PJ_STUN_ATTR_END_MANDATORY_ATTR].decode_attr + == NULL); + pj_assert(mandatory_attr_desc[PJ_STUN_ATTR_USE_CANDIDATE].decode_attr + == &decode_empty_attr); + pj_assert(PJ_ARRAY_SIZE(extended_attr_desc) == + PJ_STUN_ATTR_END_EXTENDED_ATTR-PJ_STUN_ATTR_START_EXTENDED_ATTR); + + if (attr_type < PJ_STUN_ATTR_START_EXTENDED_ATTR) + desc = &mandatory_attr_desc[attr_type]; + else if (attr_type >= PJ_STUN_ATTR_START_EXTENDED_ATTR && + attr_type < PJ_STUN_ATTR_END_EXTENDED_ATTR) + desc = &extended_attr_desc[attr_type-PJ_STUN_ATTR_START_EXTENDED_ATTR]; + else + return NULL; + + return desc->decode_attr == NULL ? NULL : desc; +} + + +/* + * Get STUN attribute name. + */ +PJ_DEF(const char*) pj_stun_get_attr_name(unsigned attr_type) +{ + const struct attr_desc *attr_desc; + + attr_desc = find_attr_desc(attr_type); + if (!attr_desc || attr_desc->name==NULL) + return "???"; + + return attr_desc->name; +} + + +/** + * Get STUN standard reason phrase for the specified error code. + */ +PJ_DEF(pj_str_t) pj_stun_get_err_reason(int err_code) +{ + unsigned i; + + for (i=0; ihdr.type=(pj_uint16_t)(t), \ + (a)->hdr.length=(pj_uint16_t)(l) +#define ATTR_HDR_LEN 4 + +#define getval16(p, pos) (pj_uint16_t)(((p)[(pos)] << 8) | \ + ((p)[(pos) + 1] << 0)) + + +////////////////////////////////////////////////////////////////////////////// +/* + * STUN generic IP address container + */ +#define STUN_GENERIC_IP_ADDR_LEN 8 + +/* + * Create a generic STUN IP address attribute for IPv4 address. + */ +PJ_DEF(pj_status_t) +pj_stun_generic_ip_addr_attr_create(pj_pool_t *pool, + int attr_type, + pj_bool_t xor_ed, + unsigned addr_len, + const pj_sockaddr_t *addr, + pj_stun_generic_ip_addr_attr **p_attr) +{ + pj_stun_generic_ip_addr_attr *attr; + + PJ_ASSERT_RETURN(pool && addr_len && addr && p_attr, PJ_EINVAL); + PJ_ASSERT_RETURN(addr_len == sizeof(pj_sockaddr_in) || + addr_len == sizeof(pj_sockaddr_in6), PJ_EINVAL); + + attr = PJ_POOL_ZALLOC_TYPE(pool, pj_stun_generic_ip_addr_attr); + INIT_ATTR(attr, attr_type, STUN_GENERIC_IP_ADDR_LEN); + + if (!xor_ed) { + pj_memcpy(&attr->addr, addr, addr_len); + } else if (addr_len == sizeof(pj_sockaddr_in)) { + const pj_sockaddr_in *addr4 = (const pj_sockaddr_in*) addr; + + pj_sockaddr_in_init(&attr->addr.ipv4, NULL, 0); + attr->addr.ipv4.sin_port = (pj_uint16_t)(addr4->sin_port ^ 0x2112); + attr->addr.ipv4.sin_addr.s_addr = (addr4->sin_addr.s_addr ^ + pj_htonl(0x2112A442)); + } else if (addr_len == sizeof(pj_sockaddr_in6)) { + return PJLIB_UTIL_ESTUNIPV6NOTSUPP; + } else { + return PJLIB_UTIL_ESTUNINADDRLEN; + } + + *p_attr = attr; + + return PJ_SUCCESS; +} + + +static pj_status_t decode_generic_ip_addr_attr(pj_pool_t *pool, + const pj_uint8_t *buf, + void **p_attr) +{ + pj_stun_generic_ip_addr_attr *attr; + pj_uint32_t val; + + /* Create the attribute */ + attr = PJ_POOL_ZALLOC_TYPE(pool, pj_stun_generic_ip_addr_attr); + pj_memcpy(attr, buf, ATTR_HDR_LEN); + + /* Convert to host byte order */ + attr->hdr.type = pj_ntohs(attr->hdr.type); + attr->hdr.length = pj_ntohs(attr->hdr.length); + + /* Check that the attribute length is valid */ + if (attr->hdr.length != STUN_GENERIC_IP_ADDR_LEN) + return PJLIB_UTIL_ESTUNINATTRLEN; + + /* Check address family */ + val = *(pj_uint8_t*)(buf + ATTR_HDR_LEN + 1); + + /* Check address family is valid (only supports ipv4 for now) */ + if (val != 1) + return PJLIB_UTIL_ESTUNIPV6NOTSUPP; + + /* Get port and address */ + pj_sockaddr_in_init(&attr->addr.ipv4, NULL, 0); + attr->addr.ipv4.sin_port = getval16(buf, ATTR_HDR_LEN + 2); + pj_memcpy(&attr->addr.ipv4.sin_addr, buf+ATTR_HDR_LEN+4, 4); + + /* Done */ + *p_attr = attr; + + return PJ_SUCCESS; +} + + +static pj_status_t encode_generic_ip_addr_attr(const void *a, pj_uint8_t *buf, + unsigned len, unsigned *printed) +{ + enum { + ATTR_LEN = ATTR_HDR_LEN + STUN_GENERIC_IP_ADDR_LEN + }; + pj_uint8_t *start_buf = buf; + const pj_stun_generic_ip_addr_attr *ca = + (const pj_stun_generic_ip_addr_attr *)a; + pj_stun_generic_ip_addr_attr *attr; + + if (len < ATTR_LEN) + return PJ_ETOOSMALL; + + /* Copy and convert headers to network byte order */ + pj_memcpy(buf, a, ATTR_HDR_LEN); + attr = (pj_stun_generic_ip_addr_attr*) buf; + attr->hdr.type = pj_htons(attr->hdr.type); + attr->hdr.length = pj_htons((pj_uint16_t)STUN_GENERIC_IP_ADDR_LEN); + buf += ATTR_HDR_LEN; + + /* Ignored */ + *buf++ = '\0'; + + /* Family (IPv4 only for now) */ + PJ_ASSERT_RETURN(ca->addr.addr.sa_family == PJ_AF_INET, PJ_EINVAL); + *buf++ = 1; + + /* Port */ + pj_memcpy(buf, &ca->addr.ipv4.sin_port, 2); + buf += 2; + + /* Address */ + pj_memcpy(buf, &ca->addr.ipv4.sin_addr, 4); + buf += 4; + + pj_assert(buf - start_buf == ATTR_LEN); + + /* Done */ + *printed = buf - start_buf; + + return PJ_SUCCESS; +} + + +////////////////////////////////////////////////////////////////////////////// +/* + * STUN generic string attribute + */ + +/* + * Create a STUN generic string attribute. + */ +PJ_DEF(pj_status_t) +pj_stun_generic_string_attr_create(pj_pool_t *pool, + int attr_type, + const pj_str_t *value, + pj_stun_generic_string_attr **p_attr) +{ + pj_stun_generic_string_attr *attr; + + PJ_ASSERT_RETURN(pool && value && p_attr, PJ_EINVAL); + + attr = PJ_POOL_ZALLOC_TYPE(pool, pj_stun_generic_string_attr); + INIT_ATTR(attr, attr_type, value->slen); + pj_strdup(pool, &attr->value, value); + + *p_attr = attr; + + return PJ_SUCCESS; +} + + +static pj_status_t decode_generic_string_attr(pj_pool_t *pool, + const pj_uint8_t *buf, + void **p_attr) +{ + pj_stun_generic_string_attr *attr; + pj_str_t value; + + /* Create the attribute */ + attr = PJ_POOL_ZALLOC_TYPE(pool, pj_stun_generic_string_attr); + + /* Copy the header */ + pj_memcpy(attr, buf, ATTR_HDR_LEN); + + /* Convert to host byte order */ + attr->hdr.type = pj_ntohs(attr->hdr.type); + attr->hdr.length = pj_ntohs(attr->hdr.length); + + /* Get pointer to the string in the message */ + value.ptr = ((char*)buf + ATTR_HDR_LEN); + value.slen = attr->hdr.length; + + /* Copy the string to the attribute */ + pj_strdup(pool, &attr->value, &value); + + /* Done */ + *p_attr = attr; + + return PJ_SUCCESS; + +} + + +static pj_status_t encode_generic_string_attr(const void *a, pj_uint8_t *buf, + unsigned len, unsigned *printed) +{ + const pj_stun_generic_string_attr *ca = + (const pj_stun_generic_string_attr*)a; + pj_stun_attr_hdr *attr; + + /* Calculated total attr_len (add padding if necessary) */ + *printed = (ca->value.slen + ATTR_HDR_LEN + 3) & (~3); + if (len < *printed) { + *printed = 0; + return PJ_ETOOSMALL; + } + + /* Copy header */ + pj_memcpy(buf, a, ATTR_HDR_LEN); + attr = (pj_stun_attr_hdr*)buf; + + /* Set the correct length */ + attr->length = (pj_uint16_t) ca->value.slen; + + /* Convert to network byte order */ + attr->type = pj_htons(attr->type); + attr->length = pj_htons(attr->length); + + /* Copy the string */ + pj_memcpy(buf+ATTR_HDR_LEN, ca->value.ptr, ca->value.slen); + + /* Done */ + return PJ_SUCCESS; +} + + +////////////////////////////////////////////////////////////////////////////// +/* + * STUN empty attribute (used by USE-CANDIDATE). + */ + +/* + * Create a STUN empty attribute. + */ +PJ_DEF(pj_status_t) +pj_stun_empty_attr_create(pj_pool_t *pool, + int attr_type, + pj_stun_empty_attr **p_attr) +{ + pj_stun_empty_attr *attr; + + PJ_ASSERT_RETURN(pool && p_attr, PJ_EINVAL); + + attr = PJ_POOL_ZALLOC_TYPE(pool, pj_stun_empty_attr); + INIT_ATTR(attr, attr_type, sizeof(pj_stun_empty_attr)); + + *p_attr = attr; + + return PJ_SUCCESS; +} + + +static pj_status_t decode_empty_attr(pj_pool_t *pool, + const pj_uint8_t *buf, + void **p_attr) +{ + pj_stun_empty_attr *attr; + + /* Check that the struct address is valid */ + pj_assert(sizeof(pj_stun_empty_attr) == ATTR_HDR_LEN); + + /* Create the attribute */ + attr = PJ_POOL_ZALLOC_TYPE(pool, pj_stun_empty_attr); + pj_memcpy(attr, buf, ATTR_HDR_LEN); + + /* Convert to host byte order */ + attr->hdr.type = pj_ntohs(attr->hdr.type); + attr->hdr.length = pj_ntohs(attr->hdr.length); + + /* Check that the attribute length is valid */ + if (attr->hdr.length != ATTR_HDR_LEN) + return PJLIB_UTIL_ESTUNINATTRLEN; + + /* Done */ + *p_attr = attr; + + return PJ_SUCCESS; +} + + +static pj_status_t encode_empty_attr(const void *a, pj_uint8_t *buf, + unsigned len, unsigned *printed) +{ + pj_stun_empty_attr *attr; + + if (len < ATTR_HDR_LEN) + return PJ_ETOOSMALL; + + /* Copy and convert attribute to network byte order */ + pj_memcpy(buf, a, ATTR_HDR_LEN); + attr = (pj_stun_empty_attr*) buf; + attr->hdr.type = pj_htons(attr->hdr.type); + pj_assert(attr->hdr.length == ATTR_HDR_LEN); + attr->hdr.length = pj_htons(ATTR_HDR_LEN); + + /* Done */ + *printed = ATTR_HDR_LEN; + + return PJ_SUCCESS; +} + + +////////////////////////////////////////////////////////////////////////////// +/* + * STUN generic 32bit integer attribute. + */ +#define STUN_UINT_LEN 4 + +/* + * Create a STUN generic 32bit value attribute. + */ +PJ_DEF(pj_status_t) +pj_stun_generic_uint_attr_create(pj_pool_t *pool, + int attr_type, + pj_uint32_t value, + pj_stun_generic_uint_attr **p_attr) +{ + pj_stun_generic_uint_attr *attr; + + PJ_ASSERT_RETURN(pool && p_attr, PJ_EINVAL); + + attr = PJ_POOL_ZALLOC_TYPE(pool, pj_stun_generic_uint_attr); + INIT_ATTR(attr, attr_type, STUN_UINT_LEN); + attr->value = value; + + *p_attr = attr; + + return PJ_SUCCESS; +} + + +static pj_status_t decode_generic_uint_attr(pj_pool_t *pool, + const pj_uint8_t *buf, + void **p_attr) +{ + enum + { + ATTR_LEN = STUN_UINT_LEN + ATTR_HDR_LEN + }; + pj_stun_generic_uint_attr *attr; + + /* Check that the struct address is valid */ + pj_assert(sizeof(pj_stun_generic_uint_attr) == ATTR_LEN); + + /* Create the attribute */ + attr = PJ_POOL_ZALLOC_TYPE(pool, pj_stun_generic_uint_attr); + pj_memcpy(attr, buf, ATTR_LEN); + + /* Convert to host byte order */ + attr->hdr.type = pj_ntohs(attr->hdr.type); + attr->hdr.length = pj_ntohs(attr->hdr.length); + attr->value = pj_ntohl(attr->value); + + /* Check that the attribute length is valid */ + if (attr->hdr.length != STUN_UINT_LEN) + return PJLIB_UTIL_ESTUNINATTRLEN; + + /* Done */ + *p_attr = attr; + + return PJ_SUCCESS; +} + + +static pj_status_t encode_generic_uint_attr(const void *a, pj_uint8_t *buf, + unsigned len, unsigned *printed) +{ + enum + { + ATTR_LEN = STUN_UINT_LEN + ATTR_HDR_LEN + }; + pj_stun_generic_uint_attr *attr; + + if (len < ATTR_LEN) + return PJ_ETOOSMALL; + + /* Copy and convert attribute to network byte order */ + pj_memcpy(buf, a, ATTR_LEN); + attr = (pj_stun_generic_uint_attr*) buf; + attr->hdr.type = pj_htons(attr->hdr.type); + pj_assert(attr->hdr.length == STUN_UINT_LEN); + attr->hdr.length = pj_htons(STUN_UINT_LEN); + attr->value = pj_htonl(attr->value); + + /* Done */ + *printed = ATTR_LEN; + + return PJ_SUCCESS; +} + +////////////////////////////////////////////////////////////////////////////// +/* + * STUN MESSAGE-INTEGRITY attribute. + */ + +#define STUN_MSG_INTEGRITY_LEN 20 + +/* + * Create a STUN MESSAGE-INTEGRITY attribute. + */ +PJ_DEF(pj_status_t) +pj_stun_msg_integrity_attr_create(pj_pool_t *pool, + pj_stun_msg_integrity_attr **p_attr) +{ + pj_stun_msg_integrity_attr *attr; + + PJ_ASSERT_RETURN(pool && p_attr, PJ_EINVAL); + + attr = PJ_POOL_ZALLOC_TYPE(pool, pj_stun_msg_integrity_attr); + INIT_ATTR(attr, PJ_STUN_ATTR_MESSAGE_INTEGRITY, STUN_MSG_INTEGRITY_LEN); + + *p_attr = attr; + + return PJ_SUCCESS; +} + + +static pj_status_t decode_msg_integrity_attr(pj_pool_t *pool, + const pj_uint8_t *buf, + void **p_attr) +{ + enum + { + ATTR_LEN = STUN_MSG_INTEGRITY_LEN + ATTR_HDR_LEN + }; + pj_stun_msg_integrity_attr *attr; + + /* Check that struct size is valid */ + pj_assert(sizeof(pj_stun_msg_integrity_attr)==STUN_MSG_INTEGRITY_LEN); + + /* Create attribute */ + attr = PJ_POOL_ZALLOC_TYPE(pool, pj_stun_msg_integrity_attr); + pj_memcpy(attr, buf, sizeof(pj_stun_msg_integrity_attr)); + attr->hdr.type = pj_ntohs(attr->hdr.type); + attr->hdr.length = pj_ntohs(attr->hdr.length); + + /* Check that the attribute length is valid */ + if (attr->hdr.length != STUN_MSG_INTEGRITY_LEN) + return PJLIB_UTIL_ESTUNINATTRLEN; + + /* Done */ + *p_attr = attr; + return PJ_SUCCESS; +} + + +static pj_status_t encode_msg_integrity_attr(const void *a, pj_uint8_t *buf, + unsigned len, unsigned *printed) +{ + enum + { + ATTR_LEN = STUN_MSG_INTEGRITY_LEN + ATTR_HDR_LEN + }; + pj_stun_msg_integrity_attr *attr; + + if (len < ATTR_LEN) + return PJ_ETOOSMALL; + + /* Copy and convert attribute to network byte order */ + pj_memcpy(buf, a, ATTR_LEN); + attr = (pj_stun_msg_integrity_attr*) buf; + attr->hdr.type = pj_htons(attr->hdr.type); + pj_assert(attr->hdr.length == STUN_MSG_INTEGRITY_LEN); + attr->hdr.length = pj_htons(STUN_MSG_INTEGRITY_LEN); + + /* Done */ + *printed = ATTR_LEN; + + return PJ_SUCCESS; +} + +////////////////////////////////////////////////////////////////////////////// +/* + * STUN ERROR-CODE + */ + +/* + * Create a STUN ERROR-CODE attribute. + */ +PJ_DEF(pj_status_t) +pj_stun_error_code_attr_create(pj_pool_t *pool, + int err_code, + const pj_str_t *err_reason, + pj_stun_error_code_attr **p_attr) +{ + pj_stun_error_code_attr *attr; + char err_buf[80]; + pj_str_t str; + + PJ_ASSERT_RETURN(pool && err_code && p_attr, PJ_EINVAL); + + if (err_reason == NULL) { + str = pj_stun_get_err_reason(err_code); + if (str.slen == 0) { + str.slen = pj_ansi_snprintf(err_buf, sizeof(err_buf), + "Unknown error %d", err_code); + str.ptr = err_buf; + } + err_reason = &str; + } + + attr = PJ_POOL_ZALLOC_TYPE(pool, pj_stun_error_code_attr); + INIT_ATTR(attr, PJ_STUN_ATTR_ERROR_CODE, 4+err_reason->slen); + attr->err_class = (pj_uint8_t)(err_code / 100); + attr->number = (pj_uint8_t) (err_code % 100); + pj_strdup(pool, &attr->reason, err_reason); + + *p_attr = attr; + + return PJ_SUCCESS; +} + + +static pj_status_t decode_error_code_attr(pj_pool_t *pool, + const pj_uint8_t *buf, + void **p_attr) +{ + pj_stun_error_code_attr *attr; + pj_str_t value; + + /* Create the attribute */ + attr = PJ_POOL_ZALLOC_TYPE(pool, pj_stun_error_code_attr); + + /* Copy the header */ + pj_memcpy(attr, buf, ATTR_HDR_LEN + 4); + + /* Convert to host byte order */ + attr->hdr.type = pj_ntohs(attr->hdr.type); + attr->hdr.length = pj_ntohs(attr->hdr.length); + + /* Get pointer to the string in the message */ + value.ptr = ((char*)buf + ATTR_HDR_LEN + 4); + value.slen = attr->hdr.length - 4; + + /* Copy the string to the attribute */ + pj_strdup(pool, &attr->reason, &value); + + /* Done */ + *p_attr = attr; + + return PJ_SUCCESS; +} + + +static pj_status_t encode_error_code_attr(const void *a, pj_uint8_t *buf, + unsigned len, unsigned *printed) +{ + const pj_stun_error_code_attr *ca = + (const pj_stun_error_code_attr*)a; + pj_stun_error_code_attr *attr; + + if (len < ATTR_HDR_LEN + 4 + (unsigned)ca->reason.slen) + return PJ_ETOOSMALL; + + /* Copy and convert attribute to network byte order */ + pj_memcpy(buf, ca, ATTR_HDR_LEN + 4); + + /* Update length */ + attr = (pj_stun_error_code_attr*) buf; + attr->hdr.length = (pj_uint16_t)(4 + ca->reason.slen); + + /* Convert fiends to network byte order */ + attr->hdr.type = pj_htons(attr->hdr.type); + attr->hdr.length = pj_htons(attr->hdr.length); + + /* Copy error string */ + pj_memcpy(buf + ATTR_HDR_LEN + 4, ca->reason.ptr, ca->reason.slen); + + /* Done */ + *printed = (ATTR_HDR_LEN + 4 + ca->reason.slen + 3) & (~3); + + return PJ_SUCCESS; +} + +////////////////////////////////////////////////////////////////////////////// +/* + * STUN UNKNOWN-ATTRIBUTES attribute + */ + +/* + * Create an empty instance of STUN UNKNOWN-ATTRIBUTES attribute. + * + * @param pool The pool to allocate memory from. + * @param p_attr Pointer to receive the attribute. + * + * @return PJ_SUCCESS on success or the appropriate error code. + */ +PJ_DEF(pj_status_t) +pj_stun_unknown_attr_create(pj_pool_t *pool, + unsigned attr_cnt, + pj_uint16_t attr_array[], + pj_stun_unknown_attr **p_attr) +{ + pj_stun_unknown_attr *attr; + unsigned i; + + PJ_ASSERT_RETURN(pool && attr_cnt < PJ_STUN_MAX_ATTR && p_attr, PJ_EINVAL); + + attr = PJ_POOL_ZALLOC_TYPE(pool, pj_stun_unknown_attr); + INIT_ATTR(attr, PJ_STUN_ATTR_UNKNOWN_ATTRIBUTES, attr_cnt * 2); + + attr->attr_count = attr_cnt; + for (i=0; iattrs[i] = attr_array[i]; + } + + /* If the number of unknown attributes is an odd number, one of the + * attributes MUST be repeated in the list. + */ + if ((attr_cnt & 0x01)) { + attr->attrs[attr_cnt] = attr_array[attr_cnt-1]; + } + + *p_attr = NULL; + + return PJ_SUCCESS; +} + + +static pj_status_t decode_unknown_attr(pj_pool_t *pool, + const pj_uint8_t *buf, + void **p_attr) +{ + pj_stun_unknown_attr *attr; + const pj_uint16_t *punk_attr; + unsigned i; + + attr = PJ_POOL_ZALLOC_TYPE(pool, pj_stun_unknown_attr); + pj_memcpy(attr, buf, ATTR_HDR_LEN); + + attr->hdr.type = pj_ntohs(attr->hdr.type); + attr->hdr.length = pj_ntohs(attr->hdr.length); + + attr->attr_count = (attr->hdr.length >> 1); + if (attr->attr_count > PJ_STUN_MAX_ATTR) + return PJ_ETOOMANY; + + punk_attr = (const pj_uint16_t*)(buf + ATTR_HDR_LEN); + for (i=0; iattr_count; ++i) { + attr->attrs[i] = pj_ntohs(punk_attr[i]); + } + + /* Done */ + *p_attr = attr; + + return PJ_SUCCESS; +} + + +static pj_status_t encode_unknown_attr(const void *a, pj_uint8_t *buf, + unsigned len, unsigned *printed) +{ + const pj_stun_unknown_attr *ca = (const pj_stun_unknown_attr*) a; + pj_stun_unknown_attr *attr; + pj_uint16_t *dst_unk_attr; + unsigned i; + + /* Check that buffer is enough */ + if (len < ATTR_HDR_LEN + (ca->attr_count << 1)) + return PJ_ETOOSMALL; + + /* Copy to message */ + pj_memcpy(buf, ca, ATTR_HDR_LEN); + + /* Set the correct length */ + attr = (pj_stun_unknown_attr *) buf; + attr->hdr.length = (pj_uint16_t)(ca->attr_count << 1); + + /* Convert to network byte order */ + attr->hdr.type = pj_htons(attr->hdr.type); + attr->hdr.length = pj_htons(attr->hdr.length); + + /* Copy individual attribute */ + dst_unk_attr = (pj_uint16_t*)(buf + ATTR_HDR_LEN); + for (i=0; i < ca->attr_count; ++i, ++dst_unk_attr) { + *dst_unk_attr = pj_htons(attr->attrs[i]); + } + + /* Done */ + *printed = (ATTR_HDR_LEN + (ca->attr_count << 1) + 3) & (~3); + + return PJ_SUCCESS; +} + + +////////////////////////////////////////////////////////////////////////////// +/* + * STUN generic binary attribute + */ + +/* + * Create a blank binary attribute. + */ +PJ_DEF(pj_status_t) +pj_stun_binary_attr_create(pj_pool_t *pool, + int attr_type, + pj_stun_binary_attr **p_attr) +{ + pj_stun_binary_attr *attr; + + PJ_ASSERT_RETURN(pool && attr_type && p_attr, PJ_EINVAL); + + attr = PJ_POOL_ZALLOC_TYPE(pool, pj_stun_binary_attr); + INIT_ATTR(attr, attr_type, sizeof(pj_stun_binary_attr)); + + *p_attr = attr; + + return PJ_SUCCESS; +} + + +static pj_status_t decode_binary_attr(pj_pool_t *pool, + const pj_uint8_t *buf, + void **p_attr) +{ + pj_stun_binary_attr *attr; + + /* Create the attribute */ + attr = PJ_POOL_ZALLOC_TYPE(pool, pj_stun_binary_attr); + + /* Copy the header */ + pj_memcpy(attr, buf, ATTR_HDR_LEN); + + /* Convert to host byte order */ + attr->hdr.type = pj_ntohs(attr->hdr.type); + attr->hdr.length = pj_ntohs(attr->hdr.length); + + /* Copy the data to the attribute */ + attr->length = attr->hdr.length; + attr->data = pj_pool_alloc(pool, attr->length); + pj_memcpy(attr->data, buf+ATTR_HDR_LEN, attr->length); + + /* Done */ + *p_attr = attr; + + return PJ_SUCCESS; + +} + + +static pj_status_t encode_binary_attr(const void *a, pj_uint8_t *buf, + unsigned len, unsigned *printed) +{ + const pj_stun_binary_attr *ca = (const pj_stun_binary_attr*)a; + pj_stun_attr_hdr *attr; + + /* Calculated total attr_len (add padding if necessary) */ + *printed = (ca->length + ATTR_HDR_LEN + 3) & (~3); + if (len < *printed) + return PJ_ETOOSMALL; + + /* Copy header */ + pj_memcpy(buf, a, ATTR_HDR_LEN); + + /* Set the correct length */ + attr = (pj_stun_attr_hdr*)buf; + attr->length = (pj_uint16_t) ca->length; + + /* Convert to network byte order */ + attr->type = pj_htons(attr->type); + attr->length = pj_htons(attr->length); + + /* Copy the data */ + pj_memcpy(buf+ATTR_HDR_LEN, ca->data, ca->length); + + /* Done */ + return PJ_SUCCESS; +} + + +////////////////////////////////////////////////////////////////////////////// + +/* + * Create a blank STUN message. + */ +PJ_DEF(pj_status_t) pj_stun_msg_create( pj_pool_t *pool, + unsigned msg_type, + pj_uint32_t magic, + const pj_uint8_t tsx_id[12], + pj_stun_msg **p_msg) +{ + pj_stun_msg *msg; + + PJ_ASSERT_RETURN(pool && msg_type && p_msg, PJ_EINVAL); + + msg = PJ_POOL_ZALLOC_TYPE(pool, pj_stun_msg); + msg->hdr.type = (pj_uint16_t) msg_type; + msg->hdr.magic = magic; + + if (tsx_id) { + pj_memcpy(&msg->hdr.tsx_id, tsx_id, sizeof(msg->hdr.tsx_id)); + } else { + struct transaction_id + { + pj_uint32_t proc_id; + pj_uint32_t random; + pj_uint32_t counter; + } id; + static pj_uint32_t pj_stun_tsx_id_counter; + + id.proc_id = pj_getpid(); + id.random = pj_rand(); + id.counter = pj_stun_tsx_id_counter++; + + pj_memcpy(&msg->hdr.tsx_id, &id, sizeof(msg->hdr.tsx_id)); + } + + *p_msg = msg; + + return PJ_SUCCESS; +} + + +/* + * Add STUN attribute to STUN message. + */ +PJ_DEF(pj_status_t) pj_stun_msg_add_attr(pj_stun_msg *msg, + pj_stun_attr_hdr *attr) +{ + PJ_ASSERT_RETURN(msg && attr, PJ_EINVAL); + PJ_ASSERT_RETURN(msg->attr_count < PJ_STUN_MAX_ATTR, PJ_ETOOMANY); + + msg->attr[msg->attr_count++] = attr; + return PJ_SUCCESS; +} + + +/* + * Check that the PDU is potentially a valid STUN message. + */ +PJ_DEF(pj_status_t) pj_stun_msg_check(const void *pdu, unsigned pdu_len, + unsigned options) +{ + pj_stun_msg_hdr *hdr; + + PJ_ASSERT_RETURN(pdu && pdu_len > sizeof(pj_stun_msg_hdr), + PJLIB_UTIL_ESTUNINMSGLEN); + + PJ_UNUSED_ARG(options); + + hdr = (pj_stun_msg_hdr*) pdu; + + /* First byte of STUN message is always 0x00 or 0x01. */ + if ((*(const char*)pdu) != 0x00 && (*(const char*)pdu) != 0x01) + return PJLIB_UTIL_ESTUNINMSGTYPE; + + /* If magic is set, then there is great possibility that this is + * a STUN message. + */ + if (pj_ntohl(hdr->magic) == PJ_STUN_MAGIC) + return PJ_SUCCESS; + + /* Check the PDU length */ + if (pj_ntohs(hdr->length) > pdu_len) + return PJLIB_UTIL_ESTUNINMSGLEN; + + /* Could be a STUN message */ + return PJ_SUCCESS; +} + + +/* + * Parse incoming packet into STUN message. + */ +PJ_DEF(pj_status_t) pj_stun_msg_decode(pj_pool_t *pool, + const pj_uint8_t *pdu, + unsigned pdu_len, + unsigned options, + pj_stun_msg **p_msg, + unsigned *p_parsed_len, + unsigned *p_err_code, + unsigned *p_uattr_cnt, + pj_uint16_t uattr[]) +{ + + pj_stun_msg *msg; + unsigned uattr_cnt; + const pj_uint8_t *start_pdu = pdu; + pj_status_t status; + + PJ_UNUSED_ARG(options); + + PJ_ASSERT_RETURN(pool && pdu && pdu_len && p_msg, PJ_EINVAL); + PJ_ASSERT_RETURN(sizeof(pj_stun_msg_hdr) == 20, PJ_EBUG); + + /* Application should have checked that this is a valid STUN msg */ + PJ_ASSERT_RETURN((status=pj_stun_msg_check(pdu, pdu_len, options)) + == PJ_SUCCESS, status); + + /* Create the message, copy the header, and convert to host byte order */ + msg = PJ_POOL_ZALLOC_TYPE(pool, pj_stun_msg); + pj_memcpy(&msg->hdr, pdu, sizeof(pj_stun_msg_hdr)); + msg->hdr.type = pj_ntohs(msg->hdr.type); + msg->hdr.length = pj_ntohs(msg->hdr.length); + msg->hdr.magic = pj_ntohl(msg->hdr.magic); + + pdu += sizeof(pj_stun_msg_hdr); + pdu_len -= sizeof(pj_stun_msg_hdr); + + if (p_err_code) + *p_err_code = 0; + + /* Parse attributes */ + uattr_cnt = 0; + while (pdu_len > 0) { + unsigned attr_type, attr_val_len; + const struct attr_desc *adesc; + + /* Get attribute type and length. If length is not aligned + * to 4 bytes boundary, add padding. + */ + attr_type = pj_ntohs(*(pj_uint16_t*)pdu); + attr_val_len = pj_ntohs(*(pj_uint16_t*)(pdu+2)); + attr_val_len = (attr_val_len + 3) & (~3); + + /* Check length */ + if (pdu_len < attr_val_len) + return PJLIB_UTIL_ESTUNINATTRLEN; + + /* Get the attribute descriptor */ + adesc = find_attr_desc(attr_type); + + if (adesc == NULL) { + /* Unrecognized attribute */ + + PJ_LOG(4,(THIS_FILE, "Unrecognized attribute type %d", + attr_type)); + + /* Put to unrecognized attribute array */ + if (p_uattr_cnt && uattr && uattr_cnt < *p_uattr_cnt) { + uattr[uattr_cnt++] = (pj_uint16_t)attr_type; + } + + /* Is this a fatal condition? */ + if (attr_type <= 0x7FFF) { + /* This is a mandatory attribute, we must return error + * if we don't understand the attribute. + */ + if (p_err_code && *p_err_code == 0) + *p_err_code = PJ_STUN_STATUS_UNKNOWN_ATTRIBUTE; + + return PJLIB_UTIL_ESTUNUNKNOWNATTR; + } + + } else { + void *attr; + + /* Parse the attribute */ + status = (adesc->decode_attr)(pool, pdu, &attr); + + if (status != PJ_SUCCESS) { + PJ_LOG(4,(THIS_FILE, + "Error parsing STUN attribute type %d: status=%d", + attr_type, status)); + return status; + } + + /* Make sure we have rooms for the new attribute */ + if (msg->attr_count >= PJ_STUN_MAX_ATTR) + return PJLIB_UTIL_ESTUNTOOMANYATTR; + + /* Add the attribute */ + msg->attr[msg->attr_count++] = (pj_stun_attr_hdr*)attr; + } + + pdu += (attr_val_len + 4); + pdu_len -= (attr_val_len + 4); + } + + *p_msg = msg; + + if (p_uattr_cnt) + *p_uattr_cnt = uattr_cnt; + + if (p_parsed_len) + *p_parsed_len = (pdu - start_pdu); + + return PJ_SUCCESS; +} + + +/* + * Print the message structure to a buffer. + */ +PJ_DEF(pj_status_t) pj_stun_msg_encode(const pj_stun_msg *msg, + pj_uint8_t *buf, unsigned buf_size, + unsigned options, + unsigned *p_msg_len) +{ + pj_stun_msg_hdr *hdr; + pj_uint8_t *start = buf; + unsigned i; + + PJ_ASSERT_RETURN(msg && buf && buf_size, PJ_EINVAL); + + PJ_UNUSED_ARG(options); + + /* Copy the message header part and convert the header fields to + * network byte order + */ + if (buf_size < sizeof(pj_stun_msg_hdr)) + return PJ_ETOOSMALL; + pj_memcpy(buf, &msg->hdr, sizeof(pj_stun_msg_hdr)); + hdr = (pj_stun_msg_hdr*) buf; + hdr->magic = pj_htonl(hdr->magic); + hdr->type = pj_htons(hdr->type); + /* We'll fill in the length later */ + + buf += sizeof(pj_stun_msg_hdr); + buf_size -= sizeof(pj_stun_msg_hdr); + + /* Print each attribute */ + for (i=0; iattr_count; ++i) { + const struct attr_desc *adesc; + const pj_stun_attr_hdr *attr_hdr; + unsigned printed; + pj_status_t status; + + attr_hdr = msg->attr[i]; + + adesc = find_attr_desc(attr_hdr->type); + PJ_ASSERT_RETURN(adesc != NULL, PJ_EBUG); + + status = adesc->encode_attr(attr_hdr, buf, buf_size, &printed); + if (status != PJ_SUCCESS) + return status; + + buf += printed; + buf_size -= printed; + } + + /* Update the message length in the header. + * Note that length is not including the 20 bytes header. + */ + hdr->length = (pj_uint16_t)((buf - start) - 20); + hdr->length = pj_htons(hdr->length); + + /* Done */ + if (p_msg_len) + *p_msg_len = (buf - start); + + return PJ_SUCCESS; +} + + +/* + * Find STUN attribute in the STUN message, starting from the specified + * index. + */ +PJ_DEF(pj_stun_attr_hdr*) pj_stun_msg_find_attr( const pj_stun_msg *msg, + int attr_type, + unsigned index) +{ + PJ_ASSERT_RETURN(msg, NULL); + + for (; index < msg->attr_count; ++index) { + if (msg->attr[index]->type == attr_type) + return (pj_stun_attr_hdr*) &msg->attr[index]; + } + + return NULL; +} + diff --git a/pjlib-util/src/pjlib-util/stun_server.c b/pjlib-util/src/pjlib-util/stun_server.c new file mode 100644 index 00000000..d4d7fa8f --- /dev/null +++ b/pjlib-util/src/pjlib-util/stun_server.c @@ -0,0 +1,129 @@ +/* $Id$ */ +/* + * Copyright (C) 2003-2005 Benny Prijono + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + */ +#include +#include +#include +#include +#include + + +struct pj_stun_service +{ + pj_str_t name; + unsigned options; + void *user_data; + unsigned cb_cnt; + pj_stun_service_handler *cb; +}; + + +/* + * Create STUN service. + */ +PJ_DEF(pj_status_t) pj_stun_service_create( pj_pool_t *pool, + const char *name, + unsigned options, + unsigned handler_cnt, + pj_stun_service_handler cb[], + void *user_data, + pj_stun_service **p_svc) +{ + pj_stun_service *svc; + + PJ_ASSERT_RETURN(pool && handler_cnt && cb && p_svc, PJ_EINVAL); + + svc = PJ_POOL_ZALLOC_TYPE(pool, pj_stun_service); + svc->options = options; + svc->user_data = user_data; + + if (!name) name = "pj_stun_service"; + + pj_strdup2_with_null(pool, &svc->name, name); + + svc->cb_cnt = handler_cnt; + svc->cb = pj_pool_calloc(pool, handler_cnt, + sizeof(pj_stun_service_handler)); + pj_memcpy(svc->cb, cb, handler_cnt * sizeof(pj_stun_service_handler)); + + *p_svc = svc; + + return PJ_SUCCESS; +} + + +/* + * Destroy STUN service + */ +PJ_DEF(pj_status_t) pj_stun_service_destroy(pj_stun_service *svc) +{ + PJ_ASSERT_RETURN(svc, PJ_EINVAL); + return PJ_SUCCESS; +} + + +/* + * Get user data associated with the STUN service. + */ +PJ_DEF(void*) pj_stun_service_get_user_data(pj_stun_service *svc) +{ + PJ_ASSERT_RETURN(svc, NULL); + return svc->user_data; +} + + +/* + * Find handler. + */ +static pj_stun_service_handler *find_handler(pj_stun_service *svc, + int msg_type) +{ + unsigned i; + + for (i=0; icb_cnt; ++i) { + if (svc->cb[i].msg_type == msg_type) + return &svc->cb[i]; + } + + return NULL; +} + + +/* + * Instruct the STUN service to handle incoming STUN message. + */ +PJ_DEF(pj_status_t) pj_stun_service_handle_msg( pj_stun_service *svc, + void *handle_data, + const pj_stun_msg *msg) +{ + pj_stun_service_handler *handler; + + PJ_ASSERT_RETURN(svc && msg, PJ_EINVAL); + + handler = find_handler(svc, msg->hdr.type); + if (handler == NULL) + return PJLIB_UTIL_ESTUNNOHANDLER; + + return (*handler->handle_msg)(svc, handle_data, msg); +} + + +////////////////////////////////////////////////////////////////////////////// + + + diff --git a/pjlib-util/src/pjlib-util/stun_simple.c b/pjlib-util/src/pjlib-util/stun_simple.c new file mode 100644 index 00000000..6ab6dbbe --- /dev/null +++ b/pjlib-util/src/pjlib-util/stun_simple.c @@ -0,0 +1,125 @@ +/* $Id$ */ +/* + * Copyright (C) 2003-2006 Benny Prijono + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + */ +#include +#include +#include +#include +#include +#include + +#define THIS_FILE "stun.c" + +PJ_DEF(pj_status_t) pjstun_create_bind_req( pj_pool_t *pool, + void **msg, pj_size_t *len, + pj_uint32_t id_hi, + pj_uint32_t id_lo) +{ + pjstun_msg_hdr *hdr; + + PJ_CHECK_STACK(); + + + hdr = pj_pool_calloc(pool, 1, sizeof(pjstun_msg_hdr)); + if (!hdr) + return PJ_ENOMEM; + + hdr->type = pj_htons(PJSTUN_BINDING_REQUEST); + hdr->tsx[2] = pj_htonl(id_hi); + hdr->tsx[3] = pj_htonl(id_lo); + *msg = hdr; + *len = sizeof(pjstun_msg_hdr); + + return PJ_SUCCESS; +} + +PJ_DEF(pj_status_t) pjstun_parse_msg( void *buf, pj_size_t len, + pjstun_msg *msg) +{ + pj_uint16_t msg_type, msg_len; + char *p_attr; + + PJ_CHECK_STACK(); + + msg->hdr = (pjstun_msg_hdr*)buf; + msg_type = pj_ntohs(msg->hdr->type); + + switch (msg_type) { + case PJSTUN_BINDING_REQUEST: + case PJSTUN_BINDING_RESPONSE: + case PJSTUN_BINDING_ERROR_RESPONSE: + case PJSTUN_SHARED_SECRET_REQUEST: + case PJSTUN_SHARED_SECRET_RESPONSE: + case PJSTUN_SHARED_SECRET_ERROR_RESPONSE: + break; + default: + PJ_LOG(4,(THIS_FILE, "Error: unknown msg type %d", msg_type)); + return PJLIB_UTIL_ESTUNINMSGTYPE; + } + + msg_len = pj_ntohs(msg->hdr->length); + if (msg_len != len - sizeof(pjstun_msg_hdr)) { + PJ_LOG(4,(THIS_FILE, "Error: invalid msg_len %d (expecting %d)", + msg_len, len - sizeof(pjstun_msg_hdr))); + return PJLIB_UTIL_ESTUNINMSGLEN; + } + + msg->attr_count = 0; + p_attr = (char*)buf + sizeof(pjstun_msg_hdr); + + while (msg_len > 0) { + pjstun_attr_hdr **attr = &msg->attr[msg->attr_count]; + pj_uint32_t len; + + *attr = (pjstun_attr_hdr*)p_attr; + len = pj_ntohs((pj_uint16_t) ((*attr)->length)) + sizeof(pjstun_attr_hdr); + + if (msg_len < len) { + PJ_LOG(4,(THIS_FILE, "Error: length mismatch in attr %d", + msg->attr_count)); + return PJLIB_UTIL_ESTUNINATTRLEN; + } + + if (pj_ntohs((*attr)->type) > PJSTUN_ATTR_REFLECTED_FORM) { + PJ_LOG(5,(THIS_FILE, "Warning: unknown attr type %x in attr %d. " + "Attribute was ignored.", + pj_ntohs((*attr)->type), msg->attr_count)); + } + + msg_len = (pj_uint16_t)(msg_len - len); + p_attr += len; + ++msg->attr_count; + } + + return PJ_SUCCESS; +} + +PJ_DEF(void*) pjstun_msg_find_attr( pjstun_msg *msg, pjstun_attr_type t) +{ + int i; + + PJ_CHECK_STACK(); + + for (i=0; iattr_count; ++i) { + pjstun_attr_hdr *attr = msg->attr[i]; + if (pj_ntohs(attr->type) == t) + return attr; + } + + return 0; +} diff --git a/pjlib-util/src/pjlib-util/stun_simple_client.c b/pjlib-util/src/pjlib-util/stun_simple_client.c new file mode 100644 index 00000000..7d6e472b --- /dev/null +++ b/pjlib-util/src/pjlib-util/stun_simple_client.c @@ -0,0 +1,241 @@ +/* $Id$ */ +/* + * Copyright (C) 2003-2006 Benny Prijono + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + */ +#include +#include +#include +#include +#include +#include +#include + + +enum { MAX_REQUEST = 3 }; +static int stun_timer[] = {1600, 1600, 1600 }; + +#define THIS_FILE "stun_client.c" +#define LOG_ADDR(addr) pj_inet_ntoa(addr.sin_addr), pj_ntohs(addr.sin_port) + + +PJ_DECL(pj_status_t) pjstun_get_mapped_addr( pj_pool_factory *pf, + int sock_cnt, pj_sock_t sock[], + const pj_str_t *srv1, int port1, + const pj_str_t *srv2, int port2, + pj_sockaddr_in mapped_addr[]) +{ + pj_sockaddr_in srv_addr[2]; + int i, j, send_cnt = 0; + pj_pool_t *pool; + struct { + struct { + pj_uint32_t mapped_addr; + pj_uint32_t mapped_port; + } srv[2]; + } *rec; + void *out_msg; + pj_size_t out_msg_len; + int wait_resp = 0; + pj_status_t status; + + PJ_CHECK_STACK(); + + /* Create pool. */ + pool = pj_pool_create(pf, "stun%p", 1024, 1024, NULL); + if (!pool) + return PJ_ENOMEM; + + + /* Allocate client records */ + rec = pj_pool_calloc(pool, sock_cnt, sizeof(*rec)); + if (!rec) { + status = PJ_ENOMEM; + goto on_error; + } + + + /* Create the outgoing BIND REQUEST message template */ + status = pjstun_create_bind_req( pool, &out_msg, &out_msg_len, + pj_rand(), pj_rand()); + if (status != PJ_SUCCESS) + goto on_error; + + /* Resolve servers. */ + status = pj_sockaddr_in_init(&srv_addr[0], srv1, (pj_uint16_t)port1); + if (status != PJ_SUCCESS) + goto on_error; + + status = pj_sockaddr_in_init(&srv_addr[1], srv2, (pj_uint16_t)port2); + if (status != PJ_SUCCESS) + goto on_error; + + /* Init mapped addresses to zero */ + pj_memset(mapped_addr, 0, sock_cnt * sizeof(pj_sockaddr_in)); + + /* Main retransmission loop. */ + for (send_cnt=0; send_cnttsx[2] = pj_htonl(i); + msg_hdr->tsx[3] = pj_htonl(j); + + /* Send! */ + sent_len = out_msg_len; + status = pj_sock_sendto(sock[i], out_msg, &sent_len, 0, + (pj_sockaddr_t*)&srv_addr[j], + sizeof(pj_sockaddr_in)); + if (status == PJ_SUCCESS) + ++wait_resp; + } + } + + /* All requests sent. + * The loop below will wait for responses until all responses have + * been received (i.e. wait_resp==0) or timeout occurs, which then + * we'll go to the next retransmission iteration. + */ + + /* Calculate time of next retransmission. */ + pj_gettimeofday(&next_tx); + next_tx.sec += (stun_timer[send_cnt]/1000); + next_tx.msec += (stun_timer[send_cnt]%1000); + pj_time_val_normalize(&next_tx); + + for (pj_gettimeofday(&now), select_rc=1; + status==PJ_SUCCESS && select_rc==1 && wait_resp>0 + && PJ_TIME_VAL_LT(now, next_tx); + pj_gettimeofday(&now)) + { + pj_time_val timeout; + + timeout = next_tx; + PJ_TIME_VAL_SUB(timeout, now); + + for (i=0; itsx[2]); + srv_idx = pj_ntohl(msg.hdr->tsx[3]); + + if (sock_idx<0 || sock_idx>=sock_cnt || srv_idx<0 || srv_idx>=2) { + status = PJLIB_UTIL_ESTUNININDEX; + continue; + } + + if (pj_ntohs(msg.hdr->type) != PJSTUN_BINDING_RESPONSE) { + status = PJLIB_UTIL_ESTUNNOBINDRES; + continue; + } + + if (pjstun_msg_find_attr(&msg, PJSTUN_ATTR_ERROR_CODE) != NULL) { + status = PJLIB_UTIL_ESTUNRECVERRATTR; + continue; + } + + attr = (void*)pjstun_msg_find_attr(&msg, PJSTUN_ATTR_MAPPED_ADDR); + if (!attr) { + status = PJLIB_UTIL_ESTUNNOMAP; + continue; + } + + rec[sock_idx].srv[srv_idx].mapped_addr = attr->addr; + rec[sock_idx].srv[srv_idx].mapped_port = attr->port; + } + } + + /* The best scenario is if all requests have been replied. + * Then we don't need to go to the next retransmission iteration. + */ + if (wait_resp <= 0) + break; + } + + for (i=0; i + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + */ +#include +#include +#include +#include +#include +#include +#include + + +#define TIMER_ACTIVE 1 + + +struct pj_stun_client_tsx +{ + char obj_name[PJ_MAX_OBJ_NAME]; + pj_pool_t *pool; + pj_stun_endpoint *endpt; + pj_stun_tsx_cb cb; + void *user_data; + + pj_uint32_t tsx_id[4]; + + pj_bool_t require_retransmit; + pj_timer_entry timer; + unsigned transmit_count; + pj_time_val retransmit_time; + + pj_uint8_t last_pkt[PJ_STUN_MAX_PKT_LEN]; + unsigned last_pkt_size; +}; + + +static void retransmit_timer_callback(pj_timer_heap_t *timer_heap, + pj_timer_entry *timer); + +static void stun_perror(pj_stun_client_tsx *tsx, const char *title, + pj_status_t status) +{ + char errmsg[PJ_ERR_MSG_SIZE]; + + pj_strerror(status, errmsg, sizeof(errmsg)); + PJ_LOG(4,(tsx->obj_name, "%s: %s", title, errmsg)); +} + + +/* + * Create a STUN client transaction. + */ +PJ_DEF(pj_status_t) pj_stun_client_tsx_create(pj_stun_endpoint *endpt, + const pj_stun_tsx_cb *cb, + pj_stun_client_tsx **p_tsx) +{ + pj_pool_t *pool; + pj_stun_client_tsx *tsx; + + PJ_ASSERT_RETURN(endpt && cb && p_tsx, PJ_EINVAL); + PJ_ASSERT_RETURN(cb->on_send_msg, PJ_EINVAL); + + pool = pj_pool_create(endpt->pf, "tsx", 1000, 1000, NULL); + tsx = PJ_POOL_ZALLOC_TYPE(pool, pj_stun_client_tsx); + tsx->pool = pool; + tsx->endpt = endpt; + pj_memcpy(&tsx->cb, cb, sizeof(*cb)); + + tsx->timer.cb = &retransmit_timer_callback; + tsx->timer.user_data = tsx; + + pj_ansi_snprintf(tsx->obj_name, sizeof(tsx->obj_name), "stuntsx%p", pool); + + *p_tsx = tsx; + + PJ_LOG(4,(tsx->obj_name, "STUN client transaction created")); + return PJ_SUCCESS; +} + + +/* + * . + */ +PJ_DEF(pj_status_t) pj_stun_client_tsx_destroy(pj_stun_client_tsx *tsx) +{ + PJ_ASSERT_RETURN(tsx, PJ_EINVAL); + + if (tsx->timer.id != 0) { + pj_timer_heap_cancel(tsx->endpt->timer_heap, &tsx->timer); + tsx->timer.id = 0; + } + pj_pool_release(tsx->pool); + return PJ_SUCCESS; +} + + +/* + * Set user data. + */ +PJ_DEF(pj_status_t) pj_stun_client_tsx_set_data(pj_stun_client_tsx *tsx, + void *data) +{ + PJ_ASSERT_RETURN(tsx, PJ_EINVAL); + tsx->user_data = data; + return PJ_SUCCESS; +} + + +/* + * Get the user data + */ +PJ_DEF(void*) pj_stun_client_tsx_get_data(pj_stun_client_tsx *tsx) +{ + PJ_ASSERT_RETURN(tsx, NULL); + return tsx->user_data; +} + + +/* + * Transmit message. + */ +static pj_status_t tsx_transmit_msg(pj_stun_client_tsx *tsx) +{ + pj_status_t status; + + PJ_ASSERT_RETURN(tsx->timer.id == 0, PJ_EBUSY); + + if (tsx->require_retransmit) { + /* Calculate retransmit/timeout delay */ + if (tsx->transmit_count == 0) { + tsx->retransmit_time.sec = 0; + tsx->retransmit_time.msec = tsx->endpt->rto_msec; + + } else if (tsx->transmit_count < PJ_STUN_MAX_RETRANSMIT_COUNT) { + unsigned msec; + + msec = PJ_TIME_VAL_MSEC(tsx->retransmit_time); + msec = (msec >> 1) + 100; + tsx->retransmit_time.sec = msec / 1000; + tsx->retransmit_time.msec = msec % 100; + + } else { + tsx->retransmit_time.sec = PJ_STUN_TIMEOUT_VALUE / 1000; + tsx->retransmit_time.msec = PJ_STUN_TIMEOUT_VALUE % 1000; + } + + /* Schedule timer first because when send_msg() failed we can + * cancel it (as opposed to when schedule_timer() failed we cannot + * cancel transmission). + */ + status = pj_timer_heap_schedule(tsx->endpt->timer_heap, &tsx->timer, + &tsx->retransmit_time); + if (status != PJ_SUCCESS) { + tsx->timer.id = 0; + return status; + } + } + + + /* Send message */ + status = tsx->cb.on_send_msg(tsx, tsx->last_pkt, tsx->last_pkt_size); + if (status != PJ_SUCCESS) { + if (tsx->timer.id != 0) { + pj_timer_heap_cancel(tsx->endpt->timer_heap, &tsx->timer); + tsx->timer.id = 0; + } + stun_perror(tsx, "STUN error sending message", status); + return status; + } + + tsx->transmit_count++; + + PJ_LOG(4,(tsx->obj_name, "STUN sending message (transmit count=%d)", + tsx->transmit_count)); + return status; +} + + +/* + * Send outgoing message and start STUN transaction. + */ +PJ_DEF(pj_status_t) pj_stun_client_tsx_send_msg(pj_stun_client_tsx *tsx, + pj_bool_t retransmit, + const pj_stun_msg *msg) +{ + pj_status_t status; + + PJ_ASSERT_RETURN(tsx && msg, PJ_EINVAL); + PJ_ASSERT_RETURN(tsx->timer.id != 0, PJ_EBUSY); + + /* Encode message */ + status = pj_stun_msg_encode(msg, tsx->last_pkt, sizeof(tsx->last_pkt), + 0, &tsx->last_pkt_size); + if (status != PJ_SUCCESS) { + stun_perror(tsx, "STUN msg_encode() failed", status); + return status; + } + + /* Update STUN transaction ID */ + tsx->tsx_id[0] = msg->hdr.magic; + pj_memcpy(&tsx->tsx_id[1], msg->hdr.tsx_id, 12); + + /* Update STUN retransmit flag */ + tsx->require_retransmit = retransmit; + + /* Send the message */ + return tsx_transmit_msg(tsx); +} + + +/* Retransmit timer callback */ +static void retransmit_timer_callback(pj_timer_heap_t *timer_heap, + pj_timer_entry *timer) +{ + pj_stun_client_tsx *tsx = (pj_stun_client_tsx *) timer->user_data; + pj_status_t status; + + PJ_UNUSED_ARG(timer_heap); + + if (tsx->transmit_count >= PJ_STUN_MAX_RETRANSMIT_COUNT) { + /* Retransmission count exceeded. Transaction has failed */ + tsx->timer.id = 0; + PJ_LOG(4,(tsx->obj_name, "STUN timeout waiting for response")); + if (tsx->cb.on_complete) { + tsx->cb.on_complete(tsx, PJLIB_UTIL_ESTUNNOTRESPOND, NULL); + } + return; + } + + tsx->timer.id = 0; + status = tsx_transmit_msg(tsx); + if (status != PJ_SUCCESS) { + tsx->timer.id = 0; + if (tsx->cb.on_complete) { + tsx->cb.on_complete(tsx, status, NULL); + } + } +} + + +/* + * Notify the STUN transaction about the arrival of STUN response. + */ +PJ_DEF(pj_status_t) pj_stun_client_tsx_on_rx_msg(pj_stun_client_tsx *tsx, + const void *packet, + pj_size_t pkt_size, + unsigned *parsed_len) +{ + pj_stun_msg *msg; + pj_stun_error_code_attr *err_attr; + pj_status_t status; + + PJ_ASSERT_RETURN(tsx && packet && pkt_size, PJ_EINVAL); + + /* Try to parse the message */ + status = pj_stun_msg_decode(tsx->pool, (const pj_uint8_t*)packet, + pkt_size, 0, &msg, parsed_len, + NULL, NULL, NULL); + if (status != PJ_SUCCESS) { + stun_perror(tsx, "STUN msg_decode() error", status); + return status; + } + + /* Must be STUN response message */ + if (!PJ_STUN_IS_RESPONSE(msg->hdr.type) && + !PJ_STUN_IS_ERROR_RESPONSE(msg->hdr.type)) + { + PJ_LOG(4,(tsx->obj_name, + "STUN rx_msg() error: not response message")); + return PJLIB_UTIL_ESTUNNOTRESPONSE; + } + + /* Compare response's transaction ID */ + if (msg->hdr.magic != tsx->tsx_id[0] || + pj_memcmp(msg->hdr.tsx_id, &tsx->tsx_id[1], 12) != 0) + { + return PJLIB_UTIL_ESTUNINVALIDID; + } + + /* We have a response with matching transaction ID. + * We can cancel retransmit timer now. + */ + if (tsx->timer.id) { + pj_timer_heap_cancel(tsx->endpt->timer_heap, &tsx->timer); + tsx->timer.id = 0; + } + + /* Find STUN error code attribute */ + err_attr = (pj_stun_error_code_attr*) + pj_stun_msg_find_attr(msg, PJ_STUN_ATTR_ERROR_CODE, 0); + + if (err_attr && err_attr->err_class <= 2) { + /* draft-ietf-behave-rfc3489bis-05.txt Section 8.3.2: + * Any response between 100 and 299 MUST result in the cessation + * of request retransmissions, but otherwise is discarded. + */ + PJ_LOG(4,(tsx->obj_name, + "STUN rx_msg() error: received provisional %d code (%.*s)", + err_attr->err_class * 100 + err_attr->number, + (int)err_attr->reason.slen, + err_attr->reason.ptr)); + return PJ_SUCCESS; + } + + if (err_attr == NULL) { + status = PJ_SUCCESS; + } else { + status = PJ_STATUS_FROM_STUN_CODE(err_attr->err_class * 100 + + err_attr->number); + } + + /* Call callback */ + if (tsx->cb.on_complete) { + tsx->cb.on_complete(tsx, status, msg); + } + + return PJ_SUCCESS; +} + diff --git a/pjlib-util/src/pjstun-srv/server.h b/pjlib-util/src/pjstun-srv/server.h new file mode 100644 index 00000000..9de5adce --- /dev/null +++ b/pjlib-util/src/pjstun-srv/server.h @@ -0,0 +1,61 @@ +/* $Id$ */ +/* + * Copyright (C) 2003-2005 Benny Prijono + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + */ +#ifndef __PJSTUN_SERVER_H__ +#define __PJSTUN_SERVER_H__ + + +#define MAX_SERVICE 16 +#define MAX_PKT_LEN 512 + +struct service +{ + unsigned index; + pj_uint16_t port; + pj_bool_t is_stream; + pj_sock_t sock; + pj_ioqueue_key_t *key; + pj_ioqueue_op_key_t recv_opkey, + send_opkey; + + int src_addr_len; + pj_sockaddr_in src_addr; + pj_ssize_t rx_pkt_len; + pj_uint8_t rx_pkt[MAX_PKT_LEN]; + pj_uint8_t tx_pkt[MAX_PKT_LEN]; +}; + +struct stun_server_tag +{ + pj_caching_pool cp; + pj_pool_t *pool; + pj_ioqueue_t *ioqueue; + unsigned service_cnt; + struct service services[MAX_SERVICE]; + + pj_bool_t thread_quit_flag; + unsigned thread_cnt; + pj_thread_t *threads[16]; + +}; + +extern struct stun_server_tag server; + + +#endif /* __PJSTUN_SERVER_H__ */ + diff --git a/pjlib-util/src/pjstun-srv/server_main.c b/pjlib-util/src/pjstun-srv/server_main.c new file mode 100644 index 00000000..56a1bf85 --- /dev/null +++ b/pjlib-util/src/pjstun-srv/server_main.c @@ -0,0 +1,452 @@ +/* $Id$ */ +/* + * Copyright (C) 2003-2005 Benny Prijono + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + */ +#include +#include +#include "server.h" + +#include +#include + + +#define THIS_FILE "server_main.c" +#define MAX_THREADS 8 + +struct stun_server_tag server; + + +pj_status_t server_perror(const char *sender, const char *title, + pj_status_t status) +{ + char errmsg[PJ_ERR_MSG_SIZE]; + pj_strerror(status, errmsg, sizeof(errmsg)); + + PJ_LOG(3,(sender, "%s: %s", title, errmsg)); + + return status; +} + + +static pj_status_t create_response(pj_pool_t *pool, + const pj_stun_msg *req_msg, + unsigned err_code, + unsigned uattr_cnt, + pj_uint16_t uattr_types[], + pj_stun_msg **p_response) +{ + pj_uint32_t msg_type = req_msg->hdr.type; + pj_stun_msg *response; + pj_stun_error_code_attr *err_attr; + pj_status_t status; + + /* Create response or error response */ + if (err_code) + msg_type |= PJ_STUN_ERROR_RESPONSE_BIT; + else + msg_type |= PJ_STUN_RESPONSE_BIT; + + status = pj_stun_msg_create(pool, msg_type, req_msg->hdr.magic, + req_msg->hdr.tsx_id, &response); + if (status != PJ_SUCCESS) { + return status; + } + + /* Add error code attribute */ + if (err_code) { + status = pj_stun_error_code_attr_create(pool, err_code, NULL, + &err_attr); + if (status != PJ_SUCCESS) { + return status; + } + + pj_stun_msg_add_attr(response, &err_attr->hdr); + } + + /* Add unknown_attribute attributes if err_code is 420 */ + if (err_code == PJ_STUN_STATUS_UNKNOWN_ATTRIBUTE) { + pj_stun_unknown_attr *uattr; + + status = pj_stun_unknown_attr_create(pool, uattr_cnt, uattr_types, + &uattr); + if (status != PJ_SUCCESS) + return status; + + pj_stun_msg_add_attr(response, &uattr->hdr); + } + + *p_response = response; + return PJ_SUCCESS; +} + + +static pj_status_t send_msg(struct service *svc, const pj_stun_msg *msg) +{ + unsigned tx_pkt_len; + pj_ssize_t length; + pj_status_t status; + + /* Encode packet */ + tx_pkt_len = sizeof(svc->tx_pkt); + status = pj_stun_msg_encode(msg, svc->tx_pkt, tx_pkt_len, 0, + &tx_pkt_len); + if (status != PJ_SUCCESS) + return status; + + length = tx_pkt_len; + + /* Send packet */ + if (svc->is_stream) { + status = pj_ioqueue_send(svc->key, &svc->send_opkey, svc->tx_pkt, + &length, 0); + } else { + status = pj_ioqueue_sendto(svc->key, &svc->send_opkey, svc->tx_pkt, + &length, 0, &svc->src_addr, + svc->src_addr_len); + } + + PJ_LOG(4,(THIS_FILE, "Sending STUN %s %s", + pj_stun_get_method_name(msg->hdr.type), + pj_stun_get_class_name(msg->hdr.type))); + + return (status == PJ_SUCCESS || status == PJ_EPENDING) ? + PJ_SUCCESS : status; +} + + +static pj_status_t err_respond(struct service *svc, + pj_pool_t *pool, + const pj_stun_msg *req_msg, + unsigned err_code, + unsigned uattr_cnt, + pj_uint16_t uattr_types[]) +{ + pj_stun_msg *response; + pj_status_t status; + + /* Create the error response */ + status = create_response(pool, req_msg, err_code, + uattr_cnt, uattr_types, &response); + if (status != PJ_SUCCESS) { + server_perror(THIS_FILE, "Error creating response", status); + return status; + } + + /* Send response */ + status = send_msg(svc, response); + if (status != PJ_SUCCESS) { + server_perror(THIS_FILE, "Error sending response", status); + return status; + } + + return PJ_SUCCESS; +} + + +static void handle_binding_request(struct service *svc, pj_pool_t *pool, + const pj_stun_msg *rx_msg) +{ + pj_stun_msg *response; + pj_stun_generic_ip_addr_attr *m_attr; + pj_status_t status; + + status = create_response(pool, rx_msg, 0, 0, NULL, &response); + if (status != PJ_SUCCESS) { + server_perror(THIS_FILE, "Error creating response", status); + return; + } + + /* Create MAPPED-ADDRESS attribute */ + status = pj_stun_generic_ip_addr_attr_create(pool, + PJ_STUN_ATTR_MAPPED_ADDR, + PJ_FALSE, + svc->src_addr_len, + &svc->src_addr, &m_attr); + if (status != PJ_SUCCESS) { + server_perror(THIS_FILE, "Error creating response", status); + return; + } + pj_stun_msg_add_attr(response, &m_attr->hdr); + + /* On the presence of magic, create XOR-MAPPED-ADDRESS attribute */ + if (rx_msg->hdr.magic == PJ_STUN_MAGIC) { + status = + pj_stun_generic_ip_addr_attr_create(pool, + PJ_STUN_ATTR_XOR_MAPPED_ADDRESS, + PJ_TRUE, + svc->src_addr_len, + &svc->src_addr, &m_attr); + if (status != PJ_SUCCESS) { + server_perror(THIS_FILE, "Error creating response", status); + return; + } + } + + /* Send */ + status = send_msg(svc, response); + if (status != PJ_SUCCESS) + server_perror(THIS_FILE, "Error sending response", status); +} + + +static void handle_unknown_request(struct service *svc, pj_pool_t *pool, + pj_stun_msg *rx_msg) +{ + err_respond(svc, pool, rx_msg, PJ_STUN_STATUS_BAD_REQUEST, 0, NULL); +} + + +static void on_read_complete(pj_ioqueue_key_t *key, + pj_ioqueue_op_key_t *op_key, + pj_ssize_t bytes_read) +{ + struct service *svc = (struct service *) pj_ioqueue_get_user_data(key); + pj_pool_t *pool = NULL; + pj_stun_msg *rx_msg; + unsigned err_code; + unsigned uattr_cnt; + pj_uint16_t uattr_types[16]; + pj_status_t status; + + if (bytes_read <= 0) + goto next_read; + + pool = pj_pool_create(&server.cp.factory, "service", 4000, 4000, NULL); + + err_code = 0; + uattr_cnt = PJ_ARRAY_SIZE(uattr_types); + rx_msg = NULL; + status = pj_stun_msg_decode(pool, svc->rx_pkt, bytes_read, 0, &rx_msg, + NULL, &err_code, &uattr_cnt, uattr_types); + if (status != PJ_SUCCESS) { + server_perror(THIS_FILE, "STUN msg_decode() error", status); + if (err_code != 0 && rx_msg && PJ_STUN_IS_REQUEST(rx_msg->hdr.type)) { + err_respond(svc, pool, rx_msg, err_code, + uattr_cnt, uattr_types); + } + goto next_read; + } + + PJ_LOG(4,(THIS_FILE, "RX STUN %s %s message", + pj_stun_get_method_name(rx_msg->hdr.type), + pj_stun_get_class_name(rx_msg->hdr.type))); + + if (PJ_STUN_IS_REQUEST(rx_msg->hdr.type)) { + switch (rx_msg->hdr.type) { + case PJ_STUN_BINDING_REQUEST: + handle_binding_request(svc, pool, rx_msg); + break; + default: + handle_unknown_request(svc, pool, rx_msg); + } + + } + +next_read: + if (pool != NULL) + pj_pool_release(pool); + + if (bytes_read < 0) { + server_perror(THIS_FILE, "on_read_complete()", -bytes_read); + } + + svc->rx_pkt_len = sizeof(svc->rx_pkt); + svc->src_addr_len = sizeof(svc->src_addr); + + status = pj_ioqueue_recvfrom(svc->key, &svc->recv_opkey, + svc->rx_pkt, &svc->rx_pkt_len, + PJ_IOQUEUE_ALWAYS_ASYNC, + &svc->src_addr, &svc->src_addr_len); + if (status != PJ_EPENDING) + server_perror(THIS_FILE, "error starting async read", status); +} + + +static pj_status_t init_service(struct service *svc) +{ + pj_status_t status; + pj_ioqueue_callback service_callback; + pj_sockaddr_in addr; + + status = pj_sock_socket(PJ_AF_INET, PJ_SOCK_DGRAM, 0, &svc->sock); + if (status != PJ_SUCCESS) + return status; + + status = pj_sockaddr_in_init(&addr, NULL, svc->port); + if (status != PJ_SUCCESS) + goto on_error; + + status = pj_sock_bind(svc->sock, &addr, sizeof(addr)); + if (status != PJ_SUCCESS) + goto on_error; + + pj_bzero(&service_callback, sizeof(service_callback)); + service_callback.on_read_complete = &on_read_complete; + + status = pj_ioqueue_register_sock(server.pool, server.ioqueue, svc->sock, + svc, &service_callback, &svc->key); + if (status != PJ_SUCCESS) + goto on_error; + + + pj_ioqueue_op_key_init(&svc->recv_opkey, sizeof(svc->recv_opkey)); + pj_ioqueue_op_key_init(&svc->send_opkey, sizeof(svc->send_opkey)); + + on_read_complete(svc->key, &svc->recv_opkey, 0); + + PJ_LOG(4,(THIS_FILE, "Service started on port %d", svc->port)); + return PJ_SUCCESS; + +on_error: + if (svc->key != NULL) { + pj_ioqueue_unregister(svc->key); + svc->key = NULL; + svc->sock = PJ_INVALID_SOCKET; + } else if (svc->sock != 0 && svc->sock != PJ_INVALID_SOCKET) { + pj_sock_close(svc->sock); + svc->sock = PJ_INVALID_SOCKET; + } + + return status; +} + + +static int worker_thread(void *p) +{ + PJ_UNUSED_ARG(p); + + while (!server.thread_quit_flag) { + pj_time_val timeout = { 0, 50 }; + pj_ioqueue_poll(server.ioqueue, &timeout); + } + + return 0; +} + + +pj_status_t server_init(void) +{ + pj_status_t status; + + status = pj_init(); + if (status != PJ_SUCCESS) + return server_perror(THIS_FILE, "pj_init() error", status); + + status = pjlib_util_init(); + if (status != PJ_SUCCESS) + return server_perror(THIS_FILE, "pjlib_util_init() error", status); + + pj_caching_pool_init(&server.cp, + &pj_pool_factory_default_policy, 0); + + + server.pool = pj_pool_create(&server.cp.factory, "server", 4000, 4000, + NULL); + + status = pj_ioqueue_create(server.pool, PJ_IOQUEUE_MAX_HANDLES, + &server.ioqueue); + if (status != PJ_SUCCESS) + return server_perror(THIS_FILE, "pj_ioqueue_create()", status); + + server.service_cnt = 1; + server.services[0].index = 0; + server.services[0].port = PJ_STUN_PORT; + + status = init_service(&server.services[0]); + if (status != PJ_SUCCESS) + return server_perror(THIS_FILE, "init_service() error", status); + + return PJ_SUCCESS; +} + + +pj_status_t server_main(void) +{ +#if 1 + for (;;) { + pj_time_val timeout = { 0, 50 }; + pj_ioqueue_poll(server.ioqueue, &timeout); + + if (kbhit() && _getch()==27) + break; + } +#else + pj_status_t status; + char line[10]; + + status = pj_thread_create(server.pool, "stun_server", &worker_thread, NULL, + 0, 0, &server.threads[0]); + if (status != PJ_SUCCESS) + return server_perror(THIS_FILE, "create_thread() error", status); + + puts("Press ENTER to quit"); + fgets(line, sizeof(line), stdin); + +#endif + + return PJ_SUCCESS; +} + + +pj_status_t server_destroy(void) +{ + unsigned i; + + for (i=0; ikey != NULL) { + pj_ioqueue_unregister(svc->key); + svc->key = NULL; + svc->sock = PJ_INVALID_SOCKET; + } else if (svc->sock != 0 && svc->sock != PJ_INVALID_SOCKET) { + pj_sock_close(svc->sock); + svc->sock = PJ_INVALID_SOCKET; + } + } + + server.thread_quit_flag = PJ_TRUE; + for (i=0; i