/* $Id: stun.c 974 2007-02-19 01:13:53Z bennylp $ *//*  * 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 <pjlib-util/stun.h>#include <pjlib-util/errno.h>#include <pj/pool.h>#include <pj/log.h>#include <pj/sock.h>#include <pj/os.h>#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; i<msg->attr_count; ++i) {	pj_stun_attr_hdr *attr = msg->attr[i];	if (pj_ntohs(attr->type) == t)	    return attr;    }    return 0;}