mqsink 0.0.5

data/bin/mqsinkd

@@ -0,0 +1,26 @@
+#!/usr/bin/env ruby
+
+require 'json'
+require 'mqsink'
+
+def local_boot
+ if not Dir.exists? "/etc/mqsink"
+  system("mkdir /etc/mqsink")
+  system("touch /etc/mqsink/acl.conf")
+  system("touch /etc/mqsink/classifiers.conf")
+  system("touch /etc/mqsink/mqsink.json")
+ end
+end
+
+local_boot
+jconfig = ARGV[0]
+jconfig = '/etc/mqsink/mqsink.json' if not ARGV[0]
+cfg_file = open(jconfig).readlines.join
+puts "+ FATAL : the configuration file is empty!" if cfg_file == ''
+exit 1 if cfg_file == ''
+begin
+ cfg = JSON.parse cfg_file
+ debug = 0
+ debug = 1 if cfg['debug'] == "on"
+ sink = MQSink::Server.new cfg, debug
+end

data/lib/mqsink/acl.rb

@@ -0,0 +1,27 @@
+module MQSink
+ class ACL
+  def initialize clientip, acl
+    @clientip 	= clientip
+    @acl	= acl
+  end
+
+  def dynamic_reconstruct acl, pool=[] # hacky removal of line breaks .. need to change it really #
+   acl.each do |line| 
+    pool << line.chop
+   end
+   return pool
+  end
+
+  def lookup
+   pool = dynamic_reconstruct @acl
+   if pool.include?(@clientip)
+    return 0
+   elsif pool[0] == "0.0.0.0"
+    return 0
+   else
+    return 1
+   end
+  end
+
+ end
+end

data/lib/mqsink/amqp.rb

@@ -0,0 +1,48 @@
+module MQSink
+ class AMQP
+
+  def initialize msg, remotehost, mode, mqserver, classifiers, static_queue, debug
+   @msg                 = msg
+   @remotehost          = remotehost
+   @mode                = mode
+   @mqserver		= mqserver
+   @classifiers		= classifiers
+   @static_queue	= static_queue
+   @debug               = debug
+  end
+
+  def publish
+   if @mode == 'xlate'
+    queue         = @msg.split(" ")[0]
+    datag         = @msg.split(" ")[1]
+   elsif @mode == 'classify'
+    File.open(@classifiers).each do |line|
+     if line.match /\[.*?\]/
+      queue, regex = line.split ' '
+      refine_regex = regex.split('[')[1].split(']')[0..-1].join
+      re = Regexp.new(refine_regex)
+      if @msg.match re
+       datag = @msg
+      else
+       return
+      end
+     end
+    end
+   elsif @mode == 'raw'
+    queue         = @static_queue
+    datag         = @msg
+   end
+   connection = Bunny.new(:hostname => @mqserver)
+   connection.start
+   ch           = connection.create_channel
+   q            = ch.queue(queue)
+   message      = datag
+   mssg         = { :timestamp => Time.now, :service => 'mqsink', :fwd_host => @remotehost, :payload => message }
+   mssg         = JSON.dump mssg
+   puts "+ mqsink [debug]: outbound message to queue '#{queue}' -> #{mssg}" if @debug == 1
+   ch.default_exchange.publish(mssg, :routing_key => q.name)
+   connection.close
+  end
+
+ end
+end

data/lib/mqsink/server.rb

@@ -0,0 +1,57 @@
+module MQSink
+ class Server
+
+  def initialize cfg, debug
+   @localhost   = cfg['localhost']
+   @localport   = cfg['localport']
+   @firewall    = cfg['firewall']
+   @fw_file	= cfg['fw_file']
+   @mode        = cfg['mode']
+   @mqserver    = cfg['amqp_server']
+   @classifiers	= cfg['classifiers']
+   @static_queue = cfg['static_queue']
+   @debug	= debug
+   @acl   = build_acl @fw_file
+   @acl_refresh = cfg['fw_refresh']
+   listen
+  end
+
+  def build_acl fw_file
+   puts "+ mqsink [debug]: rebuilding ACL .." if @debug == 1
+   @acl         = open(fw_file).readlines
+   return @acl
+  end
+
+  def listen
+   @sock = UDPSocket.new
+   puts "+ mqsink [debug]: loading server .." if @debug == 1
+   @sock.bind @localhost, @localport
+   count = 1
+   while true do
+    r, w, e = IO.select([@sock], nil, nil)
+    if r[0]
+      buffer, sockaddr = @sock.recvfrom_nonblock(1500)
+      if @firewall == "on"
+       @acl = build_acl @fw_file if count == 5 and @acl_refresh == 'on'
+       count = 0 if count == 5 and @acl_refresh == 'on'
+       count += 1 
+       puts "+ mqsink [debug]: verifying authorization for client -> #{sockaddr[3]} .." if @debug == 1
+       acl = MQSink::ACL.new sockaddr[3], @acl
+       lookup = acl.lookup
+      else # do not apply ACL when firewall is disabled #
+       lookup = 0
+      end
+      if lookup == 0
+       remotehost = Resolv.getname sockaddr[3]
+       puts "+ mqsink [debug]: buffer accepted -> #{buffer}" if @debug == 1
+       relay = MQSink::AMQP.new buffer, remotehost, @mode, @mqserver, @classifiers, @static_queue, @debug
+       relay.publish
+      else
+       puts "+ mqsink [debug]: client is not authorized .. ignoring buffer .." if @debug == 1
+      end
+    end
+   end
+  end
+
+ end
+end

data/lib/mqsink.rb

@@ -0,0 +1,9 @@
+module MQSink
+ Directory = File.expand_path File.join File.dirname(__FILE__), '../'
+ require 'bunny'
+ require 'socket'
+ require 'resolv'
+ require 'mqsink/server'
+ require 'mqsink/amqp'
+ require 'mqsink/acl'
+end

metadata

@@ -0,0 +1,51 @@
+--- !ruby/object:Gem::Specification
+name: mqsink
+version: !ruby/object:Gem::Version
+  version: 0.0.5
+  prerelease: 
+platform: ruby
+authors:
+- Samer Abdel-Hafez
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-12-30 00:00:00.000000000 Z
+dependencies: []
+description: a smart and lightweight udp to amqp translator
+email:
+- sam@ip.fi
+executables:
+- mqsinkd
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/mqsink.rb
+- lib/mqsink/acl.rb
+- lib/mqsink/amqp.rb
+- lib/mqsink/server.rb
+- bin/mqsinkd
+homepage: http://github.com/nopedial/mqsink
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: mqsink
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: udp to amqp bridge
+test_files: []