text: /**
   * WinPR: Windows Portable Runtime
   * Schannel Security Package (OpenSSL)
   *
   * Copyright 2012-2014 Marc-Andre Moreau <marcandre.moreau@gmail.com>
   *
   * 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.
   */
  
  #ifdef HAVE_CONFIG_H
  #include "config.h"
  #endif
  
  #include "schannel_openssl.h"
  
  #ifdef WITH_OPENSSL
  
  #include <winpr/crt.h>
  #include <winpr/sspi.h>
  #include <winpr/ssl.h>
  #include <winpr/print.h>
  
  #include <openssl/ssl.h>
  #include <openssl/err.h>
  #include <openssl/bio.h>
  
  struct _SCHANNEL_OPENSSL
  {
  	SSL* ssl;
  	SSL_CTX* ctx;
  	BOOL connected;
  	BIO* bioRead;
  	BIO* bioWrite;
  	BYTE* ReadBuffer;
  	BYTE* WriteBuffer;
  };
  
  #include "../../log.h"
  #define TAG WINPR_TAG("sspi.schannel")
  
  char* openssl_get_ssl_error_string(int ssl_error)
  {
  	switch (ssl_error)
  	{
  		case SSL_ERROR_ZERO_RETURN:
  			return "SSL_ERROR_ZERO_RETURN";
  
  		case SSL_ERROR_WANT_READ:
  			return "SSL_ERROR_WANT_READ";
  
  		case SSL_ERROR_WANT_WRITE:
  			return "SSL_ERROR_WANT_WRITE";
  
  		case SSL_ERROR_SYSCALL:
  			return "SSL_ERROR_SYSCALL";
  
  		case SSL_ERROR_SSL:
  			return "SSL_ERROR_SSL";
  	}
  
  	return "SSL_ERROR_UNKNOWN";
  }
  
  int schannel_openssl_client_init(SCHANNEL_OPENSSL* context)
  {
  	int status;
  	long options = 0;
  	context->ctx = SSL_CTX_new(TLSv1_client_method());
  
  	if (!context->ctx)
  	{
  		WLog_ERR(TAG, "SSL_CTX_new failed");
  		return -1;
  	}
  
  	/**
  	 * SSL_OP_NO_COMPRESSION:
  	 *
  	 * The Microsoft RDP server does not advertise support
  	 * for TLS compression, but alternative servers may support it.
  	 * This was observed between early versions of the FreeRDP server
  	 * and the FreeRDP client, and caused major performance issues,
  	 * which is why we're disabling it.
  	 */
  #ifdef SSL_OP_NO_COMPRESSION
  	options |= SSL_OP_NO_COMPRESSION;
  #endif
  	/**
  	 * SSL_OP_TLS_BLOCK_PADDING_BUG:
  	 *
  	 * The Microsoft RDP server does *not* support TLS padding.
  	 * It absolutely needs to be disabled otherwise it won't work.
  	 */
  	options |= SSL_OP_TLS_BLOCK_PADDING_BUG;
  	/**
  	 * SSL_OP_DONT_INSERT_EMPTY_FRAGMENTS:
  	 *
  	 * Just like TLS padding, the Microsoft RDP server does not
  	 * support empty fragments. This needs to be disabled.
  	 */
  	options |= SSL_OP_DONT_INSERT_EMPTY_FRAGMENTS;
  	SSL_CTX_set_options(context->ctx, options);
  	context->ssl = SSL_new(context->ctx);
  
  	if (!context->ssl)
  	{
  		WLog_ERR(TAG, "SSL_new failed");
  		goto out_ssl_new_failed;
  	}
  
  	context->bioRead = BIO_new(BIO_s_mem());
  
  	if (!context->bioRead)
  	{
  		WLog_ERR(TAG, "BIO_new failed");
  		goto out_bio_read_failed;
  		return -1;
  	}
  
  	status = BIO_set_write_buf_size(context->bioRead, SCHANNEL_CB_MAX_TOKEN);
  	if (status != 1)
  	{
  		WLog_ERR(TAG, "BIO_set_write_buf_size on bioRead failed");
  		goto out_set_write_buf_read;
  	}
  
  	context->bioWrite = BIO_new(BIO_s_mem());
  
  	if (!context->bioWrite)
  	{
  		WLog_ERR(TAG, "BIO_new failed");
  		goto out_bio_write_failed;
  	}
  
  	status = BIO_set_write_buf_size(context->bioWrite, SCHANNEL_CB_MAX_TOKEN);
  	if (status != 1)
  	{
  		WLog_ERR(TAG, "BIO_set_write_buf_size on bioWrite failed");
  		goto out_set_write_buf_write;
  	}
  	status = BIO_make_bio_pair(context->bioRead, context->bioWrite);
  	if (status != 1)
  	{
  		WLog_ERR(TAG, "BIO_make_bio_pair failed");
  		goto out_bio_pair;
  	}
  	SSL_set_bio(context->ssl, context->bioRead, context->bioWrite);
  	context->ReadBuffer = (BYTE*) malloc(SCHANNEL_CB_MAX_TOKEN);
  	if (!context->ReadBuffer)
  	{
  		WLog_ERR(TAG, "Failed to allocate ReadBuffer");
  		goto out_read_alloc;
  	}
  	context->WriteBuffer = (BYTE*) malloc(SCHANNEL_CB_MAX_TOKEN);
  	if (!context->WriteBuffer)
  	{
  		WLog_ERR(TAG, "Failed to allocate ReadBuffer");
  		goto out_write_alloc;
  	}
  	return 0;
  
  out_write_alloc:
  	free(context->ReadBuffer);
  out_read_alloc:
  out_bio_pair:
  out_set_write_buf_write:
  	BIO_free(context->bioWrite);
  out_bio_write_failed:
  out_set_write_buf_read:
  	BIO_free(context->bioRead);
  out_bio_read_failed:
  	SSL_free(context->ssl);
  out_ssl_new_failed:
  	SSL_CTX_free(context->ctx);
  	return -1;
  }
  
  int schannel_openssl_server_init(SCHANNEL_OPENSSL* context)
  {
  	int status;
  	long options = 0;
  
  	context->ctx = SSL_CTX_new(TLSv1_server_method());
  
  	if (!context->ctx)
  	{
  		WLog_ERR(TAG, "SSL_CTX_new failed");
  		return -1;
  	}
  
  	/*
  	 * SSL_OP_NO_SSLv2:
  	 *
  	 * We only want SSLv3 and TLSv1, so disable SSLv2.
  	 * SSLv3 is used by, eg. Microsoft RDC for Mac OS X.
  	 */
  	options |= SSL_OP_NO_SSLv2;
  	/**
  	 * SSL_OP_NO_COMPRESSION:
  	 *
  	 * The Microsoft RDP server does not advertise support
  	 * for TLS compression, but alternative servers may support it.
  	 * This was observed between early versions of the FreeRDP server
  	 * and the FreeRDP client, and caused major performance issues,
  	 * which is why we're disabling it.
  	 */
  #ifdef SSL_OP_NO_COMPRESSION
  	options |= SSL_OP_NO_COMPRESSION;
  #endif
  	/**
  	 * SSL_OP_TLS_BLOCK_PADDING_BUG:
  	 *
  	 * The Microsoft RDP server does *not* support TLS padding.
  	 * It absolutely needs to be disabled otherwise it won't work.
  	 */
  	options |= SSL_OP_TLS_BLOCK_PADDING_BUG;
  	/**
  	 * SSL_OP_DONT_INSERT_EMPTY_FRAGMENTS:
  	 *
  	 * Just like TLS padding, the Microsoft RDP server does not
  	 * support empty fragments. This needs to be disabled.
  	 */
  	options |= SSL_OP_DONT_INSERT_EMPTY_FRAGMENTS;
  	SSL_CTX_set_options(context->ctx, options);
  
  	if (SSL_CTX_use_RSAPrivateKey_file(context->ctx, "/tmp/localhost.key", SSL_FILETYPE_PEM) <= 0)
  	{
  		WLog_ERR(TAG, "SSL_CTX_use_RSAPrivateKey_file failed");
  		goto out_rsa_key;
  	}
  
  	context->ssl = SSL_new(context->ctx);
  
  	if (!context->ssl)
  	{
  		WLog_ERR(TAG, "SSL_new failed");
  		goto out_ssl_new;
  	}
  
  	if (SSL_use_certificate_file(context->ssl, "/tmp/localhost.crt", SSL_FILETYPE_PEM) <= 0)
  	{
  		WLog_ERR(TAG, "SSL_use_certificate_file failed");
  		goto out_use_certificate;
  	}
  
  	context->bioRead = BIO_new(BIO_s_mem());
  
  	if (!context->bioRead)
  	{
  		WLog_ERR(TAG, "BIO_new failed");
  		goto out_bio_read;
  	}
  
  	status = BIO_set_write_buf_size(context->bioRead, SCHANNEL_CB_MAX_TOKEN);
  	if (status != 1)
  	{
  		WLog_ERR(TAG, "BIO_set_write_buf_size failed for bioRead");
  		goto out_set_write_buf_read;
  	}
  	context->bioWrite = BIO_new(BIO_s_mem());
  
  	if (!context->bioWrite)
  	{
  		WLog_ERR(TAG, "BIO_new failed");
  		goto out_bio_write;
  	}
  
  	status = BIO_set_write_buf_size(context->bioWrite, SCHANNEL_CB_MAX_TOKEN);
  	if (status != 1)
  	{
  		WLog_ERR(TAG, "BIO_set_write_buf_size failed for bioWrite");
  		goto out_set_write_buf_write;
  	}
  	status = BIO_make_bio_pair(context->bioRead, context->bioWrite);
  	if (status != 1)
  	{
  		WLog_ERR(TAG, "BIO_make_bio_pair failed");
  		goto out_bio_pair;
  	}
  	SSL_set_bio(context->ssl, context->bioRead, context->bioWrite);
  	context->ReadBuffer = (BYTE*) malloc(SCHANNEL_CB_MAX_TOKEN);
  	if (!context->ReadBuffer)
  	{
  		WLog_ERR(TAG, "Failed to allocate memory for ReadBuffer");
  		goto out_read_buffer;
  	}
  	context->WriteBuffer = (BYTE*) malloc(SCHANNEL_CB_MAX_TOKEN);
  	if (!context->WriteBuffer)
  	{
  		WLog_ERR(TAG, "Failed to allocate memory for WriteBuffer");
  		goto out_write_buffer;
  	}
  	return 0;
  
  out_write_buffer:
  	free(context->ReadBuffer);
  out_read_buffer:
  out_bio_pair:
  out_set_write_buf_write:
  	BIO_free(context->bioWrite);
  out_bio_write:
  out_set_write_buf_read:
  	BIO_free(context->bioRead);
  out_bio_read:
  out_use_certificate:
  	SSL_free(context->ssl);
  out_ssl_new:
  out_rsa_key:
  	SSL_CTX_free(context->ctx);
  	return -1;
  }
  
  SECURITY_STATUS schannel_openssl_client_process_tokens(SCHANNEL_OPENSSL* context, PSecBufferDesc pInput, PSecBufferDesc pOutput)
  {
  	int status;
  	int ssl_error;
  	PSecBuffer pBuffer;
  
  	if (!context->connected)
  	{
  		if (pInput)
  		{
  			if (pInput->cBuffers < 1)
  				return SEC_E_INVALID_TOKEN;
  
  			pBuffer = sspi_FindSecBuffer(pInput, SECBUFFER_TOKEN);
  
  			if (!pBuffer)
  				return SEC_E_INVALID_TOKEN;
  
  			status = BIO_write(context->bioRead, pBuffer->pvBuffer, pBuffer->cbBuffer);
  		}
  
  		status = SSL_connect(context->ssl);
  
  		if (status < 0)
  		{
  			ssl_error = SSL_get_error(context->ssl, status);
  			WLog_ERR(TAG, "SSL_connect error: %s", openssl_get_ssl_error_string(ssl_error));
  		}
  
  		if (status == 1)
  			context->connected = TRUE;
  
  		status = BIO_read(context->bioWrite, context->ReadBuffer, SCHANNEL_CB_MAX_TOKEN);
  
  		if (pOutput->cBuffers < 1)
  			return SEC_E_INVALID_TOKEN;
  
  		pBuffer = sspi_FindSecBuffer(pOutput, SECBUFFER_TOKEN);
  
  		if (!pBuffer)
  			return SEC_E_INVALID_TOKEN;
  
  		if (status > 0)
  		{
  			if (pBuffer->cbBuffer < (unsigned long) status)
  				return SEC_E_INSUFFICIENT_MEMORY;
  
  			CopyMemory(pBuffer->pvBuffer, context->ReadBuffer, status);
  			pBuffer->cbBuffer = status;
  			return (context->connected) ? SEC_E_OK : SEC_I_CONTINUE_NEEDED;
  		}
  		else
  		{
  			pBuffer->cbBuffer = 0;
  			return (context->connected) ? SEC_E_OK : SEC_I_CONTINUE_NEEDED;
  		}
  	}
  
  	return SEC_E_OK;
  }
  
  SECURITY_STATUS schannel_openssl_server_process_tokens(SCHANNEL_OPENSSL* context, PSecBufferDesc pInput, PSecBufferDesc pOutput)
  {
  	int status;
  	int ssl_error;
  	PSecBuffer pBuffer;
  
  	if (!context->connected)
  	{
  		if (pInput->cBuffers < 1)
  			return SEC_E_INVALID_TOKEN;
  
  		pBuffer = sspi_FindSecBuffer(pInput, SECBUFFER_TOKEN);
  
  		if (!pBuffer)
  			return SEC_E_INVALID_TOKEN;
  
  		status = BIO_write(context->bioRead, pBuffer->pvBuffer, pBuffer->cbBuffer);
  		status = SSL_accept(context->ssl);
  
  		if (status < 0)
  		{
  			ssl_error = SSL_get_error(context->ssl, status);
  			WLog_ERR(TAG, "SSL_accept error: %s", openssl_get_ssl_error_string(ssl_error));
  		}
  
  		if (status == 1)
  			context->connected = TRUE;
  
  		status = BIO_read(context->bioWrite, context->ReadBuffer, SCHANNEL_CB_MAX_TOKEN);
  
  		if (pOutput->cBuffers < 1)
  			return SEC_E_INVALID_TOKEN;
  
  		pBuffer = sspi_FindSecBuffer(pOutput, SECBUFFER_TOKEN);
  
  		if (!pBuffer)
  			return SEC_E_INVALID_TOKEN;
  
  		if (status > 0)
  		{
  			if (pBuffer->cbBuffer < (unsigned long) status)
  				return SEC_E_INSUFFICIENT_MEMORY;
  
  			CopyMemory(pBuffer->pvBuffer, context->ReadBuffer, status);
  			pBuffer->cbBuffer = status;
  			return (context->connected) ? SEC_E_OK : SEC_I_CONTINUE_NEEDED;
  		}
  		else
  		{
  			pBuffer->cbBuffer = 0;
  			return (context->connected) ? SEC_E_OK : SEC_I_CONTINUE_NEEDED;
  		}
  	}
  
  	return SEC_E_OK;
  }
  
  SECURITY_STATUS schannel_openssl_encrypt_message(SCHANNEL_OPENSSL* context, PSecBufferDesc pMessage)
  {
  	int status;
  	int length;
  	int offset;
  	int ssl_error;
  	PSecBuffer pStreamBodyBuffer;
  	PSecBuffer pStreamHeaderBuffer;
  	PSecBuffer pStreamTrailerBuffer;
  	pStreamHeaderBuffer = sspi_FindSecBuffer(pMessage, SECBUFFER_STREAM_HEADER);
  	pStreamBodyBuffer = sspi_FindSecBuffer(pMessage, SECBUFFER_DATA);
  	pStreamTrailerBuffer = sspi_FindSecBuffer(pMessage, SECBUFFER_STREAM_TRAILER);
  
  	if ((!pStreamHeaderBuffer) || (!pStreamBodyBuffer) || (!pStreamTrailerBuffer))
  		return SEC_E_INVALID_TOKEN;
  
  	status = SSL_write(context->ssl, pStreamBodyBuffer->pvBuffer, pStreamBodyBuffer->cbBuffer);
  
  	if (status < 0)
  	{
  		ssl_error = SSL_get_error(context->ssl, status);
  		WLog_ERR(TAG, "SSL_write: %s", openssl_get_ssl_error_string(ssl_error));
  	}
  
  	status = BIO_read(context->bioWrite, context->ReadBuffer, SCHANNEL_CB_MAX_TOKEN);
  
  	if (status > 0)
  	{
  		offset = 0;
  		length = (pStreamHeaderBuffer->cbBuffer > (unsigned long) status) ? status : pStreamHeaderBuffer->cbBuffer;
  		CopyMemory(pStreamHeaderBuffer->pvBuffer, &context->ReadBuffer[offset], length);
  		status -= length;
  		offset += length;
  		length = (pStreamBodyBuffer->cbBuffer > (unsigned long) status) ? status : pStreamBodyBuffer->cbBuffer;
  		CopyMemory(pStreamBodyBuffer->pvBuffer, &context->ReadBuffer[offset], length);
  		status -= length;
  		offset += length;
  		length = (pStreamTrailerBuffer->cbBuffer > (unsigned long) status) ? status : pStreamTrailerBuffer->cbBuffer;
  		CopyMemory(pStreamTrailerBuffer->pvBuffer, &context->ReadBuffer[offset], length);
  		status -= length;
  	}
  
  	return SEC_E_OK;
  }
  
  SECURITY_STATUS schannel_openssl_decrypt_message(SCHANNEL_OPENSSL* context, PSecBufferDesc pMessage)
  {
  	int status;
  	int length;
  	BYTE* buffer;
  	int ssl_error;
  	PSecBuffer pBuffer;
  	pBuffer = sspi_FindSecBuffer(pMessage, SECBUFFER_DATA);
  
  	if (!pBuffer)
  		return SEC_E_INVALID_TOKEN;
  
  	status = BIO_write(context->bioRead, pBuffer->pvBuffer, pBuffer->cbBuffer);
  	status = SSL_read(context->ssl, pBuffer->pvBuffer, pBuffer->cbBuffer);
  
  	if (status < 0)
  	{
  		ssl_error = SSL_get_error(context->ssl, status);
  		WLog_ERR(TAG, "SSL_read: %s", openssl_get_ssl_error_string(ssl_error));
  	}
  
  	length = status;
  	buffer = pBuffer->pvBuffer;
  	pMessage->pBuffers[0].BufferType = SECBUFFER_STREAM_HEADER;
  	pMessage->pBuffers[0].cbBuffer = 5;
  	pMessage->pBuffers[1].BufferType = SECBUFFER_DATA;
  	pMessage->pBuffers[1].pvBuffer = buffer;
  	pMessage->pBuffers[1].cbBuffer = length;
  	pMessage->pBuffers[2].BufferType = SECBUFFER_STREAM_TRAILER;
  	pMessage->pBuffers[2].cbBuffer = 36;
  	pMessage->pBuffers[3].BufferType = SECBUFFER_EMPTY;
  	pMessage->pBuffers[3].cbBuffer = 0;
  	return SEC_E_OK;
  }
  
  SCHANNEL_OPENSSL* schannel_openssl_new()
  {
  	SCHANNEL_OPENSSL* context;
  	context = (SCHANNEL_OPENSSL*) calloc(1, sizeof(SCHANNEL_OPENSSL));
  
  	if (context != NULL)
  	{
  		winpr_InitializeSSL(WINPR_SSL_INIT_DEFAULT);
  		context->connected = FALSE;
  	}
  
  	return context;
  }
  
  void schannel_openssl_free(SCHANNEL_OPENSSL* context)
  {
  	if (context)
  	{
  		free(context->ReadBuffer);
  		free(context->WriteBuffer);
  		free(context);
  	}
  }
  
  #else
  
  int schannel_openssl_client_init(SCHANNEL_OPENSSL* context)
  {
  	return 0;
  }
  
  int schannel_openssl_server_init(SCHANNEL_OPENSSL* context)
  {
  	return 0;
  }
  
  SECURITY_STATUS schannel_openssl_client_process_tokens(SCHANNEL_OPENSSL* context, PSecBufferDesc pInput, PSecBufferDesc pOutput)
  {
  	return SEC_E_OK;
  }
  
  SECURITY_STATUS schannel_openssl_server_process_tokens(SCHANNEL_OPENSSL* context, PSecBufferDesc pInput, PSecBufferDesc pOutput)
  {
  	return SEC_E_OK;
  }
  
  SECURITY_STATUS schannel_openssl_encrypt_message(SCHANNEL_OPENSSL* context, PSecBufferDesc pMessage)
  {
  	return SEC_E_OK;
  }
  
  SECURITY_STATUS schannel_openssl_decrypt_message(SCHANNEL_OPENSSL* context, PSecBufferDesc pMessage)
  {
  	return SEC_E_OK;
  }
  
  SCHANNEL_OPENSSL* schannel_openssl_new(void)
  {
  	return NULL;
  }
  
  void schannel_openssl_free(SCHANNEL_OPENSSL* context)
  {
  
  }
  
  #endif
  