|
1 | | -var chownr = require('chownr') |
2 | | -var dezalgo = require('dezalgo') |
3 | | -var fs = require('graceful-fs') |
4 | | -var inflight = require('inflight') |
5 | | -var log = require('npmlog') |
6 | | -var mkdirp = require('mkdirp') |
| 1 | +const chownr = require('chownr') |
| 2 | +const fs = require('graceful-fs') |
| 3 | +const inflight = require('inflight') |
| 4 | +const log = require('npmlog') |
| 5 | +const mkdirp = require('mkdirp') |
| 6 | +const { dirname } = require('path') |
| 7 | + |
| 8 | +// retain ownership of the parent dir |
| 9 | +// this matches behavior in cacache to infer the cache ownership |
| 10 | +// based on the ownership of the cache folder or it is parent. |
7 | 11 |
|
8 | | -// memoize the directories created by this step |
9 | | -var stats = {} |
10 | | -var effectiveOwner |
11 | 12 | module.exports = function correctMkdir (path, cb) { |
12 | | - cb = dezalgo(cb) |
13 | | - cb = inflight('correctMkdir:' + path, cb) |
| 13 | + cb = inflight('correctMkdir: ' + path, cb) |
14 | 14 | if (!cb) { |
15 | 15 | return log.verbose('correctMkdir', path, 'correctMkdir already in flight; waiting') |
16 | 16 | } else { |
17 | 17 | log.verbose('correctMkdir', path, 'correctMkdir not in flight; initializing') |
18 | 18 | } |
19 | 19 |
|
20 | | - if (stats[path]) return cb(null, stats[path]) |
21 | | - |
22 | | - fs.stat(path, function (er, st) { |
23 | | - if (er) return makeDirectory(path, cb) |
| 20 | + if (!process.getuid) { |
| 21 | + log.verbose('makeCacheDir', 'UID & GID are irrelevant on', process.platform) |
| 22 | + return mkdirp(path, (er, made) => cb(er, { uid: 0, gid: 0 })) |
| 23 | + } |
24 | 24 |
|
25 | | - if (!st.isDirectory()) { |
26 | | - log.error('correctMkdir', 'invalid dir %s', path) |
| 25 | + inferOwner(path, (er, owner) => { |
| 26 | + if (er) { |
| 27 | + log.error('correctMkdir', 'failed to infer path ownership %s', path) |
27 | 28 | return cb(er) |
28 | 29 | } |
29 | 30 |
|
30 | | - var ownerStats = calculateOwner() |
31 | | - // there's always a chance the permissions could have been frobbed, so fix |
32 | | - if (st.uid !== ownerStats.uid) { |
33 | | - stats[path] = ownerStats |
34 | | - setPermissions(path, ownerStats, cb) |
35 | | - } else { |
36 | | - stats[path] = st |
37 | | - cb(null, stats[path]) |
38 | | - } |
39 | | - }) |
40 | | -} |
41 | | - |
42 | | -function calculateOwner () { |
43 | | - if (!effectiveOwner) { |
44 | | - effectiveOwner = { uid: 0, gid: 0 } |
45 | | - |
46 | | - // Pretty much only on windows |
47 | | - if (!process.getuid) { |
48 | | - return effectiveOwner |
49 | | - } |
50 | | - |
51 | | - effectiveOwner.uid = +process.getuid() |
52 | | - effectiveOwner.gid = +process.getgid() |
53 | | - |
54 | | - if (effectiveOwner.uid === 0) { |
55 | | - if (process.env.SUDO_UID) effectiveOwner.uid = +process.env.SUDO_UID |
56 | | - if (process.env.SUDO_GID) effectiveOwner.gid = +process.env.SUDO_GID |
57 | | - } |
58 | | - } |
59 | | - |
60 | | - return effectiveOwner |
61 | | -} |
62 | | - |
63 | | -function makeDirectory (path, cb) { |
64 | | - cb = inflight('makeDirectory:' + path, cb) |
65 | | - if (!cb) { |
66 | | - return log.verbose('makeDirectory', path, 'creation already in flight; waiting') |
67 | | - } else { |
68 | | - log.verbose('makeDirectory', path, 'creation not in flight; initializing') |
69 | | - } |
70 | | - |
71 | | - var owner = calculateOwner() |
72 | | - |
73 | | - if (!process.getuid) { |
74 | | - return mkdirp(path, function (er) { |
75 | | - log.verbose('makeCacheDir', 'UID & GID are irrelevant on', process.platform) |
76 | | - |
77 | | - stats[path] = owner |
78 | | - return cb(er, stats[path]) |
79 | | - }) |
80 | | - } |
81 | | - |
82 | | - if (owner.uid !== 0 || !process.env.HOME) { |
83 | | - log.silly( |
84 | | - 'makeDirectory', path, |
85 | | - 'uid:', owner.uid, |
86 | | - 'gid:', owner.gid |
87 | | - ) |
88 | | - stats[path] = owner |
89 | | - mkdirp(path, afterMkdir) |
90 | | - } else { |
91 | | - fs.stat(process.env.HOME, function (er, st) { |
| 31 | + mkdirp(path, (er, made) => { |
92 | 32 | if (er) { |
93 | | - log.error('makeDirectory', 'homeless?') |
| 33 | + log.error('correctMkdir', 'failed to make directory %s', path) |
94 | 34 | return cb(er) |
95 | 35 | } |
96 | | - |
97 | | - log.silly( |
98 | | - 'makeDirectory', path, |
99 | | - 'uid:', st.uid, |
100 | | - 'gid:', st.gid |
101 | | - ) |
102 | | - stats[path] = st |
103 | | - mkdirp(path, afterMkdir) |
| 36 | + chownr(made || path, owner.uid, owner.gid, (er) => cb(er, owner)) |
104 | 37 | }) |
105 | | - } |
106 | | - |
107 | | - function afterMkdir (er, made) { |
108 | | - if (er || !stats[path] || isNaN(stats[path].uid) || isNaN(stats[path].gid)) { |
109 | | - return cb(er, stats[path]) |
110 | | - } |
111 | | - |
112 | | - if (!made) return cb(er, stats[path]) |
| 38 | + }) |
| 39 | +} |
113 | 40 |
|
114 | | - setPermissions(made, stats[path], cb) |
| 41 | +const pathToOwner = new Map() |
| 42 | +const inferOwner = (path, cb) => { |
| 43 | + if (pathToOwner.has(path)) { |
| 44 | + return cb(null, pathToOwner.get(path)) |
115 | 45 | } |
116 | | -} |
117 | 46 |
|
118 | | -function setPermissions (path, st, cb) { |
119 | | - chownr(path, st.uid, st.gid, function (er) { |
120 | | - if (er && er.code === 'ENOENT') return cb(null, st) |
121 | | - return cb(er, st) |
| 47 | + const parent = dirname(path) |
| 48 | + fs.lstat(path, (er, st) => { |
| 49 | + if (er && parent !== path) { |
| 50 | + inferOwner(parent, (er, owner) => { |
| 51 | + if (er) { |
| 52 | + return cb(er) |
| 53 | + } |
| 54 | + pathToOwner.set(path, owner) |
| 55 | + return cb(null, owner) |
| 56 | + }) |
| 57 | + } else if (er) { |
| 58 | + cb(er) |
| 59 | + } else { |
| 60 | + const owner = { uid: st.uid, gid: st.gid } |
| 61 | + pathToOwner.set(path, owner) |
| 62 | + cb(null, owner) |
| 63 | + } |
122 | 64 | }) |
123 | 65 | } |
0 commit comments