tools/chrome_proxy/common/chrome_proxy_metrics.py

   1 # Copyright 2014 The Chromium Authors. All rights reserved.
   2 # Use of this source code is governed by a BSD-style license that can be
   3 # found in the LICENSE file.
   4
   5 import logging
   6 import time
   7
   8 from common import network_metrics
   9 from telemetry.page import page_test
  10 from telemetry.value import scalar
  11
  12
  13 CHROME_PROXY_VIA_HEADER = 'Chrome-Compression-Proxy'
  14
  15
  16 class ChromeProxyMetricException(page_test.MeasurementFailure):
  17   pass
  18
  19
  20 class ChromeProxyResponse(network_metrics.HTTPResponse):
  21   """ Represents an HTTP response from a timeline event."""
  22   def __init__(self, event):
  23     super(ChromeProxyResponse, self).__init__(event)
  24
  25   def ShouldHaveChromeProxyViaHeader(self):
  26     resp = self.response
  27     # Ignore https and data url
  28     if resp.url.startswith('https') or resp.url.startswith('data:'):
  29       return False
  30     # Ignore 304 Not Modified and cache hit.
  31     if resp.status == 304 or resp.served_from_cache:
  32       return False
  33     # Ignore invalid responses that don't have any header. Log a warning.
  34     if not resp.headers:
  35       logging.warning('response for %s does not any have header '
  36                       '(refer=%s, status=%s)',
  37                       resp.url, resp.GetHeader('Referer'), resp.status)
  38       return False
  39     return True
  40
  41   def HasResponseHeader(self, key, value):
  42     response_header = self.response.GetHeader(key)
  43     if not response_header:
  44       return False
  45     values = [v.strip() for v in response_header.split(',')]
  46     return any(v == value for v in values)
  47
  48   def HasRequestHeader(self, key, value):
  49     if key not in self.response.request_headers:
  50       return False
  51     request_header = self.response.request_headers[key]
  52     values = [v.strip() for v in request_header.split(',')]
  53     return any(v == value for v in values)
  54
  55   def HasChromeProxyViaHeader(self):
  56     via_header = self.response.GetHeader('Via')
  57     if not via_header:
  58       return False
  59     vias = [v.strip(' ') for v in via_header.split(',')]
  60     # The Via header is valid if it has a 4-character version prefix followed by
  61     # the proxy name, for example, "1.1 Chrome-Compression-Proxy".
  62     return any(v[4:] == CHROME_PROXY_VIA_HEADER for v in vias)
  63
  64   def HasExtraViaHeader(self, extra_header):
  65     return self.HasResponseHeader('Via', extra_header)
  66
  67   def IsValidByViaHeader(self):
  68     return (not self.ShouldHaveChromeProxyViaHeader() or
  69             self.HasChromeProxyViaHeader())
  70
  71   def GetChromeProxyClientType(self):
  72     """Get the client type directive from the Chrome-Proxy request header.
  73
  74     Returns:
  75         The client type directive from the Chrome-Proxy request header for the
  76         request that lead to this response. For example, if the request header
  77         "Chrome-Proxy: c=android" is present, then this method would return
  78         "android". Returns None if no client type directive is present.
  79     """
  80     if 'Chrome-Proxy' not in self.response.request_headers:
  81       return None
  82
  83     chrome_proxy_request_header = self.response.request_headers['Chrome-Proxy']
  84     values = [v.strip() for v in chrome_proxy_request_header.split(',')]
  85     for value in values:
  86       kvp = value.split('=', 1)
  87       if len(kvp) == 2 and kvp[0].strip() == 'c':
  88         return kvp[1].strip()
  89     return None
  90
  91   def HasChromeProxyLoFiRequest(self):
  92     return self.HasRequestHeader('Chrome-Proxy', "q=low")
  93
  94   def HasChromeProxyLoFiResponse(self):
  95     return self.HasResponseHeader('Chrome-Proxy', "q=low")
  96
  97   def HasChromeProxyPassThroughRequest(self):
  98     return self.HasRequestHeader('Chrome-Proxy', "pass-through")