r_socks 0.2.8 → 0.2.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f9f3daabf6b654e432702a1f8ec25cb8a31360cbd199badd407d3f8b0beb4a75
-  data.tar.gz: 20a333f1b28ffd89094362e27b7dccd4c0b7053bd8f5e83154fd29c2485a11ab
+  metadata.gz: 46f22ff3b97fbaadccab67bf228ca67015b12edb0395ee8d7ef74e9b9327a591
+  data.tar.gz: 13f8da021a4f3693c68f1f6260a059a0dac445a37c3d602ad0e1c5bbcfb159ca
 SHA512:
-  metadata.gz: 9cbb6b31615e0448dab30fb0eb6a6b39485415d091d8f559b01d4a60a1982496c998c45b0870971ee002b3ac4d9e0eba078b5efce546baa5909170600b525624
-  data.tar.gz: 991c43cbb67f8f52c2ffce971bfd0e7cecdaad8a02565e0489494ac6893036e2464e5404053887f8b5cbaeca840927c698a1a41b9023afc34f41698d9c1d3fbc
+  metadata.gz: 30beb9843a14a9851372be0fb1cdc14677bdf2339c93dddc9a65ec2962c27d2a1dd69ca518a443040eba645d2c461943b6bb11cfdae0c1eeaaf8f1543a4dfe38
+  data.tar.gz: 7d43b76ad191f450fc68f7e54f04dea783c574d57e367acfbe3a781fb2586882c6f896f93009472bf15efee163aee24444d2fa13aae760550cb1705fd0824048

data/lib/r_socks/config.rb

@@ -118,5 +118,13 @@ module RSocks
     def forward_addr=(value)
       @store[:forward_addr] = value.to_s
     end
+
+    def forward_white_list=(arr)
+      @store[:white_list] = arr
+    end
+
+    def forward_white_list
+      @store[:white_list] || []
+    end
   end
 end

data/lib/r_socks/connection_handler.rb

@@ -26,6 +26,11 @@ module RSocks
           )
         end
         @port, @ip = Socket.unpack_sockaddr_in(get_peername)
+
+        white_list = @config.forward_white_list
+        if !white_list.empty? && !white_list.include?(@ip.to_s)
+          raise Error, "#{@ip} not in white list"
+        end
       rescue => e
         puts "post_init error: #{e.message}"
         close_connection

data/lib/r_socks/http_proxy_parser.rb

@@ -12,6 +12,7 @@ module RSocks
       @default_password = ENV['RSOCKS_PASSWORD'] || 'default'
       @adaptor = config.auth_adaptor
       @health_check_route = config.health_check_route
+      @need_auth = config.auth_method == :no_auth
     end
 
     def call(data)
@@ -27,9 +28,13 @@ module RSocks
       host_format_checking(temp.shift)
       generate_header(temp)
 
-      state = auth_user
-      raise RSocks::HttpAuthFailed unless state
-      state
+      if @need_auth
+        state = auth_user
+        raise RSocks::HttpAuthFailed unless state
+        return state
+      end
+
+      true
     end
 
     def auth_user

data/lib/r_socks/version.rb

@@ -1,3 +1,3 @@
 module RSocks
-  VERSION = "0.2.8"
+  VERSION = "0.2.9"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: r_socks
 version: !ruby/object:Gem::Version
-  version: 0.2.8
+  version: 0.2.9
 platform: ruby
 authors:
 - Nick An