搜索

[10370] 2016-08-04_Python渗透工具的架构探讨

文档创建者：s7ckTeam
浏览次数：6
最后更新：2025-01-18
2016-08-04_Python渗透工具的架构探讨 P y t h o n 渗 透 工 具 的 架 构 探 讨 V i l l a n C h   F r e e B u f   2 0 1 6 - 0 8 - 0 4 本 文 原 创 作 者 ： V i l l a n C h ， 本 文 属 F r e e B u f 原 创 奖 励 计 划 ， 未 经 许 可 禁 止 转 载 在 最 近 忙 活 的 一 些 事 情 中 ， 体 会 到 ： 如 果 你 写 的 不 是 一 个 脚 本 ， 那 么 作 为 一 个 有 命 令 行 交 互 的 工 具 ， 整 个 工 具 的 结 构 和 可 在 最 近 忙 活 的 一 些 事 情 中 ， 体 会 到 ： 如 果 你 写 的 不 是 一 个 脚 本 ， 那 么 作 为 一 个 有 命 令 行 交 互 的 工 具 ， 整 个 工 具 的 结 构 和 可 拓 展 性 就 应 该 称 为 一 个 非 常 重 要 的 部 分 。 本 文 就 带 大 家 来 探 讨 一 下 这 个 问 题 。 当 然 前 提 是 读 者 能 写 得 出 这 个 工 具 ， 本 文 理 拓 展 性 就 应 该 称 为 一 个 非 常 重 要 的 部 分 。 本 文 就 带 大 家 来 探 讨 一 下 这 个 问 题 。 当 然 前 提 是 读 者 能 写 得 出 这 个 工 具 ， 本 文 理 论 比 较 多 ， 实 践 部 分 读 者 可 以 期 待 以 后 的 文 章 。 论 比 较 多 ， 实 践 部 分 读 者 可 以 期 待 以 后 的 文 章 。 0 × 0 0   背 景 介 绍 背 景 介 绍 其 实 就 我 自 己 来 说 ， 我 是 非 常 喜 欢   S Q L m a p   的 架 构 。 当 然 不 得 不 承 认 ， S Q L m a p   源 码 也 是 非 常 优 雅 。 当 然 网 上 也 有 了 关 于   S Q L m a p   源 码 分 析 的 文 章 ， 两 篇 比 较 好 的 文 章 都 算 是 虎 头 蛇 尾 。 我 的 这 篇 文 章 不 会 只 以   S Q L m a p   为 例 子 。 同 样 还 有 我 自 己 上 个 月 完 成 的 一 个 很 有 趣 的 匿 名 代 理 收 集 工 具 。       在 一 开 始 我 在 想 怎 么 样 把 构 造 一 个 工 具 的 每 一 个 步 骤 都 给 大 家 呈 现 出 来 这 个 问 题 上 ， 伤 了 不 少 脑 筋 ， 因 为 需 要 涉 及 的 方 面 还 是 非 常 的 多 的 。 仅 仅 是 整 体 大 致 来 讲 ， 非 常 的 抽 象 而 且 ， 肯 定 难 免 划 水 吧 。       于 是 我 们 还 是 以 解 决 具 体 问 题 的 视 角 来 讨 论 这 个 工 具 应 该 怎 么 样 完 成 吧 ！ 0 × 0 1   交 互 控 制 交 互 控 制 关 于 交 互 控 制 ， 我 们 大 致 能 想 到 的 就 是 三 种 方 式 ：       1 .   直 接 命 令 行 交 互         2 .   s h e l l   交 互         3 .   提 供   A P I     这 三 种 交 互 方 式 如 果 都 能 提 供 最 好 ， 如 果 工 具 简 单 ， 功 能 单 一 ， 结 构 也 不 打 算 设 计 多 么 复 杂 。 一 般 来 说 ， 只 需 要 提 供 一 个 命 令 行 交 互 就 可 以 了 ， 但 是 我 个 人 推 荐 至 少 还 是 需 要 提 供   A P I ，   单 独 把   A P I   抽 离 出 来 ， 这 样 让 你 的 功 能 脚 本 （ 核 心 层 ） 有 更 大 的 自 由 度 去 完 成 一 些 事 情 。 不 需 要 为 了 命 令 行 的 问 题 去 妥 协 ， 不 断 修 改 功 能 脚 本 。 只 需 要 在   A P I   构 造 的 时 候 完 成 对 核 心 功 能 脚 本 的 适 应 。 与 此 同 时 ， 你 在 构 造   s h e l l   或 构 造 命 令 行 解 析 器 的 时 候 ， 可 以 无 后 顾 之 忧 的 调 用   A P I ， 不 必 牵 一 发 而 动 全 身 。         那 么 我 们 如 何 完 成 用 户 交 互 的 构 造 呢 ？ 既 然 我 们 选 择   P y t h o n   作 为 工 具 开 发 语 言 ， P y t h o n   强 大 的 模 块 肯 定 帮 的 上 我 们 的 忙 。 c m d                 o p t p a r s e r c m d 是 一 个 构 造   s h e l l   的 一 个 简 易 框 架 ， 通 过 继 承   c m d   中 的   C m d   类 ， 添 加 相 应 方 法 可 以 构 造 出 一 个 自 己 的   s h e l l   用 于 实 时 与 用 户 交 互 。 如 果 大 家 有 过   P y t h o n   使 用 经 验 的 话 ， 应 该 对   o p t p a r s e r   比 较 熟 悉 吧 ， 这 是 一 个 非 常 优 秀 的 构 造 命 令 行 交 互 的 框 架 ， 方 便 解 析 命 令 行 。         那 么 在   S Q L m a p   中 ， 三 种 交 互 方 式 都 是 存 在 的 ， 可 能 大 家 用 的 更 多 的 是 直 接 命 令 行 交 互 ， 而 很 少 用   S Q L m a p   的   s h e l l   交 互 。         但 是 值 得 一 提 的 是 但 是 值 得 一 提 的 是   S Q L m a p   的 的   A P I   并 不 是 我 提 到 的 并 不 是 我 提 到 的   A P I   概 念 一 样 。 概 念 一 样 。   使 用 过 使 用 过   S Q L m a p   A P I   的 朋 友 们 肯 定 都 知 道 的 朋 友 们 肯 定 都 知 道   S Q L m a p   的 的   A P I   是 要 设 定 服 务 器 的 ， 通 过 本 地 是 要 设 定 服 务 器 的 ， 通 过 本 地   R E S T f u l   风 格 的 风 格 的   A P I   调 用 来 使 用 调 用 来 使 用   S Q L m a p   完 成 功 能 。 而 我 这 里 所 说 的 完 成 功 能 。 而 我 这 里 所 说 的   A P I   是 介 于 核 心 功 能 是 介 于 核 心 功 能
层 与 交 互 层 之 间 的 层 ， 提 供 交 互 与 核 心 功 能 脚 本 的 隔 离 。 层 与 交 互 层 之 间 的 层 ， 提 供 交 互 与 核 心 功 能 脚 本 的 隔 离 。 0 × 0 2   配 置 问 题 配 置 问 题 我 私 以 为 这 是 非 常 有 必 要 重 视 的 一 个 地 方 ， 在 我 刚 开 始 接 触 这 玩 意 的 时 候 ， 经 常 性 的 改 动 了 一 个 地 方 的 数 据 ， 还 要 手 动 翻 到 别 的 脚 本 中 去 改 相 应 的 配 置 。 道 理 大 家 都 懂 ， 但 实 际 动 手 的 时 候 ， 自 己 仍 然 是 犯 蠢 了 。 那 么 这 里 其 实 很 好 解 决 。 你 可 以 设 置 配 置 的 全 局 的 变 量 都 放 在 一 个   y o u r d a t a . p y   中 ， 然 后 当 需 要 的 时 候 虽 然 可 能 在 大 家 看 来 这 是 很 小 的 事 情 ， 也 并 不 用 提 到 台 面 上 ， 但 是 作 为 吃 过 亏 的 人 来 讲 ， 我 还 是 乐 意 带 大 家 看 看   S Q L m a p   是 怎 么 解 决 这 个 问 题 的 。         下 图 是   S Q L m a p   的 部 分 目 录 结 构 ：         这 个   d a t a . p y   中 ： 注 意 ： 注 意 ：   A t t r i b D i c t   是 是   S Q L m a p   自 己 继 承 了 一 个 自 己 继 承 了 一 个   D i c t   完 成 了 对 原 来 完 成 了 对 原 来   D i c t   的 一 些 改 进 ， 可 以 简 单 就 当 成 一 个 的 一 些 改 进 ， 可 以 简 单 就 当 成 一 个   D i c t   来 看 就 可 以 了 来 看 就 可 以 了 。 。 然 后   S Q L m a p   在 测 试   S Q L   注 入 的 过 程 中 ， 使 用 的 配 置 文 件 事 先 全 部 被 加 载 到 了   c o n f   和   k b   中 了 ， 因 此 ， 在 程 序 执 行 的 过 程 中 ， 凡 是 需 要 用 到 配 置 文 件 的 地 方 ， 只 需 要   i m p o r t   进   c o n f   与   k b   就 可 以 很 方 便 完 成 任 务 了 。   同 时 默 认 配 置 在 同 目 录 下 的   d e f a u l t s . p y   中 ， 各 种 数 据 （ 比 如 重 要 的 枚 举 类 型 ， 重 要 的   d i c t s ） 都 在   c o r e   这 个 包 中 ， 当   S Q L m a p   需 要 的 时 候 只 需 要   i m p o r t   进 实 现 设 定 好 的 数 据 就 可 以 了 。 可 能 有 些 读 者 还 是 觉 得 ， 这 并 没 有 什 么 了 不 起 ， 或 者 并 没 有 体 会 到 这 样 做 的 便 捷 性 （ 我 直 接 写 在 功 能 脚 本 里 快 得 很 ， 而 且 从 来 不 管 你 所 谓 的 架 构 ， 我 的 程 序 照 样 跑 啊 。 ） ， 如 果 这 样 想 的 话 ， 大 概 这 样 可 能 程 序 的 扩 展 性 也 健 壮 性 都 会 受 到 很 大 的 影 响 。 下 面 是 我 认 为 比 较 优 化 的 一 个 结 构 f r o m   y o u r d a t a   i m p o r t   X X X X X X 或 者         f r o m   d a t a . y o u r d a t a   i m p o r t   X X X X X X # ! / u s r / b i n / e n v   p y t h o n f r o m   l i b . c o r e . d a t a t y p e   i m p o r t   A t t r i b D i c t f r o m   l i b . c o r e . l o g   i m p o r t   L O G G E R #   s q l m a p   p a t h s p a t h s   =   A t t r i b D i c t ( ) #   o b j e c t   t o   s t o r e   o r i g i n a l   c o m m a n d   l i n e   o p t i o n s c m d L i n e O p t i o n s   =   A t t r i b D i c t ( ) #   o b j e c t   t o   s t o r e   m e r g e d   o p t i o n s   ( c o m m a n d   l i n e ,   c o n f i g u r a t i o n   f i l e   a n d   d e f a u l t   o p t i o n s ) m e r g e d O p t i o n s   =   A t t r i b D i c t ( ) #   o b j e c t   t o   s h a r e   w i t h i n   f u n c t i o n   a n d   c l a s s e s   c o m m a n d #   l i n e   o p t i o n s   a n d   s e t t i n g s c o n f   =   A t t r i b D i c t ( ) #   o b j e c t   t o   s h a r e   w i t h i n   f u n c t i o n   a n d   c l a s s e s   r e s u l t s k b   = A t t r i b D i c t ( ) #   o b j e c t   w i t h   e a c h   d a t a b a s e   m a n a g e m e n t   s y s t e m   s p e c i f i c   q u e r i e s q u e r i e s   =   { } #   l o g g e r l o g g e r   =   L O G G E R    
同 样 对 比 我 们 看 一 下 比 较 糟 糕 的 结 构 问 题 还 是 相 当 明 显 的 ： 配 置 1 ， 配 置 2 ， 配 置 3 很 有 可 能 有 交 集 ， 当 你 需 要 改 动 交 集 中 的 某 项 配 置 的 时 候 ， 你 需 要 改 动 三 个 功 能 脚 本 ， 这 是 非 常 难 受 的 一 件 事 情 。 0 × 0 3   I / O   控 制 控 制 所 谓 的   I / O   控 制 并 不 是 说 像 操 作 系 统   I / O   那 样 的 概 念 ， 我 们 要 讨 论 的 是 程 序 的   I / O   问 题 ： 1 .   用 户 输 入 与 呈 现 给 用 户 ( 交 互 与 输 出 )                 2 .   网 络   I / O   ( 访 问 网 络 )                 3 .   文 件 与 数 据   I / O   ( 数 据 库 与   c s v ,   x m l ,   j s o n   等 类 型 文 件 读 取 与 导 出 )                 4 .   日 志 问 题 ： 网 络 问 题 ： 网 络   I / O   ? 读 者 肯 定 要 问 ， 网 络   I / O   也 需 要 我 们 专 门 注 意 么 ？         肯 定 需 要 ！ 在 大 家 使 用 一 个 渗 透 工 具 的 时 候 ， 肯 定 多 多 少 少 注 意 隐 匿 自 己 的 身 份 ， 当 然 事 先 布 置 好   T o r   或 者   V P N   或 者   使 用   H T T P   代 理 ， 当 然 ， 在 国 内 的 我 们 可 能 会 花 了 好 大 的 力 气 配 置   T o r   的 上 游 服 务 器 ， 会 花 钱 购 买   V P N （ 当 然 会 被 告 知 禁 止 使 用   V P N   做 非 法 的 勾 当 ） ， 或 者 说 上 公 共 代 理 网 站 上 寻 找 一 些 公 共 的 代 理 拿 来 用 。 （ 当 然 ， 我 也 经 常 干 这 些 事 情 。 所 以 写 了 一 个 公 共 代 理 收 集 过 滤 的 工 具 ） 。 所 谓 的 网 络   I / O   控 制 就 是 说 控 制 你 连 接 网 络 的 某 些 方 式 ， 比 如   S Q L m a p   提 供 了 使 用   T o r   的 接 口 ， 也 提 供 了         H T T P   代 理 的 接 口 （ – p r
o x y ） 这 些 功 能 都 是 在 网 络   I / O   控 制 实 现 的 。         可 能 有 朋 友 就 会 说 ， P y t h o n   中 有   r e q u e s t s   还 不 够 么 ？ 自 己 重 新 写 一 个 是 什 么 鬼 啊 ？         那 么 我 们 就 具 体 来 看 一 下   S Q L m a p   中 的 网 络 控 制 ， 大 家 一 切 都 明 了 了 。 对 代 理 的 细 节 处 理 显 然 不 是 其 他 的 自 带 的 模 块 所 有 的 ， 这 里 的 对   T o r   和   h t t p   代 理 的 处 理 就 数 据 网 络   I / O   的 控 制 。         当 然 这 里 可 能 看 不 到 对   T o r   处 理 的 细 节 ， 但 是 打 开   c o n n R e a d P r o x y 那 么 对 于 网 络   I / O   我 们 就 不 谈 更 多 了 。 在 用 户 输 入 与 展 现 给 用 户 的 输 出 上 ， 还 是 有 话 要 说 的 。 因 为 ， 如 果 很 随 意 的 直 接   p r i n t   出 去 的 话 非 常 的 尴 尬 ， 我 们 可 以 单 独 控 制 一 个 输 出 函 数 ， 程 序 中 所 有 的 输 出 都 调 用 某 个 函 数 或 者 以 同 一 种 方 式 输 出 ， 而 不 是 到 处 都 存 在 的 调 试 专 用   p r i n t   。   其 实 这 样 做 非 常 的 合 理 ， 当 输 出 可 控 时 ， 你 可 以 控 制 输 出 的 等 级 ， 比 如   D e b u g   级 别 ， 在   D e b u g   开 启 的 时 候 ， 所 有 的 信 息 都 会 输 出 。   E r r o r   级 别 ，   还 有   I n f o ,   W a r n i n g ,   C r i t i c a l   在 输 入 的 时 候 ， 你 决 定 这 条 信 息 展 示 在 用 户 面 前 的 级 别 ， 一 般 来 说 ， 如 果   I n f o   级 别 的 信 息 优 先 级 要 低 于   C r i t i c a l ，   你 （ 或 者 ） 可 以 选 择   I n f o   级 别 的 输 出 要 不 要 用 户 看 到 ， 只 展 示   E r r o r   或 者 只 展 示   C r i t i c a l   以 上 的 信 息 。 同 样 的 对 于 日 志 来 说 也 一 样 ， 如 果 大 家 有 使 用 过   l o g g i n g   模 块 应 该 会 熟 悉 这 个 非 常 好 用 的 日 志 模 块 ， 替 我 们 剩 下 了 很 大 的 力 气 ， 不 用 自 己 去 写 日 志 的 输 出 格 式 什 么 了 。         哎 ， 数 据 库 读 写 ， 核 心 功 能 脚 本 执 行 的 结 果 要 支 持 不 同 数 据 类 型 的 导 出 ， 这 些 也 应 该 是 要 具 备 的 ， 因 为 导 出 的 易 用 的 数 据 文 件 ， 可 以 很 容 易 的 被 其 他 工 具 使 用 ， 可 以 以 更 好 的 方 式 被 加 载 ， 使 用 ， 而 不 是 手 动 的 再 次 输 入 结 果 ， 这 样 很 蠢 的 。 在   S Q L m a p   中 就 使 用 了 很 多 专 门 处 理 各 种 特 定 文 件 类 型 的 脚 本 来 完 成 这 样 的 基 础 性 的 工 作 。 0 × 0 4   差 错 控 制 差 错 控 制 编 写 渗 透 测 试 工 具 的 人 员 一 般 不 是 专 业 的 程 序 员 ， 那 么 差 错 控 制 肯 定 也 是 一 个 值 得 注 意 的 问 题 。 并 不 推 崇 一 个   t r y   一 个   e x c e p t   捕 捉 所 有 的 错 误 ， 这 样 真 的 出 事 的 时 候 会 让 人 一 头 雾 水 的 。 那 么 正 确 的 做 法 应 该 是 什 么 呢 ？ 编 写 你 自 己 的 异 常 ， 灵 活 使 用   P y t h o n   提 供 的 例 如   V a l u e E r r o r   I n d e x E r r o r   和 键 盘 中 断 等 ， 然 后 让 出 问 题 的 部 分 了 属 于 胸 。 可 能 我 在 这 里 说 这 个 ， 懂 得 编 程 的 读 者 会 觉 得 可 笑 ， 说 实 话 我 也 觉 得 这 非 常 的 班 门 弄 斧 ， 但 是 我 相 信 点 开 这 篇 文 章 的 大 多 数 还 是 一 个 渗 透 测 试 的 爱 好 者 或 者 从 业 者 ， 在 进 行 渗 透 测 试 的 时 候 经 常 会 需 要 写 脚 本 ， 完 成 自 己 的 目 的 。 到 一 定 程 度 以 后 ， 希 望 把 自 己 的 脚 本 整 合 起 来 ， 做 一 个 更 好 的 工 具 。 自 己 使 用 或 者 是 留 给 别 人 用 。 毕 竟 写 脚 本 和 写 程 序 做 工 具 还 是 完 全 不 同 的 两 个 概 念 ， 作 为 一 个 写 得 出 脚 本 的 渗 透 测 试 者 ， 还 是 需 要 多 少 有 一 些 编 程 的 经 验 。
0 × 0 5   我 心 中 的 理 想 架 构 我 心 中 的 理 想 架 构 上 面 讲 了 这 么 多 ， 可 能 有 些 人 会 有 所 收 获 ， 有 些 人 会 一 头 雾 水 ， 至 少 我 觉 得 ， 结 合 我 自 己 这 段 时 间 的 体 会 ， 和 几 个 具 体 的 例 子 ， 一 个   P y t h o n   工 具 该 有 的 架 构 ， 该 有 的 内 容 ， 我 还 算 是 比 较 了 解 的 。 那 么 我 就 专 门 做 了 一 个   P y t h o n   工 具 的 架 构 图 给 大 家 看 ， 当 然 这 是 我 心 目 中 比 较 不 错 的 结 构 了 。 如 果 有 兴 趣 的 话 ， 大 家 可 以 实 现 一 下 ， 开 发 一 个 更 高 效 更 好 用 的 工 具 。 上 图 的 工 具 比 较 适 合 脚 本 集 类 型 的 工 具 开 发 ， 功 能 可 以 比 较 分 散 ， 在 具 体 的 功 能 上 可 以 不 要 求 太 大 的 连 续 性 ， 但 是 交 互 性 更 加 强 ， 建 议 使 用   s h e l l   做 交 互 以 便 不 同 功 能 随 意 切 换 而 且 不 需 要 重 新 加 载 程 序 。 这 第 二 种 架 构 就 是   S Q L m a p   所 使 用 的 架 构 的 ‘ 改 变 ’ 版 本 （ S Q L m a p   没 有 提 供 功 能 开 放 的   A P I ，   但 是 我 还 是 希 望 这 样 来 让 所 有 的   A P I   更 加 好 找 易 于 二 次 开 发 ） 。 特 别 适 合 流 程 类 的 工 具 开 发 。 比 如   S Q L   注 入 检 测 就 是 一 个 可 以 抽 象 出 流 程 的 测 试 过 程 。
0 × 0 6   结 语 结 语 一 款 牛 B 的 工 具 的 诞 生 ， 绝 对 不 是 简 单 的 功 能 拼 凑 ， 而 是 不 断 重 构 ， 思 考 ， 注 意 细 节 。         显 然 我 们 足 够 幸 运 ， 可 以 使 用   P y t h o n ，   很 多 细 节 的 工 作 ， 前 人 已 经 替 我 们 完 成 了 ， 站 在 巨 人 的 肩 膀 上 ， 风 景 显 然 更 加 的 壮 观 。         那 么 ， 如 果 大 家 还 是 不 太 名 如 何 完 成 一 个 工 具 的 话 ， 如 果 感 兴 趣 的 人 多 的 话 ， 我 就 继 续 讲 非 常 具 体 的 细 节 吧 ， 很 乐 意 和 大 家 分 享 我 微 薄 的 经 验 。 * 本 文 原 创 作 者 ： V i l l a n C h ， 本 文 属 F r e e B u f 原 创 奖 励 计 划 ， 未 经 许 可 禁 止 转 载 阅 读 原 文
回复

举报

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

本版积分规则

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

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