/*
    This file is part of libcapwap.

    libcapwap 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 3 of the License, or
    (at your option) any later version.

    libcapwap 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 Foobar.  If not, see <http://www.gnu.org/licenses/>.

*/

/**
 * @file 
 * @brief Implementation output handler of capwap cw_out_local_ip_address
 */

#include <errno.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>

#include <string.h>
#include "capwap.h"
#include "capwap_items.h"

#include "dbg.h"
#include "log.h"
#include "capwap.h"
#include "conn.h"


/**
 * Output handler for Capwap Local IP Address message element.
 *
 * @param conn Connection object
 * @param action Pointer to action which called this handler
 * @param dst Destination buffer
 *
 * This handler determines the IP address from #conn->sock.
 * It can deal both with IPv4 and IPv6 sockets.
 *
 */
int cw_out_capwap_local_ip_address(struct conn *conn, struct cw_action_out *action,
				   uint8_t * dst)
{
	struct sockaddr_storage a;
	socklen_t alen = sizeof(struct sockaddr_storage);
	int rc = getsockname(conn->sock, (struct sockaddr *) &a, &alen);

	if (rc != 0){
		cw_log(LOG_ERR,"Can't determine sock IP address - %s\n",strerror(errno));
		return 0;
	}


	int id; 

	switch (((struct sockaddr *) &a)->sa_family) {
		case AF_INET:
		{
			struct sockaddr_in *sain = (struct sockaddr_in *) &a;

			id  = CW_ELEM_CAPWAP_LOCAL_IPV4_ADDRESS;

/*			if (conn->capwap_mode == CW_MODE_CISCO)
				id = CW_ELEM_WTP_IPV4_IP_ADDRESS;
			else
				id = CW_ELEM_CAPWAP_LOCAL_IPV4_ADDRESS;
*/
			cw_put_data(dst + 4, (uint8_t *) & sain->sin_addr, 4);
			return 4 + cw_put_elem_hdr(dst, id, 4);
		}

		case AF_INET6:
		{
			id  = CW_ELEM_CAPWAP_LOCAL_IPV6_ADDRESS;

/*			if (conn->capwap_mode == CW_MODE_CISCO)
				id = CW_ELEM_WTP_IPV6_IP_ADDRESS;
			else
				id = CW_ELEM_CAPWAP_LOCAL_IPV6_ADDRESS;
*/

			struct sockaddr_in6 *sain = (struct sockaddr_in6 *) &a;
			cw_put_data(dst + 4, (uint8_t *) & sain->sin6_addr, 16);
			return 16 + cw_put_elem_hdr(dst, id, 16);

		}
	}

	cw_log(DBG_MSG_ERR, "Can't determine CAPWAP Local IP Adress");
	return 0;
}