58 lines
978 B
Ruby
58 lines
978 B
Ruby
# -*- coding: binary -*-
|
|
|
|
module Rex
|
|
module Proto
|
|
module SMB
|
|
class SimpleClient
|
|
|
|
class OpenPipe < OpenFile
|
|
|
|
# Valid modes are: 'trans' and 'rw'
|
|
attr_accessor :mode
|
|
|
|
def initialize(*args)
|
|
super(*args)
|
|
self.mode = 'rw'
|
|
@buff = ''
|
|
end
|
|
|
|
def read_buffer(length, offset=0)
|
|
length ||= @buff.length
|
|
@buff.slice!(0, length)
|
|
end
|
|
|
|
def read(length = nil, offset = 0)
|
|
case self.mode
|
|
when 'trans'
|
|
read_buffer(length, offset)
|
|
when 'rw'
|
|
super(length, offset)
|
|
else
|
|
raise ArgumentError
|
|
end
|
|
end
|
|
|
|
def write(data, offset = 0)
|
|
case self.mode
|
|
|
|
when 'trans'
|
|
write_trans(data, offset)
|
|
when 'rw'
|
|
super(data, offset)
|
|
else
|
|
raise ArgumentError
|
|
end
|
|
end
|
|
|
|
def write_trans(data, offset=0)
|
|
ack = self.client.trans_named_pipe(self.file_id, data)
|
|
doff = ack['Payload'].v['DataOffset']
|
|
dlen = ack['Payload'].v['DataCount']
|
|
@buff << ack.to_s[4+doff, dlen]
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|