617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan/*******************************************************************************
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan * Copyright (C) 2004-2008 Intel Corp. All rights reserved.
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan *
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan * Redistribution and use in source and binary forms, with or without
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan * modification, are permitted provided that the following conditions are met:
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan *
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan * - Redistributions of source code must retain the above copyright notice,
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan * this list of conditions and the following disclaimer.
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan *
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan * - Redistributions in binary form must reproduce the above copyright notice,
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan * this list of conditions and the following disclaimer in the documentation
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan * and/or other materials provided with the distribution.
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan *
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan * - Neither the name of Intel Corp. nor the names of its
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan * contributors may be used to endorse or promote products derived from this
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan * software without specific prior written permission.
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan *
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS ``AS IS''
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan * ARE DISCLAIMED. IN NO EVENT SHALL Intel Corp. OR THE CONTRIBUTORS
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan * POSSIBILITY OF SUCH DAMAGE.
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan *******************************************************************************/
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan#ifndef _PORT_FORWARD_REQUEST_H_
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan#define _PORT_FORWARD_REQUEST_H_
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan#include <string>
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan#if defined(__sun) || defined(_LINUX)
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan#ifndef SOCKET
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan#define SOCKET int
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan#endif
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan#endif
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logantypedef int (*IsConnectionPermittedCallback)(void * const param, SOCKET s);
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Loganclass PortForwardRequest
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan{
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Loganpublic:
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan enum PORT_FORWARD_REQUEST_STATUS {
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan NOT_ACTIVE,
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan PENDING_REQUEST,
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan LISTENING
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan };
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan PortForwardRequest(std::string bindedAddress, int port,
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan SOCKET listeningSocket,
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan IsConnectionPermittedCallback cb, bool isLocal) :
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan _bindedAddress(bindedAddress),
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan _port(port),
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan _local(isLocal),
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan _listeningSocket(listeningSocket),
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan _cb(cb),
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan _status(NOT_ACTIVE),
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan _channelCount(0) {}
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan const std::string GetBindedAddress() const { return _bindedAddress; }
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan const unsigned int GetPort() const { return _port; }
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan SOCKET GetListeningSocket() const { return _listeningSocket; }
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan int IsConnectionPermitted(void *param, SOCKET s)
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan {
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan if (_cb != NULL) {
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan return _cb(param, s);
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan } else {
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan return -1;
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan }
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan }
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan PORT_FORWARD_REQUEST_STATUS GetStatus() { return _status; }
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan bool IsLocal() { return _local; }
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan bool SetStatus(PORT_FORWARD_REQUEST_STATUS newStatus) { _status = newStatus; return true; }
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan unsigned int GetChannelCount() { return _channelCount; }
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan unsigned int IncreaseChannelCount() { return ++_channelCount; }
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan unsigned int DecreaseChannelCount()
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan {
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan if (_channelCount > 0) {
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan --_channelCount;
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan }
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan return _channelCount;
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan }
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Loganprivate:
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan const std::string _bindedAddress;
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan const unsigned int _port;
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan const bool _local;
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan const SOCKET _listeningSocket;
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan const IsConnectionPermittedCallback _cb;
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan PORT_FORWARD_REQUEST_STATUS _status;
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan unsigned int _channelCount;
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan};
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan
617e2443dfc17fe44fd44c0675d6aad2ffc9df42Mark Logan#endif