metasploit-framework/lib/rex/proto/dcerpc/ndr.rb

74 lines
1.8 KiB
Ruby

# -*- coding: binary -*-
require "rex/text"
module Rex
module Proto
module DCERPC
class NDR
# Provide padding to align the string to the 32bit boundary
def self.align(string)
warn 'should be using Rex::Encoder::NDR'
return "\x00" * ((4 - (string.length & 3)) & 3)
end
# Encode a 4 byte long
# use to encode:
# long element_1;
def self.long(string)
warn 'should be using Rex::Encoder::NDR'
return [string].pack('V')
end
# Encode a 2 byte short
# use to encode:
# short element_1;
def self.short(string)
warn 'should be using Rex::Encoder::NDR'
return [string].pack('v')
end
# Encode a single byte
# use to encode:
# byte element_1;
def self.byte(string)
warn 'should be using Rex::Encoder::NDR'
return [string].pack('C')
end
# Encode a byte array
# use to encode:
# char element_1
def self.UniConformantArray(string)
warn 'should be using Rex::Encoder::NDR'
return long(string.length) + string + align(string)
end
# Encode a string
# use to encode:
# w_char *element_1;
def self.UnicodeConformantVaryingString(string)
warn 'should be using Rex::Encoder::NDR'
string += "\x00" # null pad
return long(string.length) + long(0) + long(string.length) + Rex::Text.to_unicode(string) + align(Rex::Text.to_unicode(string))
end
# Encode a string that is already unicode encoded
# use to encode:
# w_char *element_1;
def self.UnicodeConformantVaryingStringPreBuilt(string)
warn 'should be using Rex::Encoder::NDR'
# if the string len is odd, thats bad!
if string.length % 2 > 0
string += "\x00"
end
len = string.length / 2;
return long(len) + long(0) + long(len) + string + align(string)
end
end
end
end
end