Class: Rex::Proto::SMB::SimpleClient::OpenPipe

Inherits:
OpenFile
  • Object
show all
Defined in:
lib/rex/proto/smb/simpleclient/open_pipe.rb

Instance Attribute Summary collapse

Attributes inherited from OpenFile

#chunk_size, #client, #file_id, #name, #tree_id

Instance Method Summary collapse

Methods inherited from OpenFile

#<<, #close, #delete

Constructor Details

#initialize(*args) ⇒ OpenPipe


13
14
15
16
17
# File 'lib/rex/proto/smb/simpleclient/open_pipe.rb', line 13

def initialize(*args)
  super(*args)
  self.mode = 'rw'
  @buff = ''
end

Instance Attribute Details

#modeObject

Valid modes are: 'trans' and 'rw'


11
12
13
# File 'lib/rex/proto/smb/simpleclient/open_pipe.rb', line 11

def mode
  @mode
end

Instance Method Details

#read(length = nil, offset = 0) ⇒ Object


24
25
26
27
28
29
30
31
32
33
# File 'lib/rex/proto/smb/simpleclient/open_pipe.rb', line 24

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

#read_buffer(length, offset = 0) ⇒ Object


19
20
21
22
# File 'lib/rex/proto/smb/simpleclient/open_pipe.rb', line 19

def read_buffer(length, offset=0)
  length ||= @buff.length
  @buff.slice!(0, length)
end

#write(data, offset = 0) ⇒ Object


35
36
37
38
39
40
41
42
43
44
45
# File 'lib/rex/proto/smb/simpleclient/open_pipe.rb', line 35

def write(data, offset = 0)
  case self.mode

  when 'trans'
    write_trans(data, offset)
  when 'rw'
    super(data, offset)
  else
    raise ArgumentError
  end
end

#write_trans(data, offset = 0) ⇒ Object


47
48
49
50
51
52
# File 'lib/rex/proto/smb/simpleclient/open_pipe.rb', line 47

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