Class: Rex::MIME::Header

Inherits:
Object
  • Object
show all
Defined in:
lib/rex/mime/header.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(data = '') ⇒ Header

Returns a new instance of Header.



10
11
12
13
 
# File 'lib/rex/mime/header.rb', line 10

def initialize(data='')
  self.headers = []
  parse(data)
end

Instance Attribute Details

#headers ⇒ Object

Returns the value of attribute headers.



8
9
10
 
# File 'lib/rex/mime/header.rb', line 8

def headers
  @headers
end

Instance Method Details

#add(var, val) ⇒ Object 



57
58
59
60
 
# File 'lib/rex/mime/header.rb', line 57

def add(var, val)
  self.headers << [var, val]
  self.headers[-1]
end

#find(idx) ⇒ Object 



39
40
41
42
43
44
45
46
47
48
49
50
 
# File 'lib/rex/mime/header.rb', line 39

def find(idx)
  if (idx.class == ::Fixnum)
    return self.headers[idx]
  else
    self.headers.each do |pair|
      if (pair[0] == idx.to_s)
        return pair
      end
    end
  end
  nil
end

#parse(data) ⇒ Object 



15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
 
# File 'lib/rex/mime/header.rb', line 15

def parse(data)
  prev = nil
  data.gsub("\r", '').split("\n").each do |line|

    # Handle header folding
    if (line =~ /^\s+/)
      # Ignore if there is no previous header
      next if not prev
      next if not self.headers[prev]
      self.headers[prev][1] << line.strip
      next
    end

    var,val = line.split(':')
    next if not val
    self.headers << [ var.to_s.strip, val.to_s.strip ]
    prev = self.headers.length - 1
  end
end

#remove(idx) ⇒ Object 



62
63
64
65
66
67
68
69
70
71
72
73
 
# File 'lib/rex/mime/header.rb', line 62

def remove(idx)
  if (idx.class == ::Fixnum)
    self.headers.delete_at(idx)
  else
    self.headers.each_index do |i|
      pair = self.headers[i]
      if (pair[0] == idx.to_s)
        self.headers.delete_at(i)
      end
    end
  end
end

#set(var, val) ⇒ Object 



52
53
54
55
 
# File 'lib/rex/mime/header.rb', line 52

def set(var, val)
  hdr = self.find(var) || self.add(var, '')
  hdr[1] = val
end

#to_s ⇒ Object 



35
36
37
 
# File 'lib/rex/mime/header.rb', line 35

def to_s
  self.headers.map{ |pair| "#{pair[0]}: #{pair[1]}\r\n" }.join
end