From c9f58ce4a73e57f4d23db9e99315b5bfd982fb16 Mon Sep 17 00:00:00 2001
From: Simon Zolin <s.zolin@adguard.com>
Date: Tue, 18 Aug 2020 17:34:02 +0300
Subject: [PATCH] * upgrade yaml schema 6->7: DHCP settings

---
 home/upgrade.go | 92 ++++++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 91 insertions(+), 1 deletion(-)

diff --git a/home/upgrade.go b/home/upgrade.go
index b2936a3e..4021c206 100644
--- a/home/upgrade.go
+++ b/home/upgrade.go
@@ -13,7 +13,7 @@ import (
 	yaml "gopkg.in/yaml.v2"
 )
 
-const currentSchemaVersion = 6 // used for upgrading from old configs to new config
+const currentSchemaVersion = 7 // used for upgrading from old configs to new config
 
 // Performs necessary upgrade operations if needed
 func upgradeConfig() error {
@@ -90,6 +90,12 @@ func upgradeConfigSchema(oldVersion int, diskConfig *map[string]interface{}) err
 		if err != nil {
 			return err
 		}
+		fallthrough
+	case 6:
+		err := upgradeSchema6to7(diskConfig)
+		if err != nil {
+			return err
+		}
 	default:
 		err := fmt.Errorf("configuration file contains unknown schema_version, abort")
 		log.Println(err)
@@ -345,3 +351,87 @@ func upgradeSchema5to6(diskConfig *map[string]interface{}) error {
 
 	return nil
 }
+
+// dhcp:
+//   enabled: false
+//   interface_name: vboxnet0
+//   gateway_ip: 192.168.56.1
+//   ...
+//
+// ->
+//
+// dhcp:
+//   enabled: false
+//   interface_name: vboxnet0
+//   dhcpv4:
+//     gateway_ip: 192.168.56.1
+//     ...
+func upgradeSchema6to7(diskConfig *map[string]interface{}) error {
+	log.Printf("Upgrade yaml: 6 to 7")
+
+	(*diskConfig)["schema_version"] = 7
+
+	_dhcp, ok := (*diskConfig)["dhcp"]
+	if !ok {
+		return nil
+	}
+
+	switch dhcp := _dhcp.(type) {
+	case map[interface{}]interface{}:
+		dhcpv4 := map[string]interface{}{}
+		val, ok := dhcp["gateway_ip"].(string)
+		if !ok {
+			log.Fatalf("expecting dhcp.%s to be a string", "gateway_ip")
+			return nil
+		}
+		dhcpv4["gateway_ip"] = val
+		delete(dhcp, "gateway_ip")
+
+		val, ok = dhcp["subnet_mask"].(string)
+		if !ok {
+			log.Fatalf("expecting dhcp.%s to be a string", "subnet_mask")
+			return nil
+		}
+		dhcpv4["subnet_mask"] = val
+		delete(dhcp, "subnet_mask")
+
+		val, ok = dhcp["range_start"].(string)
+		if !ok {
+			log.Fatalf("expecting dhcp.%s to be a string", "range_start")
+			return nil
+		}
+		dhcpv4["range_start"] = val
+		delete(dhcp, "range_start")
+
+		val, ok = dhcp["range_end"].(string)
+		if !ok {
+			log.Fatalf("expecting dhcp.%s to be a string", "range_end")
+			return nil
+		}
+		dhcpv4["range_end"] = val
+		delete(dhcp, "range_end")
+
+		intVal, ok := dhcp["lease_duration"].(int)
+		if !ok {
+			log.Fatalf("expecting dhcp.%s to be an integer", "lease_duration")
+			return nil
+		}
+		dhcpv4["lease_duration"] = intVal
+		delete(dhcp, "lease_duration")
+
+		intVal, ok = dhcp["icmp_timeout_msec"].(int)
+		if !ok {
+			log.Fatalf("expecting dhcp.%s to be an integer", "icmp_timeout_msec")
+			return nil
+		}
+		dhcpv4["icmp_timeout_msec"] = intVal
+		delete(dhcp, "icmp_timeout_msec")
+
+		dhcp["dhcpv4"] = dhcpv4
+
+	default:
+		return nil
+	}
+
+	return nil
+}