62 lines
1.6 KiB
Ruby
Executable File
62 lines
1.6 KiB
Ruby
Executable File
#!/usr/bin/env ruby
|
|
# -*- coding: binary -*-
|
|
|
|
# $Id$
|
|
# $Revision$
|
|
|
|
msfbase = __FILE__
|
|
while File.symlink?(msfbase)
|
|
msfbase = File.expand_path(File.readlink(msfbase), File.dirname(msfbase))
|
|
end
|
|
|
|
@args = ARGV.dup
|
|
|
|
Dir.chdir(File.dirname(msfbase))
|
|
|
|
$stderr.puts "[*]"
|
|
$stderr.puts "[*] Attempting to update the Metasploit Framework..."
|
|
$stderr.puts "[*]"
|
|
$stderr.puts ""
|
|
|
|
if not (Process.uid == 0 or File.stat(msfbase).owned?)
|
|
$stderr.puts "[-] ERROR: User running msfupdate does not own the metasploit install"
|
|
$stderr.puts "Please run msfupdate as the same user who installed metasploit."
|
|
end
|
|
|
|
wait = (@args.shift.to_s == "wait")
|
|
|
|
have_configdir = false
|
|
@args.each do |arg|
|
|
next unless arg =~ /--config-dir/
|
|
have_configdir = true
|
|
end
|
|
|
|
unless have_configdir
|
|
configdir = File.join(File.dirname(msfbase), "data", "svn")
|
|
# Spaces in the directory should be fine since this whole thing is passed
|
|
# as a single argument via the multi-arg syntax for system() below.
|
|
@args.push("--config-dir=#{configdir}")
|
|
end
|
|
|
|
@args.push("--non-interactive")
|
|
|
|
res = system("svn", "cleanup")
|
|
if res.nil?
|
|
$stderr.puts "[-] ERROR: Failed to run svn"
|
|
$stderr.puts ""
|
|
$stderr.puts "[-] If you used a binary installer, make sure you run the symlink in"
|
|
$stderr.puts "[-] /usr/local/bin instead of running this file directly (e.g.: ./msfupdate)"
|
|
$stderr.puts "[-] to ensure a proper environment."
|
|
else
|
|
# Cleanup worked, go ahead and update
|
|
$stderr.puts "DEBUG: Going with *@args: #{@args.inspect}"
|
|
system("svn", "update", *@args)
|
|
end
|
|
|
|
if wait
|
|
$stderr.puts ""
|
|
$stderr.puts "[*] Please hit enter to exit"
|
|
$stderr.puts ""
|
|
$stdin.readline
|
|
end
|