diff options
Diffstat (limited to 'pjnath/src/pjturn-srv/listener_udp.c')
-rw-r--r-- | pjnath/src/pjturn-srv/listener_udp.c | 219 |
1 files changed, 219 insertions, 0 deletions
diff --git a/pjnath/src/pjturn-srv/listener_udp.c b/pjnath/src/pjturn-srv/listener_udp.c new file mode 100644 index 00000000..d8f90ca4 --- /dev/null +++ b/pjnath/src/pjturn-srv/listener_udp.c @@ -0,0 +1,219 @@ +/* $Id$ */ +/* + * Copyright (C) 2003-2007 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 "turn.h" + +struct read_op +{ + pj_ioqueue_op_key_t op_key; + pjturn_pkt pkt; +}; + +struct udp_listener +{ + pjturn_listener base; + pj_ioqueue_key_t *key; + unsigned read_cnt; + struct read_op **read_op; /* Array of read_op's */ +}; + + +static pj_status_t udp_sendto(pjturn_listener *listener, + const void *packet, + pj_size_t size, + unsigned flag, + const pj_sockaddr_t *addr, + int addr_len); +static pj_status_t udp_destroy(pjturn_listener *udp); +static void on_read_complete(pj_ioqueue_key_t *key, + pj_ioqueue_op_key_t *op_key, + pj_ssize_t bytes_read); + + +/* + * Create a new listener on the specified port. + */ +PJ_DEF(pj_status_t) pjturn_listener_create_udp( pjturn_srv *srv, + int af, + const pj_str_t *bound_addr, + unsigned port, + unsigned concurrency_cnt, + unsigned flags, + pjturn_listener **p_listener) +{ + pj_pool_t *pool; + struct udp_listener *udp; + pj_ioqueue_callback ioqueue_cb; + unsigned i; + pj_status_t status; + + /* Create structure */ + pool = pj_pool_create(srv->core.pf, "udplis%p", 1000, 1000, NULL); + udp = PJ_POOL_ZALLOC_T(pool, struct udp_listener); + udp->base.pool = pool; + udp->base.server = srv; + udp->base.tp_type = PJTURN_TP_UDP; + udp->base.sock = PJ_INVALID_SOCKET; + udp->base.sendto = &udp_sendto; + udp->base.destroy = &udp_destroy; + udp->read_cnt = concurrency_cnt; + udp->base.flags = flags; + + /* Create socket */ + status = pj_sock_socket(af, pj_SOCK_DGRAM(), 0, &udp->base.sock); + if (status != PJ_SUCCESS) + goto on_error; + + /* Init bind address */ + status = pj_sockaddr_init(af, &udp->base.addr, bound_addr, + (pj_uint16_t)port); + if (status != PJ_SUCCESS) + goto on_error; + + /* Bind socket */ + status = pj_sock_bind(udp->base.sock, &udp->base.addr, + pj_sockaddr_get_len(&udp->base.addr)); + if (status != PJ_SUCCESS) + goto on_error; + + /* Register to ioqueue */ + pj_bzero(&ioqueue_cb, sizeof(ioqueue_cb)); + ioqueue_cb.on_read_complete = on_read_complete; + status = pj_ioqueue_register_sock(pool, srv->core.ioqueue, udp->base.sock, + udp, &ioqueue_cb, &udp->key); + + /* Create op keys */ + udp->read_op = (struct read_op**)pj_pool_calloc(pool, concurrency_cnt, + sizeof(struct read_op*)); + + /* Create each read_op and kick off read operation */ + for (i=0; i<concurrency_cnt; ++i) { + pj_pool_t *rpool = pj_pool_create(srv->core.pf, "rop%p", + 1000, 1000, NULL); + + udp->read_op[i] = PJ_POOL_ZALLOC_T(rpool, struct read_op); + udp->read_op[i]->pkt.pool = rpool; + + on_read_complete(udp->key, &udp->read_op[i]->op_key, 0); + } + + /* Done */ + *p_listener = &udp->base; + return PJ_SUCCESS; + + +on_error: + udp_destroy(&udp->base); + return status; +} + + +/* + * Destroy listener. + */ +static pj_status_t udp_destroy(pjturn_listener *listener) +{ + struct udp_listener *udp = (struct udp_listener *)listener; + unsigned i; + + if (udp->key) { + pj_ioqueue_unregister(udp->key); + udp->key = NULL; + udp->base.sock = PJ_INVALID_SOCKET; + } else if (udp->base.sock != PJ_INVALID_SOCKET) { + pj_sock_close(udp->base.sock); + udp->base.sock = PJ_INVALID_SOCKET; + } + + for (i=0; i<udp->read_cnt; ++i) { + if (udp->read_op[i]->pkt.pool) { + pj_pool_t *rpool = udp->read_op[i]->pkt.pool; + udp->read_op[i]->pkt.pool = NULL; + pj_pool_release(rpool); + } + } + + if (udp->base.pool) { + pj_pool_release(udp->base.pool); + udp->base.pool = NULL; + } + return PJ_SUCCESS; +} + +/* + * Callback to send packet. + */ +static pj_status_t udp_sendto(pjturn_listener *listener, + const void *packet, + pj_size_t size, + unsigned flag, + const pj_sockaddr_t *addr, + int addr_len) +{ + pj_ssize_t len = size; + return pj_sock_sendto(listener->sock, packet, &len, flag, addr, + pj_sockaddr_get_len(addr)); +} + +/* + * Callback on received packet. + */ +static void on_read_complete(pj_ioqueue_key_t *key, + pj_ioqueue_op_key_t *op_key, + pj_ssize_t bytes_read) +{ + struct udp_listener *udp; + struct read_op *read_op = (struct read_op*) op_key; + pj_status_t status; + + udp = (struct udp_listener*) pj_ioqueue_get_user_data(key); + + do { + pj_pool_t *rpool; + + /* Report to server */ + if (bytes_read > 0) { + read_op->pkt.len = bytes_read; + pj_gettimeofday(&read_op->pkt.rx_time); + + pjturn_srv_on_rx_pkt(udp->base.server, &read_op->pkt); + } + + /* Reset pool */ + rpool = read_op->pkt.pool; + pj_pool_reset(rpool); + read_op->pkt.pool = rpool; + read_op->pkt.listener = &udp->base; + read_op->pkt.src.tp_type = udp->base.tp_type; + + /* Read next packet */ + bytes_read = sizeof(read_op->pkt.pkt); + read_op->pkt.src_addr_len = sizeof(read_op->pkt.src.clt_addr); + pj_bzero(&read_op->pkt.src.clt_addr, sizeof(read_op->pkt.src.clt_addr)); + + status = pj_ioqueue_recvfrom(udp->key, op_key, + read_op->pkt.pkt, &bytes_read, 0, + &read_op->pkt.src.clt_addr, + &read_op->pkt.src_addr_len); + + if (status != PJ_EPENDING && status != PJ_SUCCESS) + bytes_read = -status; + + } while (status != PJ_EPENDING && status != PJ_ECANCELLED); +} + |