Class: Rex::OLE::DIFAT

Inherits:
Object
  • Object
show all
Defined in:
lib/rex/ole/difat.rb

Direct Known Subclasses

FAT, MiniFAT

Instance Method Summary collapse

Constructor Details

#initialize(stg) ⇒ DIFAT

Returns a new instance of DIFAT.



17
18
19
20
 
# File 'lib/rex/ole/difat.rb', line 17

def initialize stg
	@stg = stg
	@entries = []
end

Instance Method Details

#+(expr) ⇒ Object 



33
34
35
36
 
# File 'lib/rex/ole/difat.rb', line 33

def +(expr)
	@entries += expr
	self
end

#<<(expr) ⇒ Object 



38
39
40
 
# File 'lib/rex/ole/difat.rb', line 38

def <<(expr)
	@entries << expr
end

#[](idx) ⇒ Object 



29
30
31
 
# File 'lib/rex/ole/difat.rb', line 29

def [](idx)
	@entries[idx]
end

#[]=(idx, expr) ⇒ Object

convenience access to entries



25
26
27
 
# File 'lib/rex/ole/difat.rb', line 25

def []=(idx,expr)
	@entries[idx] = expr
end

#each ⇒ Object 



54
55
56
57
58
 
# File 'lib/rex/ole/difat.rb', line 54

def each
	@entries.each { |el|
		yield el
	}
end

#length ⇒ Object 



42
43
44
 
# File 'lib/rex/ole/difat.rb', line 42

def length
	@entries.length
end

#read ⇒ Object

low-level functions



87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
 
# File 'lib/rex/ole/difat.rb', line 87

def read
	@entries = []

	# start with the header part
	@entries += @stg.header._sectFat

	# double indirect fat
	sect = @stg.header._sectDifStart
	while (sect != SECT_END)
		if (@entries.include?(sect))
			raise RuntimeError, 'Sector chain loop detected (0x%08x)' % sect
		end

		@entries << sect
		buf = @stg.read_sector(sect, @stg.header.sector_size)

		# the last sect ptr in the block becomes the next entry
		sect = Util.get32(buf, ((@stg.header.idx_per_sect)-1) * 4)
	end

	# don't need these free ones, but it doesn't hurt to keep them.
	#@difat.delete(SECT_FREE)
end

#reset ⇒ Object 



50
51
52
 
# File 'lib/rex/ole/difat.rb', line 50

def reset
	@entries = []
end

#slice!(start, stop) ⇒ Object 



46
47
48
 
# File 'lib/rex/ole/difat.rb', line 46

def slice!(start,stop)
	@entries.slice!(start,stop)
end

#to_s ⇒ Object

woop



63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
 
# File 'lib/rex/ole/difat.rb', line 63

def to_s
	ret = "{ "
	@entries.each { |el|
		ret << ", " if (ret.length > 2)
		case el
		when SECT_END
			ret << "END"
		when SECT_DIF
			ret << "DIF"
		when SECT_FAT
			ret << "FAT"
		when SECT_FREE
			ret << "FREE"
		else
			ret << "0x%x" % el
		end
	}
	ret << " }"
	ret
end

#write ⇒ Object 



111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
 
# File 'lib/rex/ole/difat.rb', line 111

def write
	len = @entries.length
	first109 = @entries.dup

	rest = nil
	if (len > 109)
		rest = first109.slice!(109,len)
	end

	@stg.header._sectFat = []
	@stg.header._sectFat += first109
	if (len < 109)
		need = 109 - len
		need.times {
			@stg.header._sectFat << SECT_FREE
		}
	end

	if (rest and rest.length > 0)
		raise RuntimeError, 'TODO: support writing DIF properly!'
		# may require adding more fat sectors :-/
		#@stg.header._csectDif = rest.length
		#@stg.header._sectDifStart = idx
	end

	@stg.header._csectFat = len
end