/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/urlutils.py

  • Committer: Robert Collins
  • Date: 2010-05-06 11:08:10 UTC
  • mto: This revision was merged to the branch mainline in revision 5223.
  • Revision ID: robertc@robertcollins.net-20100506110810-h3j07fh5gmw54s25
Cleaner matcher matching revised unlocking protocol.

Show diffs side-by-side

added added

removed removed

Lines of Context:
16
16
 
17
17
"""A collection of function for handling URL operations."""
18
18
 
19
 
from __future__ import absolute_import
20
 
 
21
19
import os
22
20
import re
23
21
import sys
24
22
 
25
 
try:
26
 
    import urlparse
27
 
except ImportError:
28
 
    from urllib import parse as urlparse
 
23
from bzrlib.lazy_import import lazy_import
 
24
lazy_import(globals(), """
 
25
from posixpath import split as _posix_split, normpath as _posix_normpath
 
26
import urllib
 
27
import urlparse
29
28
 
30
 
from . import (
 
29
from bzrlib import (
31
30
    errors,
32
31
    osutils,
33
32
    )
34
 
 
35
 
from .lazy_import import lazy_import
36
 
lazy_import(globals(), """
37
 
from posixpath import split as _posix_split
38
33
""")
39
34
 
40
 
from .sixish import (
41
 
    int2byte,
42
 
    PY3,
43
 
    text_type,
44
 
    unichr,
45
 
    )
46
 
 
47
 
 
48
 
class InvalidURL(errors.PathError):
49
 
 
50
 
    _fmt = 'Invalid url supplied to transport: "%(path)s"%(extra)s'
51
 
 
52
 
 
53
 
class InvalidURLJoin(errors.PathError):
54
 
 
55
 
    _fmt = "Invalid URL join request: %(reason)s: %(base)r + %(join_args)r"
56
 
 
57
 
    def __init__(self, reason, base, join_args):
58
 
        self.reason = reason
59
 
        self.base = base
60
 
        self.join_args = join_args
61
 
        errors.PathError.__init__(self, base, reason)
62
 
 
63
 
 
64
 
class InvalidRebaseURLs(errors.PathError):
65
 
 
66
 
    _fmt = "URLs differ by more than path: %(from_)r and %(to)r"
67
 
 
68
 
    def __init__(self, from_, to):
69
 
        self.from_ = from_
70
 
        self.to = to
71
 
        errors.PathError.__init__(
72
 
            self, from_, 'URLs differ by more than path.')
73
 
 
74
35
 
75
36
def basename(url, exclude_trailing_slash=True):
76
37
    """Return the last component of a URL.
99
60
    return split(url, exclude_trailing_slash=exclude_trailing_slash)[0]
100
61
 
101
62
 
102
 
if PY3:
103
 
    quote_from_bytes = urlparse.quote_from_bytes
104
 
    quote = urlparse.quote
105
 
    unquote_to_bytes = urlparse.unquote_to_bytes
106
 
else:
107
 
    # Private copies of quote and unquote, copied from Python's urllib module
108
 
    # because urllib unconditionally imports socket, which imports ssl.
109
 
 
110
 
    always_safe = ('ABCDEFGHIJKLMNOPQRSTUVWXYZ'
111
 
                   'abcdefghijklmnopqrstuvwxyz'
112
 
                   '0123456789' '_.-')
113
 
    _safe_map = {}
114
 
    for i, c in zip(range(256), ''.join(map(chr, range(256)))):
115
 
        _safe_map[c] = c if (
116
 
            i < 128 and c in always_safe) else '%{0:02X}'.format(i)
117
 
    _safe_quoters = {}
118
 
 
119
 
    def quote_from_bytes(s, safe='/'):
120
 
        """quote('abc def') -> 'abc%20def'
121
 
 
122
 
        Each part of a URL, e.g. the path info, the query, etc., has a
123
 
        different set of reserved characters that must be quoted.
124
 
 
125
 
        RFC 2396 Uniform Resource Identifiers (URI): Generic Syntax lists
126
 
        the following reserved characters.
127
 
 
128
 
        reserved    = ";" | "/" | "?" | ":" | "@" | "&" | "=" | "+" |
129
 
                      "$" | ","
130
 
 
131
 
        Each of these characters is reserved in some component of a URL,
132
 
        but not necessarily in all of them.
133
 
 
134
 
        By default, the quote function is intended for quoting the path
135
 
        section of a URL.  Thus, it will not encode '/'.  This character
136
 
        is reserved, but in typical usage the quote function is being
137
 
        called on a path where the existing slash characters are used as
138
 
        reserved characters.
139
 
        """
140
 
        # fastpath
141
 
        if not s:
142
 
            if s is None:
143
 
                raise TypeError('None object cannot be quoted')
144
 
            return s
145
 
        cachekey = (safe, always_safe)
146
 
        try:
147
 
            (quoter, safe) = _safe_quoters[cachekey]
148
 
        except KeyError:
149
 
            safe_map = _safe_map.copy()
150
 
            safe_map.update([(c, c) for c in safe])
151
 
            quoter = safe_map.__getitem__
152
 
            safe = always_safe + safe
153
 
            _safe_quoters[cachekey] = (quoter, safe)
154
 
        if not s.rstrip(safe):
155
 
            return s
156
 
        return ''.join(map(quoter, s))
157
 
 
158
 
    quote = quote_from_bytes
159
 
    unquote_to_bytes = urlparse.unquote
160
 
 
161
 
 
162
 
unquote = urlparse.unquote
163
 
 
164
 
 
165
63
def escape(relpath):
166
64
    """Escape relpath to be a valid url."""
167
 
    if not isinstance(relpath, str) and sys.version_info[0] == 2:
 
65
    if isinstance(relpath, unicode):
168
66
        relpath = relpath.encode('utf-8')
169
 
    return quote(relpath, safe='/~')
 
67
    # After quoting and encoding, the path should be perfectly
 
68
    # safe as a plain ASCII string, str() just enforces this
 
69
    return str(urllib.quote(relpath, safe='/~'))
170
70
 
171
71
 
172
72
def file_relpath(base, path):
176
76
    """
177
77
    if len(base) < MIN_ABS_FILEURL_LENGTH:
178
78
        raise ValueError('Length of base (%r) must equal or'
179
 
                         ' exceed the platform minimum url length (which is %d)' %
180
 
                         (base, MIN_ABS_FILEURL_LENGTH))
181
 
    base = osutils.normpath(local_path_from_url(base))
182
 
    path = osutils.normpath(local_path_from_url(path))
 
79
            ' exceed the platform minimum url length (which is %d)' %
 
80
            (base, MIN_ABS_FILEURL_LENGTH))
 
81
    base = local_path_from_url(base)
 
82
    path = local_path_from_url(path)
183
83
    return escape(osutils.relpath(base, path))
184
84
 
185
85
 
201
101
    first_path_slash = path.find('/')
202
102
    if first_path_slash == -1:
203
103
        return len(scheme), None
204
 
    return len(scheme), first_path_slash + m.start('path')
205
 
 
206
 
 
207
 
def is_url(url):
208
 
    """Tests whether a URL is in actual fact a URL."""
209
 
    return _url_scheme_re.match(url) is not None
 
104
    return len(scheme), first_path_slash+len(scheme)+3
210
105
 
211
106
 
212
107
def join(base, *args):
223
118
    """
224
119
    if not args:
225
120
        return base
226
 
    scheme_end, path_start = _find_scheme_and_separator(base)
227
 
    if scheme_end is None and path_start is None:
228
 
        path_start = 0
229
 
    elif path_start is None:
230
 
        path_start = len(base)
231
 
    path = base[path_start:]
 
121
    match = _url_scheme_re.match(base)
 
122
    scheme = None
 
123
    if match:
 
124
        scheme = match.group('scheme')
 
125
        path = match.group('path').split('/')
 
126
        if path[-1:] == ['']:
 
127
            # Strip off a trailing slash
 
128
            # This helps both when we are at the root, and when
 
129
            # 'base' has an extra slash at the end
 
130
            path = path[:-1]
 
131
    else:
 
132
        path = base.split('/')
 
133
 
 
134
    if scheme is not None and len(path) >= 1:
 
135
        host = path[:1]
 
136
        # the path should be represented as an abs path.
 
137
        # we know this must be absolute because of the presence of a URL scheme.
 
138
        remove_root = True
 
139
        path = [''] + path[1:]
 
140
    else:
 
141
        # create an empty host, but dont alter the path - this might be a
 
142
        # relative url fragment.
 
143
        host = []
 
144
        remove_root = False
 
145
 
232
146
    for arg in args:
233
 
        arg_scheme_end, arg_path_start = _find_scheme_and_separator(arg)
234
 
        if arg_scheme_end is None and arg_path_start is None:
235
 
            arg_path_start = 0
236
 
        elif arg_path_start is None:
237
 
            arg_path_start = len(arg)
238
 
        if arg_scheme_end is not None:
239
 
            base = arg
240
 
            path = arg[arg_path_start:]
241
 
            scheme_end = arg_scheme_end
242
 
            path_start = arg_path_start
 
147
        match = _url_scheme_re.match(arg)
 
148
        if match:
 
149
            # Absolute URL
 
150
            scheme = match.group('scheme')
 
151
            # this skips .. normalisation, making http://host/../../..
 
152
            # be rather strange.
 
153
            path = match.group('path').split('/')
 
154
            # set the host and path according to new absolute URL, discarding
 
155
            # any previous values.
 
156
            # XXX: duplicates mess from earlier in this function.  This URL
 
157
            # manipulation code needs some cleaning up.
 
158
            if scheme is not None and len(path) >= 1:
 
159
                host = path[:1]
 
160
                path = path[1:]
 
161
                # url scheme implies absolute path.
 
162
                path = [''] + path
 
163
            else:
 
164
                # no url scheme we take the path as is.
 
165
                host = []
243
166
        else:
 
167
            path = '/'.join(path)
244
168
            path = joinpath(path, arg)
245
 
    return base[:path_start] + path
 
169
            path = path.split('/')
 
170
    if remove_root and path[0:1] == ['']:
 
171
        del path[0]
 
172
    if host:
 
173
        # Remove the leading slash from the path, so long as it isn't also the
 
174
        # trailing slash, which we want to keep if present.
 
175
        if path and path[0] == '' and len(path) > 1:
 
176
            del path[0]
 
177
        path = host + path
 
178
 
 
179
    if scheme is None:
 
180
        return '/'.join(path)
 
181
    return scheme + '://' + '/'.join(path)
246
182
 
247
183
 
248
184
def joinpath(base, *args):
257
193
    """
258
194
    path = base.split('/')
259
195
    if len(path) > 1 and path[-1] == '':
260
 
        # If the path ends in a trailing /, remove it.
 
196
        #If the path ends in a trailing /, remove it.
261
197
        path.pop()
262
198
    for arg in args:
263
199
        if arg.startswith('/'):
267
203
                continue
268
204
            elif chunk == '..':
269
205
                if path == ['']:
270
 
                    raise InvalidURLJoin('Cannot go above root',
271
 
                                         base, args)
 
206
                    raise errors.InvalidURLJoin('Cannot go above root',
 
207
                            base, args)
272
208
                path.pop()
273
209
            else:
274
210
                path.append(chunk)
281
217
# jam 20060502 Sorted to 'l' because the final target is 'local_path_from_url'
282
218
def _posix_local_path_from_url(url):
283
219
    """Convert a url like file:///path/to/foo into /path/to/foo"""
284
 
    url = split_segment_parameters_raw(url)[0]
285
220
    file_localhost_prefix = 'file://localhost/'
286
221
    if url.startswith(file_localhost_prefix):
287
222
        path = url[len(file_localhost_prefix) - 1:]
288
223
    elif not url.startswith('file:///'):
289
 
        raise InvalidURL(
 
224
        raise errors.InvalidURL(
290
225
            url, 'local urls must start with file:/// or file://localhost/')
291
226
    else:
292
227
        path = url[len('file://'):]
301
236
    """
302
237
    # importing directly from posixpath allows us to test this
303
238
    # on non-posix platforms
304
 
    return 'file://' + escape(osutils._posix_abspath(path))
 
239
    return 'file://' + escape(_posix_normpath(
 
240
        osutils._posix_abspath(path)))
305
241
 
306
242
 
307
243
def _win32_local_path_from_url(url):
308
244
    """Convert a url like file:///C:/path/to/foo into C:/path/to/foo"""
309
245
    if not url.startswith('file://'):
310
 
        raise InvalidURL(url, 'local urls must start with file:///, '
311
 
                         'UNC path urls must start with file://')
312
 
    url = split_segment_parameters_raw(url)[0]
 
246
        raise errors.InvalidURL(url, 'local urls must start with file:///, '
 
247
                                     'UNC path urls must start with file://')
313
248
    # We strip off all 3 slashes
314
249
    win32_url = url[len('file:'):]
315
250
    # check for UNC path: //HOST/path
316
251
    if not win32_url.startswith('///'):
317
252
        if (win32_url[2] == '/'
318
 
                or win32_url[3] in '|:'):
319
 
            raise InvalidURL(url, 'Win32 UNC path urls'
320
 
                             ' have form file://HOST/path')
 
253
            or win32_url[3] in '|:'):
 
254
            raise errors.InvalidURL(url, 'Win32 UNC path urls'
 
255
                ' have form file://HOST/path')
321
256
        return unescape(win32_url)
322
257
 
323
258
    # allow empty paths so we can serve all roots
325
260
        return '/'
326
261
 
327
262
    # usual local path with drive letter
328
 
    if (len(win32_url) < 6
329
 
        or win32_url[3] not in ('abcdefghijklmnopqrstuvwxyz'
330
 
                                'ABCDEFGHIJKLMNOPQRSTUVWXYZ') or
331
 
        win32_url[4] not in '|:'
332
 
            or win32_url[5] != '/'):
333
 
        raise InvalidURL(url, 'Win32 file urls start with'
334
 
                         ' file:///x:/, where x is a valid drive letter')
 
263
    if (win32_url[3] not in ('abcdefghijklmnopqrstuvwxyz'
 
264
                             'ABCDEFGHIJKLMNOPQRSTUVWXYZ')
 
265
        or win32_url[4] not in  '|:'
 
266
        or win32_url[5] != '/'):
 
267
        raise errors.InvalidURL(url, 'Win32 file urls start with'
 
268
                ' file:///x:/, where x is a valid drive letter')
335
269
    return win32_url[3].upper() + u':' + unescape(win32_url[5:])
336
270
 
337
271
 
344
278
    # on non-win32 platform
345
279
    # FIXME: It turns out that on nt, ntpath.abspath uses nt._getfullpathname
346
280
    #       which actually strips trailing space characters.
347
 
    #       The worst part is that on linux ntpath.abspath has different
 
281
    #       The worst part is that under linux ntpath.abspath has different
348
282
    #       semantics, since 'nt' is not an available module.
349
283
    if path == '/':
350
284
        return 'file:///'
354
288
    if win32_path.startswith('//'):
355
289
        return 'file:' + escape(win32_path)
356
290
    return ('file:///' + str(win32_path[0].upper()) + ':' +
357
 
            escape(win32_path[2:]))
 
291
        escape(win32_path[2:]))
358
292
 
359
293
 
360
294
local_path_to_url = _posix_local_path_to_url
369
303
    MIN_ABS_FILEURL_LENGTH = WIN32_MIN_ABS_FILEURL_LENGTH
370
304
 
371
305
 
372
 
_url_scheme_re = re.compile('^(?P<scheme>[^:/]{2,}):(//)?(?P<path>.*)$')
373
 
_url_hex_escapes_re = re.compile('(%[0-9a-fA-F]{2})')
 
306
_url_scheme_re = re.compile(r'^(?P<scheme>[^:/]{2,})://(?P<path>.*)$')
 
307
_url_hex_escapes_re = re.compile(r'(%[0-9a-fA-F]{2})')
374
308
 
375
309
 
376
310
def _unescape_safe_chars(matchobj):
405
339
    :param url: Either a hybrid URL or a local path
406
340
    :return: A normalized URL which only includes 7-bit ASCII characters.
407
341
    """
408
 
    scheme_end, path_start = _find_scheme_and_separator(url)
409
 
    if scheme_end is None:
 
342
    m = _url_scheme_re.match(url)
 
343
    if not m:
410
344
        return local_path_to_url(url)
411
 
    prefix = url[:path_start]
412
 
    path = url[path_start:]
413
 
    if not isinstance(url, text_type):
 
345
    scheme = m.group('scheme')
 
346
    path = m.group('path')
 
347
    if not isinstance(url, unicode):
414
348
        for c in url:
415
349
            if c not in _url_safe_characters:
416
 
                raise InvalidURL(url, 'URLs can only contain specific'
417
 
                                 ' safe characters (not %r)' % c)
 
350
                raise errors.InvalidURL(url, 'URLs can only contain specific'
 
351
                                            ' safe characters (not %r)' % c)
418
352
        path = _url_hex_escapes_re.sub(_unescape_safe_chars, path)
419
 
        return str(prefix + ''.join(path))
 
353
        return str(scheme + '://' + ''.join(path))
420
354
 
421
355
    # We have a unicode (hybrid) url
422
356
    path_chars = list(path)
423
357
 
424
 
    for i in range(len(path_chars)):
 
358
    for i in xrange(len(path_chars)):
425
359
        if path_chars[i] not in _url_safe_characters:
 
360
            chars = path_chars[i].encode('utf-8')
426
361
            path_chars[i] = ''.join(
427
 
                ['%%%02X' % c for c in bytearray(path_chars[i].encode('utf-8'))])
 
362
                ['%%%02X' % ord(c) for c in path_chars[i].encode('utf-8')])
428
363
    path = ''.join(path_chars)
429
364
    path = _url_hex_escapes_re.sub(_unescape_safe_chars, path)
430
 
    return str(prefix + path)
 
365
    return str(scheme + '://' + path)
431
366
 
432
367
 
433
368
def relative_url(base, other):
450
385
    if base_scheme != other_scheme:
451
386
        return other
452
387
    elif sys.platform == 'win32' and base_scheme == 'file://':
453
 
        base_drive = base[base_first_slash + 1:base_first_slash + 3]
454
 
        other_drive = other[other_first_slash + 1:other_first_slash + 3]
 
388
        base_drive = base[base_first_slash+1:base_first_slash+3]
 
389
        other_drive = other[other_first_slash+1:other_first_slash+3]
455
390
        if base_drive != other_drive:
456
391
            return other
457
392
 
458
 
    base_path = base[base_first_slash + 1:]
459
 
    other_path = other[other_first_slash + 1:]
 
393
    base_path = base[base_first_slash+1:]
 
394
    other_path = other[other_first_slash+1:]
460
395
 
461
396
    if base_path.endswith('/'):
462
397
        base_path = base_path[:-1]
486
421
    """On win32 the drive letter needs to be added to the url base."""
487
422
    # Strip off the drive letter
488
423
    # path is currently /C:/foo
489
 
    if len(path) < 4 or path[2] not in ':|' or path[3] != '/':
490
 
        raise InvalidURL(url_base + path,
491
 
                         'win32 file:/// paths need a drive letter')
492
 
    url_base += path[0:3]  # file:// + /C:
493
 
    path = path[3:]  # /foo
 
424
    if len(path) < 3 or path[2] not in ':|' or path[3] != '/':
 
425
        raise errors.InvalidURL(url_base + path,
 
426
            'win32 file:/// paths need a drive letter')
 
427
    url_base += path[0:3] # file:// + /C:
 
428
    path = path[3:] # /foo
494
429
    return url_base, path
495
430
 
496
431
 
501
436
    :param exclude_trailing_slash: Strip off a final '/' if it is part
502
437
        of the path (but not if it is part of the protocol specification)
503
438
 
504
 
    :return: (parent_url, child_dir).  child_dir may be the empty string if
505
 
        we're at the root.
 
439
    :return: (parent_url, child_dir).  child_dir may be the empty string if we're at
 
440
        the root.
506
441
    """
507
442
    scheme_loc, first_path_slash = _find_scheme_and_separator(url)
508
443
 
518
453
            return url, ''
519
454
 
520
455
    # We have a fully defined path
521
 
    url_base = url[:first_path_slash]  # http://host, file://
522
 
    path = url[first_path_slash:]  # /file/foo
 
456
    url_base = url[:first_path_slash] # http://host, file://
 
457
    path = url[first_path_slash:] # /file/foo
523
458
 
524
459
    if sys.platform == 'win32' and url.startswith('file:///'):
525
460
        # Strip off the drive letter
534
469
    return url_base + head, tail
535
470
 
536
471
 
537
 
def split_segment_parameters_raw(url):
538
 
    """Split the subsegment of the last segment of a URL.
539
 
 
540
 
    :param url: A relative or absolute URL
541
 
    :return: (url, subsegments)
542
 
    """
543
 
    # GZ 2011-11-18: Dodgy removing the terminal slash like this, function
544
 
    #                operates on urls not url+segments, and Transport classes
545
 
    #                should not be blindly adding slashes in the first place.
546
 
    lurl = strip_trailing_slash(url)
547
 
    # Segments begin at first comma after last forward slash, if one exists
548
 
    segment_start = lurl.find(",", lurl.rfind("/") + 1)
549
 
    if segment_start == -1:
550
 
        return (url, [])
551
 
    return (lurl[:segment_start],
552
 
            [str(s) for s in lurl[segment_start + 1:].split(",")])
553
 
 
554
 
 
555
 
def split_segment_parameters(url):
556
 
    """Split the segment parameters of the last segment of a URL.
557
 
 
558
 
    :param url: A relative or absolute URL
559
 
    :return: (url, segment_parameters)
560
 
    """
561
 
    (base_url, subsegments) = split_segment_parameters_raw(url)
562
 
    parameters = {}
563
 
    for subsegment in subsegments:
564
 
        (key, value) = subsegment.split("=", 1)
565
 
        if not isinstance(key, str):
566
 
            raise TypeError(key)
567
 
        if not isinstance(value, str):
568
 
            raise TypeError(value)
569
 
        parameters[key] = value
570
 
    return (base_url, parameters)
571
 
 
572
 
 
573
 
def join_segment_parameters_raw(base, *subsegments):
574
 
    """Create a new URL by adding subsegments to an existing one.
575
 
 
576
 
    This adds the specified subsegments to the last path in the specified
577
 
    base URL. The subsegments should be bytestrings.
578
 
 
579
 
    :note: You probably want to use join_segment_parameters instead.
580
 
    """
581
 
    if not subsegments:
582
 
        return base
583
 
    for subsegment in subsegments:
584
 
        if not isinstance(subsegment, str):
585
 
            raise TypeError("Subsegment %r is not a bytestring" % subsegment)
586
 
        if "," in subsegment:
587
 
            raise InvalidURLJoin(", exists in subsegments",
588
 
                                 base, subsegments)
589
 
    return ",".join((base,) + subsegments)
590
 
 
591
 
 
592
 
def join_segment_parameters(url, parameters):
593
 
    """Create a new URL by adding segment parameters to an existing one.
594
 
 
595
 
    The parameters of the last segment in the URL will be updated; if a
596
 
    parameter with the same key already exists it will be overwritten.
597
 
 
598
 
    :param url: A URL, as string
599
 
    :param parameters: Dictionary of parameters, keys and values as bytestrings
600
 
    """
601
 
    (base, existing_parameters) = split_segment_parameters(url)
602
 
    new_parameters = {}
603
 
    new_parameters.update(existing_parameters)
604
 
    for key, value in parameters.items():
605
 
        if not isinstance(key, str):
606
 
            raise TypeError("parameter key %r is not a str" % key)
607
 
        if not isinstance(value, str):
608
 
            raise TypeError("parameter value %r for %r is not a str" %
609
 
                            (value, key))
610
 
        if "=" in key:
611
 
            raise InvalidURLJoin("= exists in parameter key", url,
612
 
                                 parameters)
613
 
        new_parameters[key] = value
614
 
    return join_segment_parameters_raw(
615
 
        base, *["%s=%s" % item for item in sorted(new_parameters.items())])
616
 
 
617
 
 
618
472
def _win32_strip_local_trailing_slash(url):
619
473
    """Strip slashes after the drive letter"""
620
474
    if len(url) > WIN32_MIN_ABS_FILEURL_LENGTH:
656
510
        # so just chop off the last character
657
511
        return url[:-1]
658
512
 
659
 
    if first_path_slash is None or first_path_slash == len(url) - 1:
 
513
    if first_path_slash is None or first_path_slash == len(url)-1:
660
514
        # Don't chop off anything if the only slash is the path
661
515
        # separating slash
662
516
        return url
670
524
    This returns a Unicode path from a URL
671
525
    """
672
526
    # jam 20060427 URLs are supposed to be ASCII only strings
673
 
    #       If they are passed in as unicode, unquote
 
527
    #       If they are passed in as unicode, urllib.unquote
674
528
    #       will return a UNICODE string, which actually contains
675
529
    #       utf-8 bytes. So we have to ensure that they are
676
530
    #       plain ASCII strings, or the final .decode will
677
531
    #       try to encode the UNICODE => ASCII, and then decode
678
532
    #       it into utf-8.
 
533
    try:
 
534
        url = str(url)
 
535
    except UnicodeError, e:
 
536
        raise errors.InvalidURL(url, 'URL was not a plain ASCII url: %s' % (e,))
679
537
 
680
 
    if PY3:
681
 
        if isinstance(url, text_type):
682
 
            try:
683
 
                url.encode("ascii")
684
 
            except UnicodeError as e:
685
 
                raise InvalidURL(
686
 
                    url, 'URL was not a plain ASCII url: %s' % (e,))
687
 
        return urlparse.unquote(url)
688
 
    else:
689
 
        if isinstance(url, text_type):
690
 
            try:
691
 
                url = url.encode("ascii")
692
 
            except UnicodeError as e:
693
 
                raise InvalidURL(
694
 
                    url, 'URL was not a plain ASCII url: %s' % (e,))
695
 
        unquoted = unquote(url)
696
 
        try:
697
 
            unicode_path = unquoted.decode('utf-8')
698
 
        except UnicodeError as e:
699
 
            raise InvalidURL(
700
 
                url, 'Unable to encode the URL as utf-8: %s' % (e,))
701
 
        return unicode_path
 
538
    unquoted = urllib.unquote(url)
 
539
    try:
 
540
        unicode_path = unquoted.decode('utf-8')
 
541
    except UnicodeError, e:
 
542
        raise errors.InvalidURL(url, 'Unable to encode the URL as utf-8: %s' % (e,))
 
543
    return unicode_path
702
544
 
703
545
 
704
546
# These are characters that if escaped, should stay that way
705
547
_no_decode_chars = ';/?:@&=+$,#'
706
548
_no_decode_ords = [ord(c) for c in _no_decode_chars]
707
549
_no_decode_hex = (['%02x' % o for o in _no_decode_ords]
708
 
                  + ['%02X' % o for o in _no_decode_ords])
709
 
_hex_display_map = dict(([('%02x' % o, int2byte(o)) for o in range(256)]
710
 
                         + [('%02X' % o, int2byte(o)) for o in range(256)]))
711
 
# These entries get mapped to themselves
712
 
_hex_display_map.update((hex, b'%' + hex.encode('ascii'))
713
 
                        for hex in _no_decode_hex)
 
550
                + ['%02X' % o for o in _no_decode_ords])
 
551
_hex_display_map = dict(([('%02x' % o, chr(o)) for o in range(256)]
 
552
                    + [('%02X' % o, chr(o)) for o in range(256)]))
 
553
#These entries get mapped to themselves
 
554
_hex_display_map.update((hex,'%'+hex) for hex in _no_decode_hex)
714
555
 
715
556
# These characters shouldn't be percent-encoded, and it's always safe to
716
557
# unencode them if they are.
717
558
_url_dont_escape_characters = set(
718
 
    "abcdefghijklmnopqrstuvwxyz"  # Lowercase alpha
719
 
    "ABCDEFGHIJKLMNOPQRSTUVWXYZ"  # Uppercase alpha
720
 
    "0123456789"  # Numbers
721
 
    "-._~"  # Unreserved characters
 
559
   "abcdefghijklmnopqrstuvwxyz" # Lowercase alpha
 
560
   "ABCDEFGHIJKLMNOPQRSTUVWXYZ" # Uppercase alpha
 
561
   "0123456789" # Numbers
 
562
   "-._~"  # Unreserved characters
722
563
)
723
564
 
724
565
# These characters should not be escaped
725
566
_url_safe_characters = set(
726
 
    "abcdefghijklmnopqrstuvwxyz"  # Lowercase alpha
727
 
    "ABCDEFGHIJKLMNOPQRSTUVWXYZ"  # Uppercase alpha
728
 
    "0123456789"  # Numbers
729
 
    "_.-!~*'()"  # Unreserved characters
730
 
    "/;?:@&=+$,"  # Reserved characters
731
 
    "%#"         # Extra reserved characters
 
567
   "abcdefghijklmnopqrstuvwxyz" # Lowercase alpha
 
568
   "ABCDEFGHIJKLMNOPQRSTUVWXYZ" # Uppercase alpha
 
569
   "0123456789" # Numbers
 
570
   "_.-!~*'()"  # Unreserved characters
 
571
   "/;?:@&=+$," # Reserved characters
 
572
   "%#"         # Extra reserved characters
732
573
)
733
574
 
734
 
 
735
 
def _unescape_segment_for_display(segment, encoding):
736
 
    """Unescape a segment for display.
737
 
 
738
 
    Helper for unescape_for_display
739
 
 
740
 
    :param url: A 7-bit ASCII URL
741
 
    :param encoding: The final output encoding
742
 
 
743
 
    :return: A unicode string which can be safely encoded into the
744
 
         specified encoding.
745
 
    """
746
 
    escaped_chunks = segment.split('%')
747
 
    escaped_chunks[0] = escaped_chunks[0].encode('utf-8')
748
 
    for j in range(1, len(escaped_chunks)):
749
 
        item = escaped_chunks[j]
750
 
        try:
751
 
            escaped_chunks[j] = _hex_display_map[item[:2]]
752
 
        except KeyError:
753
 
            # Put back the percent symbol
754
 
            escaped_chunks[j] = b'%' + \
755
 
                (item[:2].encode('utf-8') if PY3 else item[:2])
756
 
        except UnicodeDecodeError:
757
 
            escaped_chunks[j] = unichr(int(item[:2], 16)).encode('utf-8')
758
 
        escaped_chunks[j] += (item[2:].encode('utf-8') if PY3 else item[2:])
759
 
    unescaped = b''.join(escaped_chunks)
760
 
    try:
761
 
        decoded = unescaped.decode('utf-8')
762
 
    except UnicodeDecodeError:
763
 
        # If this path segment cannot be properly utf-8 decoded
764
 
        # after doing unescaping we will just leave it alone
765
 
        return segment
766
 
    else:
767
 
        try:
768
 
            decoded.encode(encoding)
769
 
        except UnicodeEncodeError:
770
 
            # If this chunk cannot be encoded in the local
771
 
            # encoding, then we should leave it alone
772
 
            return segment
773
 
        else:
774
 
            # Otherwise take the url decoded one
775
 
            return decoded
776
 
 
777
 
 
778
575
def unescape_for_display(url, encoding):
779
576
    """Decode what you can for a URL, so that we get a nice looking path.
780
577
 
802
599
 
803
600
    # Split into sections to try to decode utf-8
804
601
    res = url.split('/')
805
 
    for i in range(1, len(res)):
806
 
        res[i] = _unescape_segment_for_display(res[i], encoding)
 
602
    for i in xrange(1, len(res)):
 
603
        escaped_chunks = res[i].split('%')
 
604
        for j in xrange(1, len(escaped_chunks)):
 
605
            item = escaped_chunks[j]
 
606
            try:
 
607
                escaped_chunks[j] = _hex_display_map[item[:2]] + item[2:]
 
608
            except KeyError:
 
609
                # Put back the percent symbol
 
610
                escaped_chunks[j] = '%' + item
 
611
            except UnicodeDecodeError:
 
612
                escaped_chunks[j] = unichr(int(item[:2], 16)) + item[2:]
 
613
        unescaped = ''.join(escaped_chunks)
 
614
        try:
 
615
            decoded = unescaped.decode('utf-8')
 
616
        except UnicodeDecodeError:
 
617
            # If this path segment cannot be properly utf-8 decoded
 
618
            # after doing unescaping we will just leave it alone
 
619
            pass
 
620
        else:
 
621
            try:
 
622
                decoded.encode(encoding)
 
623
            except UnicodeEncodeError:
 
624
                # If this chunk cannot be encoded in the local
 
625
                # encoding, then we should leave it alone
 
626
                pass
 
627
            else:
 
628
                # Otherwise take the url decoded one
 
629
                res[i] = decoded
807
630
    return u'/'.join(res)
808
631
 
809
632
 
818
641
    is used without a path, e.g. c:foo-bar => foo-bar.
819
642
    If no /, path separator or : is found, the from_location is returned.
820
643
    """
821
 
    from_location, unused_params = split_segment_parameters(from_location)
822
644
    if from_location.find("/") >= 0 or from_location.find(os.sep) >= 0:
823
645
        return os.path.basename(from_location.rstrip("/\\"))
824
646
    else:
825
647
        sep = from_location.find(":")
826
648
        if sep > 0:
827
 
            return from_location[sep + 1:]
 
649
            return from_location[sep+1:]
828
650
        else:
829
651
            return from_location
830
652
 
847
669
    old_parsed = urlparse.urlparse(old_base)
848
670
    new_parsed = urlparse.urlparse(new_base)
849
671
    if (old_parsed[:2]) != (new_parsed[:2]):
850
 
        raise InvalidRebaseURLs(old_base, new_base)
 
672
        raise errors.InvalidRebaseURLs(old_base, new_base)
851
673
    return determine_relative_path(new_parsed[2],
852
674
                                   join(old_parsed[2], url))
853
675
 
858
680
    to_segments = osutils.splitpath(to_path)
859
681
    count = -1
860
682
    for count, (from_element, to_element) in enumerate(zip(from_segments,
861
 
                                                           to_segments)):
 
683
                                                       to_segments)):
862
684
        if from_element != to_element:
863
685
            break
864
686
    else:
871
693
    return osutils.pathjoin(*segments)
872
694
 
873
695
 
874
 
class URL(object):
875
 
    """Parsed URL."""
876
 
 
877
 
    def __init__(self, scheme, quoted_user, quoted_password, quoted_host,
878
 
                 port, quoted_path):
879
 
        self.scheme = scheme
880
 
        self.quoted_host = quoted_host
881
 
        self.host = unquote(self.quoted_host)
882
 
        self.quoted_user = quoted_user
883
 
        if self.quoted_user is not None:
884
 
            self.user = unquote(self.quoted_user)
885
 
        else:
886
 
            self.user = None
887
 
        self.quoted_password = quoted_password
888
 
        if self.quoted_password is not None:
889
 
            self.password = unquote(self.quoted_password)
890
 
        else:
891
 
            self.password = None
892
 
        self.port = port
893
 
        self.quoted_path = _url_hex_escapes_re.sub(
894
 
            _unescape_safe_chars, quoted_path)
895
 
        self.path = unquote(self.quoted_path)
896
 
 
897
 
    def __eq__(self, other):
898
 
        return (isinstance(other, self.__class__) and
899
 
                self.scheme == other.scheme and
900
 
                self.host == other.host and
901
 
                self.user == other.user and
902
 
                self.password == other.password and
903
 
                self.path == other.path)
904
 
 
905
 
    def __repr__(self):
906
 
        return "<%s(%r, %r, %r, %r, %r, %r)>" % (
907
 
            self.__class__.__name__,
908
 
            self.scheme, self.quoted_user, self.quoted_password,
909
 
            self.quoted_host, self.port, self.quoted_path)
910
 
 
911
 
    @classmethod
912
 
    def from_string(cls, url):
913
 
        """Create a URL object from a string.
914
 
 
915
 
        :param url: URL as bytestring
916
 
        """
917
 
        # GZ 2017-06-09: Actually validate ascii-ness
918
 
        # pad.lv/1696545: For the moment, accept both native strings and
919
 
        # unicode.
920
 
        if isinstance(url, str):
921
 
            pass
922
 
        elif isinstance(url, text_type):
923
 
            try:
924
 
                url = url.encode()
925
 
            except UnicodeEncodeError:
926
 
                raise InvalidURL(url)
927
 
        else:
928
 
            raise InvalidURL(url)
929
 
        (scheme, netloc, path, params,
930
 
         query, fragment) = urlparse.urlparse(url, allow_fragments=False)
931
 
        user = password = host = port = None
932
 
        if '@' in netloc:
933
 
            user, host = netloc.rsplit('@', 1)
934
 
            if ':' in user:
935
 
                user, password = user.split(':', 1)
936
 
        else:
937
 
            host = netloc
938
 
 
939
 
        if ':' in host and not (host[0] == '[' and host[-1] == ']'):
940
 
            # there *is* port
941
 
            host, port = host.rsplit(':', 1)
942
 
            if port:
943
 
                try:
944
 
                    port = int(port)
945
 
                except ValueError:
946
 
                    raise InvalidURL('invalid port number %s in url:\n%s' %
947
 
                                     (port, url))
948
 
            else:
949
 
                port = None
950
 
        if host != "" and host[0] == '[' and host[-1] == ']':  # IPv6
951
 
            host = host[1:-1]
952
 
 
953
 
        return cls(scheme, user, password, host, port, path)
954
 
 
955
 
    def __str__(self):
956
 
        netloc = self.quoted_host
957
 
        if ":" in netloc:
958
 
            netloc = "[%s]" % netloc
959
 
        if self.quoted_user is not None:
960
 
            # Note that we don't put the password back even if we
961
 
            # have one so that it doesn't get accidentally
962
 
            # exposed.
963
 
            netloc = '%s@%s' % (self.quoted_user, netloc)
964
 
        if self.port is not None:
965
 
            netloc = '%s:%d' % (netloc, self.port)
966
 
        return urlparse.urlunparse(
967
 
            (self.scheme, netloc, self.quoted_path, None, None, None))
968
 
 
969
 
    @staticmethod
970
 
    def _combine_paths(base_path, relpath):
971
 
        """Transform a Transport-relative path to a remote absolute path.
972
 
 
973
 
        This does not handle substitution of ~ but does handle '..' and '.'
974
 
        components.
975
 
 
976
 
        Examples::
977
 
 
978
 
            t._combine_paths('/home/sarah', 'project/foo')
979
 
                => '/home/sarah/project/foo'
980
 
            t._combine_paths('/home/sarah', '../../etc')
981
 
                => '/etc'
982
 
            t._combine_paths('/home/sarah', '/etc')
983
 
                => '/etc'
984
 
 
985
 
        :param base_path: base path
986
 
        :param relpath: relative url string for relative part of remote path.
987
 
        :return: urlencoded string for final path.
988
 
        """
989
 
        # pad.lv/1696545: For the moment, accept both native strings and
990
 
        # unicode.
991
 
        if isinstance(relpath, str):
992
 
            pass
993
 
        elif isinstance(relpath, text_type):
994
 
            try:
995
 
                relpath = relpath.encode()
996
 
            except UnicodeEncodeError:
997
 
                raise InvalidURL(relpath)
998
 
        else:
999
 
            raise InvalidURL(relpath)
1000
 
        relpath = _url_hex_escapes_re.sub(_unescape_safe_chars, relpath)
1001
 
        if relpath.startswith('/'):
1002
 
            base_parts = []
1003
 
        else:
1004
 
            base_parts = base_path.split('/')
1005
 
        if len(base_parts) > 0 and base_parts[-1] == '':
1006
 
            base_parts = base_parts[:-1]
1007
 
        for p in relpath.split('/'):
1008
 
            if p == '..':
1009
 
                if len(base_parts) == 0:
1010
 
                    # In most filesystems, a request for the parent
1011
 
                    # of root, just returns root.
1012
 
                    continue
1013
 
                base_parts.pop()
1014
 
            elif p == '.':
1015
 
                continue  # No-op
1016
 
            elif p != '':
1017
 
                base_parts.append(p)
1018
 
        path = '/'.join(base_parts)
1019
 
        if not path.startswith('/'):
1020
 
            path = '/' + path
1021
 
        return path
1022
 
 
1023
 
    def clone(self, offset=None):
1024
 
        """Return a new URL for a path relative to this URL.
1025
 
 
1026
 
        :param offset: A relative path, already urlencoded
1027
 
        :return: `URL` instance
1028
 
        """
1029
 
        if offset is not None:
1030
 
            relative = unescape(offset)
1031
 
            if sys.version_info[0] == 2:
1032
 
                relative = relative.encode('utf-8')
1033
 
            path = self._combine_paths(self.path, relative)
1034
 
            path = quote(path, safe="/~")
1035
 
        else:
1036
 
            path = self.quoted_path
1037
 
        return self.__class__(self.scheme, self.quoted_user,
1038
 
                              self.quoted_password, self.quoted_host, self.port,
1039
 
                              path)
1040
 
 
1041
696
 
1042
697
def parse_url(url):
1043
698
    """Extract the server address, the credentials and the path from the url.
1046
701
    chars.
1047
702
 
1048
703
    :param url: an quoted url
 
704
 
1049
705
    :return: (scheme, user, password, host, port, path) tuple, all fields
1050
706
        are unquoted.
1051
707
    """
1052
 
    parsed_url = URL.from_string(url)
1053
 
    return (parsed_url.scheme, parsed_url.user, parsed_url.password,
1054
 
            parsed_url.host, parsed_url.port, parsed_url.path)
 
708
    if isinstance(url, unicode):
 
709
        raise errors.InvalidURL('should be ascii:\n%r' % url)
 
710
    url = url.encode('utf-8')
 
711
    (scheme, netloc, path, params,
 
712
     query, fragment) = urlparse.urlparse(url, allow_fragments=False)
 
713
    user = password = host = port = None
 
714
    if '@' in netloc:
 
715
        user, host = netloc.rsplit('@', 1)
 
716
        if ':' in user:
 
717
            user, password = user.split(':', 1)
 
718
            password = urllib.unquote(password)
 
719
        user = urllib.unquote(user)
 
720
    else:
 
721
        host = netloc
 
722
 
 
723
    if ':' in host and not (host[0] == '[' and host[-1] == ']'): #there *is* port
 
724
        host, port = host.rsplit(':',1)
 
725
        try:
 
726
            port = int(port)
 
727
        except ValueError:
 
728
            raise errors.InvalidURL('invalid port number %s in url:\n%s' %
 
729
                                    (port, url))
 
730
    if host != "" and host[0] == '[' and host[-1] == ']': #IPv6
 
731
        host = host[1:-1]
 
732
 
 
733
    host = urllib.unquote(host)
 
734
    path = urllib.unquote(path)
 
735
 
 
736
    return (scheme, user, password, host, port, path)