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

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

Direct Known Subclasses

OpenPipe

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

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

Returns a new instance of OpenFile.



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

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

Instance Attribute Details

#chunk_size ⇒ Object

Returns the value of attribute chunk_size.



25
26
27
 
# File 'lib/rex/proto/smb/simpleclient.rb', line 25

def chunk_size
  @chunk_size
end

#client ⇒ Object

Returns the value of attribute client.



25
26
27
 
# File 'lib/rex/proto/smb/simpleclient.rb', line 25

def client
  @client
end

#file_id ⇒ Object

Returns the value of attribute file_id.



25
26
27
 
# File 'lib/rex/proto/smb/simpleclient.rb', line 25

def file_id
  @file_id
end

#mode ⇒ Object

Returns the value of attribute mode.



25
26
27
 
# File 'lib/rex/proto/smb/simpleclient.rb', line 25

def mode
  @mode
end

#name ⇒ Object

Returns the value of attribute name.



25
26
27
 
# File 'lib/rex/proto/smb/simpleclient.rb', line 25

def name
  @name
end

#tree_id ⇒ Object

Returns the value of attribute tree_id.



25
26
27
 
# File 'lib/rex/proto/smb/simpleclient.rb', line 25

def tree_id
  @tree_id
end

Instance Method Details

#<<(data) ⇒ Object 



89
90
91
 
# File 'lib/rex/proto/smb/simpleclient.rb', line 89

def << (data)
	self.write(data)
end

#close ⇒ Object

Close this open file



44
45
46
 
# File 'lib/rex/proto/smb/simpleclient.rb', line 44

def close
	self.client.close(self.file_id, self.tree_id)
end

#delete ⇒ Object 



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

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

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

Read data from the file



49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
 
# File 'lib/rex/proto/smb/simpleclient.rb', line 49

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

			begin
				ok = self.client.read(self.file_id, fptr, self.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

		return data
	else
		ok = self.client.read(self.file_id, offset, length)
		data = ok.to_s.slice(
			ok['Payload'].v['DataOffset'] + 4,
			ok['Payload'].v['DataLenLow']
		)
		return data
	end
end

#write(data, offset = 0) ⇒ Object

Write data to the file



94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
 
# File 'lib/rex/proto/smb/simpleclient.rb', line 94

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, self.chunk_size)

	# Keep writing data until we run out
	while (chunk.length > 0)
		ok = self.client.write(self.file_id, fptr, chunk)
		cl = ok['Payload'].v['CountLow']

		# 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, self.chunk_size)
	end
end