X-Git-Url: https://git.sesse.net/?p=cubemap;a=blobdiff_plain;f=parse.cpp;h=9c7fd10b2f0c87a4f01e1e6112d683c07ccd3e18;hp=37b5436bd6d7bc82cd7890e0082ad6a26543f727;hb=d34b94a858c08d64eddfb9c115719fd9129be933;hpb=8d15ccbfc46a7f0b96a47109d8f8f2495082b8f6

diff --git a/parse.cpp b/parse.cpp
index 37b5436..9c7fd10 100644
--- a/parse.cpp
+++ b/parse.cpp
@@ -13,8 +13,23 @@ vector<string> split_tokens(const string &line)
 	vector<string> ret;
 	string current_token;
 
+	bool in_quote = false;
+
 	for (size_t i = 0; i < line.size(); ++i) {
-		if (isspace(line[i])) {
+		// Handle all escaped characters.
+		if (line[i] == '\\' && i < line.size() - 1) {
+			current_token.push_back(line[++i]);
+			continue;
+		}
+
+		// Handle start and end quote.
+		if (line[i] == '"') {
+			in_quote = !in_quote;
+			continue;
+		}
+
+		// Handle break.
+		if (isspace(line[i]) && !in_quote) {
 			if (!current_token.empty()) {
 				ret.push_back(current_token);
 			}
@@ -56,9 +71,9 @@ vector<string> split_lines(const string &str)
 	return ret;
 }
 
-multimap<string, string> extract_headers(const vector<string> &lines, const string &log_context)
+HTTPHeaderMultimap extract_headers(const vector<string> &lines, const string &log_context)
 {
-	multimap<string, string> parameters;
+	HTTPHeaderMultimap parameters;
 	for (size_t i = 1; i < lines.size(); ++i) {
 		size_t split = lines[i].find(":");
 		if (split == string::npos) {
@@ -100,7 +115,7 @@ RequestParseStatus wait_for_double_newline(string *existing_data, const char *ne
 	const char *ptr = reinterpret_cast<char *>(
 		memmem(existing_data->data() + start_at, existing_data->size() - start_at,
 		       "\r\n\r\n", 4));
-	if (ptr == NULL) {
+	if (ptr == nullptr) {
 		return RP_NOT_FINISHED_YET;
 	}
 	if (ptr != existing_data->data() + existing_data->size() - 4) {