biryani 0.0.1

data/lib/biryani/hpack/huffman.rb

@@ -0,0 +1,309 @@
+module Biryani
+  module HPACK
+    # rubocop: disable Metrics/ModuleLength
+    module Huffman
+      # https://datatracker.ietf.org/doc/html/rfc7541#appendix-B
+      ENCODE_TABLE = %w[
+        1111111111000
+        11111111111111111011000
+        1111111111111111111111100010
+        1111111111111111111111100011
+        1111111111111111111111100100
+        1111111111111111111111100101
+        1111111111111111111111100110
+        1111111111111111111111100111
+        1111111111111111111111101000
+        111111111111111111101010
+        111111111111111111111111111100
+        1111111111111111111111101001
+        1111111111111111111111101010
+        111111111111111111111111111101
+        1111111111111111111111101011
+        1111111111111111111111101100
+        1111111111111111111111101101
+        1111111111111111111111101110
+        1111111111111111111111101111
+        1111111111111111111111110000
+        1111111111111111111111110001
+        1111111111111111111111110010
+        111111111111111111111111111110
+        1111111111111111111111110011
+        1111111111111111111111110100
+        1111111111111111111111110101
+        1111111111111111111111110110
+        1111111111111111111111110111
+        1111111111111111111111111000
+        1111111111111111111111111001
+        1111111111111111111111111010
+        1111111111111111111111111011
+        010100
+        1111111000
+        1111111001
+        111111111010
+        1111111111001
+        010101
+        11111000
+        11111111010
+        1111111010
+        1111111011
+        11111001
+        11111111011
+        11111010
+        010110
+        010111
+        011000
+        00000
+        00001
+        00010
+        011001
+        011010
+        011011
+        011100
+        011101
+        011110
+        011111
+        1011100
+        11111011
+        111111111111100
+        100000
+        111111111011
+        1111111100
+        1111111111010
+        100001
+        1011101
+        1011110
+        1011111
+        1100000
+        1100001
+        1100010
+        1100011
+        1100100
+        1100101
+        1100110
+        1100111
+        1101000
+        1101001
+        1101010
+        1101011
+        1101100
+        1101101
+        1101110
+        1101111
+        1110000
+        1110001
+        1110010
+        11111100
+        1110011
+        11111101
+        1111111111011
+        1111111111111110000
+        1111111111100
+        11111111111100
+        100010
+        111111111111101
+        00011
+        100011
+        00100
+        100100
+        00101
+        100101
+        100110
+        100111
+        00110
+        1110100
+        1110101
+        101000
+        101001
+        101010
+        00111
+        101011
+        1110110
+        101100
+        01000
+        01001
+        101101
+        1110111
+        1111000
+        1111001
+        1111010
+        1111011
+        111111111111110
+        11111111100
+        11111111111101
+        1111111111101
+        1111111111111111111111111100
+        11111111111111100110
+        1111111111111111010010
+        11111111111111100111
+        11111111111111101000
+        1111111111111111010011
+        1111111111111111010100
+        1111111111111111010101
+        11111111111111111011001
+        1111111111111111010110
+        11111111111111111011010
+        11111111111111111011011
+        11111111111111111011100
+        11111111111111111011101
+        11111111111111111011110
+        111111111111111111101011
+        11111111111111111011111
+        111111111111111111101100
+        111111111111111111101101
+        1111111111111111010111
+        11111111111111111100000
+        111111111111111111101110
+        11111111111111111100001
+        11111111111111111100010
+        11111111111111111100011
+        11111111111111111100100
+        111111111111111011100
+        1111111111111111011000
+        11111111111111111100101
+        1111111111111111011001
+        11111111111111111100110
+        11111111111111111100111
+        111111111111111111101111
+        1111111111111111011010
+        111111111111111011101
+        11111111111111101001
+        1111111111111111011011
+        1111111111111111011100
+        11111111111111111101000
+        11111111111111111101001
+        111111111111111011110
+        11111111111111111101010
+        1111111111111111011101
+        1111111111111111011110
+        111111111111111111110000
+        111111111111111011111
+        1111111111111111011111
+        11111111111111111101011
+        11111111111111111101100
+        111111111111111100000
+        111111111111111100001
+        1111111111111111100000
+        111111111111111100010
+        11111111111111111101101
+        1111111111111111100001
+        11111111111111111101110
+        11111111111111111101111
+        11111111111111101010
+        1111111111111111100010
+        1111111111111111100011
+        1111111111111111100100
+        11111111111111111110000
+        1111111111111111100101
+        1111111111111111100110
+        11111111111111111110001
+        11111111111111111111100000
+        11111111111111111111100001
+        11111111111111101011
+        1111111111111110001
+        1111111111111111100111
+        11111111111111111110010
+        1111111111111111101000
+        1111111111111111111101100
+        11111111111111111111100010
+        11111111111111111111100011
+        11111111111111111111100100
+        111111111111111111111011110
+        111111111111111111111011111
+        11111111111111111111100101
+        111111111111111111110001
+        1111111111111111111101101
+        1111111111111110010
+        111111111111111100011
+        11111111111111111111100110
+        111111111111111111111100000
+        111111111111111111111100001
+        11111111111111111111100111
+        111111111111111111111100010
+        111111111111111111110010
+        111111111111111100100
+        111111111111111100101
+        11111111111111111111101000
+        11111111111111111111101001
+        1111111111111111111111111101
+        111111111111111111111100011
+        111111111111111111111100100
+        111111111111111111111100101
+        11111111111111101100
+        111111111111111111110011
+        11111111111111101101
+        111111111111111100110
+        1111111111111111101001
+        111111111111111100111
+        111111111111111101000
+        11111111111111111110011
+        1111111111111111101010
+        1111111111111111101011
+        1111111111111111111101110
+        1111111111111111111101111
+        111111111111111111110100
+        111111111111111111110101
+        11111111111111111111101010
+        11111111111111111110100
+        11111111111111111111101011
+        111111111111111111111100110
+        11111111111111111111101100
+        11111111111111111111101101
+        111111111111111111111100111
+        111111111111111111111101000
+        111111111111111111111101001
+        111111111111111111111101010
+        111111111111111111111101011
+        1111111111111111111111111110
+        111111111111111111111101100
+        111111111111111111111101101
+        111111111111111111111101110
+        111111111111111111111101111
+        111111111111111111111110000
+        11111111111111111111101110
+        111111111111111111111111111111
+      ].freeze
+      DECODE_TABLE = ENCODE_TABLE.each_with_index.to_h.freeze
+      EOS = '1' * 30
+
+      private_constant :ENCODE_TABLE, :DECODE_TABLE, :EOS
+      Ractor.make_shareable(ENCODE_TABLE)
+      Ractor.make_shareable(DECODE_TABLE)
+      Ractor.make_shareable(EOS)
+
+      # @param s [String]
+      #
+      # @return [String]
+      def self.encode(s)
+        bits = s.bytes.map { |sym| ENCODE_TABLE[sym] }.join
+        bits << '1' * ((8 - bits.bytesize) % 8)
+        [bits].pack('B*')
+      end
+
+      Accumulator = Struct.new(:s, :buf)
+
+      # @param encoded [String]
+      #
+      # @raise [HuffmanDecodeError]
+      #
+      # @return [String]
+      def self.decode(encoded)
+        bits = encoded.unpack1('B*').chars
+        res = bits.each_with_object(Accumulator.new(s: '', buf: '')) do |bit, acc|
+          acc.buf << bit
+          raise Error::HuffmanDecodeError if acc.buf == EOS
+
+          if (chr = DECODE_TABLE[acc.buf])
+            acc.s << chr
+            acc.buf.clear
+          end
+
+          acc
+        end
+        raise Error::HuffmanDecodeError if res.buf.chars.any? { |c| c != '1' } || res.buf.bytesize >= 8
+
+        res.s
+      end
+    end
+    # rubocop: enable Metrics/ModuleLength
+  end
+end

data/lib/biryani/hpack/integer.rb

@@ -0,0 +1,66 @@
+module Biryani
+  module HPACK
+    module Integer
+      # if I < 2^N - 1, encode I on N bits
+      # else
+      #     encode (2^N - 1) on N bits
+      #     I = I - (2^N - 1)
+      #     while I >= 128
+      #          encode (I % 128 + 128) on 8 bits
+      #          I = I / 128
+      #     encode I on 8 bits
+      # decode I from the next N bits
+      # if I < 2^N - 1, return I
+      # else
+      #     M = 0
+      #     repeat
+      #         B = next octet
+      #         I = I + (B & 127) * 2^M
+      #         M = M + 7
+      #     while B & 128 == 128
+      #     return I
+      # https://datatracker.ietf.org/doc/html/rfc7541#section-5.1
+      #
+      # @param i [Integer]
+      # @param n [Integer]
+      # @param mask [Integer]
+      #
+      # @return [Array]
+      def self.encode(i, n, mask)
+        limit = (1 << n) - 1
+        return [i | mask].pack('C*') if i < limit
+
+        bytes = [limit | mask]
+        i -= limit
+        while i > 128
+          bytes << i % 128 + 128
+          i /= 128
+        end
+        bytes << i
+        bytes.pack('C*')
+      end
+
+      # @param s [String]
+      # @param cursor [Integer]
+      # @param n [Integer]
+      #
+      # @return [Integer]
+      # @return [Integer]
+      def self.decode(s, n, cursor)
+        limit = (1 << n) - 1
+        h = s.getbyte(cursor)
+        return [h & limit, cursor + 1] if (h & limit) != limit
+
+        res = limit
+        s[cursor + 1..].each_byte.each_with_index.each do |byte, i|
+          res += (byte & 127) * 2**(i * 7)
+          cursor += 1
+
+          break if (byte & 128).zero?
+        end
+
+        [res, cursor + 1]
+      end
+    end
+  end
+end

data/lib/biryani/hpack/option.rb

@@ -0,0 +1,24 @@
+module Biryani
+  module HPACK
+    class Some
+      attr_reader :index, :value
+
+      def initialize(index, value)
+        @index = index
+        @value = value
+      end
+
+      def deconstruct
+        [@index, @value]
+      end
+    end
+
+    class None
+      def initialize; end
+
+      def deconstruct
+        []
+      end
+    end
+  end
+end

data/lib/biryani/hpack/string.rb

@@ -0,0 +1,40 @@
+module Biryani
+  module HPACK
+    module String
+      #   0   1   2   3   4   5   6   7
+      # +---+---+---+---+---+---+---+---+
+      # | H |    String Length (7+)     |
+      # +---+---------------------------+
+      # |  String Data (Length octets)  |
+      # +-------------------------------+
+      # https://datatracker.ietf.org/doc/html/rfc7541#section-5.2
+      #
+      # @param s [String]
+      #
+      # @return [String]
+      def self.encode(s)
+        res = Huffman.encode(s)
+        mask = 0b10000000
+        if res.bytesize > s.bytesize
+          res = s
+          mask = 0b00000000
+        end
+
+        Integer.encode(res.bytesize, 7, mask) + res
+      end
+
+      # @param s [String]
+      # @param cursor [Integer]
+      #
+      # @return [String]
+      # @return [Integer]
+      def self.decode(s, cursor)
+        h = (s.getbyte(cursor) & 0b10000000).positive?
+        len, c = Integer.decode(s, 7, cursor)
+        return [Huffman.decode(s[c...c + len]), c + len] if h
+
+        [s[c...c + len], c + len]
+      end
+    end
+  end
+end

data/lib/biryani/hpack.rb

@@ -0,0 +1,84 @@
+module Biryani
+  module HPACK
+    # https://datatracker.ietf.org/doc/html/rfc7541#appendix-A
+    STATIC_TABLE = [
+      [':authority',                  ''],
+      [':method',                     'GET'],
+      [':method',                     'POST'],
+      [':path',                       '/'],
+      [':path',                       '/index.html'],
+      [':scheme',                     'http'],
+      [':scheme',                     'https'],
+      [':status',                     '200'],
+      [':status',                     '204'],
+      [':status',                     '206'],
+      [':status',                     '304'],
+      [':status',                     '400'],
+      [':status',                     '404'],
+      [':status',                     '500'],
+      ['accept-charset',              ''],
+      ['accept-encoding',             'gzip, deflate'],
+      ['accept-language',             ''],
+      ['accept-ranges',               ''],
+      ['accept',                      ''],
+      ['access-control-allow-origin', ''],
+      ['age',                         ''],
+      ['allow',                       ''],
+      ['authorization',               ''],
+      ['cache-control',               ''],
+      ['content-disposition',         ''],
+      ['content-encoding',            ''],
+      ['content-language',            ''],
+      ['content-length',              ''],
+      ['content-location',            ''],
+      ['content-range',               ''],
+      ['content-type',                ''],
+      ['cookie',                      ''],
+      ['date',                        ''],
+      ['etag',                        ''],
+      ['expect',                      ''],
+      ['expires',                     ''],
+      ['from',                        ''],
+      ['host',                        ''],
+      ['if-match',                    ''],
+      ['if-modified-since',           ''],
+      ['if-none-match',               ''],
+      ['if-range',                    ''],
+      ['if-unmodified-since',         ''],
+      ['last-modified',               ''],
+      ['link',                        ''],
+      ['location',                    ''],
+      ['max-forwards',                ''],
+      ['proxy-authenticate',          ''],
+      ['proxy-authorization',         ''],
+      ['range',                       ''],
+      ['referer',                     ''],
+      ['refresh',                     ''],
+      ['retry-after',                 ''],
+      ['server',                      ''],
+      ['set-cookie',                  ''],
+      ['strict-transport-security',   ''],
+      ['transfer-encoding',           ''],
+      ['user-agent',                  ''],
+      ['vary',                        ''],
+      ['via',                         ''],
+      ['www-authenticate',            '']
+    ].freeze
+    STATIC_TABLE_SIZE = STATIC_TABLE.length
+
+    private_constant :STATIC_TABLE, :STATIC_TABLE_SIZE
+    Ractor.make_shareable(STATIC_TABLE)
+    Ractor.make_shareable(STATIC_TABLE_SIZE)
+  end
+end
+
+require_relative 'hpack/decoder'
+require_relative 'hpack/dynamic_table'
+require_relative 'hpack/encoder'
+require_relative 'hpack/error'
+require_relative 'hpack/field'
+require_relative 'hpack/fields'
+require_relative 'hpack/huffman'
+require_relative 'hpack/integer'
+require_relative 'hpack/option'
+require_relative 'hpack/string'

data/lib/biryani/http_request.rb

@@ -0,0 +1,83 @@
+module Biryani
+  class HTTPRequest
+    attr_accessor :method, :uri, :fields, :content
+
+    # @param method [String]
+    # @param uri [URI]
+    # @param fields [Hash<String, String>]
+    # @param content [String]
+    def initialize(method, uri, fields, content)
+      @method = method
+      @uri = uri
+      @fields = fields
+      @content = content
+    end
+  end
+
+  class HTTPRequestBuilder
+    PSEUDO_HEADER_FIELDS = [':authority', ':method', ':path', ':scheme'].freeze
+    Ractor.make_shareable(PSEUDO_HEADER_FIELDS)
+
+    def initialize
+      @h = {}
+    end
+
+    # @param name [String]
+    # @param value [String]
+    #
+    # @return [nil, ConnectioError]
+    # rubocop: disable Metrics/CyclomaticComplexity
+    # rubocop: disable Metrics/PerceivedComplexity
+    def field(name, value)
+      return ConnectionError.new(ErrorCode::PROTOCOL_ERROR, 'field name has uppercase letter') if name.downcase != name
+      return ConnectionError.new(ErrorCode::PROTOCOL_ERROR, 'unknown pseudo-header field name') if name[0] == ':' && !PSEUDO_HEADER_FIELDS.include?(name)
+      return ConnectionError.new(ErrorCode::PROTOCOL_ERROR, 'appear pseudo-header fields after regular fields') if name[0] == ':' && @h.any? { |name_, _| name_[0] != ':' }
+      return ConnectionError.new(ErrorCode::PROTOCOL_ERROR, 'duplicated pseudo-header fields') if PSEUDO_HEADER_FIELDS.include?(name) && @h.key?(name)
+      return ConnectionError.new(ErrorCode::PROTOCOL_ERROR, "invalid `#{name}` field") if PSEUDO_HEADER_FIELDS.include?(name) && value.empty?
+      return ConnectionError.new(ErrorCode::PROTOCOL_ERROR, 'connection-specific field is forbidden') if name == 'connection'
+
+      # TODO: trailers
+
+      if name == 'cookie' && @h.key?('cookie')
+        @h[name] += "; #{value}"
+      else
+        @h[name] = value
+      end
+
+      nil
+    end
+    # rubocop: enable Metrics/CyclomaticComplexity
+    # rubocop: enable Metrics/PerceivedComplexity
+
+    # @param arr [Array]
+    #
+    # @return [nil, ConnectioError]
+    def fields(arr)
+      arr.each do |name, value|
+        err = field(name, value)
+        return err unless err.nil?
+      end
+
+      nil
+    end
+
+    # @param s [String]
+    #
+    # @return [HTTPRequest]
+    def build(s)
+      self.class.http_request(@h, s)
+    end
+
+    # @param fields [Hash<String, String>]
+    # @param s [String]
+    #
+    # @return [HTTPRequest, ConnectionError]
+    def self.http_request(fields, s)
+      return ConnectionError.new(ErrorCode::PROTOCOL_ERROR, 'missing pseudo-header fields') unless PSEUDO_HEADER_FIELDS.all? { |x| fields.key?(x) }
+      return ConnectionError.new(ErrorCode::PROTOCOL_ERROR, 'invalid content-length') if fields.key?('content-length') && !s.empty? && s.length != fields['content-length'].to_i
+
+      uri = URI("#{fields[':scheme']}://#{fields[':authority']}#{fields[':path']}")
+      HTTPRequest.new(fields[':method'], uri, fields.reject { |name, _| PSEUDO_HEADER_FIELDS.include?(name) }, s)
+    end
+  end
+end

data/lib/biryani/http_response.rb

@@ -0,0 +1,61 @@
+module Biryani
+  class HTTPResponse
+    attr_accessor :status, :fields, :content
+
+    # @param status [Integer]
+    # @param fields [Hash]
+    # @param content [String]
+    def initialize(status, fields, content)
+      @status = status
+      @fields = fields
+      @content = content
+    end
+  end
+
+  class HTTPResponseParser
+    # @param res [HTTPResponse]
+    def initialize(res)
+      @res = res
+    end
+
+    # @return [Array]
+    def fields
+      fields = [[':status', @res.status.to_s]]
+      @res.fields.each do |name, value|
+        # TODO: validate fields
+        fields << [name.to_s.downcase, value.to_s]
+      end
+
+      fields
+    end
+
+    # @return [String]
+    def content
+      @res.content
+    end
+
+    # @param stream_id [Integer]
+    # @param encoder [Encoder]
+    # @param max_frame_size [Integer]
+    #
+    # @return [Array<Object>] frames
+    def parse(stream_id, encoder, max_frame_size)
+      fragment = encoder.encode(fields)
+      len = (fragment.bytesize + max_frame_size - 1) / max_frame_size
+      frames = fragment.gsub(/.{1,#{max_frame_size}}/m).with_index.map do |s, index|
+        if index.zero?
+          Frame::Headers.new(len < 2, content.empty?, stream_id, nil, nil, s, nil)
+        else
+          Frame::Continuation.new(index == len - 1, stream_id, s)
+        end
+      end
+
+      len = (content.bytesize + max_frame_size - 1) / max_frame_size
+      frames += content.gsub(/.{1,#{max_frame_size}}/m).with_index.map do |s, index|
+        Frame::Data.new(index == len - 1, stream_id, s, nil)
+      end
+
+      frames
+    end
+  end
+end

data/lib/biryani/server.rb

@@ -0,0 +1,19 @@
+module Biryani
+  class Server
+    # @param proc [Proc]
+    def initialize(proc)
+      @proc = proc
+    end
+
+    # @param socket [Socket]
+    def run(socket)
+      loop do
+        Ractor.new(socket.accept, @proc) do |io, proc|
+          conn = Connection.new(proc)
+          conn.serve(io)
+          io&.close
+        end
+      end
+    end
+  end
+end