class Rex::MIME::Message

Attributes

bound[RW]
content[RW]
header[RW]
parts[RW]

Public Class Methods

new(data=nil) click to toggle source
# File lib/rex/mime/message.rb, line 15
def initialize(data=nil)
  self.header = Rex::MIME::Header.new
  self.parts  = []
  self.bound  = "_Part_#{rand(1024)}_#{rand(0xffffffff)}_#{rand(0xffffffff)}"
  self.content = ''
  if data
    head,body = data.split(/\r?\n\r?\n/, 2)

    self.header.parse(head)
    ctype = self.header.find('Content-Type')

    if ctype and ctype[1] and ctype[1] =~ /multipart\/mixed;\s*boundary=([^\s]+)/
      self.bound = $1

      chunks = body.to_s.split(/--#{self.bound}(--)?\r?\n/)
      self.content = chunks.shift.to_s.gsub(/\s+$/, '')
      self.content << "\r\n" if not self.content.empty?

      chunks.each do |chunk|
        break if chunk == "--"
        head,body = chunk.split(/\r?\n\r?\n/, 2)
        part = Rex::MIME::Part.new
        part.header.parse(head)
        part.content = body.gsub(/\s+$/, '')
        self.parts << part
      end
    else
      self.content = body.to_s.gsub(/\s+$/, '') + "\r\n"
    end
  end
end

Public Instance Methods

add_part(data='', content_type='text/plain', transfer_encoding="8bit", content_disposition=nil) click to toggle source
# File lib/rex/mime/message.rb, line 88
def add_part(data='', content_type='text/plain', transfer_encoding="8bit", content_disposition=nil)
  part = Rex::MIME::Part.new

  if (content_disposition)
    part.header.set("Content-Disposition", content_disposition)
  end

  if (content_type)
    part.header.set("Content-Type", content_type)
  end

  if (transfer_encoding)
    part.header.set("Content-Transfer-Encoding", transfer_encoding)
  end

  part.content = data
  self.parts << part
  part
end
add_part_attachment(data, name) click to toggle source
# File lib/rex/mime/message.rb, line 108
def add_part_attachment(data, name)
  self.add_part(
    Rex::Text.encode_base64(data, "\r\n"),
    "application/octet-stream; name=\"#{name}\"",
    "base64",
    "attachment; filename=\"#{name}\""
  )
end
add_part_inline_attachment(data, name) click to toggle source
# File lib/rex/mime/message.rb, line 118
def add_part_inline_attachment(data, name)
  self.add_part(
    Rex::Text.encode_base64(data, "\r\n"),
    "application/octet-stream; name=\"#{name}\"",
    "base64",
    "inline; filename=\"#{name}\""
  )
end
from() click to toggle source
# File lib/rex/mime/message.rb, line 59
def from
  (self.header.find('From') || [nil, nil])[1]
end
from=(val) click to toggle source
# File lib/rex/mime/message.rb, line 55
def from=(val)
  self.header.set("From", val)
end
mime_defaults() click to toggle source
# File lib/rex/mime/message.rb, line 71
def mime_defaults
  self.header.set("MIME-Version", "1.0")
  self.header.set("Content-Type", "multipart/mixed; boundary=\"#{self.bound}\"")
  self.header.set("Subject", '') # placeholder
  self.header.set("Date", Time.now.strftime("%a,%e %b %Y %H:%M:%S %z"))
  self.header.set("Message-ID",
    "<"+
    Rex::Text.rand_text_alphanumeric(rand(20)+40)+
    "@"+
    Rex::Text.rand_text_alpha(rand(20)+3)+
    ">"
  )
  self.header.set("From", '')    # placeholder
  self.header.set("To", '')      # placeholder
end
subject() click to toggle source
# File lib/rex/mime/message.rb, line 67
def subject
  (self.header.find('Subject') || [nil, nil])[1]
end
subject=(val) click to toggle source
# File lib/rex/mime/message.rb, line 63
def subject=(val)
  self.header.set("Subject", val)
end
to() click to toggle source
# File lib/rex/mime/message.rb, line 47
def to
  (self.header.find('To') || [nil, nil])[1]
end
to=(val) click to toggle source
# File lib/rex/mime/message.rb, line 51
def to=(val)
  self.header.set("To", val)
end
to_s() click to toggle source
# File lib/rex/mime/message.rb, line 127
def to_s
  msg = force_crlf(self.header.to_s + "\r\n")

  unless self.content.blank?
    msg << force_crlf(self.content + "\r\n")
  end

  self.parts.each do |part|
    msg << force_crlf("--" + self.bound + "\r\n")
    msg << part.to_s
  end

  if self.parts.length > 0
    msg << force_crlf("--" + self.bound + "--\r\n")
  end

  msg
end