class Grack::Server

Constants

F

logic helping functions


PLAIN_TYPE

HTTP error response handling functions


SERVICES

Public Class Methods

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

Public Instance Methods

call(env) click to toggle source
# File lib/grack/server.rb, line 37
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 if !cmd

  @dir = get_git_dir(path)
  return render_not_found if !@dir

  Dir.chdir(@dir) do
    self.method(cmd).call()
  end
end
dumb_info_refs() click to toggle source
# File lib/grack/server.rb, line 98
def dumb_info_refs
  update_server_info
  send_file(@reqfile, "text/plain; charset=utf-8") do
    hdr_nocache
  end
end
get_config_setting(service_name) click to toggle source
# File lib/grack/server.rb, line 217
def get_config_setting(service_name)
  service_name = service_name.gsub('-', '')
  setting = get_git_config("http.#{service_name}")
  if service_name == 'uploadpack'
    return setting != 'false'
  else
    return setting == 'true'
  end
end
get_git_config(config_name) click to toggle source
# File lib/grack/server.rb, line 227
def get_git_config(config_name)
  cmd = git_command("config #{config_name}")
  %x#{cmd}`.chomp
end
get_git_dir(path) click to toggle source
# File lib/grack/server.rb, line 172
def get_git_dir(path)
  root = @config[:project_root] || %xpwd`
  path = File.join(root, path)
  if File.exists?(path) # TODO: check is a valid git directory
    return path
  end
  false
end
get_idx_file() click to toggle source
# File lib/grack/server.rb, line 124
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 105
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 78
def get_info_refs
  service_name = get_service_type

  if has_access(service_name)
    cmd = git_command("#{service_name} --stateless-rpc --advertise-refs .")
    refs = %x#{cmd}`

    @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
  else
    dumb_info_refs
  end
end
get_loose_object() click to toggle source
# File lib/grack/server.rb, line 112
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 118
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 181
def get_service_type
  service_type = @req.params['service']
  return false if !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 130
def get_text_file
  send_file(@reqfile, "text/plain") do
    hdr_nocache
  end
end
git_command(command) click to toggle source
# File lib/grack/server.rb, line 245
def git_command(command)
  git_bin = @config[:git_path] || 'git'
  command = "#{git_bin} #{command}"
  command
end
has_access(rpc, check_content_type = false) click to toggle source
# File lib/grack/server.rb, line 203
def has_access(rpc, check_content_type = false)
  if check_content_type
    return false if @req.content_type != "application/x-git-%s-request" % rpc
  end
  return false if !['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
  return get_config_setting(rpc)
end
hdr_cache_forever() click to toggle source
# File lib/grack/server.rb, line 297
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 291
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 188
def match_routing
  cmd = nil
  path = nil
  SERVICES.each do |method, handler, match, rpc|
    if m = Regexp.new(match).match(@req.path_info)
      return ['not_allowed'] if method != @req.request_method
      cmd = handler
      path = m[1]
      file = @req.path_info.sub(path + '/', '')
      return [cmd, path, file, rpc]
    end
  end
  return nil
end
pkt_flush() click to toggle source

packet-line handling functions


# File lib/grack/server.rb, line 278
def pkt_flush
  '0000'
end
pkt_write(str) click to toggle source
# File lib/grack/server.rb, line 282
def pkt_write(str)
  (str.size + 4).to_s(base=16).rjust(4, '0') + str
end
read_body() click to toggle source
# File lib/grack/server.rb, line 232
def read_body
  if @env["HTTP_CONTENT_ENCODING"] =~ /gzip/
    input = Zlib::GzipReader.new(@req.body).read
  else
    input = @req.body.read
  end
end
render_method_not_allowed() click to toggle source
# File lib/grack/server.rb, line 257
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 269
def render_no_access
  [403, PLAIN_TYPE, ["Forbidden"]]
end
render_not_found() click to toggle source
# File lib/grack/server.rb, line 265
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 143
def send_file(reqfile, content_type)
  reqfile = File.join(@dir, reqfile)
  return render_not_found if !F.exists?(reqfile)

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

  yield

  if size = F.size?(reqfile)
    @res["Content-Length"] = size.to_s
    @res.finish do
      F.open(reqfile, "rb") do |file|
        while part = file.read(8192)
          @res.write part
        end
      end
    end
  else
    body = [F.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

actual command handling functions


# File lib/grack/server.rb, line 58
def service_rpc
  return render_no_access if !has_access(@rpc, true)
  input = read_body

  @res = Rack::Response.new
  @res.status = 200
  @res["Content-Type"] = "application/x-git-%s-result" % @rpc
  @res.finish do
    command = git_command("#{@rpc} --stateless-rpc #{@dir}")
    IO.popen(command, File::RDWR) do |pipe|
      pipe.write(input)
      pipe.close_write
      while !pipe.eof?
        block = pipe.read(8192) # 8M at a time
        @res.write block        # steam it to the client
      end
    end
  end
end
set_config(config) click to toggle source
# File lib/grack/server.rb, line 29
def set_config(config)
  @config = config || {}
end
set_config_setting(key, value) click to toggle source
# File lib/grack/server.rb, line 33
def set_config_setting(key, value)
  @config[key] = value
end
update_server_info() click to toggle source
# File lib/grack/server.rb, line 240
def update_server_info
  cmd = git_command("update-server-info")
  %x#{cmd}`
end