/**

 * FreeRDP: A Remote Desktop Protocol Implementation

 * Network Transport Layer

 *

 * Copyright 2011 Vic Lee

 *

 * Licensed under the Apache License, Version 2.0 (the "License");

 * you may not use this file except in compliance with the License.

 * You may obtain a copy of the License at

 *

 *     http://www.apache.org/licenses/LICENSE-2.0

 *

 * Unless required by applicable law or agreed to in writing, software

 * distributed under the License is distributed on an "AS IS" BASIS,

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

 * See the License for the specific language governing permissions and

 * limitations under the License.

 */

#ifndef FREERDP_LIB_CORE_TRANSPORT_H

#define FREERDP_LIB_CORE_TRANSPORT_H

typedef enum

{

	TRANSPORT_LAYER_TCP,

	TRANSPORT_LAYER_TLS,

	TRANSPORT_LAYER_TSG,

	TRANSPORT_LAYER_TSG_TLS,

	TRANSPORT_LAYER_CLOSED

} TRANSPORT_LAYER;

typedef struct rdp_transport rdpTransport;

#include "tcp.h"

#include "nla.h"

#include "gateway/tsg.h"

#include "gateway/rdg.h"

#include <winpr/sspi.h>

#include <winpr/wlog.h>

#include <winpr/synch.h>

#include <winpr/thread.h>

#include <winpr/stream.h>

#include <winpr/collections.h>

#include <freerdp/api.h>

#include <freerdp/crypto/tls.h>

#include <time.h>

#include <freerdp/types.h>

#include <freerdp/settings.h>

typedef int (*TransportRecv)(rdpTransport* transport, wStream* stream, void* extra);

struct rdp_transport

{

	TRANSPORT_LAYER layer;

	BIO* frontBio;

	rdpRdg* rdg;

	rdpTsg* tsg;

	rdpTls* tls;

	rdpContext* context;

	rdpNla* nla;

	rdpSettings* settings;

	void* ReceiveExtra;

	wStream* ReceiveBuffer;

	TransportRecv ReceiveCallback;

	wStreamPool* ReceivePool;

	HANDLE connectedEvent;

	BOOL NlaMode;

	BOOL blocking;

	BOOL GatewayEnabled;

	CRITICAL_SECTION ReadLock;

	CRITICAL_SECTION WriteLock;

	ULONG written;

	HANDLE rereadEvent;

	BOOL haveMoreBytesToRead;

	wLog* log;

};

FREERDP_LOCAL wStream* transport_send_stream_init(rdpTransport* transport, int size);

FREERDP_LOCAL BOOL transport_connect(rdpTransport* transport, const char* hostname, UINT16 port,

                                     DWORD timeout);

FREERDP_LOCAL BOOL transport_attach(rdpTransport* transport, int sockfd);

FREERDP_LOCAL BOOL transport_disconnect(rdpTransport* transport);

FREERDP_LOCAL BOOL transport_connect_rdp(rdpTransport* transport);

FREERDP_LOCAL BOOL transport_connect_tls(rdpTransport* transport);

FREERDP_LOCAL BOOL transport_connect_nla(rdpTransport* transport);

FREERDP_LOCAL BOOL transport_accept_rdp(rdpTransport* transport);

FREERDP_LOCAL BOOL transport_accept_tls(rdpTransport* transport);

FREERDP_LOCAL BOOL transport_accept_nla(rdpTransport* transport);

FREERDP_LOCAL int transport_read_pdu(rdpTransport* transport, wStream* s);

FREERDP_LOCAL int transport_write(rdpTransport* transport, wStream* s);

FREERDP_LOCAL void transport_get_fds(rdpTransport* transport, void** rfds, int* rcount);

FREERDP_LOCAL int transport_check_fds(rdpTransport* transport);

FREERDP_LOCAL DWORD transport_get_event_handles(rdpTransport* transport, HANDLE* events,

                                                DWORD nCount);

FREERDP_LOCAL BOOL transport_set_blocking_mode(rdpTransport* transport, BOOL blocking);

FREERDP_LOCAL void transport_set_gateway_enabled(rdpTransport* transport, BOOL GatewayEnabled);

FREERDP_LOCAL void transport_set_nla_mode(rdpTransport* transport, BOOL NlaMode);

FREERDP_LOCAL BOOL transport_is_write_blocked(rdpTransport* transport);

FREERDP_LOCAL int transport_drain_output_buffer(rdpTransport* transport);

FREERDP_LOCAL wStream* transport_receive_pool_take(rdpTransport* transport);

FREERDP_LOCAL int transport_receive_pool_return(rdpTransport* transport, wStream* pdu);

FREERDP_LOCAL rdpTransport* transport_new(rdpContext* context);

FREERDP_LOCAL void transport_free(rdpTransport* transport);

#endif /* FREERDP_LIB_CORE_TRANSPORT_H */