From cbcf4ec9b60ca8d7a684d3a4b4c1ec6b11d500de Mon Sep 17 00:00:00 2001
From: Omar Polo <op@omarpolo.com>
Date: Sat, 6 Mar 2021 20:31:39 +0000
Subject: [PATCH] fix redirect when vhost support is enabled

---
 main.c | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/main.c b/main.c
index 351f7ad..08661a9 100644
--- a/main.c
+++ b/main.c
@@ -26,6 +26,7 @@
  */
 #define GEMINI_REQUEST_MAX 1025
 
+int virtualhost;
 
 void        autoindex(const char *);
 void        cgi(const char *cgicmd);
@@ -160,7 +161,9 @@ display_file(const char *uri)
 	if (S_ISDIR(sb.st_mode) != 0) {
 		if (fp[strlen(fp) -1 ] != '/') {
 			/* no ending "/", redirect to "path/" */
-			estrlcpy(tmp, uri,  sizeof(tmp));
+			if (virtualhost)
+				estrlcat(tmp, "gemini://", sizeof(tmp));
+			estrlcat(tmp, uri,  sizeof(tmp));
 			estrlcat(tmp, "/", sizeof(tmp));
 			status_redirect(31, tmp);
 			return;
@@ -331,7 +334,6 @@ main(int argc, char **argv)
 	char 		uri      [PATH_MAX]           = {'\0'};
 	char 		user     [_SC_LOGIN_NAME_MAX] = "";
 	char        query[PATH_MAX]               = {'\0'};
-	int 		virtualhost = 0;
 	int 		option = 0;
 	char        *pos = NULL;