cmogstored.git  about / heads / tags
alternative mogstored implementation for MogileFS
blob 9217661a0ed719730399b3d8cc1f1d67eebc3e2d 2630 bytes (raw)
$ git show HEAD:test/http_put6_fail.rb	# shows this blob on the CLI

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
 
#!/usr/bin/env ruby
# -*- encoding: binary -*-
# Copyright (C) 2012-2020 all contributors <cmogstored-public@yhbt.net>
# License: GPL-3.0+ <https://www.gnu.org/licenses/gpl-3.0.txt>
require 'test/test_helper'
require 'net/http'
require 'stringio'

class TestHTTPPut6Fail < Test::Unit::TestCase
  def setup
    @skip = true
    @tmpdir = Dir.mktmpdir('cmogstored-httpput6-test')
    Dir.mkdir("#@tmpdir/dev666")
    @to_close = []
    @host = TEST_HOST6
    srv = TCPServer.new(@host, 0)
    @port = srv.addr[1]
    srv.close
    @err = Tempfile.new("stderr")
    cmd = [ "cmogstored", "--docroot=#@tmpdir", "--httplisten=[#@host]:#@port",
            "--maxconns=500" ]
    vg = ENV["VALGRIND"] and cmd = vg.split(/\s+/).concat(cmd)
    @pid = fork {
      $stderr.reopen(@err)
      @err.close
      exec(*cmd)
    }
    @client = get_client
    @skip = false
  rescue Errno::EAFNOSUPPORT
    # host does not have IPv6
  end

  def teardown
    return if @skip
    Process.kill(:QUIT, @pid) rescue nil
    _, status = Process.waitpid2(@pid)
    @to_close.each { |io| io.close unless io.closed? }
    @err.rewind
    assert status.success?, status.inspect
  ensure
    FileUtils.rm_rf(@tmpdir)
  end

  def test_put_premature_eof
    return if @skip
    path = "/dev666/foo"
    url = "http://[#@host]:#@port#{path}"
    req = "PUT #{url} HTTP/1.1\r\n" \
          "Host: [#@host]:#@port\r\n" \
          "Content-Length: 666\r\n" \
          "\r\n" \
          "abcde"
    @client.write(req)
    @client.shutdown(Socket::SHUT_WR)
    addr = Regexp.escape("[#{@client.addr[3]}]:#{@client.addr[1]}")
    assert_nil @client.read(1)
    assert ! File.exist?("#@tmpdir#{path}")
    buf = File.read(@err.path)
    assert_match(%r{PUT #{path} failed from #{addr} after 5 bytes:}, buf)
    if RUBY_PLATFORM =~ /linux/
      assert_match(%r{last_data_recv=\d+ms from #{addr} for PUT #{path}}, buf)
    end
  end

  def test_put_premature_eof_chunked
    return if @skip
    path = "/dev666/foo"
    url = "http://[#@host]:#@port#{path}"
    req = "PUT #{url} HTTP/1.1\r\n" \
          "Host: [#@host]:#@port\r\n" \
          "Transfer-Encoding: chunked\r\n" \
          "\r\n" \
          "666\r\nf"
    @client.write(req)
    @client.shutdown(Socket::SHUT_WR)
    addr = Regexp.escape("[#{@client.addr[3]}]:#{@client.addr[1]}")
    assert_nil @client.read(1)
    assert ! File.exist?("#@tmpdir#{path}")
    buf = File.read(@err.path)
    assert_match(%r{PUT #{path} failed from #{addr} after 1 bytes:}, buf)
    if RUBY_PLATFORM =~ /linux/
      assert_match(%r{last_data_recv=\d+ms from #{addr} for PUT #{path}}, buf)
    end
  end
end

git clone https://yhbt.net/cmogstored.git