cmd/lms/PortForwardRequest.h

/*******************************************************************************
 * Copyright (C) 2004-2008 Intel Corp. All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 *  - Redistributions of source code must retain the above copyright notice,
 *    this list of conditions and the following disclaimer.
 *
 *  - Redistributions in binary form must reproduce the above copyright notice,
 *    this list of conditions and the following disclaimer in the documentation
 *    and/or other materials provided with the distribution.
 *
 *  - Neither the name of Intel Corp. nor the names of its
 *    contributors may be used to endorse or promote products derived from this
 *    software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS ``AS IS''
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL Intel Corp. OR THE CONTRIBUTORS
 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 *******************************************************************************/

#ifndef _PORT_FORWARD_REQUEST_H_
#define _PORT_FORWARD_REQUEST_H_

#include <string>

#if defined(__sun) || defined(_LINUX)
#ifndef SOCKET
#define SOCKET int
#endif
#endif

typedef int (*IsConnectionPermittedCallback)(void * const param, SOCKET s);

class PortForwardRequest
{
public:
    enum PORT_FORWARD_REQUEST_STATUS {
        NOT_ACTIVE,
        PENDING_REQUEST,
        LISTENING
    };

    PortForwardRequest(std::string bindedAddress, int port,
               SOCKET listeningSocket,
               IsConnectionPermittedCallback cb, bool isLocal) :
        _bindedAddress(bindedAddress),
        _port(port),
        _local(isLocal),
        _listeningSocket(listeningSocket),
        _cb(cb),
        _status(NOT_ACTIVE),
        _channelCount(0) {}

    const std::string GetBindedAddress() const { return _bindedAddress; }
    const unsigned int GetPort() const { return _port; }
    SOCKET GetListeningSocket() const { return _listeningSocket; }

    int IsConnectionPermitted(void *param, SOCKET s)
    {
        if (_cb != NULL) {
            return _cb(param, s);
        } else {
            return -1;
        }
    }

    PORT_FORWARD_REQUEST_STATUS GetStatus() { return _status; }
    bool IsLocal() { return _local; }
    bool SetStatus(PORT_FORWARD_REQUEST_STATUS newStatus) { _status = newStatus; return true; }
    unsigned int GetChannelCount() { return _channelCount; }
    unsigned int IncreaseChannelCount() { return ++_channelCount; }

    unsigned int DecreaseChannelCount()
    {
        if (_channelCount > 0) {
            --_channelCount;
        }
        return _channelCount;
    }


private:
    const std::string _bindedAddress;
    const unsigned int _port;
    const bool _local;
    const SOCKET _listeningSocket;
    const IsConnectionPermittedCallback _cb;
    PORT_FORWARD_REQUEST_STATUS _status;
    unsigned int _channelCount;
};

#endif