/*
 * Copyright 2010-2014 Haiku Inc. All rights reserved.
 * Distributed under the terms of the MIT License.
 *
 * Authors:
 *              Christophe Huriaux, c.huriaux@gmail.com
 *              Niels Sascha Reedijk, niels.reedijk@gmail.com
 *              Adrien Destugues, pulkomandy@pulkomandy.tk
 */


#include <NetworkRequest.h>

#include <AbstractSocket.h>

using namespace BPrivate::Network;


BNetworkRequest::BNetworkRequest(const BUrl& url, BDataIO* output,
        BUrlProtocolListener* listener, BUrlContext* context,
        const char* threadName, const char* protocolName)
        :
        BUrlRequest(url, output, listener, context, threadName, protocolName),
        fSocket(NULL)
{
}


status_t
BNetworkRequest::Stop()
{
        status_t threadStatus = BUrlRequest::Stop();

        if (threadStatus != B_OK)
                return threadStatus;

        send_signal(fThreadId, SIGUSR1); // unblock blocking syscalls.
        wait_for_thread(fThreadId, &threadStatus);
        return threadStatus;
}


void
BNetworkRequest::SetTimeout(bigtime_t timeout)
{
        if (fSocket != NULL)
                fSocket->SetTimeout(timeout);
}


bool
BNetworkRequest::_ResolveHostName(BString host, uint16_t port)
{
        _EmitDebug(B_URL_PROTOCOL_DEBUG_TEXT, "Resolving %s",
                fUrl.UrlString().String());

        fRemoteAddr = BNetworkAddress(host, port);
        if (fRemoteAddr.InitCheck() != B_OK)
                return false;

        //! ProtocolHook:HostnameResolved
        if (fListener != NULL)
                fListener->HostnameResolved(this, fRemoteAddr.ToString().String());

        _EmitDebug(B_URL_PROTOCOL_DEBUG_TEXT, "Hostname resolved to: %s",
                fRemoteAddr.ToString().String());

        return true;
}


static void
empty(int)
{
}


void
BNetworkRequest::_ProtocolSetup()
{
        // Setup an (empty) signal handler so we can be stopped by a signal,
        // without the whole process being killed.
        // TODO make connect() properly unlock when close() is called on the
        // socket, and remove this.
        struct sigaction action;
        action.sa_handler = empty;
        action.sa_mask = 0;
        action.sa_flags = 0;
        sigaction(SIGUSR1, &action, NULL);
}


status_t
BNetworkRequest::_GetLine(BString& destString)
{
        // Find a complete line in inputBuffer
        uint32 characterIndex = 0;

        while ((characterIndex < fInputBuffer.Size())
                && ((fInputBuffer.Data())[characterIndex] != '\n'))
                characterIndex++;

        if (characterIndex == fInputBuffer.Size())
                return B_ERROR;

        char* temporaryBuffer = new(std::nothrow) char[characterIndex + 1];
        if (temporaryBuffer == NULL)
                return B_NO_MEMORY;

        fInputBuffer.RemoveData(temporaryBuffer, characterIndex + 1);

        // Strip end-of-line character(s)
        if (characterIndex != 0 && temporaryBuffer[characterIndex - 1] == '\r')
                destString.SetTo(temporaryBuffer, characterIndex - 1);
        else
                destString.SetTo(temporaryBuffer, characterIndex);

        delete[] temporaryBuffer;
        return B_OK;
}