More checks for printable asciis

Something at some point changed so this becomes necessar

gateway/httpgateway.cpp

@@ -20,6 +20,7 @@ SOFTWARE.
 */
 #include "httpgateway.h"
 #include "../logger.h"
+#include <stdexcept>
 HttpGateway::HttpGateway(std::string listenaddr, int port, uint64_t maxPayloadLength)
 {
 	this->listenaddr = listenaddr;
@@ -34,6 +35,11 @@ bool HttpGateway::keepReading()
 
 Request HttpGateway::convertRequest(httplib::Request request)
 {
+	if(!utils::is_printable_ascii(request.target))
+	{
+		throw std::runtime_error("Invalid chars in URI: " + utils::catv(request.target));
+	}
+
 	Request result;
 	result.setRequestMethod(request.method);
 	result.setUrl(request.target);
@@ -55,6 +61,12 @@ Request HttpGateway::convertRequest(httplib::Request request)
 
 	if(request.has_header("COOKIE"))
 	{
+		std::string cookie = request.get_header_value("COOKIE");
+		if(!utils::is_printable_ascii(cookie))
+		{
+			/* We better bail */
+			throw std::runtime_error("Cookie with non printable chars sent");
+		}
 		result.initCookies(request.get_header_value("COOKIE"));
 	}
 	result.setIp("127.0.0.1");
@@ -82,6 +94,27 @@ httplib::Response HttpGateway::convertResponse(Response response)
 void HttpGateway::work(RequestWorker &worker)
 {
 	httplib::Server server;
+	server.set_exception_handler([](const httplib::Request& req, httplib::Response& res, std::exception_ptr ep) {
+		auto fmt = "<h1>Error 500</h1><p>%s</p>";
+		char buf[BUFSIZ];
+		try
+		{
+			std::rethrow_exception(ep);
+		}
+		catch (std::exception &e)
+		{
+			std::string exception = utils::html_xss(e.what());
+			snprintf(buf, sizeof(buf), fmt, exception.c_str());
+			Logger::error() << "Exception caught in Httpgateway::work():" << utils::html_xss(utils::catv(e.what()));
+		}
+		catch (...)
+		{
+			snprintf(buf, sizeof(buf), fmt, "Unknown Exception");
+			Logger::error() << "Unknown exception caught in Httpgateway::work()";
+		}
+		res.set_content(buf, "text/html");
+		res.status = 500;
+	});
 	server.set_payload_max_length(this->maxPayloadLength);
 	auto handler = [&](const httplib::Request &req, httplib::Response &res)
 	{
@@ -94,4 +127,5 @@ void HttpGateway::work(RequestWorker &worker)
 	server.Get("/(.*)", handler);
 	server.Post("/(.*)", handler);
 	server.listen(this->listenaddr.c_str(), this->listenport);
+
 }

logger.h

@@ -2,6 +2,7 @@
 #define LOGGER_H
 #include <iostream>
 #include <chrono>
+#include "utils.h"
 class Logger
 {
   private:
@@ -24,7 +25,15 @@ class Logger
 			{
 				(*out) << time(0) << " " << prefix;
 			}
-			(*out) << val;
+
+			if constexpr (std::is_convertible_v<T, std::string_view>)
+			{
+				(*out) << utils::catv(val);
+			}
+			else
+			{
+				(*out) << val;
+			}
 			headerSent = true;
 			return *this; // or maybe out itself? probably not.
 		}

qswiki.cpp

@@ -230,8 +230,8 @@ int main(int argc, char **argv)
 	}
 	catch(const std::exception &e)
 	{
-		Logger::error() << e.what();
-		std::cerr << e.what() << std::endl;
+		Logger::error() << utils::catv(e.what());
+		std::cerr << utils::catv(e.what()) << std::endl;
 	}
 	return 0;
 }

utils.cpp

@@ -214,3 +214,40 @@ std::string utils::trim(std::string_view view)
 	}
 	return std::string{view};
 }
+
+std::string utils::catv(std::string_view view)
+{
+	std::string result;
+	result.reserve(view.length());
+	for(auto c : view)
+	{
+		if (!isascii(c))
+		{
+			result += "M-";
+			result += toascii(c);
+		}
+		else if(iscntrl(c))
+		{
+			result += '^';
+			result += c == '\177' ? '?': c | 0100;
+		}
+		else
+		{
+			result += c;
+		}
+	}
+	return result;
+}
+
+bool utils::is_printable_ascii(std::string view)
+{
+	for(char c : view)
+	{
+		if( !(c >= ' ' && c <= '~'))
+		{
+			return false;
+		}
+	}
+	return true;
+}
+

utils.h

@@ -91,6 +91,10 @@ template <class T> inline std::string toString(const T &v)
 }
 
 std::string trim(std::string_view view);
+std::string catv(std::string_view view);
+
+bool is_printable_ascii(std::string view);
+
 
 } // namespace utils
 #endif