From 0eb4afb0465a0f2dfdf2a9d356488ab09fe66fa0 Mon Sep 17 00:00:00 2001
From: John Moscato <jmoscato@gmail.com>
Date: Fri, 14 Apr 2017 01:40:59 -0400
Subject: [PATCH] Add option to skip server certificate validation

This option will allow gobench to make successful HTTPS requests to
servers that have a self-signed certificate, without the need to
manually import it as a trusted certificate. The flag to control the
new option is named "nc" and it can be set to true to disable the
validation check of the server's certificate. The "nc" flag defaults
to false so server certificate validation is performed by default.
---
 gobench.go | 32 +++++++++++++++++++-------------
 1 file changed, 19 insertions(+), 13 deletions(-)

diff --git a/gobench.go b/gobench.go
index d186b67..2c4f723 100644
--- a/gobench.go
+++ b/gobench.go
@@ -3,6 +3,7 @@ package main
 import (
 	"bufio"
 	"bytes"
+	"crypto/tls"
 	"flag"
 	"fmt"
 	"io"
@@ -30,16 +31,18 @@ var (
 	writeTimeout     int
 	readTimeout      int
 	authHeader       string
+	noCheckCert      bool
 )
 
 type Configuration struct {
-	urls       []string
-	method     string
-	postData   []byte
-	requests   int64
-	period     int64
-	keepAlive  bool
-	authHeader string
+	urls        []string
+	method      string
+	postData    []byte
+	requests    int64
+	period      int64
+	keepAlive   bool
+	authHeader  string
+	noCheckCert bool
 
 	myClient fasthttp.Client
 }
@@ -89,6 +92,7 @@ func init() {
 	flag.IntVar(&writeTimeout, "tw", 5000, "Write timeout (in milliseconds)")
 	flag.IntVar(&readTimeout, "tr", 5000, "Read timeout (in milliseconds)")
 	flag.StringVar(&authHeader, "auth", "", "Authorization header")
+	flag.BoolVar(&noCheckCert, "nc", false, "Do not validate server's certificate")
 }
 
 func printResults(results map[int]*Result, startTime time.Time) {
@@ -170,12 +174,13 @@ func NewConfiguration() *Configuration {
 	}
 
 	configuration := &Configuration{
-		urls:       make([]string, 0),
-		method:     "GET",
-		postData:   nil,
-		keepAlive:  keepAlive,
-		requests:   int64((1 << 63) - 1),
-		authHeader: authHeader}
+		urls:        make([]string, 0),
+		method:      "GET",
+		postData:    nil,
+		keepAlive:   keepAlive,
+		requests:    int64((1 << 63) - 1),
+		authHeader:  authHeader,
+		noCheckCert: noCheckCert}
 
 	if period != -1 {
 		configuration.period = period
@@ -231,6 +236,7 @@ func NewConfiguration() *Configuration {
 	configuration.myClient.ReadTimeout = time.Duration(readTimeout) * time.Millisecond
 	configuration.myClient.WriteTimeout = time.Duration(writeTimeout) * time.Millisecond
 	configuration.myClient.MaxConnsPerHost = clients
+	configuration.myClient.TLSConfig = &tls.Config{InsecureSkipVerify: noCheckCert}
 
 	configuration.myClient.Dial = MyDialer()