a180a41bba1d50822df23fff0099e90b86638b89vboxsync/* -*- c-basic-offset: 8 -*-
a180a41bba1d50822df23fff0099e90b86638b89vboxsync rdesktop: A Remote Desktop Protocol client.
a180a41bba1d50822df23fff0099e90b86638b89vboxsync Protocol services - TCP layer
a180a41bba1d50822df23fff0099e90b86638b89vboxsync Copyright (C) Matthew Chapman <matthewc.unsw.edu.au> 1999-2008
a180a41bba1d50822df23fff0099e90b86638b89vboxsync Copyright 2005-2011 Peter Astrand <astrand@cendio.se> for Cendio AB
a180a41bba1d50822df23fff0099e90b86638b89vboxsync Copyright 2012-2013 Henrik Andersson <hean01@cendio.se> for Cendio AB
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync This program is free software: you can redistribute it and/or modify
a180a41bba1d50822df23fff0099e90b86638b89vboxsync it under the terms of the GNU General Public License as published by
a180a41bba1d50822df23fff0099e90b86638b89vboxsync the Free Software Foundation, either version 3 of the License, or
a180a41bba1d50822df23fff0099e90b86638b89vboxsync (at your option) any later version.
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync This program is distributed in the hope that it will be useful,
a180a41bba1d50822df23fff0099e90b86638b89vboxsync but WITHOUT ANY WARRANTY; without even the implied warranty of
a180a41bba1d50822df23fff0099e90b86638b89vboxsync MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
a180a41bba1d50822df23fff0099e90b86638b89vboxsync GNU General Public License for more details.
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync You should have received a copy of the GNU General Public License
a180a41bba1d50822df23fff0099e90b86638b89vboxsync along with this program. If not, see <http://www.gnu.org/licenses/>.
a180a41bba1d50822df23fff0099e90b86638b89vboxsync*/
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
6e9aa255e3376b2da5824c09c4c62bc233463bfevboxsync/*
6e9aa255e3376b2da5824c09c4c62bc233463bfevboxsync * Oracle GPL Disclaimer: For the avoidance of doubt, except that if any license choice
6e9aa255e3376b2da5824c09c4c62bc233463bfevboxsync * other than GPL or LGPL is available it will apply instead, Oracle elects to use only
6e9aa255e3376b2da5824c09c4c62bc233463bfevboxsync * the General Public License version 2 (GPLv2) at this time for any software where
6e9aa255e3376b2da5824c09c4c62bc233463bfevboxsync * a choice of GPL license versions is made available with the language indicating
6e9aa255e3376b2da5824c09c4c62bc233463bfevboxsync * that GPLv2 or any later version may be used, or where a choice of which version
6e9aa255e3376b2da5824c09c4c62bc233463bfevboxsync * of the GPL is applied is otherwise unspecified.
6e9aa255e3376b2da5824c09c4c62bc233463bfevboxsync */
6e9aa255e3376b2da5824c09c4c62bc233463bfevboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#ifndef _WIN32
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#include <unistd.h> /* select read write close */
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#include <sys/socket.h> /* socket connect setsockopt */
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#include <sys/time.h> /* timeval */
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#include <netdb.h> /* gethostbyname */
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#include <netinet/in.h> /* sockaddr_in */
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#include <netinet/tcp.h> /* TCP_NODELAY */
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#include <arpa/inet.h> /* inet_addr */
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#include <errno.h> /* errno */
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#endif
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#include <openssl/ssl.h>
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#include <openssl/x509.h>
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#include <openssl/err.h>
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#include "rdesktop.h"
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#ifdef _WIN32
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#define socklen_t int
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#define TCP_CLOSE(_sck) closesocket(_sck)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#define TCP_STRERROR "tcp error"
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#define TCP_BLOCKS (WSAGetLastError() == WSAEWOULDBLOCK)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#else
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#define TCP_CLOSE(_sck) close(_sck)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#define TCP_STRERROR strerror(errno)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#define TCP_BLOCKS (errno == EWOULDBLOCK)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#endif
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#ifndef INADDR_NONE
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#define INADDR_NONE ((unsigned long) -1)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#endif
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#ifdef WITH_SCARD
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#define STREAM_COUNT 8
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#else
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#define STREAM_COUNT 1
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#endif
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsyncstatic RD_BOOL g_ssl_initialized = False;
a180a41bba1d50822df23fff0099e90b86638b89vboxsyncstatic SSL *g_ssl = NULL;
a180a41bba1d50822df23fff0099e90b86638b89vboxsyncstatic SSL_CTX *g_ssl_ctx = NULL;
a180a41bba1d50822df23fff0099e90b86638b89vboxsyncstatic int g_sock;
a180a41bba1d50822df23fff0099e90b86638b89vboxsyncstatic RD_BOOL g_run_ui = False;
a180a41bba1d50822df23fff0099e90b86638b89vboxsyncstatic struct stream g_in;
a180a41bba1d50822df23fff0099e90b86638b89vboxsyncstatic struct stream g_out[STREAM_COUNT];
a180a41bba1d50822df23fff0099e90b86638b89vboxsyncint g_tcp_port_rdp = TCP_PORT_RDP;
a180a41bba1d50822df23fff0099e90b86638b89vboxsyncextern RD_BOOL g_user_quit;
a180a41bba1d50822df23fff0099e90b86638b89vboxsyncextern RD_BOOL g_network_error;
a180a41bba1d50822df23fff0099e90b86638b89vboxsyncextern RD_BOOL g_reconnect_loop;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync/* wait till socket is ready to write or timeout */
a180a41bba1d50822df23fff0099e90b86638b89vboxsyncstatic RD_BOOL
a180a41bba1d50822df23fff0099e90b86638b89vboxsynctcp_can_send(int sck, int millis)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync{
a180a41bba1d50822df23fff0099e90b86638b89vboxsync fd_set wfds;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync struct timeval time;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync int sel_count;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync time.tv_sec = millis / 1000;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync time.tv_usec = (millis * 1000) % 1000000;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync FD_ZERO(&wfds);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync FD_SET(sck, &wfds);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync sel_count = select(sck + 1, 0, &wfds, 0, &time);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (sel_count > 0)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync return True;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync return False;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync}
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync/* Initialise TCP transport data packet */
a180a41bba1d50822df23fff0099e90b86638b89vboxsyncSTREAM
a180a41bba1d50822df23fff0099e90b86638b89vboxsynctcp_init(uint32 maxlen)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync{
a180a41bba1d50822df23fff0099e90b86638b89vboxsync static int cur_stream_id = 0;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync STREAM result = NULL;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#ifdef WITH_SCARD
a180a41bba1d50822df23fff0099e90b86638b89vboxsync scard_lock(SCARD_LOCK_TCP);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#endif
a180a41bba1d50822df23fff0099e90b86638b89vboxsync result = &g_out[cur_stream_id];
a180a41bba1d50822df23fff0099e90b86638b89vboxsync cur_stream_id = (cur_stream_id + 1) % STREAM_COUNT;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (maxlen > result->size)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync result->data = (uint8 *) xrealloc(result->data, maxlen);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync result->size = maxlen;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync result->p = result->data;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync result->end = result->data + result->size;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#ifdef WITH_SCARD
a180a41bba1d50822df23fff0099e90b86638b89vboxsync scard_unlock(SCARD_LOCK_TCP);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#endif
a180a41bba1d50822df23fff0099e90b86638b89vboxsync return result;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync}
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync/* Send TCP transport data packet */
a180a41bba1d50822df23fff0099e90b86638b89vboxsyncvoid
a180a41bba1d50822df23fff0099e90b86638b89vboxsynctcp_send(STREAM s)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync{
a180a41bba1d50822df23fff0099e90b86638b89vboxsync int ssl_err;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync int length = s->end - s->data;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync int sent, total = 0;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (g_network_error == True)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync return;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#ifdef WITH_SCARD
a180a41bba1d50822df23fff0099e90b86638b89vboxsync scard_lock(SCARD_LOCK_TCP);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#endif
a180a41bba1d50822df23fff0099e90b86638b89vboxsync while (total < length)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (g_ssl)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync sent = SSL_write(g_ssl, s->data + total, length - total);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (sent <= 0)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync ssl_err = SSL_get_error(g_ssl, sent);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (sent < 0 && (ssl_err == SSL_ERROR_WANT_READ ||
a180a41bba1d50822df23fff0099e90b86638b89vboxsync ssl_err == SSL_ERROR_WANT_WRITE))
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync tcp_can_send(g_sock, 100);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync sent = 0;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync else
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#ifdef WITH_SCARD
a180a41bba1d50822df23fff0099e90b86638b89vboxsync scard_unlock(SCARD_LOCK_TCP);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#endif
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync error("SSL_write: %d (%s)\n", ssl_err, TCP_STRERROR);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_network_error = True;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync return;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync else
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync sent = send(g_sock, s->data + total, length - total, 0);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (sent <= 0)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (sent == -1 && TCP_BLOCKS)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync tcp_can_send(g_sock, 100);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync sent = 0;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync else
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#ifdef WITH_SCARD
a180a41bba1d50822df23fff0099e90b86638b89vboxsync scard_unlock(SCARD_LOCK_TCP);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#endif
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync error("send: %s\n", TCP_STRERROR);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_network_error = True;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync return;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync total += sent;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#ifdef WITH_SCARD
a180a41bba1d50822df23fff0099e90b86638b89vboxsync scard_unlock(SCARD_LOCK_TCP);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#endif
a180a41bba1d50822df23fff0099e90b86638b89vboxsync}
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync/* Receive a message on the TCP layer */
a180a41bba1d50822df23fff0099e90b86638b89vboxsyncSTREAM
a180a41bba1d50822df23fff0099e90b86638b89vboxsynctcp_recv(STREAM s, uint32 length)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync{
a180a41bba1d50822df23fff0099e90b86638b89vboxsync uint32 new_length, end_offset, p_offset;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync int rcvd = 0, ssl_err;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (g_network_error == True)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync return NULL;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (s == NULL)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync /* read into "new" stream */
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (length > g_in.size)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_in.data = (uint8 *) xrealloc(g_in.data, length);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_in.size = length;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_in.end = g_in.p = g_in.data;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync s = &g_in;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync else
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync /* append to existing stream */
a180a41bba1d50822df23fff0099e90b86638b89vboxsync new_length = (s->end - s->data) + length;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (new_length > s->size)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync p_offset = s->p - s->data;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync end_offset = s->end - s->data;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync s->data = (uint8 *) xrealloc(s->data, new_length);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync s->size = new_length;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync s->p = s->data + p_offset;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync s->end = s->data + end_offset;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync while (length > 0)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if ((!g_ssl || SSL_pending(g_ssl) <= 0) && g_run_ui)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (!ui_select(g_sock))
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync /* User quit */
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_user_quit = True;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync return NULL;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (g_ssl)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync rcvd = SSL_read(g_ssl, s->end, length);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync ssl_err = SSL_get_error(g_ssl, rcvd);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (ssl_err == SSL_ERROR_SSL)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (SSL_get_shutdown(g_ssl) & SSL_RECEIVED_SHUTDOWN)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync error("Remote peer initiated ssl shutdown.\n");
a180a41bba1d50822df23fff0099e90b86638b89vboxsync return NULL;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync ERR_print_errors_fp(stdout);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_network_error = True;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync return NULL;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (ssl_err == SSL_ERROR_WANT_READ || ssl_err == SSL_ERROR_WANT_WRITE)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync rcvd = 0;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync else if (ssl_err != SSL_ERROR_NONE)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync error("SSL_read: %d (%s)\n", ssl_err, TCP_STRERROR);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_network_error = True;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync return NULL;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync else
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync rcvd = recv(g_sock, s->end, length, 0);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (rcvd < 0)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (rcvd == -1 && TCP_BLOCKS)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync rcvd = 0;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync else
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync error("recv: %s\n", TCP_STRERROR);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_network_error = True;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync return NULL;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync else if (rcvd == 0)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync error("Connection closed\n");
a180a41bba1d50822df23fff0099e90b86638b89vboxsync return NULL;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync s->end += rcvd;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync length -= rcvd;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync return s;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync}
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync/* Establish a SSL/TLS 1.0 connection */
a180a41bba1d50822df23fff0099e90b86638b89vboxsyncRD_BOOL
a180a41bba1d50822df23fff0099e90b86638b89vboxsynctcp_tls_connect(void)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync{
a180a41bba1d50822df23fff0099e90b86638b89vboxsync int err;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync long options;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (!g_ssl_initialized)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync SSL_load_error_strings();
a180a41bba1d50822df23fff0099e90b86638b89vboxsync SSL_library_init();
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_ssl_initialized = True;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync /* create process context */
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (g_ssl_ctx == NULL)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_ssl_ctx = SSL_CTX_new(TLSv1_client_method());
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (g_ssl_ctx == NULL)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync error("tcp_tls_connect: SSL_CTX_new() failed to create TLS v1.0 context\n");
a180a41bba1d50822df23fff0099e90b86638b89vboxsync goto fail;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync options = 0;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#ifdef SSL_OP_NO_COMPRESSION
a180a41bba1d50822df23fff0099e90b86638b89vboxsync options |= SSL_OP_NO_COMPRESSION;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#endif // __SSL_OP_NO_COMPRESSION
a180a41bba1d50822df23fff0099e90b86638b89vboxsync options |= SSL_OP_DONT_INSERT_EMPTY_FRAGMENTS;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync SSL_CTX_set_options(g_ssl_ctx, options);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync /* free old connection */
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (g_ssl)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync SSL_free(g_ssl);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync /* create new ssl connection */
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_ssl = SSL_new(g_ssl_ctx);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (g_ssl == NULL)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync error("tcp_tls_connect: SSL_new() failed\n");
a180a41bba1d50822df23fff0099e90b86638b89vboxsync goto fail;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (SSL_set_fd(g_ssl, g_sock) < 1)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync error("tcp_tls_connect: SSL_set_fd() failed\n");
a180a41bba1d50822df23fff0099e90b86638b89vboxsync goto fail;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync do
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync err = SSL_connect(g_ssl);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync while (SSL_get_error(g_ssl, err) == SSL_ERROR_WANT_READ);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (err < 0)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync ERR_print_errors_fp(stdout);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync goto fail;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync return True;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync fail:
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (g_ssl)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync SSL_free(g_ssl);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (g_ssl_ctx)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync SSL_CTX_free(g_ssl_ctx);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_ssl = NULL;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_ssl_ctx = NULL;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync return False;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync}
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync/* Get public key from server of TLS 1.0 connection */
a180a41bba1d50822df23fff0099e90b86638b89vboxsyncRD_BOOL
a180a41bba1d50822df23fff0099e90b86638b89vboxsynctcp_tls_get_server_pubkey(STREAM s)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync{
a180a41bba1d50822df23fff0099e90b86638b89vboxsync X509 *cert = NULL;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync EVP_PKEY *pkey = NULL;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync s->data = s->p = NULL;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync s->size = 0;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (g_ssl == NULL)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync goto out;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync cert = SSL_get_peer_certificate(g_ssl);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (cert == NULL)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync error("tcp_tls_get_server_pubkey: SSL_get_peer_certificate() failed\n");
a180a41bba1d50822df23fff0099e90b86638b89vboxsync goto out;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync pkey = X509_get_pubkey(cert);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (pkey == NULL)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync error("tcp_tls_get_server_pubkey: X509_get_pubkey() failed\n");
a180a41bba1d50822df23fff0099e90b86638b89vboxsync goto out;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync s->size = i2d_PublicKey(pkey, NULL);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (s->size < 1)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync error("tcp_tls_get_server_pubkey: i2d_PublicKey() failed\n");
a180a41bba1d50822df23fff0099e90b86638b89vboxsync goto out;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync s->data = s->p = xmalloc(s->size);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync i2d_PublicKey(pkey, &s->p);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync s->p = s->data;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync s->end = s->p + s->size;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync out:
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (cert)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync X509_free(cert);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (pkey)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync EVP_PKEY_free(pkey);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync return (s->size != 0);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync}
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync/* Establish a connection on the TCP layer */
a180a41bba1d50822df23fff0099e90b86638b89vboxsyncRD_BOOL
a180a41bba1d50822df23fff0099e90b86638b89vboxsynctcp_connect(char *server)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync{
a180a41bba1d50822df23fff0099e90b86638b89vboxsync socklen_t option_len;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync uint32 option_value;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync int i;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#ifdef IPv6
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync int n;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync struct addrinfo hints, *res, *ressave;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync char tcp_port_rdp_s[10];
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync snprintf(tcp_port_rdp_s, 10, "%d", g_tcp_port_rdp);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync memset(&hints, 0, sizeof(struct addrinfo));
a180a41bba1d50822df23fff0099e90b86638b89vboxsync hints.ai_family = AF_UNSPEC;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync hints.ai_socktype = SOCK_STREAM;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if ((n = getaddrinfo(server, tcp_port_rdp_s, &hints, &res)))
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync error("getaddrinfo: %s\n", gai_strerror(n));
a180a41bba1d50822df23fff0099e90b86638b89vboxsync return False;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync ressave = res;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_sock = -1;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync while (res)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_sock = socket(res->ai_family, res->ai_socktype, res->ai_protocol);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (!(g_sock < 0))
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (connect(g_sock, res->ai_addr, res->ai_addrlen) == 0)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync break;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync TCP_CLOSE(g_sock);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_sock = -1;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync res = res->ai_next;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync freeaddrinfo(ressave);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (g_sock == -1)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync error("%s: unable to connect\n", server);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync return False;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#else /* no IPv6 support */
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync struct hostent *nslookup;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync struct sockaddr_in servaddr;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if ((nslookup = gethostbyname(server)) != NULL)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync memcpy(&servaddr.sin_addr, nslookup->h_addr, sizeof(servaddr.sin_addr));
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync else if ((servaddr.sin_addr.s_addr = inet_addr(server)) == INADDR_NONE)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync error("%s: unable to resolve host\n", server);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync return False;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if ((g_sock = socket(AF_INET, SOCK_STREAM, 0)) < 0)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync error("socket: %s\n", TCP_STRERROR);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync return False;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync servaddr.sin_family = AF_INET;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync servaddr.sin_port = htons((uint16) g_tcp_port_rdp);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (connect(g_sock, (struct sockaddr *) &servaddr, sizeof(struct sockaddr)) < 0)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (!g_reconnect_loop)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync error("connect: %s\n", TCP_STRERROR);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync TCP_CLOSE(g_sock);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_sock = -1;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync return False;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync#endif /* IPv6 */
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync option_value = 1;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync option_len = sizeof(option_value);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync setsockopt(g_sock, IPPROTO_TCP, TCP_NODELAY, (void *) &option_value, option_len);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync /* receive buffer must be a least 16 K */
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (getsockopt(g_sock, SOL_SOCKET, SO_RCVBUF, (void *) &option_value, &option_len) == 0)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (option_value < (1024 * 16))
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync option_value = 1024 * 16;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync option_len = sizeof(option_value);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync setsockopt(g_sock, SOL_SOCKET, SO_RCVBUF, (void *) &option_value,
a180a41bba1d50822df23fff0099e90b86638b89vboxsync option_len);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_in.size = 4096;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_in.data = (uint8 *) xmalloc(g_in.size);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync for (i = 0; i < STREAM_COUNT; i++)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_out[i].size = 4096;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_out[i].data = (uint8 *) xmalloc(g_out[i].size);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync return True;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync}
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync/* Disconnect on the TCP layer */
a180a41bba1d50822df23fff0099e90b86638b89vboxsyncvoid
a180a41bba1d50822df23fff0099e90b86638b89vboxsynctcp_disconnect(void)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync{
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (g_ssl)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (!g_network_error)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync (void) SSL_shutdown(g_ssl);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync SSL_free(g_ssl);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_ssl = NULL;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync SSL_CTX_free(g_ssl_ctx);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_ssl_ctx = NULL;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync TCP_CLOSE(g_sock);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_sock = -1;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync}
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsyncchar *
a180a41bba1d50822df23fff0099e90b86638b89vboxsynctcp_get_address()
a180a41bba1d50822df23fff0099e90b86638b89vboxsync{
a180a41bba1d50822df23fff0099e90b86638b89vboxsync static char ipaddr[32];
a180a41bba1d50822df23fff0099e90b86638b89vboxsync struct sockaddr_in sockaddr;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync socklen_t len = sizeof(sockaddr);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (getsockname(g_sock, (struct sockaddr *) &sockaddr, &len) == 0)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync uint8 *ip = (uint8 *) & sockaddr.sin_addr;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync sprintf(ipaddr, "%d.%d.%d.%d", ip[0], ip[1], ip[2], ip[3]);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync else
a180a41bba1d50822df23fff0099e90b86638b89vboxsync strcpy(ipaddr, "127.0.0.1");
a180a41bba1d50822df23fff0099e90b86638b89vboxsync return ipaddr;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync}
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsyncRD_BOOL
a180a41bba1d50822df23fff0099e90b86638b89vboxsynctcp_is_connected()
a180a41bba1d50822df23fff0099e90b86638b89vboxsync{
a180a41bba1d50822df23fff0099e90b86638b89vboxsync struct sockaddr_in sockaddr;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync socklen_t len = sizeof(sockaddr);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (getpeername(g_sock, (struct sockaddr *) &sockaddr, &len))
a180a41bba1d50822df23fff0099e90b86638b89vboxsync return True;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync return False;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync}
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync/* reset the state of the tcp layer */
a180a41bba1d50822df23fff0099e90b86638b89vboxsync/* Support for Session Directory */
a180a41bba1d50822df23fff0099e90b86638b89vboxsyncvoid
a180a41bba1d50822df23fff0099e90b86638b89vboxsynctcp_reset_state(void)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync{
a180a41bba1d50822df23fff0099e90b86638b89vboxsync int i;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync /* Clear the incoming stream */
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (g_in.data != NULL)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync xfree(g_in.data);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_in.p = NULL;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_in.end = NULL;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_in.data = NULL;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_in.size = 0;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_in.iso_hdr = NULL;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_in.mcs_hdr = NULL;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_in.sec_hdr = NULL;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_in.rdp_hdr = NULL;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_in.channel_hdr = NULL;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsync /* Clear the outgoing stream(s) */
a180a41bba1d50822df23fff0099e90b86638b89vboxsync for (i = 0; i < STREAM_COUNT; i++)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync {
a180a41bba1d50822df23fff0099e90b86638b89vboxsync if (g_out[i].data != NULL)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync xfree(g_out[i].data);
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_out[i].p = NULL;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_out[i].end = NULL;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_out[i].data = NULL;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_out[i].size = 0;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_out[i].iso_hdr = NULL;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_out[i].mcs_hdr = NULL;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_out[i].sec_hdr = NULL;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_out[i].rdp_hdr = NULL;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_out[i].channel_hdr = NULL;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync }
a180a41bba1d50822df23fff0099e90b86638b89vboxsync}
a180a41bba1d50822df23fff0099e90b86638b89vboxsync
a180a41bba1d50822df23fff0099e90b86638b89vboxsyncvoid
a180a41bba1d50822df23fff0099e90b86638b89vboxsynctcp_run_ui(RD_BOOL run)
a180a41bba1d50822df23fff0099e90b86638b89vboxsync{
a180a41bba1d50822df23fff0099e90b86638b89vboxsync g_run_ui = run;
a180a41bba1d50822df23fff0099e90b86638b89vboxsync}