Whitelist for storing request headers

This commit is contained in:
Ian Gulliver
2019-04-30 02:07:48 +00:00
parent 205b8253b2
commit 87bc2c1611
6 changed files with 23 additions and 13 deletions

View File

@@ -2,6 +2,7 @@
#include <functional>
#include <unordered_map>
#include <unordered_set>
#include "stream_buffer.h"
@@ -10,7 +11,7 @@ class FastCGIRequest;
class FastCGIConn {
public:
FastCGIConn(int sock, const sockaddr_in6& client_addr, const std::function<void(std::unique_ptr<FastCGIRequest>)>& callback);
FastCGIConn(int sock, const sockaddr_in6& client_addr, const std::function<void(std::unique_ptr<FastCGIRequest>)>& callback, const std::unordered_set<std::string_view>& headers);
~FastCGIConn();
void Serve();
@@ -19,7 +20,8 @@ class FastCGIConn {
private:
const int sock_;
std::function<void(std::unique_ptr<FastCGIRequest>)> callback_;
const std::function<void(std::unique_ptr<FastCGIRequest>)>& callback_;
const std::unordered_set<std::string_view>& headers_;
uint64_t requests_ = 0;