lib/sockit.rb in sockit-1.1.1 vs lib/sockit.rb in sockit-1.2.0

@@ -67,10 +67,22 @@
     else
       @@config
     end
   end
 
+  def enabled
+    @@enabled ||= false
+  end
+
+  def enable
+    @@enabled = true
+  end
+
+  def disable
+    @@enabled = false
+  end
+
   extend Sockit::V5::Authentication
   extend Sockit::V5::Connection
   extend Sockit::V5::Support
 
   extend Sockit::V4::Connection
@@ -81,15 +93,17 @@
 end
 
 class TCPSocket
 
   alias :initialize_tcp :initialize
+
   def initialize(remote_host, remote_port, local_host=nil, local_port=nil)
-    if Sockit.connect_via_socks?(remote_host)
+    if Sockit.enabled? && Sockit.connect_via_socks?(remote_host)
       initialize_tcp(Sockit.config.host, Sockit.config.port)
       Sockit.perform_v5_authenticate(self) if Sockit.is_socks_v5?
       Sockit.connect(self, remote_host, remote_port)
     else
       Sockit.direct_connect(self, remote_host, remote_port, local_host, local_port)
     end
   end
+
 end