class Grack::Server

Constants

CRLF

Uses chunked (streaming) transfer, otherwise response blocks to calculate Content-Length header en.wikipedia.org/wiki/Chunked_transfer_encoding

PLAIN_TYPE

HTTP error response handling functions


SERVICES

Attributes

git[R]

Public Class Methods

new(config = false) click to toggle source
# File lib/grack/server.rb, line 28
def initialize(config = false)
  set_config(config)
end

Public Instance Methods

_call(env) click to toggle source
# File lib/grack/server.rb, line 44
def _call(env)
  @env = env
  @req = Rack::Request.new(env)

  cmd, path, @reqfile, @rpc = match_routing

  return render_method_not_allowed if cmd == 'not_allowed'
  return render_not_found unless cmd

  @git = get_git(path)
  return render_not_found unless git.valid_repo?

  self.method(cmd).call
end
call(env) click to toggle source
# File lib/grack/server.rb, line 40
def call(env)
  dup._call(env)
end
dumb_info_refs() click to toggle source
# File lib/grack/server.rb, line 121
def dumb_info_refs
  git.update_server_info
  send_file(@reqfile, "text/plain; charset=utf-8") do
    hdr_nocache
  end
end
encode_chunk(chunk) click to toggle source
# File lib/grack/server.rb, line 94
def encode_chunk(chunk)
  size_in_hex = chunk.size.to_s(16)
  [size_in_hex, CRLF, chunk, CRLF].join
end
get_git(path) click to toggle source
# File lib/grack/server.rb, line 197
def get_git(path)
  root = @config[:project_root] || Dir.pwd
  path = File.join(root, path)
  Grack::Git.new(@config[:git_path], path)
end
get_idx_file() click to toggle source
# File lib/grack/server.rb, line 147
def get_idx_file
  send_file(@reqfile, "application/x-git-packed-objects-toc") do
    hdr_cache_forever
  end
end
get_info_packs() click to toggle source
# File lib/grack/server.rb, line 128
def get_info_packs
  # objects/info/packs
  send_file(@reqfile, "text/plain; charset=utf-8") do
    hdr_nocache
  end
end
get_info_refs() click to toggle source
# File lib/grack/server.rb, line 103
def get_info_refs
  service_name = get_service_type
  return dumb_info_refs unless has_access?(service_name)

  refs = git.execute([service_name, '--stateless-rpc', '--advertise-refs', git.repo])

  @res = Rack::Response.new
  @res.status = 200
  @res["Content-Type"] = "application/x-git-%s-advertisement" % service_name
  hdr_nocache

  @res.write(pkt_write("# service=git-#{service_name}\n"))
  @res.write(pkt_flush)
  @res.write(refs)

  @res.finish
end
get_loose_object() click to toggle source
# File lib/grack/server.rb, line 135
def get_loose_object
  send_file(@reqfile, "application/x-git-loose-object") do
    hdr_cache_forever
  end
end
get_pack_file() click to toggle source
# File lib/grack/server.rb, line 141
def get_pack_file
  send_file(@reqfile, "application/x-git-packed-objects") do
    hdr_cache_forever
  end
end
get_service_type() click to toggle source
# File lib/grack/server.rb, line 203
def get_service_type
  service_type = @req.params['service']
  return false unless service_type
  return false if service_type[0, 4] != 'git-'
  service_type.gsub('git-', '')
end
get_text_file() click to toggle source
# File lib/grack/server.rb, line 153
def get_text_file
  send_file(@reqfile, "text/plain") do
    hdr_nocache
  end
end
has_access?(rpc, check_content_type = false) click to toggle source
# File lib/grack/server.rb, line 229
def has_access?(rpc, check_content_type = false)
  if check_content_type
    conten_type = "application/x-git-%s-request" % rpc
    return false unless @req.content_type == conten_type
  end

  return false unless ['upload-pack', 'receive-pack'].include?(rpc)

  if rpc == 'receive-pack'
    return @config[:receive_pack] if @config.include?(:receive_pack)
  end

  if rpc == 'upload-pack'
    return @config[:upload_pack] if @config.include?(:upload_pack)
  end

  git.config_setting(rpc)
end
hdr_cache_forever() click to toggle source
# File lib/grack/server.rb, line 301
def hdr_cache_forever
  now = Time.now().to_i
  @res["Date"] = now.to_s
  @res["Expires"] = (now + 31536000).to_s;
  @res["Cache-Control"] = "public, max-age=31536000";
end
hdr_nocache() click to toggle source

header writing functions


# File lib/grack/server.rb, line 295
def hdr_nocache
  @res["Expires"] = "Fri, 01 Jan 1980 00:00:00 GMT"
  @res["Pragma"] = "no-cache"
  @res["Cache-Control"] = "no-cache, max-age=0, must-revalidate"
end
match_routing() click to toggle source
# File lib/grack/server.rb, line 210
def match_routing
  cmd = nil
  path = nil

  SERVICES.each do |method, handler, match, rpc|
    next unless m = Regexp.new(match).match(@req.path_info)

    return ['not_allowed'] unless method == @req.request_method

    cmd = handler
    path = m[1]
    file = @req.path_info.sub(path + '/', '')

    return [cmd, path, file, rpc]
  end
  
  nil
end
pkt_flush() click to toggle source

packet-line handling functions


# File lib/grack/server.rb, line 283
def pkt_flush
  '0000'
end
pkt_write(str) click to toggle source
# File lib/grack/server.rb, line 287
def pkt_write(str)
  (str.size + 4).to_s(16).rjust(4, '0') + str
end
read_body() click to toggle source
# File lib/grack/server.rb, line 248
def read_body
  if @env["HTTP_CONTENT_ENCODING"] =~ /gzip/
    Zlib::GzipReader.new(@req.body).read
  else
    @req.body.read
  end
end
render_method_not_allowed() click to toggle source
# File lib/grack/server.rb, line 262
def render_method_not_allowed
  if @env['SERVER_PROTOCOL'] == "HTTP/1.1"
    [405, PLAIN_TYPE, ["Method Not Allowed"]]
  else
    [400, PLAIN_TYPE, ["Bad Request"]]
  end
end
render_no_access() click to toggle source
# File lib/grack/server.rb, line 274
def render_no_access
  [403, PLAIN_TYPE, ["Forbidden"]]
end
render_not_found() click to toggle source
# File lib/grack/server.rb, line 270
def render_not_found
  [404, PLAIN_TYPE, ["Not Found"]]
end
send_file(reqfile, content_type) { || ... } click to toggle source

some of this borrowed from the Rack::File implementation

# File lib/grack/server.rb, line 164
def send_file(reqfile, content_type)
  reqfile = File.join(git.repo, reqfile)
  return render_not_found unless File.exists?(reqfile)

  return render_not_found unless reqfile == File.realpath(reqfile)

  # reqfile looks legit: no path traversal, no leading '|'

  @res = Rack::Response.new
  @res.status = 200
  @res["Content-Type"]  = content_type
  @res["Last-Modified"] = File.mtime(reqfile).httpdate

  yield

  if size = File.size?(reqfile)
    @res["Content-Length"] = size.to_s
    @res.finish do
      File.open(reqfile, "rb") do |file|
        while part = file.read(8192)
          @res.write part
        end
      end
    end
  else
    body = [File.read(reqfile)]
    size = Rack::Utils.bytesize(body.first)
    @res["Content-Length"] = size
    @res.write body
    @res.finish
  end
end
service_rpc() click to toggle source
# File lib/grack/server.rb, line 69
def service_rpc
  return render_no_access unless has_access?(@rpc, true)

  input = read_body

  @res = Rack::Response.new
  @res.status = 200
  @res["Content-Type"] = "application/x-git-%s-result" % @rpc
  @res["Transfer-Encoding"] = "chunked"
  @res["Cache-Control"] = "no-cache"

  @res.finish do
    git.execute([@rpc, '--stateless-rpc', git.repo]) do |pipe|
      pipe.write(input)
      pipe.close_write

      while block = pipe.read(8192)     # 8KB at a time
        @res.write encode_chunk(block)  # stream it to the client
      end

      @res.write terminating_chunk
    end
  end
end
set_config(config) click to toggle source
# File lib/grack/server.rb, line 32
def set_config(config)
  @config = config || {}
end
set_config_setting(key, value) click to toggle source
# File lib/grack/server.rb, line 36
def set_config_setting(key, value)
  @config[key] = value
end
terminating_chunk() click to toggle source
# File lib/grack/server.rb, line 99
def terminating_chunk
  [0, CRLF, CRLF].join
end