test/test_501_paramiko.py

   1 #!/usr/bin/env python
   2 # -*- coding:utf-8 -*-
   3
   4 #  Copyright (C) 2010-2018  CEA/DEN
   5 #
   6 #  This library is free software; you can redistribute it and/or
   7 #  modify it under the terms of the GNU Lesser General Public
   8 #  License as published by the Free Software Foundation; either
   9 #  version 2.1 of the License.
  10 #
  11 #  This library is distributed in the hope that it will be useful,
  12 #  but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14 #  Lesser General Public License for more details.
  15 #
  16 #  You should have received a copy of the GNU Lesser General Public
  17 #  License along with this library; if not, write to the Free Software
  18 #  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  19
  20
  21 """
  22 to set id_rsa from/to reflexive on local machine:
  23
  24     @is231761/home/wambeke/.ssh>ssh wambeke@is231761
  25     Password:
  26     Last login: Thu Jun  7 13:34:07 2018 from is231761.intra.cea.fr
  27     @is231761/home/wambeke>exit
  28     déconnexion
  29
  30     @is231761/home/wambeke/.ssh> ssh-keygen
  31     Generating public/private rsa key pair.
  32     Enter file in which to save the key (/home/wambeke/.ssh/id_rsa):
  33     Enter passphrase (empty for no passphrase):
  34     Enter same passphrase again:
  35     Your identification has been saved in /home/wambeke/.ssh/id_rsa.
  36     Your public key has been saved in /home/wambeke/.ssh/id_rsa.pub.
  37     The key fingerprint is:
  38     SHA256:V0IU/wkuCRw42rA5bHFgdJlzDx9EIJyWIBrkzkL3GNA wambeke@is231761
  39     The key's randomart image is:
  40     +---[RSA 2048]----+
  41     |ooo.=+o*o=*.     |
  42
  43     |                 |
  44     +----[SHA256]-----+
  45
  46     @is231761/home/wambeke/.ssh> ls
  47     id_rsa  id_rsa.pub  known_hosts
  48     @is231761/home/wambeke/.ssh> rm known_hosts
  49     @is231761/home/wambeke/.ssh> ls
  50     id_rsa  id_rsa.pub
  51
  52     @is231761/home/wambeke/.ssh> ssh wambeke@is231761
  53     The authenticity of host 'is231761 (127.0.0.1)' can't be established.
  54     ECDSA key fingerprint is SHA256:QvrU7Abrbily0bzMjYbRPeKCxDkXT9rQ6pSpcm+yFN4.
  55     ECDSA key fingerprint is MD5:6c:95:b7:c7:cd:de:c5:07:8b:3a:9b:14:d1:69:6b:c6.
  56     Are you sure you want to continue connecting (yes/no)? yes
  57     Warning: Permanently added 'is231761' (ECDSA) to the list of known hosts.
  58     Password:
  59     Last login: Thu Jun  7 13:35:07 2018 from is231761.intra.cea.fr
  60     @is231761/home/wambeke>exit
  61     déconnexion
  62     Connection to is231761 closed.
  63
  64
  65     @is231761/home/wambeke/.ssh> lst
  66     total 124K
  67     -rw-r--r--   1 wambeke lgls  170  7 juin  13:36 known_hosts
  68     drwx------   2 wambeke lgls 4,0K  7 juin  13:36 .
  69     -rw-r--r--   1 wambeke lgls  398  7 juin  13:35 id_rsa.pub
  70     -rw-------   1 wambeke lgls 1,7K  7 juin  13:35 id_rsa
  71     drwxr-xr-x 182 wambeke lmpe 104K  6 juin  13:39 ..
  72
  73
  74
  75     @is231761/home/wambeke/.ssh> ssh-copy-id -i ~/.ssh/id_rsa.pub is231761
  76     /usr/bin/ssh-copy-id: INFO: Source of key(s) to be installed: "/home/wambeke/.ssh/id_rsa.pub"
  77     /usr/bin/ssh-copy-id: INFO: attempting to log in with the new key(s), to filter out any that are already installed
  78     /usr/bin/ssh-copy-id: INFO: 1 key(s) remain to be installed -- if you are prompted now it is to install the new keys
  79     Password:
  80
  81     Number of key(s) added: 1
  82
  83     Now try logging into the machine, with:   "ssh 'is231761'"
  84     and check to make sure that only the key(s) you wanted were added.
  85
  86     @is231761/home/wambeke/.ssh> ssh wambeke@is231761
  87     Last login: Thu Jun  7 13:36:42 2018 from is231761.intra.cea.fr
  88     @is231761/home/wambeke>exit
  89     déconnexion
  90     Connection to is231761 closed.
  91
  92 """
  93
  94 import os
  95 import sys
  96 import unittest
  97 import getpass
  98
  99 verbose = False
 100
 101
 102 class TestCase(unittest.TestCase):
 103     "Test a paramiko connection" ""
 104
 105     def setLoggerParamiko(self):
 106         """to get logs of paramiko, useful if problems"""
 107         import logging as LOGI
 108
 109         loggerPrmk = LOGI.getLogger("paramiko")
 110         if len(loggerPrmk.handlers) != 0:
 111             print("logging.__file__ %s" % LOGI.__file__)
 112             print("logger paramiko have handler set yet, is a surprise")
 113             return
 114         if not verbose:
 115             # stay as it, null
 116             return
 117
 118         # set a paramiko logger verbose
 119         handler = LOGI.StreamHandler()
 120         msg = "create paramiko logger, with handler on stdout"
 121
 122         # handler = LOGI.MemoryHandler()
 123         # etc... https://docs.python.org/2/library/logging.handlers.html
 124         # msg = "create paramiko logger, with handler in memory"
 125
 126         # original frm from paramiko
 127         # frm = '%(levelname)-.3s [%(asctime)s.%(msecs)03d] thr=%(thread)-3d %(name)s: %(message)s' # noqa
 128         frm = "%(levelname)-5s :: %(asctime)s :: %(name)s :: %(message)s"
 129         handler.setFormatter(LOGI.Formatter(frm, "%y%m%d_%H%M%S"))
 130         loggerPrmk.addHandler(handler)
 131
 132         # logger is not notset but low, handlers needs setlevel greater
 133         loggerPrmk.setLevel(LOGI.DEBUG)
 134         handler.setLevel(LOGI.INFO)  # LOGI.DEBUG) # may be other one
 135
 136         loggerPrmk.info(msg)
 137
 138     """example from internet
 139   def fetch_netmask(self, hostname, port=22):
 140     private_key = os.path.expanduser('~/.ssh/id_rsa')
 141     connection = open_ssh_connection('wambeke', hostname, port=port, key=private_key)
 142
 143     get_netmask = ("ip -oneline -family inet address show | grep {}").format(hostname)
 144     stdin, stdout, stderr = connection.exec_command(get_netmask)
 145     address = parse_address(hostname, stdout)
 146     connection.close()
 147     return address
 148
 149   def open_ssh_connection(self, username, hostname, port=22, key=None):
 150     client = PK.SSHClient()
 151     client.set_missing_host_key_policy(PK.AutoAddPolicy())
 152     client.connect(hostname, port=port, timeout=5, username=username, key_filename=key)
 153     return client
 154   """
 155
 156     def test_000(self):
 157         self.setLoggerParamiko()
 158
 159     def test_010(self):
 160         # http://docs.paramiko.org/en/2.4/api/agent.html
 161
 162         try:
 163             import paramiko as PK
 164         except:
 165             print("\nproblem 'import paramiko', no tests")
 166             return
 167
 168         # port=22 # useless
 169         username = getpass.getuser()
 170         hostname = os.uname()[1]
 171         aFile = "/tmp/%s_test_paramiko.tmp" % username
 172         cmd = ("pwd; ls -alt {0}; cat {0}").format(aFile)
 173
 174         # connect
 175         client = PK.SSHClient()
 176         client.set_missing_host_key_policy(PK.AutoAddPolicy())
 177         # client.connect(hostname, username=username, password="xxxxx")
 178         # client.connect(hostname, username=username, passphrase="yyyy", key_filename="/home/wambeke/.ssh/id_rsa_satjobs_passphrase")
 179         # client.connect(hostname, username=username)
 180
 181         # timeout in seconds
 182         client.connect(hostname, username=username, timeout=1.0)
 183
 184         # obtain session
 185         session = client.get_transport().open_session()
 186         # Forward local agent
 187         PK.agent.AgentRequestHandler(session)
 188         # commands executed after this point will see the forwarded agent on the remote end.
 189
 190         # one api
 191         session.exec_command("date > %s" % aFile)
 192         cmd = ("pwd; ls -alt {0}; cat {0} && echo OK").format(aFile)
 193         # another api
 194         stdin, stdout, stderr = client.exec_command(cmd)
 195         output = stdout.read()
 196         if verbose:
 197             print("stdout:\n%s" % output)
 198         self.assertTrue(aFile in output)
 199         self.assertTrue("OK" in output)
 200         client.close()
 201
 202
 203 if __name__ == "__main__":
 204     # verbose = True # human eyes
 205     unittest.main(exit=False)
 206     pass