Browse Source

Fix XMPP

v1
Nicolas Vinot 7 years ago
parent
commit
37acf426ab
  1. 1
      Gemfile
  2. 9
      lib/cryptcheck.rb
  3. 18
      lib/cryptcheck/tls/xmpp/server.rb

1
Gemfile

@ -10,6 +10,7 @@ gem 'ruby-progressbar'
gem 'logging'
gem 'activerecord'
gem 'sqlite3'
gem 'rake'
group :test do
gem 'rspec'

9
lib/cryptcheck.rb

@ -1,10 +1,17 @@
module CryptCheck
module Tls
autoload :Server, 'cryptcheck/tls/server'
autoload :TlsNotSupportedServer, 'cryptcheck/tls/server'
autoload :Grade, 'cryptcheck/tls/grade'
autoload :TlsNotSupportedGrade, 'cryptcheck/tls/grade'
autoload :Https, 'cryptcheck/tls/https'
autoload :Xmpp, 'cryptcheck/tls/xmpp'
module Https
autoload :Server, 'cryptcheck/tls/https/server'
autoload :Grade, 'cryptcheck/tls/https/grade'
end
autoload :Xmpp, 'cryptcheck/tls/xmpp'
module Xmpp
autoload :Server, 'cryptcheck/tls/xmpp/server'
autoload :Grade, 'cryptcheck/tls/xmpp/grade'

18
lib/cryptcheck/tls/xmpp/server.rb

@ -13,11 +13,11 @@ module CryptCheck
attr_reader :domain
def initialize(domain, type=:s2s, hostname: nil)
@type, @domain = type, domain
service, port = case type
when :s2s then ['_xmpp-server', 5269]
when :c2s then ['_xmpp-client', 5222]
end
@domain = domain
unless hostname
srv = RESOLVER.getresources("#{service}._tcp.#{domain}", Resolv::DNS::Resource::IN::SRV).sort_by(&:priority).first
if srv
@ -30,8 +30,20 @@ module CryptCheck
end
def ssl_connect(socket, context, method, &block)
socket.write "<?xml version='1.0' ?><stream:stream xmlns:stream='http://etherx.jabber.org/streams' xmlns='jabber:client' to='#{@domain}' version='1.0'>"
response = ::Nokogiri::XML socket.recv 4096
type = case @type
when :s2s then 'jabber:server'
when :c2s then 'jabber:client'
end
socket.write "<?xml version='1.0' ?><stream:stream xmlns:stream='http://etherx.jabber.org/streams' xmlns='#{type}' to='#{@domain}' version='1.0'>"
response = ''
loop do
response += socket.recv 1024
xml = ::Nokogiri::XML response
unless xml.xpath('//stream:features').empty?
response = xml
break
end
end
starttls = response.xpath '//tls:starttls', tls: TLS_NAMESPACE
raise TLSNotAvailableException unless starttls
@required = !starttls.xpath('//tls:required', tls: TLS_NAMESPACE).nil?

Loading…
Cancel
Save