mirror of
https://github.com/kasmtech/KasmVNC.git
synced 2024-11-22 16:13:13 +01:00
b7dc4a16fe
There might be more bytes left in the current TLS record, even if there is nothing on the underlying stream. Make sure we properly return this when we aren't being requested to block.
130 lines
3.0 KiB
C++
130 lines
3.0 KiB
C++
/* Copyright (C) 2002-2005 RealVNC Ltd. All Rights Reserved.
|
|
* Copyright (C) 2005 Martin Koegler
|
|
* Copyright (C) 2010 TigerVNC Team
|
|
*
|
|
* This is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This software is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this software; if not, write to the Free Software
|
|
* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307,
|
|
* USA.
|
|
*/
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
#include <config.h>
|
|
#endif
|
|
|
|
#include <rdr/Exception.h>
|
|
#include <rdr/TLSException.h>
|
|
#include <rdr/TLSInStream.h>
|
|
#include <errno.h>
|
|
|
|
#ifdef HAVE_GNUTLS
|
|
using namespace rdr;
|
|
|
|
enum { DEFAULT_BUF_SIZE = 16384 };
|
|
|
|
ssize_t TLSInStream::pull(gnutls_transport_ptr_t str, void* data, size_t size)
|
|
{
|
|
TLSInStream* self= (TLSInStream*) str;
|
|
InStream *in = self->in;
|
|
|
|
try {
|
|
if (!in->check(1, 1, false)) {
|
|
gnutls_transport_set_errno(self->session, EAGAIN);
|
|
return -1;
|
|
}
|
|
|
|
if ((size_t)(in->getend() - in->getptr()) < size)
|
|
size = in->getend() - in->getptr();
|
|
|
|
in->readBytes(data, size);
|
|
|
|
} catch (Exception& e) {
|
|
gnutls_transport_set_errno(self->session, EINVAL);
|
|
return -1;
|
|
}
|
|
|
|
return size;
|
|
}
|
|
|
|
TLSInStream::TLSInStream(InStream* _in, gnutls_session_t _session)
|
|
: session(_session), in(_in), bufSize(DEFAULT_BUF_SIZE), offset(0)
|
|
{
|
|
gnutls_transport_ptr_t recv, send;
|
|
|
|
ptr = end = start = new U8[bufSize];
|
|
|
|
gnutls_transport_set_pull_function(session, pull);
|
|
gnutls_transport_get_ptr2(session, &recv, &send);
|
|
gnutls_transport_set_ptr2(session, this, send);
|
|
}
|
|
|
|
TLSInStream::~TLSInStream()
|
|
{
|
|
gnutls_transport_set_pull_function(session, NULL);
|
|
|
|
delete[] start;
|
|
}
|
|
|
|
size_t TLSInStream::pos()
|
|
{
|
|
return offset + ptr - start;
|
|
}
|
|
|
|
size_t TLSInStream::overrun(size_t itemSize, size_t nItems, bool wait)
|
|
{
|
|
if (itemSize > bufSize)
|
|
throw Exception("TLSInStream overrun: max itemSize exceeded");
|
|
|
|
if (end - ptr != 0)
|
|
memmove(start, ptr, end - ptr);
|
|
|
|
offset += ptr - start;
|
|
end -= ptr - start;
|
|
ptr = start;
|
|
|
|
while ((size_t)(end - start) < itemSize) {
|
|
size_t n = readTLS((U8*) end, start + bufSize - end, wait);
|
|
if (!wait && n == 0)
|
|
return 0;
|
|
end += n;
|
|
}
|
|
|
|
size_t nAvail;
|
|
nAvail = (end - ptr) / itemSize;
|
|
if (nAvail < nItems)
|
|
return nAvail;
|
|
|
|
return nItems;
|
|
}
|
|
|
|
size_t TLSInStream::readTLS(U8* buf, size_t len, bool wait)
|
|
{
|
|
int n;
|
|
|
|
if (gnutls_record_check_pending(session) == 0) {
|
|
n = in->check(1, 1, wait);
|
|
if (n == 0)
|
|
return 0;
|
|
}
|
|
|
|
n = gnutls_record_recv(session, (void *) buf, len);
|
|
if (n == GNUTLS_E_INTERRUPTED || n == GNUTLS_E_AGAIN)
|
|
return 0;
|
|
|
|
if (n < 0) throw TLSException("readTLS", n);
|
|
|
|
return n;
|
|
}
|
|
|
|
#endif
|