forked from snapcore/snapd
-
Notifications
You must be signed in to change notification settings - Fork 0
/
mkdirallchown.go
87 lines (72 loc) · 2.06 KB
/
mkdirallchown.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
86
87
// -*- Mode: Go; indent-tabs-mode: t -*-
/*
* Copyright (C) 2016 Canonical Ltd
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License version 3 as
* published by the Free Software Foundation.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*
*/
package osutil
import (
"os"
"path/filepath"
"sync"
"syscall"
"github.com/snapcore/snapd/osutil/sys"
)
// XXX: we need to come back and fix this; this is a hack to unblock us.
// As part of the fixing we should unify with the similar code in
// cmd/snap-update-ns/utils.(*Secure).MkdirAll
var mu sync.Mutex
// MkdirAllChown is like os.MkdirAll but it calls os.Chown on any
// directories it creates.
func MkdirAllChown(path string, perm os.FileMode, uid sys.UserID, gid sys.GroupID) error {
mu.Lock()
defer mu.Unlock()
return mkdirAllChown(filepath.Clean(path), perm, uid, gid)
}
func mkdirAllChown(path string, perm os.FileMode, uid sys.UserID, gid sys.GroupID) error {
// split out so filepath.Clean isn't called twice for each inner path
if s, err := os.Stat(path); err == nil {
if s.IsDir() {
return nil
}
// emulate os.MkdirAll
return &os.PathError{
Op: "mkdir",
Path: path,
Err: syscall.ENOTDIR,
}
}
dir := filepath.Dir(path)
if dir != "/" {
if err := mkdirAllChown(dir, perm, uid, gid); err != nil {
return err
}
}
cand := path + ".mkdir-new"
if err := os.Mkdir(cand, perm); err != nil && !os.IsExist(err) {
return err
}
if err := sys.ChownPath(cand, uid, gid); err != nil {
return err
}
if err := os.Rename(cand, path); err != nil {
return err
}
fd, err := os.Open(dir)
if err != nil {
return err
}
defer fd.Close()
return fd.Sync()
}