/*

* Copyright (c) 2008 Nokia Corporation and/or its subsidiary(-ies).

* All rights reserved.

* This component and the accompanying materials are made available

* under the terms of "Eclipse Public License v1.0"

* which accompanies this distribution, and is available

* at the URL "http://www.eclipse.org/legal/epl-v10.html".

*

* Initial Contributors:

* Nokia Corporation - initial contribution.

*

* Contributors:

*

* Description:  ?Description

*

*/

#include <sys/socket.h>

#include <netinet/in.h>

#include <netdb.h>

#include <unistd.h>

#include <errno.h>

#include <arpa/inet.h>

#include <fcntl.h>

#include <sys/select.h>

#ifndef __SYMBIAN32__

#include <netinet/tcp.h>  // TCP_NODELAY

#endif

#include <string.h>

#include "logger.h"

#include "socket.h"

const int INVALID_SOCKET = 0;

namespace java

{

namespace comms

{

Socket::Socket() : mSocket(INVALID_SOCKET)

{

}

Socket::Socket(int aSocket) : mSocket(aSocket)

{

    setSockOpts();

}

Socket::~Socket()

{

    close();

}

int Socket::open(int aPort)

{

    sockaddr_in addr;

    addr.sin_family = AF_INET;

    addr.sin_port = htons(aPort);

    addr.sin_addr.s_addr = inet_addr("127.0.0.1");

    mSocket = socket(AF_INET, SOCK_STREAM , 0);

    setSockOpts();

    int rc = connect(mSocket, (sockaddr*)&addr, sizeof(addr));

    if (rc<0)

    {

        ELOG2(EJavaComms, "connect failed %d - %s", errno, strerror(errno));

        close();

    }

    else

    {

        rc = 0;

    }

    return rc;

}

// returns: 0   ok

//          < 0 in error

int Socket::read(char *aBuf, int len)

{

    int rc = 0;

    do

    {

        if (mSocket==INVALID_SOCKET) return -1;

        fd_set readset;

        FD_ZERO(&readset);

        FD_SET(mSocket, &readset);

        timeval tv;

        tv.tv_sec = 600;

        tv.tv_usec = 0;

        rc = select(mSocket+1, &readset, 0, 0, &tv);

        if (rc < 0 && errno != EINTR)

        {

            ELOG3(EJavaComms, "select failed %d - %s - fd=%d", errno, strerror(errno), mSocket);

            break;

        }

        else if (rc > 0 && FD_ISSET(mSocket, &readset))

        {

            rc = recv(aBuf, len);

            break;

        }

    }

    while (1);

    return rc;

}

// returns: 0   ok (all bytes read)

//          < 0 in error

int Socket::recv(char *aBuf, int len)

{

    int count = len;

    while (count > 0)

    {

        int rc = ::recv(mSocket, aBuf, count, 0);

        if (rc < 0)

        {

            if (errno == EINTR) continue;

            WLOG3(EJavaComms, "recv failed %d - %s - fd=%d", errno, strerror(errno), mSocket);

            return -1;

        }

        if (rc == 0)     // EOF

        {

//            LOG1(EJavaComms, EInfo, "EOF received - exiting fd=%d", mSocket);

            return -1;

        }

        aBuf += rc;

        count -= rc;

    }

    return 0;

}

// returns: 0   in ok

//          < 0 in error

int Socket::write(const char *aBuf, int len)

{

    int sent = 0;

    do

    {

        int rc = send(mSocket, aBuf+sent, len-sent, 0);

        if (rc > 0)

            sent += rc;

        else if (rc < 0 && errno != EINTR)

        {

            WLOG3(EJavaComms, "send failed %d - %s - fd=%d", errno, strerror(errno), mSocket);

            return -1;

        }

    }

    while (len > sent);

    return 0;

}

int Socket::shutdown(int aHow)

{

    int rc = ::shutdown(mSocket, aHow);

    if (rc<0) WLOG3(EJavaComms, "shutdown failed %d - %s - fd=%d", errno, strerror(errno), mSocket);

    return rc;

}

void Socket::close()

{

    if (mSocket !=INVALID_SOCKET)

    {

        int rc = ::close(mSocket);

        if (rc<0) WLOG3(EJavaComms, "close failed %d - %s - fd=%d", errno, strerror(errno), mSocket);

        mSocket = INVALID_SOCKET;

    }

}

int Socket::getSocket() const

{

    return mSocket;

}

void Socket::setSockOpts()

{

    int flag = 1;

    int rc = setsockopt(mSocket, SOL_SOCKET, SO_REUSEADDR, (char*)&flag, sizeof(int));

    if (rc<0) ELOG3(EJavaComms, "setsockopt(SO_REUSEADDR) failed %d - %s - fd=%d", errno, strerror(errno), mSocket);

    rc = setsockopt(mSocket, IPPROTO_TCP, TCP_NODELAY, (char*)&flag, sizeof(int));

    if (rc<0) ELOG3(EJavaComms, "setsockopt(TCP_NODELAY) failed %d - %s - fd=%d", errno, strerror(errno), mSocket);

}

} // namespace comms

} // namespace java