package home import ( "fmt" "os" "path/filepath" "github.com/AdguardTeam/golibs/file" "github.com/AdguardTeam/golibs/log" "golang.org/x/crypto/bcrypt" yaml "gopkg.in/yaml.v2" ) const currentSchemaVersion = 6 // used for upgrading from old configs to new config // Performs necessary upgrade operations if needed func upgradeConfig() error { // read a config file into an interface map, so we can manipulate values without losing any diskConfig := map[string]interface{}{} body, err := readConfigFile() if err != nil { return err } err = yaml.Unmarshal(body, &diskConfig) if err != nil { log.Printf("Couldn't parse config file: %s", err) return err } schemaVersionInterface, ok := diskConfig["schema_version"] log.Tracef("got schema version %v", schemaVersionInterface) if !ok { // no schema version, set it to 0 schemaVersionInterface = 0 } schemaVersion, ok := schemaVersionInterface.(int) if !ok { err = fmt.Errorf("configuration file contains non-integer schema_version, abort") log.Println(err) return err } if schemaVersion == currentSchemaVersion { // do nothing return nil } return upgradeConfigSchema(schemaVersion, &diskConfig) } // Upgrade from oldVersion to newVersion func upgradeConfigSchema(oldVersion int, diskConfig *map[string]interface{}) error { switch oldVersion { case 0: err := upgradeSchema0to1(diskConfig) if err != nil { return err } fallthrough case 1: err := upgradeSchema1to2(diskConfig) if err != nil { return err } fallthrough case 2: err := upgradeSchema2to3(diskConfig) if err != nil { return err } fallthrough case 3: err := upgradeSchema3to4(diskConfig) if err != nil { return err } fallthrough case 4: err := upgradeSchema4to5(diskConfig) if err != nil { return err } fallthrough case 5: err := upgradeSchema5to6(diskConfig) if err != nil { return err } default: err := fmt.Errorf("configuration file contains unknown schema_version, abort") log.Println(err) return err } configFile := config.getConfigFilename() body, err := yaml.Marshal(diskConfig) if err != nil { log.Printf("Couldn't generate YAML file: %s", err) return err } config.fileData = body err = file.SafeWrite(configFile, body) if err != nil { log.Printf("Couldn't save YAML config: %s", err) return err } return nil } // The first schema upgrade: // No more "dnsfilter.txt", filters are now kept in data/filters/ func upgradeSchema0to1(diskConfig *map[string]interface{}) error { log.Printf("%s(): called", _Func()) dnsFilterPath := filepath.Join(Context.workDir, "dnsfilter.txt") if _, err := os.Stat(dnsFilterPath); !os.IsNotExist(err) { log.Printf("Deleting %s as we don't need it anymore", dnsFilterPath) err = os.Remove(dnsFilterPath) if err != nil { log.Printf("Cannot remove %s due to %s", dnsFilterPath, err) // not fatal, move on } } (*diskConfig)["schema_version"] = 1 return nil } // Second schema upgrade: // coredns is now dns in config // delete 'Corefile', since we don't use that anymore func upgradeSchema1to2(diskConfig *map[string]interface{}) error { log.Printf("%s(): called", _Func()) coreFilePath := filepath.Join(Context.workDir, "Corefile") if _, err := os.Stat(coreFilePath); !os.IsNotExist(err) { log.Printf("Deleting %s as we don't need it anymore", coreFilePath) err = os.Remove(coreFilePath) if err != nil { log.Printf("Cannot remove %s due to %s", coreFilePath, err) // not fatal, move on } } if _, ok := (*diskConfig)["dns"]; !ok { (*diskConfig)["dns"] = (*diskConfig)["coredns"] delete((*diskConfig), "coredns") } (*diskConfig)["schema_version"] = 2 return nil } // Third schema upgrade: // Bootstrap DNS becomes an array func upgradeSchema2to3(diskConfig *map[string]interface{}) error { log.Printf("%s(): called", _Func()) // Let's read dns configuration from diskConfig dnsConfig, ok := (*diskConfig)["dns"] if !ok { return fmt.Errorf("no DNS configuration in config file") } // Convert interface{} to map[string]interface{} newDNSConfig := make(map[string]interface{}) switch v := dnsConfig.(type) { case map[interface{}]interface{}: for k, v := range v { newDNSConfig[fmt.Sprint(k)] = v } default: return fmt.Errorf("DNS configuration is not a map") } // Replace bootstrap_dns value filed with new array contains old bootstrap_dns inside if bootstrapDNS, ok := (newDNSConfig)["bootstrap_dns"]; ok { newBootstrapConfig := []string{fmt.Sprint(bootstrapDNS)} (newDNSConfig)["bootstrap_dns"] = newBootstrapConfig (*diskConfig)["dns"] = newDNSConfig } else { return fmt.Errorf("no bootstrap DNS in DNS config") } // Bump schema version (*diskConfig)["schema_version"] = 3 return nil } // Add use_global_blocked_services=true setting for existing "clients" array func upgradeSchema3to4(diskConfig *map[string]interface{}) error { log.Printf("%s(): called", _Func()) (*diskConfig)["schema_version"] = 4 clients, ok := (*diskConfig)["clients"] if !ok { return nil } switch arr := clients.(type) { case []interface{}: for i := range arr { switch c := arr[i].(type) { case map[interface{}]interface{}: c["use_global_blocked_services"] = true default: continue } } default: return nil } return nil } // Replace "auth_name", "auth_pass" string settings with an array: // users: // - name: "..." // password: "..." // ... func upgradeSchema4to5(diskConfig *map[string]interface{}) error { log.Printf("%s(): called", _Func()) (*diskConfig)["schema_version"] = 5 name, ok := (*diskConfig)["auth_name"] if !ok { return nil } nameStr, ok := name.(string) if !ok { log.Fatal("Please use double quotes in your user name in \"auth_name\" and restart AdGuardHome") return nil } pass, ok := (*diskConfig)["auth_pass"] if !ok { return nil } passStr, ok := pass.(string) if !ok { log.Fatal("Please use double quotes in your password in \"auth_pass\" and restart AdGuardHome") return nil } if len(nameStr) == 0 { return nil } hash, err := bcrypt.GenerateFromPassword([]byte(passStr), bcrypt.DefaultCost) if err != nil { log.Fatalf("Can't use password \"%s\": bcrypt.GenerateFromPassword: %s", passStr, err) return nil } u := User{ Name: nameStr, PasswordHash: string(hash), } users := []User{u} (*diskConfig)["users"] = users return nil } // clients: // ... // ip: 127.0.0.1 // mac: ... // // -> // // clients: // ... // ids: // - 127.0.0.1 // - ... func upgradeSchema5to6(diskConfig *map[string]interface{}) error { log.Printf("%s(): called", _Func()) (*diskConfig)["schema_version"] = 6 clients, ok := (*diskConfig)["clients"] if !ok { return nil } switch arr := clients.(type) { case []interface{}: for i := range arr { switch c := arr[i].(type) { case map[interface{}]interface{}: _ip, ok := c["ip"] ids := []string{} if ok { ip, ok := _ip.(string) if !ok { log.Fatalf("client.ip is not a string: %v", _ip) return nil } if len(ip) != 0 { ids = append(ids, ip) } } _mac, ok := c["mac"] if ok { mac, ok := _mac.(string) if !ok { log.Fatalf("client.mac is not a string: %v", _mac) return nil } if len(mac) != 0 { ids = append(ids, mac) } } c["ids"] = ids default: continue } } default: return nil } return nil }