src/connection.c: Move code to get proxy certificate path to verify.c.

[tlsproxy/tlsproxy.git] / src / verify.c

diff --git a/src/verify.c b/src/verify.c
index 63ca6da2d982a121d36cb8b6779e795ba0c72f30..e3a86294755c4283bc3c899a03b72501659b61f3 100644 (file)
--- a/src/verify.c
+++ b/src/verify.c
@@ -155,6 +155,38 @@ int verify_tls_connection(gnutls_session_t session, const char *hostname) {
     return 0;
 }
 
+
+int proxy_certificate_path(const char *hostname, char *path, size_t size) {
+    int result;
+
+    /* Hostname too long. */
+    if (size - strlen(PROXY_SERVER_CERT_FORMAT) <= strlen(hostname)) {
+        LOG(LOG_WARNING,
+            "proxy_certificate_path(): hostname too long: '%s'",
+            hostname);
+        return -1;
+    }
+    /* Try to prevent path traversals in hostnames. */
+    if (NULL != strstr(hostname, "..")) {
+        LOG(LOG_WARNING,
+            "proxy_certificate_path(): possible path traversal: '%s'",
+            hostname);
+        return -1;
+    }
+    result = snprintf(path, size, PROXY_SERVER_CERT_FORMAT, hostname);
+    if (result < 0) {
+        LOG_PERROR(LOG_ERROR,
+                   "proxy_certificate_path(): snprintf failed");
+        return -1;
+    } else if ((size_t)result >= size) {
+        LOG(LOG_ERROR,
+            "proxy_certificate_path(): snprintf buffer too short");
+        return -1;
+    }
+
+    return 0;
+}
+
 int server_certificate_path(FILE **file, const char *hostname,
                             char *path, size_t size) {
     int result;