From f8533738a9e9032d07270a09fe6dfdc2e19f64d1 Mon Sep 17 00:00:00 2001 From: v-byte-cpu <65545655+v-byte-cpu@users.noreply.github.com> Date: Tue, 9 Mar 2021 22:50:59 +0300 Subject: [PATCH] Add ARP scan --- .gitignore | 1 + README.md | 20 ++ command/arp.go | 145 ++++++++++++ command/root.go | 25 +++ go.mod | 15 ++ go.sum | 339 +++++++++++++++++++++++++++++ main.go | 7 + pkg/ip/ip.go | 59 +++++ pkg/ip/ip_test.go | 99 +++++++++ pkg/packet/afpacket/readwriter.go | 59 +++++ pkg/packet/mock_receiver_test.go | 126 +++++++++++ pkg/packet/mock_sender_test.go | 87 ++++++++ pkg/packet/readwriter.go | 6 + pkg/packet/receiver.go | 94 ++++++++ pkg/packet/receiver_test.go | 217 ++++++++++++++++++ pkg/packet/sender.go | 59 +++++ pkg/packet/sender_test.go | 162 ++++++++++++++ pkg/packet/utils_test.go | 40 ++++ pkg/scan/arp/arp.go | 149 +++++++++++++ pkg/scan/arp/arp_test.go | 65 ++++++ pkg/scan/arp/filter.go | 19 ++ pkg/scan/arp/filter_test.go | 43 ++++ pkg/scan/arp/result_easyjson.go | 99 +++++++++ pkg/scan/engine.go | 75 +++++++ pkg/scan/engine_test.go | 154 +++++++++++++ pkg/scan/generator.go | 100 +++++++++ pkg/scan/generator_test.go | 330 ++++++++++++++++++++++++++++ pkg/scan/inject_engine.go | 25 +++ pkg/scan/mock_engine_test.go | 50 +++++ pkg/scan/mock_generator_test.go | 49 +++++ pkg/scan/mock_sendreceiver_test.go | 88 ++++++++ pkg/scan/range.go | 47 ++++ pkg/scan/range_test.go | 219 +++++++++++++++++++ pkg/scan/wire_gen.go | 26 +++ 34 files changed, 3098 insertions(+) create mode 100644 .gitignore create mode 100644 command/arp.go create mode 100644 command/root.go create mode 100644 go.mod create mode 100644 go.sum create mode 100644 main.go create mode 100644 pkg/ip/ip.go create mode 100644 pkg/ip/ip_test.go create mode 100644 pkg/packet/afpacket/readwriter.go create mode 100644 pkg/packet/mock_receiver_test.go create mode 100644 pkg/packet/mock_sender_test.go create mode 100644 pkg/packet/readwriter.go create mode 100644 pkg/packet/receiver.go create mode 100644 pkg/packet/receiver_test.go create mode 100644 pkg/packet/sender.go create mode 100644 pkg/packet/sender_test.go create mode 100644 pkg/packet/utils_test.go create mode 100644 pkg/scan/arp/arp.go create mode 100644 pkg/scan/arp/arp_test.go create mode 100644 pkg/scan/arp/filter.go create mode 100644 pkg/scan/arp/filter_test.go create mode 100644 pkg/scan/arp/result_easyjson.go create mode 100644 pkg/scan/engine.go create mode 100644 pkg/scan/engine_test.go create mode 100644 pkg/scan/generator.go create mode 100644 pkg/scan/generator_test.go create mode 100644 pkg/scan/inject_engine.go create mode 100644 pkg/scan/mock_engine_test.go create mode 100644 pkg/scan/mock_generator_test.go create mode 100644 pkg/scan/mock_sendreceiver_test.go create mode 100644 pkg/scan/range.go create mode 100644 pkg/scan/range_test.go create mode 100644 pkg/scan/wire_gen.go diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..51b8fc1 --- /dev/null +++ b/.gitignore @@ -0,0 +1 @@ +sx diff --git a/README.md b/README.md index 8d7aa19..28e0317 100644 --- a/README.md +++ b/README.md @@ -1 +1,21 @@ # sx + +## Purpose + +The goal of this project is to create the fastest network scanner with clean and simple code. + +Right now, only ARP scan is supported. + +## Building + +From the root of the source tree, run: + +``` +go build +``` + +## Using + +``` +./sx help +``` diff --git a/command/arp.go b/command/arp.go new file mode 100644 index 0000000..b71a2fe --- /dev/null +++ b/command/arp.go @@ -0,0 +1,145 @@ +package command + +import ( + "bufio" + "context" + "errors" + "net" + "os" + "os/signal" + "strings" + "sync" + "time" + + "github.com/spf13/cobra" + "github.com/v-byte-cpu/sx/pkg/ip" + "github.com/v-byte-cpu/sx/pkg/packet/afpacket" + "github.com/v-byte-cpu/sx/pkg/scan" + "github.com/v-byte-cpu/sx/pkg/scan/arp" + "go.uber.org/zap" +) + +var errSrcIP = errors.New("invalid source IP") + +var interfaceFlag string +var srcIPFlag string +var srcMACFlag string + +func init() { + arpCmd.Flags().StringVarP(&interfaceFlag, "iface", "i", "", "set interface to send/receive packets") + arpCmd.Flags().StringVar(&srcIPFlag, "srcip", "", "set source IP address for generated packets") + arpCmd.Flags().StringVar(&srcMACFlag, "srcmac", "", "set source MAC address for generated packets") + rootCmd.AddCommand(arpCmd) +} + +var arpCmd = &cobra.Command{ + Use: "arp [flags] subnet", + Example: strings.Join([]string{"arp 192.168.0.1/24", "arp 10.0.0.1"}, "\n"), + Short: "Perform ARP scan", + Args: func(cmd *cobra.Command, args []string) error { + if len(args) != 1 { + return errors.New("requires one ip subnet argument") + } + return nil + }, + RunE: func(cmd *cobra.Command, args []string) (err error) { + dstSubnet, err := ip.ParseIPNet(args[0]) + if err != nil { + return err + } + + var iface *net.Interface + var srcIP net.IP + + if len(interfaceFlag) > 0 { + if iface, err = net.InterfaceByName(interfaceFlag); err != nil { + return err + } + } else { + if iface, srcIP, err = ip.GetSubnetInterface(dstSubnet); err != nil { + return err + } + } + + if len(srcIPFlag) > 0 { + if srcIP = net.ParseIP(srcIPFlag); srcIP == nil { + return errSrcIP + } + } + + srcMAC := iface.HardwareAddr + if len(srcMACFlag) > 0 { + if srcMAC, err = net.ParseMAC(srcMACFlag); err != nil { + return err + } + } + + r := &scan.Range{Subnet: dstSubnet, Interface: iface, SrcIP: srcIP, SrcMAC: srcMAC} + return startEngine(r) + }, +} + +func startEngine(r *scan.Range) error { + bw := bufio.NewWriter(os.Stdout) + defer bw.Flush() + logger, err := zap.NewProduction() + if err != nil { + return err + } + + ctx, cancel := signal.NotifyContext(context.Background(), os.Interrupt) + defer cancel() + + // setup network interface to read/write packets + rw, err := afpacket.NewPacketSource(r.Interface.Name) + if err != nil { + return err + } + defer rw.Close() + err = rw.SetBPFFilter(arp.BPFFilter(r)) + if err != nil { + return err + } + + m := arp.NewScanMethod(ctx) + + // setup result logging + var wg sync.WaitGroup + wg.Add(1) + go func() { + defer wg.Done() + for result := range m.Results() { + // TODO extract it + if jsonFlag { + data, err := result.MarshalJSON() + if err != nil { + logger.Error("arp", zap.Error(err)) + } + bw.Write(data) + } else { + bw.WriteString(result.String()) + } + bw.WriteByte('\n') + } + }() + + // start scan + engine := scan.SetupEngine(rw, m) + done, errc := engine.Start(ctx, r) + go func() { + defer cancel() + <-done + <-time.After(300 * time.Millisecond) + }() + + // error logging + wg.Add(1) + go func() { + defer wg.Done() + for err := range errc { + logger.Error("arp", zap.Error(err)) + } + }() + wg.Wait() + return nil +} diff --git a/command/root.go b/command/root.go new file mode 100644 index 0000000..20cefef --- /dev/null +++ b/command/root.go @@ -0,0 +1,25 @@ +package command + +import ( + "os" + + "github.com/spf13/cobra" +) + +var rootCmd = &cobra.Command{ + Use: "sx", + Short: "Fast, modern, easy-to-use network scanner", + Version: "0.1.0", +} + +var jsonFlag bool + +func init() { + rootCmd.PersistentFlags().BoolVar(&jsonFlag, "json", false, "enable JSON output") +} + +func Main() { + if err := rootCmd.Execute(); err != nil { + os.Exit(1) + } +} diff --git a/go.mod b/go.mod new file mode 100644 index 0000000..cd5b9fa --- /dev/null +++ b/go.mod @@ -0,0 +1,15 @@ +module github.com/v-byte-cpu/sx + +go 1.15 + +require ( + github.com/golang/mock v1.5.0 + github.com/google/gopacket v1.1.20-0.20210304165259-20562ffb40f8 + github.com/google/wire v0.5.0 + github.com/mailru/easyjson v0.7.7 + github.com/spf13/cobra v1.1.3 + github.com/stretchr/testify v1.7.0 + go.uber.org/multierr v1.6.0 // indirect + go.uber.org/zap v1.16.0 + golang.org/x/net v0.0.0-20210226172049-e18ecbb05110 +) diff --git a/go.sum b/go.sum new file mode 100644 index 0000000..fcf2db1 --- /dev/null +++ b/go.sum @@ -0,0 +1,339 @@ +cloud.google.com/go v0.26.0/go.mod h1:aQUYkXzVsufM+DwF1aE+0xfcU+56JwCaLick0ClmMTw= +cloud.google.com/go v0.34.0/go.mod h1:aQUYkXzVsufM+DwF1aE+0xfcU+56JwCaLick0ClmMTw= +cloud.google.com/go v0.38.0/go.mod h1:990N+gfupTy94rShfmMCWGDn0LpTmnzTp2qbd1dvSRU= +cloud.google.com/go v0.44.1/go.mod h1:iSa0KzasP4Uvy3f1mN/7PiObzGgflwredwwASm/v6AU= +cloud.google.com/go v0.44.2/go.mod h1:60680Gw3Yr4ikxnPRS/oxxkBccT6SA1yMk63TGekxKY= +cloud.google.com/go v0.45.1/go.mod h1:RpBamKRgapWJb87xiFSdk4g1CME7QZg3uwTez+TSTjc= +cloud.google.com/go v0.46.3/go.mod h1:a6bKKbmY7er1mI7TEI4lsAkts/mkhTSZK8w33B4RAg0= +cloud.google.com/go/bigquery v1.0.1/go.mod h1:i/xbL2UlR5RvWAURpBYZTtm/cXjCha9lbfbpx4poX+o= +cloud.google.com/go/datastore v1.0.0/go.mod h1:LXYbyblFSglQ5pkeyhO+Qmw7ukd3C+pD7TKLgZqpHYE= +cloud.google.com/go/firestore v1.1.0/go.mod h1:ulACoGHTpvq5r8rxGJ4ddJZBZqakUQqClKRT5SZwBmk= +cloud.google.com/go/pubsub v1.0.1/go.mod h1:R0Gpsv3s54REJCy4fxDixWD93lHJMoZTyQ2kNxGRt3I= +cloud.google.com/go/storage v1.0.0/go.mod h1:IhtSnM/ZTZV8YYJWCY8RULGVqBDmpoyjwiyrjsg+URw= +dmitri.shuralyov.com/gpu/mtl v0.0.0-20190408044501-666a987793e9/go.mod h1:H6x//7gZCb22OMCxBHrMx7a5I7Hp++hsVxbQ4BYO7hU= +github.com/BurntSushi/toml v0.3.1 h1:WXkYYl6Yr3qBf1K79EBnL4mak0OimBfB0XUf9Vl28OQ= +github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU= +github.com/BurntSushi/xgb v0.0.0-20160522181843-27f122750802/go.mod h1:IVnqGOEym/WlBOVXweHU+Q+/VP0lqqI8lqeDx9IjBqo= +github.com/OneOfOne/xxhash v1.2.2/go.mod h1:HSdplMjZKSmBqAxg5vPj2TmRDmfkzw+cTzAElWljhcU= +github.com/alecthomas/template v0.0.0-20160405071501-a0175ee3bccc/go.mod h1:LOuyumcjzFXgccqObfd/Ljyb9UuFJ6TxHnclSeseNhc= +github.com/alecthomas/units v0.0.0-20151022065526-2efee857e7cf/go.mod h1:ybxpYRFXyAe+OPACYpWeL0wqObRcbAqCMya13uyzqw0= +github.com/armon/circbuf v0.0.0-20150827004946-bbbad097214e/go.mod h1:3U/XgcO3hCbHZ8TKRvWD2dDTCfh9M9ya+I9JpbB7O8o= +github.com/armon/go-metrics v0.0.0-20180917152333-f0300d1749da/go.mod h1:Q73ZrmVTwzkszR9V5SSuryQ31EELlFMUz1kKyl939pY= +github.com/armon/go-radix v0.0.0-20180808171621-7fddfc383310/go.mod h1:ufUuZ+zHj4x4TnLV4JWEpy2hxWSpsRywHrMgIH9cCH8= +github.com/beorn7/perks v0.0.0-20180321164747-3a771d992973/go.mod h1:Dwedo/Wpr24TaqPxmxbtue+5NUziq4I4S80YR8gNf3Q= +github.com/beorn7/perks v1.0.0/go.mod h1:KWe93zE9D1o94FZ5RNwFwVgaQK1VOXiVxmqh+CedLV8= +github.com/bgentry/speakeasy v0.1.0/go.mod h1:+zsyZBPWlz7T6j88CTgSN5bM796AkVf0kBD4zp0CCIs= +github.com/bketelsen/crypt v0.0.3-0.20200106085610-5cbc8cc4026c/go.mod h1:MKsuJmJgSg28kpZDP6UIiPt0e0Oz0kqKNGyRaWEPv84= +github.com/cespare/xxhash v1.1.0/go.mod h1:XrSqR1VqqWfGrhpAt58auRo0WTKS1nRRg3ghfAqPWnc= +github.com/client9/misspell v0.3.4/go.mod h1:qj6jICC3Q7zFZvVWo7KLAzC3yx5G7kyvSDkc90ppPyw= +github.com/coreos/bbolt v1.3.2/go.mod h1:iRUV2dpdMOn7Bo10OQBFzIJO9kkE559Wcmn+qkEiiKk= +github.com/coreos/etcd v3.3.13+incompatible/go.mod h1:uF7uidLiAD3TWHmW31ZFd/JWoc32PjwdhPthX9715RE= +github.com/coreos/go-semver v0.3.0/go.mod h1:nnelYz7RCh+5ahJtPPxZlU+153eP4D4r3EedlOD2RNk= +github.com/coreos/go-systemd v0.0.0-20190321100706-95778dfbb74e/go.mod h1:F5haX7vjVVG0kc13fIWeqUViNPyEJxv/OmvnBo0Yme4= +github.com/coreos/pkg v0.0.0-20180928190104-399ea9e2e55f/go.mod h1:E3G3o1h8I7cfcXa63jLwjI0eiQQMgzzUDFVpN/nH/eA= +github.com/cpuguy83/go-md2man/v2 v2.0.0/go.mod h1:maD7wRr/U5Z6m/iR4s+kqSMx2CaBsrgA7czyZG/E6dU= +github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c= +github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/dgrijalva/jwt-go v3.2.0+incompatible/go.mod h1:E3ru+11k8xSBh+hMPgOLZmtrrCbhqsmaPHjLKYnJCaQ= +github.com/dgryski/go-sip13 v0.0.0-20181026042036-e10d5fee7954/go.mod h1:vAd38F8PWV+bWy6jNmig1y/TA+kYO4g3RSRF0IAv0no= +github.com/fatih/color v1.7.0/go.mod h1:Zm6kSWBoL9eyXnKyktHP6abPY2pDugNf5KwzbycvMj4= +github.com/fsnotify/fsnotify v1.4.7/go.mod h1:jwhsz4b93w/PPRr/qN1Yymfu8t87LnFCMoQvtojpjFo= +github.com/ghodss/yaml v1.0.0/go.mod h1:4dBDuWmgqj2HViK6kFavaiC9ZROes6MMH2rRYeMEF04= +github.com/go-gl/glfw v0.0.0-20190409004039-e6da0acd62b1/go.mod h1:vR7hzQXu2zJy9AVAgeJqvqgH9Q5CA+iKCZ2gyEVpxRU= +github.com/go-kit/kit v0.8.0/go.mod h1:xBxKIO96dXMWWy0MnWVtmwkA9/13aqxPnvrjFYMA2as= +github.com/go-logfmt/logfmt v0.3.0/go.mod h1:Qt1PoO58o5twSAckw1HlFXLmHsOX5/0LbT9GBnD5lWE= +github.com/go-logfmt/logfmt v0.4.0/go.mod h1:3RMwSq7FuexP4Kalkev3ejPJsZTpXXBr9+V4qmtdjCk= +github.com/go-stack/stack v1.8.0/go.mod h1:v0f6uXyyMGvRgIKkXu+yp6POWl0qKG85gN/melR3HDY= +github.com/gogo/protobuf v1.1.1/go.mod h1:r8qH/GZQm5c6nD/R0oafs1akxWv10x8SbQlK7atdtwQ= +github.com/gogo/protobuf v1.2.1/go.mod h1:hp+jE20tsWTFYpLwKvXlhS1hjn+gTNwPg2I6zVXpSg4= +github.com/golang/glog v0.0.0-20160126235308-23def4e6c14b/go.mod h1:SBH7ygxi8pfUlaOkMMuAQtPIUF8ecWP5IEl/CR7VP2Q= +github.com/golang/groupcache v0.0.0-20190129154638-5b532d6fd5ef/go.mod h1:cIg4eruTrX1D+g88fzRXU5OdNfaM+9IcxsU14FzY7Hc= +github.com/golang/mock v1.1.1/go.mod h1:oTYuIxOrZwtPieC+H1uAHpcLFnEyAGVDL/k47Jfbm0A= +github.com/golang/mock v1.2.0/go.mod h1:oTYuIxOrZwtPieC+H1uAHpcLFnEyAGVDL/k47Jfbm0A= +github.com/golang/mock v1.3.1/go.mod h1:sBzyDLLjw3U8JLTeZvSv8jJB+tU5PVekmnlKIyFUx0Y= +github.com/golang/mock v1.5.0 h1:jlYHihg//f7RRwuPfptm04yp4s7O6Kw8EZiVYIGcH0g= +github.com/golang/mock v1.5.0/go.mod h1:CWnOUgYIOo4TcNZ0wHX3YZCqsaM1I1Jvs6v3mP3KVu8= +github.com/golang/protobuf v1.2.0/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U= +github.com/golang/protobuf v1.3.1/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U= +github.com/golang/protobuf v1.3.2/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U= +github.com/google/btree v0.0.0-20180813153112-4030bb1f1f0c/go.mod h1:lNA+9X1NB3Zf8V7Ke586lFgjr2dZNuvo3lPJSGZ5JPQ= +github.com/google/btree v1.0.0/go.mod h1:lNA+9X1NB3Zf8V7Ke586lFgjr2dZNuvo3lPJSGZ5JPQ= +github.com/google/go-cmp v0.2.0/go.mod h1:oXzfMopK8JAjlY9xF4vHSVASa0yLyX7SntLO5aqRK0M= +github.com/google/go-cmp v0.3.0/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU= +github.com/google/gopacket v1.1.20-0.20210304165259-20562ffb40f8 h1:FU2/d0krhJFVXjbGP3S9dJJFLOfSG0drhIZuTdyvzqE= +github.com/google/gopacket v1.1.20-0.20210304165259-20562ffb40f8/go.mod h1:iJ8V8n6KS+z2U1A8pUwu8bW5SyEMkXJB8Yo/Vo+TKTo= +github.com/google/martian v2.1.0+incompatible/go.mod h1:9I4somxYTbIHy5NJKHRl3wXiIaQGbYVAs8BPL6v8lEs= +github.com/google/pprof v0.0.0-20181206194817-3ea8567a2e57/go.mod h1:zfwlbNMJ+OItoe0UupaVj+oy1omPYYDuagoSzA8v9mc= +github.com/google/pprof v0.0.0-20190515194954-54271f7e092f/go.mod h1:zfwlbNMJ+OItoe0UupaVj+oy1omPYYDuagoSzA8v9mc= +github.com/google/renameio v0.1.0/go.mod h1:KWCgfxg9yswjAJkECMjeO8J8rahYeXnNhOm40UhjYkI= +github.com/google/subcommands v1.0.1/go.mod h1:ZjhPrFU+Olkh9WazFPsl27BQ4UPiG37m3yTrtFlrHVk= +github.com/google/wire v0.5.0 h1:I7ELFeVBr3yfPIcc8+MWvrjk+3VjbcSzoXm3JVa+jD8= +github.com/google/wire v0.5.0/go.mod h1:ngWDr9Qvq3yZA10YrxfyGELY/AFWGVpy9c1LTRi1EoU= +github.com/googleapis/gax-go/v2 v2.0.4/go.mod h1:0Wqv26UfaUD9n4G6kQubkQ+KchISgw+vpHVxEJEs9eg= +github.com/googleapis/gax-go/v2 v2.0.5/go.mod h1:DWXyrwAJ9X0FpwwEdw+IPEYBICEFu5mhpdKc/us6bOk= +github.com/gopherjs/gopherjs v0.0.0-20181017120253-0766667cb4d1/go.mod h1:wJfORRmW1u3UXTncJ5qlYoELFm8eSnnEO6hX4iZ3EWY= +github.com/gorilla/websocket v1.4.2/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE= +github.com/grpc-ecosystem/go-grpc-middleware v1.0.0/go.mod h1:FiyG127CGDf3tlThmgyCl78X/SZQqEOJBCDaAfeWzPs= +github.com/grpc-ecosystem/go-grpc-prometheus v1.2.0/go.mod h1:8NvIoxWQoOIhqOTXgfV/d3M/q6VIi02HzZEHgUlZvzk= +github.com/grpc-ecosystem/grpc-gateway v1.9.0/go.mod h1:vNeuVxBJEsws4ogUvrchl83t/GYV9WGTSLVdBhOQFDY= +github.com/hashicorp/consul/api v1.1.0/go.mod h1:VmuI/Lkw1nC05EYQWNKwWGbkg+FbDBtguAZLlVdkD9Q= +github.com/hashicorp/consul/sdk v0.1.1/go.mod h1:VKf9jXwCTEY1QZP2MOLRhb5i/I/ssyNV1vwHyQBF0x8= +github.com/hashicorp/errwrap v1.0.0/go.mod h1:YH+1FKiLXxHSkmPseP+kNlulaMuP3n2brvKWEqk/Jc4= +github.com/hashicorp/go-cleanhttp v0.5.1/go.mod h1:JpRdi6/HCYpAwUzNwuwqhbovhLtngrth3wmdIIUrZ80= +github.com/hashicorp/go-immutable-radix v1.0.0/go.mod h1:0y9vanUI8NX6FsYoO3zeMjhV/C5i9g4Q3DwcSNZ4P60= +github.com/hashicorp/go-msgpack v0.5.3/go.mod h1:ahLV/dePpqEmjfWmKiqvPkv/twdG7iPBM1vqhUKIvfM= +github.com/hashicorp/go-multierror v1.0.0/go.mod h1:dHtQlpGsu+cZNNAkkCN/P3hoUDHhCYQXV3UM06sGGrk= +github.com/hashicorp/go-rootcerts v1.0.0/go.mod h1:K6zTfqpRlCUIjkwsN4Z+hiSfzSTQa6eBIzfwKfwNnHU= +github.com/hashicorp/go-sockaddr v1.0.0/go.mod h1:7Xibr9yA9JjQq1JpNB2Vw7kxv8xerXegt+ozgdvDeDU= +github.com/hashicorp/go-syslog v1.0.0/go.mod h1:qPfqrKkXGihmCqbJM2mZgkZGvKG1dFdvsLplgctolz4= +github.com/hashicorp/go-uuid v1.0.0/go.mod h1:6SBZvOh/SIDV7/2o3Jml5SYk/TvGqwFJ/bN7x4byOro= +github.com/hashicorp/go-uuid v1.0.1/go.mod h1:6SBZvOh/SIDV7/2o3Jml5SYk/TvGqwFJ/bN7x4byOro= +github.com/hashicorp/go.net v0.0.1/go.mod h1:hjKkEWcCURg++eb33jQU7oqQcI9XDCnUzHA0oac0k90= +github.com/hashicorp/golang-lru v0.5.0/go.mod h1:/m3WP610KZHVQ1SGc6re/UDhFvYD7pJ4Ao+sR/qLZy8= +github.com/hashicorp/golang-lru v0.5.1/go.mod h1:/m3WP610KZHVQ1SGc6re/UDhFvYD7pJ4Ao+sR/qLZy8= +github.com/hashicorp/hcl v1.0.0/go.mod h1:E5yfLk+7swimpb2L/Alb/PJmXilQ/rhwaUYs4T20WEQ= +github.com/hashicorp/logutils v1.0.0/go.mod h1:QIAnNjmIWmVIIkWDTG1z5v++HQmx9WQRO+LraFDTW64= +github.com/hashicorp/mdns v1.0.0/go.mod h1:tL+uN++7HEJ6SQLQ2/p+z2pH24WQKWjBPkE0mNTz8vQ= +github.com/hashicorp/memberlist v0.1.3/go.mod h1:ajVTdAv/9Im8oMAAj5G31PhhMCZJV2pPBoIllUwCN7I= +github.com/hashicorp/serf v0.8.2/go.mod h1:6hOLApaqBFA1NXqRQAsxw9QxuDEvNxSQRwA/JwenrHc= +github.com/inconshreveable/mousetrap v1.0.0 h1:Z8tu5sraLXCXIcARxBp/8cbvlwVa7Z1NHg9XEKhtSvM= +github.com/inconshreveable/mousetrap v1.0.0/go.mod h1:PxqpIevigyE2G7u3NXJIT2ANytuPF1OarO4DADm73n8= +github.com/jonboulle/clockwork v0.1.0/go.mod h1:Ii8DK3G1RaLaWxj9trq07+26W01tbo22gdxWY5EU2bo= +github.com/josharian/intern v1.0.0 h1:vlS4z54oSdjm0bgjRigI+G1HpF+tI+9rE5LLzOg8HmY= +github.com/josharian/intern v1.0.0/go.mod h1:5DoeVV0s6jJacbCEi61lwdGj/aVlrQvzHFFd8Hwg//Y= +github.com/json-iterator/go v1.1.6/go.mod h1:+SdeFBvtyEkXs7REEP0seUULqWtbJapLOCVDaaPEHmU= +github.com/jstemmer/go-junit-report v0.0.0-20190106144839-af01ea7f8024/go.mod h1:6v2b51hI/fHJwM22ozAgKL4VKDeJcHhJFhtBdhmNjmU= +github.com/jtolds/gls v4.20.0+incompatible/go.mod h1:QJZ7F/aHp+rZTRtaJ1ow/lLfFfVYBRgL+9YlvaHOwJU= +github.com/julienschmidt/httprouter v1.2.0/go.mod h1:SYymIcj16QtmaHHD7aYtjjsJG7VTCxuUUipMqKk8s4w= +github.com/kisielk/errcheck v1.1.0/go.mod h1:EZBBE59ingxPouuu3KfxchcWSUPOHkagtvWXihfKN4Q= +github.com/kisielk/gotool v1.0.0/go.mod h1:XhKaO+MFFWcvkIS/tQcRk01m1F5IRFswLeQ+oQHNcck= +github.com/konsorten/go-windows-terminal-sequences v1.0.1/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ= +github.com/kr/logfmt v0.0.0-20140226030751-b84e30acd515/go.mod h1:+0opPa2QZZtGFBFZlji/RkVcI2GknAs/DXo4wKdlNEc= +github.com/kr/pretty v0.1.0 h1:L/CwN0zerZDmRFUapSPitk6f+Q3+0za1rQkzVuMiMFI= +github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo= +github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ= +github.com/kr/text v0.1.0 h1:45sCR5RtlFHMR4UwH9sdQ5TC8v0qDQCHnXt+kaKSTVE= +github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI= +github.com/magiconair/properties v1.8.1/go.mod h1:PppfXfuXeibc/6YijjN8zIbojt8czPbwD3XqdrwzmxQ= +github.com/mailru/easyjson v0.7.7 h1:UGYAvKxe3sBsEDzO8ZeWOSlIQfWFlxbzLZe7hwFURr0= +github.com/mailru/easyjson v0.7.7/go.mod h1:xzfreul335JAWq5oZzymOObrkdz5UnU4kGfJJLY9Nlc= +github.com/mattn/go-colorable v0.0.9/go.mod h1:9vuHe8Xs5qXnSaW/c/ABM9alt+Vo+STaOChaDxuIBZU= +github.com/mattn/go-isatty v0.0.3/go.mod h1:M+lRXTBqGeGNdLjl/ufCoiOlB5xdOkqRJdNxMWT7Zi4= +github.com/matttproud/golang_protobuf_extensions v1.0.1/go.mod h1:D8He9yQNgCq6Z5Ld7szi9bcBfOoFv/3dc6xSMkL2PC0= +github.com/miekg/dns v1.0.14/go.mod h1:W1PPwlIAgtquWBMBEV9nkV9Cazfe8ScdGz/Lj7v3Nrg= +github.com/mitchellh/cli v1.0.0/go.mod h1:hNIlj7HEI86fIcpObd7a0FcrxTWetlwJDGcceTlRvqc= +github.com/mitchellh/go-homedir v1.0.0/go.mod h1:SfyaCUpYCn1Vlf4IUYiD9fPX4A5wJrkLzIz1N1q0pr0= +github.com/mitchellh/go-homedir v1.1.0/go.mod h1:SfyaCUpYCn1Vlf4IUYiD9fPX4A5wJrkLzIz1N1q0pr0= +github.com/mitchellh/go-testing-interface v1.0.0/go.mod h1:kRemZodwjscx+RGhAo8eIhFbs2+BFgRtFPeD/KE+zxI= +github.com/mitchellh/gox v0.4.0/go.mod h1:Sd9lOJ0+aimLBi73mGofS1ycjY8lL3uZM3JPS42BGNg= +github.com/mitchellh/iochan v1.0.0/go.mod h1:JwYml1nuB7xOzsp52dPpHFffvOCDupsG0QubkSMEySY= +github.com/mitchellh/mapstructure v0.0.0-20160808181253-ca63d7c062ee/go.mod h1:FVVH3fgwuzCH5S8UJGiWEs2h04kUh9fWfEaFds41c1Y= +github.com/mitchellh/mapstructure v1.1.2/go.mod h1:FVVH3fgwuzCH5S8UJGiWEs2h04kUh9fWfEaFds41c1Y= +github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q= +github.com/modern-go/reflect2 v1.0.1/go.mod h1:bx2lNnkwVCuqBIxFjflWJWanXIb3RllmbCylyMrvgv0= +github.com/mwitkow/go-conntrack v0.0.0-20161129095857-cc309e4a2223/go.mod h1:qRWi+5nqEBWmkhHvq77mSJWrCKwh8bxhgT7d/eI7P4U= +github.com/oklog/ulid v1.3.1/go.mod h1:CirwcVhetQ6Lv90oh/F+FBtV6XMibvdAFo93nm5qn4U= +github.com/pascaldekloe/goe v0.0.0-20180627143212-57f6aae5913c/go.mod h1:lzWF7FIEvWOWxwDKqyGYQf6ZUaNfKdP144TG7ZOy1lc= +github.com/pelletier/go-toml v1.2.0/go.mod h1:5z9KED0ma1S8pY6P1sdut58dfprrGBbd/94hg7ilaic= +github.com/pkg/errors v0.8.0/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0= +github.com/pkg/errors v0.8.1 h1:iURUrRGxPUNPdy5/HRSm+Yj6okJ6UtLINN0Q9M4+h3I= +github.com/pkg/errors v0.8.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0= +github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM= +github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= +github.com/posener/complete v1.1.1/go.mod h1:em0nMJCgc9GFtwrmVmEMR/ZL6WyhyjMBndrE9hABlRI= +github.com/prometheus/client_golang v0.9.1/go.mod h1:7SWBe2y4D6OKWSNQJUaRYU/AaXPKyh/dDVn+NZz0KFw= +github.com/prometheus/client_golang v0.9.3/go.mod h1:/TN21ttK/J9q6uSwhBd54HahCDft0ttaMvbicHlPoso= +github.com/prometheus/client_model v0.0.0-20180712105110-5c3871d89910/go.mod h1:MbSGuTsp3dbXC40dX6PRTWyKYBIrTGTE9sqQNg2J8bo= +github.com/prometheus/client_model v0.0.0-20190129233127-fd36f4220a90/go.mod h1:xMI15A0UPsDsEKsMN9yxemIoYk6Tm2C1GtYGdfGttqA= +github.com/prometheus/common v0.0.0-20181113130724-41aa239b4cce/go.mod h1:daVV7qP5qjZbuso7PdcryaAu0sAZbrN9i7WWcTMWvro= +github.com/prometheus/common v0.4.0/go.mod h1:TNfzLD0ON7rHzMJeJkieUDPYmFC7Snx/y86RQel1bk4= +github.com/prometheus/procfs v0.0.0-20181005140218-185b4288413d/go.mod h1:c3At6R/oaqEKCNdg8wHV1ftS6bRYblBhIjjI8uT2IGk= +github.com/prometheus/procfs v0.0.0-20190507164030-5867b95ac084/go.mod h1:TjEm7ze935MbeOT/UhFTIMYKhuLP4wbCsTZCD3I8kEA= +github.com/prometheus/tsdb v0.7.1/go.mod h1:qhTCs0VvXwvX/y3TZrWD7rabWM+ijKTux40TwIPHuXU= +github.com/rogpeppe/fastuuid v0.0.0-20150106093220-6724a57986af/go.mod h1:XWv6SoW27p1b0cqNHllgS5HIMJraePCO15w5zCzIWYg= +github.com/rogpeppe/go-internal v1.3.0/go.mod h1:M8bDsm7K2OlrFYOpmOWEs/qY81heoFRclV5y23lUDJ4= +github.com/russross/blackfriday/v2 v2.0.1/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM= +github.com/ryanuber/columnize v0.0.0-20160712163229-9b3edd62028f/go.mod h1:sm1tb6uqfes/u+d4ooFouqFdy9/2g9QGwK3SQygK0Ts= +github.com/sean-/seed v0.0.0-20170313163322-e2103e2c3529/go.mod h1:DxrIzT+xaE7yg65j358z/aeFdxmN0P9QXhEzd20vsDc= +github.com/shurcooL/sanitized_anchor_name v1.0.0/go.mod h1:1NzhyTcUVG4SuEtjjoZeVRXNmyL/1OwPU0+IJeTBvfc= +github.com/sirupsen/logrus v1.2.0/go.mod h1:LxeOpSwHxABJmUn/MG1IvRgCAasNZTLOkJPxbbu5VWo= +github.com/smartystreets/assertions v0.0.0-20180927180507-b2de0cb4f26d/go.mod h1:OnSkiWE9lh6wB0YB77sQom3nweQdgAjqCqsofrRNTgc= +github.com/smartystreets/goconvey v1.6.4/go.mod h1:syvi0/a8iFYH4r/RixwvyeAJjdLS9QV7WQ/tjFTllLA= +github.com/soheilhy/cmux v0.1.4/go.mod h1:IM3LyeVVIOuxMH7sFAkER9+bJ4dT7Ms6E4xg4kGIyLM= +github.com/spaolacci/murmur3 v0.0.0-20180118202830-f09979ecbc72/go.mod h1:JwIasOWyU6f++ZhiEuf87xNszmSA2myDM2Kzu9HwQUA= +github.com/spf13/afero v1.1.2/go.mod h1:j4pytiNVoe2o6bmDsKpLACNPDBIoEAkihy7loJ1B0CQ= +github.com/spf13/cast v1.3.0/go.mod h1:Qx5cxh0v+4UWYiBimWS+eyWzqEqokIECu5etghLkUJE= +github.com/spf13/cobra v1.1.3 h1:xghbfqPkxzxP3C/f3n5DdpAbdKLj4ZE4BWQI362l53M= +github.com/spf13/cobra v1.1.3/go.mod h1:pGADOWyqRD/YMrPZigI/zbliZ2wVD/23d+is3pSWzOo= +github.com/spf13/jwalterweatherman v1.0.0/go.mod h1:cQK4TGJAtQXfYWX+Ddv3mKDzgVb68N+wFjFa4jdeBTo= +github.com/spf13/pflag v1.0.3/go.mod h1:DYY7MBk1bdzusC3SYhjObp+wFpr4gzcvqqNjLnInEg4= +github.com/spf13/pflag v1.0.5 h1:iy+VFUOCP1a+8yFto/drg2CJ5u0yRoB7fZw3DKv/JXA= +github.com/spf13/pflag v1.0.5/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg= +github.com/spf13/viper v1.7.0/go.mod h1:8WkrPz2fc9jxqZNCJI/76HCieCp4Q8HaLFoCha5qpdg= +github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= +github.com/stretchr/objx v0.1.1/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= +github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs= +github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI= +github.com/stretchr/testify v1.4.0/go.mod h1:j7eGeouHqKxXV5pUuKE4zz7dFj8WfuZ+81PSLYec5m4= +github.com/stretchr/testify v1.7.0 h1:nwc3DEeHmmLAfoZucVR881uASk0Mfjw8xYJ99tb5CcY= +github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= +github.com/subosito/gotenv v1.2.0/go.mod h1:N0PQaV/YGNqwC0u51sEeR/aUtSLEXKX9iv69rRypqCw= +github.com/tmc/grpc-websocket-proxy v0.0.0-20190109142713-0ad062ec5ee5/go.mod h1:ncp9v5uamzpCO7NfCPTXjqaC+bZgJeR0sMTm6dMHP7U= +github.com/xiang90/probing v0.0.0-20190116061207-43a291ad63a2/go.mod h1:UETIi67q53MR2AWcXfiuqkDkRtnGDLqkBTpCHuJHxtU= +go.etcd.io/bbolt v1.3.2/go.mod h1:IbVyRI1SCnLcuJnV2u8VeU0CEYM7e686BmAb1XKL+uU= +go.opencensus.io v0.21.0/go.mod h1:mSImk1erAIZhrmZN+AvHh14ztQfjbGwt4TtuofqLduU= +go.opencensus.io v0.22.0/go.mod h1:+kGneAE2xo2IficOXnaByMWTGM9T73dGwxeWcUqIpI8= +go.uber.org/atomic v1.4.0/go.mod h1:gD2HeocX3+yG+ygLZcrzQJaqmWj9AIm7n08wl/qW/PE= +go.uber.org/atomic v1.6.0/go.mod h1:sABNBOSYdrvTF6hTgEIbc7YasKWGhgEQZyfxyTvoXHQ= +go.uber.org/atomic v1.7.0 h1:ADUqmZGgLDDfbSL9ZmPxKTybcoEYHgpYfELNoN+7hsw= +go.uber.org/atomic v1.7.0/go.mod h1:fEN4uk6kAWBTFdckzkM89CLk9XfWZrxpCo0nPH17wJc= +go.uber.org/multierr v1.1.0/go.mod h1:wR5kodmAFQ0UK8QlbwjlSNy0Z68gJhDJUG5sjR94q/0= +go.uber.org/multierr v1.5.0/go.mod h1:FeouvMocqHpRaaGuG9EjoKcStLC43Zu/fmqdUMPcKYU= +go.uber.org/multierr v1.6.0 h1:y6IPFStTAIT5Ytl7/XYmHvzXQ7S3g/IeZW9hyZ5thw4= +go.uber.org/multierr v1.6.0/go.mod h1:cdWPpRnG4AhwMwsgIHip0KRBQjJy5kYEpYjJxpXp9iU= +go.uber.org/tools v0.0.0-20190618225709-2cfd321de3ee/go.mod h1:vJERXedbb3MVM5f9Ejo0C68/HhF8uaILCdgjnY+goOA= +go.uber.org/zap v1.10.0/go.mod h1:vwi/ZaCAaUcBkycHslxD9B2zi4UTXhF60s6SWpuDF0Q= +go.uber.org/zap v1.16.0 h1:uFRZXykJGK9lLY4HtgSw44DnIcAM+kRBP7x5m+NpAOM= +go.uber.org/zap v1.16.0/go.mod h1:MA8QOfq0BHJwdXa996Y4dYkAqRKB8/1K1QMMZVaNZjQ= +golang.org/x/crypto v0.0.0-20180904163835-0709b304e793/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4= +golang.org/x/crypto v0.0.0-20181029021203-45a5f77698d3/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4= +golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w= +golang.org/x/crypto v0.0.0-20190510104115-cbcb75029529/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI= +golang.org/x/crypto v0.0.0-20190605123033-f99c8df09eb5/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI= +golang.org/x/crypto v0.0.0-20191011191535-87dc89f01550/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI= +golang.org/x/exp v0.0.0-20190121172915-509febef88a4/go.mod h1:CJ0aWSM057203Lf6IL+f9T1iT9GByDxfZKAQTCR3kQA= +golang.org/x/exp v0.0.0-20190306152737-a1d7652674e8/go.mod h1:CJ0aWSM057203Lf6IL+f9T1iT9GByDxfZKAQTCR3kQA= +golang.org/x/exp v0.0.0-20190510132918-efd6b22b2522/go.mod h1:ZjyILWgesfNpC6sMxTJOJm9Kp84zZh5NQWvqDGG3Qr8= +golang.org/x/exp v0.0.0-20190829153037-c13cbed26979/go.mod h1:86+5VVa7VpoJ4kLfm080zCjGlMRFzhUhsZKEZO7MGek= +golang.org/x/exp v0.0.0-20191030013958-a1ab85dbe136/go.mod h1:JXzH8nQsPlswgeRAPE3MuO9GYsAcnJvJ4vnMwN/5qkY= +golang.org/x/image v0.0.0-20190227222117-0694c2d4d067/go.mod h1:kZ7UVZpmo3dzQBMxlp+ypCbDeSB+sBbTgSJuh5dn5js= +golang.org/x/image v0.0.0-20190802002840-cff245a6509b/go.mod h1:FeLwcggjj3mMvU+oOTbSwawSJRM1uh48EjtB4UJZlP0= +golang.org/x/lint v0.0.0-20181026193005-c67002cb31c3/go.mod h1:UVdnD1Gm6xHRNCYTkRU2/jEulfH38KcIWyp/GAMgvoE= +golang.org/x/lint v0.0.0-20190227174305-5b3e6a55c961/go.mod h1:wehouNa3lNwaWXcvxsM5YxQ5yQlVC4a0KAMCusXpPoU= +golang.org/x/lint v0.0.0-20190301231843-5614ed5bae6f/go.mod h1:UVdnD1Gm6xHRNCYTkRU2/jEulfH38KcIWyp/GAMgvoE= +golang.org/x/lint v0.0.0-20190313153728-d0100b6bd8b3/go.mod h1:6SW0HCj/g11FgYtHlgUYUwCkIfeOF89ocIRzGO/8vkc= +golang.org/x/lint v0.0.0-20190409202823-959b441ac422/go.mod h1:6SW0HCj/g11FgYtHlgUYUwCkIfeOF89ocIRzGO/8vkc= +golang.org/x/lint v0.0.0-20190909230951-414d861bb4ac/go.mod h1:6SW0HCj/g11FgYtHlgUYUwCkIfeOF89ocIRzGO/8vkc= +golang.org/x/lint v0.0.0-20190930215403-16217165b5de/go.mod h1:6SW0HCj/g11FgYtHlgUYUwCkIfeOF89ocIRzGO/8vkc= +golang.org/x/lint v0.0.0-20200302205851-738671d3881b h1:Wh+f8QHJXR411sJR8/vRBTZ7YapZaRvUcLFFJhusH0k= +golang.org/x/lint v0.0.0-20200302205851-738671d3881b/go.mod h1:3xt1FjdF8hUf6vQPIChWIBhFzV8gjjsPE/fR3IyQdNY= +golang.org/x/mobile v0.0.0-20190312151609-d3739f865fa6/go.mod h1:z+o9i4GpDbdi3rU15maQ/Ox0txvL9dWGYEHz965HBQE= +golang.org/x/mobile v0.0.0-20190719004257-d2bd2a29d028/go.mod h1:E/iHnbuqvinMTCcRqshq8CkpyQDoeVncDDYHnLhea+o= +golang.org/x/mod v0.0.0-20190513183733-4bf6d317e70e/go.mod h1:mXi4GBBbnImb6dmsKGUJ2LatrhH/nqhxcFungHvyanc= +golang.org/x/mod v0.1.0/go.mod h1:0QHyrYULN0/3qlju5TqG8bIK38QM8yzMo5ekMj3DlcY= +golang.org/x/mod v0.1.1-0.20191105210325-c90efee705ee/go.mod h1:QqPTAvyqsEbceGzBzNggFXnrqF1CaUcvgkdR5Ot7KZg= +golang.org/x/mod v0.3.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA= +golang.org/x/net v0.0.0-20180724234803-3673e40ba225/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4= +golang.org/x/net v0.0.0-20180826012351-8a410e7b638d/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4= +golang.org/x/net v0.0.0-20181023162649-9b4f9f5ad519/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4= +golang.org/x/net v0.0.0-20181114220301-adae6a3d119a/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4= +golang.org/x/net v0.0.0-20181201002055-351d144fa1fc/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4= +golang.org/x/net v0.0.0-20181220203305-927f97764cc3/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4= +golang.org/x/net v0.0.0-20190108225652-1e06a53dbb7e/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4= +golang.org/x/net v0.0.0-20190213061140-3a22650c66bd/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4= +golang.org/x/net v0.0.0-20190311183353-d8887717615a/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg= +golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg= +golang.org/x/net v0.0.0-20190501004415-9ce7a6920f09/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg= +golang.org/x/net v0.0.0-20190503192946-f4e77d36d62c/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg= +golang.org/x/net v0.0.0-20190603091049-60506f45cf65/go.mod h1:HSz+uSET+XFnRR8LxR5pz3Of3rY3CfYBVs4xY44aLks= +golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s= +golang.org/x/net v0.0.0-20210226172049-e18ecbb05110 h1:qWPm9rbaAMKs8Bq/9LRpbMqxWRVUAQwMI9fVrssnTfw= +golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg= +golang.org/x/oauth2 v0.0.0-20180821212333-d2e6202438be/go.mod h1:N/0e6XlmueqKjAGxoOufVs8QHGRruUQn6yWY3a++T0U= +golang.org/x/oauth2 v0.0.0-20190226205417-e64efc72b421/go.mod h1:gOpvHmFTYa4IltrdGE7lF6nIHvwfUNPOp7c8zoXwtLw= +golang.org/x/oauth2 v0.0.0-20190604053449-0f29369cfe45/go.mod h1:gOpvHmFTYa4IltrdGE7lF6nIHvwfUNPOp7c8zoXwtLw= +golang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.0.0-20181108010431-42b317875d0f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.0.0-20181221193216-37e7f081c4d4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.0.0-20190227155943-e225da77a7e6/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sys v0.0.0-20180823144017-11551d06cbcc/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= +golang.org/x/sys v0.0.0-20180830151530-49385e6e1522/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= +golang.org/x/sys v0.0.0-20180905080454-ebe1bf3edb33/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= +golang.org/x/sys v0.0.0-20181026203630-95b1ffbd15a5/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= +golang.org/x/sys v0.0.0-20181107165924-66b7b1311ac8/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= +golang.org/x/sys v0.0.0-20181116152217-5ac8a444bdc5/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= +golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= +golang.org/x/sys v0.0.0-20190312061237-fead79001313/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20190502145724-3ef323f4f1fd/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20190507160741-ecd444e8653b/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20190606165138-5da285871e9c/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20190624142023-c5567b49c5d0/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20201119102817-f84b799fce68 h1:nxC68pudNYkKU6jWhgrqdreuFiOQWj1Fs7T3VrH4Pjw= +golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo= +golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ= +golang.org/x/text v0.3.1-0.20180807135948-17ff2d5776d2/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ= +golang.org/x/text v0.3.2/go.mod h1:bEr9sfX3Q8Zfm5fL9x+3itogRgK3+ptLWKqgva+5dAk= +golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= +golang.org/x/time v0.0.0-20181108054448-85acf8d2951c/go.mod h1:tRJNPiyCQ0inRvYxbN9jk5I+vvW/OXSQhTDSoE431IQ= +golang.org/x/time v0.0.0-20190308202827-9d24e82272b4/go.mod h1:tRJNPiyCQ0inRvYxbN9jk5I+vvW/OXSQhTDSoE431IQ= +golang.org/x/tools v0.0.0-20180221164845-07fd8470d635/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= +golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= +golang.org/x/tools v0.0.0-20190114222345-bf090417da8b/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= +golang.org/x/tools v0.0.0-20190226205152-f727befe758c/go.mod h1:9Yl7xja0Znq3iFh3HoIrodX9oNMXvdceNzlUR8zjMvY= +golang.org/x/tools v0.0.0-20190311212946-11955173bddd/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs= +golang.org/x/tools v0.0.0-20190312151545-0bb0c0a6e846/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs= +golang.org/x/tools v0.0.0-20190312170243-e65039ee4138/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs= +golang.org/x/tools v0.0.0-20190328211700-ab21143f2384/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs= +golang.org/x/tools v0.0.0-20190422233926-fe54fb35175b/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs= +golang.org/x/tools v0.0.0-20190425150028-36563e24a262/go.mod h1:RgjU9mgBXZiqYHBnxXauZ1Gv1EHHAz9KjViQ78xBX0Q= +golang.org/x/tools v0.0.0-20190506145303-2d16b83fe98c/go.mod h1:RgjU9mgBXZiqYHBnxXauZ1Gv1EHHAz9KjViQ78xBX0Q= +golang.org/x/tools v0.0.0-20190606124116-d0a3d012864b/go.mod h1:/rFqwRUd4F7ZHNgwSSTFct+R/Kf4OFW1sUzUTQQTgfc= +golang.org/x/tools v0.0.0-20190621195816-6e04913cbbac/go.mod h1:/rFqwRUd4F7ZHNgwSSTFct+R/Kf4OFW1sUzUTQQTgfc= +golang.org/x/tools v0.0.0-20190628153133-6cdbf07be9d0/go.mod h1:/rFqwRUd4F7ZHNgwSSTFct+R/Kf4OFW1sUzUTQQTgfc= +golang.org/x/tools v0.0.0-20190816200558-6889da9d5479/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo= +golang.org/x/tools v0.0.0-20190911174233-4f2ddba30aff/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo= +golang.org/x/tools v0.0.0-20191012152004-8de300cfc20a/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo= +golang.org/x/tools v0.0.0-20191029041327-9cc4af7d6b2c/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo= +golang.org/x/tools v0.0.0-20191029190741-b9c20aec41a5/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo= +golang.org/x/tools v0.0.0-20191112195655-aa38f8e97acc/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo= +golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo= +golang.org/x/tools v0.0.0-20200130002326-2f3ba24bd6e7 h1:EBZoQjiKKPaLbPrbpssUfuHtwM6KV/vb4U85g/cigFY= +golang.org/x/tools v0.0.0-20200130002326-2f3ba24bd6e7/go.mod h1:TB2adYChydJhpapKDTa4BR/hXlZSLoq2Wpct/0txZ28= +golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +golang.org/x/xerrors v0.0.0-20191011141410-1b5146add898/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +google.golang.org/api v0.4.0/go.mod h1:8k5glujaEP+g9n7WNsDg8QP6cUVNI86fCNMcbazEtwE= +google.golang.org/api v0.7.0/go.mod h1:WtwebWUNSVBH/HAw79HIFXZNqEvBhG+Ra+ax0hx3E3M= +google.golang.org/api v0.8.0/go.mod h1:o4eAsZoiT+ibD93RtjEohWalFOjRDx6CVaqeizhEnKg= +google.golang.org/api v0.9.0/go.mod h1:o4eAsZoiT+ibD93RtjEohWalFOjRDx6CVaqeizhEnKg= +google.golang.org/api v0.13.0/go.mod h1:iLdEw5Ide6rF15KTC1Kkl0iskquN2gFfn9o9XIsbkAI= +google.golang.org/appengine v1.1.0/go.mod h1:EbEs0AVv82hx2wNQdGPgUI5lhzA/G0D9YwlJXL52JkM= +google.golang.org/appengine v1.4.0/go.mod h1:xpcJRLb0r/rnEns0DIKYYv+WjYCduHsrkT7/EB5XEv4= +google.golang.org/appengine v1.5.0/go.mod h1:xpcJRLb0r/rnEns0DIKYYv+WjYCduHsrkT7/EB5XEv4= +google.golang.org/appengine v1.6.1/go.mod h1:i06prIuMbXzDqacNJfV5OdTW448YApPu5ww/cMBSeb0= +google.golang.org/genproto v0.0.0-20180817151627-c66870c02cf8/go.mod h1:JiN7NxoALGmiZfu7CAH4rXhgtRTLTxftemlI0sWmxmc= +google.golang.org/genproto v0.0.0-20190307195333-5fe7a883aa19/go.mod h1:VzzqZJRnGkLBvHegQrXjBqPurQTc5/KpmUdxsrq26oE= +google.golang.org/genproto v0.0.0-20190418145605-e7d98fc518a7/go.mod h1:VzzqZJRnGkLBvHegQrXjBqPurQTc5/KpmUdxsrq26oE= +google.golang.org/genproto v0.0.0-20190425155659-357c62f0e4bb/go.mod h1:VzzqZJRnGkLBvHegQrXjBqPurQTc5/KpmUdxsrq26oE= +google.golang.org/genproto v0.0.0-20190502173448-54afdca5d873/go.mod h1:VzzqZJRnGkLBvHegQrXjBqPurQTc5/KpmUdxsrq26oE= +google.golang.org/genproto v0.0.0-20190801165951-fa694d86fc64/go.mod h1:DMBHOl98Agz4BDEuKkezgsaosCRResVns1a3J2ZsMNc= +google.golang.org/genproto v0.0.0-20190819201941-24fa4b261c55/go.mod h1:DMBHOl98Agz4BDEuKkezgsaosCRResVns1a3J2ZsMNc= +google.golang.org/genproto v0.0.0-20190911173649-1774047e7e51/go.mod h1:IbNlFCBrqXvoKpeg0TB2l7cyZUmoaFKYIwrEpbDKLA8= +google.golang.org/genproto v0.0.0-20191108220845-16a3f7862a1a/go.mod h1:n3cpQtvxv34hfy77yVDNjmbRyujviMdxYliBSkLhpCc= +google.golang.org/grpc v1.19.0/go.mod h1:mqu4LbDTu4XGKhr4mRzUsmM4RtVoemTSY81AxZiDr8c= +google.golang.org/grpc v1.20.1/go.mod h1:10oTOabMzJvdu6/UiuZezV6QK5dSlG84ov/aaiqXj38= +google.golang.org/grpc v1.21.1/go.mod h1:oYelfM1adQP15Ek0mdvEgi9Df8B9CZIaU1084ijfRaM= +gopkg.in/alecthomas/kingpin.v2 v2.2.6/go.mod h1:FMv+mEhP44yOT+4EoQTLFTRgOQ1FBLkstjWtayDeSgw= +gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= +gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127 h1:qIbj1fsPNlZgppZ+VLlY7N33q108Sa+fhmuc+sWQYwY= +gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= +gopkg.in/errgo.v2 v2.1.0/go.mod h1:hNsd1EY+bozCKY1Ytp96fpM3vjJbqLJn88ws8XvfDNI= +gopkg.in/ini.v1 v1.51.0/go.mod h1:pNLf8WUiyNEtQjuu5G5vTm06TEv9tsIgeAvK8hOrP4k= +gopkg.in/resty.v1 v1.12.0/go.mod h1:mDo4pnntr5jdWRML875a/NmxYqAlA73dVijT2AXvQQo= +gopkg.in/yaml.v2 v2.0.0-20170812160011-eb3733d160e7/go.mod h1:JAlM8MvJe8wmxCU4Bli9HhUf9+ttbYbLASfIpnQbh74= +gopkg.in/yaml.v2 v2.2.1/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= +gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= +gopkg.in/yaml.v2 v2.2.4/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= +gopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY= +gopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ= +gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c h1:dUUwHk2QECo/6vqA44rthZ8ie2QXMNeKRTHCNY2nXvo= +gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= +honnef.co/go/tools v0.0.0-20190102054323-c2f93a96b099/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4= +honnef.co/go/tools v0.0.0-20190106161140-3f1c8253044a/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4= +honnef.co/go/tools v0.0.0-20190418001031-e561f6794a2a/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4= +honnef.co/go/tools v0.0.1-2019.2.3 h1:3JgtbtFHMiCmsznwGVTUWbgGov+pVqnlf1dEJTNAXeM= +honnef.co/go/tools v0.0.1-2019.2.3/go.mod h1:a3bituU0lyd329TUQxRnasdCoJDkEUEAqEt0JzvZhAg= +rsc.io/binaryregexp v0.2.0/go.mod h1:qTv7/COck+e2FymRvadv62gMdZztPaShugOCi3I+8D8= diff --git a/main.go b/main.go new file mode 100644 index 0000000..2ac1f84 --- /dev/null +++ b/main.go @@ -0,0 +1,7 @@ +package main + +import "github.com/v-byte-cpu/sx/command" + +func main() { + command.Main() +} diff --git a/pkg/ip/ip.go b/pkg/ip/ip.go new file mode 100644 index 0000000..af6b104 --- /dev/null +++ b/pkg/ip/ip.go @@ -0,0 +1,59 @@ +package ip + +import ( + "errors" + "net" +) + +var ( + ErrInvalidAddr = errors.New("invalid IP subnet/host") + ErrSubnetInterface = errors.New("no directly connected interfaces to destination subnet") +) + +func Inc(ip net.IP) { + for j := len(ip) - 1; j >= 0; j-- { + ip[j]++ + if ip[j] > 0 { + break + } + } +} + +func DupIP(ip net.IP) net.IP { + dup := make([]byte, 4) + copy(dup, ip.To4()) + return dup +} + +func ParseIPNet(subnet string) (*net.IPNet, error) { + _, result, err := net.ParseCIDR(subnet) + if err == nil { + return result, err + } + // try to parse host IP address instead + ipAddr := net.ParseIP(subnet) + if ipAddr == nil { + return nil, ErrInvalidAddr + } + return &net.IPNet{IP: ipAddr.To4(), Mask: net.CIDRMask(32, 32)}, nil +} + +func GetSubnetInterface(dstSubnet *net.IPNet) (*net.Interface, net.IP, error) { + dstSubnetIP := dstSubnet.IP.Mask(dstSubnet.Mask) + ifaces, err := net.Interfaces() + if err != nil { + return nil, nil, err + } + for _, iface := range ifaces { + addrs, err := iface.Addrs() + if err != nil { + return nil, nil, err + } + for _, addr := range addrs { + if ipnet, ok := addr.(*net.IPNet); ok && ipnet.Contains(dstSubnetIP) { + return &iface, ipnet.IP.To4(), nil + } + } + } + return nil, nil, ErrSubnetInterface +} diff --git a/pkg/ip/ip_test.go b/pkg/ip/ip_test.go new file mode 100644 index 0000000..4e6a95a --- /dev/null +++ b/pkg/ip/ip_test.go @@ -0,0 +1,99 @@ +package ip + +import ( + "net" + "testing" + + "github.com/stretchr/testify/assert" +) + +func TestInc(t *testing.T) { + t.Parallel() + tests := []struct { + name string + input net.IP + expected net.IP + }{ + { + name: "ZeroNet", + input: net.IPv4(0, 0, 0, 0), + expected: net.IPv4(0, 0, 0, 1), + }, + { + name: "Inc3rd", + input: net.IPv4(1, 1, 0, 255), + expected: net.IPv4(1, 1, 1, 0), + }, + { + name: "Inc2nd", + input: net.IPv4(1, 1, 255, 255), + expected: net.IPv4(1, 2, 0, 0), + }, + { + name: "Inc1st", + input: net.IPv4(1, 255, 255, 255), + expected: net.IPv4(2, 0, 0, 0), + }, + } + + for _, vtt := range tests { + tt := vtt + t.Run(tt.name, func(t *testing.T) { + t.Parallel() + Inc(tt.input) + assert.Equal(t, tt.expected, tt.input) + }) + } +} + +func TestDupIP(t *testing.T) { + t.Parallel() + ipAddr := net.IPv4(192, 168, 0, 1).To4() + + dupAddr := DupIP(ipAddr) + assert.Equal(t, ipAddr, dupAddr) + + dupAddr[3]++ + assert.Equal(t, net.IPv4(192, 168, 0, 1).To4(), ipAddr) +} + +func TestParseIPNetWithError(t *testing.T) { + t.Parallel() + _, err := ParseIPNet("") + assert.Error(t, err) +} + +func TestParseIPNet(t *testing.T) { + t.Parallel() + tests := []struct { + name string + in string + expected *net.IPNet + }{ + { + name: "subnet", + in: "192.168.0.1/24", + expected: &net.IPNet{ + IP: net.IPv4(192, 168, 0, 0).To4(), + Mask: net.CIDRMask(24, 32), + }, + }, + { + name: "host", + in: "10.0.0.1", + expected: &net.IPNet{ + IP: net.IPv4(10, 0, 0, 1).To4(), + Mask: net.CIDRMask(32, 32), + }, + }, + } + for _, vtt := range tests { + tt := vtt + t.Run(tt.name, func(t *testing.T) { + result, err := ParseIPNet(tt.in) + assert.NoError(t, err) + assert.Equal(t, tt.expected, result) + + }) + } +} diff --git a/pkg/packet/afpacket/readwriter.go b/pkg/packet/afpacket/readwriter.go new file mode 100644 index 0000000..2a8356d --- /dev/null +++ b/pkg/packet/afpacket/readwriter.go @@ -0,0 +1,59 @@ +package afpacket + +import ( + "github.com/google/gopacket" + afp "github.com/google/gopacket/afpacket" + "github.com/google/gopacket/layers" + "github.com/google/gopacket/pcap" + "github.com/v-byte-cpu/sx/pkg/packet" + "golang.org/x/net/bpf" +) + +type AfPacketSource struct { + handle *afp.TPacket +} + +// Assert that AfPacketSource conforms to the packet.ReadWriter interface +var _ packet.ReadWriter = (*AfPacketSource)(nil) + +func NewPacketSource(iface string) (*AfPacketSource, error) { + handle, err := afp.NewTPacket(afp.SocketRaw, afp.OptInterface(iface)) + if err != nil { + return nil, err + } + return &AfPacketSource{handle}, nil +} + +// maxPacketLength is the maximum size of packets to capture in bytes. +// pcap calls it "snaplen" and default value used in tcpdump is 262144 bytes, +// that is redundant for most scans, see pcap(3) and tcpdump(1) for more info +func (s *AfPacketSource) SetBPFFilter(bpfFilter string, maxPacketLength int) error { + pcapBPF, err := pcap.CompileBPFFilter(layers.LinkTypeEthernet, maxPacketLength, bpfFilter) + if err != nil { + return err + } + bpfIns := make([]bpf.RawInstruction, 0, len(pcapBPF)) + for _, ins := range pcapBPF { + rawIns := bpf.RawInstruction{ + Op: ins.Code, + Jt: ins.Jt, + Jf: ins.Jf, + K: ins.K, + } + bpfIns = append(bpfIns, rawIns) + } + return s.handle.SetBPF(bpfIns) +} + +func (s *AfPacketSource) Close() { + s.handle.Close() +} + +func (s *AfPacketSource) ReadPacketData() ([]byte, *gopacket.CaptureInfo, error) { + data, ci, err := s.handle.ZeroCopyReadPacketData() + return data, &ci, err +} + +func (s *AfPacketSource) WritePacketData(pkt []byte) error { + return s.handle.WritePacketData(pkt) +} diff --git a/pkg/packet/mock_receiver_test.go b/pkg/packet/mock_receiver_test.go new file mode 100644 index 0000000..a818ecd --- /dev/null +++ b/pkg/packet/mock_receiver_test.go @@ -0,0 +1,126 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: receiver.go + +// Package packet is a generated GoMock package. +package packet + +import ( + context "context" + reflect "reflect" + + gomock "github.com/golang/mock/gomock" + gopacket "github.com/google/gopacket" +) + +// MockProcessor is a mock of Processor interface. +type MockProcessor struct { + ctrl *gomock.Controller + recorder *MockProcessorMockRecorder +} + +// MockProcessorMockRecorder is the mock recorder for MockProcessor. +type MockProcessorMockRecorder struct { + mock *MockProcessor +} + +// NewMockProcessor creates a new mock instance. +func NewMockProcessor(ctrl *gomock.Controller) *MockProcessor { + mock := &MockProcessor{ctrl: ctrl} + mock.recorder = &MockProcessorMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockProcessor) EXPECT() *MockProcessorMockRecorder { + return m.recorder +} + +// ProcessPacketData mocks base method. +func (m *MockProcessor) ProcessPacketData(data []byte, ci *gopacket.CaptureInfo) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ProcessPacketData", data, ci) + ret0, _ := ret[0].(error) + return ret0 +} + +// ProcessPacketData indicates an expected call of ProcessPacketData. +func (mr *MockProcessorMockRecorder) ProcessPacketData(data, ci interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ProcessPacketData", reflect.TypeOf((*MockProcessor)(nil).ProcessPacketData), data, ci) +} + +// MockReader is a mock of Reader interface. +type MockReader struct { + ctrl *gomock.Controller + recorder *MockReaderMockRecorder +} + +// MockReaderMockRecorder is the mock recorder for MockReader. +type MockReaderMockRecorder struct { + mock *MockReader +} + +// NewMockReader creates a new mock instance. +func NewMockReader(ctrl *gomock.Controller) *MockReader { + mock := &MockReader{ctrl: ctrl} + mock.recorder = &MockReaderMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockReader) EXPECT() *MockReaderMockRecorder { + return m.recorder +} + +// ReadPacketData mocks base method. +func (m *MockReader) ReadPacketData() ([]byte, *gopacket.CaptureInfo, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ReadPacketData") + ret0, _ := ret[0].([]byte) + ret1, _ := ret[1].(*gopacket.CaptureInfo) + ret2, _ := ret[2].(error) + return ret0, ret1, ret2 +} + +// ReadPacketData indicates an expected call of ReadPacketData. +func (mr *MockReaderMockRecorder) ReadPacketData() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReadPacketData", reflect.TypeOf((*MockReader)(nil).ReadPacketData)) +} + +// MockReceiver is a mock of Receiver interface. +type MockReceiver struct { + ctrl *gomock.Controller + recorder *MockReceiverMockRecorder +} + +// MockReceiverMockRecorder is the mock recorder for MockReceiver. +type MockReceiverMockRecorder struct { + mock *MockReceiver +} + +// NewMockReceiver creates a new mock instance. +func NewMockReceiver(ctrl *gomock.Controller) *MockReceiver { + mock := &MockReceiver{ctrl: ctrl} + mock.recorder = &MockReceiverMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockReceiver) EXPECT() *MockReceiverMockRecorder { + return m.recorder +} + +// ReceivePackets mocks base method. +func (m *MockReceiver) ReceivePackets(ctx context.Context) <-chan error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ReceivePackets", ctx) + ret0, _ := ret[0].(<-chan error) + return ret0 +} + +// ReceivePackets indicates an expected call of ReceivePackets. +func (mr *MockReceiverMockRecorder) ReceivePackets(ctx interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReceivePackets", reflect.TypeOf((*MockReceiver)(nil).ReceivePackets), ctx) +} diff --git a/pkg/packet/mock_sender_test.go b/pkg/packet/mock_sender_test.go new file mode 100644 index 0000000..146a968 --- /dev/null +++ b/pkg/packet/mock_sender_test.go @@ -0,0 +1,87 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: sender.go + +// Package packet is a generated GoMock package. +package packet + +import ( + context "context" + reflect "reflect" + + gomock "github.com/golang/mock/gomock" +) + +// MockSender is a mock of Sender interface. +type MockSender struct { + ctrl *gomock.Controller + recorder *MockSenderMockRecorder +} + +// MockSenderMockRecorder is the mock recorder for MockSender. +type MockSenderMockRecorder struct { + mock *MockSender +} + +// NewMockSender creates a new mock instance. +func NewMockSender(ctrl *gomock.Controller) *MockSender { + mock := &MockSender{ctrl: ctrl} + mock.recorder = &MockSenderMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockSender) EXPECT() *MockSenderMockRecorder { + return m.recorder +} + +// SendPackets mocks base method. +func (m *MockSender) SendPackets(ctx context.Context, in <-chan *BufferData) (<-chan interface{}, <-chan error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "SendPackets", ctx, in) + ret0, _ := ret[0].(<-chan interface{}) + ret1, _ := ret[1].(<-chan error) + return ret0, ret1 +} + +// SendPackets indicates an expected call of SendPackets. +func (mr *MockSenderMockRecorder) SendPackets(ctx, in interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendPackets", reflect.TypeOf((*MockSender)(nil).SendPackets), ctx, in) +} + +// MockWriter is a mock of Writer interface. +type MockWriter struct { + ctrl *gomock.Controller + recorder *MockWriterMockRecorder +} + +// MockWriterMockRecorder is the mock recorder for MockWriter. +type MockWriterMockRecorder struct { + mock *MockWriter +} + +// NewMockWriter creates a new mock instance. +func NewMockWriter(ctrl *gomock.Controller) *MockWriter { + mock := &MockWriter{ctrl: ctrl} + mock.recorder = &MockWriterMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockWriter) EXPECT() *MockWriterMockRecorder { + return m.recorder +} + +// WritePacketData mocks base method. +func (m *MockWriter) WritePacketData(pkt []byte) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "WritePacketData", pkt) + ret0, _ := ret[0].(error) + return ret0 +} + +// WritePacketData indicates an expected call of WritePacketData. +func (mr *MockWriterMockRecorder) WritePacketData(pkt interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WritePacketData", reflect.TypeOf((*MockWriter)(nil).WritePacketData), pkt) +} diff --git a/pkg/packet/readwriter.go b/pkg/packet/readwriter.go new file mode 100644 index 0000000..efccbb5 --- /dev/null +++ b/pkg/packet/readwriter.go @@ -0,0 +1,6 @@ +package packet + +type ReadWriter interface { + Reader + Writer +} diff --git a/pkg/packet/receiver.go b/pkg/packet/receiver.go new file mode 100644 index 0000000..c182992 --- /dev/null +++ b/pkg/packet/receiver.go @@ -0,0 +1,94 @@ +//go:generate mockgen -destination=mock_receiver_test.go -package=packet -source receiver.go + +package packet + +import ( + "context" + "io" + "net" + "strings" + "syscall" + "time" + + "github.com/google/gopacket" +) + +type Processor interface { + ProcessPacketData(data []byte, ci *gopacket.CaptureInfo) error +} + +type Reader interface { + ReadPacketData() (data []byte, ci *gopacket.CaptureInfo, err error) +} + +type Receiver interface { + ReceivePackets(ctx context.Context) <-chan error +} + +type receiver struct { + sr Reader + p Processor +} + +func NewReceiver(sr Reader, p Processor) Receiver { + return &receiver{sr, p} +} + +func isTemporaryError(err error) bool { + if err == syscall.EAGAIN { + return true + } + nerr, ok := err.(net.Error) + return ok && nerr.Temporary() +} + +func isUnrecoverableError(err error) bool { + switch err { + case io.EOF, io.ErrUnexpectedEOF, io.ErrNoProgress, + io.ErrClosedPipe, io.ErrShortBuffer, syscall.EBADF: + return true + default: + return strings.Contains(err.Error(), "use of closed file") + } +} + +func (r *receiver) ReceivePackets(ctx context.Context) <-chan error { + errc := make(chan error, 100) + go func() { + defer close(errc) + for { + select { + case <-ctx.Done(): + return + default: + data, ci, err := r.sr.ReadPacketData() + if err != nil { + // Immediately retry for temporary errors + if isTemporaryError(err) { + continue + } + if isUnrecoverableError(err) { + return + } + // Log unknown error + select { + case <-ctx.Done(): + return + case errc <- err: + } + // Sleep briefly and try again + time.Sleep(time.Millisecond * time.Duration(5)) + continue + } + if err := r.p.ProcessPacketData(data, ci); err != nil { + select { + case <-ctx.Done(): + return + case errc <- err: + } + } + } + } + }() + return errc +} diff --git a/pkg/packet/receiver_test.go b/pkg/packet/receiver_test.go new file mode 100644 index 0000000..e771da3 --- /dev/null +++ b/pkg/packet/receiver_test.go @@ -0,0 +1,217 @@ +package packet + +import ( + "context" + "errors" + "io" + "net" + "syscall" + "testing" + "time" + + "github.com/golang/mock/gomock" + "github.com/google/gopacket" + "github.com/stretchr/testify/assert" +) + +func newCaptureInfo() *gopacket.CaptureInfo { + return &gopacket.CaptureInfo{} +} + +func TestReceivePacketsWithUnrecoverableError(t *testing.T) { + t.Parallel() + tests := []struct { + name string + exitError error + }{ + { + name: "io.EOF", + exitError: io.EOF, + }, + { + name: "io.ErrUnexpectedEOF", + exitError: io.ErrUnexpectedEOF, + }, + { + name: "io.ErrNoProgress", + exitError: io.ErrNoProgress, + }, + { + name: "io.ErrClosedPipe", + exitError: io.ErrClosedPipe, + }, + { + name: "io.ErrShortBuffer", + exitError: io.ErrShortBuffer, + }, + { + name: "syscall.EBADF", + exitError: syscall.EBADF, + }, + { + name: "poll.ErrFileClosing", + exitError: errors.New("use of closed file"), + }, + } + + for _, vtt := range tests { + tt := vtt + t.Run(tt.name, func(t *testing.T) { + t.Parallel() + ctrl := gomock.NewController(t) + sr := NewMockReader(ctrl) + sr.EXPECT().ReadPacketData().Return(nil, nil, io.EOF) + p := NewMockProcessor(ctrl) + r := NewReceiver(sr, p) + + out := chanErrToGeneric(r.ReceivePackets(context.Background())) + result := chanToSlice(t, out, 0, 3*time.Second) + assert.Equal(t, 0, len(result), "error slice is not empty") + }) + } +} + +func TestReceivePacketsOnePacket(t *testing.T) { + t.Parallel() + + data := []byte{0x1, 0x2, 0x3} + expectedData := make([]byte, len(data)) + copy(expectedData, data) + + ctrl := gomock.NewController(t) + sr := NewMockReader(ctrl) + gomock.InOrder( + sr.EXPECT().ReadPacketData().Return(data, newCaptureInfo(), nil), + sr.EXPECT().ReadPacketData().Return(nil, nil, io.EOF), + ) + + p := NewMockProcessor(ctrl) + p.EXPECT(). + ProcessPacketData(expectedData, newCaptureInfo()).Return(nil) + r := NewReceiver(sr, p) + + out := chanErrToGeneric(r.ReceivePackets(context.Background())) + result := chanToSlice(t, out, 0, 3*time.Second) + assert.Equal(t, 0, len(result), "error slice is not empty") +} + +func TestReceivePacketsOnePacketWithProcessError(t *testing.T) { + t.Parallel() + + data := []byte{0x1, 0x2, 0x3} + + ctrl := gomock.NewController(t) + sr := NewMockReader(ctrl) + gomock.InOrder( + sr.EXPECT().ReadPacketData().Return(data, newCaptureInfo(), nil), + sr.EXPECT().ReadPacketData().Return(nil, nil, io.EOF), + ) + + p := NewMockProcessor(ctrl) + notNil := gomock.Not(gomock.Nil()) + p.EXPECT(). + ProcessPacketData(notNil, notNil).Return(errors.New("process error")) + r := NewReceiver(sr, p) + + out := chanErrToGeneric(r.ReceivePackets(context.Background())) + result := chanToSlice(t, out, 1, 3*time.Second) + assert.Equal(t, 1, len(result), "error slice is invalid") + assert.Error(t, result[0].(error)) +} + +func TestReceivePacketsOnePacketWithRetryError(t *testing.T) { + t.Parallel() + + tests := []struct { + name string + exitError error + }{ + { + name: "syscall.EAGAIN", + exitError: syscall.EAGAIN, + }, + { + name: "temporaryNetError", + exitError: &net.OpError{Op: "accept", Err: syscall.ECONNRESET}, + }, + } + + for _, vtt := range tests { + tt := vtt + t.Run(tt.name, func(t *testing.T) { + t.Parallel() + data := []byte{0x1, 0x2, 0x3} + expectedData := make([]byte, len(data)) + copy(expectedData, data) + + ctrl := gomock.NewController(t) + sr := NewMockReader(ctrl) + gomock.InOrder( + sr.EXPECT().ReadPacketData().Return(nil, nil, tt.exitError), + sr.EXPECT().ReadPacketData().Return(data, newCaptureInfo(), nil), + sr.EXPECT().ReadPacketData().Return(nil, nil, io.EOF), + ) + + p := NewMockProcessor(ctrl) + p.EXPECT(). + ProcessPacketData(expectedData, newCaptureInfo()).Return(nil) + r := NewReceiver(sr, p) + + out := chanErrToGeneric(r.ReceivePackets(context.Background())) + result := chanToSlice(t, out, 0, 3*time.Second) + assert.Equal(t, 0, len(result), "error slice is not empty") + }) + } +} + +func TestReceivePacketsOnePacketWithUnknownError(t *testing.T) { + t.Parallel() + + data := []byte{0x1, 0x2, 0x3} + expectedData := make([]byte, len(data)) + copy(expectedData, data) + + ctrl := gomock.NewController(t) + sr := NewMockReader(ctrl) + gomock.InOrder( + sr.EXPECT().ReadPacketData().Return(nil, nil, errors.New("unknown error")), + sr.EXPECT().ReadPacketData().Return(data, newCaptureInfo(), nil), + sr.EXPECT().ReadPacketData().Return(nil, nil, io.EOF), + ) + + p := NewMockProcessor(ctrl) + p.EXPECT(). + ProcessPacketData(expectedData, newCaptureInfo()).Return(nil) + r := NewReceiver(sr, p) + + out := chanErrToGeneric(r.ReceivePackets(context.Background())) + result := chanToSlice(t, out, 1, 3*time.Second) + assert.Equal(t, 1, len(result), "error slice length is invalid") + assert.Error(t, result[0].(error)) +} + +func TestReceivePacketsOnePacketWithContextCancel(t *testing.T) { + t.Parallel() + + data := []byte{0x1, 0x2, 0x3} + expectedData := make([]byte, len(data)) + copy(expectedData, data) + + ctrl := gomock.NewController(t) + sr := NewMockReader(ctrl) + sr.EXPECT().ReadPacketData().Return(data, newCaptureInfo(), nil) + + p := NewMockProcessor(ctrl) + ctx, cancel := context.WithCancel(context.Background()) + p.EXPECT(). + ProcessPacketData(expectedData, newCaptureInfo()).Return(nil). + Do(func(data []byte, ci *gopacket.CaptureInfo) error { + cancel() + return nil + }) + r := NewReceiver(sr, p) + + out := chanErrToGeneric(r.ReceivePackets(ctx)) + result := chanToSlice(t, out, 0, 3*time.Second) + assert.Equal(t, 0, len(result), "error slice is not empty") +} diff --git a/pkg/packet/sender.go b/pkg/packet/sender.go new file mode 100644 index 0000000..ecfb042 --- /dev/null +++ b/pkg/packet/sender.go @@ -0,0 +1,59 @@ +//go:generate mockgen -destination=mock_sender_test.go -package=packet -source sender.go + +package packet + +import ( + "context" + + "github.com/google/gopacket" +) + +type BufferData struct { + Buf gopacket.SerializeBuffer + Err error +} + +type Sender interface { + SendPackets(ctx context.Context, in <-chan *BufferData) (done <-chan interface{}, errc <-chan error) +} + +type Writer interface { + WritePacketData(pkt []byte) error +} + +type sender struct { + w Writer +} + +func NewSender(w Writer) Sender { + return &sender{w} +} + +func (s *sender) SendPackets(ctx context.Context, in <-chan *BufferData) (<-chan interface{}, <-chan error) { + done := make(chan interface{}) + errc := make(chan error, 100) + go func() { + defer func() { + close(done) + close(errc) + }() + for { + select { + case <-ctx.Done(): + return + case pkt, ok := <-in: + if !ok { + return + } + if pkt.Err != nil { + errc <- pkt.Err + continue + } + if err := s.w.WritePacketData(pkt.Buf.Bytes()); err != nil { + errc <- err + } + } + } + }() + return done, errc +} diff --git a/pkg/packet/sender_test.go b/pkg/packet/sender_test.go new file mode 100644 index 0000000..ed69f59 --- /dev/null +++ b/pkg/packet/sender_test.go @@ -0,0 +1,162 @@ +package packet + +import ( + "context" + "errors" + "testing" + "time" + + "github.com/golang/mock/gomock" + "github.com/google/gopacket" + "github.com/stretchr/testify/assert" + "github.com/stretchr/testify/require" +) + +func TestSenderWithEmptyChannel(t *testing.T) { + t.Parallel() + in := make(chan *BufferData) + close(in) + + ctrl := gomock.NewController(t) + w := NewMockWriter(ctrl) + s := NewSender(w) + + done, errc := s.SendPackets(context.Background(), in) + + out := chanErrToGeneric(errc) + result := chanToSlice(t, out, 0, 3*time.Second) + assert.Equal(t, 0, len(result), "error slice is not empty") + result = chanToSlice(t, done, 0, 3*time.Second) + assert.Equal(t, 0, len(result), "error slice is not empty") +} + +func TestSenderWithOnePacket(t *testing.T) { + t.Parallel() + in := make(chan *BufferData, 1) + data := []byte{0x1, 0x2, 0x3} + buffer := gopacket.NewSerializeBuffer() + gopacket.SerializeLayers(buffer, gopacket.SerializeOptions{}, gopacket.Payload(data)) + in <- &BufferData{Buf: buffer} + close(in) + + expectedData := make([]byte, len(data)) + copy(expectedData, data) + + ctrl := gomock.NewController(t) + w := NewMockWriter(ctrl) + w.EXPECT().WritePacketData(expectedData).Return(nil) + s := NewSender(w) + + done, errc := s.SendPackets(context.Background(), in) + + out := chanErrToGeneric(errc) + result := chanToSlice(t, out, 0, 3*time.Second) + assert.Equal(t, 0, len(result), "error slice is not empty") + result = chanToSlice(t, done, 0, 3*time.Second) + assert.Equal(t, 0, len(result), "error slice is not empty") +} + +func TestSenderWithTwoPackets(t *testing.T) { + t.Parallel() + in := make(chan *BufferData, 2) + + data := []byte{0x1, 0x2, 0x3} + buffer := gopacket.NewSerializeBuffer() + gopacket.SerializeLayers(buffer, gopacket.SerializeOptions{}, gopacket.Payload(data)) + in <- &BufferData{Buf: buffer} + + data2 := []byte{0x2, 0x3, 0x4} + buffer2 := gopacket.NewSerializeBuffer() + gopacket.SerializeLayers(buffer2, gopacket.SerializeOptions{}, gopacket.Payload(data2)) + in <- &BufferData{Buf: buffer2} + + close(in) + + expectedData := make([]byte, len(data)) + copy(expectedData, data) + expectedData2 := make([]byte, len(data2)) + copy(expectedData2, data2) + + ctrl := gomock.NewController(t) + w := NewMockWriter(ctrl) + gomock.InOrder( + w.EXPECT().WritePacketData(expectedData).Return(nil), + w.EXPECT().WritePacketData(expectedData2).Return(nil), + ) + s := NewSender(w) + + done, errc := s.SendPackets(context.Background(), in) + + out := chanErrToGeneric(errc) + result := chanToSlice(t, out, 0, 3*time.Second) + assert.Equal(t, 0, len(result), "error slice is not empty") + result = chanToSlice(t, done, 0, 3*time.Second) + assert.Equal(t, 0, len(result), "error slice is not empty") +} + +func TestSenderWithInvalidPacketReturnsError(t *testing.T) { + t.Parallel() + in := make(chan *BufferData, 1) + in <- &BufferData{Err: errors.New("invalid data")} + close(in) + + ctrl := gomock.NewController(t) + w := NewMockWriter(ctrl) + s := NewSender(w) + + done, errc := s.SendPackets(context.Background(), in) + + out := chanErrToGeneric(errc) + result := chanToSlice(t, out, 1, 3*time.Second) + assert.Equal(t, 1, len(result), "error slice size is invalid") + assert.Error(t, result[0].(error)) + + result = chanToSlice(t, done, 0, 3*time.Second) + assert.Equal(t, 0, len(result), "error slice is not empty") +} + +func TestSenderWithWriteErrorReturnsError(t *testing.T) { + t.Parallel() + in := make(chan *BufferData, 1) + + data := []byte{0x1, 0x2, 0x3} + buffer := gopacket.NewSerializeBuffer() + gopacket.SerializeLayers(buffer, gopacket.SerializeOptions{}, gopacket.Payload(data)) + in <- &BufferData{Buf: buffer} + close(in) + + ctrl := gomock.NewController(t) + w := NewMockWriter(ctrl) + w.EXPECT().WritePacketData(data).Return(errors.New("write error")) + s := NewSender(w) + + done, errc := s.SendPackets(context.Background(), in) + + out := chanErrToGeneric(errc) + result := chanToSlice(t, out, 1, 3*time.Second) + assert.Equal(t, 1, len(result), "error slice size is invalid") + assert.Error(t, result[0].(error)) + + result = chanToSlice(t, done, 0, 3*time.Second) + assert.Equal(t, 0, len(result), "error slice is not empty") +} + +func TestSenderWithTimeout(t *testing.T) { + t.Parallel() + + ctrl := gomock.NewController(t) + w := NewMockWriter(ctrl) + s := NewSender(w) + + ctx, cancel := context.WithTimeout(context.Background(), 1*time.Millisecond) + defer cancel() + + done, errc := s.SendPackets(ctx, nil) + select { + case <-errc: + case <-time.After(1 * time.Second): + require.FailNow(t, "exit timeout") + } + result := chanToSlice(t, done, 0, 3*time.Second) + assert.Equal(t, 0, len(result), "error slice is not empty") +} diff --git a/pkg/packet/utils_test.go b/pkg/packet/utils_test.go new file mode 100644 index 0000000..29ba0a7 --- /dev/null +++ b/pkg/packet/utils_test.go @@ -0,0 +1,40 @@ +package packet + +import ( + "testing" + "time" + + "github.com/stretchr/testify/require" +) + +func chanToSlice(t *testing.T, in <-chan interface{}, expectedLen int, timeout time.Duration) []interface{} { + t.Helper() + result := []interface{}{} +loop: + for { + select { + case data, ok := <-in: + if !ok { + break loop + } + if len(result) == expectedLen { + require.FailNow(t, "chan size is greater than expected, data:", data) + } + result = append(result, data) + case <-time.After(timeout): + t.Fatal("read timeout") + } + } + return result +} + +func chanErrToGeneric(in <-chan error) <-chan interface{} { + out := make(chan interface{}, cap(in)) + go func() { + defer close(out) + for i := range in { + out <- i + } + }() + return out +} diff --git a/pkg/scan/arp/arp.go b/pkg/scan/arp/arp.go new file mode 100644 index 0000000..4db4a36 --- /dev/null +++ b/pkg/scan/arp/arp.go @@ -0,0 +1,149 @@ +//go:generate easyjson -output_filename result_easyjson.go arp.go + +package arp + +import ( + "context" + "fmt" + "net" + "runtime" + + "github.com/google/gopacket" + "github.com/google/gopacket/layers" + "github.com/google/gopacket/macs" + "github.com/v-byte-cpu/sx/pkg/packet" + "github.com/v-byte-cpu/sx/pkg/scan" +) + +type ScanMethod struct { + gen *scan.PacketMultiGenerator + results chan *ScanResult + internalResults chan *ScanResult + ctx context.Context + + rcvEth layers.Ethernet + rcvARP layers.ARP + rcvDecoded []gopacket.LayerType + rcvMacPrefix [3]byte + parser *gopacket.DecodingLayerParser +} + +//easyjson:json +type ScanResult struct { + IP string `json:"ip"` + MAC string `json:"mac"` + Vendor string `json:"vendor"` +} + +func (r *ScanResult) String() string { + return fmt.Sprintf("%-20s %-20s %s", r.IP, r.MAC, r.Vendor) +} + +// Assert that arp.ScanMethod conforms to the scan.Method interface +var _ scan.Method = (*ScanMethod)(nil) + +func NewScanMethod(ctx context.Context) *ScanMethod { + f := newPacketFiller() + gen := scan.NewPacketMultiGenerator(f, runtime.NumCPU()) + + results := make(chan *ScanResult, 1000) + internalResults := make(chan *ScanResult, 1000) + + copyChans := func() { + defer close(results) + for { + select { + case <-ctx.Done(): + return + case v := <-internalResults: + select { + case <-ctx.Done(): + return + case results <- v: + } + } + } + } + go copyChans() + + sm := &ScanMethod{gen: gen, results: results, internalResults: internalResults, ctx: ctx} + parser := gopacket.NewDecodingLayerParser(layers.LayerTypeEthernet, &sm.rcvEth, &sm.rcvARP) + parser.IgnoreUnsupported = true + sm.parser = parser + return sm +} + +func (s *ScanMethod) Results() <-chan *ScanResult { + return s.results +} + +func (s *ScanMethod) Packets(ctx context.Context, r *scan.Range) <-chan *packet.BufferData { + pairs, err := scan.IPPortPairs(ctx, r) + if err != nil { + out := make(chan *packet.BufferData, 1) + out <- &packet.BufferData{Err: err} + close(out) + return out + } + return s.gen.Packets(ctx, pairs) +} + +func (s *ScanMethod) ProcessPacketData(data []byte, ci *gopacket.CaptureInfo) error { + // try to exit as early as possible + select { + case <-s.ctx.Done(): + return nil + default: + } + + err := s.parser.DecodeLayers(data, &s.rcvDecoded) + if err != nil { + return err + } + for _, layerType := range s.rcvDecoded { + if layerType == layers.LayerTypeARP { + copy(s.rcvMacPrefix[:], s.rcvARP.SourceHwAddress[:3]) + hwVendor := macs.ValidMACPrefixMap[s.rcvMacPrefix] + + select { + case <-s.ctx.Done(): + case s.internalResults <- &ScanResult{ + IP: net.IP(s.rcvARP.SourceProtAddress).String(), + MAC: net.HardwareAddr(s.rcvARP.SourceHwAddress).String(), + Vendor: hwVendor, + }: + } + return nil + } + } + return nil +} + +type packetFiller struct{} + +func newPacketFiller() *packetFiller { + return &packetFiller{} +} + +func (f *packetFiller) Fill(packet gopacket.SerializeBuffer, pair *scan.Request) error { + eth := &layers.Ethernet{ + SrcMAC: pair.SrcMAC, + DstMAC: net.HardwareAddr{0xff, 0xff, 0xff, 0xff, 0xff, 0xff}, + EthernetType: layers.EthernetTypeARP, + } + + a := &layers.ARP{ + AddrType: layers.LinkTypeEthernet, + Protocol: layers.EthernetTypeIPv4, + HwAddressSize: uint8(6), + ProtAddressSize: uint8(4), + Operation: layers.ARPRequest, + SourceHwAddress: pair.SrcMAC, + SourceProtAddress: pair.SrcIP, + DstHwAddress: net.HardwareAddr{0x00, 0x00, 0x00, 0x00, 0x00, 0x00}, + DstProtAddress: pair.DstIP.To4(), + } + + var opt gopacket.SerializeOptions + return gopacket.SerializeLayers(packet, opt, eth, a) +} diff --git a/pkg/scan/arp/arp_test.go b/pkg/scan/arp/arp_test.go new file mode 100644 index 0000000..2e48d5f --- /dev/null +++ b/pkg/scan/arp/arp_test.go @@ -0,0 +1,65 @@ +package arp + +import ( + "context" + "net" + "testing" + "time" + + "github.com/google/gopacket" + "github.com/google/gopacket/layers" + "github.com/stretchr/testify/assert" + "github.com/stretchr/testify/require" +) + +func TestProcessPacketData(t *testing.T) { + t.Parallel() + + ctx, cancel := context.WithCancel(context.Background()) + defer cancel() + sm := NewScanMethod(ctx) + + // generate packet data + packet := gopacket.NewSerializeBuffer() + eth := &layers.Ethernet{ + SrcMAC: net.HardwareAddr{0x1, 0x2, 0x3, 0x4, 0x5, 0x6}, + DstMAC: net.HardwareAddr{0x10, 0x11, 0x12, 0x13, 0x14, 0x15}, + EthernetType: layers.EthernetTypeARP, + } + + a := &layers.ARP{ + AddrType: layers.LinkTypeEthernet, + Protocol: layers.EthernetTypeIPv4, + HwAddressSize: uint8(6), + ProtAddressSize: uint8(4), + Operation: layers.ARPRequest, + SourceHwAddress: net.HardwareAddr{0x1, 0x2, 0x3, 0x4, 0x5, 0x6}, + SourceProtAddress: net.IPv4(192, 168, 0, 3).To4(), + DstHwAddress: net.HardwareAddr{0x10, 0x11, 0x12, 0x13, 0x14, 0x15}, + DstProtAddress: net.IPv4(192, 168, 0, 2).To4(), + } + var opt gopacket.SerializeOptions + err := gopacket.SerializeLayers(packet, opt, eth, a) + require.NoError(t, err) + + err = sm.ProcessPacketData(packet.Bytes(), &gopacket.CaptureInfo{}) + require.NoError(t, err) + + select { + case result, ok := <-sm.Results(): + if !ok { + require.FailNow(t, "results chan is empty") + } + assert.Equal(t, net.HardwareAddr{0x1, 0x2, 0x3, 0x4, 0x5, 0x6}.String(), result.MAC) + assert.Equal(t, net.IPv4(192, 168, 0, 3).To4().String(), result.IP) + cancel() + select { + case _, ok := <-sm.Results(): + require.False(t, ok, "results chan is not closed") + case <-time.After(1 * time.Second): + t.Fatal("read timeout") + } + case <-time.After(3 * time.Second): + t.Fatal("read timeout") + } +} diff --git a/pkg/scan/arp/filter.go b/pkg/scan/arp/filter.go new file mode 100644 index 0000000..f4d19e4 --- /dev/null +++ b/pkg/scan/arp/filter.go @@ -0,0 +1,19 @@ +package arp + +import ( + "fmt" + + "github.com/v-byte-cpu/sx/pkg/scan" +) + +// For ARP packets: +// Ethernet header (14 bytes) + ARP packet (28 bytes) + FCS (4 bytes) = 46 bytes +// So it is less than Ethernet minimum frame size = 64 bytes +const MaxPacketLength = 64 + +func BPFFilter(r *scan.Range) (filter string, maxPacketLength int) { + if r.Subnet == nil { + return "arp", MaxPacketLength + } + return fmt.Sprintf("arp src net %s", r.Subnet.String()), MaxPacketLength +} diff --git a/pkg/scan/arp/filter_test.go b/pkg/scan/arp/filter_test.go new file mode 100644 index 0000000..5f3240d --- /dev/null +++ b/pkg/scan/arp/filter_test.go @@ -0,0 +1,43 @@ +package arp + +import ( + "net" + "testing" + + "github.com/stretchr/testify/assert" + "github.com/v-byte-cpu/sx/pkg/scan" +) + +func TestBPFFilter(t *testing.T) { + t.Parallel() + + tests := []struct { + name string + scanRange *scan.Range + expectedFilter string + }{ + { + name: "EmptySubnet", + expectedFilter: "arp", + scanRange: &scan.Range{}, + }, + { + name: "OneSubnet", + scanRange: &scan.Range{ + Subnet: &net.IPNet{ + IP: net.IPv4(192, 168, 0, 0), + Mask: net.CIDRMask(24, 32), + }, + }, + expectedFilter: "arp src net 192.168.0.0/24", + }, + } + + for _, tt := range tests { + t.Run(tt.name, func(t *testing.T) { + filter, maxPacketLength := BPFFilter(tt.scanRange) + assert.Equal(t, tt.expectedFilter, filter) + assert.Equal(t, maxPacketLength, MaxPacketLength) + }) + } +} diff --git a/pkg/scan/arp/result_easyjson.go b/pkg/scan/arp/result_easyjson.go new file mode 100644 index 0000000..3dd59a1 --- /dev/null +++ b/pkg/scan/arp/result_easyjson.go @@ -0,0 +1,99 @@ +// Code generated by easyjson for marshaling/unmarshaling. DO NOT EDIT. + +package arp + +import ( + json "encoding/json" + easyjson "github.com/mailru/easyjson" + jlexer "github.com/mailru/easyjson/jlexer" + jwriter "github.com/mailru/easyjson/jwriter" +) + +// suppress unused package warning +var ( + _ *json.RawMessage + _ *jlexer.Lexer + _ *jwriter.Writer + _ easyjson.Marshaler +) + +func easyjsonD3b49167DecodeGithubComVByteCpuSxPkgScanArp(in *jlexer.Lexer, out *ScanResult) { + isTopLevel := in.IsStart() + if in.IsNull() { + if isTopLevel { + in.Consumed() + } + in.Skip() + return + } + in.Delim('{') + for !in.IsDelim('}') { + key := in.UnsafeFieldName(false) + in.WantColon() + if in.IsNull() { + in.Skip() + in.WantComma() + continue + } + switch key { + case "ip": + out.IP = string(in.String()) + case "mac": + out.MAC = string(in.String()) + case "vendor": + out.Vendor = string(in.String()) + default: + in.SkipRecursive() + } + in.WantComma() + } + in.Delim('}') + if isTopLevel { + in.Consumed() + } +} +func easyjsonD3b49167EncodeGithubComVByteCpuSxPkgScanArp(out *jwriter.Writer, in ScanResult) { + out.RawByte('{') + first := true + _ = first + { + const prefix string = ",\"ip\":" + out.RawString(prefix[1:]) + out.String(string(in.IP)) + } + { + const prefix string = ",\"mac\":" + out.RawString(prefix) + out.String(string(in.MAC)) + } + { + const prefix string = ",\"vendor\":" + out.RawString(prefix) + out.String(string(in.Vendor)) + } + out.RawByte('}') +} + +// MarshalJSON supports json.Marshaler interface +func (v ScanResult) MarshalJSON() ([]byte, error) { + w := jwriter.Writer{} + easyjsonD3b49167EncodeGithubComVByteCpuSxPkgScanArp(&w, v) + return w.Buffer.BuildBytes(), w.Error +} + +// MarshalEasyJSON supports easyjson.Marshaler interface +func (v ScanResult) MarshalEasyJSON(w *jwriter.Writer) { + easyjsonD3b49167EncodeGithubComVByteCpuSxPkgScanArp(w, v) +} + +// UnmarshalJSON supports json.Unmarshaler interface +func (v *ScanResult) UnmarshalJSON(data []byte) error { + r := jlexer.Lexer{Data: data} + easyjsonD3b49167DecodeGithubComVByteCpuSxPkgScanArp(&r, v) + return r.Error() +} + +// UnmarshalEasyJSON supports easyjson.Unmarshaler interface +func (v *ScanResult) UnmarshalEasyJSON(l *jlexer.Lexer) { + easyjsonD3b49167DecodeGithubComVByteCpuSxPkgScanArp(l, v) +} diff --git a/pkg/scan/engine.go b/pkg/scan/engine.go new file mode 100644 index 0000000..bfd4f5a --- /dev/null +++ b/pkg/scan/engine.go @@ -0,0 +1,75 @@ +//go:generate mockgen -package scan -destination=mock_engine_test.go -source=engine.go + +package scan + +import ( + "context" + "net" + "sync" + + "github.com/v-byte-cpu/sx/pkg/packet" +) + +type Range struct { + SrcIP net.IP + SrcMAC net.HardwareAddr + Interface *net.Interface + Subnet *net.IPNet + StartPort uint16 + EndPort uint16 +} + +type PacketSource interface { + Packets(ctx context.Context, r *Range) <-chan *packet.BufferData +} + +type Engine struct { + src PacketSource + rcv packet.Receiver + snd packet.Sender +} + +func NewEngine(ps PacketSource, s packet.Sender, r packet.Receiver) *Engine { + return &Engine{src: ps, snd: s, rcv: r} +} + +func (e *Engine) Start(ctx context.Context, r *Range) (<-chan interface{}, <-chan error) { + packets := e.src.Packets(ctx, r) + errc1 := e.rcv.ReceivePackets(ctx) + done, errc2 := e.snd.SendPackets(ctx, packets) + return done, mergeErrChan(ctx, errc1, errc2) +} + +// generics would be helpful :) +func mergeErrChan(ctx context.Context, channels ...<-chan error) <-chan error { + var wg sync.WaitGroup + wg.Add(len(channels)) + + out := make(chan error) + multiplex := func(c <-chan error) { + defer wg.Done() + for { + select { + case <-ctx.Done(): + return + case e, ok := <-c: + if !ok { + return + } + select { + case <-ctx.Done(): + return + case out <- e: + } + } + } + } + for _, c := range channels { + go multiplex(c) + } + go func() { + wg.Wait() + close(out) + }() + return out +} diff --git a/pkg/scan/engine_test.go b/pkg/scan/engine_test.go new file mode 100644 index 0000000..74711f8 --- /dev/null +++ b/pkg/scan/engine_test.go @@ -0,0 +1,154 @@ +//go:generate mockgen -package scan -destination=mock_sendreceiver_test.go github.com/v-byte-cpu/sx/pkg/packet Sender,Receiver + +package scan + +import ( + "context" + "errors" + "net" + "testing" + "time" + + "github.com/golang/mock/gomock" + "github.com/stretchr/testify/assert" + "github.com/stretchr/testify/require" + "github.com/v-byte-cpu/sx/pkg/packet" +) + +func chanErrorToSlice(t *testing.T, in <-chan error, expectedLen int, timeout time.Duration) []error { + t.Helper() + result := []error{} +loop: + for { + select { + case data, ok := <-in: + if !ok { + break loop + } + if len(result) == expectedLen { + require.FailNow(t, "chan size is greater than expected, data:", data) + } + result = append(result, data) + case <-time.After(timeout): + t.Fatal("read timeout") + } + } + return result +} + +// generics would be helpful :) +func chanRequestToSlice(t *testing.T, in <-chan *Request, expectedLen int, timeout time.Duration) []*Request { + t.Helper() + result := []*Request{} +loop: + for { + select { + case data, ok := <-in: + if !ok { + break loop + } + if len(result) == expectedLen { + require.FailNow(t, "chan size is greater than expected, data:", data) + } + result = append(result, data) + case <-time.After(timeout): + t.Fatal("read timeout") + } + } + return result +} + +func TestMergeErrChanEmptyChannels(t *testing.T) { + t.Parallel() + c1 := make(chan error) + close(c1) + c2 := make(chan error) + close(c2) + out := mergeErrChan(context.Background(), c1, c2) + + result := chanErrorToSlice(t, out, 0, 3*time.Second) + assert.Equal(t, 0, len(result), "error slice is not empty") +} + +func TestMergeErrChanOneElementAndEmptyChannel(t *testing.T) { + t.Parallel() + c1 := make(chan error, 1) + c1 <- errors.New("test error") + close(c1) + c2 := make(chan error) + close(c2) + out := mergeErrChan(context.Background(), c1, c2) + + result := chanErrorToSlice(t, out, 1, 3*time.Second) + assert.Equal(t, 1, len(result), "error slice size is invalid") + assert.Error(t, result[0]) +} + +func TestMergeErrChanTwoElements(t *testing.T) { + t.Parallel() + c1 := make(chan error, 1) + c1 <- errors.New("test error") + close(c1) + c2 := make(chan error, 1) + c2 <- errors.New("test error") + close(c2) + out := mergeErrChan(context.Background(), c1, c2) + + result := chanErrorToSlice(t, out, 2, 3*time.Second) + assert.Equal(t, 2, len(result), "error slice size is invalid") + assert.Error(t, result[0]) + assert.Error(t, result[1]) +} + +func TestMergeErrChanContextExit(t *testing.T) { + t.Parallel() + c1 := make(chan error) + defer close(c1) + c2 := make(chan error) + defer close(c2) + + ctx, cancel := context.WithTimeout(context.Background(), 1*time.Millisecond) + defer cancel() + out := mergeErrChan(ctx, c1, c2) + + result := chanErrorToSlice(t, out, 0, 3*time.Second) + assert.Equal(t, 0, len(result), "error slice is not empty") +} + +func TestEngineStartCollectsAllErrors(t *testing.T) { + t.Parallel() + + ctrl := gomock.NewController(t) + ps := NewMockPacketSource(ctrl) + notNil := gomock.Not(gomock.Nil()) + packets := make(chan *packet.BufferData) + close(packets) + ps.EXPECT().Packets(notNil, notNil).Return(packets).AnyTimes() + + errc1 := make(chan error, 1) + errc1 <- errors.New("send error") + close(errc1) + s := NewMockSender(ctrl) + s.EXPECT().SendPackets(notNil, notNil).Return(nil, errc1) + + errc2 := make(chan error, 1) + errc2 <- errors.New("receive error") + close(errc2) + r := NewMockReceiver(ctrl) + r.EXPECT().ReceivePackets(notNil).Return(errc2) + e := NewEngine(ps, s, r) + + _, out := e.Start(context.Background(), &Range{ + Subnet: &net.IPNet{ + IP: net.IPv4(192, 168, 0, 1), + Mask: net.CIDRMask(32, 32), + }, + StartPort: 888, + EndPort: 888, + }) + + result := chanErrorToSlice(t, out, 2, 3*time.Second) + assert.Equal(t, 2, len(result), "error slice is invalid") + assert.Error(t, result[0]) + assert.Error(t, result[1]) +} diff --git a/pkg/scan/generator.go b/pkg/scan/generator.go new file mode 100644 index 0000000..fdcbc19 --- /dev/null +++ b/pkg/scan/generator.go @@ -0,0 +1,100 @@ +//go:generate mockgen -package scan -destination=mock_generator_test.go -source generator.go + +package scan + +import ( + "context" + "sync" + + "github.com/google/gopacket" + "github.com/v-byte-cpu/sx/pkg/packet" +) + +type PacketFiller interface { + Fill(packet gopacket.SerializeBuffer, pair *Request) error +} + +type PacketGenerator struct { + filler PacketFiller +} + +func NewPacketGenerator(filler PacketFiller) *PacketGenerator { + return &PacketGenerator{filler} +} + +func (g *PacketGenerator) Packets(ctx context.Context, in <-chan *Request) <-chan *packet.BufferData { + out := make(chan *packet.BufferData) + go func() { + defer close(out) + for { + select { + case <-ctx.Done(): + return + case pair, ok := <-in: + if !ok { + return + } + // TODO buffer pool + buf := gopacket.NewSerializeBuffer() + if err := g.filler.Fill(buf, pair); err != nil { + out <- &packet.BufferData{Err: err} + continue + } + out <- &packet.BufferData{Buf: buf} + } + } + }() + return out +} + +type PacketMultiGenerator struct { + gen *PacketGenerator + numWorkers int +} + +func NewPacketMultiGenerator(filler PacketFiller, numWorkers int) *PacketMultiGenerator { + gen := &PacketGenerator{filler} + return &PacketMultiGenerator{gen, numWorkers} +} + +func (g *PacketMultiGenerator) Packets(ctx context.Context, in <-chan *Request) <-chan *packet.BufferData { + workers := make([]<-chan *packet.BufferData, g.numWorkers) + for i := 0; i < g.numWorkers; i++ { + workers[i] = g.gen.Packets(ctx, in) + } + return MergeBufferDataChan(ctx, workers...) +} + +// generics would be helpful :) +func MergeBufferDataChan(ctx context.Context, channels ...<-chan *packet.BufferData) <-chan *packet.BufferData { + var wg sync.WaitGroup + wg.Add(len(channels)) + + out := make(chan *packet.BufferData) + multiplex := func(c <-chan *packet.BufferData) { + defer wg.Done() + for { + select { + case <-ctx.Done(): + return + case e, ok := <-c: + if !ok { + return + } + select { + case <-ctx.Done(): + return + case out <- e: + } + } + } + } + for _, c := range channels { + go multiplex(c) + } + go func() { + wg.Wait() + close(out) + }() + return out +} diff --git a/pkg/scan/generator_test.go b/pkg/scan/generator_test.go new file mode 100644 index 0000000..6efe3a0 --- /dev/null +++ b/pkg/scan/generator_test.go @@ -0,0 +1,330 @@ +package scan + +import ( + "context" + "errors" + "net" + "runtime" + "testing" + "time" + + "github.com/golang/mock/gomock" + "github.com/stretchr/testify/assert" + "github.com/stretchr/testify/require" + "github.com/v-byte-cpu/sx/pkg/packet" +) + +func chanBufferDataToGeneric(in <-chan *packet.BufferData) <-chan interface{} { + out := make(chan interface{}, cap(in)) + go func() { + defer close(out) + for i := range in { + out <- i + } + }() + return out +} + +func chanToSlice(t *testing.T, in <-chan interface{}, expectedLen int, timeout time.Duration) []interface{} { + t.Helper() + result := []interface{}{} +loop: + for { + select { + case data, ok := <-in: + if !ok { + break loop + } + if len(result) == expectedLen { + require.FailNow(t, "chan size is greater than expected, data:", data) + } + result = append(result, data) + case <-time.After(timeout): + t.Fatal("read timeout") + } + } + return result +} + +func TestGeneratorPacketsWithEmptyChannel(t *testing.T) { + t.Parallel() + in := make(chan *Request) + close(in) + + ctrl := gomock.NewController(t) + f := NewMockPacketFiller(ctrl) + g := NewPacketGenerator(f) + + out := chanBufferDataToGeneric(g.Packets(context.Background(), in)) + result := chanToSlice(t, out, 0, 3*time.Second) + assert.Equal(t, 0, len(result), "result is not empty") +} + +func TestMultiGeneratorPacketsWithEmptyChannel(t *testing.T) { + t.Parallel() + in := make(chan *Request) + close(in) + + ctrl := gomock.NewController(t) + f := NewMockPacketFiller(ctrl) + g := NewPacketMultiGenerator(f, runtime.NumCPU()) + + out := chanBufferDataToGeneric(g.Packets(context.Background(), in)) + result := chanToSlice(t, out, 0, 3*time.Second) + assert.Equal(t, 0, len(result), "result is not empty") +} + +func TestGeneratorPacketsWithOnePair(t *testing.T) { + t.Parallel() + port := uint16(888) + + in := make(chan *Request, 1) + in <- &Request{DstIP: net.IPv4(192, 168, 0, 1).To4(), DstPort: port} + close(in) + + ctrl := gomock.NewController(t) + f := NewMockPacketFiller(ctrl) + f.EXPECT(). + Fill(gomock.Not(gomock.Nil()), + &Request{DstIP: net.IPv4(192, 168, 0, 1).To4(), DstPort: port}) + + g := NewPacketGenerator(f) + + out := chanBufferDataToGeneric(g.Packets(context.Background(), in)) + results := chanToSlice(t, out, 1, 3*time.Second) + + assert.Equal(t, 1, len(results), "result size is invalid") + result := results[0].(*packet.BufferData) + assert.NoError(t, result.Err) + assert.NotNil(t, result.Buf) +} + +func TestMultiGeneratorPacketsWithOnePair(t *testing.T) { + t.Parallel() + port := uint16(888) + + in := make(chan *Request, 1) + in <- &Request{DstIP: net.IPv4(192, 168, 0, 1).To4(), DstPort: port} + close(in) + + ctrl := gomock.NewController(t) + f := NewMockPacketFiller(ctrl) + f.EXPECT(). + Fill(gomock.Not(gomock.Nil()), + &Request{DstIP: net.IPv4(192, 168, 0, 1).To4(), DstPort: port}) + + g := NewPacketMultiGenerator(f, runtime.NumCPU()) + + out := chanBufferDataToGeneric(g.Packets(context.Background(), in)) + results := chanToSlice(t, out, 1, 3*time.Second) + + assert.Equal(t, 1, len(results), "result size is invalid") + result := results[0].(*packet.BufferData) + assert.NoError(t, result.Err) + assert.NotNil(t, result.Buf) +} + +func TestGeneratorPacketsWithTwoPairs(t *testing.T) { + t.Parallel() + port := uint16(888) + + in := make(chan *Request, 2) + in <- &Request{DstIP: net.IPv4(192, 168, 0, 1).To4(), DstPort: port} + in <- &Request{DstIP: net.IPv4(192, 168, 0, 1).To4(), DstPort: port + 1} + close(in) + + ctrl := gomock.NewController(t) + f := NewMockPacketFiller(ctrl) + f.EXPECT(). + Fill(gomock.Not(gomock.Nil()), + &Request{DstIP: net.IPv4(192, 168, 0, 1).To4(), DstPort: port}) + f.EXPECT(). + Fill(gomock.Not(gomock.Nil()), + &Request{DstIP: net.IPv4(192, 168, 0, 1).To4(), DstPort: port + 1}) + g := NewPacketGenerator(f) + + out := chanBufferDataToGeneric(g.Packets(context.Background(), in)) + results := chanToSlice(t, out, 2, 3*time.Second) + + assert.Equal(t, 2, len(results), "result size is invalid") + result1 := results[0].(*packet.BufferData) + result2 := results[1].(*packet.BufferData) + assert.NoError(t, result1.Err) + assert.NotNil(t, result1.Buf) + assert.NoError(t, result2.Err) + assert.NotNil(t, result2.Buf) +} + +func TestMultiGeneratorPacketsWithTwoPairs(t *testing.T) { + t.Parallel() + port := uint16(888) + + in := make(chan *Request, 2) + in <- &Request{DstIP: net.IPv4(192, 168, 0, 1).To4(), DstPort: port} + in <- &Request{DstIP: net.IPv4(192, 168, 0, 1).To4(), DstPort: port + 1} + close(in) + + ctrl := gomock.NewController(t) + f := NewMockPacketFiller(ctrl) + f.EXPECT(). + Fill(gomock.Not(gomock.Nil()), + &Request{DstIP: net.IPv4(192, 168, 0, 1).To4(), DstPort: port}) + f.EXPECT(). + Fill(gomock.Not(gomock.Nil()), + &Request{DstIP: net.IPv4(192, 168, 0, 1).To4(), DstPort: port + 1}) + + g := NewPacketMultiGenerator(f, runtime.NumCPU()) + + out := chanBufferDataToGeneric(g.Packets(context.Background(), in)) + results := chanToSlice(t, out, 2, 3*time.Second) + + assert.Equal(t, 2, len(results), "result size is invalid") + result1 := results[0].(*packet.BufferData) + result2 := results[1].(*packet.BufferData) + assert.NoError(t, result1.Err) + assert.NotNil(t, result1.Buf) + assert.NoError(t, result2.Err) + assert.NotNil(t, result2.Buf) +} + +func TestGeneratorPacketsWithOnePairReturnsError(t *testing.T) { + t.Parallel() + port := uint16(888) + + in := make(chan *Request, 1) + in <- &Request{DstIP: net.IPv4(192, 168, 0, 1).To4(), DstPort: port} + close(in) + + ctrl := gomock.NewController(t) + f := NewMockPacketFiller(ctrl) + f.EXPECT(). + Fill(gomock.Not(gomock.Nil()), + &Request{DstIP: net.IPv4(192, 168, 0, 1).To4(), DstPort: port}). + Return(errors.New("failed request")) + g := NewPacketGenerator(f) + + out := chanBufferDataToGeneric(g.Packets(context.Background(), in)) + results := chanToSlice(t, out, 1, 3*time.Second) + + assert.Equal(t, 1, len(results), "result size is invalid") + result := results[0].(*packet.BufferData) + assert.Error(t, result.Err) + assert.Nil(t, result.Buf) +} + +func TestMultiGeneratorPacketsWithOnePairReturnsError(t *testing.T) { + t.Parallel() + port := uint16(888) + + in := make(chan *Request, 1) + in <- &Request{DstIP: net.IPv4(192, 168, 0, 1).To4(), DstPort: port} + close(in) + + ctrl := gomock.NewController(t) + f := NewMockPacketFiller(ctrl) + f.EXPECT(). + Fill(gomock.Not(gomock.Nil()), + &Request{DstIP: net.IPv4(192, 168, 0, 1).To4(), DstPort: port}). + Return(errors.New("failed request")) + + g := NewPacketMultiGenerator(f, runtime.NumCPU()) + + out := chanBufferDataToGeneric(g.Packets(context.Background(), in)) + results := chanToSlice(t, out, 1, 3*time.Second) + + assert.Equal(t, 1, len(results), "result size is invalid") + result := results[0].(*packet.BufferData) + assert.Error(t, result.Err) + assert.Nil(t, result.Buf) +} + +func TestGeneratorPacketsWithTimeout(t *testing.T) { + t.Parallel() + + ctrl := gomock.NewController(t) + f := NewMockPacketFiller(ctrl) + g := NewPacketGenerator(f) + + ctx, cancel := context.WithTimeout(context.Background(), 1*time.Millisecond) + defer cancel() + out := g.Packets(ctx, nil) + select { + case <-out: + case <-time.After(1 * time.Second): + require.FailNow(t, "exit timeout") + } +} + +func TestMultiGeneratorPacketsWithTimeout(t *testing.T) { + t.Parallel() + + ctrl := gomock.NewController(t) + f := NewMockPacketFiller(ctrl) + g := NewPacketMultiGenerator(f, runtime.NumCPU()) + + ctx, cancel := context.WithTimeout(context.Background(), 1*time.Millisecond) + defer cancel() + out := g.Packets(ctx, nil) + select { + case <-out: + case <-time.After(1 * time.Second): + require.FailNow(t, "exit timeout") + } +} + +func TestMergeBufferDataChanEmptyChannels(t *testing.T) { + t.Parallel() + c1 := make(chan *packet.BufferData) + close(c1) + c2 := make(chan *packet.BufferData) + close(c2) + out := chanBufferDataToGeneric(MergeBufferDataChan(context.Background(), c1, c2)) + + result := chanToSlice(t, out, 0, 3*time.Second) + assert.Equal(t, 0, len(result), "result slice is not empty") +} + +func TestMergeBufferDataChanOneElementAndEmptyChannel(t *testing.T) { + t.Parallel() + c1 := make(chan *packet.BufferData, 1) + c1 <- &packet.BufferData{} + close(c1) + c2 := make(chan *packet.BufferData) + close(c2) + out := chanBufferDataToGeneric(MergeBufferDataChan(context.Background(), c1, c2)) + + result := chanToSlice(t, out, 1, 3*time.Second) + assert.Equal(t, 1, len(result), "result slice size is invalid") + assert.NotNil(t, result[0]) +} + +func TestMergeBufferDataChanTwoElements(t *testing.T) { + t.Parallel() + c1 := make(chan *packet.BufferData, 1) + c1 <- &packet.BufferData{} + close(c1) + c2 := make(chan *packet.BufferData, 1) + c2 <- &packet.BufferData{} + close(c2) + out := chanBufferDataToGeneric(MergeBufferDataChan(context.Background(), c1, c2)) + + result := chanToSlice(t, out, 2, 3*time.Second) + assert.Equal(t, 2, len(result), "result slice size is invalid") + assert.NotNil(t, result[0]) + assert.NotNil(t, result[1]) +} + +func TestMergeBufferDataChanContextExit(t *testing.T) { + t.Parallel() + c1 := make(chan *packet.BufferData) + defer close(c1) + c2 := make(chan *packet.BufferData) + defer close(c2) + + ctx, cancel := context.WithTimeout(context.Background(), 1*time.Millisecond) + defer cancel() + out := chanBufferDataToGeneric(MergeBufferDataChan(ctx, c1, c2)) + + result := chanToSlice(t, out, 0, 3*time.Second) + assert.Equal(t, 0, len(result), "result slice is not empty") +} diff --git a/pkg/scan/inject_engine.go b/pkg/scan/inject_engine.go new file mode 100644 index 0000000..b5a50c1 --- /dev/null +++ b/pkg/scan/inject_engine.go @@ -0,0 +1,25 @@ +//+build wireinject + +package scan + +import ( + "github.com/google/wire" + "github.com/v-byte-cpu/sx/pkg/packet" +) + +type Method interface { + PacketSource + packet.Processor +} + +func SetupEngine(rw packet.ReadWriter, m Method) *Engine { + wire.Build(NewEngine, + packet.NewSender, + packet.NewReceiver, + wire.Bind(new(PacketSource), new(Method)), + wire.Bind(new(packet.Processor), new(Method)), + wire.Bind(new(packet.Reader), new(packet.ReadWriter)), + wire.Bind(new(packet.Writer), new(packet.ReadWriter)), + ) + return nil +} diff --git a/pkg/scan/mock_engine_test.go b/pkg/scan/mock_engine_test.go new file mode 100644 index 0000000..d515a75 --- /dev/null +++ b/pkg/scan/mock_engine_test.go @@ -0,0 +1,50 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: engine.go + +// Package scan is a generated GoMock package. +package scan + +import ( + context "context" + reflect "reflect" + + gomock "github.com/golang/mock/gomock" + packet "github.com/v-byte-cpu/sx/pkg/packet" +) + +// MockPacketSource is a mock of PacketSource interface. +type MockPacketSource struct { + ctrl *gomock.Controller + recorder *MockPacketSourceMockRecorder +} + +// MockPacketSourceMockRecorder is the mock recorder for MockPacketSource. +type MockPacketSourceMockRecorder struct { + mock *MockPacketSource +} + +// NewMockPacketSource creates a new mock instance. +func NewMockPacketSource(ctrl *gomock.Controller) *MockPacketSource { + mock := &MockPacketSource{ctrl: ctrl} + mock.recorder = &MockPacketSourceMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockPacketSource) EXPECT() *MockPacketSourceMockRecorder { + return m.recorder +} + +// Packets mocks base method. +func (m *MockPacketSource) Packets(ctx context.Context, r *Range) <-chan *packet.BufferData { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Packets", ctx, r) + ret0, _ := ret[0].(<-chan *packet.BufferData) + return ret0 +} + +// Packets indicates an expected call of Packets. +func (mr *MockPacketSourceMockRecorder) Packets(ctx, r interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Packets", reflect.TypeOf((*MockPacketSource)(nil).Packets), ctx, r) +} diff --git a/pkg/scan/mock_generator_test.go b/pkg/scan/mock_generator_test.go new file mode 100644 index 0000000..f21ed82 --- /dev/null +++ b/pkg/scan/mock_generator_test.go @@ -0,0 +1,49 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: generator.go + +// Package scan is a generated GoMock package. +package scan + +import ( + reflect "reflect" + + gomock "github.com/golang/mock/gomock" + gopacket "github.com/google/gopacket" +) + +// MockPacketFiller is a mock of PacketFiller interface. +type MockPacketFiller struct { + ctrl *gomock.Controller + recorder *MockPacketFillerMockRecorder +} + +// MockPacketFillerMockRecorder is the mock recorder for MockPacketFiller. +type MockPacketFillerMockRecorder struct { + mock *MockPacketFiller +} + +// NewMockPacketFiller creates a new mock instance. +func NewMockPacketFiller(ctrl *gomock.Controller) *MockPacketFiller { + mock := &MockPacketFiller{ctrl: ctrl} + mock.recorder = &MockPacketFillerMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockPacketFiller) EXPECT() *MockPacketFillerMockRecorder { + return m.recorder +} + +// Fill mocks base method. +func (m *MockPacketFiller) Fill(packet gopacket.SerializeBuffer, pair *Request) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Fill", packet, pair) + ret0, _ := ret[0].(error) + return ret0 +} + +// Fill indicates an expected call of Fill. +func (mr *MockPacketFillerMockRecorder) Fill(packet, pair interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Fill", reflect.TypeOf((*MockPacketFiller)(nil).Fill), packet, pair) +} diff --git a/pkg/scan/mock_sendreceiver_test.go b/pkg/scan/mock_sendreceiver_test.go new file mode 100644 index 0000000..8d0a877 --- /dev/null +++ b/pkg/scan/mock_sendreceiver_test.go @@ -0,0 +1,88 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: github.com/v-byte-cpu/sx/pkg/packet (interfaces: Sender,Receiver) + +// Package scan is a generated GoMock package. +package scan + +import ( + context "context" + reflect "reflect" + + gomock "github.com/golang/mock/gomock" + packet "github.com/v-byte-cpu/sx/pkg/packet" +) + +// MockSender is a mock of Sender interface. +type MockSender struct { + ctrl *gomock.Controller + recorder *MockSenderMockRecorder +} + +// MockSenderMockRecorder is the mock recorder for MockSender. +type MockSenderMockRecorder struct { + mock *MockSender +} + +// NewMockSender creates a new mock instance. +func NewMockSender(ctrl *gomock.Controller) *MockSender { + mock := &MockSender{ctrl: ctrl} + mock.recorder = &MockSenderMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockSender) EXPECT() *MockSenderMockRecorder { + return m.recorder +} + +// SendPackets mocks base method. +func (m *MockSender) SendPackets(arg0 context.Context, arg1 <-chan *packet.BufferData) (<-chan interface{}, <-chan error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "SendPackets", arg0, arg1) + ret0, _ := ret[0].(<-chan interface{}) + ret1, _ := ret[1].(<-chan error) + return ret0, ret1 +} + +// SendPackets indicates an expected call of SendPackets. +func (mr *MockSenderMockRecorder) SendPackets(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendPackets", reflect.TypeOf((*MockSender)(nil).SendPackets), arg0, arg1) +} + +// MockReceiver is a mock of Receiver interface. +type MockReceiver struct { + ctrl *gomock.Controller + recorder *MockReceiverMockRecorder +} + +// MockReceiverMockRecorder is the mock recorder for MockReceiver. +type MockReceiverMockRecorder struct { + mock *MockReceiver +} + +// NewMockReceiver creates a new mock instance. +func NewMockReceiver(ctrl *gomock.Controller) *MockReceiver { + mock := &MockReceiver{ctrl: ctrl} + mock.recorder = &MockReceiverMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockReceiver) EXPECT() *MockReceiverMockRecorder { + return m.recorder +} + +// ReceivePackets mocks base method. +func (m *MockReceiver) ReceivePackets(arg0 context.Context) <-chan error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ReceivePackets", arg0) + ret0, _ := ret[0].(<-chan error) + return ret0 +} + +// ReceivePackets indicates an expected call of ReceivePackets. +func (mr *MockReceiverMockRecorder) ReceivePackets(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReceivePackets", reflect.TypeOf((*MockReceiver)(nil).ReceivePackets), arg0) +} diff --git a/pkg/scan/range.go b/pkg/scan/range.go new file mode 100644 index 0000000..f0761ba --- /dev/null +++ b/pkg/scan/range.go @@ -0,0 +1,47 @@ +package scan + +import ( + "context" + "errors" + "net" + + "github.com/v-byte-cpu/sx/pkg/ip" +) + +var ErrPortRange = errors.New("invalid port range") +var ErrSubnet = errors.New("invalid subnet") + +type Request struct { + SrcIP net.IP + SrcMAC []byte + DstIP net.IP + DstMAC []byte + DstPort uint16 + Meta map[string]interface{} +} + +func IPPortPairs(ctx context.Context, r *Range) (<-chan *Request, error) { + if r.StartPort > r.EndPort { + return nil, ErrPortRange + } + if r.Subnet == nil { + return nil, ErrSubnet + } + out := make(chan *Request) + go func() { + defer close(out) + for port := r.StartPort; port <= r.EndPort; port++ { + ipnet := r.Subnet + for ipaddr := ipnet.IP.Mask(ipnet.Mask); ipnet.Contains(ipaddr); ip.Inc(ipaddr) { + select { + case <-ctx.Done(): + return + case out <- &Request{ + SrcIP: r.SrcIP, SrcMAC: r.SrcMAC, + DstIP: ip.DupIP(ipaddr), DstPort: port}: + } + } + } + }() + return out, nil +} diff --git a/pkg/scan/range_test.go b/pkg/scan/range_test.go new file mode 100644 index 0000000..adeecaf --- /dev/null +++ b/pkg/scan/range_test.go @@ -0,0 +1,219 @@ +package scan + +import ( + "context" + "net" + "testing" + "time" + + "github.com/stretchr/testify/assert" + "github.com/stretchr/testify/require" +) + +func newScanRange(opts ...scanRangeOption) *Range { + sr := &Range{ + SrcIP: net.IPv4(192, 168, 0, 3), + SrcMAC: net.HardwareAddr{0x1, 0x2, 0x3, 0x4, 0x5, 0x6}, + StartPort: 22, + EndPort: 888, + Subnet: &net.IPNet{ + IP: net.IPv4(192, 168, 0, 0), + Mask: net.CIDRMask(24, 32), + }, + } + for _, o := range opts { + o(sr) + } + return sr +} + +type scanRangeOption func(sr *Range) + +func withStartPort(startPort uint16) scanRangeOption { + return func(sr *Range) { + sr.StartPort = startPort + } +} + +func withEndPort(endPort uint16) scanRangeOption { + return func(sr *Range) { + sr.EndPort = endPort + } +} + +func withSubnet(subnet *net.IPNet) scanRangeOption { + return func(sr *Range) { + sr.Subnet = subnet + } +} + +func newScanRequest(opts ...scanRequestOption) *Request { + r := &Request{ + SrcIP: net.IPv4(192, 168, 0, 3), + SrcMAC: net.HardwareAddr{0x1, 0x2, 0x3, 0x4, 0x5, 0x6}, + } + for _, o := range opts { + o(r) + } + return r +} + +type scanRequestOption func(sr *Request) + +func withDstIP(dstIP net.IP) scanRequestOption { + return func(sr *Request) { + sr.DstIP = dstIP + } +} + +func withDstPort(dstPort uint16) scanRequestOption { + return func(sr *Request) { + sr.DstPort = dstPort + } +} + +func TestIPPortPairsWithInvalidInput(t *testing.T) { + tests := []struct { + name string + startPort uint16 + endPort uint16 + subnets []net.IPNet + scanRange *Range + }{ + { + name: "InvalidPortRange", + scanRange: newScanRange(withStartPort(5000), withEndPort(2000)), + }, + { + name: "NilSubnet", + scanRange: newScanRange(withSubnet(nil)), + }, + } + for _, vtt := range tests { + tt := vtt + t.Run(tt.name, func(t *testing.T) { + _, err := IPPortPairs(context.Background(), tt.scanRange) + assert.Error(t, err) + }) + } +} + +func comparePairChanToSlice(t *testing.T, expected []*Request, in <-chan *Request, timeout time.Duration) { + t.Helper() + result := pairChanToSlice(t, in, len(expected), timeout) + require.Equal(t, expected, result) +} + +func pairChanToSlice(t *testing.T, in <-chan *Request, expectedLen int, timeout time.Duration) []*Request { + t.Helper() + result := []*Request{} +loop: + for { + select { + case data, ok := <-in: + if !ok { + break loop + } + if len(result) == expectedLen { + require.FailNow(t, "chan size is greater than expected, data:", data) + } + result = append(result, data) + case <-time.After(timeout): + t.Fatal("read timeout") + } + } + return result +} + +func TestIPPortPairsWithOneIpOnePort(t *testing.T) { + t.Parallel() + port := uint16(888) + pairs, err := IPPortPairs(context.Background(), + newScanRange( + withSubnet(&net.IPNet{IP: net.IPv4(192, 168, 0, 1), Mask: net.CIDRMask(32, 32)}), + withStartPort(port), + withEndPort(port), + )) + assert.NoError(t, err) + + expected := []*Request{ + newScanRequest(withDstIP(net.IPv4(192, 168, 0, 1).To4()), withDstPort(port)), + } + comparePairChanToSlice(t, expected, pairs, 5*time.Second) +} + +func TestIPPortPairsWithOneIpTwoPorts(t *testing.T) { + t.Parallel() + port := uint16(888) + pairs, err := IPPortPairs(context.Background(), + newScanRange( + withSubnet(&net.IPNet{IP: net.IPv4(192, 168, 0, 1), Mask: net.CIDRMask(32, 32)}), + withStartPort(port), + withEndPort(port+1), + )) + assert.NoError(t, err) + + expected := []*Request{ + newScanRequest(withDstIP(net.IPv4(192, 168, 0, 1).To4()), withDstPort(port)), + newScanRequest(withDstIP(net.IPv4(192, 168, 0, 1).To4()), withDstPort(port+1)), + } + comparePairChanToSlice(t, expected, pairs, 5*time.Second) +} + +func TestIPPortPairsWithTwoIpsOnePort(t *testing.T) { + t.Parallel() + port := uint16(888) + pairs, err := IPPortPairs(context.Background(), + newScanRange( + withSubnet(&net.IPNet{IP: net.IPv4(192, 168, 0, 1), Mask: net.CIDRMask(31, 32)}), + withStartPort(port), + withEndPort(port), + )) + assert.NoError(t, err) + + expected := []*Request{ + newScanRequest(withDstIP(net.IPv4(192, 168, 0, 0).To4()), withDstPort(port)), + newScanRequest(withDstIP(net.IPv4(192, 168, 0, 1).To4()), withDstPort(port)), + } + comparePairChanToSlice(t, expected, pairs, 5*time.Second) +} + +func TestIPPortPairsWithFourIpsOnePort(t *testing.T) { + t.Parallel() + port := uint16(888) + pairs, err := IPPortPairs(context.Background(), + newScanRange( + withSubnet(&net.IPNet{IP: net.IPv4(192, 168, 0, 1), Mask: net.CIDRMask(30, 32)}), + withStartPort(port), + withEndPort(port), + )) + assert.NoError(t, err) + + expected := []*Request{ + newScanRequest(withDstIP(net.IPv4(192, 168, 0, 0).To4()), withDstPort(port)), + newScanRequest(withDstIP(net.IPv4(192, 168, 0, 1).To4()), withDstPort(port)), + newScanRequest(withDstIP(net.IPv4(192, 168, 0, 2).To4()), withDstPort(port)), + newScanRequest(withDstIP(net.IPv4(192, 168, 0, 3).To4()), withDstPort(port)), + } + comparePairChanToSlice(t, expected, pairs, 5*time.Second) +} + +func TestIPPortPairsWithTwoIpsTwoPorts(t *testing.T) { + t.Parallel() + port := uint16(888) + pairs, err := IPPortPairs(context.Background(), + newScanRange( + withSubnet(&net.IPNet{IP: net.IPv4(192, 168, 0, 1), Mask: net.CIDRMask(31, 32)}), + withStartPort(port), + withEndPort(port+1), + )) + assert.NoError(t, err) + + expected := []*Request{ + newScanRequest(withDstIP(net.IPv4(192, 168, 0, 0).To4()), withDstPort(port)), + newScanRequest(withDstIP(net.IPv4(192, 168, 0, 1).To4()), withDstPort(port)), + newScanRequest(withDstIP(net.IPv4(192, 168, 0, 0).To4()), withDstPort(port+1)), + newScanRequest(withDstIP(net.IPv4(192, 168, 0, 1).To4()), withDstPort(port+1)), + } + comparePairChanToSlice(t, expected, pairs, 5*time.Second) +} diff --git a/pkg/scan/wire_gen.go b/pkg/scan/wire_gen.go new file mode 100644 index 0000000..61021b4 --- /dev/null +++ b/pkg/scan/wire_gen.go @@ -0,0 +1,26 @@ +// Code generated by Wire. DO NOT EDIT. + +//go:generate go run github.com/google/wire/cmd/wire +//+build !wireinject + +package scan + +import ( + "github.com/v-byte-cpu/sx/pkg/packet" +) + +// Injectors from inject_engine.go: + +func SetupEngine(rw packet.ReadWriter, m Method) *Engine { + sender := packet.NewSender(rw) + receiver := packet.NewReceiver(rw, m) + engine := NewEngine(m, sender, receiver) + return engine +} + +// inject_engine.go: + +type Method interface { + PacketSource + packet.Processor +}