| Current File : //proc/self/root/usr/lib/python3/dist-packages/lazr/uri/_uri.py |
# Copyright 2009 Canonical Ltd. All rights reserved.
#
# This file is part of lazr.uri
#
# lazr.uri is free software: you can redistribute it and/or modify it
# under the terms of the GNU Lesser General Public License as published by
# the Free Software Foundation, version 3 of the License.
#
# lazr.uri is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
# FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public
# License for more details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with lazr.uri. If not, see <http://www.gnu.org/licenses/>.
"""Functions for working with generic syntax URIs."""
__metaclass__ = type
__all__ = [
'URI',
'InvalidURIError',
'find_uris_in_text',
'possible_uri_re',
'merge',
'remove_dot_segments',
]
import re
try:
unicode
except NameError:
unicode = str
# Default port numbers for different URI schemes
# The registered URI schemes comes from
# http://www.iana.org/assignments/uri-schemes.html
# The default ports come from the relevant RFCs
_default_port = {
# Official schemes
'acap': '674',
'dav': '80',
'dict': '2628',
'dns': '53',
'ftp': '21',
'go': '1096',
'gopher': '70',
'h323': '1720',
'http': '80',
'https': '443',
'imap': '143',
'ipp': '631',
'iris.beep': '702',
'ldap': '389',
'mtqp': '1038',
'mupdate': '3905',
'nfs': '2049',
'nntp': '119',
'pop': '110',
'rtsp': '554',
'sip': '5060',
'sips': '5061',
'snmp': '161',
'soap.beep': '605',
'soap.beeps': '605',
'telnet': '23',
'tftp': '69',
'tip': '3372',
'vemmi': '575',
'xmlrpc.beep': '602',
'xmlrpc.beeps': '602',
'z39.50r': '210',
'z39.50s': '210',
# Historical schemes
'prospero': '1525',
'wais': '210',
# Common but unregistered schemes
'bzr+http': '80',
'bzr+ssh': '22',
'irc': '6667',
'sftp': '22',
'ssh': '22',
'svn': '3690',
'svn+ssh': '22',
}
# Regular expressions adapted from the ABNF in the RFC
scheme_re = r"(?P<scheme>[a-z][-a-z0-9+.]*)"
userinfo_re = r"(?P<userinfo>(?:[-a-z0-9._~!$&\'()*+,;=:]|%[0-9a-f]{2})*)"
# The following regular expression will match some IP address style
# host names that the RFC would not (e.g. leading zeros on the
# components), but is signficantly simpler.
host_re = (r"(?P<host>[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}|"
r"(?:[-a-z0-9._~!$&\'()*+,;=]|%[0-9a-f]{2})*|"
r"\[[0-9a-z:.]+\])")
port_re = r"(?P<port>[0-9]*)"
authority_re = r"(?P<authority>(?:%s@)?%s(?::%s)?)" % (
userinfo_re, host_re, port_re)
path_abempty_re = r"(?:/(?:[-a-z0-9._~!$&\'()*+,;=:@]|%[0-9a-f]{2})*)*"
path_noscheme_re = (r"(?:[-a-z0-9._~!$&\'()*+,;=@]|%[0-9a-f]{2})+"
r"(?:/(?:[-a-z0-9._~!$&\'()*+,;=:@]|%[0-9a-f]{2})*)*")
path_rootless_re = (r"(?:[-a-z0-9._~!$&\'()*+,;=:@]|%[0-9a-f]{2})+"
r"(?:/(?:[-a-z0-9._~!$&\'()*+,;=:@]|%[0-9a-f]{2})*)*")
path_absolute_re = r"/(?:%s)?" % path_rootless_re
path_empty_re = r""
hier_part_re = r"(?P<hierpart>//%s%s|%s|%s|%s)" % (
authority_re, path_abempty_re, path_absolute_re, path_rootless_re,
path_empty_re)
relative_part_re = r"(?P<relativepart>//%s%s|%s|%s|%s)" % (
authority_re, path_abempty_re, path_absolute_re, path_noscheme_re,
path_empty_re)
# Additionally we also permit square braces in the query portion to
# accomodate real-world URIs.
query_re = r"(?P<query>(?:[-a-z0-9._~!$&\'()*+,;=:@/?\[\]]|%[0-9a-f]{2})*)"
fragment_re = r"(?P<fragment>(?:[-a-z0-9._~!$&\'()*+,;=:@/?]|%[0-9a-f]{2})*)"
uri_re = r"%s:%s(?:\?%s)?(?:#%s)?$" % (
scheme_re, hier_part_re, query_re, fragment_re)
relative_ref_re = r"%s(?:\?%s)?(?:#%s)?$" % (
relative_part_re, query_re, fragment_re)
uri_pat = re.compile(uri_re, re.IGNORECASE)
relative_ref_pat = re.compile(relative_ref_re, re.IGNORECASE)
def merge(basepath, relpath, has_authority):
"""Merge two URI path components into a single path component.
Follows rules specified in Section 5.2.3 of RFC 3986.
The algorithm in the RFC treats the empty basepath edge case
differently for URIs with and without an authority section, which
is why the third argument is necessary.
"""
if has_authority and basepath == '':
return '/' + relpath
slash = basepath.rfind('/')
return basepath[:slash+1] + relpath
def remove_dot_segments(path):
"""Remove '.' and '..' segments from a URI path.
Follows the rules specified in Section 5.2.4 of RFC 3986.
"""
output = []
while path:
if path.startswith('../'):
path = path[3:]
elif path.startswith('./'):
path = path[2:]
elif path.startswith('/./') or path == '/.':
path = '/' + path[3:]
elif path.startswith('/../') or path == '/..':
path = '/' + path[4:]
if len(output) > 0:
del output[-1]
elif path in ['.', '..']:
path = ''
else:
if path.startswith('/'):
slash = path.find('/', 1)
else:
slash = path.find('/')
if slash < 0:
slash = len(path)
output.append(path[:slash])
path = path[slash:]
return ''.join(output)
def normalise_unreserved(string):
"""Return a version of 's' where no unreserved characters are encoded.
Unreserved characters are defined in Section 2.3 of RFC 3986.
Percent encoded sequences are normalised to upper case.
"""
result = string.split('%')
unreserved = ('ABCDEFGHIJKLMNOPQRSTUVWXYZ'
'abcdefghijklmnopqrstuvwxyz'
'0123456789-._~')
for index, item in enumerate(result):
if index == 0:
continue
try:
ch = int(item[:2], 16)
except ValueError:
continue
if chr(ch) in unreserved:
result[index] = chr(ch) + item[2:]
else:
result[index] = '%%%02X%s' % (ch, item[2:])
return ''.join(result)
class InvalidURIError(Exception):
"""Invalid URI"""
class URI:
"""A class that represents a URI.
This class can represent arbitrary URIs that conform to the
generic syntax described in RFC 3986.
"""
def __init__(self, uri=None, scheme=None, userinfo=None, host=None,
port=None, path=None, query=None, fragment=None):
"""Create a URI instance.
Can be called with either a string URI or the component parts
of the URI as keyword arguments.
In either case, all arguments are expected to be appropriately
URI encoded.
"""
assert (uri is not None and scheme is None and userinfo is None and
host is None and port is None and path is None and
query is None and fragment is None) or uri is None, (
"URI() must be called with a single string argument or "
"with URI components given as keyword arguments.")
if uri is not None:
if isinstance(uri, unicode):
try:
uri.encode('ASCII')
except UnicodeEncodeError:
raise InvalidURIError(
'URIs must consist of ASCII characters')
match = uri_pat.match(uri)
if match is None:
raise InvalidURIError('"%s" is not a valid URI' % uri)
self.scheme = match.group('scheme')
self.userinfo = match.group('userinfo')
self.host = match.group('host')
self.port = match.group('port')
hierpart = match.group('hierpart')
authority = match.group('authority')
if authority is None:
self.path = hierpart
else:
# Skip past the //authority part
self.path = hierpart[2+len(authority):]
self.query = match.group('query')
self.fragment = match.group('fragment')
else:
if scheme is None:
raise InvalidURIError('URIs must have a scheme')
if host is None and (userinfo is not None or port is not None):
raise InvalidURIError(
'host must be given if userinfo or port are')
if path is None:
raise InvalidURIError('URIs must have a path')
self.scheme = scheme
self.userinfo = userinfo
self.host = host
self.port = port
self.path = path
self.query = query
self.fragment = fragment
self._normalise()
if (self.scheme in ['http', 'https', 'ftp', 'gopher', 'telnet',
'imap', 'mms', 'rtsp', 'svn', 'svn+ssh',
'bzr', 'bzr+http', 'bzr+ssh'] and
not self.host):
raise InvalidURIError('%s URIs must have a host name' %
self.scheme)
def _normalise(self):
"""Perform normalisation of URI components."""
self.scheme = self.scheme.lower()
if self.userinfo is not None:
self.userinfo = normalise_unreserved(self.userinfo)
if self.host is not None:
self.host = normalise_unreserved(self.host.lower())
if self.port == '':
self.port = None
elif self.port is not None:
if self.port == _default_port.get(self.scheme):
self.port = None
if self.host is not None and self.path == '':
self.path = '/'
self.path = normalise_unreserved(remove_dot_segments(self.path))
if self.query is not None:
self.query = normalise_unreserved(self.query)
if self.fragment is not None:
self.fragment = normalise_unreserved(self.fragment)
@property
def authority(self):
"""The authority part of the URI"""
if self.host is None:
return None
authority = self.host
if self.userinfo is not None:
authority = '%s@%s' % (self.userinfo, authority)
if self.port is not None:
authority = '%s:%s' % (authority, self.port)
return authority
@property
def hier_part(self):
"""The hierarchical part of the URI"""
authority = self.authority
if authority is None:
return self.path
else:
return '//%s%s' % (authority, self.path)
def __str__(self):
uri = '%s:%s' % (self.scheme, self.hier_part)
if self.query is not None:
uri += '?%s' % self.query
if self.fragment is not None:
uri += '#%s' % self.fragment
return uri
def __repr__(self):
return '%s(%r)' % (self.__class__.__name__, str(self))
def __eq__(self, other):
if isinstance(other, self.__class__):
return (self.scheme == other.scheme and
self.authority == other.authority and
self.path == other.path and
self.query == other.query and
self.fragment == other.fragment)
else:
return NotImplemented
def __ne__(self, other):
equal = self.__eq__(other)
if equal == NotImplemented:
return NotImplemented
else:
return not equal
def __hash__(self):
return hash((
self.scheme, self.authority, self.path, self.query, self.fragment))
def replace(self, **parts):
"""Replace one or more parts of the URI, returning the result."""
if not parts:
return self
baseparts = dict(
scheme=self.scheme,
userinfo=self.userinfo,
host=self.host,
port=self.port,
path=self.path,
query=self.query,
fragment=self.fragment)
baseparts.update(parts)
return self.__class__(**baseparts)
def resolve(self, reference):
"""Resolve the given URI reference relative to this URI.
Uses the rules from Section 5.2 of RFC 3986 to resolve the new
URI.
"""
# If the reference is a full URI, then return it as is.
try:
return self.__class__(reference)
except InvalidURIError:
pass
match = relative_ref_pat.match(reference)
if match is None:
raise InvalidURIError("Invalid relative reference")
parts = dict(scheme=self.scheme)
authority = match.group('authority')
if authority is not None:
parts['userinfo'] = match.group('userinfo')
parts['host'] = match.group('host')
parts['port'] = match.group('port')
# Skip over the //authority part
parts['path'] = remove_dot_segments(
match.group('relativepart')[2+len(authority):])
parts['query'] = match.group('query')
else:
path = match.group('relativepart')
query = match.group('query')
if path == '':
parts['path'] = self.path
if query is not None:
parts['query'] = query
else:
parts['query'] = self.query
else:
if path.startswith('/'):
parts['path'] = remove_dot_segments(path)
else:
parts['path'] = merge(self.path, path,
has_authority=self.host is not None)
parts['path'] = remove_dot_segments(parts['path'])
parts['query'] = query
parts['userinfo'] = self.userinfo
parts['host'] = self.host
parts['port'] = self.port
parts['fragment'] = match.group('fragment')
return self.__class__(**parts)
def append(self, path):
"""Append the given path to this URI.
The path must not start with a slash, but a slash is added to
base URI (before appending the path), in case it doesn't end
with a slash.
"""
assert not path.startswith('/')
return self.ensureSlash().resolve(path)
def contains(self, other):
"""Returns True if the URI 'other' is contained by this one."""
if (self.scheme != other.scheme or
self.authority != other.authority):
return False
if self.path == other.path:
return True
basepath = self.path
if not basepath.endswith('/'):
basepath += '/'
otherpath = other.path
if not otherpath.endswith('/'):
otherpath += '/'
return otherpath.startswith(basepath)
def underDomain(self, domain):
"""Return True if the given domain name a parent of the URL's host."""
if len(domain) == 0:
return True
our_segments = self.host.split('.')
domain_segments = domain.split('.')
return our_segments[-len(domain_segments):] == domain_segments
def ensureSlash(self):
"""Return a URI with the path normalised to end with a slash."""
if self.path.endswith('/'):
return self
else:
return self.replace(path=self.path + '/')
def ensureNoSlash(self):
"""Return a URI with the path normalised to not end with a slash."""
if self.path.endswith('/'):
return self.replace(path=self.path.rstrip('/'))
else:
return self
# Regular expression for finding URIs in a body of text:
#
# From RFC 3986 ABNF for URIs:
#
# URI = scheme ":" hier-part [ "?" query ] [ "#" fragment ]
# hier-part = "//" authority path-abempty
# / path-absolute
# / path-rootless
# / path-empty
#
# authority = [ userinfo "@" ] host [ ":" port ]
# userinfo = *( unreserved / pct-encoded / sub-delims / ":" )
# host = IP-literal / IPv4address / reg-name
# reg-name = *( unreserved / pct-encoded / sub-delims )
# port = *DIGIT
#
# path-abempty = *( "/" segment )
# path-absolute = "/" [ segment-nz *( "/" segment ) ]
# path-rootless = segment-nz *( "/" segment )
# path-empty = 0<pchar>
#
# segment = *pchar
# segment-nz = 1*pchar
# pchar = unreserved / pct-encoded / sub-delims / ":" / "@"
#
# query = *( pchar / "/" / "?" )
# fragment = *( pchar / "/" / "?" )
#
# unreserved = ALPHA / DIGIT / "-" / "." / "_" / "~"
# pct-encoded = "%" HEXDIG HEXDIG
# sub-delims = "!" / "$" / "&" / "'" / "(" / ")"
# / "*" / "+" / "," / ";" / "="
#
# We only match a set of known scheme names. We don't handle
# IP-literal either.
#
# We will simplify "unreserved / pct-encoded / sub-delims" as the
# following regular expression:
# [-a-zA-Z0-9._~%!$&'()*+,;=]
#
# We also require that the path-rootless form not begin with a
# colon to avoid matching strings like "http::foo" (to avoid bug
# #40255).
#
# The path-empty pattern is not matched either, due to false
# positives.
#
# Some allowed URI punctuation characters will be trimmed if they
# appear at the end of the URI since they may be incidental in the
# flow of the text.
#
# apport has at one time produced query strings containing sqaure
# braces (that are not percent-encoded). In RFC 2986 they seem to be
# allowed by section 2.2 "Reserved Characters", yet section 3.4
# "Query" appears to provide a strict definition of the query string
# that would forbid square braces. Either way, links with
# non-percent-encoded square braces are being used on Launchpad so
# it's probably best to accomodate them.
possible_uri_re = r'''
\b
(?:about|gopher|http|https|sftp|news|ftp|mailto|file|irc|jabber|xmpp)
:
(?:
(?:
# "//" authority path-abempty
//
(?: # userinfo
[%(unreserved)s:]*
@
)?
(?: # host
\d+\.\d+\.\d+\.\d+ |
[%(unreserved)s]*
)
(?: # port
: \d*
)?
(?: / [%(unreserved)s:@]* )*
) | (?:
# path-absolute
/
(?: [%(unreserved)s:@]+
(?: / [%(unreserved)s:@]* )* )?
) | (?:
# path-rootless
[%(unreserved)s@]
[%(unreserved)s:@]*
(?: / [%(unreserved)s:@]* )*
)
)
(?: # query
\?
[%(unreserved)s:@/\?\[\]]*
)?
(?: # fragment
\#
[%(unreserved)s:@/\?]*
)?
''' % {'unreserved': "-a-zA-Z0-9._~%!$&'()*+,;="}
possible_uri_pat = re.compile(possible_uri_re, re.IGNORECASE | re.VERBOSE)
uri_trailers_pat = re.compile(r'([,.?:);>]+)$')
def find_uris_in_text(text):
"""Scan a block of text for URIs, and yield the ones found."""
for match in possible_uri_pat.finditer(text):
uri_string = match.group()
# remove characters from end of URI that are not likely to be
# part of the URI.
uri_string = uri_trailers_pat.sub('', uri_string)
try:
uri = URI(uri_string)
except InvalidURIError:
continue
yield uri