/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 breezy/transport/ssh.py

  • Committer: Jelmer Vernooij
  • Date: 2017-05-21 19:09:26 UTC
  • mfrom: (6622.1.36 breezy)
  • Revision ID: jelmer@jelmer.uk-20170521190926-5vtz8xaf0e9ylrpc
Merge rename to breezy.

Show diffs side-by-side

added added

removed removed

Lines of Context:
breezy/transport/ssh.py
28
28
import sys
29
29
from binascii import hexlify
30
30
 
31
 
from bzrlib import (
 
31
from breezy import (
32
32
    config,
33
33
    errors,
34
34
    osutils,
47
47
 
48
48
 
49
49
SYSTEM_HOSTKEYS = {}
50
 
BZR_HOSTKEYS = {}
 
50
BRZ_HOSTKEYS = {}
51
51
 
52
52
 
53
53
_paramiko_version = getattr(paramiko, '__version_info__', (0, 0, 0))
63
63
    """Manager for manage SSH vendors."""
64
64
 
65
65
    # Note, although at first sign the class interface seems similar to
66
 
    # bzrlib.registry.Registry it is not possible/convenient to directly use
 
66
    # breezy.registry.Registry it is not possible/convenient to directly use
67
67
    # the Registry because the class just has "get()" interface instead of the
68
68
    # Registry's "get(key)".
69
69
 
85
85
        self._cached_ssh_vendor = None
86
86
 
87
87
    def _get_vendor_by_environment(self, environment=None):
88
 
        """Return the vendor or None based on BZR_SSH environment variable.
 
88
        """Return the vendor or None based on BRZ_SSH environment variable.
89
89
 
90
 
        :raises UnknownSSH: if the BZR_SSH environment variable contains
 
90
        :raises UnknownSSH: if the BRZ_SSH environment variable contains
91
91
                            unknown vendor name
92
92
        """
93
93
        if environment is None:
94
94
            environment = os.environ
95
 
        if 'BZR_SSH' in environment:
96
 
            vendor_name = environment['BZR_SSH']
 
95
        if 'BRZ_SSH' in environment:
 
96
            vendor_name = environment['BRZ_SSH']
97
97
            try:
98
98
                vendor = self._ssh_vendors[vendor_name]
99
99
            except KeyError:
134
134
            vendor = LSHSubprocessVendor()
135
135
        # As plink user prompts are not handled currently, don't auto-detect
136
136
        # it by inspection below, but keep this vendor detection for if a path
137
 
        # is given in BZR_SSH. See https://bugs.launchpad.net/bugs/414743
 
137
        # is given in BRZ_SSH. See https://bugs.launchpad.net/bugs/414743
138
138
        elif 'plink' in version and progname == 'plink':
139
139
            # Checking if "plink" was the executed argument as Windows
140
140
            # sometimes reports 'ssh -V' incorrectly with 'plink' in its
158
158
        """Find out what version of SSH is on the system.
159
159
 
160
160
        :raises SSHVendorNotFound: if no any SSH vendor is found
161
 
        :raises UnknownSSH: if the BZR_SSH environment variable contains
 
161
        :raises UnknownSSH: if the BRZ_SSH environment variable contains
162
162
                            unknown vendor name
163
163
        """
164
164
        if self._cached_ssh_vendor is None:
281
281
        return hexlify(s).upper()
282
282
 
283
283
    def _connect(self, username, password, host, port):
284
 
        global SYSTEM_HOSTKEYS, BZR_HOSTKEYS
 
284
        global SYSTEM_HOSTKEYS, BRZ_HOSTKEYS
285
285
 
286
286
        load_host_keys()
287
287
 
298
298
        if host in SYSTEM_HOSTKEYS and keytype in SYSTEM_HOSTKEYS[host]:
299
299
            our_server_key = SYSTEM_HOSTKEYS[host][keytype]
300
300
            our_server_key_hex = self._hexify(our_server_key.get_fingerprint())
301
 
        elif host in BZR_HOSTKEYS and keytype in BZR_HOSTKEYS[host]:
302
 
            our_server_key = BZR_HOSTKEYS[host][keytype]
 
301
        elif host in BRZ_HOSTKEYS and keytype in BRZ_HOSTKEYS[host]:
 
302
            our_server_key = BRZ_HOSTKEYS[host][keytype]
303
303
            our_server_key_hex = self._hexify(our_server_key.get_fingerprint())
304
304
        else:
305
305
            trace.warning('Adding %s host key for %s: %s'
306
306
                          % (keytype, host, server_key_hex))
307
 
            add = getattr(BZR_HOSTKEYS, 'add', None)
 
307
            add = getattr(BRZ_HOSTKEYS, 'add', None)
308
308
            if add is not None: # paramiko >= 1.X.X
309
 
                BZR_HOSTKEYS.add(host, keytype, server_key)
 
309
                BRZ_HOSTKEYS.add(host, keytype, server_key)
310
310
            else:
311
 
                BZR_HOSTKEYS.setdefault(host, {})[keytype] = server_key
 
311
                BRZ_HOSTKEYS.setdefault(host, {})[keytype] = server_key
312
312
            our_server_key = server_key
313
313
            our_server_key_hex = self._hexify(our_server_key.get_fingerprint())
314
314
            save_host_keys()
604
604
    Load system host keys (probably doesn't work on windows) and any
605
605
    "discovered" keys from previous sessions.
606
606
    """
607
 
    global SYSTEM_HOSTKEYS, BZR_HOSTKEYS
 
607
    global SYSTEM_HOSTKEYS, BRZ_HOSTKEYS
608
608
    try:
609
609
        SYSTEM_HOSTKEYS = paramiko.util.load_host_keys(
610
610
            os.path.expanduser('~/.ssh/known_hosts'))
612
612
        trace.mutter('failed to load system host keys: ' + str(e))
613
613
    bzr_hostkey_path = osutils.pathjoin(config.config_dir(), 'ssh_host_keys')
614
614
    try:
615
 
        BZR_HOSTKEYS = paramiko.util.load_host_keys(bzr_hostkey_path)
 
615
        BRZ_HOSTKEYS = paramiko.util.load_host_keys(bzr_hostkey_path)
616
616
    except IOError, e:
617
617
        trace.mutter('failed to load bzr host keys: ' + str(e))
618
618
        save_host_keys()
622
622
    """
623
623
    Save "discovered" host keys in $(config)/ssh_host_keys/.
624
624
    """
625
 
    global SYSTEM_HOSTKEYS, BZR_HOSTKEYS
 
625
    global SYSTEM_HOSTKEYS, BRZ_HOSTKEYS
626
626
    bzr_hostkey_path = osutils.pathjoin(config.config_dir(), 'ssh_host_keys')
627
627
    config.ensure_config_dir_exists()
628
628
 
629
629
    try:
630
630
        f = open(bzr_hostkey_path, 'w')
631
631
        f.write('# SSH host keys collected by bzr\n')
632
 
        for hostname, keys in BZR_HOSTKEYS.iteritems():
 
632
        for hostname, keys in BRZ_HOSTKEYS.iteritems():
633
633
            for keytype, key in keys.iteritems():
634
634
                f.write('%s %s %s\n' % (hostname, keytype, key.get_base64()))
635
635
        f.close()
718
718
 
719
719
        :param proc: a subprocess.Popen
720
720
        :param sock: if proc.stdin/out is a socket from a socketpair, then sock
721
 
            should bzrlib's half of that socketpair.  If not passed, proc's
 
721
            should breezy's half of that socketpair.  If not passed, proc's
722
722
            stdin/out is assumed to be ordinary pipes.
723
723
        """
724
724
        self.proc = proc