Remove empty lines

Author: krhubert

cmd/run.go

@@ -17,7 +17,7 @@ import (
 var (
 	fast           bool
 	ifaceName      string
-	simulatorNames = []string{"c2-dns", "dga", "scan", "spambot", "tunnel"}
+	simulatorNames = []string{"c2-dns", "c2-ip", "dga", "scan", "spambot", "tunnel"}
 )
 
 func newRunCommand() *cobra.Command {
@@ -86,6 +86,14 @@ var allsimualtors = []simulatorInfo{
 		1 * time.Second,
 		false,
 	},
+	{
+		"c2-ip",
+		[]string{"Preparing random sample of current C2 IP:port pairs"},
+		"Connecting to %s",
+		simulator.NewC2IP(),
+		1 * time.Second,
+		true,
+	},
 	{
 		"dga",
 		[]string{"Generating list of DGA domains"},

simulator/c2_ip.go

@@ -0,0 +1,66 @@
+package simulator
+
+import (
+	"context"
+	"encoding/json"
+	"io/ioutil"
+	"math/rand"
+	"net"
+	"net/http"
+
+	"github.com/pkg/errors"
+)
+
+// C2IP simulator.
+type C2IP struct{}
+
+// NewC2IP creates c2 dns simulator.
+func NewC2IP() *C2IP {
+	return &C2IP{}
+}
+
+// Simulate c2 dns traffic.
+func (*C2IP) Simulate(ctx context.Context, extIP net.IP, host string) error {
+	d := &net.Dialer{
+		LocalAddr: &net.TCPAddr{IP: extIP},
+	}
+
+	conn, err := d.DialContext(ctx, "tcp", host)
+	if err != nil {
+		return err
+	}
+	conn.Close()
+	return nil
+}
+
+// Hosts returns hosts marked c2 dns threat.
+func (t *C2IP) Hosts() ([]string, error) {
+	const nLookup = 10
+	resp, err := http.Get("https://api.open.wisdom.alphasoc.net/v1/c2-ip")
+	if err != nil {
+		return nil, err
+	}
+	defer resp.Body.Close()
+
+	b, err := ioutil.ReadAll(resp.Body)
+	if err != nil {
+		return nil, errors.Wrapf(err, "alnfo read body error")
+	}
+
+	response := &struct {
+		Hosts []string `json:"hosts"`
+	}{}
+
+	if err := json.Unmarshal(b, response); err != nil {
+		return nil, errors.Wrapf(err, "alnfo parse body error")
+	}
+
+	var (
+		hosts []string
+		idx   = rand.Perm(len(response.Hosts))
+	)
+	for n, i := 0, 0; n < len(response.Hosts) && i < nLookup; n, i = n+1, i+1 {
+		hosts = append(hosts, response.Hosts[idx[n]])
+	}
+	return hosts, nil
+}

simulator/scan.go

@@ -48,7 +48,6 @@ func (*PortScan) Simulate(ctx context.Context, extIP net.IP, host string) error
 	conn, err := d.DialContext(ctx, "tcp", host)
 	if err != nil {
 		return err
-
 	}
 	conn.Close()
 	return nil