Attachment 639349 Details for Bug 873657 – The patch allows proxy between CDS and client systems in rhui-lb.py (yum plugin for RHUI)

[patch] The patch allows proxy between CDS and client systems in rhui-lb.py (yum plugin for RHUI)

rh-rhui-tools-2.1.10_client-yum-plugin_proxy.patch (text/plain), 4.97 KB, created by Satoru SATOH on 2012-11-06 12:53:32 UTC

(hide)

Description:

Filename:

MIME Type:

Creator: Satoru SATOH

Created: 2012-11-06 12:53:32 UTC

Size: 4.97 KB

patch

obsolete

>diff -up rh-rhui-tools-2.1.10/etc/rhui/client-yum-plugin/rhui-lb.py.proxy rh-rhui-tools-2.1.10/etc/rhui/client-yum-plugin/rhui-lb.py
>--- rh-rhui-tools-2.1.10/etc/rhui/client-yum-plugin/rhui-lb.py.proxy	2012-11-02 11:19:49.423934938 +0900
>+++ rh-rhui-tools-2.1.10/etc/rhui/client-yum-plugin/rhui-lb.py	2012-11-02 14:20:04.405136870 +0900
>@@ -11,7 +11,9 @@
> # have received a copy of GPLv2 along with this software; if not, see
> # http://www.gnu.org/licenses/old-licenses/gpl-2.0.txt.
> 
>-import httplib
>+import re
>+import urlgrabber
>+import urllib
> import urlparse
> 
> from yum.plugins import TYPE_CORE
>@@ -21,6 +23,38 @@ requires_api_version = '2.5'
> plugin_type = (TYPE_CORE,)
> 
> 
>+def __make_ugopts(conduit):
>+    """
>+    Setup urlgrabber options w/ parameters from config files, both main
>+    (/etc/yum.conf) and plugin's conf (/etc/yum/pluginconf.d/rhui-lb.conf).
>+
>+    NOTE: plugin's conf is given priority over yum's main conf.
>+    """
>+    conf = conduit.getConf()
>+    opts = dict(timeout=10, ssl_ca_cert="/etc/pki/entitlement/ca.crt")
>+
>+    proxy = conduit.confString("main", "proxy", conf.proxy)
>+    if proxy:
>+        proxy_username = conduit.confString("main", "proxy_username",
>+                                            conf.proxy_username).strip()
>+
>+        # Stolen from yum.yumRepo.YumRepository.doProxyDict()
>+        if proxy_username:
>+            auth = urllib.quote(proxy_username)
>+
>+            proxy_password = conduit.confString("main", "proxy_password",
>+                                                conf.proxy_password).strip()
>+            if proxy_password:
>+                auth += ':' + urllib.quote(proxy_password)
>+
>+            proto, rest = re.match("(\w+://)(.+)", proxy).groups()
>+            proxy = "%s%s@%s" % (proto, auth, rest)
>+
>+        opts["proxies"] = dict(https=proxy, HTTPS=proxy)
>+
>+    return opts
>+
>+
> def postreposetup_hook(conduit):
>     """
>     Once the repo has been set up, look to the load balancer listing file
>@@ -51,38 +85,39 @@ def postreposetup_hook(conduit):
> 
>     # Figure out which load balancer is up
>     up_lb = None
>+    ugopts = __make_ugopts(conduit)
>+
>     for lb in cds_balancers:
>         try:
>-            conn = httplib.HTTPSConnection(lb)
>-            conn.request('GET', '/pulp/mirror/?members')
>-            conn.sock.settimeout(10)
>-            response = conn.getresponse()
>-
>-            if response.status == 200:
>-                up_lb = lb
>-
>-                # Make sure the known list of load balancers is accurate.
>-                # If not, update the listing file.
>-                response_body = response.read()
>-                response_balancers = response_body.split()
>-
>-                # We want to attempt to retain the order of the balancer list file,
>-                # so it's not as simple as sorting them and doing an equality check.
>-
>-                # First, remove any from the local list that aren't in the new list.
>-                modified = [x for x in cds_balancers if x in response_balancers]
>-
>-                # Then, add any new ones to the end of the local list.
>-                modified = modified + [x for x in response_balancers if x not in modified]
>-
>-                # If any changes have been made, save to the listing file.
>-                if modified != cds_balancers:
>-                    f = open(cds_list_filename, 'w')
>-                    f.write('\n'.join(modified))
>-                    f.close()
>+            url = "https://%s/pulp/mirror/?members" % lb
>+            data = urlgrabber.urlread(url, **ugopts)
>+
>+            up_lb = lb
>+
>+            # Make sure the known list of load balancers is accurate.
>+            # If not, update the listing file.
>+            response_balancers = data.split()
>+
>+            # We want to attempt to retain the order of the balancer list file,
>+            # so it's not as simple as sorting them and doing an equality check.
>+
>+            # First, remove any from the local list that aren't in the new list.
>+            modified = [x for x in cds_balancers if x in response_balancers]
> 
>-                break
>-        except:
>+            # Then, add any new ones to the end of the local list.
>+            modified = modified + [x for x in response_balancers if x not in modified]
>+
>+            # If any changes have been made, save to the listing file.
>+            if modified != cds_balancers:
>+                f = open(cds_list_filename, 'w')
>+                f.write('\n'.join(modified))
>+                f.close()
>+
>+            break
>+
>+        except urlgrabber.grabber.URLGrabError, e:
>+            conduit.info(2, "Could not retrieve members list from " + lb)
>+            conduit.info(2, "  Reason= " + str(e))
>             pass
> 
>     for repo in rhui_repos:
>@@ -90,3 +125,7 @@ def postreposetup_hook(conduit):
>         # the good one
>         path = urlparse.urlparse(repo.mirrorlist)[2]
>         repo.mirrorlist = 'https://%s%s' % (up_lb, path)
>+
>+        # FIXME: Hack to override proxy configuration of repo object:
>+        repo._proxy_dict = ugopts.get("proxies", None)
>+

Actions: View | Diff

Attachments on bug 873657: 639349 | 684895