summaryrefslogtreecommitdiff
path: root/pjlib-util/src
diff options
context:
space:
mode:
authorBenny Prijono <bennylp@teluu.com>2007-02-21 00:40:05 +0000
committerBenny Prijono <bennylp@teluu.com>2007-02-21 00:40:05 +0000
commit1a5b8c19989c4ab4e1eee7f712913f8f3e459e71 (patch)
tree94d7851e91849c6b60420150c8944db9bcf4b298 /pjlib-util/src
parentb69a9ab56a1df4a41bfa9132d5aa5263fa80b38b (diff)
Merged the ICE branch into the trunk
git-svn-id: http://svn.pjsip.org/repos/pjproject/trunk@992 74dad513-b988-da41-8d7b-12977e46ad98
Diffstat (limited to 'pjlib-util/src')
-rw-r--r--pjlib-util/src/pjlib-util/stun_endpoint.c68
-rw-r--r--pjlib-util/src/pjlib-util/stun_msg.c1562
-rw-r--r--pjlib-util/src/pjlib-util/stun_server.c129
-rw-r--r--pjlib-util/src/pjlib-util/stun_simple.c (renamed from pjlib-util/src/pjlib-util/stun.c)50
-rw-r--r--pjlib-util/src/pjlib-util/stun_simple_client.c (renamed from pjlib-util/src/pjlib-util/stun_client.c)22
-rw-r--r--pjlib-util/src/pjlib-util/stun_transaction.c333
-rw-r--r--pjlib-util/src/pjstun-srv/server.h61
-rw-r--r--pjlib-util/src/pjstun-srv/server_main.c452
8 files changed, 2641 insertions, 36 deletions
diff --git a/pjlib-util/src/pjlib-util/stun_endpoint.c b/pjlib-util/src/pjlib-util/stun_endpoint.c
new file mode 100644
index 00000000..6a1de943
--- /dev/null
+++ b/pjlib-util/src/pjlib-util/stun_endpoint.c
@@ -0,0 +1,68 @@
+/* $Id$ */
+/*
+ * Copyright (C) 2003-2005 Benny Prijono <benny@prijono.org>
+ *
+ * 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 <pjlib-util/stun_endpoint.h>
+#include <pjlib-util/errno.h>
+#include <pj/assert.h>
+#include <pj/pool.h>
+
+
+/*
+ * 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 <benny@prijono.org>
+ *
+ * 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 <pjlib-util/stun_msg.h>
+#include <pjlib-util/errno.h>
+#include <pj/assert.h>
+#include <pj/log.h>
+#include <pj/os.h>
+#include <pj/pool.h>
+#include <pj/rand.h>
+#include <pj/string.h>
+
+#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; i<PJ_ARRAY_SIZE(stun_err_msg_map); ++i) {
+ if (stun_err_msg_map[i].err_code == err_code)
+ return pj_str((char*)stun_err_msg_map[i].err_msg);
+ }
+ return pj_str(NULL);
+}
+
+
+
+#define INIT_ATTR(a,t,l) (a)->hdr.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; i<attr_cnt; ++i) {
+ attr->attrs[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; i<attr->attr_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; i<msg->attr_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 <benny@prijono.org>
+ *
+ * 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 <pjlib-util/stun_server.h>
+#include <pjlib-util/errno.h>
+#include <pj/assert.h>
+#include <pj/pool.h>
+#include <pj/string.h>
+
+
+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; i<svc->cb_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.c b/pjlib-util/src/pjlib-util/stun_simple.c
index c06479ac..6ab6dbbe 100644
--- a/pjlib-util/src/pjlib-util/stun.c
+++ b/pjlib-util/src/pjlib-util/stun_simple.c
@@ -1,6 +1,6 @@
/* $Id$ */
/*
- * Copyright (C) 2003-2007 Benny Prijono <benny@prijono.org>
+ * Copyright (C) 2003-2006 Benny Prijono <benny@prijono.org>
*
* 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
@@ -16,7 +16,7 @@
* along with this program; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*/
-#include <pjlib-util/stun.h>
+#include <pjlib-util/stun_simple.h>
#include <pjlib-util/errno.h>
#include <pj/pool.h>
#include <pj/log.h>
@@ -25,47 +25,47 @@
#define THIS_FILE "stun.c"
-PJ_DEF(pj_status_t) pj_stun_create_bind_req( pj_pool_t *pool,
+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)
{
- pj_stun_msg_hdr *hdr;
+ pjstun_msg_hdr *hdr;
PJ_CHECK_STACK();
- hdr = pj_pool_calloc(pool, 1, sizeof(pj_stun_msg_hdr));
+ hdr = pj_pool_calloc(pool, 1, sizeof(pjstun_msg_hdr));
if (!hdr)
return PJ_ENOMEM;
- hdr->type = pj_htons(PJ_STUN_BINDING_REQUEST);
+ 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(pj_stun_msg_hdr);
+ *len = sizeof(pjstun_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_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 = (pj_stun_msg_hdr*)buf;
+ msg->hdr = (pjstun_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:
+ 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));
@@ -73,21 +73,21 @@ PJ_DEF(pj_status_t) pj_stun_parse_msg( void *buf, pj_size_t len,
}
msg_len = pj_ntohs(msg->hdr->length);
- if (msg_len != len - sizeof(pj_stun_msg_hdr)) {
+ if (msg_len != len - sizeof(pjstun_msg_hdr)) {
PJ_LOG(4,(THIS_FILE, "Error: invalid msg_len %d (expecting %d)",
- msg_len, len - sizeof(pj_stun_msg_hdr)));
+ msg_len, len - sizeof(pjstun_msg_hdr)));
return PJLIB_UTIL_ESTUNINMSGLEN;
}
msg->attr_count = 0;
- p_attr = (char*)buf + sizeof(pj_stun_msg_hdr);
+ p_attr = (char*)buf + sizeof(pjstun_msg_hdr);
while (msg_len > 0) {
- pj_stun_attr_hdr **attr = &msg->attr[msg->attr_count];
+ pjstun_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);
+ *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",
@@ -95,7 +95,7 @@ PJ_DEF(pj_status_t) pj_stun_parse_msg( void *buf, pj_size_t len,
return PJLIB_UTIL_ESTUNINATTRLEN;
}
- if (pj_ntohs((*attr)->type) > PJ_STUN_ATTR_REFLECTED_FORM) {
+ 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));
@@ -109,14 +109,14 @@ PJ_DEF(pj_status_t) pj_stun_parse_msg( void *buf, pj_size_t len,
return PJ_SUCCESS;
}
-PJ_DEF(void*) pj_stun_msg_find_attr( pj_stun_msg *msg, pj_stun_attr_type t)
+PJ_DEF(void*) pjstun_msg_find_attr( pjstun_msg *msg, pjstun_attr_type t)
{
int i;
PJ_CHECK_STACK();
for (i=0; i<msg->attr_count; ++i) {
- pj_stun_attr_hdr *attr = msg->attr[i];
+ pjstun_attr_hdr *attr = msg->attr[i];
if (pj_ntohs(attr->type) == t)
return attr;
}
diff --git a/pjlib-util/src/pjlib-util/stun_client.c b/pjlib-util/src/pjlib-util/stun_simple_client.c
index 6005694c..7d6e472b 100644
--- a/pjlib-util/src/pjlib-util/stun_client.c
+++ b/pjlib-util/src/pjlib-util/stun_simple_client.c
@@ -1,6 +1,6 @@
/* $Id$ */
/*
- * Copyright (C) 2003-2007 Benny Prijono <benny@prijono.org>
+ * Copyright (C) 2003-2006 Benny Prijono <benny@prijono.org>
*
* 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
@@ -16,7 +16,7 @@
* along with this program; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*/
-#include <pjlib-util/stun.h>
+#include <pjlib-util/stun_simple.h>
#include <pjlib-util/errno.h>
#include <pj/os.h>
#include <pj/pool.h>
@@ -32,7 +32,7 @@ static int stun_timer[] = {1600, 1600, 1600 };
#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,
+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,
@@ -69,7 +69,7 @@ PJ_DECL(pj_status_t) pj_stun_get_mapped_addr( pj_pool_factory *pf,
/* Create the outgoing BIND REQUEST message template */
- status = pj_stun_create_bind_req( pool, &out_msg, &out_msg_len,
+ status = pjstun_create_bind_req( pool, &out_msg, &out_msg_len,
pj_rand(), pj_rand());
if (status != PJ_SUCCESS)
goto on_error;
@@ -97,7 +97,7 @@ PJ_DECL(pj_status_t) pj_stun_get_mapped_addr( pj_pool_factory *pf,
/* Send messages to servers that has not given us response. */
for (i=0; i<sock_cnt && status==PJ_SUCCESS; ++i) {
for (j=0; j<2 && status==PJ_SUCCESS; ++j) {
- pj_stun_msg_hdr *msg_hdr = out_msg;
+ pjstun_msg_hdr *msg_hdr = out_msg;
pj_ssize_t sent_len;
if (rec[i].srv[j].mapped_port != 0)
@@ -150,10 +150,10 @@ PJ_DECL(pj_status_t) pj_stun_get_mapped_addr( pj_pool_factory *pf,
for (i=0; i<sock_cnt; ++i) {
int sock_idx, srv_idx;
pj_ssize_t len;
- pj_stun_msg msg;
+ pjstun_msg msg;
pj_sockaddr_in addr;
int addrlen = sizeof(addr);
- pj_stun_mapped_addr_attr *attr;
+ pjstun_mapped_addr_attr *attr;
char recv_buf[128];
if (!PJ_FD_ISSET(sock[i], &r))
@@ -170,7 +170,7 @@ PJ_DECL(pj_status_t) pj_stun_get_mapped_addr( pj_pool_factory *pf,
if (status != PJ_SUCCESS)
continue;
- status = pj_stun_parse_msg(recv_buf, len, &msg);
+ status = pjstun_parse_msg(recv_buf, len, &msg);
if (status != PJ_SUCCESS) {
continue;
}
@@ -184,17 +184,17 @@ PJ_DECL(pj_status_t) pj_stun_get_mapped_addr( pj_pool_factory *pf,
continue;
}
- if (pj_ntohs(msg.hdr->type) != PJ_STUN_BINDING_RESPONSE) {
+ if (pj_ntohs(msg.hdr->type) != PJSTUN_BINDING_RESPONSE) {
status = PJLIB_UTIL_ESTUNNOBINDRES;
continue;
}
- if (pj_stun_msg_find_attr(&msg, PJ_STUN_ATTR_ERROR_CODE) != NULL) {
+ if (pjstun_msg_find_attr(&msg, PJSTUN_ATTR_ERROR_CODE) != NULL) {
status = PJLIB_UTIL_ESTUNRECVERRATTR;
continue;
}
- attr = (void*)pj_stun_msg_find_attr(&msg, PJ_STUN_ATTR_MAPPED_ADDR);
+ attr = (void*)pjstun_msg_find_attr(&msg, PJSTUN_ATTR_MAPPED_ADDR);
if (!attr) {
status = PJLIB_UTIL_ESTUNNOMAP;
continue;
diff --git a/pjlib-util/src/pjlib-util/stun_transaction.c b/pjlib-util/src/pjlib-util/stun_transaction.c
new file mode 100644
index 00000000..6720b8d7
--- /dev/null
+++ b/pjlib-util/src/pjlib-util/stun_transaction.c
@@ -0,0 +1,333 @@
+/* $Id$ */
+/*
+ * Copyright (C) 2003-2005 Benny Prijono <benny@prijono.org>
+ *
+ * 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 <pjlib-util/stun_transaction.h>
+#include <pjlib-util/errno.h>
+#include <pj/assert.h>
+#include <pj/log.h>
+#include <pj/pool.h>
+#include <pj/string.h>
+#include <pj/timer.h>
+
+
+#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 <benny@prijono.org>
+ *
+ * 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 <benny@prijono.org>
+ *
+ * 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 <pjlib-util.h>
+#include <pjlib.h>
+#include "server.h"
+
+#include <stdio.h>
+#include <conio.h>
+
+
+#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; i<PJ_ARRAY_SIZE(server.services); ++i) {
+ struct service *svc = &server.services[i];
+
+ 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;
+ }
+ }
+
+ server.thread_quit_flag = PJ_TRUE;
+ for (i=0; i<PJ_ARRAY_SIZE(server.threads); ++i) {
+ if (server.threads[i]) {
+ pj_thread_join(server.threads[i]);
+ pj_thread_destroy(server.threads[i]);
+ server.threads[i] = NULL;
+ }
+ }
+
+ pj_ioqueue_destroy(server.ioqueue);
+ pj_pool_release(server.pool);
+ pj_caching_pool_destroy(&server.cp);
+
+ pj_shutdown();
+
+ return PJ_SUCCESS;
+}
+
+
+int main()
+{
+ if (server_init()) {
+ server_destroy();
+ return 1;
+ }
+
+ server_main();
+ server_destroy();
+
+ return 0;
+}