thm 0.2.1 → 0.2.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 7b09ec5d25d74981d6de95c8ff62b81e52e6bf50
-  data.tar.gz: 572aeb78c445a7519d9a7db03c5fe99ea8437bf6
+  metadata.gz: 14298155e86d49a7a209f664112f24a831608ab5
+  data.tar.gz: 36f3b7465a8bda21887bc76760d83e18e1f53aac
 SHA512:
-  metadata.gz: 9a3da3536826878934cec22453d65f39dc4952af97368003835f7b04f186415f74cf7f611fd2a58fde883d6dfd4bec8f2432885c997dd8d0750b268f5cfba3e9
-  data.tar.gz: 6eca45c81f9aab95254f09138ad6f0a43541e5581d8e7d4886ee8dafe41378db3ba7c8100e2a9b849a9f5d3cdf6e9431a73fd0329f8d0a7b41821c8a4439586f
+  metadata.gz: 4ea375fc0eeac8aa7c01f5849ab71b6994ef5a38f6f1756bbd7bc933a87e24d0dbc6a3330c575cb8243e4e3ac42c1a9ad1a54d541f66be0320174aa20dbfd776
+  data.tar.gz: 52f8c8f450769a074d3f4797dfe366a635ba307e50d13b76ddda15bde8ce86756bc8736fee2cdf5f08b5d710687425add72002f68207e6578bdf5e4fe078fb1f

data/{datalayerlight.rb → lib/thm/datalayerlight.rb}

@@ -82,6 +82,7 @@ module DatalayerLight
 
   end
 
+  # Fine for small capture projects
   class MySQLDrv
     
     require 'mysql'
@@ -139,5 +140,73 @@ module DatalayerLight
     end
     
   end
+
+  # Metrics / Measurements Engine InfluxDB RestAPI
+  class InfluxDB
+
+    require "net/http"
+    require "uri"
+    require "json"
+    require "pp"
   
+    attr_accessor :dbhost, :dbuser, :dbpass, :dbport, :dburl
+
+    def initialize
+      @dbhost = "127.0.0.1"
+      @dbuser = "threatmonitor"
+      @dbpass = "dk3rbi9l"
+      @dbport = 8086
+      @dbname = "threatmonitor"
+    end
+    
+    def apiget(sql)
+      @dburl = "http://#{@dbhost}:#{@dbport}"
+      sqlunicode = URI.encode(sql)
+      puts "InfluxDB SQL URL: #{@dburl}/query?db=#{@dbname}&q=#{sqlunicode}"
+      uri = URI.parse("#{@dburl}/query?db=#{@dbname}&q=#{sqlunicode}")
+      puts "Request URI: #{uri}"
+      http = Net::HTTP.new(uri.host, uri.port)
+      begin
+        response = http.request(Net::HTTP::Get.new(uri.request_uri))
+        begin
+          j = JSON.parse(response.body)
+        rescue JSON::ParserError
+          puts "Could not read JSON data"
+        end
+      rescue
+        puts "Error retrieving data"
+      end
+    end
+    
+    def apipost(data)
+      @dburl = "http://#{@dbhost}:#{@dbport}"
+      #puts "InfluxDB SQL URL: #{@dburl}/query?db=#{@dbname}"
+      uri = URI.parse("#{@dburl}/write?db=#{@dbname}")
+      #puts "Request URI: #{uri}"
+      http = Net::HTTP.new(uri.host, uri.port)
+      request = Net::HTTP::Post.new(uri.request_uri)
+      request.set_content_type("application/x-www-form-urlencoded")
+      begin
+        request.body = data unless data.empty?
+        response = http.request(request)
+        if response.code == 204 # Good response
+          puts "OK"
+        elsif response.code == 200 or response.code == 400 # 200 can be an error in some cases !!
+          puts "Error code #{response.code}"
+        end
+      rescue
+        puts "Error posting data"
+      end
+    end
+    
+    def query(sql, mode="r")
+      if mode == "r"
+        apiget("#{sql}")
+      elsif mode == "w"
+        apipost(sql)
+      end
+    end
+    
+  end
+
 end

data/lib/thm/version.rb

@@ -4,7 +4,7 @@ module Thm #:nodoc:
 
     MAJOR = 0
     MINOR = 2
-    TINY = 1
+    TINY = 2
     CODENAME = "Deedrah"
 
     STRING = [MAJOR, MINOR, TINY].join('.')

data/lib/thm.rb

@@ -16,9 +16,6 @@ require 'guid'
 require 'yaml'
 require 'pcaplet'
 require 'pcaprub' # For Live capture / write
-require File.expand_path(File.join(
-          File.dirname(__FILE__),
-          "../datalayerlight.rb"))
 include Pcap
 
 # TODO
@@ -38,7 +35,12 @@ module Tools
   
 end
 
-# Load Datasources
+# Load Database drivers
+require File.expand_path(File.join(
+          File.dirname(__FILE__),
+          "../lib/thm/datalayerlight.rb"))
+          
+# Load Datasources / Services contains defaults
 require File.expand_path(File.join(
           File.dirname(__FILE__),
           "../lib/thm/dataservices.rb"))

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: thm
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.2.2
 platform: ruby
 authors:
 - puppetpies
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-07-18 00:00:00.000000000 Z
+date: 2015-07-20 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -185,7 +185,6 @@ files:
 - bin/thm-session
 - bin/thm-useradmin
 - config.rb
-- datalayerlight.rb
 - js/JSXTransformer.js
 - js/chartkick.js
 - js/files/authenticate.jsx
@@ -195,6 +194,7 @@ files:
 - js/react.js
 - lib/thm.rb
 - lib/thm/consumer.rb
+- lib/thm/datalayerlight.rb
 - lib/thm/dataservices.rb
 - lib/thm/fileservices.rb
 - lib/thm/localmachine.rb