搜索

[13765] 2019-02-17_为Nginx加入一个使用深度学习的软WAF

文档创建者：s7ckTeam
浏览次数：2
最后更新：2025-01-18
2019-02-17_为Nginx加入一个使用深度学习的软WAF 为 N g i n x 加 入 一 个 使 用 深 度 学 习 的 软 W A F r o c h e k   F r e e B u f   2 0 1 9 - 0 2 - 1 7 一 、 前 言 一 、 前 言 本 文 介 绍 如 何 向 本 文 介 绍 如 何 向 N g i n x 增 加 了 一 个 使 用 增 加 了 一 个 使 用 T e n s o r f l o w   C 库 的 软 库 的 软 W A F 模 块 ， 模 块 主 体 基 于 模 块 ， 模 块 主 体 基 于 N a x s i 。 。 二 、 获 取 数 据 及 训 练 数 据 二 、 获 取 数 据 及 训 练 数 据 这 里 ， 之 前 有 D a l a o 发 表 过 这 样 一 篇 文 章 ： 基 于 卷 积 神 经 网 络 的 S Q L 注 入 检 测 。 这 是 一 个 开 源 的 项 目 ， 但 是 由 于 速 度 的 关 系 ， 我 不 打 算 使 用 这 篇 文 章 的 模 型 ， 仅 仅 采 用 这 篇 文 章 使 用 的 数 据 集 。 这 样 可 以 节 省 很 多 特 征 工 程 的 时 间 。 数 据 训 练 并 不 是 这 篇 文 章 的 重 点 ， 这 里 仅 仅 说 一 下 训 练 结 果 ， 这 里 为 了 防 止 C U D A 周 期 对 检 测 时 间 的 影 响 ， 使 用 C P U 跑 推 理 过 程 。 如 果 您 对 数 据 的 训 练 感 兴 趣 ， 可 以 看 我 之 前 写 的 一 篇 文 章 ： 使 用 C N N 做 S Q L 和 X S S 的 识 别 。 三 、 使 用 三 、 使 用 T e n s o r f l o w   C 库 做 推 理 库 做 推 理 我 们 的 目 标 是 向 N g i n x 加 入 一 个 使 用 T e n s o r f l o w   C 库 的 软 W A F 模 块 。 如 果 从 头 开 始 写 一 个 软 W A F ， 想 必 会 占 用 相 当 多 的 时 间 ， 并 且 这 个 也 和 这 篇 文 章 的 主 旨 偏 离 。 在 N g i n x 的 开 源 的 软 W A F 模 块 中 ， N a x s i 是 一 个 很 受 欢 迎 的 模 块 。 这 个 模 块 使 用 C 作 为 主 开 发 语 言 ， 因 此 ， 如 果 我 们 想 基 于 这 个 模 块 加 一 个 推 理 过 程 ， 很 大 可 能 性 需 要 加 入 T e n s o r f l o w   C 库 。 那 么 ， 首 先 应 该 做 的 是 ， 试 着 使 用 T e n s o r f l o w   C 库 做 单 次 推 理 ， 并 做 好 模 块 测 试 。 M a i n 文 件 如 下 ：
编 译 完 成 后 ， 可 以 跑 一 下 数 据 ， 这 里 为 了 节 省 篇 幅 ， 仅 测 试 一 条 可 见 这 部 分 代 码 已 经 可 以 正 常 工 作 了 。 四 、 向 四 、 向 N a x s i 内 加 入 代 码 内 加 入 代 码 首 先 ， 回 忆 下 N g i n x 的 一 些 原 理 ， N g i n x 在 运 行 时 使 用 f o r k ， 创 建 了 一 个 m a s t e r 进 程 和 若 干 w o r k e r 进 程 ， w o r k e r 进 程 是 实 际 处 理 数 据 的 进 程 。 每 个 模 块 的 初 始 化 函 数 ， 实 际 上 是 由 初 始 化 的 进 程 来 完 成 的 ， 在 这 之 后 ， 如 果 配 置 了 d a e m o n ， 初 始 化 的 进 程 自 动 退 出 。 同 时 ， 为 了 便 于 理 解 ， 我 们 可 以 把 推 理 流 程 拆 分 成 这 样 几 个 部 分 ： 1 .   初 始 化 模 型
2 .   将 输 入 转 化 为 T e n s o r f l o w 识 别 的 格 式 3 .   运 行 模 型 ， 获 取 结 果 其 中 ， 步 骤 1 仅 需 运 行 一 次 ， 步 骤 2 ， 3 在 每 次 运 行 这 个 模 块 时 都 需 要 进 行 。 我 对 于 N g i n x 理 解 不 深 ， D a L a o 轻 拍 。 由 于 w o r k e r 进 程 是 f o r k 产 生 的 ， 实 际 上 是 无 法 使 用 初 始 化 进 程 产 生 的 m o d e l ， 所 以 ， 每 个 w o r k e r 进 程 需 要 自 己 初 始 化 一 次 自 己 的 m o d e l 相 关 资 源 。 因 此 ， 新 加 入 的 函 数 应 该 类 似 这 样 ： 这 里 ， 我 仅 仅 检 测 u r i 内 部 的 注 入 请 求 ， 其 它 部 分 的 检 测 代 码 应 该 是 非 常 相 似 的 ， 这 里 就 不 再 重 复 罗 嗦 了 。 由 于 t f _ m o d e l 是 在 第 一 次 调 用 模 块 时 才 会 自 动 载 入 ， 因 此 ， 在 这 个 服 务 器 的 每 个 w o r k e r 进 程 第 一 次 接 受 到 数 据 时 ， 可 能 会 稍 卡 顿 一 下 。 五 、 编 译 运 行 五 、 编 译 运 行 将 N a x s i 和 N g i n x 的 代 码 同 时 复 制 到 编 译 服 务 器 内 部 ， 然 后 在 编 译 N g i n x 时 ， 包 含 下 N a x s i 的 代 码 部 分 。 由 于 N g i n x 在 编 译 时 是 不 使 用 T e n s o r f l o w 库 的 ， 所 以 我 们 需 要 手 动 修 改 下 M a k e f i l e ， 在 链 接 时 自 动 加 入 T e n s o r f l o w 库 ， 最 后 编 译 。 然 后 ， 需 要 将 N a x s i 的 配 置 文 件 复 制 到 N g i n x 的 c o n f 文 件 夹 内 ， 配 置 N a x s i ， 并 修 改 N g i n x 的 配 置 文 件 ， 加 入 N a x s i 模 块 。
自 定 义 一 个 4 0 3 的 P a g e ， 作 为 注 入 发 生 时 的 替 换 界 面 。 然 后 ， 运 行 N g i n x 。 六 、 手 注 测 试 六 、 手 注 测 试 这 里 使 用 简 单 的 手 注 ， 测 试 模 块 是 否 正 常 运 行 。 这 里 实 际 是 不 可 能 出 现 注 入 的 情 况 的 ， 如 果 有 兴 趣 的 话 ， 可 以 加 入 D V W A 等 靶 机 ， 使 用 s q l m a p 等 工 具 实 际 攻 击 。 七 、 总 结 七 、 总 结 经 过 以 上 几 个 步 骤 ， 我 们 就 得 到 了 一 个 使 用 D L 的 N g i n x 的 软 W A F 模 块 ， 并 把 它 加 入 到 了 N g i n x 内 。 相 比 普 通 的 规 则 匹 配 类 的 引 擎 ， 一 般 而 言 ， 这 种 方 法 的 防 御 效 果 更 加 优 秀 。 考 虑 漏 报 和 误 报 率 的 话 ， 使 用 测 试 集 测 试 ， 可 靠 性 提 高 了 数 倍 。 当 然 ， 推 理 流 程 会 对 服 务 器 本 身 的 性 能 造 成 一 定 的 影 响 ， 因 此 在 实 际 的 环 境 中 ， 可 能 需 要 使 用 T e n s o r f l o w   C   G P U 库 来 跑 运 算 ， 或 者 将 N g i n x 作 为 代 理 使 用 。 * 本 文 作 者 ： 本 文 作 者 ： r o c h e k ， 转 载 请 注 明 来 自 ， 转 载 请 注 明 来 自 F r e e B u f . C O M
阅 读 原 文
回复

举报

高级模式
B Color Image Link Quote Code Smilies
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

使用Markdown编辑器编辑 使用富文本编辑器编辑

Archiver| 手机版| 小黑屋|
Powered by Discuz! X3.4 Copyright © 2001-2020, Tencent Cloud. 商业源码建议获取授权，如侵犯您的权益，请联系客服处理