36 lines
1.0 KiB
Ruby
36 lines
1.0 KiB
Ruby
|
# encoding: UTF-8
|
||
|
|
|
||
|
|
require 'rex/proto/sip/util'
|
||
|
|
|
||
|
|
module Rex
|
||
|
|
module Proto
|
||
|
|
# SIP protocol support
|
||
|
|
module SIP
|
||
|
|
SIP_STATUS_REGEX = /^SIP\/(\d\.\d) (\d{3})\s*(.*)$/
|
||
|
|
|
||
|
|
# Represents a SIP response message
|
||
|
|
class Response
|
||
|
|
attr_accessor :version, :code, :message, :headers
|
||
|
|
|
||
|
|
def header(name)
|
||
|
|
@headers.select { |k, _| k.downcase == name.downcase }.last
|
||
|
|
end
|
||
|
|
|
||
|
|
def self.parse(data)
|
||
|
|
response = Response.new
|
||
|
|
# do some basic sanity checking on this response to ensure that it is SIP
|
||
|
|
status_line = data.split(/\r\n/)[0]
|
||
|
|
unless status_line && status_line =~ SIP_STATUS_REGEX
|
||
|
|
fail(ArgumentError, 'Does not start with a valid SIP status line')
|
||
|
|
end
|
||
|
|
response.version = Regexp.last_match(1)
|
||
|
|
response.code = Regexp.last_match(2)
|
||
|
|
response.message = Regexp.last_match(3)
|
||
|
|
response.headers = ::Rex::Proto::SIP.extract_headers(data)
|
||
|
|
response
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|