Requests db storage

git-svn-id: file:///home/svn/framework3/trunk@8997 4d416f70-5f16-0410-b530-b9f4589650da
This commit is contained in:
et 2010-04-03 07:24:48 +00:00
parent eac38aaa23
commit e4a17dd0ca
1 changed files with 30 additions and 11 deletions

View File

@ -117,25 +117,44 @@ class HttpCrawler
db = SQLite3::Database.new(dbpath) db = SQLite3::Database.new(dbpath)
#db = Mysql.new("127.0.0.1", username, password, databasename) #db = Mysql.new("127.0.0.1", username, password, databasename)
until !db.transaction_active? until !db.transaction_active?
puts "Waiting for db" #puts "Waiting for db"
#wait #wait
end end
#puts "db: #{db.transaction_active?}" #puts "db: #{db.transaction_active?}"
db.transaction db.execute( "insert into wmap_requests values (?,?,?,?,?,?,?,?,?,?,?,?,?,?,?)",
nil, #CREATE TABLE "wmap_requests" (
# "id" INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,
# "host" varchar(255),
# "address" varchar(16),
# "address6" varchar(255),
# "port" integer,
# "ssl" integer,
# "meth" varchar(32),
# "path" text,
# "headers" text,
# "query" text,
# "body" text,
# "respcode" varchar(16),
# "resphead" text,
# "response" text,
# "created_at" datetime);
db.transaction db.execute( "insert into wmap_requests (host,address,address6,port,ssl,meth,path,headers,query,body,respcode,resphead,response,created_at,updated_at) values (?,?,?,?,?,?,?,?,?,?,?,?,?,?,?)",
hashreq['rhost'], hashreq['rhost'],
hashreq['rhost'], hashreq['rhost'],
hashreq['rhost'], hashreq['rhost'],
hashreq['rport'].to_i, hashreq['rport'].to_i,
hashreq['ssl'], hashreq['ssl']? 1:0,
hashreq['method'], hashreq['method'],
SQLite3::Blob.new(hashreq['uri']), SQLite3::Blob.new(hashreq['uri']),
SQLite3::Blob.new("a"), SQLite3::Blob.new(''),
SQLite3::Blob.new("b"), SQLite3::Blob.new(hashreq['query']? hashreq['query']:''),
SQLite3::Blob.new("c"), SQLite3::Blob.new(hashreq['data']? hashreq['data']:''),
"200", response.code.to_s,
SQLite3::Blob.new("d"), SQLite3::Blob.new(''),
SQLite3::Blob.new("e"), SQLite3::Blob.new(response.body.to_s),
Time.new,
Time.new Time.new
) )
db.commit db.commit
@ -297,7 +316,7 @@ class HttpCrawler
#rescue ::Rex::ConnectionRefused, ::Rex::HostUnreachable, ::Rex::ConnectionTimeout #rescue ::Rex::ConnectionRefused, ::Rex::HostUnreachable, ::Rex::ConnectionTimeout
#rescue ::Timeout::Error, ::Errno::EPIPE #rescue ::Timeout::Error, ::Errno::EPIPE
rescue rescue
puts "ERROR" puts "ERROR #{$!.backtrace}"
end end
end end