fix -init without cn

nothinux · Mar 5, 2022 · 3bba86f · 3bba86f
1 parent e4a3a27
commit 3bba86f
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/cmd/certify/main.go b/cmd/certify/main.go
@@ -5,6 +5,7 @@ import (
 	"fmt"
 	"log"
 	"os"
+	"strings"
 )
 
 const usage = `Usage of certify:
@@ -43,7 +44,15 @@ func main() {
 		}
 		fmt.Println("CA private key file generated", caKeyPath)
 
-		if err := generateCA(pkey.PrivateKey, os.Args[2], caPath); err != nil {
+		var cn string
+
+		if len(os.Args) > 2 {
+			if strings.Contains(os.Args[2], "cn:") {
+				cn = os.Args[2]
+			}
+		}
+
+		if err := generateCA(pkey.PrivateKey, cn, caPath); err != nil {
 			log.Fatal(err)
 		}
 		fmt.Println("CA certificate file generated", caPath)