aws-eventstream 1.0.1 → 1.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 2d0f5881812afed9948b7dbbd8a12d35d90a22ea
-  data.tar.gz: b020ea7e1445e00af035c21b282252879e77cd62
+  metadata.gz: 91e0e9ec59b3137e556a697e02b94a1bea99041f
+  data.tar.gz: 53cd60928f4434173f2e01798256d40e21e8e401
 SHA512:
-  metadata.gz: de4844d722291b990106e03b385aabcfc4ff23de66f2ba4251a1ff6069ef050d20d7cfdcb2b37f8c1707f8857e50f0c3e8aac009f28df56ae21822c1d5130a5c
-  data.tar.gz: 45e7a1242f1628933acccff62f45f785127261ec8ef27327cf9fb7642f5551b8d0cb0d728faaa1fa9a952c6f9fedeabf1eee07132cc85be88ca13cbdd21bbaef
+  metadata.gz: 5c4a00f2542e62fadd50744026151d8ad7df6682e020a48e194a570377442e64cd8673ccc4464b552cda952631d4cf457b5df2c4f6b91cc7089b66541bb26bfb
+  data.tar.gz: 4765ce5c6e944b58a44bb50e3921601932d2c9a6ac9eb33c8edb2f86ad1b031cb8dfaf88ef73303073931093122101720e8faceb2b616f2515395e3459239aac

data/lib/aws-eventstream/encoder.rb

@@ -39,6 +39,12 @@ module Aws
       # and 4 bytes total message crc checksum
       OVERHEAD_LENGTH = 16
 
+      # Maximum header length allowed (after encode) 128kb
+      MAX_HEADERS_LENGTH = 131072
+
+      # Maximum payload length allowed (after encode) 16mb
+      MAX_PAYLOAD_LENGTH = 16777216
+
       # Encodes Aws::EventStream::Message to output IO when
       #   provided, else return the encoded binary string
       #
@@ -62,13 +68,20 @@ module Aws
         end
       end
 
-      private
-
+      # Encodes an Aws::EventStream::Message
+      #   into Aws::EventStream::BytesBuffer
+      #
+      # @param [Aws::EventStream::Message] msg
+      #
+      # @return [Aws::EventStream::BytesBuffer]
       def encode_message(message)
         # create context buffer with encode headers
         ctx_buffer = encode_headers(message)
         headers_len = ctx_buffer.bytesize
         # encode payload
+        if message.payload.length > MAX_PAYLOAD_LENGTH
+          raise Aws::EventStream::Errors::EventPayloadLengthExceedError.new
+        end
         ctx_buffer << message.payload.read
         total_len = ctx_buffer.bytesize + OVERHEAD_LENGTH
 
@@ -85,6 +98,12 @@ module Aws
         buffer
       end
 
+      # Encodes headers part of an Aws::EventStream::Message
+      #   into Aws::EventStream::BytesBuffer
+      #
+      # @param [Aws::EventStream::Message] msg
+      #
+      # @return [Aws::EventStream::BytesBuffer]
       def encode_headers(msg)
         buffer = BytesBuffer.new('')
         msg.headers.each do |k, v|
@@ -101,9 +120,14 @@ module Aws
           pattern ? buffer << [v.value].pack(pattern) :
             buffer << v.value
         end
+        if buffer.bytesize > MAX_HEADERS_LENGTH
+          raise Aws::EventStream::Errors::EventHeadersLengthExceedError.new
+        end
         buffer
       end
 
+      private
+
       def prelude(total_len, headers_len)
         BytesBuffer.new(pack_uint32([
           total_len,

data/lib/aws-eventstream/errors.rb

@@ -30,6 +30,18 @@ module Aws
         end
       end
 
+      class EventPayloadLengthExceedError < RuntimeError
+        def initialize(*args)
+          super("Payload length of a message should be under 16mb.")
+        end
+      end
+
+      class EventHeadersLengthExceedError < RuntimeError
+        def initialize(*args)
+          super("Encoded headers length of a message should be under 128kb.")
+        end
+      end
+
     end
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: aws-eventstream
 version: !ruby/object:Gem::Version
-  version: 1.0.1
+  version: 1.0.2
 platform: ruby
 authors:
 - Amazon Web Services
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2018-06-15 00:00:00.000000000 Z
+date: 2019-03-11 00:00:00.000000000 Z
 dependencies: []
 description: Amazon Web Services event stream library. Decodes and encodes binary
   stream under `vnd.amazon.event-stream` content-type