← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/osutils.py

(jameinel) Bug #581311,
 treat WSAECONNABORTED as ConnectionReset. (John A Meinel)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/osutils.py
53
53
    deprecated_in,
54
54
    )
55
55
 
56
 
from hashlib import (
57
 
    md5,
58
 
    sha1 as sha,
59
 
    )
 
56
# sha and md5 modules are deprecated in python2.6 but hashlib is available as
 
57
# of 2.5
 
58
if sys.version_info < (2, 5):
 
59
    import md5 as _mod_md5
 
60
    md5 = _mod_md5.new
 
61
    import sha as _mod_sha
 
62
    sha = _mod_sha.new
 
63
else:
 
64
    from hashlib import (
 
65
        md5,
 
66
        sha1 as sha,
 
67
        )
60
68
 
61
69
 
62
70
import bzrlib
88
96
        user_encoding = get_user_encoding()
89
97
        return [a.decode(user_encoding) for a in sys.argv[1:]]
90
98
    except UnicodeDecodeError:
91
 
        raise errors.BzrError("Parameter %r encoding is unsupported by %s "
92
 
            "application locale." % (a, user_encoding))
 
99
        raise errors.BzrError(("Parameter '%r' is unsupported by the current "
 
100
                                                            "encoding." % a))
93
101
 
94
102
 
95
103
def make_readonly(filename):
261
269
            else:
262
270
                rename_func(tmp_name, new)
263
271
    if failure_exc is not None:
264
 
        try:
265
 
            raise failure_exc[0], failure_exc[1], failure_exc[2]
266
 
        finally:
267
 
            del failure_exc
 
272
        raise failure_exc[0], failure_exc[1], failure_exc[2]
268
273
 
269
274
 
270
275
# In Python 2.4.2 and older, os.path.abspath and os.path.realpath
387
392
# These were already lazily imported into local scope
388
393
# mkdtemp = tempfile.mkdtemp
389
394
# rmtree = shutil.rmtree
390
 
lstat = os.lstat
391
 
fstat = os.fstat
392
 
 
393
 
def wrap_stat(st):
394
 
    return st
395
 
 
396
395
 
397
396
MIN_ABS_PATHLENGTH = 1
398
397
 
408
407
    getcwd = _win32_getcwd
409
408
    mkdtemp = _win32_mkdtemp
410
409
    rename = _win32_rename
411
 
    try:
412
 
        from bzrlib import _walkdirs_win32
413
 
    except ImportError:
414
 
        pass
415
 
    else:
416
 
        lstat = _walkdirs_win32.lstat
417
 
        fstat = _walkdirs_win32.fstat
418
 
        wrap_stat = _walkdirs_win32.wrap_stat
419
410
 
420
411
    MIN_ABS_PATHLENGTH = 3
421
412
 
2252
2243
            termios.tcsetattr(fd, termios.TCSADRAIN, settings)
2253
2244
        return ch
2254
2245
 
 
2246
 
2255
2247
if sys.platform == 'linux2':
2256
2248
    def _local_concurrency():
2257
 
        try:
2258
 
            return os.sysconf('SC_NPROCESSORS_ONLN')
2259
 
        except (ValueError, OSError, AttributeError):
2260
 
            return None
 
2249
        concurrency = None
 
2250
        prefix = 'processor'
 
2251
        for line in file('/proc/cpuinfo', 'rb'):
 
2252
            if line.startswith(prefix):
 
2253
                concurrency = int(line[line.find(':')+1:]) + 1
 
2254
        return concurrency
2261
2255
elif sys.platform == 'darwin':
2262
2256
    def _local_concurrency():
2263
2257
        return subprocess.Popen(['sysctl', '-n', 'hw.availcpu'],
2264
2258
                                stdout=subprocess.PIPE).communicate()[0]
2265
 
elif "bsd" in sys.platform:
 
2259
elif sys.platform[0:7] == 'freebsd':
2266
2260
    def _local_concurrency():
2267
2261
        return subprocess.Popen(['sysctl', '-n', 'hw.ncpu'],
2268
2262
                                stdout=subprocess.PIPE).communicate()[0]
2296
2290
    concurrency = os.environ.get('BZR_CONCURRENCY', None)
2297
2291
    if concurrency is None:
2298
2292
        try:
2299
 
            import multiprocessing
2300
 
        except ImportError:
2301
 
            # multiprocessing is only available on Python >= 2.6
2302
 
            try:
2303
 
                concurrency = _local_concurrency()
2304
 
            except (OSError, IOError):
2305
 
                pass
2306
 
        else:
2307
 
            concurrency = multiprocessing.cpu_count()
 
2293
            concurrency = _local_concurrency()
 
2294
        except (OSError, IOError):
 
2295
            pass
2308
2296
    try:
2309
2297
        concurrency = int(concurrency)
2310
2298
    except (TypeError, ValueError):
2381
2369
    except UnicodeDecodeError:
2382
2370
        raise errors.BzrError("Can't decode username as %s." % \
2383
2371
                user_encoding)
2384
 
    except ImportError, e:
2385
 
        if sys.platform != 'win32':
2386
 
            raise
2387
 
        if str(e) != 'No module named pwd':
2388
 
            raise
2389
 
        # https://bugs.launchpad.net/bzr/+bug/660174
2390
 
        # getpass.getuser() is unable to return username on Windows
2391
 
        # if there is no USERNAME environment variable set.
2392
 
        # That could be true if bzr is running as a service,
2393
 
        # e.g. running `bzr serve` as a service on Windows.
2394
 
        # We should not fail with traceback in this case.
2395
 
        username = u'UNKNOWN'
2396
2372
    return username
2397
2373
 
2398
2374
 
2427
2403
    except (ImportError, AttributeError):
2428
2404
        # Either the fcntl module or specific constants are not present
2429
2405
        pass
2430
 
 
2431
 
 
2432
 
def find_executable_on_path(name):
2433
 
    """Finds an executable on the PATH.
2434
 
    
2435
 
    On Windows, this will try to append each extension in the PATHEXT
2436
 
    environment variable to the name, if it cannot be found with the name
2437
 
    as given.
2438
 
    
2439
 
    :param name: The base name of the executable.
2440
 
    :return: The path to the executable found or None.
2441
 
    """
2442
 
    path = os.environ.get('PATH')
2443
 
    if path is None:
2444
 
        return None
2445
 
    path = path.split(os.pathsep)
2446
 
    if sys.platform == 'win32':
2447
 
        exts = os.environ.get('PATHEXT', '').split(os.pathsep)
2448
 
        exts = [ext.lower() for ext in exts]
2449
 
        base, ext = os.path.splitext(name)
2450
 
        if ext != '':
2451
 
            if ext.lower() not in exts:
2452
 
                return None
2453
 
            name = base
2454
 
            exts = [ext]
2455
 
    else:
2456
 
        exts = ['']
2457
 
    for ext in exts:
2458
 
        for d in path:
2459
 
            f = os.path.join(d, name) + ext
2460
 
            if os.access(f, os.X_OK):
2461
 
                return f
2462
 
    return None
2463
 
 
2464
 
 
2465
 
def _posix_is_local_pid_dead(pid):
2466
 
    """True if pid doesn't correspond to live process on this machine"""
2467
 
    try:
2468
 
        # Special meaning of unix kill: just check if it's there.
2469
 
        os.kill(pid, 0)
2470
 
    except OSError, e:
2471
 
        if e.errno == errno.ESRCH:
2472
 
            # On this machine, and really not found: as sure as we can be
2473
 
            # that it's dead.
2474
 
            return True
2475
 
        elif e.errno == errno.EPERM:
2476
 
            # exists, though not ours
2477
 
            return False
2478
 
        else:
2479
 
            mutter("os.kill(%d, 0) failed: %s" % (pid, e))
2480
 
            # Don't really know.
2481
 
            return False
2482
 
    else:
2483
 
        # Exists and our process: not dead.
2484
 
        return False
2485
 
 
2486
 
if sys.platform == "win32":
2487
 
    is_local_pid_dead = win32utils.is_local_pid_dead
2488
 
else:
2489
 
    is_local_pid_dead = _posix_is_local_pid_dead