/****************************************************************************

**

** Copyright (C) 2010 Nokia Corporation and/or its subsidiary(-ies).

** All rights reserved.

** Contact: Nokia Corporation (qt-info@nokia.com)

**

** This file is part of the QtNetwork module of the Qt Toolkit.

**

** $QT_BEGIN_LICENSE:LGPL$

** No Commercial Usage

** This file contains pre-release code and may not be distributed.

** You may use this file in accordance with the terms and conditions

** contained in the Technology Preview License Agreement accompanying

** this package.

**

** GNU Lesser General Public License Usage

** Alternatively, this file may be used under the terms of the GNU Lesser

** General Public License version 2.1 as published by the Free Software

** Foundation and appearing in the file LICENSE.LGPL included in the

** packaging of this file.  Please review the following information to

** ensure the GNU Lesser General Public License version 2.1 requirements

** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.

**

** In addition, as a special exception, Nokia gives you certain additional

** rights.  These rights are described in the Nokia Qt LGPL Exception

** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.

**

** If you have questions regarding the use of this file, please contact

** Nokia at qt-info@nokia.com.

**

**

**

**

**

**

**

**

** $QT_END_LICENSE$

**

****************************************************************************/

#include "qlocalserver.h"

#include "qlocalserver_p.h"

#include "qlocalsocket.h"

#include <qdebug.h>

// The buffer size need to be 0 otherwise data could be

// lost if the socket that has written data closes the connection

// before it is read.  Pipewriter is used for write buffering.

#define BUFSIZE 0

// ###: This should be a property. Should replace the insane 50 on unix as well.

#define SYSTEM_MAX_PENDING_SOCKETS 8

QT_BEGIN_NAMESPACE

bool QLocalServerPrivate::addListener()

{

    // The object must not change its address once the

    // contained OVERLAPPED struct is passed to Windows.

    listeners << Listener();

    Listener &listener = listeners.last();

    listener.handle = CreateNamedPipe(

                 (const wchar_t *)fullServerName.utf16(), // pipe name

                 PIPE_ACCESS_DUPLEX | FILE_FLAG_OVERLAPPED,       // read/write access

                 PIPE_TYPE_MESSAGE |       // message type pipe

                 PIPE_READMODE_MESSAGE |   // message-read mode

                 PIPE_WAIT,                // blocking mode

                 PIPE_UNLIMITED_INSTANCES, // max. instances

                 BUFSIZE,                  // output buffer size

                 BUFSIZE,                  // input buffer size

                 3000,                     // client time-out

                 NULL);

    if (listener.handle == INVALID_HANDLE_VALUE) {

        setError(QLatin1String("QLocalServerPrivate::addListener"));

        listeners.removeLast();

        return false;

    }

    memset(&listener.overlapped, 0, sizeof(listener.overlapped));

    listener.overlapped.hEvent = eventHandle;

    if (!ConnectNamedPipe(listener.handle, &listener.overlapped)) {

        switch (GetLastError()) {

        case ERROR_IO_PENDING:

            break;

        case ERROR_PIPE_CONNECTED:

            SetEvent(eventHandle);

            break;

        default:

            CloseHandle(listener.handle);

            setError(QLatin1String("QLocalServerPrivate::addListener"));

            listeners.removeLast();

            return false;

        }

    } else {

        Q_ASSERT_X(false, "QLocalServerPrivate::addListener", "The impossible happened");

        SetEvent(eventHandle);

    }

    return true;

}

void QLocalServerPrivate::setError(const QString &function)

{

    int windowsError = GetLastError();

    errorString = QString::fromLatin1("%1: %2").arg(function).arg(qt_error_string(windowsError));

    error = QAbstractSocket::UnknownSocketError;

}

void QLocalServerPrivate::init()

{

}

bool QLocalServerPrivate::removeServer(const QString &name)

{

    Q_UNUSED(name);

    return true;

}

bool QLocalServerPrivate::listen(const QString &name)

{

    Q_Q(QLocalServer);

    QString pipePath = QLatin1String("\\\\.\\pipe\\");

    if (name.startsWith(pipePath))

        fullServerName = name;

    else

        fullServerName = pipePath + name;

    // Use only one event for all listeners of one socket.

    // The idea is that listener events are rare, so polling all listeners once in a while is

    // cheap compared to waiting for N additional events in each iteration of the main loop.

    eventHandle = CreateEvent(NULL, TRUE, FALSE, NULL);

    connectionEventNotifier = new QWinEventNotifier(eventHandle , q);

    q->connect(connectionEventNotifier, SIGNAL(activated(HANDLE)), q, SLOT(_q_onNewConnection()));

    for (int i = 0; i < SYSTEM_MAX_PENDING_SOCKETS; ++i)

        if (!addListener())

            return false;

    return true;

}

void QLocalServerPrivate::_q_onNewConnection()

{

    Q_Q(QLocalServer);

    DWORD dummy;

    // Reset first, otherwise we could reset an event which was asserted

    // immediately after we checked the conn status.

    ResetEvent(eventHandle);

    // Testing shows that there is indeed absolutely no guarantee which listener gets

    // a client connection first, so there is no way around polling all of them.

    for (int i = 0; i < listeners.size(); ) {

        HANDLE handle = listeners[i].handle;

        if (GetOverlappedResult(handle, &listeners[i].overlapped, &dummy, FALSE)) {

            listeners.removeAt(i);

            addListener();

            if (pendingConnections.size() > maxPendingConnections)

                connectionEventNotifier->setEnabled(false);

            // Make this the last thing so connected slots can wreak the least havoc

            q->incomingConnection((quintptr)handle);

        } else {

            if (GetLastError() != ERROR_IO_INCOMPLETE) {

                setError(QLatin1String("QLocalServerPrivate::_q_onNewConnection"));

                return;

            }

            ++i;

        }

    }

}

void QLocalServerPrivate::closeServer()

{

    connectionEventNotifier->setEnabled(false); // Otherwise, closed handle is checked before deleter runs

    connectionEventNotifier->deleteLater();

    connectionEventNotifier = 0;

    CloseHandle(eventHandle);

    for (int i = 0; i < listeners.size(); ++i)

        CloseHandle(listeners[i].handle);

    listeners.clear();

}

void QLocalServerPrivate::waitForNewConnection(int msecs, bool *timedOut)

{

    Q_Q(QLocalServer);

    if (!pendingConnections.isEmpty() || !q->isListening())

        return;

    DWORD result = WaitForSingleObject(eventHandle, (msecs == -1) ? INFINITE : msecs);

    if (result == WAIT_TIMEOUT) {

        if (timedOut)

            *timedOut = true;

    } else {

        _q_onNewConnection();

    }

}

QT_END_NAMESPACE