|
6 | 6 | package conf |
7 | 7 |
|
8 | 8 | import ( |
| 9 | + "io/ioutil" |
9 | 10 | "log" |
| 11 | + "os" |
10 | 12 | "path/filepath" |
11 | | - "strings" |
12 | 13 |
|
13 | 14 | "golang.org/x/sys/windows" |
14 | 15 | ) |
15 | 16 |
|
16 | | -func maybeMigrate(c string) { |
| 17 | +func maybeMigrateConfiguration(c string) { |
17 | 18 | if disableAutoMigration { |
18 | 19 | return |
19 | 20 | } |
20 | | - |
21 | | - vol := filepath.VolumeName(c) |
22 | | - withoutVol := strings.TrimPrefix(c, vol) |
23 | | - oldRoot := filepath.Join(vol, "\\windows.old") |
24 | | - oldC := filepath.Join(oldRoot, withoutVol) |
25 | | - |
26 | | - sd, err := windows.GetNamedSecurityInfo(oldRoot, windows.SE_FILE_OBJECT, windows.OWNER_SECURITY_INFORMATION) |
27 | | - if err == windows.ERROR_PATH_NOT_FOUND || err == windows.ERROR_FILE_NOT_FOUND { |
28 | | - return |
29 | | - } |
| 21 | + oldRoot, err := windows.KnownFolderPath(windows.FOLDERID_LocalAppData, windows.KF_FLAG_DEFAULT) |
30 | 22 | if err != nil { |
31 | | - log.Printf("Not migrating configuration from ‘%s’ due to GetNamedSecurityInfo error: %v", oldRoot, err) |
32 | 23 | return |
33 | 24 | } |
34 | | - owner, defaulted, err := sd.Owner() |
| 25 | + oldC := filepath.Join(oldRoot, "WireGuard", "Configurations") |
| 26 | + files, err := ioutil.ReadDir(oldC) |
35 | 27 | if err != nil { |
36 | | - log.Printf("Not migrating configuration from ‘%s’ due to GetSecurityDescriptorOwner error: %v", oldRoot, err) |
37 | | - return |
38 | | - } |
39 | | - if defaulted || (!owner.IsWellKnown(windows.WinLocalSystemSid) && !owner.IsWellKnown(windows.WinBuiltinAdministratorsSid)) { |
40 | | - log.Printf("Not migrating configuration from ‘%s’, as it is not explicitly owned by SYSTEM or Built-in Administrators, but rather ‘%v’", oldRoot, owner) |
41 | 28 | return |
42 | 29 | } |
43 | | - err = windows.MoveFileEx(windows.StringToUTF16Ptr(oldC), windows.StringToUTF16Ptr(c), windows.MOVEFILE_COPY_ALLOWED) |
44 | | - if err != nil { |
45 | | - if err != windows.ERROR_FILE_NOT_FOUND && err != windows.ERROR_ALREADY_EXISTS { |
46 | | - log.Printf("Not migrating configuration from ‘%s’ due to error when moving files: %v", oldRoot, err) |
| 30 | + for i := range files { |
| 31 | + if files[i].IsDir() { |
| 32 | + continue |
47 | 33 | } |
48 | | - return |
| 34 | + newPath := filepath.Join(c, files[i].Name()) |
| 35 | + newFile, err := os.OpenFile(newPath, os.O_EXCL | os.O_CREATE | os.O_WRONLY, 0600) |
| 36 | + if err != nil { |
| 37 | + continue |
| 38 | + } |
| 39 | + oldPath := filepath.Join(oldC, files[i].Name()) |
| 40 | + oldConfig, err := ioutil.ReadFile(oldPath) |
| 41 | + if err != nil { |
| 42 | + newFile.Close() |
| 43 | + os.Remove(newPath) |
| 44 | + continue |
| 45 | + } |
| 46 | + _, err = newFile.Write(oldConfig) |
| 47 | + if err != nil { |
| 48 | + newFile.Close() |
| 49 | + os.Remove(newPath) |
| 50 | + continue |
| 51 | + } |
| 52 | + newFile.Close() |
| 53 | + os.Remove(oldPath) |
| 54 | + log.Printf("Migrated configuration from ‘%s’ to ‘%s’", oldPath, newPath) |
49 | 55 | } |
50 | | - log.Printf("Migrated configuration from ‘%s’", oldRoot) |
| 56 | + os.Remove(oldC) |
51 | 57 | } |
0 commit comments