Class: Rex::Proto::SMB::SimpleClient::OpenFile

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

Overview

This represents an open file, which can be read, written, or closed

Direct Known Subclasses

OpenPipe

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(client, name, tree_id, file_id, versions) ⇒ OpenFile

Returns a new instance of OpenFile.


10
11
12
13
14
15
16
17
# File 'lib/rex/proto/smb/simpleclient/open_file.rb', line 10

def initialize(client, name, tree_id, file_id, versions)
  self.client = client
  self.name = name
  self.tree_id = tree_id
  self.file_id = file_id
  self.chunk_size = 48000
  self.versions = versions
end

Instance Attribute Details

#chunk_sizeObject

Returns the value of attribute chunk_size


8
9
10
# File 'lib/rex/proto/smb/simpleclient/open_file.rb', line 8

def chunk_size
  @chunk_size
end

#clientObject

Returns the value of attribute client


8
9
10
# File 'lib/rex/proto/smb/simpleclient/open_file.rb', line 8

def client
  @client
end

#file_idObject

Returns the value of attribute file_id


8
9
10
# File 'lib/rex/proto/smb/simpleclient/open_file.rb', line 8

def file_id
  @file_id
end

#modeObject

Returns the value of attribute mode


8
9
10
# File 'lib/rex/proto/smb/simpleclient/open_file.rb', line 8

def mode
  @mode
end

#nameObject

Returns the value of attribute name


8
9
10
# File 'lib/rex/proto/smb/simpleclient/open_file.rb', line 8

def name
  @name
end

#tree_idObject

Returns the value of attribute tree_id


8
9
10
# File 'lib/rex/proto/smb/simpleclient/open_file.rb', line 8

def tree_id
  @tree_id
end

#versionsObject

Returns the value of attribute versions


8
9
10
# File 'lib/rex/proto/smb/simpleclient/open_file.rb', line 8

def versions
  @versions
end

Instance Method Details

#<<(data) ⇒ Object


109
110
111
# File 'lib/rex/proto/smb/simpleclient/open_file.rb', line 109

def <<(data)
  write(data)
end

#closeObject

Close this open file


28
29
30
# File 'lib/rex/proto/smb/simpleclient/open_file.rb', line 28

def close
  client.close(file_id, tree_id)
end

#deleteObject


19
20
21
22
23
24
25
# File 'lib/rex/proto/smb/simpleclient/open_file.rb', line 19

def delete
  begin
    close
  rescue StandardError
  end
  client.delete(name, tree_id)
end

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

Read data from the file


101
102
103
104
105
106
107
# File 'lib/rex/proto/smb/simpleclient/open_file.rb', line 101

def read(length = nil, offset = 0)
  if self.client.is_a?(RubySMB::Client)
    read_ruby_smb(length, offset)
  else
    read_rex_smb(length, offset)
  end
end

#read_rex_smb(length, offset) ⇒ Object


64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
# File 'lib/rex/proto/smb/simpleclient/open_file.rb', line 64

def read_rex_smb(length, offset)
  if length.nil?
    data = ''
    fptr = offset
    ok = client.read(file_id, fptr, chunk_size)
    while ok && ok['Payload'].v['DataLenLow'] > 0
      buff = ok.to_s.slice(
        ok['Payload'].v['DataOffset'] + 4,
        ok['Payload'].v['DataLenLow']
      )
      data << buff
      break if ok['Payload'].v['Remaining'] == 0
      fptr += ok['Payload'].v['DataLenLow']

      begin
        ok = client.read(file_id, fptr, chunk_size)
      rescue XCEPT::ErrorCode => e
        case e.error_code
        when 0x00050001
          # Novell fires off an access denied error on EOF
          ok = nil
        else
          raise e
        end
      end
    end
  else
    ok = client.read(file_id, offset, length)
    data = ok.to_s.slice(
      ok['Payload'].v['DataOffset'] + 4,
      ok['Payload'].v['DataLenLow']
    )
  end
  data
end

#read_ruby_smb(length, offset, depth = 0) ⇒ Object


32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
# File 'lib/rex/proto/smb/simpleclient/open_file.rb', line 32

def read_ruby_smb(length, offset, depth = 0)
  if length.nil?
    max_size = client.open_files[client.last_file_id].size
    fptr = offset

    chunk = [max_size, chunk_size].min

    data = client.read(file_id, fptr, chunk).pack('C*')
    fptr = data.length

    while data.length < max_size
      if (max_size - data.length) < chunk
        chunk = max_size - data.length
      end
      data << client.read(file_id, fptr, chunk).pack('C*')
      fptr = data.length
    end
  else
    begin
      data = client.read(file_id, offset, length).pack('C*')
    rescue RubySMB::Error::UnexpectedStatusCode => e
      if e.message == 'STATUS_PIPE_EMPTY' && depth < 20
        data = read_ruby_smb(length, offset, depth + 1)
      else
        raise e
      end
    end
  end

  data
end

#write(data, offset = 0) ⇒ Object

Write data to the file


114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
# File 'lib/rex/proto/smb/simpleclient/open_file.rb', line 114

def write(data, offset = 0)
  # Track our offset into the remote file
  fptr = offset

  # Duplicate the data so we can use slice!
  data = data.dup

  # Take our first chunk of bytes
  chunk = data.slice!(0, chunk_size)

  # Keep writing data until we run out
  until chunk.empty?
    ok = client.write(file_id, fptr, chunk)
    if self.client.is_a?(RubySMB::Client)
      cl = ok
    else
      cl = ok['Payload'].v['CountLow']
    end

    # Partial write, push the failed data back into the queue
    if cl != chunk.length
      data = chunk.slice(cl - 1, chunk.length - cl) + data
    end

    # Increment our painter and grab the next chunk
    fptr += cl
    chunk = data.slice!(0, chunk_size)
  end
end