forked from kelseyhightower/confd
-
Notifications
You must be signed in to change notification settings - Fork 0
/
file.go
85 lines (75 loc) · 1.9 KB
/
file.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
package util
import (
"crypto/md5"
"fmt"
"io"
"os"
"syscall"
"github.com/golang/glog"
)
// FileInfo describes a configuration file and is returned by fileStat.
type fileInfo struct {
Uid uint32
Gid uint32
Mode os.FileMode
Md5 string
}
// IsFileExist reports whether path exits.
func IsFileExist(fpath string) bool {
if _, err := os.Stat(fpath); os.IsNotExist(err) {
return false
}
return true
}
// IsSameConfig reports whether src and dest config files are equal.
// Two config files are equal when they have the same file contents and
// Unix permissions. The owner, group, and mode must match.
// It return false in other cases.
func IsSameConfig(src, dest string) (bool, error) {
if !IsFileExist(dest) {
return false, nil
}
dfi, err := getFileInfo(dest)
if err != nil {
return false, err
}
sfi, err := getFileInfo(src)
if err != nil {
return false, err
}
if dfi.Uid != sfi.Uid {
glog.Infof("%s has UID %d should be %d", dest, dfi.Uid, sfi.Uid)
}
if dfi.Gid != sfi.Gid {
glog.Infof("%s has GID %d should be %d", dest, dfi.Gid, sfi.Gid)
}
if dfi.Mode != sfi.Mode {
glog.Infof("%s has mode %s should be %s", dest, os.FileMode(dfi.Mode), os.FileMode(sfi.Mode))
}
if dfi.Md5 != sfi.Md5 {
glog.Infof("%s has md5sum %s should be %s", dest, dfi.Md5, sfi.Md5)
}
if dfi.Uid != sfi.Uid || dfi.Gid != sfi.Gid || dfi.Mode != sfi.Mode || dfi.Md5 != sfi.Md5 {
return false, nil
}
return true, nil
}
// getFileInfo returns a FileInfo describing the named file.
func getFileInfo(name string) (fi fileInfo, err error) {
if !IsFileExist(name) {
return fi, fmt.Errorf("%s file not found", name)
}
f, err := os.Open(name)
if err != nil {
return fi, err
}
defer f.Close()
stats, _ := f.Stat()
fi.Uid = stats.Sys().(*syscall.Stat_t).Uid
fi.Gid = stats.Sys().(*syscall.Stat_t).Gid
fi.Mode = stats.Mode()
h := md5.New()
io.Copy(h, f)
fi.Md5 = fmt.Sprintf("%x", h.Sum(nil))
return fi, nil
}