virtualx-engine/drivers/openssl/stream_peer_ssl.cpp

#include "stream_peer_ssl.h"


Error StreamPeerSSL::connect(const String &p_host,int p_port,int p_flags) {

	// Set up a SSL_CTX object, which will tell our BIO object how to do its work
	ctx = SSL_CTX_new(SSLv23_client_method());
	// Create our BIO object for SSL connections.
	BIO* bio = BIO_new_ssl_connect(ctx);
	// Failure?
	if (bio == NULL) {

		// We need to free up the SSL_CTX before we leave.
		ERR_FAIL_COND_V(bio==NULL,ERR_CANT_CREATE);
	}
	// Makes ssl point to bio's SSL object.
	BIO_get_ssl(bio, &ssl);
	// Set the SSL to automatically retry on failure.
	SSL_set_mode(ssl, SSL_MODE_AUTO_RETRY);
	// We're connection to google.com on port 443.
	BIO_set_conn_hostname(bio, (p_host+":"+itos(p_port)).utf8().get_data());

	// Same as before, try to connect.
	if (BIO_do_connect(bio) <= 0) {

		ERR_EXPLAIN("Failed to connect to '"+p_host+"'' port "+itos(p_port));
		BIO_free_all(bio);
		SSL_CTX_free(ctx);
		ERR_FAIL_V(ERR_CANT_CONNECT);
	}

	// Now we need to do the SSL handshake, so we can communicate.
	if (BIO_do_handshake(bio) <= 0) {
		ERR_EXPLAIN("Failed to handshake to '"+p_host+"'' port "+itos(p_port));
		BIO_free_all(bio);
		SSL_CTX_free(ctx);
		ERR_FAIL_V(ERR_CANT_CONNECT);
	}


	// Create a buffer for grabbing information from the page.
	char buf[1024];
	memset(buf, 0, sizeof(buf));
	// Create a buffer for the reqest we'll send to the server
	char send[1024];
	memset(send, 0, sizeof(send));
	// Create our GET request.
	strcat(send, "GET / HTTP/1.1\nHost:google.com\nUser Agent: Mozilla/5.0 (compatible; MSIE 9.0; Windows NT 6.1; Trident/5.0)\nConnection: Close\n\n");
	// BIO_puts sends a null-terminated string to the server. In this case it's our GET request.
	BIO_puts(bio, send);
	// Loop while there's information to be read.
	while (1) {
		// BIO_read() reads data from the server into a buffer. It returns the number of characters read in.
		int x = BIO_read(bio, buf, sizeof(buf) - 1);
		// If we haven't read in anything, assume there's nothing more to be sent since we used Connection: Close.
		if (x == 0) {
			break;
		}
		// If BIO_read() returns a negative number, there was an error
		else if (x < 0) {
			// BIO_should_retry lets us know if we should keep trying to read data or not.
			if (!BIO_should_retry(bio)) {
				printf("\nRead Failed!\n");
				BIO_free_all(bio);
				SSL_CTX_free(ctx);
				return;
			}
		}
		// We actually got some data, without errors!
		else {
			// Null-terminate our buffer, just in case
			buf[x] = 0;
			// Echo what the server sent to the screen
			printf("%s", buf);
		}
	}
	// Free up that BIO object we created.
	BIO_free_all(bio);
	// Remember, we also need to free up that SSL_CTX object!
	SSL_CTX_free(ctx);
	// Return.

}

void StreamPeerSSL::initialize_ssl() {

	CRYPTO_malloc_init(); // Initialize malloc, free, etc for OpenSSL's use
	SSL_library_init(); // Initialize OpenSSL's SSL libraries
	SSL_load_error_strings(); // Load SSL error strings
	ERR_load_BIO_strings(); // Load BIO error strings
	OpenSSL_add_all_algorithms(); // Load all available encryption algorithms
}

void StreamPeerSSL::finalize_ssl(){


}