From edcec9fa9086599d56bffe7fdb2c1f05ae7a3ad8 Mon Sep 17 00:00:00 2001
From: cbdev <cb@cbcdn.com>
Date: Sun, 2 Jun 2019 15:45:12 +0200
Subject: File backend peer query

---
 plugins/backend_file.c | 111 ++++++++++++++++++++++++++++++++++++++++++++-----
 plugins/makefile       |   2 +-
 2 files changed, 102 insertions(+), 11 deletions(-)

(limited to 'plugins')

diff --git a/plugins/backend_file.c b/plugins/backend_file.c
index c3ed8ea..a858ef4 100644
--- a/plugins/backend_file.c
+++ b/plugins/backend_file.c
@@ -84,7 +84,11 @@ static int expression_resolve(char* template, size_t length, char* endpoint, siz
 					return 1;
 				}
 
-				//TODO rtrim slash
+				//rtrim slash so test and test/ are the same file
+				if(strlen(endpoint) >= 2 && endpoint[strlen(endpoint) - 1] == '/'){
+					endpoint[strlen(endpoint) - 1] = 0;
+				}
+
 				//replace with sanitized endpoint string
 				for(p = 0; endpoint[p]; p++){
 					if(endpoint[p] == '/'){
@@ -106,7 +110,45 @@ static int expression_resolve(char* template, size_t length, char* endpoint, siz
 					return 1;
 				}
 
-				//TODO iterate cookies
+				//find the cookie header
+				for(p = 0; p < headers; p++){
+					if(!strcmp(header[p].tag, "Cookie")){
+						value = header[p].value;
+						break;
+					}
+				}
+
+				//no cookie header, fail the expression
+				if(p == headers){
+					return 1;
+				}
+
+				do{
+					//ensure min length of cookie name
+					if(strlen(value) < index_len + 1){
+						return 1;
+					}
+					//check if cookie found
+					if(value[index_len] == '='
+							&& !strncmp(value, index, index_len)){
+						value += index_len + 1;
+						for(value_len = 0; value[value_len] && value[value_len] != ';'; value_len++){
+						}
+						break;
+					}
+					//skip to next cookie
+					for(; *value && strncmp(value, "; ", 2); value++){
+					}
+					if(*value){
+						value += 2;
+					}
+				} while(*value);
+
+				if(!*value){
+					return 1;
+				}
+
+				variable_len = 8 + index_len + 1;
 			}
 			else if(!strncmp(template + u, "%header:", 8)){
 				//scan headers
@@ -140,9 +182,10 @@ static int expression_resolve(char* template, size_t length, char* endpoint, siz
 				fprintf(stderr, "Expression replacement failed\n");
 				return 1;
 			}
-
-			//skip the inserted value
-			u += value_len - 1;
+			if(value){
+				//skip the inserted value
+				u += value_len - 1;
+			}
 		}
 	}
 
@@ -150,13 +193,14 @@ static int expression_resolve(char* template, size_t length, char* endpoint, siz
 }
 
 ws_peer_info query(char* endpoint, size_t protocols, char** protocol, size_t headers, ws_http_header* header, websocket* ws){
-	size_t u, line_alloc = 0;
+	size_t u, p, line_alloc = 0;
 	ssize_t line_length = 0;
-	char* line = NULL;
+	char* line = NULL, *components[3];
 	FILE* input = NULL;
 	char target_path[BACKEND_FILE_MAX_PATH];
 	ws_peer_info peer = {
 		.transport = peer_transport_detect,
+		.protocol = protocols
 	};
 
 	for(u = 0; u < expressions; u++){
@@ -165,7 +209,7 @@ ws_peer_info query(char* endpoint, size_t protocols, char** protocol, size_t hea
 		if(expression_resolve(target_path + strlen(backend_path) + 1, sizeof(target_path) - strlen(backend_path) - 1, endpoint, headers, header)){
 			continue;
 		}
-	
+
 		//check whether the file exists
 		input = fopen(target_path, "r");
 		if(!input){
@@ -174,8 +218,55 @@ ws_peer_info query(char* endpoint, size_t protocols, char** protocol, size_t hea
 
 		//read it
 		for(line_length = getline(&line, &line_alloc, input); line_length >= 0; line_length = getline(&line, &line_alloc, input)){
-			//TODO evaluate line in file
-			fprintf(stderr, "File %s, line %s\n", target_path, line);
+			memset(components, 0, sizeof(components));
+			//TODO rtrim line
+
+			//read lines of host subproto framing framing-config
+			components[0] = strchr(line, ' ');
+			if(protocols && !components[0]){
+				continue;
+			}
+
+			//terminate host
+			components[0][0] = 0;
+			components[0]++;
+
+			//find framing, terminate subprotocol
+			components[1] = strchr(components[0], ' ');
+			if(components[1]){
+				components[1][0] = 0;
+				components[1]++;
+
+				//find find framing config, terminate framing
+				components[2] = strchr(components[1], ' ');
+				if(components[2]){
+					components[2][0] = 0;
+					components[2]++;
+				}
+			}
+
+			//find a match for any indicated protocol
+			for(p = 0; p < protocols; p++){
+				if(!strcmp(components[0], protocol[p])){
+					peer.protocol = p;
+					break;
+				}
+			}
+
+			//for '*' use the first available protocol
+			if(components[0] && !strcmp(components[0], "*")){
+				peer.protocol = 0;
+			}
+
+			//the current line does not match any indicated protocols
+			if(protocols && peer.protocol == protocols){
+				continue;
+			}
+
+			peer.host = strdup(line);
+			peer.framing = core_framing(components[1]);
+			peer.framing_config = components[2] ? strdup(components[2]) : NULL;
+			break;
 		}
 		fclose(input);
 
diff --git a/plugins/makefile b/plugins/makefile
index 6b413db..1a6ef36 100644
--- a/plugins/makefile
+++ b/plugins/makefile
@@ -1,7 +1,7 @@
 .PHONY: all clean
 PLUGINS = backend_file.so
 
-CFLAGS += -fPIC -I../
+CFLAGS += -fPIC -g -I../
 LDFLAGS += -shared
 
 %.so :: %.c %.h
-- 
cgit v1.2.3