From 0c3a3e69d96bf7536f816b32d954b48ec217ef63 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Mitja=20=C5=BDivkovi=C4=87?= <mitja.zivkovic@3fs.si>
Date: Mon, 16 Nov 2020 09:19:29 +0100
Subject: [PATCH] Adds tls command line parameters validation

---
 main.go | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/main.go b/main.go
index 730f40e..e7d75a4 100644
--- a/main.go
+++ b/main.go
@@ -83,6 +83,14 @@ func main() {
 		log.Fatalf("Invalid arguments: messages count should be > 1, given: %v", *count)
 	}
 
+	if *cert != "" && *key == "" {
+		log.Fatal("Invalid arguments: private key path missing")
+	}
+
+	if *cert == "" && *key != "" {
+		log.Fatalf("Invalid arguments: certificate path missing")
+	}
+
 	var tlsConfig *tls.Config
 	if *cert != "" && *key != "" {
 		tlsConfig = generateTlsConfig(*cert, *key)