Commit 25ab6a34 authored by Aaron Gable's avatar Aaron Gable Committed by Commit Bot

Revert "gerrit_util: Use httplib2 for communication instead of httplib."

This reverts commit d7568645.

Reason for revert:
third_party.httplib2.ServerNotFoundError: Unable to find the server at metadata.google.internal

Original change's description:
> gerrit_util: Use httplib2 for communication instead of httplib.
> 
> Retain the httplib import to continue using its constants, but actually make
> the http(s) connections using httplib2. The latter has built-in support for
> proxy settings, which then actually allows people behind proxies to interact
> with Gerrit.
> 
> Compared to httplib, the biggest changes are:
> - There's only one Http class instead of HTTPConnection and HTTPSConnection.
> - Http.request() returns a tuple (response, contents).
> - Http.request() expects a full URI instead of just a path, as Http's
>   constructor does not take a host parameter.
> - The response object inherits from dict.
> - All headers in a response are lower-cased.
> 
> All in all, it is possible to see that httplib2 support was retro-fitted
> into the code, but that should not worsen its readability overall.
> 
> Patch written in collaboration with Alexis Menard <alexis.menard@intel.com>.
> 
> BUG=672729
> R=​alexis.menard@intel.com,agable@chromium.org,tandrii@chromium.org
> 
> Change-Id: Ic40e804064e74e89bc2ad979572628f1bd78c19a
> Reviewed-on: https://chromium-review.googlesource.com/458221
> Reviewed-by: Aaron Gable <agable@chromium.org>
> Commit-Queue: Raphael Kubo da Costa (rakuco) <raphael.kubo.da.costa@intel.com>
> 

TBR=agable@chromium.org,alexis.menard@intel.com,raphael.kubo.da.costa@intel.com,tandrii@chromium.org,chromium-reviews@chromium.org
NOPRESUBMIT=true
NOTREECHECKS=true
NOTRY=true
BUG=672729

Change-Id: Idfacd314b381232733bb92a02ec2fb85f016effd
Reviewed-on: https://chromium-review.googlesource.com/457792Reviewed-by: 's avatarAaron Gable <agable@chromium.org>
Commit-Queue: Aaron Gable <agable@chromium.org>
parent d7568645
...@@ -11,7 +11,7 @@ https://gerrit-review.googlesource.com/Documentation/rest-api.html ...@@ -11,7 +11,7 @@ https://gerrit-review.googlesource.com/Documentation/rest-api.html
import base64 import base64
import contextlib import contextlib
import cookielib import cookielib
import httplib # Still used for its constants. import httplib
import json import json
import logging import logging
import netrc import netrc
...@@ -27,7 +27,6 @@ import urlparse ...@@ -27,7 +27,6 @@ import urlparse
from cStringIO import StringIO from cStringIO import StringIO
import gclient_utils import gclient_utils
from third_party import httplib2
LOGGER = logging.getLogger() LOGGER = logging.getLogger()
TRY_LIMIT = 5 TRY_LIMIT = 5
...@@ -63,8 +62,10 @@ def _QueryString(param_dict, first_param=None): ...@@ -63,8 +62,10 @@ def _QueryString(param_dict, first_param=None):
def GetConnectionClass(protocol=None): def GetConnectionClass(protocol=None):
if protocol is None: if protocol is None:
protocol = GERRIT_PROTOCOL protocol = GERRIT_PROTOCOL
if protocol in ('http', 'https'): if protocol == 'https':
return httplib2.Http return httplib.HTTPSConnection
elif protocol == 'http':
return httplib.HTTPConnection
else: else:
raise RuntimeError( raise RuntimeError(
"Don't know how to work with protocol '%s'" % protocol) "Don't know how to work with protocol '%s'" % protocol)
...@@ -228,11 +229,11 @@ class GceAuthenticator(Authenticator): ...@@ -228,11 +229,11 @@ class GceAuthenticator(Authenticator):
def _test_is_gce(cls): def _test_is_gce(cls):
# Based on https://cloud.google.com/compute/docs/metadata#runninggce # Based on https://cloud.google.com/compute/docs/metadata#runninggce
try: try:
resp, _ = cls._get(cls._INFO_URL) resp = cls._get(cls._INFO_URL)
except socket.error: except socket.error:
# Could not resolve URL. # Could not resolve URL.
return False return False
return resp.get('metadata-flavor') == 'Google' return resp.getheader('Metadata-Flavor', None) == 'Google'
@staticmethod @staticmethod
def _get(url, **kwargs): def _get(url, **kwargs):
...@@ -246,11 +247,12 @@ class GceAuthenticator(Authenticator): ...@@ -246,11 +247,12 @@ class GceAuthenticator(Authenticator):
next_delay_sec *= 2 next_delay_sec *= 2
p = urlparse.urlparse(url) p = urlparse.urlparse(url)
c = GetConnectionClass(protocol=p.scheme)() c = GetConnectionClass(protocol=p.scheme)(p.netloc)
resp, contents = c.request(url, 'GET', **kwargs) c.request('GET', url, **kwargs)
resp = c.getresponse()
LOGGER.debug('GET [%s] #%d/%d (%d)', url, i+1, TRY_LIMIT, resp.status) LOGGER.debug('GET [%s] #%d/%d (%d)', url, i+1, TRY_LIMIT, resp.status)
if resp.status < httplib.INTERNAL_SERVER_ERROR: if resp.status < httplib.INTERNAL_SERVER_ERROR:
return (resp, contents) return resp
@classmethod @classmethod
def _get_token_dict(cls): def _get_token_dict(cls):
...@@ -259,10 +261,10 @@ class GceAuthenticator(Authenticator): ...@@ -259,10 +261,10 @@ class GceAuthenticator(Authenticator):
if cls._token_expiration < time.time() - 25: if cls._token_expiration < time.time() - 25:
return cls._token_cache return cls._token_cache
resp, contents = cls._get(cls._ACQUIRE_URL, headers=cls._ACQUIRE_HEADERS) resp = cls._get(cls._ACQUIRE_URL, headers=cls._ACQUIRE_HEADERS)
if resp.status != httplib.OK: if resp.status != httplib.OK:
return None return None
cls._token_cache = json.loads(contents) cls._token_cache = json.load(resp)
cls._token_expiration = cls._token_cache['expires_in'] + time.time() cls._token_expiration = cls._token_cache['expires_in'] + time.time()
return cls._token_cache return cls._token_cache
...@@ -304,11 +306,12 @@ def CreateHttpConn(host, path, reqtype='GET', headers=None, body=None): ...@@ -304,11 +306,12 @@ def CreateHttpConn(host, path, reqtype='GET', headers=None, body=None):
conn = GetConnectionClass()(host) conn = GetConnectionClass()(host)
conn.req_host = host conn.req_host = host
conn.req_params = { conn.req_params = {
'uri': urlparse.urljoin('%s://%s' % (GERRIT_PROTOCOL, host), url), 'url': url,
'method': reqtype, 'method': reqtype,
'headers': headers, 'headers': headers,
'body': body, 'body': body,
} }
conn.request(**conn.req_params)
return conn return conn
...@@ -316,7 +319,7 @@ def ReadHttpResponse(conn, expect_status=200, ignore_404=True): ...@@ -316,7 +319,7 @@ def ReadHttpResponse(conn, expect_status=200, ignore_404=True):
"""Reads an http response from a connection into a string buffer. """Reads an http response from a connection into a string buffer.
Args: Args:
conn: An Http object created by CreateHttpConn above. conn: An HTTPSConnection or HTTPConnection created by CreateHttpConn, above.
expect_status: Success is indicated by this status in the response. expect_status: Success is indicated by this status in the response.
ignore_404: For many requests, gerrit-on-borg will return 404 if the request ignore_404: For many requests, gerrit-on-borg will return 404 if the request
doesn't match the database contents. In most such cases, we doesn't match the database contents. In most such cases, we
...@@ -326,10 +329,10 @@ def ReadHttpResponse(conn, expect_status=200, ignore_404=True): ...@@ -326,10 +329,10 @@ def ReadHttpResponse(conn, expect_status=200, ignore_404=True):
sleep_time = 0.5 sleep_time = 0.5
for idx in range(TRY_LIMIT): for idx in range(TRY_LIMIT):
response, contents = conn.request(**conn.req_params) response = conn.getresponse()
# Check if this is an authentication issue. # Check if this is an authentication issue.
www_authenticate = response.get('www-authenticate') www_authenticate = response.getheader('www-authenticate')
if (response.status in (httplib.UNAUTHORIZED, httplib.FOUND) and if (response.status in (httplib.UNAUTHORIZED, httplib.FOUND) and
www_authenticate): www_authenticate):
auth_match = re.search('realm="([^"]+)"', www_authenticate, re.I) auth_match = re.search('realm="([^"]+)"', www_authenticate, re.I)
...@@ -341,25 +344,31 @@ def ReadHttpResponse(conn, expect_status=200, ignore_404=True): ...@@ -341,25 +344,31 @@ def ReadHttpResponse(conn, expect_status=200, ignore_404=True):
# If response.status < 500 then the result is final; break retry loop. # If response.status < 500 then the result is final; break retry loop.
if response.status < 500: if response.status < 500:
LOGGER.debug('got response %d for %s %s', response.status, LOGGER.debug('got response %d for %s %s', response.status,
conn.req_params['method'], conn.req_params['uri']) conn.req_params['method'], conn.req_params['url'])
break break
# A status >=500 is assumed to be a possible transient error; retry. # A status >=500 is assumed to be a possible transient error; retry.
http_version = 'HTTP/%s' % ('1.1' if response.version == 11 else '1.0') http_version = 'HTTP/%s' % ('1.1' if response.version == 11 else '1.0')
LOGGER.warn('A transient error occurred while querying %s:\n' LOGGER.warn('A transient error occurred while querying %s:\n'
'%s %s %s\n' '%s %s %s\n'
'%s %d %s', '%s %d %s',
conn.host, conn.req_params['method'], conn.req_params['uri'], conn.host, conn.req_params['method'], conn.req_params['url'],
http_version, http_version, response.status, response.reason) http_version, http_version, response.status, response.reason)
if TRY_LIMIT - idx > 1: if TRY_LIMIT - idx > 1:
LOGGER.warn('... will retry %d more times.', TRY_LIMIT - idx - 1) LOGGER.warn('... will retry %d more times.', TRY_LIMIT - idx - 1)
time.sleep(sleep_time) time.sleep(sleep_time)
sleep_time = sleep_time * 2 sleep_time = sleep_time * 2
req_host = conn.req_host
req_params = conn.req_params
conn = GetConnectionClass()(req_host)
conn.req_host = req_host
conn.req_params = req_params
conn.request(**req_params)
if ignore_404 and response.status == 404: if ignore_404 and response.status == 404:
return StringIO() return StringIO()
if response.status != expect_status: if response.status != expect_status:
reason = '%s: %s' % (response.reason, contents) reason = '%s: %s' % (response.reason, response.read())
raise GerritError(response.status, reason) raise GerritError(response.status, reason)
return StringIO(contents) return StringIO(response.read())
def ReadHttpJsonResponse(conn, expect_status=200, ignore_404=True): def ReadHttpJsonResponse(conn, expect_status=200, ignore_404=True):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment