Sha256: 5f35791e7cb6bbbf73c01a53430876981b72233c58b713b8f5c951c3ae3197d9
Contents?: true
Size: 1.81 KB
Versions: 37
Compression:
Stored size: 1.81 KB
Contents
# Copyright 2016 Matt Wrock <matt@mattwrock.com>
#
# Licensed under the Apache License, Version 2.0 (the 'License');
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an 'AS IS' BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
require_relative 'fragment'
module WinRM
# PowerShell Remoting Protcol module
module PSRP
# PowerShell Remoting Protocol message fragmenter.
class MessageFragmenter
DEFAULT_BLOB_LENGTH = 32_768
def initialize(max_blob_length = DEFAULT_BLOB_LENGTH)
@object_id = 0
@max_blob_length = max_blob_length || DEFAULT_BLOB_LENGTH
end
attr_reader :object_id
attr_accessor :max_blob_length
def fragment(message)
@object_id += 1
message_bytes = message.bytes
bytes_fragmented = 0
fragment_id = 0
fragment = nil
while bytes_fragmented < message_bytes.length
last_byte = bytes_fragmented + max_blob_length
last_byte = message_bytes.length if last_byte > message_bytes.length
fragment = Fragment.new(
object_id,
message.bytes[bytes_fragmented..last_byte - 1],
fragment_id,
bytes_fragmented.zero?,
last_byte == message_bytes.length
)
fragment_id += 1
bytes_fragmented = last_byte
yield fragment if block_given?
end
fragment
end
end
end
end
Version data entries
37 entries across 28 versions & 3 rubygems