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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
|
# A module to collect utility functions.
require 'sync'
require 'puppet/lock'
module Puppet
module Util
# Create a sync point for any threads
@@sync = Sync.new
# Execute a block as a given user or group
def self.asuser(user = nil, group = nil)
require 'etc'
uid = nil
gid = nil
olduid = nil
oldgid = nil
begin
# the groupid, if we got passed a group
# The gid has to be changed first, because, well, otherwise we won't
# be able to
if group
if group.is_a? Integer
gid = group
else
gid = self.gid(group)
end
if Process.gid != gid
oldgid = Process.gid
begin
Process.egid = gid
rescue => detail
raise Puppet::Error, "Could not change GID: %s" % detail
end
end
end
if user
if user.is_a? Integer
uid = user
else
uid = self.uid(user)
end
uid = self.uid(user)
# Now change the uid
if Process.uid != uid
olduid = Process.uid
begin
Process.euid = uid
rescue => detail
raise Puppet::Error, "Could not change UID: %s" % detail
end
end
end
retval = yield
ensure
if olduid
Process.euid = olduid
end
if oldgid
Process.egid = oldgid
end
end
return retval
end
# Change the process to a different user
def self.chuser
if group = Puppet[:group]
group = self.gid(group)
unless group
raise Puppet::Error, "No such group %s" % Puppet[:group]
end
unless Process.gid == group
begin
Process.egid = group
Process.gid = group
rescue
$stderr.puts "could not change to group %s" % group
exit(74)
end
end
end
if user = Puppet[:user]
user = self.uid(user)
unless user
raise Puppet::Error, "No such user %s" % Puppet[:user]
end
unless Process.uid == user
begin
Process.euid = user
Process.uid = user
rescue
$stderr.puts "could not change to user %s" % user
exit(74)
end
end
end
end
# Create a shared lock for reading
def self.readlock(file)
@@sync.synchronize(Sync::SH) do
File.open(file) { |f|
f.lock_shared { |lf| yield lf }
}
end
end
# Create an exclusive lock fro writing, and do the writing in a
# tmp file.
def self.writelock(file, mode = 0600)
tmpfile = file + ".tmp"
@@sync.synchronize(Sync::EX) do
File.open(file, "w", mode) do |rf|
rf.lock_exclusive do |lrf|
yield lrf
File.open(tmpfile, "w", mode) do |tf|
yield tf
tf.flush
end
begin
File.rename(tmpfile, file)
rescue => detail
Puppet.err "Could not rename %s to %s: %s" %
[file, tmpfile, detail]
end
end
end
end
end
# Get the GID of a given group, provided either a GID or a name
def self.gid(group)
if group =~ /^\d+$/
group = Integer(group)
end
unless group
raise Puppet::DevError, "Invalid group %s" % group.inspect
end
gid = nil
obj = nil
# We want to look the group up either way
if group.is_a?(Integer)
# If this doesn't find anything
obj = Puppet.type(:group).find { |gobj|
gobj.should(:gid) == group ||
gobj.is(:gid) == group
}
unless obj
begin
gobj = Etc.getgrgid(group)
gid = gobj.gid
rescue ArgumentError => detail
# ignore it; we couldn't find the group
end
end
else
unless obj = Puppet.type(:group)[group]
obj = Puppet.type(:group).create(
:name => group,
:check => [:gid]
)
obj.retrieve
end
end
if obj
gid = obj.should(:gid) || obj.is(:gid)
if gid == :absent
gid = nil
end
end
return gid
end
# Get the UID of a given user, whether a UID or name is provided
def self.uid(user)
uid = nil
if user =~ /^\d+$/
user = Integer(user)
end
if user.is_a?(Integer)
# If this doesn't find anything
obj = Puppet.type(:user).find { |uobj|
uobj.should(:uid) == user ||
uobj.is(:uid) == user
}
unless obj
begin
uobj = Etc.getpwuid(user)
uid = uobj.uid
rescue ArgumentError => detail
# ignore it; we couldn't find the user
end
end
else
unless obj = Puppet.type(:user)[user]
obj = Puppet.type(:user).create(
:name => user,
:check => [:uid, :gid]
)
end
end
if obj
obj.retrieve
uid = obj.should(:uid) || obj.is(:uid)
if uid == :absent
uid = nil
end
end
return uid
end
# Create instance methods for each of the log levels. This allows
# the messages to be a little richer. Most classes will be calling this
# method.
def self.logmethods(klass, useself = true)
Puppet::Log.eachlevel { |level|
klass.send(:define_method, level, proc { |args|
if args.is_a?(Array)
args = args.join(" ")
end
if useself
Puppet::Log.create(
:level => level,
:source => self,
:message => args
)
else
Puppet::Log.create(
:level => level,
:message => args
)
end
})
}
end
# XXX this should all be done using puppet objects, not using
# normal mkdir
def self.recmkdir(dir,mode = 0755)
if FileTest.exist?(dir)
return false
else
tmp = dir.sub(/^\//,'')
path = [File::SEPARATOR]
tmp.split(File::SEPARATOR).each { |dir|
path.push dir
if ! FileTest.exist?(File.join(path))
Dir.mkdir(File.join(path), mode)
elsif FileTest.directory?(File.join(path))
next
else FileTest.exist?(File.join(path))
raise "Cannot create %s: basedir %s is a file" %
[dir, File.join(path)]
end
}
return true
end
end
def self.symbolize(value)
case value
when String: value = value.intern
when Symbol: # nothing
else
raise ArgumentError, "'%s' must be a string or symbol" % value
end
end
end
end
# $Id$
|
