iframe-proxy | Sunbelt Computer Software

199 lines (171 loc) · 4.4 KB
# A very simple Web server written on top of Foundation's run loop and
# dispatching new requests to a pool of threads.
framework 'Foundation'
class HTTPMessage
  attr_accessor :code, :content_type, :content
    str = "HTTP/1.1 #{code} "
    str << code == 404 ? "Not Found\n" : "OK\n"
    str << "Date: #{Time.now}\n"
    str << "Server: #{__FILE__} (MacRuby, Mac OS X)\n"
    str << "Content-Length: #{@content.length}\n"
    str << "Keep-Alive: timeout=5, max=100\n"
    str << "Connection: Keep-Alive\n"
    str << "Content-Type: #{@content_type}"
    if content_type.start_with?('text')
      str << "; charset=utf-8"
    str << "\n\n"
    data = nil
    if @content.is_a?(NSData)
      data = str.dataUsingEncoding(NSUTF8StringEncoding)
      data = data.mutableCopy
      data.appendData(@content)
      str << @content
      data = str.dataUsingEncoding(NSUTF8StringEncoding)
class WebServer
  def initialize(dir, port, debug)
    @debug = debug
    log "Preparing server to serve #{dir} on http://localhost:#{port}"
    @dir = dir
    @port = NSSocketPort.alloc.initWithTCPPort(port)
    unless @port
      raise "Can't open local TCP socket on port #{port}"
    fd = @port.socket
    @handle = NSFileHandle.alloc.initWithFileDescriptor fd,
      closeOnDealloc:true
    NSNotificationCenter.defaultCenter.addObserver self,
      selector:'new_connection:',
      name:NSFileHandleConnectionAcceptedNotification,
      object:nil
    @handle.acceptConnectionInBackgroundAndNotify
  THREADS_POOL = 4
    log "Starting server"
    @threads = []
    @work_queue = []
    @work_mutex = Mutex.new
    THREADS_POOL.times do
      @threads << Thread.new do
        begin
          loop do
            sleep 0.1
            handle = @work_mutex.synchronize { @work_queue.shift }
            if handle
              handle_request(handle)
          end
        rescue => e
          puts "ERROR: #{e}"
        end
    NSRunLoop.currentRunLoop.run
  def new_connection(notification)
    dict = notification.userInfo
    remote_handle = dict[NSFileHandleNotificationFileHandleItem]
    @handle.acceptConnectionInBackgroundAndNotify
    if remote_handle
      @work_mutex.synchronize { @work_queue << remote_handle }
      @threads.each { |t| t.wakeup }
  def handle_request(handle)
     data = handle.availableData
     str = NSString.alloc.initWithData(data, encoding:NSUTF8StringEncoding)
     if str.start_with?('GET ')
       i = str.index(' ', 4)
       path = str[4..(i-1)]
       data = serve_data(path)
       handle.writeData(data)
     handle.closeFile
  def serve_file(msg, path, real_path)
    data = NSData.dataWithContentsOfFile(real_path)
    msg.content = data
    msg.content_type = case File.extname(real_path)
      when '.gif'
        'image/gif'
      when '.png'
        'image/png'
      when '.jpg', '.jpeg'
        'image/jpg'
      when '.html'
        'text/html'
        'text/plain'
  def serve_directory(msg, path, real_path)
    str = <<EOS
<!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 3.2 Final//EN">
    <title>Index of #{path}</title>
  <h1>Index of #{path}</h1>
    Dir.entries(real_path).each do |entry|
      entry_path = File.join(path, entry)
      str << "  <a href=\"#{entry_path}\">#{entry_path}</a>\n"
    str << <<EOS
    msg.content_type = 'text/html'
    msg.content = str
  def serve_data(path)
    real_path = File.join(@dir, path)
    msg = HTTPMessage.new
    if File.exist?(real_path)
      msg.code = 200
      if File.directory?(real_path)
        serve_directory(msg, path, real_path)
        serve_file(msg, path, real_path)
      msg.code = 404
      msg.content_type = 'text/html'
      msg.content = <<EOS
<!DOCTYPE HTML PUBLIC "-//IETF//DTD HTML 2.0//EN">
    <title>404 Not Found</title>
    <h1>Not Found</h1>
    <p>The requested URL #{path} was not found on this server.</p>
    log "#{msg.code} #{real_path}"
    msg.data
  def log(msg)
    puts "#{Thread.current} LOG: #{msg}" if @debug
dir, port, debug = ARGV[0], ARGV[1], ARGV[2]
dir ||= '/'
port ||= 8080
debug ||= false
server = WebServer.new(dir, port, debug)
Sunbelt Computer Software

PL/B Language Development and Support

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Sunbelt Computer Software

PL/B Language Development and Support

FilesExpand file tree

web_server.rb

Latest commit

History

web_server.rb

File metadata and controls
