From 73f136ef9a8161deb2dda0a3d6b59ea0fb2aa1f3 Mon Sep 17 00:00:00 2001
From: Brandon Turner <brandon_turner@rapid7.com>
Date: Mon, 4 Feb 2013 15:01:35 -0600
Subject: [PATCH] Update msfupdate to work with debian packages

If apt was used to install framework, use apt-get to update.
---
 msfupdate | 43 +++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 41 insertions(+), 2 deletions(-)

diff --git a/msfupdate b/msfupdate
index 058fd8f51e..6b168d0091 100755
--- a/msfupdate
+++ b/msfupdate
@@ -30,9 +30,13 @@ if not (Process.uid == 0 or File.stat(msfbase).owned?)
 	exit 0x10
 end
 
+def is_apt
+	File.exists?(File.expand_path(File.join(@msfbase_dir, '.apt')))
+end
+
 # Are you an installer, or did you get here via a source checkout?
 def is_installed
-	File.exists?(File.expand_path(File.join(@msfbase_dir, "..", "engine", "update.rb")))
+	File.exists?(File.expand_path(File.join(@msfbase_dir, "..", "engine", "update.rb"))) && !is_apt
 end
 
 def is_git
@@ -69,6 +73,24 @@ def maybe_wait_and_exit(exit_code=0)
 	end
 end
 
+def apt_upgrade_available(package)
+	require 'open3'
+	installed = nil
+	upgrade = nil
+	::Open3.popen3("apt-cache", "policy", package) do |stdin, stdout, stderr|
+		stdout.each do |line|
+			installed = $1 if line =~ /Installed: ([\w\-+.:~]+)$/
+			upgrade = $1 if line =~ /Candidate: ([\w\-+.:~]+)$/
+			break if installed && upgrade
+		end
+	end
+	if installed && installed != upgrade
+		upgrade
+	else
+		nil
+	end
+end
+
 # Some of these args are meaningful for SVN, some for Git,
 # some for both. Fun times.
 @args.each_with_index do |arg,i|
@@ -186,7 +208,24 @@ if is_installed
 	end
 end
 
-unless is_svn || is_git || is_installed
+if is_apt
+	$stdout.puts "[*] Checking for updates"
+	system("apt-get", "-qq", "update")
+
+	packages = []
+	packages << 'metasploit-framework' if framework_version = apt_upgrade_available('metasploit-framework')
+	packages << 'metasploit' if pro_version = apt_upgrade_available('metasploit')
+
+	if packages.empty?
+		$stdout.puts "[*] No updates available"
+	else
+		$stdout.puts "[*] Updating to version #{pro_version || framework_version}"
+		system("apt-get", "install", "--assume-yes", *packages)
+		system("/etc/init.d/metasploit start") if packages.include?('metasploit')
+	end
+end
+
+unless is_svn || is_git || is_installed || is_apt
 	raise RuntimeError, "Cannot determine checkout type: `#{@msfbase_dir}'"
 end