搜索

[14862] 2019-11-25_记一次开源工具某模块的基础二次开发

文档创建者：s7ckTeam
浏览次数：3
最后更新：2025-01-18
2019-11-25_记一次开源工具某模块的基础二次开发 记 一 次 开 源 工 具 某 模 块 的 基 础 二 次 开 发 原 创   煜 阳 y u y a n g   F r e e B u f   2 0 1 9 - 1 1 - 2 5 近 日 看 到 一 些 关 于 工 具 的 文 章 ， 很 多 人 对 于 写 工 具 类 文 章 都 是 蜻 蜓 点 水 ， 写 搭 建 ， 写 基 础 使 用 ， 但 是 基 础 的 开 近 日 看 到 一 些 关 于 工 具 的 文 章 ， 很 多 人 对 于 写 工 具 类 文 章 都 是 蜻 蜓 点 水 ， 写 搭 建 ， 写 基 础 使 用 ， 但 是 基 础 的 开 源 工 具 在 生 产 环 境 上 的 使 用 其 实 是 比 较 困 难 的 ， 有 很 多 需 要 二 次 开 发 做 到 与 生 产 环 境 适 配 ， 今 天 写 这 篇 文 章 的 源 工 具 在 生 产 环 境 上 的 使 用 其 实 是 比 较 困 难 的 ， 有 很 多 需 要 二 次 开 发 做 到 与 生 产 环 境 适 配 ， 今 天 写 这 篇 文 章 的 目 的 就 在 于 开 启 开 源 工 具 二 次 开 发 的 思 路 ， 其 中 一 个 模 块 的 二 次 开 发 很 简 单 ， 大 家 基 本 都 能 看 懂 ， 就 以 这 篇 简 目 的 就 在 于 开 启 开 源 工 具 二 次 开 发 的 思 路 ， 其 中 一 个 模 块 的 二 次 开 发 很 简 单 ， 大 家 基 本 都 能 看 懂 ， 就 以 这 篇 简 单 易 懂 的 代 码 改 动 举 例 。 单 易 懂 的 代 码 改 动 举 例 。 本 次 开 源 工 具 二 次 开 发 以 本 次 开 源 工 具 二 次 开 发 以 c o b r a 代 码 审 计 为 例 。 代 码 审 计 为 例 。   适 用 范 围 ： 代 码 初 学 者 、 有 一 定 p y t h o n 基 础 与 协 议 基 础 需 求 明 确 需 求 明 确 首 先 要 先 明 确 自 己 的 需 求 ， 我 们 的 需 求 是 在 上 线 前 做 关 联 性 代 码 审 计 ， 我 采 用 的 是 c o b r a ， 至 于 为 什 么 要 使 用 c o b r a 就 不 多 说 了 ， 这 是 评 估 期 间 做 的 事 情 ， 文 章 主 要 写 二 次 开 发 。 显 示 情 况 显 示 情 况 我 们 的 生 产 环 境 使 用 g i t l a b 作 为 代 码 库 ， 仅 允 许 s s h 方 式 下 拉 代 码 ， 而 不 允 许 h t t p 公 开 拉 代 码 ， 这 么 做 是 为 了 安 全 ， 但 是 同 样 的 也 对 c o b r a 的 使 用 造 成 了 一 定 的 麻 烦 。 因 为 基 础 的 c o b r a 是 没 有 s s h 下 拉 功 能 的 ， 需 要 我 们 自 己 做 二 次 开 发 代 码 改 动 代 码 改 动 首 先 在 原 基 础 的 c o b r a 上 我 们 做 一 个 拉 代 码 的 测 试 好 像 很 叼 的 样 子 ， 没 有 漏 洞 ， 但 是 有 层 s c r i p t 提 示 没 有 选 择 目 标
后 台 定 位 问 题 发 现 如 下 错 误 一 般 出 现 这 个 问 题 不 是 分 支 错 误 就 是 路 径 错 误 ， 反 正 是 找 不 到 文 件 的 ， 那 直 接 用 g i t   c l o n e 试 试 原 路 径 是 否 存 在 吧 5 0 0 ， 询 问 了 g i t l a b 负 责 人 才 知 道 是 根 本 没 有 开 放 h t t p 下 拉 代 码 的 功 能 ， 统 一 使 用 s s h 进 行 下 拉 。 那 原 基 础 的 c o b r a 能 直 接 用 s s h 拉 代 码 吗 ？ 试 试 看 好 了 提 示 请 输 入 U R L ， 也 就 是 输 入 格 式 不 对 ， 那 我 们 换 一 种 方 式 输 入
还 是 一 样 的 提 示 ， 那 看 看 后 端 是 什 么 情 况 吧 一 切 正 常 。 首 先 对 问 题 进 行 定 位 命 令 为 f i n d   . /   - n a m e   “ * . p y ” | x a r g s   g r e p   “ P l e a s e   i n p u t   a   v a l i d   U R L ” 命 令 意 思 为 在 本 级 目 录 级 下 级 目 录 搜 索 内 容 为 P l e a s e   i n p u t   a   v a l i d   U R L 的 p y 脚 本 找 到 了 a p i . p y ， 让 我 们 进 去 看 看 这 是 什 么
果 然 只 限 定 了 h t t p 与 h t t p s ， 那 根 据 咱 们 的 需 求 加 上 s s h 吧 原 代 码 ： 改 动 后 代 码 ：     i f   r e . m a t c h ( r ' h t t p : / / | h t t p s : / / ' ,   t ) :                                         a r g   =   ( t ,   f o r m a t t e r ,   o u t p u t ,   r u l e ,   a _ s i d ,   i s _ d e l )                                         p r o d u c e r ( t a s k = a r g )                                 e l s e :                                         r e t u r n   { " c o d e " :   1 0 0 4 ,   " m s g " :   " P l e a s e   i n p u t   a   v a l i d   U R L " }                         r e s u l t   =   {                                 ' m s g ' :   ' A d d   s c a n   j o b   s u c c e s s f u l l y . ' ,                                 ' s i d ' :   a _ s i d ,                                 ' t o t a l _ t a r g e t _ n u m ' :   l e n ( t a r g e t ) ,                         }                 e l s e :                         i f   r e . m a t c h ( r ' h t t p : / / | h t t p s : / / ' ,   t a r g e t ) :                                 a r g   =   ( t a r g e t ,   f o r m a t t e r ,   o u t p u t ,   r u l e ,   a _ s i d ,   i s _ d e l )                                 p r o d u c e r ( t a s k = a r g )                         e l s e :                                 r e t u r n   { " c o d e " :   1 0 0 4 ,   " m s g " :   " P l e a s e   i n p u t   a   v a l i d   U R L " }         i f   r e . m a t c h ( r ' h t t p : / / | h t t p s : / / | s s h : / / ' ,   t ) :                                         a r g   =   ( t ,   f o r m a t t e r ,   o u t p u t ,   r u l e ,   a _ s i d ,   i s _ d e l )                                         p r o d u c e r ( t a s k = a r g )                                 e l s e :                                         r e t u r n   { " c o d e " :   1 0 0 4 ,   " m s g " :   " P l e a s e   i n p u t   a   v a l i d   U R L " }                         r e s u l t   =   {                                 ' m s g ' :   ' A d d   s c a n   j o b   s u c c e s s f u l l y . ' ,                                 ' s i d ' :   a _ s i d ,                                 ' t o t a l _ t a r g e t _ n u m ' :   l e n ( t a r g e t ) ,                         }                 e l s e :
重 新 试 一 下 吧 后 台 提 示 需 要 输 入 r o o t 密 码 我 的 环 境 是 做 了 g i t @ c o d e . x x x 的 免 密 推 送 ， r o o t 用 户 密 码 我 不 知 道 啊 ， 就 算 有 权 限 难 道 把 密 码 写 在 配 置 文 件 里 又 安 全 吗 ？ 明 文 密 码 泄 露 的 事 情 可 是 发 生 不 少 ， 于 是 继 续 看 代 码 找 问 题 看 他 的 推 送 方 式 是 什 么 配 置 文 件 为 . / c o b r a / p i c k u p . p y 如 果 配 置 文 件 中 没 有 输 入 用 户 名 或 密 码 ， 便 是 公 开 链 接 ， 直 接 c l o n e ， 如 果 有 用 户 名 密 码 ， 则 分 割 填 入 用 户 名 和 密 码 进 行 加 密 c l o n e ， 如 果 这 放 在 h t t p 协 议 中 这 个 逻 辑 完 全 没 有 问 题 ， 但 是 放 在 s s h 下 拉 代 码 里 ， 就 会 存 在 很 大 的 逻 辑 问 题 了                         i f   r e . m a t c h ( r ' h t t p : / / | h t t p s : / / | s s h : / / ' ,   t a r g e t ) :                                 a r g   =   ( t a r g e t ,   f o r m a t t e r ,   o u t p u t ,   r u l e ,   a _ s i d ,   i s _ d e l )                                 p r o d u c e r ( t a s k = a r g )                         e l s e :                                 r e t u r n   { " c o d e " :   1 0 0 4 ,   " m s g " :   " P l e a s e   i n p u t   a   v a l i d   U R L " } f i n d   . /   - n a m e   " * . p y "   | x a r g s   g r e p   " g i t   c l o n e "
逻 辑 问 题 在 于 s s h 的 免 密 钥 登 录 不 需 要 密 码 ， 如 果 单 纯 以 用 户 名 密 码 作 为 判 断 依 据 那 免 密 钥 的 作 用 就 为 零 了 ， 为 了 适 配 免 密 钥 ， 实 际 上 在 代 码 中 指 定 免 密 钥 的 用 户 即 可 ， 同 时 为 保 证 其 他 基 础 功 能 不 遭 受 破 坏 ， 增 加 的 功 能 应 使 用 判 断 前 缀 的 方 式 进 行 代 码 改 写 ， 具 体 如 下 ： 原 代 码 ： 改 后 代 码 ：                   i f   s e l f . r e p o _ u s e r n a m e   i s   N o n e   o r   s e l f . r e p o _ p a s s w o r d   i s   N o n e :                         #   p u b l i c   r e p o                         c l o n e _ a d d r e s s   =   s e l f . r e p o _ a d d r e s s                 e l s e :                         #   p r i v a t e   r e p o                         c l o n e _ a d d r e s s   =   s e l f . r e p o _ a d d r e s s . s p l i t ( ' : / / ' ) [ 0 ]   +   ' : / / '   +   q u o t e ( s e l f . r e p o _ u s e r n a m e )   +   ' : '   +                                                           q u o t e ( s e l f . r e p o _ p a s s w o r d )   +   ' @ '   +   s e l f . r e p o _ a d d r e s s . s p l i t ( ' : / / ' ) [ 1 ]                 #   c l o n e   r e p o   w i t h   u s e r n a m e   a n d   p a s s w o r d                                 #   " h t t p [ s ] : / / u s e r n a m e : p a s s w o r d @ g i t l a b . c o m / u s e r n a m e / r e p o n a m e "                 #   ! ! !   i f   a d d   p a s s w o r d   i n   t h e   u r l ,   . g i t / c o n f i g   w i l l   l o g   y o u r   u r l   w i t h   p a s s w o r d                 c m d   =   ' g i t   c l o n e   '   +   c l o n e _ a d d r e s s   +   '   " '   +   s e l f . r e p o _ d i r e c t o r y   +   ' "   - b   '   +   s e l f . r e p o _ b r a n c h           i f   s e l f . r e p o _ u s e r n a m e   i s   N o n e   o r   s e l f . r e p o _ p a s s w o r d   i s   N o n e :                         #   p u b l i c   r e p o                         i f   ( s e l f . r e p o _ a d d r e s s . s p l i t ( ' : / / ' ) [ 0 ]   = =   ' s s h ' ) :                                 c l o n e _ a d d r e s s   =   ' s s h : / / '   +   ' g i t @ '   +                                   s e l f . r e p o _ a d d r e s s . s p l i t ( ' : / / ' ) [ 1 ]                         e l s e :                         c l o n e _ a d d r e s s   =   s e l f . r e p o _ a d d r e s s                 e l s e :                         #   p r i v a t e   r e p o                         i f   ( s e l f . r e p o _ a d d r e s s . s p l i t ( ' : / / ' ) [ 0 ]   = =   ' s s h ' ) :                                 c l o n e _ a d d r e s s   =   ' s s h : / / '   +   ' g i t @ '   +                                   s e l f . r e p o _ a d d r e s s . s p l i t ( ' : / / ' ) [ 1 ]                         e l s e :                         c l o n e _ a d d r e s s   =   s e l f . r e p o _ a d d r e s s . s p l i t ( ' : / / ' ) [ 0 ]   +   ' : / / '   +   q u o t e ( s e l f . r e p o _ u s e r n a m e )   +   ' : '   +                                                           q u o t e ( s e l f . r e p o _ p a s s w o r d )   +   ' @ '   +   s e l f . r e p o _ a d d r e s s . s p l i t ( ' : / / ' ) [ 1 ]                 #   c l o n e   r e p o   w i t h   u s e r n a m e   a n d   p a s s w o r d                 #   " h t t p [ s ] : / / u s e r n a m e : p a s s w o r d @ g i t l a b . c o m / u s e r n a m e / r e p o n a m e "
增 加 前 缀 判 断 ， 如 果 前 缀 为 s s h ， 则 统 一 使 用 通 用 用 户 g i t 进 行 s s h 登 录 代 码 拉 取 进 行 测 试 进 行 测 试 如 此 完 成 了 一 次 很 简 单 的 模 块 二 次 开 发 。 二 次 开 发 重 要 是 需 求 与 实 现 ， 将 自 己 的 需 求 完 整 的 实 现 出 来 即 可 ， 不 一 定 是 框 架 意 义 上 的 开 发 才 算 二 次 开 发 ， 一 定 要 明 白 自 己 需 要 的 是 什 么 ， 代 码 逻 辑 是 什 么 ， 掌 握 好 代 码 逻 辑 ， 二 次 开 发 也 不 算 是 很 难 的 事 情 。                 #   ! ! !   i f   a d d   p a s s w o r d   i n   t h e   u r l ,   . g i t / c o n f i g   w i l l   l o g   y o u r   u r l   w i t h   p a s s w o r d                 c m d   =   ' g i t   c l o n e   '   +   c l o n e _ a d d r e s s   +   '   " '   +   s e l f . r e p o _ d i r e c t o r y   +   ' "   - b   '   +   s e l f . r e p o _ b r a n c h
* 本 文 原 创 作 者 ： 本 文 原 创 作 者 ： 煜 阳 煜 阳 y u y a n g ， 本 文 属 于 ， 本 文 属 于 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. 商业源码建议获取授权，如侵犯您的权益，请联系客服处理