aspera-cli 4.5.0 → 4.8.0

data/lib/aspera/{proxy_auto_config.erb.js → proxy_auto_config.js}

@@ -1,16 +1,8 @@
 // inspired by PACSuppport.js, 2003-2004 by Apple Computer, Inc., all rights reserved
-function dnsResolve(host) {
-	if (host == '<%=context_host%>' || host == '<%=context_ip%>')
-		return '<%=context_ip%>';
-	throw 'only DNS for initial host, not '+host;  
-}
-function myIpAddress() {
-	return '<%=context_self%>';
-}
 function isPlainHostName(host) {
 	return (host.indexOf('.') == -1 ? true : false);
 }
-function dnsDomainIs(host, domain) {    
+function dnsDomainIs(host, domain) {
 	var h = host.toLowerCase();
 	var d = domain.toLowerCase();
 	var sub = h.substring(h.length - d.length, h.length);
@@ -35,7 +27,7 @@ function isInNet(host, pattern, mask) {
 		var a = ip.split('.');
 		if ((p.length == m.length) && (m.length == a.length)) {
 			for (i = 0; i < p.length; i++) {
-				if ((p[i]& m[i]) != (m[i]& a[i]))
+				if ((p[i] & m[i]) != (m[i] & a[i]))
 					return false;
 			}
 			return true;
@@ -108,7 +100,7 @@ function dateRange() {
 	var today = new Date();
 	var num = arguments.length;
 	var gmt = arguments[num - 1];
-	if (typeof gmt !=  "string")
+	if (typeof gmt != "string")
 		gmt = false;
 	else {
 		gmt = gmt.toUpperCase();
@@ -137,8 +129,7 @@ function dateRange() {
 					y2 = arg;
 				else
 					return false;
-			}
-			else if (!arg)
+			} else if (!arg)
 				return false;
 			else if (!d1)
 				d1 = arg;
@@ -146,13 +137,12 @@ function dateRange() {
 				d2 = arg;
 			else
 				return false;
-		}
-		else if (typeof arg == "string") {
+		} else if (typeof arg == "string") {
 			var months = "JANFEBMARAPRMAYJUNJULAUGSEPOCTNOVDEC";
 			arg = arg.toUpperCase();
 			arg = months.indexOf(arg);
 			if (arg == -1)
-				return false;            
+				return false;
 			arg /= 3;
 			arg += 1;
 			if (!m1)
@@ -161,23 +151,27 @@ function dateRange() {
 				m2 = arg;
 			else
 				return false;
-		}
-		else
+		} else
 			return false;
 	}
-	if (!y1) y1 = gmt ? today.getUTCFullYear() : today.getFullYear();
-	if (!y2) y2 = y1;
-	if (!m1) m1 = (gmt ? today.getUTCMonth() : today.getMonth()) + 1;
-	if (!m2) m2 = m1;
-	if (!d1) d1 = gmt ? today.getUTCDate() : today.getDate();
-	if (!d2) d2 = d1;
+	if (!y1)
+		y1 = gmt ? today.getUTCFullYear() : today.getFullYear();
+	if (!y2)
+		y2 = y1;
+	if (!m1)
+		m1 = (gmt ? today.getUTCMonth() : today.getMonth()) + 1;
+	if (!m2)
+		m2 = m1;
+	if (!d1)
+		d1 = gmt ? today.getUTCDate() : today.getDate();
+	if (!d2)
+		d2 = d1;
 	var date1;
 	var date2;
 	if (gmt) {
 		date1 = Date.UTC(y1, m1 - 1, d1, 0, 0, 0, 0);
 		date2 = Date.UTC(y2, m2 - 1, d2, 23, 59, 59, 999);
-	}
-	else {
+	} else {
 		date1 = (Date(y1, m1 - 1, d1, 0, 0, 0, 0)).valueOf();
 		date2 = (Date(y2, m2 - 1, d2, 23, 59, 59, 999)).valueOf();
 	}
@@ -190,7 +184,7 @@ function timeRange() {
 	var date2 = new Date();
 	var num = arguments.length;
 	var gmt = arguments[num - 1];
-	if (typeof gmt !=  "string")
+	if (typeof gmt != "string")
 		gmt = false;
 	else {
 		gmt = gmt.toUpperCase();
@@ -226,8 +220,7 @@ function timeRange() {
 				date2.setUTCSeconds(arg);
 				break;
 			}
-		}
-		else {
+		} else {
 			switch (i) {
 			case 0:
 				date1.setHours(arg);
@@ -262,8 +255,7 @@ function timeRange() {
 					date2.setUTCSeconds(arg);
 					break;
 				}
-			}
-			else {
+			} else {
 				switch (i) {
 				case 0:
 					date2.setHours(arg);

data/lib/aspera/proxy_auto_config.rb

@@ -1,34 +1,127 @@
+# frozen_string_literal: true
+
 require 'uri'
 require 'resolv'
-require 'erb'
+
+module URI
+  class Generic
+    # save original method that finds proxy in URI::Generic, it uses env var http_proxy
+    alias_method :find_proxy_orig, :find_proxy
+    def self.register_proxy_finder
+      raise 'mandatory block missing' unless Kernel.block_given?
+      # overload the method in URI : call user's provided block and fallback to original method
+      define_method(:find_proxy) {|envars=ENV| yield(to_s) || find_proxy_orig(envars)}
+    end
+  end
+end
 
 module Aspera
-  # evaluate a proxy autoconfig script
+  # Evaluate a proxy autoconfig script
   class ProxyAutoConfig
     # template file is read once, it contains functions that can be used in a proxy autoconf script
-    PAC_FUNC_TEMPLATE=File.read(__FILE__.gsub(/\.rb$/,'.erb.js'))
-    private_constant :PAC_FUNC_TEMPLATE
+    # it is similar to mozilla ascii_pac_utils.inc
+    PAC_FUNCTIONS_FILE = __FILE__.gsub(/\.rb$/,'.js').freeze
+    PAC_MAIN_FUNCTION = 'FindProxyForURL'
+    private_constant :PAC_FUNCTIONS_FILE,:PAC_MAIN_FUNCTION
+
+    private
+
+    # variables starting with "context_" are replaced in the ERB template file
+    # I did not find an easy way for the javascript to callback ruby
+    # and anyway, it only needs to get DNS translation
+    def pac_dns_functions(context_host)
+      context_self = '127.0.0.1'
+      context_ip = nil
+      Resolv::DNS.open{|dns|dns.each_address(context_host){|r_addr|context_ip = r_addr.to_s if r_addr.is_a?(Resolv::IPv4)}}
+      raise "DNS name not found: #{context_host}" if context_ip.nil?
+      # NOTE: Javascript code here with string inclusions
+      javascript = <<END_OF_JAVASCRIPT
+      function dnsResolve(host) {
+        if (host == '#{context_host}' || host == '#{context_ip}')
+          return '#{context_ip}';
+        throw 'only DNS for initial host, not ' + host;
+      }
+      function myIpAddress() {
+        return '#{context_self}';
+      }
+END_OF_JAVASCRIPT
+      return javascript
+    end
+
+    public
+
     # @param proxy_auto_config the proxy auto config script to be evaluated
     def initialize(proxy_auto_config)
-      @proxy_auto_config=proxy_auto_config
+      # user provided javascript with FindProxyForURL function
+      @proxy_auto_config = proxy_auto_config
+      # avoid multiple execution, this does not support load balancing
+      @cache = {}
+      @pac_functions = nil
     end
 
-    # execut proxy auto config script for the given URL
-    def get_proxy(service_url)
-      # require at runtime, in case there is no js engine
-      require 'execjs'
-      # variables starting with "context_" are replaced in the ERB template file
-      # I did not find an easy way for the javascript to callback ruby
-      # and anyway, it only needs to get DNS translation
-      context_self='127.0.0.1'
-      context_host=URI.parse(service_url).host
-      context_ip=nil
-      Resolv::DNS.open{|dns|dns.each_address(context_host){|r_addr|context_ip=r_addr.to_s if r_addr.is_a?(Resolv::IPv4)}}
-      raise "DNS name not found: #{context_host}" if context_ip.nil?
-      pac_functions=ERB.new(PAC_FUNC_TEMPLATE).result(binding)
-      context = ExecJS.compile(pac_functions+@proxy_auto_config)
-      return context.call("FindProxyForURL", service_url, context_host)
+    def register_uri_generic
+      URI::Generic.register_proxy_finder{|url_str|get_proxies(url_str).first}
+      # allow chaining
+      return self
+    end
+
+    # execute proxy auto config script for the given URL : https://en.wikipedia.org/wiki/Proxy_auto-config
+    # @return either nil, or a String formated following PAC standard
+    def find_proxy_for_url(service_url)
+      uri = URI.parse(service_url)
+      simple_url = "#{uri.scheme}://#{uri.host}"
+      if !@cache.has_key?(simple_url)
+        Log.log.debug("PAC: starting javascript for #{service_url}")
+        # require at runtime, in case there is no js engine
+        require 'execjs'
+        # read template lib
+        @pac_functions = File.read(PAC_FUNCTIONS_FILE).freeze if @pac_functions.nil?
+        # to be executed is dns + utils + user function
+        js_to_execute = "#{pac_dns_functions(uri.host)}#{@pac_functions}#{@proxy_auto_config}"
+        executable_js = ExecJS.compile(js_to_execute)
+        @cache[simple_url] = executable_js.call(PAC_MAIN_FUNCTION, simple_url, uri.host)
+        Log.log.debug("PAC: result: #{@cache[simple_url]}")
+      end
+      return @cache[simple_url]
+    end
+
+    # used to replace URI::Generic.find_proxy
+    # @return Array of URI, possibly empty
+    def get_proxies(service_url)
+      # prepare result
+      uri_list = []
+      # execute PAC script
+      proxy_list_str = find_proxy_for_url(service_url)
+      if !proxy_list_str.is_a?(String)
+        Log.log.warn("PAC: did not return a String, returned #{proxy_list_str.class}")
+        return uri_list
+      end
+      proxy_list_str.strip!
+      proxy_list_str.gsub!(/\s+/,' ')
+      proxy_list_str.split(';').each do |item|
+        # strip and split by space
+        parts = item.strip.split
+        case parts.shift
+        when 'DIRECT'
+          raise 'DIRECT has no param' unless parts.empty?
+        when 'PROXY'
+          addr_port = parts.shift
+          raise 'PROXY shall have one param' unless addr_port.is_a?(String) && parts.empty?
+          begin
+            # PAC proxy addresses are <host>:<port>
+            if /:[0-9]+$/.match?(addr_port)
+              # we want to return URIs, so add dummy scheme
+              uri_list.push(URI.parse("proxy://#{addr_port}"))
+            else
+              Log.log.warn("PAC: PROXY must be <address>:<port>, ignoring #{addr_port}")
+            end
+          rescue StandardError
+            Log.log.warn("PAC: cannot parse #{addr_port}")
+          end
+        else Log.log.warn("PAC: ignoring proxy type #{parts.first}: not supported")
+        end
+      end
+      return uri_list
     end
   end
 end
-