cmd/safcm/config/files_test.go

   1 // Copyright (C) 2021  Simon Ruderich
   2 //
   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.
   7 //
   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.
  12 //
  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/>.
  15
  16 package config
  17
  18 import (
  19         "fmt"
  20         "io/fs"
  21         "os"
  22         "syscall"
  23         "testing"
  24
  25         "ruderich.org/simon/safcm"
  26         "ruderich.org/simon/safcm/testutil"
  27 )
  28
  29 func chmod(name string, perm int) {
  30         err := os.Chmod(name, FullPermToFileMode(perm))
  31         if err != nil {
  32                 panic(err)
  33         }
  34 }
  35
  36 func TestLoadFiles(t *testing.T) {
  37         cwd, err := os.Getwd()
  38         if err != nil {
  39                 t.Fatal(err)
  40         }
  41         defer os.Chdir(cwd)
  42
  43         err = os.Chdir("../testdata/project")
  44         if err != nil {
  45                 t.Fatal(err)
  46         }
  47
  48         chmod("files-invalid-perm-dir/files", 0500)
  49         defer chmod("files-invalid-perm-dir/files", 0700)
  50         chmod("files-invalid-perm-dir/files/etc/", 0755)
  51         chmod("files-invalid-perm-dir/files/etc/resolv.conf", 0644)
  52         chmod("files-invalid-perm-dir-setgid/files", 0755)
  53         chmod("files-invalid-perm-dir-setgid/files/etc/", 02755)
  54         chmod("files-invalid-perm-dir-setgid/files/etc/resolv.conf", 0644)
  55         chmod("files-invalid-perm-file/files", 0755)
  56         chmod("files-invalid-perm-file/files/etc/", 0755)
  57         chmod("files-invalid-perm-file/files/etc/resolv.conf", 0600)
  58         chmod("files-invalid-perm-file-executable/files", 0755)
  59         chmod("files-invalid-perm-file-executable/files/etc", 0755)
  60         chmod("files-invalid-perm-file-executable/files/etc/rc.local", 0750)
  61         chmod("files-invalid-perm-file-sticky/files", 0755)
  62         chmod("files-invalid-perm-file-sticky/files/etc", 0755)
  63         chmod("files-invalid-perm-file-sticky/files/etc/resolv.conf", 01644)
  64
  65         err = syscall.Mkfifo("files-invalid-type/files/invalid", 0644)
  66         if err != nil {
  67                 t.Fatal(err)
  68         }
  69         defer os.Remove("files-invalid-type/files/invalid")
  70
  71         const errMsg = `
  72
  73 The actual permissions and user/group of files and directories are not used
  74 (except for +x on files). 0644/0755 and current remote user/group is used per
  75 default. Apply different file permissions via permissions.yaml. To prevent
  76 confusion files must be manually chmodded 0644/0755 and directories 0755 or
  77 via "safcm fixperms".
  78 `
  79
  80         tests := []struct {
  81                 group  string
  82                 exp    map[string]*safcm.File
  83                 expErr error
  84         }{
  85
  86                 {
  87                         "empty",
  88                         nil,
  89                         nil,
  90                 },
  91
  92                 {
  93                         "group",
  94                         map[string]*safcm.File{
  95                                 "/": {
  96                                         Path: "/",
  97                                         Mode: fs.ModeDir | 0755,
  98                                         Uid:  -1,
  99                                         Gid:  -1,
 100                                 },
 101                                 "/etc": {
 102                                         Path: "/etc",
 103                                         Mode: fs.ModeDir | 0755,
 104                                         Uid:  -1,
 105                                         Gid:  -1,
 106                                 },
 107                                 "/etc/.hidden": {
 108                                         Path: "/etc/.hidden",
 109                                         Mode: 0644,
 110                                         Uid:  -1,
 111                                         Gid:  -1,
 112                                         Data: []byte("..."),
 113                                 },
 114                                 "/etc/motd": {
 115                                         Path: "/etc/motd",
 116                                         Mode: 0644,
 117                                         Uid:  -1,
 118                                         Gid:  -1,
 119                                         Data: []byte(`Welcome to
 120 {{- if .IsHost "host1.example.org"}} Host ONE
 121 {{- else if "host2"}} Host TWO
 122 {{- end}}
 123
 124 {{if .InGroup "detected_linux"}}
 125 This is GNU/Linux host
 126 {{end}}
 127 {{if .InGroup "detected_freebsd"}}
 128 This is FreeBSD host
 129 {{end}}
 130
 131 {{if .InGroup "all"}}
 132 all
 133 {{end}}
 134 {{if .InGroup "host1.example.org"}}
 135 host1.example.org
 136 {{end}}
 137 {{if .InGroup "host2"}}
 138 host2
 139 {{end}}
 140 {{if .InGroup "host3.example.net"}}
 141 host3.example.net
 142 {{end}}
 143 `),
 144                                 },
 145                                 "/etc/rc.local": {
 146                                         Path: "/etc/rc.local",
 147                                         Mode: 0755,
 148                                         Uid:  -1,
 149                                         Gid:  -1,
 150                                         Data: []byte("#!/bin/sh\n"),
 151                                 },
 152                                 "/etc/resolv.conf": {
 153                                         Path: "/etc/resolv.conf",
 154                                         Mode: 0644,
 155                                         Uid:  -1,
 156                                         Gid:  -1,
 157                                         Data: []byte("nameserver ::1\n"),
 158                                 },
 159                                 "/etc/test": {
 160                                         Path: "/etc/test",
 161                                         Mode: fs.ModeSymlink | 0777,
 162                                         Uid:  -1,
 163                                         Gid:  -1,
 164                                         Data: []byte("doesnt-exist"),
 165                                 },
 166                         },
 167                         nil,
 168                 },
 169
 170                 {
 171                         "files-invalid-type",
 172                         nil,
 173                         fmt.Errorf("files-invalid-type: \"files-invalid-type/files/invalid\": file type not supported"),
 174                 },
 175                 {
 176                         "files-invalid-perm-dir",
 177                         nil,
 178                         fmt.Errorf("files-invalid-perm-dir: \"files-invalid-perm-dir/files\": invalid permissions 0500" + errMsg),
 179                 },
 180                 {
 181                         "files-invalid-perm-dir-setgid",
 182                         nil,
 183                         fmt.Errorf("files-invalid-perm-dir-setgid: \"files-invalid-perm-dir-setgid/files/etc\": invalid permissions 02755" + errMsg),
 184                 },
 185                 {
 186                         "files-invalid-perm-file",
 187                         nil,
 188                         fmt.Errorf("files-invalid-perm-file: \"files-invalid-perm-file/files/etc/resolv.conf\": invalid permissions 0600" + errMsg),
 189                 },
 190                 {
 191                         "files-invalid-perm-file-executable",
 192                         nil,
 193                         fmt.Errorf("files-invalid-perm-file-executable: \"files-invalid-perm-file-executable/files/etc/rc.local\": invalid permissions 0750" + errMsg),
 194                 },
 195                 {
 196                         "files-invalid-perm-file-sticky",
 197                         nil,
 198                         fmt.Errorf("files-invalid-perm-file-sticky: \"files-invalid-perm-file-sticky/files/etc/resolv.conf\": invalid permissions 01644" + errMsg),
 199                 },
 200         }
 201
 202         for _, tc := range tests {
 203                 t.Run(tc.group, func(t *testing.T) {
 204                         res, err := LoadFiles(tc.group)
 205                         testutil.AssertEqual(t, "res", res, tc.exp)
 206                         testutil.AssertErrorEqual(t, "err", err, tc.expErr)
 207                 })
 208         }
 209 }