网络安全渗透测试【免费下载链接】traitor:arrow_up: :skull_and_crossbones: :fire: Automatic Linux privesc via exploitation of low-hanging fruit e.g. gtfobins, pwnkit, dirty pipe, w docker.sock项目地址https://gitcode.com/gh_mirrors/tr/traitor点击查看免费下载Cobra 是 Go 生态中最流行的命令行CLI应用开发框架用于构建功能强大、风格现代的命令行程序被 Kubernetes、Hugo、GitHub CLI 等大量知名项目采用。本文以当前仓库 traitor一个自动化的 Linux 提权框架对 Cobra 的实际使用为实例完整讲解 Cobra 的核心概念、安装方式、命令/标志/参数体系、帮助与补全机制以及源码级的最佳实践。读完本文你将能够独立使用 Cobra 搭建多级子命令、持久化标志、参数校验完备的 Go CLI 应用并理解 traitor 项目中traitor、traitor list、traitor backdoor install等命令背后的实现原理。Cobra 概览能为你做什么Cobra 是一个提供简洁接口来创建强大现代 CLI 的 Go 库其设计目标是与git、go等工具保持一致的交互体验。根据 Cobra 官方 README它开箱即用地提供基于子命令的 CLIapp server、app fetch这类自然语言式调用完全 POSIX 兼容的标志同时支持短标志与长标志如-a与--any嵌套子命令命令之下还可以挂载子命令全局、局部与级联标志灵活控制标志的作用范围智能建议输入app srver时提示 did you meanapp server?自动帮助生成为命令与标志自动生成帮助文本自动识别帮助标志-h、--help等开箱即用自动生成 Shell 补全脚本支持 bash、zsh、fish、PowerShell自动生成 man 手册命令别名在不破坏现有使用方式的前提下调整命令名自定义帮助与用法可覆盖默认的 help、usage 模板可选与 Viper 无缝集成支持 12-factor 应用风格的配置管理。核心概念命令、参数与标志Cobra 建立在三个基本概念之上Commands命令表示动作actionArgs参数表示被操作的事物thingFlags标志是这些动作的修饰符。最好的 CLI 应用在使用时读起来像一句英文句子用户凭直觉就知道如何交互。遵循的模式是APPNAME VERB NOUN --ADJECTIVE或APPNAME COMMAND ARG --FLAG原文档给出了两个真实世界的经典例子hugo server --port1313其中server是命令port是标志git clone URL --bare告诉 Git 以 bare 方式克隆 URL。Commands应用的中心Command 是整个应用的中枢。应用支持的每次交互都包含在一个 Command 中一个命令可以拥有子命令children commands也可以选择性地执行一个动作Run 函数。上面例子中的server就是一个命令。在 traitor 项目中list、backdoor、setuid都是根命令traitor的子命令而backdoor之下又嵌套了install与uninstall构成典型的嵌套命令结构。Flags行为修饰符标志用于修改命令的行为。Cobra 支持完全 POSIX 兼容的标志同时也兼容 Go 标准库 flag 包。Cobra 命令可以定义两类标志会持久化到子命令的标志persistent flags仅对当前命令生效的局部标志local flags。标志功能由pflag 库提供——它是 Go 标准库flag的一个分支在保持相同接口的基础上增加了 POSIX 兼容性pflag 在 traitor 项目的 go.mod 中作为间接依赖出现。安装 Cobra使用 Cobra 非常简单。首先用go get安装最新版本go get -u github.com/spf13/cobralatest然后在应用中引入import github.com/spf13/cobra在 traitor 项目的 go.mod 中Cobra 以github.com/spf13/cobra v1.5.0作为直接依赖被引入项目运行环境为 Go 1.18。快速开始生成器与手动搭建两种路径方式一使用 cobra-cli 生成器cobra-cli是用于生成 Cobra 应用与命令文件的命令行程序它会自动搭建应用脚手架是让 Cobra 进入你的项目最快的方式go install github.com/spf13/cobra-clilatest生成器安装完成后可以使用cobra-cli add command添加命令也可以直接生成完整应用骨架。生成器与库的完整用法分别见 Cobra Generator 说明 与 Cobra User Guide后者是深入使用 Cobra 库的权威指南本文后续内容大量源于它。方式二手动搭建traitor 采用的方式虽然可以使用生成器但完全手动搭建也非常简单——Cobra 不要求任何特殊构造函数直接创建命令即可。一个典型的 Cobra 应用目录结构如下▾ appName/ ▾ cmd/ add.go your.go commands.go here.go main.go在这种结构中main.go通常非常单薄只承担一个职责初始化并执行 Cobra。traitor 的 cmd/traitor/main.go 就是一个标准范例package main import ( github.com/liamg/traitor/internal/cmd ) func main() { cmd.Execute() }根命令则通常放在cmd/root.gotraitor 中为 internal/cmd/root.govar rootCmd cobra.Command{ Use: hugo, Short: Hugo is a very fast static site generator, Long: A Fast and Flexible Static Site Generator built with love by spf13 and friends in Go. Complete documentation is available at https://gohugo.io/documentation/, Run: func(cmd *cobra.Command, args []string) { // Do Stuff Here }, } func Execute() { if err : rootCmd.Execute(); err ! nil { fmt.Fprintln(os.Stderr, err) os.Exit(1) } }注意Execute()的模式执行出错时把错误打印到 stderr 并以非零码退出。traitor 的Execute()internal/cmd/root.go正是这样实现的func Execute() { if err : rootCmd.Execute(); err ! nil { _, _ fmt.Fprintln(os.Stderr, err) os.Exit(1) } }附加的命令各自独立成文件放在cmd/目录下并通过init()注册。例如创建一个version命令package cmd import ( fmt github.com/spf13/cobra ) func init() { rootCmd.AddCommand(versionCmd) } var versionCmd cobra.Command{ Use: version, Short: Print the version number of Hugo, Long: All software has versions. This is Hugos, Run: func(cmd *cobra.Command, args []string) { fmt.Println(Hugo Static Site Generator v0.9 -- HEAD) }, }源码级剖析traitor 的 Cobra 实战把上面的模式应用到 traitor 的真实代码上可以看到一个生产级 Cobra 应用的全貌。根命令与持久化标志traitor 的根命令定义在 internal/cmd/root.go使用cobra.ExactArgs(0)强制不接受任何位置参数并通过PreRun钩子在正式执行前打印 ASCII 标题横幅var rootCmd cobra.Command{ Use: traitor, Short: Traitor is a privilege escalation framework for Linux, Long: An extensible privilege escalation framework for Linux Complete documentation is available at https://github.com/liamg/traitor, Args: cobra.ExactArgs(0), PreRun: func(_ *cobra.Command, args []string) { // 打印版本横幅 }, Run: func(cmd *cobra.Command, args []string) { // 评估系统状态并检测可利用漏洞 }, }根命令的四个标志全部用PersistentFlags()注册因此对list、backdoor、setuid等所有子命令都可见internal/cmd/root.gofunc init() { rootCmd.PersistentFlags().BoolVarP(runAnyExploit, any, a, runAnyExploit, Attempt to exploit a vulnerability as soon as it is detected. Provides a shell where possible.) rootCmd.PersistentFlags().BoolVarP(promptForPassword, with-password, p, promptForPassword, Prompt for the user password, if you know it. Can provide more GTFOBins possibilities via sudo.) rootCmd.PersistentFlags().StringVarP(exploitName, exploit, e, exploitName, Run the specified exploit, if the system is found to be vulnerable. Provides a shell where possible.) rootCmd.PersistentFlags().StringSliceVarP(skipExploits, skip, k, skipExploits, Exploit(s) to skip - specify multiple times to skip multiple exploits.) }这里可以看到BoolVarP/StringVarP/StringSliceVarP的典型用法把标志值绑定到包级变量同时提供短标志-a、-p、-e、-k、默认值与帮助文本。其中--exploit的值与 pkg/exploits/registration.go 中注册的漏洞名称一一对应如docker:writable-socket、polkit:CVE-2021-3560、polkit:CVE-2021-4034、kernel:CVE-2022-0847见 pkg/exploits/all.go运行时通过exploitName exploit.Name精确匹配要执行的漏洞。子命令的注册listinternal/cmd/list.go 演示了最简子命令——在init()中rootCmd.AddCommand(listCmd)Run里遍历所有已注册漏洞并打印名称var listCmd cobra.Command{ Use: list, Short: List all available methods., Run: func(cmd *cobra.Command, args []string) { allExploits : exploits.Get(exploits.SpeedAny) for _, exploit : range allExploits { fmt.Println(exploit.Name) } }, }嵌套子命令backdoor install / uninstallinternal/cmd/backdoor.go 展示了嵌套子命令与参数上限校验backdoor命令本身没有Run只有子命令其子命令install与uninstall通过cobra.MaximumNArgs(1)限制最多接收一个路径参数func init() { backdoorCmd.AddCommand(backdoorInstallCmd) backdoorCmd.AddCommand(backdoorUninstallCmd) rootCmd.AddCommand(backdoorCmd) } var backdoorCmd cobra.Command{ Use: backdoor, Short: Install a root shell backdoor, } var backdoorInstallCmd cobra.Command{ Use: install [path], Short: Install a root shell backdoor, Args: cobra.MaximumNArgs(1), Run: func(cmd *cobra.Command, args []string) { // ... }, }局部标志setuid与持久化标志相对internal/cmd/setuid.go 用setuidCmd.Flags()注册仅属于该命令的局部标志--shell/-s与--cmd/-c并通过rootCmd.AddCommand(setuidCmd)挂载func init() { setuidCmd.Flags().StringVarP(setuidShell, shell, s, setuidShell, Path to shell to execute, e.g. /bin/bash.) setuidCmd.Flags().StringVarP(setuidShellCmd, cmd, c, setuidShellCmd, Shell command to execute - leave blank to be dropped in interactive shell.) rootCmd.AddCommand(setuidCmd) }综合来看traitor 的命令树是traitor ├── (无子命令时的默认行为检测漏洞) ├── list # 列出所有可利用方法 ├── backdoor │ ├── install [path] # 安装 root shell 后门 │ └── uninstall [path] # 卸载后门 └── setuid # 以 setuid 方式包装 shell深入标志系统持久标志Persistent Flags与全局标志持久标志对该命令及其所有子孙命令可用。需要全局标志时把它挂到根命令上即可traitor 的-a/-p/-e/-k就是标准做法rootCmd.PersistentFlags().BoolVarP(Verbose, verbose, v, false, verbose output)局部标志Local Flags局部标志只对特定命令生效localCmd.Flags().StringVarP(Source, source, s, , Source directory to read from)父命令上的局部标志默认情况下Cobra 只解析目标命令上的局部标志父命令上的局部标志会被忽略。通过设置Command.TraverseChildren: trueCobra 会在执行目标命令前依次解析每个命令上的局部标志command : cobra.Command{ Use: print [OPTIONS] [COMMANDS], TraverseChildren: true, }与配置绑定Viper 集成Cobra 可以把标志与 Viper 绑定实现配置文件与命令行参数的统一管理var author string func init() { rootCmd.PersistentFlags().StringVar(author, author, YOUR NAME, Author name for copyright attribution) viper.BindPFlag(author, rootCmd.PersistentFlags().Lookup(author)) }注意当用户显式提供--author标志时变量author不再取配置文件中的值命令行优先级更高。必填标志标志默认都是可选的。如果希望未设置标志时报错可以标记为必填rootCmd.Flags().StringVarP(Region, region, r, , AWS region (required)) rootCmd.MarkFlagRequired(region)持久标志同理rootCmd.PersistentFlags().StringVarP(Region, region, r, , AWS region (required)) rootCmd.MarkPersistentFlagRequired(region)标志组Flag Groups当某些标志必须同时出现时例如提供了--username就必须提供--passwordCobra 可以强制执行rootCmd.Flags().StringVarP(u, username, u, , Username (required if password is set)) rootCmd.Flags().StringVarP(pw, password, p, , Password (required if username is set)) rootCmd.MarkFlagsRequiredTogether(username, password)也可以禁止互斥标志同时出现例如输出格式要么--json要么--yaml但绝不能同时给rootCmd.Flags().BoolVar(u, json, false, Output in JSON) rootCmd.Flags().BoolVar(pw, yaml, false, Output in YAML) rootCmd.MarkFlagsMutuallyExclusive(json, yaml)两种场景的共同规则局部标志与持久标志都可以参与标志组标志组只在所有成员标志都被定义的命令上强制生效一个标志可以属于多个组一个组可以包含任意数量的标志。位置参数与自定义校验位置参数的校验通过Command的Args字段指定。若Args未定义或为nil默认采用ArbitraryArgs接受任意参数。Cobra 内置了以下校验器校验器行为NoArgs存在任何位置参数即报错ArbitraryArgs接受任意参数OnlyValidArgs位置参数不在ValidArgs字段中即报错MinimumNArgs(int)位置参数少于 N 个即报错MaximumNArgs(int)位置参数多于 N 个即报错ExactArgs(int)位置参数不是恰好 N 个即报错ExactValidArgs(int)位置参数不是恰好 N 个或存在非法参数即报错RangeArgs(min, max)位置参数数量不在 [min, max] 区间即报错MatchAll(pargs ...PositionalArgs)组合多个校验如同时校验 ExactArgs 长度与其他性质traitor 中同时用到了ExactArgs(0)根命令拒绝位置参数与MaximumNArgs(1)backdoor 子命令限制路径参数数量。自定义校验器示例——要求第一个参数是合法颜色var cmd cobra.Command{ Short: hello, Args: func(cmd *cobra.Command, args []string) error { if len(args) 1 { return errors.New(requires a color argument) } if myapp.IsValidColor(args[0]) { return nil } return fmt.Errorf(invalid color specified: %s, args[0]) }, Run: func(cmd *cobra.Command, args []string) { fmt.Println(Hello, World!) }, }帮助命令与用法信息自动生成的帮助当应用存在子命令时Cobra 会自动添加help命令用户运行app help即可调用app help create也能显示任意命令的帮助。每个命令还会自动获得--help标志。以下输出完全由 Cobra 自动生成无需任何额外配置$ cobra help Cobra is a CLI library for Go that empowers applications. This application is a tool to generate the needed files to quickly create a Cobra application. Usage: cobra [command] Available Commands: add Add a command to a Cobra Application help Help about any command init Initialize a Cobra Application Flags: -a, --author string author name for copyright attribution (default YOUR NAME) --config string config file (default is $HOME/.cobra.yaml) -h, --help help for cobra -l, --license string name of license for the project --viper use Viper for configuration (default true) Use cobra [command] --help for more information about a command.help 本身只是一个普通命令没有任何特殊逻辑完全可以自定义。自定义帮助可以通过以下函数提供自己的 help 命令、help 函数或 help 模板后两者对子命令同样生效cmd.SetHelpCommand(cmd *Command) cmd.SetHelpFunc(f func(*Command, []string)) cmd.SetHelpTemplate(s string)用法信息Usage当用户输入无效标志或无效命令时Cobra 会展示用法信息默认帮助内嵌了用法部分$ cobra --invalid Error: unknown flag: --invalid Usage: cobra [command] ...同样可以覆盖用法函数与模板cmd.SetUsageFunc(f func(*Command) error) cmd.SetUsageTemplate(s string)版本标志与生命周期钩子版本标志只要在根命令上设置了Version字段Cobra 就会自动添加顶层--version标志运行时按版本模板把版本打印到 stdout。模板可通过cmd.SetVersionTemplate(s string)定制。PreRun / PostRun 钩子可以在命令的Run前后挂接函数。PersistentPreRun与PreRun在Run之前执行PersistentPostRun与PostRun在Run之后执行Persistent*Run系列会被未声明自己的子命令继承。执行顺序固定为PersistentPreRunPreRunRunPostRunPersistentPostRun需要注意当执行子命令时会运行根命令的PersistentPreRun但不会运行根命令的PersistentPostRun后者仅作用于声明它的命令自身。traitor 根命令正是利用PreRun钩子在每次执行前打印版本横幅属于这一机制的直接应用。未知命令的智能建议当发生 unknown command 错误时Cobra 会自动给出建议行为与git命令类似$ hugo srever Error: unknown command srever for hugo Did you mean this? server Run hugo --help for usage.建议基于所有已注册子命令自动生成使用 Levenshtein 距离算法忽略大小写且最小距离为 2 的命令才会被展示。可以关闭建议或调整距离阈值command.DisableSuggestions truecommand.SuggestionsMinimumDistance 1也可以用SuggestFor属性为特定字符串显式指定建议对象——适用于字符串距离不近、但从语义上应当被建议的场景如kubectl remove时建议delete以及不想用别名但希望给出建议的情况。文档生成与 Shell 补全生成命令文档Cobra 可以基于子命令、标志等信息自动生成文档详见 Cobra 文档生成说明。生成 Shell 补全Cobra 可以生成 bash、zsh、fish、PowerShell 四种 Shell 的补全文件详细指南见 Shell Completions。Cobra 会自动为程序提供功能完整的completion命令类似自动提供的help命令也可以通过cobra-cli add completion创建自己的补全命令把脚本写到 stdout 以便最灵活地使用var completionCmd cobra.Command{ Use: completion [bash|zsh|fish|powershell], Short: Generate completion script, DisableFlagsInUseLine: true, ValidArgs: []string{bash, zsh, fish, powershell}, Args: cobra.ExactValidArgs(1), Run: func(cmd *cobra.Command, args []string) { switch args[0] { case bash: cmd.Root().GenBashCompletion(os.Stdout) case zsh: cmd.Root().GenZshCompletion(os.Stdout) case fish: cmd.Root().GenFishCompletion(os.Stdout, true) case powershell: cmd.Root().GenPowerShellCompletionWithDesc(os.Stdout) } }, }各 Shell 的加载方式摘自 Shell Completions 文档# Bash每次会话加载一次 $ source (traitor completion bash) # 或写入系统补全目录Linux: $ traitor completion bash /etc/bash_completion.d/traitor # Zsh需先启用 compinit $ echo autoload -U compinit; compinit ~/.zshrc $ traitor completion zsh ${fpath[1]}/_traitor # fish $ traitor completion fish | source $ traitor completion fish ~/.config/fish/completions/traitor.fish # PowerShell PS traitor completion powershell | Out-String | Invoke-Expression PS traitor completion powershell traitor.ps1自定义补全名词与标志值补全脚本默认自动处理命令与标志的补全但通过提供额外信息可以让补全更强大静态名词补全用ValidArgs字段提供预定义选项列表如kubectl get的pod、node、service名词别名用ArgAliases声明如rc是replicationcontrollers的别名别名不展示但可手动输入并被识别动态名词补全用ValidArgsFunction在运行时返回候选列表注意ValidArgs与ValidArgsFunction二选一返回值通过cobra.ShellCompDirective位字段控制行为如ShellCompDirectiveNoFileComp、ShellCompDirectiveNoSpace、ShellCompDirectiveFilterFileExt、ShellCompDirectiveFilterDirs标志值动态补全用command.RegisterFlagCompletionFunc()注册文件扩展名 / 目录过滤用MarkFlagFilename()、MarkFlagDirname()快捷实现。调试动态补全时可直接调用隐藏的__complete命令例如traitor __complete backdoor 补全代码中禁止直接向 stdout 打印调试信息会被当作补全候选应改用cobra.CompDebug/cobra.CompError系列函数。Active HelpCobra 还基于补全系统提供Active Help框架程序可以定义在使用过程中打印的提示、警告等消息帮助用户学习如何使用程序详见 Active Help 文档。Active Help 目前仅支持 bash需 bash completion V2且 bash 4.4与 zsh通过cobra.AppendActiveHelp(...)在ValidArgsFunction或标志补全函数中追加消息实现。用户可通过PROGRAM_ACTIVE_HELP环境变量控制其配置设置为 0 即全局关闭COBRA_ACTIVE_HELP0可一次性禁用所有基于 Cobra 的程序的 Active Help。许可证Cobra 以 Apache 2.0 许可证发布详见仓库内的 LICENSE.txt这也是它能够被 Kubernetes、Hugo 等大型开源项目广泛采用的原因之一。如果你想进一步了解有哪些项目在使用 Cobra可查看仓库内的 projects_using_cobra.md完整的库级使用指南见 Cobra User Guide补全机制细节见 Shell Completions 与 Active Help。赞分享网络安全渗透测试【免费下载链接】traitor:arrow_up: :skull_and_crossbones: :fire: Automatic Linux privesc via exploitation of low-hanging fruit e.g. gtfobins, pwnkit, dirty pipe, w docker.sock项目地址https://gitcode.com/gh_mirrors/tr/traitor点击查看免费下载相关推荐基于 Cobra 构建现代化 Go CLI以 k3sup 的命令行实现为实战范例基于 Cobra 构建现代化 Go CLI以 k3sup 的命令行实现为实战范例 Cobra 是 Go 生态中最流行的命令行应用开发库用于构建类似 git、云原生运维CLICobra 命令行框架实战指南在 syzkaller 仓库中构建健壮的 Go CLI 应用Cobra 命令行框架实战指南在 syzkaller 仓库中构建健壮的 Go CLI 应用 Cobra 是 Go 生态中最流行的命令行CLI框架之一被测试操作系统网络安全Cobra 入门与实战用 Go 构建现代子命令式 CLI以 CRI-O 仓库中的 Cobra v1.10.2 为例Cobra 入门与实战用 Go 构建现代子命令式 CLI以 CRI O 仓库中的 Cobra v1.10.2 为例 Cobra 是 Go 生态中广泛使用的云原生容器运行时创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考