HTTP解析层接口改进，增加功能。

1 files changed, 89 insertions, 0 deletions

diff --git a/src/httpscan.h b/src/httpscan.h
new file mode 100644
index 0000000..2665084
--- /dev/null
+++ b/src/httpscan.h
@@ -0,0 +1,89 @@
+//
+// Created by luqiu on 2018-5-15.
+//
+
+#ifndef TFE_HTTPSCAN_H
+#define TFE_HTTPSCAN_H
+
+#include <Maat_rule.h>
+#include "opts.h"
+#include "httpaction.h"
+
+/* Forward Declare */
+class HttpScanSession;
+class HttpScan;
+class HttpSession;
+
+/* HttpScan Module Init/Deinit */
+class HttpScan
+{
+public:
+    HttpScan(struct tfe_instance * instance, struct tfe_config *config);
+    ~HttpScan() = default;
+
+    void handlerConnectionCreate(HttpConnection & ct);
+    void handlerConnectionClose(HttpConnection & ct);
+
+    /* Table Symbols */
+protected:
+    /* Global Instance */
+    tfe_instance * tfe_instance;
+    /* Global Config */
+    tfe_config * tfe_config;
+    /* Maat扫描句柄 */
+    Maat_feather_t maat_feather_ref;
+    /* 控制编译表ID */
+    int table_id_ctrl_compile;
+    /* IP配置表ID */
+    int table_id_ctrl_ip;
+    /* URL配置表ID */
+    int table_id_ctrl_http_url;
+    /* HTTP请求头部配置表ID */
+    int table_id_ctrl_http_req_hdr;
+    /* HTTP请求体配置表ID */
+    int table_id_ctrl_http_req_body;
+    /* HTTP应答头部配置表ID */
+    int table_id_ctrl_http_res_hdr;
+    /* HTTP应答体配置表ID */
+    int table_id_ctrl_http_res_body;
+
+    /* IP白名单扫描 */
+    int connection_bypass_scan();
+    int connection_bypass_do_action();
+
+    friend HttpScanSession;
+};
+
+/* Httpscan Ctx per HTTP Session */
+class HttpScanSession
+{
+public:
+    explicit HttpScanSession(const HttpScan & httpscan_module);
+    ~HttpScanSession();
+
+    /* HTTP请求、应答扫描 */
+    void ScanRequestHeader(HttpSession *http_session_ctx);
+    void ScanRequestBody(HttpSession *http_session_ctx);
+    void ScanResponseHeader(HttpSession *http_session_ctx);
+    void ScanResponseBody(HttpSession *http_session_ctx);
+
+private:
+    void hit_config_and_do_action(HttpSession *http_session_ctx);
+    void hit_scan_error();
+
+    /* 最大命中结果数量 */
+    static constexpr int MAAT_SCAN_RESULT_ = 4;
+    /* 默认内容编码 */
+    static constexpr auto MAAT_DEFAULT_CHARSET_ = CHARSET_GBK;
+
+    /* HTTPSCAN Module句柄引用 */
+    const HttpScan & httpscan_module_ref_;
+    /* 扫描中间句柄 */
+    scan_status_t maat_scan_mid_{nullptr};
+    /* 扫描命中结果 */
+    Maat_rule_t maat_scan_result_[MAAT_SCAN_RESULT_];
+    /* 扫描命中数量 */
+    int nr_maat_scan_result_{0};
+};
+
+#endif //TFE_HTTPSCAN_H