1 // Copyright (C) 2021 Simon Ruderich
3 // This program is free software: you can redistribute it and/or modify
4 // it under the terms of the GNU General Public License as published by
5 // the Free Software Foundation, either version 3 of the License, or
6 // (at your option) any later version.
8 // This program is distributed in the hope that it will be useful,
9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 // GNU General Public License for more details.
13 // You should have received a copy of the GNU General Public License
14 // along with this program. If not, see <http://www.gnu.org/licenses/>.
26 "github.com/google/go-cmp/cmp"
28 "ruderich.org/simon/safcm"
31 func chmod(name string, perm int) {
32 err := os.Chmod(name, FullPermToFileMode(perm))
38 func TestLoadFiles(t *testing.T) {
39 cwd, err := os.Getwd()
45 err = os.Chdir("../testdata/project")
50 chmod("files-invalid-perm-dir/files", 0500)
51 defer chmod("files-invalid-perm-dir/files", 0700)
52 chmod("files-invalid-perm-dir/files/etc/", 0755)
53 chmod("files-invalid-perm-dir/files/etc/resolv.conf", 0644)
54 chmod("files-invalid-perm-dir-setgid/files", 0755)
55 chmod("files-invalid-perm-dir-setgid/files/etc/", 02755)
56 chmod("files-invalid-perm-dir-setgid/files/etc/resolv.conf", 0644)
57 chmod("files-invalid-perm-file/files", 0755)
58 chmod("files-invalid-perm-file/files/etc/", 0755)
59 chmod("files-invalid-perm-file/files/etc/resolv.conf", 0600)
60 chmod("files-invalid-perm-file-executable/files", 0755)
61 chmod("files-invalid-perm-file-executable/files/etc", 0755)
62 chmod("files-invalid-perm-file-executable/files/etc/rc.local", 0750)
63 chmod("files-invalid-perm-file-sticky/files", 0755)
64 chmod("files-invalid-perm-file-sticky/files/etc", 0755)
65 chmod("files-invalid-perm-file-sticky/files/etc/resolv.conf", 01644)
67 err = syscall.Mkfifo("files-invalid-type/files/invalid", 0644)
71 defer os.Remove("files-invalid-type/files/invalid")
74 The actual permissions and user/group of files and directories are not used
75 (except for +x on files). 0644/0755 and current remote user/group is used per
76 default. Apply different file permissions via permissions.yaml. To prevent
77 confusion files must be manually chmodded 0644/0755 and directories 0755 or
83 exp map[string]*safcm.File
95 map[string]*safcm.File{
98 Mode: fs.ModeDir | 0755,
104 Mode: fs.ModeDir | 0755,
109 Path: "/etc/.hidden",
120 Data: []byte(`Welcome to
121 {{- if .IsHost "host1.example.org"}} Host ONE
122 {{- else if "host2"}} Host TWO
125 {{if .InGroup "detected_linux"}}
126 This is GNU/Linux host
128 {{if .InGroup "detected_freebsd"}}
134 Path: "/etc/rc.local",
138 Data: []byte("#!/bin/sh\n"),
140 "/etc/resolv.conf": {
141 Path: "/etc/resolv.conf",
145 Data: []byte("nameserver ::1\n"),
149 Mode: fs.ModeSymlink | 0777,
152 Data: []byte("doesnt-exist"),
159 "files-invalid-type",
161 fmt.Errorf("files-invalid-type: \"files-invalid-type/files/invalid\": file type not supported"),
164 "files-invalid-perm-dir",
166 fmt.Errorf("files-invalid-perm-dir: \"files-invalid-perm-dir/files\": invalid permissions 0500" + errMsg),
169 "files-invalid-perm-dir-setgid",
171 fmt.Errorf("files-invalid-perm-dir-setgid: \"files-invalid-perm-dir-setgid/files/etc\": invalid permissions 02755" + errMsg),
174 "files-invalid-perm-file",
176 fmt.Errorf("files-invalid-perm-file: \"files-invalid-perm-file/files/etc/resolv.conf\": invalid permissions 0600" + errMsg),
179 "files-invalid-perm-file-executable",
181 fmt.Errorf("files-invalid-perm-file-executable: \"files-invalid-perm-file-executable/files/etc/rc.local\": invalid permissions 0750" + errMsg),
184 "files-invalid-perm-file-sticky",
186 fmt.Errorf("files-invalid-perm-file-sticky: \"files-invalid-perm-file-sticky/files/etc/resolv.conf\": invalid permissions 01644" + errMsg),
190 for _, tc := range tests {
191 t.Run(tc.group, func(t *testing.T) {
192 res, err := LoadFiles(tc.group)
194 if !reflect.DeepEqual(tc.exp, res) {
196 cmp.Diff(tc.exp, res))
198 // Ugly but the simplest way to compare errors (including nil)
199 if fmt.Sprintf("%s", err) != fmt.Sprintf("%s", tc.expErr) {
200 t.Errorf("err = %#v, want %#v",