"""
-# pylint: disable-msg=C0103
+# pylint: disable=C0103
# C0103: Invalid name setup-ssh
import getpass
from ganeti import constants
from ganeti import errors
from ganeti import netutils
+from ganeti import ssconf
from ganeti import ssh
from ganeti import utils
"""
+class JoinCheckError(errors.GenericError):
+ """Exception raised if join check fails.
+
+ """
+
+
+class HostKeyVerificationError(errors.GenericError):
+ """Exception if host key do not match.
+
+ """
+
+
+class AuthError(errors.GenericError):
+ """Exception for authentication errors to hosts.
+
+ """
+
+
+def _CheckJoin(transport):
+ """Checks if a join is safe or dangerous.
+
+ Note: This function relies on the fact, that all
+ hosts have the same configuration at compile time of
+ Ganeti. So that the constants do not mismatch.
+
+ @param transport: The paramiko transport instance
+ @return: True if the join is safe; False otherwise
+
+ """
+ sftp = transport.open_sftp_client()
+ ss = ssconf.SimpleStore()
+ ss_cluster_name_path = ss.KeyToFilename(constants.SS_CLUSTER_NAME)
+
+ cluster_files = [
+ (constants.NODED_CERT_FILE, utils.ReadFile(constants.NODED_CERT_FILE)),
+ (ss_cluster_name_path, utils.ReadFile(ss_cluster_name_path)),
+ ]
+
+ for (filename, local_content) in cluster_files:
+ try:
+ remote_content = _ReadSftpFile(sftp, filename)
+ except IOError, err:
+ # Assume file does not exist. Paramiko's error reporting is lacking.
+ logging.debug("Failed to read %s: %s", filename, err)
+ continue
+
+ if remote_content != local_content:
+ logging.error("File %s doesn't match local version", filename)
+ return False
+
+ return True
+
+
def _RunRemoteCommand(transport, command):
"""Invokes and wait for the command over SSH.
_RunRemoteCommand(transport, "%s %s" % (constants.DAEMON_UTIL, command))
+def _ReadSftpFile(sftp, filename):
+ """Reads a file over sftp.
+
+ @param sftp: An open paramiko SFTP client
+ @param filename: The filename of the file to read
+ @return: The content of the file
+
+ """
+ remote_file = sftp.open(filename, "r")
+ try:
+ return remote_file.read()
+ finally:
+ remote_file.close()
+
+
def _WriteSftpFile(sftp, name, perm, data):
"""SFTPs data to a remote file.
try:
sftp.mkdir(auth_path, 0700)
- except IOError:
+ except IOError, err:
# Sadly paramiko doesn't provide errno or similiar
# so we can just assume that the path already exists
- logging.info("Path %s seems already to exist on remote node. Ignoring.",
- auth_path)
+ logging.info("Assuming directory %s on remote node exists: %s",
+ auth_path, err)
for name, (data, perm) in filemap.iteritems():
_WriteSftpFile(sftp, name, perm, data)
_InvokeDaemonUtil(transport, "reload-ssh-keys")
-def SetupNodeDaemon(transport):
- """Sets the node daemon up on the other side.
-
- @param transport: The paramiko transport instance
-
- """
- noded_cert = utils.ReadFile(constants.NODED_CERT_FILE)
-
- sftp = transport.open_sftp_client()
- _WriteSftpFile(sftp, constants.NODED_CERT_FILE, 0400, noded_cert)
-
- _InvokeDaemonUtil(transport, "start %s" % constants.NODED)
-
-
def ParseOptions():
"""Parses options passed to program.
"""
program = os.path.basename(sys.argv[0])
- parser = optparse.OptionParser(usage=("%prog [--debug|--verbose] <node>"
- " <node...>"), prog=program)
+ parser = optparse.OptionParser(usage=("%prog [--debug|--verbose] [--force]"
+ " <node> <node...>"), prog=program)
parser.add_option(cli.DEBUG_OPT)
parser.add_option(cli.VERBOSE_OPT)
parser.add_option(cli.NOSSH_KEYCHECK_OPT)
parser.add_option(optparse.Option("--key-type", dest="key_type",
choices=("rsa", "dsa"), default="dsa",
help="The private key type (rsa or dsa)"))
+ parser.add_option(optparse.Option("-j", "--force-join", dest="force_join",
+ action="store_true", default=False,
+ help="Force the join of the host"))
(options, args) = parser.parse_args()
return [private_key] + list(agent_keys)
+def _FormatFingerprint(fpr):
+ """Formats a paramiko.PKey.get_fingerprint() human readable.
+
+ @param fpr: The fingerprint to be formatted
+ @return: A human readable fingerprint
+
+ """
+ return ssh.FormatParamikoFingerprint(paramiko.util.hexify(fpr))
+
+
def LoginViaKeys(transport, username, keys):
"""Try to login on the given transport via a list of keys.
for private_key in keys:
try:
transport.auth_publickey(username, private_key)
- fpr = ":".join("%02x" % ord(i) for i in private_key.get_fingerprint())
+ fpr = _FormatFingerprint(private_key.get_fingerprint())
if isinstance(private_key, paramiko.AgentKey):
logging.debug("Authentication via the ssh-agent key %s", fpr)
else:
try:
return paramiko.util.load_host_keys(known_hosts)
except EnvironmentError:
- # We didn't found the path, silently ignore and return an empty dict
+ # We didn't find the path, silently ignore and return an empty dict
return {}
+def _VerifyServerKey(transport, host, host_keys):
+ """Verify the server keys.
+
+ @param transport: A paramiko.transport instance
+ @param host: Name of the host we verify
+ @param host_keys: Loaded host keys
+ @raises HostkeyVerificationError: When the host identify couldn't be verified
+
+ """
+
+ server_key = transport.get_remote_server_key()
+ keytype = server_key.get_name()
+
+ our_server_key = host_keys.get(host, {}).get(keytype, None)
+ if not our_server_key:
+ hexified_key = _FormatFingerprint(server_key.get_fingerprint())
+ msg = ("Unable to verify hostkey of host %s: %s. Do you want to accept"
+ " it?" % (host, hexified_key))
+
+ if cli.AskUser(msg):
+ our_server_key = server_key
+
+ if our_server_key != server_key:
+ raise HostKeyVerificationError("Unable to verify host identity")
+
+
def main():
"""Main routine.
# wants to log one more message, which fails as the file is closed
# now
+ success = True
+
for host in args:
- transport = paramiko.Transport((host, ssh_port))
- transport.start_client()
- server_key = transport.get_remote_server_key()
- keytype = server_key.get_name()
-
- our_server_key = host_keys.get(host, {}).get(keytype, None)
- if options.ssh_key_check:
- if not our_server_key:
- hexified_key = ssh.FormatParamikoFingerprint(
- paramiko.util.hexify(server_key.get_fingerprint()))
- msg = ("Unable to verify hostkey of host %s: %s. Do you want to accept"
- " it?" % (host, hexified_key))
-
- if cli.AskUser(msg):
- our_server_key = server_key
-
- if our_server_key != server_key:
- logging.error("Unable to verify identity of host. Aborting")
- transport.close()
- transport.join()
- # TODO: Run over all hosts, fetch the keys and let them verify from the
- # user beforehand then proceed with actual work later on
- raise paramiko.SSHException("Unable to verify identity of host")
+ logging.info("Configuring %s", host)
- try:
- if LoginViaKeys(transport, username, all_keys):
- logging.info("Authenticated to %s via public key", host)
- else:
- logging.warning("Authentication to %s via public key failed, trying"
- " password", host)
- if passwd is None:
- passwd = getpass.getpass(prompt="%s password:" % username)
- transport.auth_password(username=username, password=passwd)
- logging.info("Authenticated to %s via password", host)
- except paramiko.SSHException, err:
- logging.error("Connection or authentication failed to host %s: %s",
- host, err)
- transport.close()
- # this is needed for compatibility with older Paramiko or Python
- # versions
- transport.join()
- continue
+ transport = paramiko.Transport((host, ssh_port))
try:
try:
+ transport.start_client()
+
+ if options.ssh_key_check:
+ _VerifyServerKey(transport, host, host_keys)
+
+ try:
+ if LoginViaKeys(transport, username, all_keys):
+ logging.info("Authenticated to %s via public key", host)
+ else:
+ if all_keys:
+ logging.warning("Authentication to %s via public key failed,"
+ " trying password", host)
+ if passwd is None:
+ passwd = getpass.getpass(prompt="%s password:" % username)
+ transport.auth_password(username=username, password=passwd)
+ logging.info("Authenticated to %s via password", host)
+ except paramiko.SSHException, err:
+ raise AuthError("Auth error TODO" % err)
+
+ if not _CheckJoin(transport):
+ if not options.force_join:
+ raise JoinCheckError(("Host %s failed join check; Please verify"
+ " that the host was not previously joined"
+ " to another cluster and use --force-join"
+ " to continue") % host)
+
+ logging.warning("Host %s failed join check, forced to continue",
+ host)
+
SetupSSH(transport)
- SetupNodeDaemon(transport)
- except errors.GenericError, err:
- logging.error("While doing setup on host %s an error occured: %s",
- host, err)
- finally:
- transport.close()
- # this is needed for compatibility with older Paramiko or Python
- # versions
- transport.join()
+ logging.info("%s successfully configured", host)
+ finally:
+ transport.close()
+ # this is needed for compatibility with older Paramiko or Python
+ # versions
+ transport.join()
+ except AuthError, err:
+ logging.error("Authentication error: %s", err)
+ success = False
+ break
+ except HostKeyVerificationError, err:
+ logging.error("Host key verification error: %s", err)
+ success = False
+ except Exception, err:
+ logging.exception("During setup of %s: %s", host, err)
+ success = False
+
+ if success:
+ sys.exit(constants.EXIT_SUCCESS)
+
+ sys.exit(constants.EXIT_FAILURE)
if __name__ == "__main__":