class Plum::Rack::Server

Attributes

app[R]
config[R]
logger[R]
threadpool[R]

Public Class Methods

new(app, config) click to toggle source
# File lib/plum/rack/server.rb, line 6
def initialize(app, config)
  @config = config
  @state = :null
  @app = config[:debug] ? ::Rack::CommonLogger.new(app) : app
  @logger = Logger.new(config[:log] || $stdout).tap { |l| l.level = config[:debug] ? Logger::DEBUG : Logger::INFO }
  @listeners = config[:listeners].map { |lc| lc[:listener].new(lc) }
  @threadpool = ThreadPool.new(@config[:threadpool_size]) if @config[:threadpool_size] > 1

  @logger.info("Plum #{::Plum::VERSION}")
  @logger.info("Config: #{config}")

  if @config[:user]
    drop_privileges
  end
end

Public Instance Methods

log_exception(e) click to toggle source
# File lib/plum/rack/server.rb, line 53
def log_exception(e)
  @logger.error("#{e.class}: #{e.message}\n#{e.backtrace.map { |b| "\t#{b}" }.join("\n")}")
end
start() click to toggle source
# File lib/plum/rack/server.rb, line 22
def start
  #trap(:INT) { @state = :ee }
  #require "lineprof"
  #Lineprof.profile(//){
  @state = :running
  while @state == :running && !@listeners.empty?
    begin
      if ss = IO.select(@listeners, nil, nil, 2.0)
        ss[0].each { |svr|
          begin
            svr.accept(self)
          rescue Errno::ECONNRESET, Errno::ECONNABORTED # closed
          rescue => e
            log_exception(e)
          end
        }
      end
    rescue Errno::EBADF # closed
    rescue => e
      log_exception(e)
    end
  end
  #}
end
stop() click to toggle source
# File lib/plum/rack/server.rb, line 47
def stop
  @state = :stop
  @listeners.map(&:stop)
  # TODO: gracefully shutdown connections (wait threadpool?)
end

Private Instance Methods

drop_privileges() click to toggle source
# File lib/plum/rack/server.rb, line 58
def drop_privileges
  begin
    user = @config[:user]
    group = @config[:group] || user
    @logger.info "Dropping process privilege to #{user}:#{group}"

    cuid, cgid = Process.euid, Process.egid
    tuid, tgid = Etc.getpwnam(user).uid, Etc.getgrnam(group).gid

    Process.initgroups(user, tgid)
    Process::GID.change_privilege(tgid)
    Process::UID.change_privilege(tuid)
  rescue Errno::EPERM => e
    @ogger.fatal "Could not change privilege: #{e}"
    exit 2
  end
end