Create flag --start

gui/gui.go

@@ -34,7 +34,7 @@ type Options struct {
 
 type runner func(ctx context.Context, t terminalapi.Terminal, c *container.Container, opts ...termdash.Option) error
 
-func Run(targetURL string, storage storage.Reader, attacker attacker.Attacker, opts Options) error {
+func Run(targetURL string, storage storage.Reader, attacker attacker.Attacker, opts Options, start bool) error {
 	var (
 		t   terminalapi.Terminal
 		err error
@@ -48,10 +48,10 @@ func Run(targetURL string, storage storage.Reader, attacker attacker.Attacker, o
 		return fmt.Errorf("failed to generate terminal interface: %w", err)
 	}
 	defer t.Close()
-	return run(t, termdash.Run, targetURL, storage, attacker, opts)
+	return run(t, termdash.Run, targetURL, storage, attacker, opts, start)
 }
 
-func run(t terminalapi.Terminal, r runner, targetURL string, storage storage.Reader, a attacker.Attacker, opts Options) error {
+func run(t terminalapi.Terminal, r runner, targetURL string, storage storage.Reader, a attacker.Attacker, opts Options, start bool) error {
 	ctx, cancel := context.WithCancel(context.Background())
 	defer cancel()
 
@@ -96,6 +96,11 @@ func run(t terminalapi.Terminal, r runner, targetURL string, storage storage.Rea
 
 	k := keybinds(ctx, cancel, c, d, a)
 
+	// Start the attack immediately if the start flag is set
+	if start {
+		attack(ctx, d, a)
+	}
+
 	return r(ctx, t, c, termdash.KeyboardSubscriber(k), termdash.RedrawInterval(opts.RedrawInternal))
 }
 

gui/gui_test.go

@@ -43,7 +43,7 @@ func TestRun(t *testing.T) {
 	}
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
-			err := run(&termbox.Terminal{}, tt.runner, "", &storage.FakeStorage{}, &attacker.FakeAttacker{}, Options{})
+			err := run(&termbox.Terminal{}, tt.runner, "", &storage.FakeStorage{}, &attacker.FakeAttacker{}, Options{}, false)
 			assert.Equal(t, tt.wantErr, err != nil)
 		})
 	}

main.go

@@ -58,6 +58,7 @@ type cli struct {
 	tlsCertFile        string
 	tlsKeyFile         string
 	caCert             string
+	start              bool
 
 	//options for gui
 	queryRange     time.Duration
@@ -100,6 +101,7 @@ func parseFlags(stdout, stderr io.Writer) (*cli, error) {
 	flagSet.StringVar(&c.localAddress, "local-addr", "0.0.0.0", "Local IP address.")
 	flagSet.BoolVar(&c.insecureSkipVerify, "insecure", false, "Skip TLS verification")
 	flagSet.StringVar(&c.caCert, "cacert", "", "PEM ca certificate file")
+	flagSet.BoolVar(&c.start, "start", false, "Start the attack immediately")
 	flagSet.StringVar(&c.tlsCertFile, "cert", "", "PEM encoded tls certificate file to use")
 	flagSet.StringVar(&c.tlsKeyFile, "key", "", "PEM encoded tls private key file to use")
 	// TODO: Re-enable when making it capable of drawing histogram bar chart.
@@ -155,12 +157,12 @@ func (c *cli) run(args []string) int {
 	}
 	setDebug(nil, c.debug)
 
-	if err := gui.Run(target, s, a,
-		gui.Options{
-			QueryRange:     c.queryRange,
-			RedrawInternal: c.redrawInterval,
-		},
-	); err != nil {
+	guiOptions := gui.Options{
+		QueryRange:     c.queryRange,
+		RedrawInternal: c.redrawInterval,
+	}
+
+	if err := gui.Run(target, s, a, guiOptions, c.start); err != nil {
 		fmt.Fprintf(c.stderr, "failed to start application: %s\n", err.Error())
 		c.usage()
 		return 1