patronus_fati 0.8.9 → 0.8.10

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 07f850ac1cb0f1dd2f7dfb513bd4ce335fbbf6a4
-  data.tar.gz: fb8c5272dca78ebf66d505ccb899fc2de44637dc
+  metadata.gz: fb413efee4a3b4a4804852f63c9d6929b6b22aa3
+  data.tar.gz: 7bf0278db9e8317ee699c2e84f6c7766080dc6b9
 SHA512:
-  metadata.gz: 16a3e594dea7679cf076e53fbd762d0387a34f81e9d94e7ced43d79045fae10f1b96a8ed85fb7cb9b5e0440917be279e9c04f9b841c0c868d84cf76bdcce0496
-  data.tar.gz: f723d26a1a9e1f143612efc29322a268d67f6a8cc7ba30d38280043b81d03e312052469ac672a4fa571b4f8e2ba3eb593ec8080b5df26e910e137ebde4f35d4f
+  metadata.gz: c7a04a984146e82ef6dee440a2dad9356ebc4efd7c12de375bda5f1866df4a67916b584de5d977701e68dcd61a740776fa70bf3b22801fabeb74b7f9b3c1cd96
+  data.tar.gz: 2d97babd5efa250bcc0dbfc187771034c93c25432be0b5f2a127c6430a5b7a5df92c7da53948ab0787f77e6c5ee9c458206733adaa8f1f742fd5b041645313c3

data/lib/patronus_fati/connection.rb

@@ -1,4 +1,10 @@
 module PatronusFati
+  PatronusFatiError = Class.new(StandardError)
+  LostConnection    = Class.new(PatronusFatiError)
+  ConnectionTimeout = Class.new(PatronusFatiError)
+  UnableToRead      = Class.new(PatronusFatiError)
+  UnableToWrite     = Class.new(PatronusFatiError)
+
   class Connection
     attr_reader :port, :read_queue, :server, :write_queue
 
@@ -54,15 +60,12 @@ module PatronusFati
           while (line = socket.readline)
             read_queue << line
           end
-        rescue Timeout::Error
-          # Connection timed out...
-          exit 1
-        rescue EOFError
-          # We lost our connection...
-          exit 2
+        rescue Timeout::Error => e
+          raise ConnectionTimeout, e.message
+        rescue EOFError => e
+          raise LostConnection, e.message
         rescue => e
-          puts ('Unknown issue reading from socket: %s' % e.message)
-          exit 3
+          raise UnableToRead, e.message
         end
       end
     end
@@ -76,8 +79,7 @@ module PatronusFati
             count += 1
           end
         rescue => e
-          puts ('Unknown issue writing to socket: %s' % e.message)
-          exit 1
+          raise UnableToWrite, e.message
         end
       end
     end

data/lib/patronus_fati/message_processor/terminate.rb

@@ -0,0 +1,9 @@
+module PatronusFati::MessageProcessor::Terminate
+  include PatronusFati::MessageProcessor
+
+  def self.process(obj)
+    puts 'Kismet announced it\'s intention to gracefully terminate'
+
+    nil
+  end
+end

data/lib/patronus_fati/message_processor.rb

@@ -68,3 +68,4 @@ require 'patronus_fati/message_processor/critfail'
 require 'patronus_fati/message_processor/error'
 require 'patronus_fati/message_processor/protocols'
 require 'patronus_fati/message_processor/ssid'
+require 'patronus_fati/message_processor/terminate'

data/lib/patronus_fati/version.rb

@@ -1,3 +1,3 @@
 module PatronusFati
-  VERSION = '0.8.9'
+  VERSION = '0.8.10'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: patronus_fati
 version: !ruby/object:Gem::Version
-  version: 0.8.9
+  version: 0.8.10
 platform: ruby
 authors:
 - Sam Stelfox
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-04-13 00:00:00.000000000 Z
+date: 2015-04-14 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: dm-constraints
@@ -344,6 +344,7 @@ files:
 - lib/patronus_fati/message_processor/error.rb
 - lib/patronus_fati/message_processor/protocols.rb
 - lib/patronus_fati/message_processor/ssid.rb
+- lib/patronus_fati/message_processor/terminate.rb
 - lib/patronus_fati/version.rb
 - patronus_fati.gemspec
 - spec/data_models/access_point_spec.rb