/*
 *
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 * 
 *   http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 *
 */

#include "qpid/sys/Socket.h"

#include "qpid/QpidError.h"
#include "check.h"
#include "PrivatePosix.h"

#include <fcntl.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/errno.h>
#include <netinet/in.h>
#include <netdb.h>

#include <boost/format.hpp>

namespace qpid {
namespace sys {

class SocketPrivate {
public:
	SocketPrivate(int f = -1) :
		fd(f)
	{}

	int fd;
};

Socket::Socket() :
	impl(new SocketPrivate)
{
	createTcp();
}

Socket::Socket(SocketPrivate* sp) :
	impl(sp)
{}

Socket::~Socket() {
	delete impl;
}

void Socket::createTcp() const
{
	int& socket = impl->fd;
	if (socket != -1) Socket::close();
    int s = ::socket (PF_INET, SOCK_STREAM, 0);
    if (s < 0) throw QPID_POSIX_ERROR(errno);
    socket = s;
}

void Socket::setTimeout(const Duration& interval) const
{
	const int& socket = impl->fd;
    struct timeval tv;
    toTimeval(tv, interval);
    setsockopt(socket, SOL_SOCKET, SO_SNDTIMEO, &tv, sizeof(tv));
    setsockopt(socket, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(tv));
}

void Socket::setNonblocking() const {
    QPID_POSIX_CHECK(::fcntl(impl->fd, F_SETFL, O_NONBLOCK));
}


void Socket::connect(const std::string& host, int port) const
{
	const int& socket = impl->fd;
    struct sockaddr_in name;
    name.sin_family = AF_INET;
    name.sin_port = htons(port);
    // TODO: Be good to make this work for IPv6 as well as IPv4
    // Use more modern lookup functions
    struct hostent* hp = gethostbyname ( host.c_str() );
    if (hp == 0) throw QPID_POSIX_ERROR(errno);
    memcpy(&name.sin_addr.s_addr, hp->h_addr_list[0], hp->h_length);
    if (::connect(socket, (struct sockaddr*)(&name), sizeof(name)) < 0)
        throw QPID_POSIX_ERROR(errno);
}

void
Socket::close() const
{
	int& socket = impl->fd;
    if (socket == -1) return;
    if (::close(socket) < 0) throw QPID_POSIX_ERROR(errno);
    socket = -1;
}

ssize_t
Socket::send(const void* data, size_t size) const
{
	const int& socket = impl->fd;
    ssize_t sent = ::send(socket, data, size, 0);
    if (sent < 0) {
        if (errno == ECONNRESET) return SOCKET_EOF;
        if (errno == ETIMEDOUT) return SOCKET_TIMEOUT;
        throw QPID_POSIX_ERROR(errno);
    }
    return sent;
}

ssize_t
Socket::recv(void* data, size_t size) const
{
	const int& socket = impl->fd;
    ssize_t received = ::recv(socket, data, size, 0);
    if (received < 0) {
        if (errno == ETIMEDOUT) return SOCKET_TIMEOUT;
        throw QPID_POSIX_ERROR(errno);
    }
    return received;
}

int Socket::listen(int port, int backlog) const
{
    const int& socket = impl->fd;
    int yes=1;
    QPID_POSIX_CHECK(setsockopt(socket,SOL_SOCKET,SO_REUSEADDR,&yes,sizeof(yes)));
    struct sockaddr_in name;
    name.sin_family = AF_INET;
    name.sin_port = htons(port);
    name.sin_addr.s_addr = 0;
    if (::bind(socket, (struct sockaddr*)&name, sizeof(name)) < 0)
        throw QPID_POSIX_ERROR(errno);
    if (::listen(socket, backlog) < 0)
        throw QPID_POSIX_ERROR(errno);
    
    socklen_t namelen = sizeof(name);
    if (::getsockname(socket, (struct sockaddr*)&name, &namelen) < 0)
        throw QPID_POSIX_ERROR(errno);

    return ntohs(name.sin_port);
}

Socket* Socket::accept(struct sockaddr *addr, socklen_t *addrlen) const
{
	int afd = ::accept(impl->fd, addr, addrlen);
	if ( afd >= 0)
		return new Socket(new SocketPrivate(afd));
	else if (errno == EAGAIN)
		return 0;
    else throw QPID_POSIX_ERROR(errno);
}

int Socket::read(void *buf, size_t count) const
{
	return ::read(impl->fd, buf, count);
}

int Socket::write(const void *buf, size_t count) const
{
	return ::write(impl->fd, buf, count);
}

std::string Socket::getSockname() const
{
    ::sockaddr_storage name; // big enough for any socket address    
    ::socklen_t namelen = sizeof(name);

	const int& socket = impl->fd;
    if (::getsockname(socket, (::sockaddr*)&name, &namelen) < 0)
        throw QPID_POSIX_ERROR(errno);
    
    char dispName[NI_MAXHOST];
    if (int rc=::getnameinfo((::sockaddr*)&name, namelen, dispName, sizeof(dispName), 0, 0, NI_NUMERICHOST) != 0)
    	throw QPID_POSIX_ERROR(rc);
    return dispName;
}

int toFd(const SocketPrivate* s)
{
    return s->fd;
}

}} // namespace qpid::sys
