Software Público Brasileiro

/***************************************************************************
 *   Copyright (C) 2005 by Jeff Ferr                                       *
 *   root@sat                                                              *
 *                                                                         *
 *   This program 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 program 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 program; if not, write to the                         *
 *   Free Software Foundation, Inc.,                                       *
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 ***************************************************************************/
#include "Stdafx.h"
#include "jsslsocketinputstream.h"
#include "jsocketexception.h"

namespace jsocket {

SSLSocketInputStream::SSLSocketInputStream(Connection *conn_, bool *is_closed_, SSL *ssl, int64_t size_):
	jio::InputStream()
{
	jcommon::Object::SetClassName("jsocket::SSLSocketInputStream");
	
	_ssl = ssl;
	_connection = conn_;
	_fd = conn_->GetHandler();
	_is_closed = is_closed_;
	_stream = true;
	_buffer_length = size_;
	_current_index = 0;
	_end_index = 0;
	_receive_bytes = 0;
	_blocked = true;
	
	try {
		_buffer = new char[_buffer_length];
	} catch (std::bad_alloc &e) {
		_buffer = NULL;

		_buffer_length = 0;
		_current_index = 0;
	}
}

SSLSocketInputStream::~SSLSocketInputStream()
{
	if (_buffer != NULL) {
		delete [] _buffer;
	}
}

int64_t SSLSocketInputStream::Read()
{
	if ((*_is_closed) == true) {
		throw SocketException("Connection closed exception");
	}
	
	int d = _end_index - _current_index;

	if (d == 0) {
		int n;

#ifdef _WIN32
		n = ::recv(_fd, _buffer, _buffer_length, flags);
#else
		n = SSL_read(_ssl, _buffer, _buffer_length);
#endif
		
#ifdef _WIN32
		if (n == SOCKET_ERROR || n == 0) {
#else 
		if (n <= 0) {
#endif
			return -1LL;
		}
			
		_current_index = 0;
		_end_index = n;
		_receive_bytes += n;
	}
	
	uint8_t c;
	
	c = _buffer[_current_index];
	
	if (++_current_index >= _end_index) {
		_current_index = _end_index = 0;
	}
	
	return (int64_t)c;
}

int64_t SSLSocketInputStream::Read(char *data_, int64_t data_length_)
{
	if ((*_is_closed) == true) {
		throw SocketException("Connection closed exception");
	}
	
	// retorna no maximo o tamanho do buffer em bytes
	
	int64_t d, 
		 r;
	
	d = _end_index - _current_index;

	if (d == 0LL) {
		int n;

#ifdef _WIN32
		n = ::recv(_fd, _buffer, _buffer_length, flags);
#else 
		n = SSL_read(_ssl, data_, _buffer_length);
#endif
		
#ifdef _WIN32
		if (n == SOCKET_ERROR) {
#else 
		if (n <= 0) {
#endif
			return -1LL;
		}
			
		_current_index = 0;
		_end_index = n;
		_receive_bytes += n;

		d = n;
	}
	
	if (data_length_ <= d) {
		memcpy(data_, (_buffer + _current_index), data_length_);
		_current_index += data_length_;
		r = data_length_;
	} else {
		memcpy(data_, (_buffer + _current_index), d);
		_current_index += d;
		r = d;
	}
	
	if (_current_index >= _end_index) {
		_current_index = _end_index = 0;
	}

	return (int64_t)r;
}

bool SSLSocketInputStream::IsEmpty()
{
	return (_current_index == _end_index);
}

int64_t SSLSocketInputStream::Available()
{
	return (int64_t)_current_index;
}

int64_t SSLSocketInputStream::GetReadedBytes()
{
	return (int64_t)_receive_bytes;
}

void SSLSocketInputStream::Close()
{
	_connection->Close();
}

void SSLSocketInputStream::Reset()
{
	_current_index = _end_index = 0;
}

int64_t SSLSocketInputStream::GetSize()
{
	return 0LL;
}

int64_t SSLSocketInputStream::GetPosition()
{
	return 0LL;
}

void SSLSocketInputStream::Skip(int64_t skip)
{
	// DO:: nothing
}

}