2009-06-30 16:43:39 +00:00
|
|
|
require File.dirname(__FILE__) + '/aws/ec2'
|
2009-05-19 06:06:49 +00:00
|
|
|
require File.dirname(__FILE__) + '/aws/simpledb'
|
|
|
|
require File.dirname(__FILE__) + '/aws/s3'
|
2009-05-21 01:49:20 +00:00
|
|
|
|
|
|
|
require 'rubygems'
|
2009-06-22 05:13:01 +00:00
|
|
|
require 'openssl'
|
|
|
|
require 'socket'
|
2009-05-21 01:49:20 +00:00
|
|
|
require 'uri'
|
|
|
|
|
|
|
|
module Fog
|
|
|
|
module AWS
|
2009-06-22 05:13:01 +00:00
|
|
|
class Connection
|
2009-06-11 17:25:05 +00:00
|
|
|
|
2009-06-22 05:13:01 +00:00
|
|
|
def initialize(url)
|
|
|
|
@uri = URI.parse(url)
|
|
|
|
@connection = TCPSocket.open(@uri.host, @uri.port)
|
|
|
|
if @uri.scheme == 'https'
|
|
|
|
@ssl_context = OpenSSL::SSL::SSLContext.new
|
|
|
|
@ssl_context.verify_mode = OpenSSL::SSL::VERIFY_NONE
|
|
|
|
@connection = OpenSSL::SSL::SSLSocket.new(@connection, @ssl_context)
|
|
|
|
@connection.sync_close = true
|
|
|
|
@connection.connect
|
|
|
|
end
|
2009-05-24 16:11:32 +00:00
|
|
|
end
|
|
|
|
|
2009-06-22 05:13:01 +00:00
|
|
|
def request(params)
|
2009-06-25 07:32:27 +00:00
|
|
|
params[:path] ||= ''
|
|
|
|
unless params[:path][0] == '/'
|
|
|
|
params[:path] = '/' + params[:path].to_s
|
|
|
|
end
|
2009-06-27 19:08:05 +00:00
|
|
|
if params[:query] && !params[:query].empty?
|
2009-06-25 07:32:27 +00:00
|
|
|
params[:path] << "?#{params[:query]}"
|
2009-06-22 17:06:49 +00:00
|
|
|
end
|
2009-06-25 03:48:51 +00:00
|
|
|
request = "#{params[:method]} #{params[:path]} HTTP/1.1\r\n"
|
2009-06-23 07:04:29 +00:00
|
|
|
params[:headers] ||= {}
|
2009-06-25 03:48:51 +00:00
|
|
|
params[:headers]['Host'] = params[:host]
|
2009-06-22 17:06:49 +00:00
|
|
|
if params[:body]
|
|
|
|
params[:headers]['Content-Length'] = params[:body].length
|
|
|
|
end
|
2009-06-22 05:13:01 +00:00
|
|
|
for key, value in params[:headers]
|
2009-06-09 05:47:45 +00:00
|
|
|
request << "#{key}: #{value}\r\n"
|
|
|
|
end
|
2009-06-22 17:06:49 +00:00
|
|
|
request << "\r\n#{params[:body]}"
|
2009-06-22 05:13:01 +00:00
|
|
|
@connection.write(request)
|
2009-05-24 16:11:32 +00:00
|
|
|
|
2009-06-22 05:13:01 +00:00
|
|
|
response = AWS::Response.new
|
2009-06-22 18:32:27 +00:00
|
|
|
response.status = @connection.readline[9..11].to_i
|
2009-06-22 05:13:01 +00:00
|
|
|
while true
|
2009-06-24 16:57:21 +00:00
|
|
|
data = @connection.readline.chomp!
|
2009-06-23 07:04:29 +00:00
|
|
|
if data == ""
|
2009-06-22 17:06:49 +00:00
|
|
|
break
|
|
|
|
end
|
2009-06-23 07:04:29 +00:00
|
|
|
header = data.split(': ')
|
2009-06-29 17:03:44 +00:00
|
|
|
response.headers[capitalize(header[0])] = header[1]
|
2009-06-22 05:13:01 +00:00
|
|
|
end
|
2009-07-10 22:40:11 +00:00
|
|
|
|
|
|
|
if params[:parser]
|
|
|
|
body = Nokogiri::XML::SAX::PushParser.new(params[:parser])
|
|
|
|
else
|
|
|
|
body = ''
|
|
|
|
end
|
|
|
|
|
2009-06-25 07:32:27 +00:00
|
|
|
unless params[:method] == 'HEAD'
|
|
|
|
if response.headers['Content-Length']
|
2009-07-10 22:40:11 +00:00
|
|
|
body << @connection.read(response.headers['Content-Length'].to_i)
|
2009-06-25 07:32:27 +00:00
|
|
|
elsif response.headers['Transfer-Encoding'] == 'chunked'
|
|
|
|
while true
|
|
|
|
# 2 == "/r/n".length
|
|
|
|
chunk_size = @connection.readline.chomp!.to_i(16) + 2
|
2009-07-10 22:40:11 +00:00
|
|
|
body << @connection.read(chunk_size)
|
2009-06-25 07:32:27 +00:00
|
|
|
if chunk_size == 2
|
|
|
|
break
|
|
|
|
end
|
2009-06-22 17:06:49 +00:00
|
|
|
end
|
2009-05-24 16:11:32 +00:00
|
|
|
end
|
|
|
|
end
|
2009-07-10 22:40:11 +00:00
|
|
|
|
|
|
|
if params[:parser]
|
|
|
|
body.finish
|
|
|
|
response.body = params[:parser].response
|
|
|
|
else
|
|
|
|
response.body = body
|
|
|
|
end
|
|
|
|
|
2009-06-22 05:13:01 +00:00
|
|
|
response
|
2009-05-24 16:11:32 +00:00
|
|
|
end
|
2009-06-22 05:13:01 +00:00
|
|
|
|
2009-06-29 17:03:44 +00:00
|
|
|
private
|
|
|
|
|
|
|
|
def capitalize(header)
|
|
|
|
words = header.split('-')
|
|
|
|
header = ''
|
|
|
|
for word in words
|
|
|
|
header << word[0..0].upcase << word[1..-1] << '-'
|
|
|
|
end
|
|
|
|
header.chop!
|
|
|
|
end
|
|
|
|
|
2009-05-21 01:49:20 +00:00
|
|
|
end
|
2009-05-24 16:11:32 +00:00
|
|
|
|
|
|
|
class Response
|
2009-06-22 05:13:01 +00:00
|
|
|
|
2009-05-24 16:11:32 +00:00
|
|
|
attr_accessor :status, :headers, :body
|
|
|
|
|
|
|
|
def initialize
|
2009-06-22 05:13:01 +00:00
|
|
|
@body = ''
|
2009-05-24 16:11:32 +00:00
|
|
|
@headers = {}
|
|
|
|
end
|
2009-06-22 05:13:01 +00:00
|
|
|
|
2009-05-24 16:11:32 +00:00
|
|
|
end
|
|
|
|
|
2009-06-22 05:13:01 +00:00
|
|
|
|
2009-05-21 01:49:20 +00:00
|
|
|
end
|
2009-06-07 07:54:40 +00:00
|
|
|
end
|