Merge pull request #114 from aykhans/core/refactor

General refactor

README.md

@@ -84,6 +84,7 @@ Send 1000 GET requests to https://example.com with 10 parallel dodos (threads),
     "dodos": 10,
     "requests": 1000,
     "duration": "250s",
+    "skip_verify": false,
 
     "params": [
         // A random value will be selected from the list for first "key1" param on each request
@@ -162,6 +163,7 @@ timeout: "800ms"
 dodos: 10
 requests: 1000
 duration: "250s"
+skip_verify: false
 
 params:
     # A random value will be selected from the list for first "key1" param on each request
@@ -248,7 +250,7 @@ If `Headers`, `Params`, `Cookies`, `Body`, or `Proxy` fields have multiple value
 
 | Parameter       | config file | CLI Flag     | CLI Short Flag | Type                           | Description                                                 | Default |
 | --------------- | ----------- | ------------ | -------------- | ------------------------------ | ----------------------------------------------------------- | ------- |
-| Config file     | -           | -config-file | -f             | String                         | Path to local config file or http(s) URL of the config file | -       |
+| Config file     |             | -config-file | -f             | String                         | Path to local config file or http(s) URL of the config file | -       |
 | Yes             | yes         | -yes         | -y             | Boolean                        | Answer yes to all questions                                 | false   |
 | URL             | url         | -url         | -u             | String                         | URL to send the request to                                  | -       |
 | Method          | method      | -method      | -m             | String                         | HTTP method                                                 | GET     |
@@ -261,3 +263,4 @@ If `Headers`, `Params`, `Cookies`, `Body`, or `Proxy` fields have multiple value
 | Cookies         | cookies     | -cookie      | -c             | [{String: String OR [String]}] | Request cookies                                             | -       |
 | Body            | body        | -body        | -b             | String OR [String]             | Request body or list of request bodies                      | -       |
 | Proxy           | proxies     | -proxy       | -x             | String OR [String]             | Proxy URL or list of proxy URLs                             | -       |
+| Skip Verify     | skip_verify | -skip-verify |                | Boolean                        | Skip SSL/TLS certificate verification                       | false   |

Taskfile.yaml

@@ -20,8 +20,16 @@ vars:
 tasks:
     run: go run main.go
 
+    ftl:
+        cmds:
+            - task: fmt
+            - task: tidy
+            - task: lint
+
     fmt: gofmt -w -d .
 
+    tidy: go mod tidy
+
     lint: golangci-lint run
 
     build: go build -ldflags "-s -w" -o "dodo"

config.json

@@ -6,6 +6,7 @@
     "dodos": 8,
     "requests": 1000,
     "duration": "10s",
+    "skip_verify": false,
 
     "params": [
         { "key1": ["value1", "value2", "value3", "value4"] },

config.yaml

@@ -5,6 +5,7 @@ timeout: "5s"
 dodos: 8
 requests: 1000
 duration: "10s"
+skip_verify: false
 
 params:
     - key1: ["value1", "value2", "value3", "value4"]

config/cli.go

@@ -31,7 +31,7 @@ Usage with all flags:
     -p "param1=value1" -param "param2=value2" \
     -c "cookie1=value1" -cookie "cookie2=value2" \
     -x "http://proxy.example.com:8080" -proxy "socks5://proxy2.example.com:8080" \
-    -y
+    -skip-verify -y
 
 Flags:
   -h, -help                   help for dodo
@@ -48,7 +48,8 @@ Flags:
   -p, -param        [string]  Parameter for the request (e.g. "key1=value1")
   -H, -header       [string]  Header for the request (e.g. "key1:value1")
   -c, -cookie       [string]  Cookie for the request (e.g. "key1=value1")
-  -x, -proxy        [string]  Proxy for the request (e.g. "http://proxy.example.com:8080")`
+  -x, -proxy        [string]  Proxy for the request (e.g. "http://proxy.example.com:8080")
+  -skip-verify      bool      Skip SSL/TLS certificate verification (default %v)`
 
 func (config *Config) ReadCLI() (types.ConfigFile, error) {
 	flag.Usage = func() {
@@ -58,6 +59,7 @@ func (config *Config) ReadCLI() (types.ConfigFile, error) {
 			DefaultDodosCount,
 			DefaultTimeout,
 			DefaultMethod,
+			DefaultSkipVerify,
 		)
 	}
 
@@ -65,6 +67,7 @@ func (config *Config) ReadCLI() (types.ConfigFile, error) {
 		version      = false
 		configFile   = ""
 		yes          = false
+		skipVerify   = false
 		method       = ""
 		url          types.RequestURL
 		dodosCount   = uint(0)
@@ -83,6 +86,8 @@ func (config *Config) ReadCLI() (types.ConfigFile, error) {
 		flag.BoolVar(&yes, "yes", false, "Answer yes to all questions")
 		flag.BoolVar(&yes, "y", false, "Answer yes to all questions")
 
+		flag.BoolVar(&skipVerify, "skip-verify", false, "Skip SSL/TLS certificate verification")
+
 		flag.StringVar(&method, "method", "", "HTTP Method")
 		flag.StringVar(&method, "m", "", "HTTP Method")
 
@@ -149,6 +154,8 @@ func (config *Config) ReadCLI() (types.ConfigFile, error) {
 			config.Timeout = &types.Timeout{Duration: timeout}
 		case "yes", "y":
 			config.Yes = utils.ToPtr(yes)
+		case "skip-verify":
+			config.SkipVerify = utils.ToPtr(skipVerify)
 		}
 	})
 

config/config.go

@@ -15,7 +15,7 @@ import (
 )
 
 const (
-	VERSION             string        = "0.6.3"
+	VERSION             string        = "0.6.5"
 	DefaultUserAgent    string        = "Dodo/" + VERSION
 	DefaultMethod       string        = "GET"
 	DefaultTimeout      time.Duration = time.Second * 10
@@ -23,6 +23,7 @@ const (
 	DefaultRequestCount uint          = 0
 	DefaultDuration     time.Duration = 0
 	DefaultYes          bool          = false
+	DefaultSkipVerify   bool          = false
 )
 
 var SupportedProxySchemes []string = []string{"http", "socks5", "socks5h"}
@@ -35,6 +36,7 @@ type RequestConfig struct {
 	RequestCount uint
 	Duration     time.Duration
 	Yes          bool
+	SkipVerify   bool
 	Params       types.Params
 	Headers      types.Headers
 	Cookies      types.Cookies
@@ -51,6 +53,7 @@ func NewRequestConfig(conf *Config) *RequestConfig {
 		RequestCount: *conf.RequestCount,
 		Duration:     conf.Duration.Duration,
 		Yes:          *conf.Yes,
+		SkipVerify:   *conf.SkipVerify,
 		Params:       conf.Params,
 		Headers:      conf.Headers,
 		Cookies:      conf.Cookies,
@@ -122,6 +125,8 @@ func (rc *RequestConfig) Print() {
 	t.AppendRow(table.Row{"Proxy", rc.Proxies.String()})
 	t.AppendSeparator()
 	t.AppendRow(table.Row{"Body", rc.Body.String()})
+	t.AppendSeparator()
+	t.AppendRow(table.Row{"Skip Verify", rc.SkipVerify})
 
 	t.Render()
 }
@@ -134,6 +139,7 @@ type Config struct {
 	RequestCount *uint             `json:"requests" yaml:"requests"`
 	Duration     *types.Duration   `json:"duration" yaml:"duration"`
 	Yes          *bool             `json:"yes" yaml:"yes"`
+	SkipVerify   *bool             `json:"skip_verify" yaml:"skip_verify"`
 	Params       types.Params      `json:"params" yaml:"params"`
 	Headers      types.Headers     `json:"headers" yaml:"headers"`
 	Cookies      types.Cookies     `json:"cookies" yaml:"cookies"`
@@ -220,6 +226,9 @@ func (config *Config) MergeConfig(newConfig *Config) {
 	if newConfig.Yes != nil {
 		config.Yes = newConfig.Yes
 	}
+	if newConfig.SkipVerify != nil {
+		config.SkipVerify = newConfig.SkipVerify
+	}
 	if len(newConfig.Params) != 0 {
 		config.Params = newConfig.Params
 	}
@@ -256,5 +265,8 @@ func (config *Config) SetDefaults() {
 	if config.Yes == nil {
 		config.Yes = utils.ToPtr(DefaultYes)
 	}
+	if config.SkipVerify == nil {
+		config.SkipVerify = utils.ToPtr(DefaultSkipVerify)
+	}
 	config.Headers.SetIfNotExists("User-Agent", DefaultUserAgent)
 }

go.mod

@@ -1,10 +1,10 @@
 module github.com/aykhans/dodo
 
-go 1.24.0
+go 1.24.2
 
 require (
 	github.com/jedib0t/go-pretty/v6 v6.6.7
-	github.com/valyala/fasthttp v1.60.0
+	github.com/valyala/fasthttp v1.62.0
 	gopkg.in/yaml.v3 v3.0.1
 )
 
@@ -14,8 +14,8 @@ require (
 	github.com/mattn/go-runewidth v0.0.16 // indirect
 	github.com/rivo/uniseg v0.4.7 // indirect
 	github.com/valyala/bytebufferpool v1.0.0 // indirect
-	golang.org/x/net v0.38.0 // indirect
-	golang.org/x/sys v0.31.0 // indirect
-	golang.org/x/term v0.30.0 // indirect
-	golang.org/x/text v0.23.0 // indirect
+	golang.org/x/net v0.40.0 // indirect
+	golang.org/x/sys v0.33.0 // indirect
+	golang.org/x/term v0.32.0 // indirect
+	golang.org/x/text v0.25.0 // indirect
 )

go.sum

@@ -17,18 +17,18 @@ github.com/stretchr/testify v1.10.0 h1:Xv5erBjTwe/5IxqUQTdXv5kgmIvbHo3QQyRwhJsOf
 github.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
 github.com/valyala/bytebufferpool v1.0.0 h1:GqA5TC/0021Y/b9FG4Oi9Mr3q7XYx6KllzawFIhcdPw=
 github.com/valyala/bytebufferpool v1.0.0/go.mod h1:6bBcMArwyJ5K/AmCkWv1jt77kVWyCJ6HpOuEn7z0Csc=
-github.com/valyala/fasthttp v1.60.0 h1:kBRYS0lOhVJ6V+bYN8PqAHELKHtXqwq9zNMLKx1MBsw=
-github.com/valyala/fasthttp v1.60.0/go.mod h1:iY4kDgV3Gc6EqhRZ8icqcmlG6bqhcDXfuHgTO4FXCvc=
+github.com/valyala/fasthttp v1.62.0 h1:8dKRBX/y2rCzyc6903Zu1+3qN0H/d2MsxPPmVNamiH0=
+github.com/valyala/fasthttp v1.62.0/go.mod h1:FCINgr4GKdKqV8Q0xv8b+UxPV+H/O5nNFo3D+r54Htg=
 github.com/xyproto/randomstring v1.0.5 h1:YtlWPoRdgMu3NZtP45drfy1GKoojuR7hmRcnhZqKjWU=
 github.com/xyproto/randomstring v1.0.5/go.mod h1:rgmS5DeNXLivK7YprL0pY+lTuhNQW3iGxZ18UQApw/E=
-golang.org/x/net v0.38.0 h1:vRMAPTMaeGqVhG5QyLJHqNDwecKTomGeqbnfZyKlBI8=
-golang.org/x/net v0.38.0/go.mod h1:ivrbrMbzFq5J41QOQh0siUuly180yBYtLp+CKbEaFx8=
-golang.org/x/sys v0.31.0 h1:ioabZlmFYtWhL+TRYpcnNlLwhyxaM9kWTDEmfnprqik=
-golang.org/x/sys v0.31.0/go.mod h1:BJP2sWEmIv4KK5OTEluFJCKSidICx8ciO85XgH3Ak8k=
-golang.org/x/term v0.30.0 h1:PQ39fJZ+mfadBm0y5WlL4vlM7Sx1Hgf13sMIY2+QS9Y=
-golang.org/x/term v0.30.0/go.mod h1:NYYFdzHoI5wRh/h5tDMdMqCqPJZEuNqVR5xJLd/n67g=
-golang.org/x/text v0.23.0 h1:D71I7dUrlY+VX0gQShAThNGHFxZ13dGLBHQLVl1mJlY=
-golang.org/x/text v0.23.0/go.mod h1:/BLNzu4aZCJ1+kcD0DNRotWKage4q2rGVAg4o22unh4=
+golang.org/x/net v0.40.0 h1:79Xs7wF06Gbdcg4kdCCIQArK11Z1hr5POQ6+fIYHNuY=
+golang.org/x/net v0.40.0/go.mod h1:y0hY0exeL2Pku80/zKK7tpntoX23cqL3Oa6njdgRtds=
+golang.org/x/sys v0.33.0 h1:q3i8TbbEz+JRD9ywIRlyRAQbM0qF7hu24q3teo2hbuw=
+golang.org/x/sys v0.33.0/go.mod h1:BJP2sWEmIv4KK5OTEluFJCKSidICx8ciO85XgH3Ak8k=
+golang.org/x/term v0.32.0 h1:DR4lr0TjUs3epypdhTOkMmuF5CDFJ/8pOnbzMZPQ7bg=
+golang.org/x/term v0.32.0/go.mod h1:uZG1FhGx848Sqfsq4/DlJr3xGGsYMu/L5GW4abiaEPQ=
+golang.org/x/text v0.25.0 h1:qVyWApTSYLk/drJRO5mDlNYskwQznZmkpV2c8q9zls4=
+golang.org/x/text v0.25.0/go.mod h1:WEdwpYrmk1qmdHvhkSTNPm3app7v4rsT8F2UD6+VHIA=
 gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
 gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
 gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=

requests/client.go

@@ -2,6 +2,7 @@ package requests
 
 import (
 	"context"
+	"crypto/tls"
 	"errors"
 	"math/rand"
 	"net/url"
@@ -17,11 +18,12 @@ type ClientGeneratorFunc func() *fasthttp.HostClient
 // getClients initializes and returns a slice of fasthttp.HostClient based on the provided parameters.
 // It can either return clients with proxies or a single client without proxies.
 func getClients(
-	ctx context.Context,
+	_ context.Context,
 	timeout time.Duration,
 	proxies []url.URL,
 	maxConns uint,
 	URL url.URL,
+	skipVerify bool,
 ) []*fasthttp.HostClient {
 	isTLS := URL.Scheme == "https"
 
@@ -41,6 +43,9 @@ func getClients(
 			clients = append(clients, &fasthttp.HostClient{
 				MaxConns: int(maxConns),
 				IsTLS:    isTLS,
+				TLSConfig: &tls.Config{
+					InsecureSkipVerify: skipVerify,
+				},
 				Addr:                addr,
 				Dial:                dialFunc,
 				MaxIdleConnDuration: timeout,
@@ -56,6 +61,9 @@ func getClients(
 	client := &fasthttp.HostClient{
 		MaxConns: int(maxConns),
 		IsTLS:    isTLS,
+		TLSConfig: &tls.Config{
+			InsecureSkipVerify: skipVerify,
+		},
 		Addr:                URL.Host,
 		MaxIdleConnDuration: timeout,
 		MaxConnDuration:     timeout,
@@ -80,6 +88,11 @@ func getDialFunc(proxy *url.URL, timeout time.Duration) (fasthttp.DialFunc, erro
 	default:
 		return nil, errors.New("unsupported proxy scheme")
 	}
+
+	if dialer == nil {
+		return nil, errors.New("internal error: proxy dialer is nil")
+	}
+
 	return dialer, nil
 }
 

requests/run.go

@@ -20,8 +20,11 @@ import (
 //   - ctx: The context for managing request lifecycle and cancellation.
 //   - requestConfig: The configuration for the request, including timeout, proxies, and other settings.
 func Run(ctx context.Context, requestConfig *config.RequestConfig) (Responses, error) {
-	ctx, cancel := context.WithCancel(ctx)
+	if requestConfig.Duration > 0 {
+		var cancel context.CancelFunc
+		ctx, cancel = context.WithTimeout(ctx, requestConfig.Duration)
 		defer cancel()
+	}
 
 	clients := getClients(
 		ctx,
@@ -29,15 +32,12 @@ func Run(ctx context.Context, requestConfig *config.RequestConfig) (Responses, e
 		requestConfig.Proxies,
 		requestConfig.GetMaxConns(fasthttp.DefaultMaxConnsPerHost),
 		requestConfig.URL,
+		requestConfig.SkipVerify,
 	)
 	if clients == nil {
 		return nil, types.ErrInterrupt
 	}
 
-	if requestConfig.Duration > 0 {
-		time.AfterFunc(requestConfig.Duration, func() { cancel() })
-	}
-
 	responses := releaseDodos(ctx, requestConfig, clients)
 	if ctx.Err() != nil && len(responses) == 0 {
 		return nil, types.ErrInterrupt

types/body.go

@@ -66,7 +66,7 @@ func (body *Body) UnmarshalJSON(b []byte) error {
 	return nil
 }
 
-func (body *Body) UnmarshalYAML(unmarshal func(interface{}) error) error {
+func (body *Body) UnmarshalYAML(unmarshal func(any) error) error {
 	var data any
 	if err := unmarshal(&data); err != nil {
 		return err

types/cookies.go

@@ -99,7 +99,7 @@ func (cookies *Cookies) UnmarshalJSON(b []byte) error {
 	return nil
 }
 
-func (cookies *Cookies) UnmarshalYAML(unmarshal func(interface{}) error) error {
+func (cookies *Cookies) UnmarshalYAML(unmarshal func(any) error) error {
 	var raw []map[string]any
 	if err := unmarshal(&raw); err != nil {
 		return err

types/duration.go

@@ -35,7 +35,7 @@ func (duration Duration) MarshalJSON() ([]byte, error) {
 	return json.Marshal(duration.String())
 }
 
-func (duration *Duration) UnmarshalYAML(unmarshal func(interface{}) error) error {
+func (duration *Duration) UnmarshalYAML(unmarshal func(any) error) error {
 	var v any
 	if err := unmarshal(&v); err != nil {
 		return err

types/headers.go

@@ -108,7 +108,7 @@ func (headers *Headers) UnmarshalJSON(b []byte) error {
 	return nil
 }
 
-func (headers *Headers) UnmarshalYAML(unmarshal func(interface{}) error) error {
+func (headers *Headers) UnmarshalYAML(unmarshal func(any) error) error {
 	var raw []map[string]any
 	if err := unmarshal(&raw); err != nil {
 		return err

types/params.go

@@ -99,7 +99,7 @@ func (params *Params) UnmarshalJSON(b []byte) error {
 	return nil
 }
 
-func (params *Params) UnmarshalYAML(unmarshal func(interface{}) error) error {
+func (params *Params) UnmarshalYAML(unmarshal func(any) error) error {
 	var raw []map[string]any
 	if err := unmarshal(&raw); err != nil {
 		return err

types/proxies.go

@@ -75,7 +75,7 @@ func (proxies *Proxies) UnmarshalJSON(b []byte) error {
 	return nil
 }
 
-func (proxies *Proxies) UnmarshalYAML(unmarshal func(interface{}) error) error {
+func (proxies *Proxies) UnmarshalYAML(unmarshal func(any) error) error {
 	var data any
 	if err := unmarshal(&data); err != nil {
 		return err

types/request_url.go

@@ -25,7 +25,7 @@ func (requestURL *RequestURL) UnmarshalJSON(data []byte) error {
 	return nil
 }
 
-func (requestURL *RequestURL) UnmarshalYAML(unmarshal func(interface{}) error) error {
+func (requestURL *RequestURL) UnmarshalYAML(unmarshal func(any) error) error {
 	var urlStr string
 	if err := unmarshal(&urlStr); err != nil {
 		return err

types/timeout.go

@@ -35,7 +35,7 @@ func (timeout Timeout) MarshalJSON() ([]byte, error) {
 	return json.Marshal(timeout.String())
 }
 
-func (timeout *Timeout) UnmarshalYAML(unmarshal func(interface{}) error) error {
+func (timeout *Timeout) UnmarshalYAML(unmarshal func(any) error) error {
 	var v any
 	if err := unmarshal(&v); err != nil {
 		return err