64 lines
2.0 KiB
Ruby
64 lines
2.0 KiB
Ruby
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
|