From abcde53d842735462a55d6aa4ae49c02efd4adf6 Mon Sep 17 00:00:00 2001
From: Matias Doyle <doyle@noisolation.com>
Date: Mon, 4 Dec 2017 17:37:34 +0100
Subject: [PATCH] Add support for setting key_mgmt value

---
 unixgram.go | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/unixgram.go b/unixgram.go
index 8a5d21c..8bbc117 100644
--- a/unixgram.go
+++ b/unixgram.go
@@ -303,7 +303,16 @@ func (uc *unixgramConn) RemoveAllNetworks() error {
 }
 
 func (uc *unixgramConn) SetNetwork(networkID int, variable string, value string) error {
-	return uc.runCommand(fmt.Sprintf("SET_NETWORK %d %s \"%s\"", networkID, variable, value))
+	var cmd string
+
+	// Since key_mgmt expects the value to not be wrapped in "" we do a little check here.
+	if variable == "key_mgmt" {
+		cmd = fmt.Sprintf("SET_NETWORK %d %s %s", networkID, variable, value)
+	} else {
+		cmd = fmt.Sprintf("SET_NETWORK %d %s \"%s\"", networkID, variable, value)
+	}
+
+	return uc.runCommand(cmd)
 }
 
 func (uc *unixgramConn) SaveConfig() error {