Module: FPM::Util

Extended by:
FFI::Library
Included in:
Command, Command::Validator, Package, Package::NPM
Defined in:
lib/fpm/util.rb,
lib/fpm/util/tar_writer.rb

Overview

Some utility functions

Defined Under Namespace

Classes: ExecutableNotFound, ProcessFailed, TarWriter

Instance Method Summary collapse

Instance Method Details

#ar_cmdObject

Get an array containing the recommended 'ar' command for this platform and the recommended options to quickly create/append to an archive without timestamps or uids (if possible).


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
# File 'lib/fpm/util.rb', line 238

def ar_cmd
  return @@ar_cmd if defined? @@ar_cmd

  @@ar_cmd_deterministic = false

  # FIXME: don't assume current directory writeable
  FileUtils.touch(["fpm-dummy.tmp"])
  ["ar", "gar"].each do |ar|
    ["-qc", "-qcD"].each do |ar_create_opts|
      FileUtils.rm_f(["fpm-dummy.ar.tmp"])
      # Return this combination if it creates archives without uids or timestamps.
      # Exitstatus will be nonzero if the archive can't be created,
      # or its table of contents doesn't match the regular expression.
      # Be extra-careful about locale and timezone when matching output.
      system("#{ar} #{ar_create_opts} fpm-dummy.ar.tmp fpm-dummy.tmp 2>/dev/null && env TZ=UTC LANG=C LC_TIME=C #{ar} -tv fpm-dummy.ar.tmp | grep '0/0.*1970' > /dev/null 2>&1")
      if $?.exitstatus == 0
         @@ar_cmd = [ar, ar_create_opts]
         @@ar_cmd_deterministic = true
         return @@ar_cmd
      end
    end
  end
  # If no combination of ar and options omits timestamps, fall back to default.
  @@ar_cmd = ["ar", "-qc"]
  return @@ar_cmd
ensure
  # Clean up
  FileUtils.rm_f(["fpm-dummy.ar.tmp", "fpm-dummy.tmp"])
end

#ar_cmd_deterministic?Boolean

Return whether the command returned by ar_cmd can create deterministic archives

Returns:

  • (Boolean)

269
270
271
272
# File 'lib/fpm/util.rb', line 269

def ar_cmd_deterministic?
  ar_cmd if not defined? @@ar_cmd_deterministic
  return @@ar_cmd_deterministic
end

#copied_entriesObject

def copy_entry


378
379
380
381
382
383
384
385
386
387
388
389
# File 'lib/fpm/util.rb', line 378

def copied_entries
  # TODO(sissel): I wonder that this entry-copy knowledge needs to be put
  # into a separate class/module. As is, calling copy_entry the same way
  # in slightly different contexts will result in weird or bad behavior.
  # What I mean is if we do:
  #   pkg = FPM::Package::Dir...
  #   pkg.output()...
  #   pkg.output()...
  # The 2nd output call will fail or behave weirdly because @copied_entries
  # is already populated. even though this is anew round of copying.
  return @copied_entries ||= {}
end

#copy_entry(src, dst, preserve = false, remove_destination = false) ⇒ Object


355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
# File 'lib/fpm/util.rb', line 355

def copy_entry(src, dst, preserve=false, remove_destination=false)
  case File.ftype(src)
  when 'fifo', 'characterSpecial', 'blockSpecial', 'socket'
    st = File.stat(src)
    rc = mknod_w(dst, st.mode, st.dev)
    raise SystemCallError.new("mknod error", FFI.errno) if rc == -1
  when 'directory'
    FileUtils.mkdir(dst) unless File.exists? dst
  else
    # if the file with the same dev and inode has been copied already -
    # hard link it's copy to `dst`, otherwise make an actual copy
    st = File.lstat(src)
    known_entry = copied_entries[[st.dev, st.ino]]
    if known_entry
      FileUtils.ln(known_entry, dst)
    else
      FileUtils.copy_entry(src, dst, preserve, false,
                           remove_destination)
      copied_entries[[st.dev, st.ino]] = dst
    end
  end # else...
end

#copy_metadata(source, destination) ⇒ Object


332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
# File 'lib/fpm/util.rb', line 332

def (source, destination)
  source_stat = File::lstat(source)
  dest_stat = File::lstat(destination)

  # If this is a hard-link, there's no metadata to copy.
  # If this is a symlink, what it points to hasn't been copied yet.
  return if source_stat.ino == dest_stat.ino || dest_stat.symlink?

  File.utime(source_stat.atime, source_stat.mtime, destination)
  mode = source_stat.mode
  begin
    File.lchown(source_stat.uid, source_stat.gid, destination)
  rescue Errno::EPERM
    # clear setuid/setgid
    mode &= 01777
  end

  unless source_stat.symlink?
    File.chmod(mode, destination)
  end
end

#default_shellObject

def program_exists?


43
44
45
46
47
# File 'lib/fpm/util.rb', line 43

def default_shell
  shell = ENV["SHELL"]
  return "/bin/sh" if shell.nil? || shell.empty?
  return shell
end

#execmd(*args) ⇒ Object

execmd( cmd [,opts])

Execute a command as a child process. The function allows to:

  • pass environment variables to child process,

  • communicate with stdin, stdout and stderr of the child process via pipes,

  • retrieve execution's status code.

—- EXAMPLE 1 (simple execution)

if execmd(['which', 'python']) == 0

p "Python is installed"

end

—- EXAMPLE 2 (custom environment variables)

execmd(:PYTHONPATH=>'/home/me/foo', [ 'python', '-m', 'bar'])

—- EXAMPLE 3 (communicating via stdin, stdout and stderr)

script = <<PYTHON import sys sys.stdout.write(“normal outputn”) sys.stdout.write(“narning or errorn”) PYTHON status = execmd('python') do |stdin,stdout,stderr|

stdin.write(script)
stdin.close
p "STDOUT: #{stdout.read}"
p "STDERR: #{stderr.read}"

end p “STATUS: #status”

—- EXAMPLE 4 (additional options)

execmd(['which', 'python'], :process=>true, :stdin=>false, :stderr=>false) do |process,stdout|

p = stdout.read.chomp
process.wait
if (x = process.exit_code) == 0
  p "PYTHON: #{p}"
else
  p "ERROR:  #{x}"
end

end

OPTIONS:

:process (default: false) -- pass process object as the first argument the to block,
:stdin   (default: true)  -- pass stdin object of the child process to the block for writting,
:stdout  (default: true)  -- pass stdout object of the child process to the block for reading,
:stderr  (default: true)  -- pass stderr object of the child process to the block for reading,

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
# File 'lib/fpm/util.rb', line 103

def execmd(*args)
  i = 0
  if i < args.size
    if args[i].kind_of?(Hash)
      # args[0] may contain environment variables
      env = args[i]
      i += 1
    else
      env = Hash[]
    end
  end

  if i < args.size
    if args[i].kind_of?(Array)
      args2 = args[i]
    else
      args2 = [ args[i] ]
    end
    program = args2[0]
    i += 1
  else
    raise ArgumentError.new("missing argument: cmd")
  end

  if i < args.size
    if args[i].kind_of?(Hash)
      opts = Hash[args[i].map {|k,v| [k.to_sym, v]} ]
      i += 1
    end
  else
    opts = Hash[]
  end

  opts[:process] = false unless opts.include?(:process)
  opts[:stdin]   = true  unless opts.include?(:stdin)
  opts[:stdout]  = true  unless opts.include?(:stdout)
  opts[:stderr]  = true  unless opts.include?(:stderr)

  if !program.include?("/") and !program_in_path?(program)
    raise ExecutableNotFound.new(program)
  end

  logger.debug("Running command", :args => args2)

  stdout_r, stdout_w = IO.pipe
  stderr_r, stderr_w = IO.pipe

  process = ChildProcess.build(*args2)
  process.environment.merge!(env)

  process.io.stdout = stdout_w
  process.io.stderr = stderr_w

  if block_given? and opts[:stdin]
    process.duplex = true
  end

  process.start

  stdout_w.close; stderr_w.close
  logger.debug("Process is running", :pid => process.pid)
  if block_given?
    args3 = []
    args3.push(process)           if opts[:process]
    args3.push(process.io.stdin)  if opts[:stdin]
    args3.push(stdout_r)          if opts[:stdout]
    args3.push(stderr_r)          if opts[:stderr]

    yield(*args3)

    process.io.stdin.close        if opts[:stdin] and not process.io.stdin.closed?
    stdout_r.close                unless stdout_r.closed?
    stderr_r.close                unless stderr_r.closed?
  else
    # Log both stdout and stderr as 'info' because nobody uses stderr for
    # actually reporting errors and as a result 'stderr' is a misnomer.
    logger.pipe(stdout_r => :info, stderr_r => :info)
  end

  process.wait if process.alive?

  return process.exit_code
end

#expand_pessimistic_constraints(constraint) ⇒ Object

def copied_entries


391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
# File 'lib/fpm/util.rb', line 391

def expand_pessimistic_constraints(constraint)
  name, op, version = constraint.split(/\s+/)

  if op == '~>'

    new_lower_constraint = "#{name} >= #{version}"

    version_components = version.split('.').collect { |v| v.to_i }

    version_prefix = version_components[0..-3].join('.')
    portion_to_work_with = version_components.last(2)

    prefix = ''
    unless version_prefix.empty?
      prefix = version_prefix + '.'
    end

    one_to_increment = portion_to_work_with[0].to_i
    incremented = one_to_increment + 1

    new_version = ''+ incremented.to_s + '.0'

    upper_version = prefix + new_version

    new_upper_constraint = "#{name} < #{upper_version}"

    return [new_lower_constraint,new_upper_constraint]
  else
    return [constraint]
  end
end

#loggerObject

def expand_pesimistic_constraints


423
424
425
# File 'lib/fpm/util.rb', line 423

def logger
  @logger ||= Cabin::Channel.get
end

#mknod_w(path, mode, dev) ⇒ Object

wrapper around mknod ffi calls


320
321
322
323
324
325
326
327
328
329
330
# File 'lib/fpm/util.rb', line 320

def mknod_w(path, mode, dev)
  rc = -1
  case %x{uname -s}.chomp
  when 'Linux'
    # bits/stat.h #define _MKNOD_VER_LINUX  0
    rc = xmknod(0, path, mode, FFI::MemoryPointer.new(dev))
  else
    rc = mknod(path, mode, dev)
  end
  rc
end

#program_exists?(program) ⇒ Boolean

def program_in_path

Returns:

  • (Boolean)

36
37
38
39
40
41
# File 'lib/fpm/util.rb', line 36

def program_exists?(program)
  # Scan path to find the executable
  # Do this to help the user get a better error message.
  return program_in_path?(program) if !program.include?("/")
  return File.executable?(program)
end

#program_in_path?(program) ⇒ Boolean

Is the given program in the system's PATH?

Returns:

  • (Boolean)

27
28
29
30
31
32
33
34
# File 'lib/fpm/util.rb', line 27

def program_in_path?(program)
  # return false if path is not set
  return false unless ENV['PATH']
  # Scan path to find the executable
  # Do this to help the user get a better error message.
  envpath = ENV["PATH"].split(":")
  return envpath.select { |p| File.executable?(File.join(p, program)) }.any?
end

#safesystem(*args) ⇒ Object

Run a command safely in a way that gets reports useful errors.


188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
# File 'lib/fpm/util.rb', line 188

def safesystem(*args)
  # ChildProcess isn't smart enough to run a $SHELL if there's
  # spaces in the first arg and there's only 1 arg.
  if args.size == 1
    args = [ default_shell, "-c", args[0] ]
  end

  if args[0].kind_of?(Hash)
    env = args.shift()
    exit_code = execmd(env, args)
  else
    exit_code = execmd(args)
  end
  program = args[0]
  success = (exit_code == 0)

  if !success
    raise ProcessFailed.new("#{program} failed (exit code #{exit_code})" \
                            ". Full command was:#{args.inspect}")
  end
  return success
end

#safesystemout(*args) ⇒ Object

Run a command safely in a way that captures output and status.


212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
# File 'lib/fpm/util.rb', line 212

def safesystemout(*args)
  if args.size == 1
    args = [ ENV["SHELL"], "-c", args[0] ]
  end
  program = args[0]

  if !program.include?("/") and !program_in_path?(program)
    raise ExecutableNotFound.new(program)
  end

  stdout_r_str = nil
  exit_code = execmd(args, :stdin=>false, :stderr=>false) do |stdout|
    stdout_r_str = stdout.read
  end
  success = (exit_code == 0)

  if !success
    raise ProcessFailed.new("#{program} failed (exit code #{exit_code})" \
                            ". Full command was:#{args.inspect}")
  end
  return stdout_r_str
end

#tar_cmdObject

Get the recommended 'tar' command for this platform.


275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
# File 'lib/fpm/util.rb', line 275

def tar_cmd
  return @@tar_cmd if defined? @@tar_cmd

  # FIXME: don't assume current directory writeable
  FileUtils.touch(["fpm-dummy.tmp"])

  # Prefer tar that supports more of the features we want, stop if we find tar of our dreams
  best="tar"
  bestscore=0
  @@tar_cmd_deterministic = false
  # GNU Tar, if not the default, is usually on the path as gtar, but
  # Mac OS X 10.8 and earlier shipped it as /usr/bin/gnutar
  ["tar", "gtar", "gnutar"].each do |tar|
    opts=[]
    score=0
    ["--sort=name", "[email protected]"].each do |opt|
      system("#{tar} #{opt} -cf fpm-dummy.tar.tmp fpm-dummy.tmp > /dev/null 2>&1")
      if $?.exitstatus == 0
        opts << opt
        score += 1
      end
    end
    if score > bestscore
      best=tar
      bestscore=score
      if score == 2
        @@tar_cmd_deterministic = true
        break
      end
    end
  end
  @@tar_cmd = best
  return @@tar_cmd
ensure
  # Clean up
  FileUtils.rm_f(["fpm-dummy.tar.tmp", "fpm-dummy.tmp"])
end

#tar_cmd_supports_sort_names_and_set_mtime?Boolean

Return whether the command returned by tar_cmd can create deterministic archives

Returns:

  • (Boolean)

314
315
316
317
# File 'lib/fpm/util.rb', line 314

def tar_cmd_supports_sort_names_and_set_mtime?
  tar_cmd if not defined? @@tar_cmd_deterministic
  return @@tar_cmd_deterministic
end