xmpp_auth

xmpp_auth.rb

#!/usr/bin/env ruby

require 'logger'
require 'rest_client'

$stdout.sync = true
$stdin.sync = true

path = "/var/log/ejabberd/auth.log"
file = File.open(path, File::WRONLY | File::APPEND | File::CREAT)
file.sync = true
logger = Logger.new(file)
logger.level = Logger::DEBUG
logger.info "Starting up"


def auth(username, password)
  RestClient.post("http://something.com/devices/xmpp_auth", :username => username, :password => password)
  return true
rescue RestClient::Exception
  return false
end

logger.info "Starting ejabberd authentication service"

loop do
  begin
    $stdin.eof? # wait for input
    logger.debug "Start request"
    start = Time.now

    msg = $stdin.read(2)
    length = msg.unpack('n').first

    msg = $stdin.read(length)
    cmd, *data = msg.split(":")

    logger.info "Incoming Request: '#{cmd}'"
    success = case cmd
              when "auth"
                logger.info "Authenticating #{data[0]}@#{data[1]}"
                auth(data[0], data[2])
              else
                false
              end

    bool = success ? 1 : 0
    $stdout.write [2, bool].pack("nn")
    logger.info "Response: #{success ? "success" : "failure"}"
  rescue => e
    logger.error "#{e.class.name}: #{e.message}"
    logger.error e.backtrace.join("\n\t")
  end
end