搜索

[13024] 2018-08-05_一个FreeBSD下的通信协议监控程序

文档创建者：s7ckTeam
浏览次数：2
最后更新：2025-01-18
2018-08-05_一个FreeBSD下的通信协议监控程序 一 个 F r e e B S D 下 的 通 信 协 议 监 控 程 序 原 创   r o c h e k   F r e e B u f   2 0 1 8 - 0 8 - 0 5 这 是 一 个 简 易 的 通 信 协 议 的 这 是 一 个 简 易 的 通 信 协 议 的 h o o k ， 可 以 在 ， 可 以 在 F r e e B S D 1 1 上 使 用 。 上 使 用 。 简 介 简 介 最 近 想 要 在 一 个 内 核 里 的 小 聊 天 服 务 器 练 手 用 ， 现 在 写 完 的 内 容 ， 可 以 作 为 一 个 通 讯 协 议 的 监 控 程 序 使 用 。 这 里 是 我 的 B L O G ， 如 果 这 个 项 目 有 更 新 ， 应 该 会 发 在 这 里 。 B L O G 使 用 H u x 的 模 板 做 的 G i t h u b   p a g e s ， 轻 拍 （ 捂 脸 ） 使 用 的 原 理 很 简 单 ， 替 换 掉 原 有 的 协 议 处 理 函 数 ， 把 协 议 载 荷 内 的 信 息 打 印 到 d m e s g 内 。 现 在 仅 仅 处 理 了 U D P 的 内 容 。 T C P ， I C M P 稍 作 改 动 应 该 也 可 以 监 控 。 实 现 实 现 1 .   替 换 通 信 协 议 的 处 理 函 数 替 换 通 信 协 议 的 处 理 函 数 在 F r e e B S D 内 核 内 ， 通 信 协 议 的 p r o t o s w 是 这 样 定 义 的 。 I C M P T C P U D P {         . p r _ t y p e   =                 S O C K _ R A W ,         . p r _ d o m a i n   =                 & i n e t d o m a i n ,         . p r _ p r o t o c o l   =                 I P P R O T O _ I C M P ,         . p r _ f l a g s   =                 P R _ A T O M I C | P R _ A D D R | P R _ L A S T H D R ,         . p r _ i n p u t   =                 i c m p _ i n p u t ,         . p r _ c t l o u t p u t   =                 r i p _ c t l o u t p u t ,         . p r _ u s r r e q s   =                 & r i p _ u s r r e q s } , {         . p r _ t y p e   =                 S O C K _ S T R E A M ,         . p r _ d o m a i n   =                 & i n e t d o m a i n ,         . p r _ p r o t o c o l   =                 I P P R O T O _ T C P ,         . p r _ f l a g s   =                 P R _ C O N N R E Q U I R E D | P R _ I M P L O P C L | P R _ W A N T R C V D ,         . p r _ i n p u t   =                 t c p _ i n p u t ,         . p r _ c t l i n p u t   =                 t c p _ c t l i n p u t ,         . p r _ c t l o u t p u t   =                 t c p _ c t l o u t p u t ,         . p r _ i n i t   =                 t c p _ i n i t ,         . p r _ s l o w t i m o   =                 t c p _ s l o w t i m o ,         . p r _ d r a i n   =                 t c p _ d r a i n ,         . p r _ u s r r e q s   =                 & t c p _ u s r r e q s } ,
想 要 获 取 通 信 协 议 m b u f 内 的 信 息 ， 替 换 掉 p r _ i n p u t 即 可 。 一 般 的 做 法 是 在 内 核 组 件 载 入 时 ， 把 通 信 协 议 对 应 的 p r _ i n p u t 替 换 ， 在 卸 载 时 还 原 。 当 然 也 可 以 写 一 个 应 用 层 控 制 程 序 ， 向 这 个 内 核 模 块 发 送 某 段 数 据 时 ， 内 核 模 块 才 替 换 p r _ i n p u t 2 .   设 计 设 计 h o o k 函 数 函 数 作 为 h o o k 时 使 用 的 函 数 ， 需 要 和 原 有 函 数 相 同 的 定 义 。 例 如 这 样 ： 经 过 以 上 这 两 段 ， U D P 报 文 的 处 理 函 数 就 直 接 进 入 了 我 们 自 定 义 的 函 数 u d p _ i n p u t _ h o o k ( ) 内 了 。 作 为 一 个 通 信 协 议 的 处 理 函 数 ， 首 先 需 要 处 理 m b u f 内 U D P 的 信 息 确 实 存 在 。 为 了 保 证 之 后 有 U D P 信 息 ， U D P 头 和 I P 头 的 长 度 一 定 要 保 证 下 来 。 剩 余 长 度 不 满 足 U D P 头 的 话 ， 接 下 来 就 没 办 法 处 理 了 ， 所 以 这 里 把 数 据 返 回 。 然 后 ， 获 取 U D P 头 可 以 从 U D P 头 内 获 取 一 些 所 需 要 的 信 息 ， 结 合 在 上 一 步 获 取 的 i p 头 ， 就 可 以 获 取 到 这 个 包 的 具 体 信 息 了 。 例 如 发 送 地 址 ， 源 端 口 号 ， 目 的 端 口 号 等 等 。 可 以 把 这 些 都 记 录 下 来 ， 以 后 扩 展 功 能 用 。 再 剥 离 头 {         . p r _ t y p e   =                   S O C K _ D G R A M ,         . p r _ d o m a i n   =                   & i n e t d o m a i n ,         . p r _ p r o t o c o l   =                   I P P R O T O _ U D P ,         . p r _ f l a g s   =                   P R _ A T O M I C | P R _ A D D R ,         . p r _ i n p u t   =                   u d p _ i n p u t ,         . p r _ c t l i n p u t   =                   u d p _ c t l i n p u t ,         . p r _ c t l o u t p u t   =           i p _ c t l o u t p u t ,         . p r _ i n i t   =                   u d p _ i n i t ,         . p r _ u s r r e q s   =                   & u d p _ u s r r e q s } , c a s e   M O D _ L O A D :                 / *   R e p l a c e   u d p _ i n p u t   w i t h   u d p _ i n p u t _ h o o k .   * /                 i n e t s w [ i p _ p r o t o x [ I P P R O T O _ U D P ] ] . p r _ i n p u t   =   u d p _ i n p u t _ h o o k ;                 b r e a k ; i n t u d p _ i n p u t _ h o o k ( s t r u c t   m b u f   * * m p ,   i n t   * o f f p ,   i n t   p r o t o ) ; i f   ( i p h l e n   >   s i z e o f   ( s t r u c t   i p ) )   {         i p _ s t r i p o p t i o n s ( m ) ;         i p h l e n   =   s i z e o f ( s t r u c t   i p ) ; } i p   =   m t o d ( m ,   s t r u c t   i p   * ) ; i f   ( m - > m _ l e n   <   i p h l e n   +   s i z e o f ( s t r u c t   u d p h d r ) )   {         i f   ( ( m   =   m _ p u l l u p ( m ,   i p h l e n   +   s i z e o f ( s t r u c t   u d p h d r ) ) )   = =   N U L L )   {                 U D P S T A T _ I N C ( u d p s _ h d r o p s ) ;                 r e t u r n   ( I P P R O T O _ D O N E ) ;         }         i p   =   m t o d ( m ,   s t r u c t   i p   * ) ; } u h   =   ( s t r u c t   u d p h d r   * ) ( ( c a d d r _ t ) i p   +   i p h l e n ) ; m - > m _ l e n   - =   i p h l e n   +   s i z e o f ( s t r u c t   u d p h d r ) ; m - > m _ d a t a   + =   i p h l e n   +   s i z e o f ( s t r u c t   u d p h d r ) ;
这 样 ， 当 前 m _ d a t a 的 位 置 就 是 U D P 数 据 区 的 位 置 了 。 这 个 位 置 就 是 U D P 载 荷 的 信 息 。 之 前 说 的 ， 直 接 打 印 到 d m e s g 内 。 为 了 让 h o o k 后 的 通 讯 协 议 正 常 工 作 ， 需 要 把 m b u f 返 回 给 原 来 的 处 理 函 数 。 否 则 这 个 机 器 的 U D P 协 议 就 没 法 正 常 运 行 了 。 首 先 ， 需 要 恢 复 m b u f 。 再 返 回 给 原 处 理 函 数 后 记 后 记 这 样 ， 就 完 成 了 把 每 次 p r _ i n p u t 都 打 印 出 的 一 个 内 核 协 议 的 h o o k 。 因 为 最 后 返 回 给 了 原 有 的 协 议 栈 ， 所 以 不 影 响 原 有 处 理 功 能 。 这 个 h o o k ， 我 原 本 的 计 划 是 作 为 一 个 非 阻 塞 的 r e c i v e _ f r o m ( ) ， 稍 作 改 动 ， 可 以 做 其 它 事 情 。 比 如 ， 接 着 写 下 去 ， 做 一 个 基 于 U D P 的 特 殊 协 议 等 等 。 本 文 作 者 ： 本 文 作 者 ： r o c h e k ， 本 文 属 ， 本 文 属   F r e e B u f   原 创 奖 励 计 划 ， 未 经 许 可 禁 止 转 载 。 原 创 奖 励 计 划 ， 未 经 许 可 禁 止 转 载 。 阅 读 原 文 p r i n t f ( " l o a d = % s   n " ,   m - > m _ d a t a ) ; m - > m _ l e n   + =   i p h l e n   +   s i z e o f ( s t r u c t   u d p h d r ) ; m - > m _ d a t a   - =   i p h l e n   +   s i z e o f ( s t r u c t   u d p h d r ) ; r e t u r n   u d p _ i n p u t ( m p ,   o f f p ,   p r o t o ) ;
回复

举报

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

本版积分规则

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

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