Fixed receive_request for HTTPS connections and enabled HTTPS through the config fileHEADmaster

2 files changed, 11 insertions, 4 deletions

diff --git a/src/net/net.c b/src/net/net.c
index 39b25e3..3d738ae 100755
--- a/src/net/net.c
+++ b/src/net/net.c
@@ -98,7 +98,6 @@ int setup_https(http_server *hs, str certfile, str keyfile){
 		log_error("Missing private key file");
 		return 1;
 	}
-
 	if(hs->ssl != NULL){
 		SSL_free(hs->ssl);
 	}
@@ -192,18 +191,20 @@ static inline int server_read(http_server *hs, str *buf){
 }
 
 int receive_request(http_server *hs, str *request){
-	// SSL_has_pending can return 0 if you havent read any bytes yet (https://stackoverflow.com/questions/6616976/why-does-this-ssl-pending-call-always-return-zero)
 	struct pollfd pfd[1] = { {.fd = hs->csocket, .events = POLLIN } };
 	while(poll(pfd, 1, 100)){
 		if(pfd[0].revents & POLLIN){
 			int rb = 0;
 			if(hs->secure){
-				if(SSL_has_pending(hs->ssl)){
+				// SSL_has_pending can return 0 if you havent read any bytes yet
+				// https://stackoverflow.com/questions/6616976/why-does-this-ssl-pending-call-always-return-zero
+				// so we must perform a read first to advance the state machine
+				do{
 					rb = server_read(hs, request);
 					if(rb == 0){
 						return pleasesslgivemetheerror(SSL_get_error(hs->ssl, rb));
 					}
-				}
+				}while(SSL_has_pending(hs->ssl));
 			}else{
 				rb = server_read(hs, request);
 				if(rb == 0){
diff --git a/src/worker.c b/src/worker.c
index 0c29ca7..8312a74 100755
--- a/src/worker.c
+++ b/src/worker.c
@@ -115,6 +115,12 @@ int init(str name){
 		log_error("Error setting up worker server");
 		return 1;
 	}
+	if(conf.secure){
+		if(setup_https(server, conf.cert, conf.key)){
+			log_error("Error setting up HTTPS in the server");
+			return 1;
+		}
+	}
 	struct sigaction rnit = { .sa_sigaction = reinit, .sa_flags = SA_SIGINFO };
 	if(sigaction(SIGUSR1, &rnit, NULL) == -1){
 		log_error("Error setting up SIGUSR1 signal handler: %s", strerror(errno));