42 lines
1.4 KiB
Go
42 lines
1.4 KiB
Go
//go:build !(windows && arm64)
|
||
|
||
package tcpkill
|
||
|
||
import (
|
||
"github.com/spf13/cobra"
|
||
"os"
|
||
"runtime"
|
||
)
|
||
|
||
var tck = &TCPKill{}
|
||
|
||
func init() {
|
||
Cmd.Flags().StringVarP(&tck.SrcIP, "src-ip", "s", "", "本地源IP")
|
||
Cmd.Flags().IntVarP(&tck.SrcPort, "src-port", "p", 0, "本地源端口")
|
||
Cmd.Flags().StringVarP(&tck.DstIP, "dst-ip", "d", "", "目标IP")
|
||
Cmd.Flags().IntVarP(&tck.DstPort, "dst-port", "P", 0, "目标端口")
|
||
Cmd.Flags().StringVarP(&tck.Status, "status", "S", "", "匹配的连接状态，如ESTABLISHED，CLOSE_WAIT等，为空则匹配所有")
|
||
Cmd.Flags().IntVarP(&tck.RstNumbers, "rst-numbers", "n", 3, "RST包数量")
|
||
Cmd.Flags().BoolVarP(&tck.WaitMode, "wait", "w", false, "等待模式")
|
||
Cmd.Flags().StringVarP(&tck.KillType, "kill-type", "t", "both", "RST通知类型,both=都通知 target=目标地址通知 reverse=来源地址通知")
|
||
if runtime.GOOS != "windows" {
|
||
Cmd.Flags().BoolVarP(&tck.AutoIptables, "auto-iptables", "a", true, "自动设置iptables")
|
||
Cmd.Flags().IntVarP(&tck.NFQNums, "nfq-nums", "q", 0, "NFQ队列号")
|
||
} else {
|
||
Cmd.Flags().StringVarP(&tck.Eth, "eth", "e", "", "网卡")
|
||
Cmd.Flags().StringVarP(&tck.BPF, "bpf", "b", "", "BPF过滤")
|
||
Cmd.Flags().StringVarP(&tck.Host, "host", "i", "", "主机")
|
||
}
|
||
}
|
||
|
||
var Cmd = &cobra.Command{
|
||
Use: "tcpkill",
|
||
Short: "杀掉那个TCP连接",
|
||
Run: func(cmd *cobra.Command, args []string) {
|
||
err := tck.Run()
|
||
if err != nil {
|
||
os.Exit(1)
|
||
}
|
||
},
|
||
}
|