gitlab-puma 4.3.1.gitlab.2

data/lib/puma/const.rb

@@ -0,0 +1,242 @@
+#encoding: utf-8
+# frozen_string_literal: true
+
+module Puma
+  class UnsupportedOption < RuntimeError
+  end
+
+
+  # Every standard HTTP code mapped to the appropriate message.  These are
+  # used so frequently that they are placed directly in Puma for easy
+  # access rather than Puma::Const itself.
+
+  # Every standard HTTP code mapped to the appropriate message.
+  # Generated with:
+  # curl -s https://www.iana.org/assignments/http-status-codes/http-status-codes-1.csv | \
+  #   ruby -ne 'm = /^(\d{3}),(?!Unassigned|\(Unused\))([^,]+)/.match($_) and \
+  #             puts "#{m[1]} => \x27#{m[2].strip}\x27,"'
+  HTTP_STATUS_CODES = {
+    100 => 'Continue',
+    101 => 'Switching Protocols',
+    102 => 'Processing',
+    200 => 'OK',
+    201 => 'Created',
+    202 => 'Accepted',
+    203 => 'Non-Authoritative Information',
+    204 => 'No Content',
+    205 => 'Reset Content',
+    206 => 'Partial Content',
+    207 => 'Multi-Status',
+    208 => 'Already Reported',
+    226 => 'IM Used',
+    300 => 'Multiple Choices',
+    301 => 'Moved Permanently',
+    302 => 'Found',
+    303 => 'See Other',
+    304 => 'Not Modified',
+    305 => 'Use Proxy',
+    307 => 'Temporary Redirect',
+    308 => 'Permanent Redirect',
+    400 => 'Bad Request',
+    401 => 'Unauthorized',
+    402 => 'Payment Required',
+    403 => 'Forbidden',
+    404 => 'Not Found',
+    405 => 'Method Not Allowed',
+    406 => 'Not Acceptable',
+    407 => 'Proxy Authentication Required',
+    408 => 'Request Timeout',
+    409 => 'Conflict',
+    410 => 'Gone',
+    411 => 'Length Required',
+    412 => 'Precondition Failed',
+    413 => 'Payload Too Large',
+    414 => 'URI Too Long',
+    415 => 'Unsupported Media Type',
+    416 => 'Range Not Satisfiable',
+    417 => 'Expectation Failed',
+    418 => 'I\'m A Teapot',
+    421 => 'Misdirected Request',
+    422 => 'Unprocessable Entity',
+    423 => 'Locked',
+    424 => 'Failed Dependency',
+    426 => 'Upgrade Required',
+    428 => 'Precondition Required',
+    429 => 'Too Many Requests',
+    431 => 'Request Header Fields Too Large',
+    451 => 'Unavailable For Legal Reasons',
+    500 => 'Internal Server Error',
+    501 => 'Not Implemented',
+    502 => 'Bad Gateway',
+    503 => 'Service Unavailable',
+    504 => 'Gateway Timeout',
+    505 => 'HTTP Version Not Supported',
+    506 => 'Variant Also Negotiates',
+    507 => 'Insufficient Storage',
+    508 => 'Loop Detected',
+    510 => 'Not Extended',
+    511 => 'Network Authentication Required'
+  }
+
+  # For some HTTP status codes the client only expects headers.
+  #
+
+  STATUS_WITH_NO_ENTITY_BODY = {
+    204 => true,
+    205 => true,
+    304 => true
+  }
+
+  # Frequently used constants when constructing requests or responses.  Many times
+  # the constant just refers to a string with the same contents.  Using these constants
+  # gave about a 3% to 10% performance improvement over using the strings directly.
+  #
+  # The constants are frozen because Hash#[]= when called with a String key dups
+  # the String UNLESS the String is frozen. This saves us therefore 2 object
+  # allocations when creating the env hash later.
+  #
+  # While Puma does try to emulate the CGI/1.2 protocol, it does not use the REMOTE_IDENT,
+  # REMOTE_USER, or REMOTE_HOST parameters since those are either a security problem or
+  # too taxing on performance.
+  module Const
+
+    PUMA_VERSION = VERSION = "4.3.1.gitlab.2".freeze
+    CODE_NAME = "Mysterious Traveller".freeze
+    PUMA_SERVER_STRING = ['puma', PUMA_VERSION, CODE_NAME].join(' ').freeze
+
+    FAST_TRACK_KA_TIMEOUT = 0.2
+
+    # The default number of seconds for another request within a persistent
+    # session.
+    PERSISTENT_TIMEOUT = 20
+
+    # The default number of seconds to wait until we get the first data
+    # for the request
+    FIRST_DATA_TIMEOUT = 30
+
+    # How long to wait when getting some write blocking on the socket when
+    # sending data back
+    WRITE_TIMEOUT = 10
+
+    # How many requests to attempt inline before sending a client back to
+    # the reactor to be subject to normal ordering. The idea here is that
+    # we amortize the cost of going back to the reactor for a well behaved
+    # but very "greedy" client across 10 requests. This prevents a not
+    # well behaved client from monopolizing the thread forever.
+    MAX_FAST_INLINE = 10
+
+    # The original URI requested by the client.
+    REQUEST_URI= 'REQUEST_URI'.freeze
+    REQUEST_PATH = 'REQUEST_PATH'.freeze
+    QUERY_STRING = 'QUERY_STRING'.freeze
+    CONTENT_LENGTH = "CONTENT_LENGTH".freeze
+
+    PATH_INFO = 'PATH_INFO'.freeze
+
+    PUMA_TMP_BASE = "puma".freeze
+
+    ERROR_RESPONSE = {
+      # Indicate that we couldn't parse the request
+      400 => "HTTP/1.1 400 Bad Request\r\n\r\n".freeze,
+      # The standard empty 404 response for bad requests.  Use Error4040Handler for custom stuff.
+      404 => "HTTP/1.1 404 Not Found\r\nConnection: close\r\nServer: Puma #{PUMA_VERSION}\r\n\r\nNOT FOUND".freeze,
+      # The standard empty 408 response for requests that timed out.
+      408 => "HTTP/1.1 408 Request Timeout\r\nConnection: close\r\nServer: Puma #{PUMA_VERSION}\r\n\r\n".freeze,
+      # Indicate that there was an internal error, obviously.
+      500 => "HTTP/1.1 500 Internal Server Error\r\n\r\n".freeze,
+      # A common header for indicating the server is too busy.  Not used yet.
+      503 => "HTTP/1.1 503 Service Unavailable\r\n\r\nBUSY".freeze
+    }
+
+    # The basic max request size we'll try to read.
+    CHUNK_SIZE = 16 * 1024
+
+    # This is the maximum header that is allowed before a client is booted.  The parser detects
+    # this, but we'd also like to do this as well.
+    MAX_HEADER = 1024 * (80 + 32)
+
+    # Maximum request body size before it is moved out of memory and into a tempfile for reading.
+    MAX_BODY = MAX_HEADER
+
+    REQUEST_METHOD = "REQUEST_METHOD".freeze
+    HEAD = "HEAD".freeze
+    # ETag is based on the apache standard of hex mtime-size-inode (inode is 0 on win32)
+    LINE_END = "\r\n".freeze
+    REMOTE_ADDR = "REMOTE_ADDR".freeze
+    HTTP_X_FORWARDED_FOR = "HTTP_X_FORWARDED_FOR".freeze
+    HTTP_X_FORWARDED_SSL = "HTTP_X_FORWARDED_SSL".freeze
+    HTTP_X_FORWARDED_SCHEME = "HTTP_X_FORWARDED_SCHEME".freeze
+    HTTP_X_FORWARDED_PROTO = "HTTP_X_FORWARDED_PROTO".freeze
+
+    SERVER_NAME = "SERVER_NAME".freeze
+    SERVER_PORT = "SERVER_PORT".freeze
+    HTTP_HOST = "HTTP_HOST".freeze
+    PORT_80 = "80".freeze
+    PORT_443 = "443".freeze
+    LOCALHOST = "localhost".freeze
+    LOCALHOST_IP = "127.0.0.1".freeze
+    LOCALHOST_ADDR = "127.0.0.1:0".freeze
+
+    SERVER_PROTOCOL = "SERVER_PROTOCOL".freeze
+    HTTP_11 = "HTTP/1.1".freeze
+
+    SERVER_SOFTWARE = "SERVER_SOFTWARE".freeze
+    GATEWAY_INTERFACE = "GATEWAY_INTERFACE".freeze
+    CGI_VER = "CGI/1.2".freeze
+
+    STOP_COMMAND = "?".freeze
+    HALT_COMMAND = "!".freeze
+    RESTART_COMMAND = "R".freeze
+
+    RACK_INPUT = "rack.input".freeze
+    RACK_URL_SCHEME = "rack.url_scheme".freeze
+    RACK_AFTER_REPLY = "rack.after_reply".freeze
+    PUMA_SOCKET = "puma.socket".freeze
+    PUMA_CONFIG = "puma.config".freeze
+    PUMA_PEERCERT = "puma.peercert".freeze
+
+    HTTP = "http".freeze
+    HTTPS = "https".freeze
+
+    HTTPS_KEY = "HTTPS".freeze
+
+    HTTP_VERSION = "HTTP_VERSION".freeze
+    HTTP_CONNECTION = "HTTP_CONNECTION".freeze
+    HTTP_EXPECT = "HTTP_EXPECT".freeze
+    CONTINUE = "100-continue".freeze
+
+    HTTP_11_100 = "HTTP/1.1 100 Continue\r\n\r\n".freeze
+    HTTP_11_200 = "HTTP/1.1 200 OK\r\n".freeze
+    HTTP_10_200 = "HTTP/1.0 200 OK\r\n".freeze
+
+    CLOSE = "close".freeze
+    KEEP_ALIVE = "keep-alive".freeze
+
+    CONTENT_LENGTH2 = "content-length".freeze
+    CONTENT_LENGTH_S = "Content-Length: ".freeze
+    TRANSFER_ENCODING = "transfer-encoding".freeze
+    TRANSFER_ENCODING2 = "HTTP_TRANSFER_ENCODING".freeze
+
+    CONNECTION_CLOSE = "Connection: close\r\n".freeze
+    CONNECTION_KEEP_ALIVE = "Connection: Keep-Alive\r\n".freeze
+
+    TRANSFER_ENCODING_CHUNKED = "Transfer-Encoding: chunked\r\n".freeze
+    CLOSE_CHUNKED = "0\r\n\r\n".freeze
+
+    CHUNKED = "chunked".freeze
+
+    COLON = ": ".freeze
+
+    NEWLINE = "\n".freeze
+
+    HIJACK_P = "rack.hijack?".freeze
+    HIJACK = "rack.hijack".freeze
+    HIJACK_IO = "rack.hijack_io".freeze
+
+    EARLY_HINTS = "rack.early_hints".freeze
+
+    # Mininum interval to checks worker health
+    WORKER_CHECK_INTERVAL = 5
+
+  end
+end

data/lib/puma/control_cli.rb

@@ -0,0 +1,289 @@
+# frozen_string_literal: true
+
+require 'optparse'
+require_relative 'state_file'
+require_relative 'const'
+require_relative 'detect'
+require_relative 'configuration'
+require 'uri'
+require 'socket'
+
+module Puma
+  class ControlCLI
+
+    COMMANDS = %w{halt restart phased-restart start stats status stop reload-worker-directory gc gc-stats}
+
+    def initialize(argv, stdout=STDOUT, stderr=STDERR)
+      @state = nil
+      @quiet = false
+      @pidfile = nil
+      @pid = nil
+      @control_url = nil
+      @control_auth_token = nil
+      @config_file = nil
+      @command = nil
+      @environment = ENV['RACK_ENV']
+
+      @argv = argv.dup
+      @stdout = stdout
+      @stderr = stderr
+      @cli_options = {}
+
+      opts = OptionParser.new do |o|
+        o.banner = "Usage: pumactl (-p PID | -P pidfile | -S status_file | -C url -T token | -F config.rb) (#{COMMANDS.join("|")})"
+
+        o.on "-S", "--state PATH", "Where the state file to use is" do |arg|
+          @state = arg
+        end
+
+        o.on "-Q", "--quiet", "Not display messages" do |arg|
+          @quiet = true
+        end
+
+        o.on "-P", "--pidfile PATH", "Pid file" do |arg|
+          @pidfile = arg
+        end
+
+        o.on "-p", "--pid PID", "Pid" do |arg|
+          @pid = arg.to_i
+        end
+
+        o.on "-C", "--control-url URL", "The bind url to use for the control server" do |arg|
+          @control_url = arg
+        end
+
+        o.on "-T", "--control-token TOKEN", "The token to use as authentication for the control server" do |arg|
+          @control_auth_token = arg
+        end
+
+        o.on "-F", "--config-file PATH", "Puma config script" do |arg|
+          @config_file = arg
+        end
+
+        o.on "-e", "--environment ENVIRONMENT",
+          "The environment to run the Rack app on (default development)" do |arg|
+          @environment = arg
+        end
+
+        o.on_tail("-H", "--help", "Show this message") do
+          @stdout.puts o
+          exit
+        end
+
+        o.on_tail("-V", "--version", "Show version") do
+          puts Const::PUMA_VERSION
+          exit
+        end
+      end
+
+      opts.order!(argv) { |a| opts.terminate a }
+      opts.parse!
+
+      @command = argv.shift
+
+      unless @config_file == '-'
+        environment = @environment || 'development'
+
+        if @config_file.nil?
+          @config_file = %W(config/puma/#{environment}.rb config/puma.rb).find do |f|
+            File.exist?(f)
+          end
+        end
+
+        if @config_file
+          config = Puma::Configuration.new({ config_files: [@config_file] }, {})
+          config.load
+          @state              ||= config.options[:state]
+          @control_url        ||= config.options[:control_url]
+          @control_auth_token ||= config.options[:control_auth_token]
+          @pidfile            ||= config.options[:pidfile]
+        end
+      end
+
+      # check present of command
+      unless @command
+        raise "Available commands: #{COMMANDS.join(", ")}"
+      end
+
+      unless COMMANDS.include? @command
+        raise "Invalid command: #{@command}"
+      end
+
+    rescue => e
+      @stdout.puts e.message
+      exit 1
+    end
+
+    def message(msg)
+      @stdout.puts msg unless @quiet
+    end
+
+    def prepare_configuration
+      if @state
+        unless File.exist? @state
+          raise "State file not found: #{@state}"
+        end
+
+        sf = Puma::StateFile.new
+        sf.load @state
+
+        @control_url = sf.control_url
+        @control_auth_token = sf.control_auth_token
+        @pid = sf.pid
+      elsif @pidfile
+        # get pid from pid_file
+        File.open(@pidfile) { |f| @pid = f.read.to_i }
+      end
+    end
+
+    def send_request
+      uri = URI.parse @control_url
+
+      # create server object by scheme
+      server = case uri.scheme
+                when "ssl"
+                  require 'openssl'
+                  OpenSSL::SSL::SSLSocket.new(
+                    TCPSocket.new(uri.host, uri.port),
+                    OpenSSL::SSL::SSLContext.new
+                  ).tap(&:connect)
+                when "tcp"
+                  TCPSocket.new uri.host, uri.port
+                when "unix"
+                  UNIXSocket.new "#{uri.host}#{uri.path}"
+                else
+                  raise "Invalid scheme: #{uri.scheme}"
+                end
+
+      if @command == "status"
+        message "Puma is started"
+      else
+        url = "/#{@command}"
+
+        if @control_auth_token
+          url = url + "?token=#{@control_auth_token}"
+        end
+
+        server << "GET #{url} HTTP/1.0\r\n\r\n"
+
+        unless data = server.read
+          raise "Server closed connection before responding"
+        end
+
+        response = data.split("\r\n")
+
+        if response.empty?
+          raise "Server sent empty response"
+        end
+
+        (@http,@code,@message) = response.first.split(" ",3)
+
+        if @code == "403"
+          raise "Unauthorized access to server (wrong auth token)"
+        elsif @code == "404"
+          raise "Command error: #{response.last}"
+        elsif @code != "200"
+          raise "Bad response from server: #{@code}"
+        end
+
+        message "Command #{@command} sent success"
+        message response.last if @command == "stats" || @command == "gc-stats"
+      end
+    ensure
+      server.close if server && !server.closed?
+    end
+
+    def send_signal
+      unless @pid
+        raise "Neither pid nor control url available"
+      end
+
+      begin
+
+        case @command
+        when "restart"
+          Process.kill "SIGUSR2", @pid
+
+        when "halt"
+          Process.kill "QUIT", @pid
+
+        when "stop"
+          Process.kill "SIGTERM", @pid
+
+        when "stats"
+          puts "Stats not available via pid only"
+          return
+
+        when "reload-worker-directory"
+          puts "reload-worker-directory not available via pid only"
+          return
+
+        when "phased-restart"
+          Process.kill "SIGUSR1", @pid
+
+        when "status"
+          begin
+            Process.kill 0, @pid
+            puts "Puma is started"
+          rescue Errno::ESRCH
+            raise "Puma is not running"
+          end
+
+          return
+
+        else
+          return
+        end
+
+      rescue SystemCallError
+        if @command == "restart"
+          start
+        else
+          raise "No pid '#{@pid}' found"
+        end
+      end
+
+      message "Command #{@command} sent success"
+    end
+
+    def run
+      return start if @command == "start"
+
+      prepare_configuration
+
+      if Puma.windows?
+        send_request
+      else
+        @control_url ? send_request : send_signal
+      end
+
+    rescue => e
+      message e.message
+      exit 1
+    end
+
+  private
+    def start
+      require 'puma/cli'
+
+      run_args = []
+
+      run_args += ["-S", @state]  if @state
+      run_args += ["-q"] if @quiet
+      run_args += ["--pidfile", @pidfile] if @pidfile
+      run_args += ["--control-url", @control_url] if @control_url
+      run_args += ["--control-token", @control_auth_token] if @control_auth_token
+      run_args += ["-C", @config_file] if @config_file
+      run_args += ["-e", @environment] if @environment
+
+      events = Puma::Events.new @stdout, @stderr
+
+      # replace $0 because puma use it to generate restart command
+      puma_cmd = $0.gsub(/pumactl$/, 'puma')
+      $0 = puma_cmd if File.exist?(puma_cmd)
+
+      cli = Puma::CLI.new run_args, events
+      cli.run
+    end
+  end
+end