Class: Rex::Proto::SMB::SimpleClient::OpenPipe
- Inherits:
-
OpenFile
- Object
- OpenFile
- Rex::Proto::SMB::SimpleClient::OpenPipe
show all
- Defined in:
- lib/rex/proto/smb/simple_client/open_pipe.rb
Constant Summary
collapse
- STATUS_BUFFER_OVERFLOW =
This will only return the bytes available and does not receive available data
0x80000005
- STATUS_PIPE_BROKEN =
0xc000014b
Instance Attribute Summary collapse
Attributes inherited from OpenFile
#chunk_size, #client, #file_id, #name, #tree_id, #versions
Instance Method Summary
collapse
Methods inherited from OpenFile
#<<, #close, #delete, #read_rex_smb, #read_ruby_smb
Constructor Details
#initialize(*args) ⇒ OpenPipe
Returns a new instance of OpenPipe.
13
14
15
16
17
|
# File 'lib/rex/proto/smb/simple_client/open_pipe.rb', line 13
def initialize(*args)
super(*args)
self.mode = 'rw'
@buff = ''
end
|
Instance Attribute Details
#mode ⇒ Object
Valid modes are: ‘trans’ and ‘rw’
11
12
13
|
# File 'lib/rex/proto/smb/simple_client/open_pipe.rb', line 11
def mode
@mode
end
|
Instance Method Details
#peek ⇒ Object
84
85
86
87
88
89
90
91
|
# File 'lib/rex/proto/smb/simple_client/open_pipe.rb', line 84
def peek
if self.client.is_a?(RubySMB::Client)
avail = peek_ruby_smb
else
avail = peek_rex_smb
end
avail
end
|
#peek_rex_smb ⇒ Object
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
|
# File 'lib/rex/proto/smb/simple_client/open_pipe.rb', line 65
def peek_rex_smb
setup = [0x23, self.file_id].pack('vv')
pkt = self.client.trans_maxzero('\\PIPE\\', '', '', 2, setup, false, true, true)
if pkt['Payload']['SMB'].v['ErrorClass'] == STATUS_PIPE_BROKEN
raise IOError
end
avail = 0
begin
avail = pkt.to_s[pkt['Payload'].v['ParamOffset']+4, 2].unpack('v')[0]
rescue
end
if (avail == 0) and (pkt['Payload']['SMB'].v['ErrorClass'] == STATUS_BUFFER_OVERFLOW)
avail = self.client.default_max_buffer_size
end
avail
end
|
#peek_ruby_smb ⇒ Object
58
59
60
|
# File 'lib/rex/proto/smb/simple_client/open_pipe.rb', line 58
def peek_ruby_smb
self.client.last_file.peek_available
end
|
#read(length = nil, offset = 0) ⇒ Object
24
25
26
27
28
29
30
31
32
33
|
# File 'lib/rex/proto/smb/simple_client/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/simple_client/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
46
47
48
49
|
# File 'lib/rex/proto/smb/simple_client/open_pipe.rb', line 35
def write(data, offset = 0)
case self.mode
when 'trans'
if self.client.is_a?(RubySMB::Client)
raise NotImplementedError, '\'trans\' mode is not supported by RubySMB'
end
write_trans(data, offset)
when 'rw'
super(data, offset)
else
raise ArgumentError
end
end
|
#write_trans(data, offset = 0) ⇒ Object
51
52
53
54
55
56
|
# File 'lib/rex/proto/smb/simple_client/open_pipe.rb', line 51
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
|