/*
 * Copyright (C) 2011, 2012, 2013 Citrix Systems
 *
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. Neither the name of the project nor the names of its contributors
 *    may be used to endorse or promote products derived from this software
 *    without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE PROJECT AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE PROJECT OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

#include "stun_buffer.h"

////////////////////// BUFFERS ///////////////////////////

int stun_init_buffer(stun_buffer *buf) {
  if(!buf) return -1;
  ns_bzero(buf->buf,sizeof(buf->buf));
  buf->len=0;
  buf->offset=0;
  buf->coffset=0;
  return 0;
}

int stun_get_size(const stun_buffer *buf) {
  if(!buf) return 0;
  return sizeof(buf->buf);
}

////////////////////////////////////////////////////////////

void stun_tid_from_message(const stun_buffer *buf, stun_tid* id) {
  stun_tid_from_message_str(buf->buf,(size_t)(buf->len), id);
}

void stun_tid_generate_in_message(stun_buffer* buf, stun_tid* id) {
  if(buf) {
    stun_tid_generate_in_message_str(buf->buf, id);
  }
}

////////////////////////////////////////////////////////

static inline int is_channel_msg(const stun_buffer* buf) {
  if(buf && buf->len>0) {
    return is_channel_msg_str(buf->buf, (size_t)(buf->len));
  }
  return 0;
}

int stun_is_command_message(const stun_buffer* buf) {
  if(!buf || buf->len<=0)
    return 0;
  else
    return stun_is_command_message_str(buf->buf,(size_t)(buf->len));
}

int stun_is_request(const stun_buffer* buf) {
  return stun_is_request_str(buf->buf,(size_t)buf->len);
}

int stun_is_success_response(const stun_buffer* buf) {
  return stun_is_success_response_str(buf->buf, (size_t)(buf->len));
}

int stun_is_error_response(const stun_buffer* buf, int *err_code, u08bits *err_msg, size_t err_msg_size) {
  return stun_is_error_response_str(buf->buf, (size_t)(buf->len), err_code, err_msg, err_msg_size);
}

int stun_is_response(const stun_buffer* buf) {
  return stun_is_response_str(buf->buf,(size_t)(buf->len));
}

int stun_is_indication(const stun_buffer* buf) {
  if(is_channel_msg(buf)) return 0;
  return IS_STUN_INDICATION(stun_get_msg_type(buf));
}

u16bits stun_get_method(const stun_buffer* buf) {
	return stun_get_method_str(buf->buf, (size_t)(buf->len));
}

u16bits stun_get_msg_type(const stun_buffer* buf) {
  if(!buf) return (u16bits)-1;
  return stun_get_msg_type_str(buf->buf,(size_t)buf->len);
}

////////////////////////////////////////////////////////////

static void stun_init_command(u16bits message_type, stun_buffer* buf) {
  buf->len=stun_get_size(buf);
  stun_init_command_str(message_type, buf->buf, (size_t*)(&(buf->len)));
}

void stun_init_request(u16bits method, stun_buffer* buf) {
  stun_init_command(stun_make_request(method), buf);
}

void stun_init_indication(u16bits method, stun_buffer* buf) {
  stun_init_command(stun_make_indication(method), buf);
}

void stun_init_success_response(u16bits method, stun_buffer* buf, stun_tid* id) {
  buf->len=stun_get_size(buf);
  stun_init_success_response_str(method, buf->buf, (size_t*)(&(buf->len)), id);
}

void stun_init_error_response(u16bits method, stun_buffer* buf, u16bits error_code, const u08bits *reason, stun_tid* id) {
  buf->len=stun_get_size(buf);
  stun_init_error_response_str(method, buf->buf, (size_t*)(&(buf->len)), error_code, reason, id);
}
 
///////////////////////////////////////////////////////////////////////////////

int stun_get_command_message_len(const stun_buffer* buf) {
  return stun_get_command_message_len_str(buf->buf, (size_t)(buf->len));
}

///////////////////////////////////////////////////////////////////////////////

int stun_init_channel_message(u16bits chnumber, stun_buffer* buf, int length, int do_padding) {
  return stun_init_channel_message_str(chnumber, buf->buf, (size_t*)(&(buf->len)), length, do_padding);
}

int stun_is_channel_message(stun_buffer* buf, u16bits* chnumber, int is_padding_mandatory) {
  if(!buf) return 0;
  size_t blen = (size_t)buf->len;
  int ret = stun_is_channel_message_str(buf->buf, &blen, chnumber, is_padding_mandatory);
  if(ret) {
	  buf->len=(ssize_t)blen;
  }
  return ret;
}

///////////////////////////////////////////////////////////////////////////////

int stun_set_allocate_request(stun_buffer* buf, u32bits lifetime, int af4, int af6, u08bits transport, int mobile, const char *rt, int ep) {
  return stun_set_allocate_request_str(buf->buf, (size_t*)(&(buf->len)), lifetime, af4, af6, transport, mobile, rt, ep);
}

int stun_set_allocate_response(stun_buffer* buf, stun_tid* tid, 
			       const ioa_addr *relayed_addr1, const ioa_addr *relayed_addr2,
			       const ioa_addr *reflexive_addr,
			       u32bits lifetime, u32bits max_lifetime, int error_code, const u08bits *reason,
			       u64bits reservation_token, char *mobile_id) {

  return stun_set_allocate_response_str(buf->buf, (size_t*)(&(buf->len)), tid, 
					relayed_addr1, relayed_addr2, reflexive_addr,
					lifetime, max_lifetime, error_code, reason,
					reservation_token, mobile_id);

}

///////////////////////////////////////////////////////////////////////////////

u16bits stun_set_channel_bind_request(stun_buffer* buf, 
				       const ioa_addr* peer_addr, u16bits channel_number) {

  return stun_set_channel_bind_request_str(buf->buf,(size_t*)(&(buf->len)), peer_addr, channel_number);
}

void stun_set_channel_bind_response(stun_buffer* buf, stun_tid* tid, int error_code, const u08bits *reason) {
  stun_set_channel_bind_response_str(buf->buf, (size_t*)(&(buf->len)), tid, error_code, reason);
}

////////////////////////////////////////////////////////////////

stun_attr_ref stun_attr_get_first(const stun_buffer* buf) {
  return stun_attr_get_first_str(buf->buf, (size_t)(buf->len));
}

stun_attr_ref stun_attr_get_next(const stun_buffer* buf, stun_attr_ref prev) {
  return stun_attr_get_next_str(buf->buf, (size_t)(buf->len), prev);
}

int stun_attr_add(stun_buffer* buf, u16bits attr, const s08bits* avalue, int alen) {
  return stun_attr_add_str(buf->buf, (size_t*)(&(buf->len)), attr, (const u08bits *)avalue, alen);
}

int stun_attr_add_channel_number(stun_buffer* buf, u16bits chnumber) {
  return stun_attr_add_channel_number_str(buf->buf, (size_t *)(&(buf->len)), chnumber);
}

int stun_attr_add_addr(stun_buffer *buf,u16bits attr_type, const ioa_addr* ca) {
  return stun_attr_add_addr_str(buf->buf,(size_t*)(&(buf->len)), attr_type, ca);
}

int stun_attr_get_addr(const stun_buffer *buf, stun_attr_ref attr, ioa_addr* ca, 
		       const ioa_addr *default_addr) {

  return stun_attr_get_addr_str(buf->buf, (size_t)(buf->len), attr, ca, default_addr);
}

int stun_attr_get_first_addr(const stun_buffer *buf, u16bits attr_type, ioa_addr* ca, 
			     const ioa_addr *default_addr) {

  return stun_attr_get_first_addr_str(buf->buf, (size_t)(buf->len), attr_type, ca, default_addr);
}

int stun_attr_add_even_port(stun_buffer* buf, uint8_t value) {
  if(value) value=0x80;
  return stun_attr_add(buf,STUN_ATTRIBUTE_EVEN_PORT,(const s08bits*)&value,1);
}

u16bits stun_attr_get_first_channel_number(const stun_buffer *buf) {
  return stun_attr_get_first_channel_number_str(buf->buf, (size_t)(buf->len));
}

stun_attr_ref stun_attr_get_first_by_type(const stun_buffer* buf, u16bits attr_type) {
  return stun_attr_get_first_by_type_str(buf->buf, (size_t)(buf->len), attr_type);
}

///////////////////////////////////////////////////////////////////////////////

void stun_set_binding_request(stun_buffer* buf) {
  stun_set_binding_request_str(buf->buf, (size_t*)(&(buf->len)));
}

int stun_set_binding_response(stun_buffer* buf, stun_tid* tid, 
			      const ioa_addr *reflexive_addr, int error_code, const u08bits *reason) {
  return stun_set_binding_response_str(buf->buf, (size_t*)(&(buf->len)), tid, 
				       reflexive_addr, error_code, reason,
				       0,0);
}

void stun_prepare_binding_request(stun_buffer* buf) {
  stun_set_binding_request_str(buf->buf, (size_t*)(&(buf->len)));
}

int stun_is_binding_response(const stun_buffer* buf) {
  return stun_is_binding_response_str(buf->buf, (size_t)(buf->len));
}

///////////////////////////////////////////////////////