make safe http_schemes.c

nanomsg · Nov 20, 2023 · 3275eda · 3275eda
1 parent a54820f
commit 3275eda
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/supplemental/http/http_schemes.c b/src/supplemental/http/http_schemes.c
@@ -76,10 +76,12 @@ static struct {
 const char *
 nni_http_stream_scheme(const char *upper)
 {
-	for (int i = 0; http_schemes[i].upper != NULL; i++) {
-		if (strcmp(http_schemes[i].upper, upper) == 0) {
-			return (http_schemes[i].lower);
-		}
+	if (upper != NULL) {
+	  for (int i = 0; http_schemes[i].upper != NULL; i++) {
+	    if (strncmp(http_schemes[i].upper, upper, 7) == 0) {
+	      return (http_schemes[i].lower);
+	    }
+	  }
 	}
 	return (NULL);
 }