Add max file size support

README.md

@@ -63,7 +63,7 @@ INPUT:
 OUTPUT:
    -es, -estimate      estimate permutation count without generating payloads
    -o, -output string  output file to write altered subdomain list
-   -v, -verbose        display verbose output
+   -ms, -max-size int  Max export data size in bytes (default 9223372036854775807)   -v, -verbose        display verbose output
    -silent             display results only
    -version            display alterx version
 

cmd/alterx/main.go

@@ -19,6 +19,7 @@ func main() {
 		Payloads: cliOpts.Payloads,
 		Limit:    cliOpts.Limit,
 		Enrich:   cliOpts.Enrich, // enrich payloads
+		MaxSize: cliOpts.MaxSize,
 	}
 
 	if cliOpts.PermutationConfig != "" {

internal/runner/runner.go

@@ -3,6 +3,7 @@ package runner
 import (
 	"fmt"
 	"io"
+	"math"
 	"os"
 	"strings"
 
@@ -26,6 +27,7 @@ type Options struct {
 	Silent             bool
 	Enrich             bool
 	Limit              int
+	MaxSize            int
 	// internal/unexported fields
 	wordlists goflags.RuntimeMap
 }
@@ -44,6 +46,7 @@ func ParseFlags() *Options {
 	flagSet.CreateGroup("output", "Output",
 		flagSet.BoolVarP(&opts.Estimate, "estimate", "es", false, "estimate permutation count without generating payloads"),
 		flagSet.StringVarP(&opts.Output, "output", "o", "", "output file to write altered subdomain list"),
+		flagSet.IntVarP(&opts.MaxSize, "max-size", "ms", math.MaxInt, "Max export data size in bytes"),
 		flagSet.BoolVarP(&opts.Verbose, "verbose", "v", false, "display verbose output"),
 		flagSet.BoolVar(&opts.Silent, "silent", false, "display results only"),
 		flagSet.CallbackVar(printVersion, "version", "display alterx version"),
@@ -71,6 +74,10 @@ func ParseFlags() *Options {
 		}
 	}
 
+	if opts.MaxSize < 0 {
+		gologger.Fatal().Msgf("max-size cannot be negative")
+	}
+
 	if opts.Silent {
 		gologger.DefaultLogger.SetMaxLevel(levels.LevelSilent)
 	} else if opts.Verbose {

mutator.go

@@ -37,6 +37,8 @@ type Options struct {
 	// Enrich when true alterx extra possible words from input
 	// and adds them to default payloads word,number
 	Enrich bool
+	// MaxSize limits output data size
+	MaxSize int
 }
 
 // Mutator
@@ -138,6 +140,7 @@ func (m *Mutator) ExecuteWithWriter(Writer io.Writer) error {
 	}
 	resChan := m.Execute(context.TODO())
 	m.payloadCount = 0
+	maxFileSize := m.Options.MaxSize
 	for {
 		value, ok := <-resChan
 		if !ok {
@@ -148,11 +151,23 @@ func (m *Mutator) ExecuteWithWriter(Writer io.Writer) error {
 			// we can't early exit, due to abstraction we have to conclude the elaboration to drain all dedupers
 			continue
 		}
-		_, err := Writer.Write([]byte(value + "\n"))
-		m.payloadCount++
+		if maxFileSize <= 0 {
+			// drain all dedupers when max-file size reached
+			continue
+		}
+		outputData := []byte(value + "\n")
+		if len(outputData) > maxFileSize {
+			maxFileSize = 0
+			continue
+		}
+
+		n, err := Writer.Write(outputData)
 		if err != nil {
 			return err
 		}
+		// update maxFileSize limit after each write
+		maxFileSize -= n
+		m.payloadCount++
 	}
 }