-
-
Save tesfabpel/7a43c54c59ce0256186b to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#!/usr/bin/env ruby | |
# A quick and dirty implementation of an HTTP proxy server in Ruby | |
# because I did not want to install anything. | |
# | |
# Copyright (C) 2009 Torsten Becker <torsten.becker@gmail.com> | |
require 'optparse' | |
require 'socket' | |
require 'uri' | |
class Proxy | |
def run options | |
begin | |
$port = options[:port] | |
$domain = options[:domain] | |
# Start our server to handle connections (will raise things on errors) | |
@socket = TCPServer.new $port | |
# Handle every request in another thread | |
loop do | |
s = @socket.accept | |
Thread.new s, &method(:handle_request) | |
end | |
# CTRL-C | |
rescue Interrupt | |
puts 'Got Interrupt..' | |
# Ensure that we release the socket on errors | |
ensure | |
if @socket | |
@socket.close | |
puts 'Socked closed..' | |
end | |
puts 'Quitting.' | |
end | |
end | |
def handle_request to_client | |
request_line = to_client.readline | |
verb = request_line[/^\w+/] | |
url = request_line[/^\w+\s+(\S+)/, 1] | |
version = request_line[/HTTP\/(1\.\d)\s*$/, 1] | |
uri = URI::parse url | |
#puts(uri) | |
uri.host = $domain | |
# Show what got requested | |
puts((" %4s "%verb) + url) | |
to_server = TCPSocket.new(uri.host, (uri.port.nil? ? 80 : uri.port)) | |
to_server.write("#{verb} #{uri.path}?#{uri.query} HTTP/#{version}\r\n") | |
content_len = 0 | |
loop do | |
line = to_client.readline | |
if line =~ /^Content-Length:\s+(\d+)\s*$/ | |
content_len = $1.to_i | |
end | |
# Strip proxy headers | |
if line =~ /^proxy/i | |
next | |
elsif line =~ /^Host\:/i | |
next | |
elsif line.strip.empty? | |
to_server.write("Host: #{uri.host}\r\n") | |
#puts("Host: #{uri.host}\r\n") | |
to_server.write("Connection: close\r\n\r\n") | |
#puts("Connection: close\r\n\r\n") | |
if content_len >= 0 | |
to_server.write(to_client.read(content_len)) | |
end | |
break | |
else | |
to_server.write(line) | |
#puts(line) | |
end | |
end | |
#to_server.each_byte do |b| | |
# to_client.putc b | |
#end | |
IO.copy_stream(to_server, to_client) | |
=begin | |
buff = "" | |
loop do | |
to_server.read(4048, buff) | |
to_client.write(buff) | |
break if buff.size < 4048 | |
end | |
=end | |
# Close the sockets | |
to_client.close | |
to_server.close | |
end | |
end | |
# Get parameters and start the server | |
=begin | |
if ARGV.empty? | |
port = 8008 | |
elsif ARGV.size == 1 | |
port = ARGV[0].to_i | |
else | |
puts 'Usage: proxy.rb [port]' | |
exit 1 | |
end | |
=end | |
options = { | |
port: 8080 | |
} | |
parser = OptionParser.new do |opts| | |
opts.banner = "Usage: script.rb [options] domain" | |
opts.on("-p", "--port PORT", "The Local Port to listen connection from") do |port| | |
options[:port] = port | |
end | |
end | |
parser.parse! | |
options[:domain] = ARGV.pop | |
unless options[:domain] then | |
puts parser.help | |
exit | |
end | |
Proxy.new.run options |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment