From d27457ad9177b1c33ff5fba14cfa45ac509a6afc Mon Sep 17 00:00:00 2001
From: HD Moore <hd_moore@rapid7.com>
Date: Fri, 22 Apr 2011 20:44:13 +0000
Subject: [PATCH] Convert binary fields from text to binary

git-svn-id: file:///home/svn/framework3/trunk@12410 4d416f70-5f16-0410-b530-b9f4589650da
---
 .../migrate/20110422000000_convert_binary.rb  | 63 +++++++++++++++++++
 1 file changed, 63 insertions(+)
 create mode 100644 data/sql/migrate/20110422000000_convert_binary.rb

diff --git a/data/sql/migrate/20110422000000_convert_binary.rb b/data/sql/migrate/20110422000000_convert_binary.rb
new file mode 100644
index 0000000000..d258f19b4b
--- /dev/null
+++ b/data/sql/migrate/20110422000000_convert_binary.rb
@@ -0,0 +1,63 @@
+class ConvertBinary < ActiveRecord::Migration
+
+
+	class WebPage < ActiveRecord::Base
+		serialize :headers
+	end
+	
+	class WebVuln < ActiveRecord::Base
+		serialize :params
+	end
+		
+	def bfilter(str)
+		str = str.to_s
+		str.encoding = 'binary' if str.respond_to?('encoding=')
+		str.gsub(/[\x00\x7f-\xff]/, '')
+	end
+	
+	def self.up
+		rename_column :web_pages, :body, :body_text
+		rename_column :web_pages, :request, :request_text
+		rename_column :web_vulns, :request, :request_text
+		rename_column :web_vulns, :proof, :proof_text
+							
+		add_column :web_pages, :body, :binary
+		add_column :web_pages, :request, :binary
+		add_column :web_vulns, :request, :binary	
+		add_column :web_vulns, :proof, :binary
+		
+		WebPage.find(:all).each { |r| r.body = r.body_text; r.save! }
+		WebPage.find(:all).each { |r| r.request = r.request_text; r.save! }
+		WebVuln.find(:all).each { |r| r.proof = r.proof_text; r.save! }
+		WebVuln.find(:all).each { |r| r.request = r.request_text; r.save! }
+				
+		remove_column :web_pages, :body_text
+		remove_column :web_pages, :request_text
+		remove_column :web_vulns, :request_text
+		remove_column :web_vulns, :proof_text
+	end
+
+	def self.down
+
+		rename_column :web_pages, :body, :body_binary
+		rename_column :web_pages, :request, :request_binary
+		rename_column :web_vulns, :request, :request_binary
+		rename_column :web_vulns, :proof, :proof_binary
+							
+		add_column :web_pages, :body, :text
+		add_column :web_pages, :request, :text
+		add_column :web_vulns, :request, :text
+		add_column :web_vulns, :proof, :text
+		
+		WebPage.find(:all).each { |r| r.body = bfilter(r.body_binary); r.save! }
+		WebPage.find(:all).each { |r| r.request = bfilter(r.request_binary); r.save! }
+		WebVuln.find(:all).each { |r| r.proof = bfilter(r.proof_binary); r.save! }
+		WebVuln.find(:all).each { |r| r.request = bfilter(r.request_binary); r.save! }
+				
+		remove_column :web_pages, :body_binary
+		remove_column :web_pages, :request_binary
+		remove_column :web_vulns, :request_binary
+		remove_column :web_vulns, :proof_binary
+			
+	end
+end