[+] Listen And Serve TLS [+]

krishpranav · krishpranav · commit aec2de9754ec · 2021-06-26T16:17:32.000+05:30
diff --git a/pkg/httpserver/httpserver.go b/pkg/httpserver/httpserver.go
@@ -2,6 +2,8 @@ package httpserver
 
 import (
 	"net/http"
+
+	"github.com/projectdiscovery/sslcert"
 )
 
 type Options struct {
@@ -39,3 +41,21 @@ func New(options *Options) (*HTTPServer, error) {
 func (t *HTTPServer) ListenAndServe() error {
 	return http.ListenAndServe(t.options.ListenAddress, t.layers)
 }
+
+func (t *HTTPServer) ListenAndServeTLS() error {
+	if t.options.Certificate == "" || t.options.CertificateKey == "" {
+		tlsOptions := sslcert.DefaultOptions
+		tlsOptions.Host = t.options.CertificateDomain
+		tlsConfig, err := sslcert.NewTLSConfig(tlsOptions)
+		if err != nil {
+			return err
+		}
+		httpServer := &http.Server{
+			Addr:      t.options.ListenAddress,
+			TLSConfig: tlsConfig,
+		}
+		httpServer.Handler = t.layers
+		return httpServer.ListenAndServeTLS("", "")
+	}
+	return http.ListenAndServeTLS(t.options.ListenAddress, t.options.Certificate, t.options.CertificateKey, t.layers)
+}
