detector 0.2.1 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ab9800bbb6b9322ca7e8115b438c89cf5da96febc891e1ae6cf55ff8de6ff011
-  data.tar.gz: 4916567db9f219e16d9fe3bef938dbc51958d786205894a850074686c5e41400
+  metadata.gz: 9180708d27a005b8af76c7451d69354f3c63084c671a87adbad2957627098445
+  data.tar.gz: 83011b8a32cbac053af086fc2d6a029daa69fc78e796075aad4d0dbd7c21b6cf
 SHA512:
-  metadata.gz: aad9b58db57085e77b3543c50297d57d779506b93e0f89fd7475df0de19c3b45170127588dc7754fae9f5e256feb1939cb5aba88c98bc56d6289f9413f866ad6
-  data.tar.gz: 3dc3ef92bfe6bb083f90e282d15ab0866722da1dfaddb30b8cae8076150755fad0c31274649eef7e16a19b993fb07443d68d301eb236ffaa5c7beaf91c234216
+  metadata.gz: b1f95d6eb75df123a8704f96b08b7ec8fa4433d556c9d0a334d25adc4e81ce4e40f2dc1ad410f63a0bbb5e9a99d75c3fd33c8d4d6add23514d8879f142c3ab59
+  data.tar.gz: 18b9fa63739174bdf3a3301f8ceb2618b693829668fa97adc4cdac43014f68d4f1fc11840fea86118edc967bfa362e0c464b7ee8833e3d9228073f558de05043

data/lib/detector/addons/mysql.rb

@@ -90,6 +90,10 @@ module Detector
         "mysql"
       end
       
+      def protocol_type
+        :tcp
+      end
+      
       def user_access_level
         return nil unless connection
         

data/lib/detector/addons/postgres.rb

@@ -115,6 +115,10 @@ module Detector
         "psql"
       end
       
+      def protocol_type
+        :tcp
+      end
+      
       def user_access_level
         return nil unless connection
         

data/lib/detector/addons/redis.rb

@@ -72,6 +72,10 @@ module Detector
         "redis-cli"
       end
       
+      def protocol_type
+        :tcp
+      end
+      
       def user_access_level
         return nil unless connection
         

data/lib/detector/addons/smtp.rb

@@ -33,6 +33,10 @@ module Detector
         "telnet"
       end
       
+      def protocol_type
+        :tcp
+      end
+      
       def user_access_level
         return nil unless connection
         

data/lib/detector/base.rb

@@ -320,11 +320,40 @@ module Detector
     
     def ping
       return nil unless valid?
-      tcp_test
+      transport?
+    end
+    
+    def transport?
+      protocol_type == :tcp ? tcp_test : udp_test
+    end
+    
+    # Should be implemented by subclasses
+    def protocol_type
+      :tcp # Default to TCP
     end
     
     def tcp_test
-      TCPSocket.new(ip, port).present? rescue nil
+      return nil unless ip && port
+      begin
+        socket = TCPSocket.new(ip, port)
+        socket.close
+        true
+      rescue => e
+        nil
+      end
+    end
+    
+    def udp_test
+      return nil unless ip && port
+      begin
+        socket = UDPSocket.new
+        socket.connect(ip, port)
+        socket.send("", 0) # Send empty packet as probe
+        socket.close
+        true
+      rescue => e
+        nil
+      end
     end
     
     def table_count

data/lib/detector/version.rb

@@ -1,3 +1,3 @@
 module Detector
-  VERSION = "0.2.1"
+  VERSION = "0.3.0"
 end 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: detector
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.3.0
 platform: ruby
 authors:
 - Jonathan Siegel