from ganeti import utils
from ganeti import errors
from ganeti import constants
-
-
-__all__ = ["SSHCall", "CopyFileToNode", "VerifyNodeHostname",
- "KNOWN_HOSTS_OPTS", "BATCH_MODE_OPTS", "ASK_KEY_OPTS"]
+from ganeti import ssconf
KNOWN_HOSTS_OPTS = [
# Note: BATCH_MODE conflicts with ASK_KEY
BATCH_MODE_OPTS = [
- "-oEscapeChar=none",
"-oBatchMode=yes",
+ "-oEscapeChar=none",
"-oStrictHostKeyChecking=yes",
]
ASK_KEY_OPTS = [
- "-oStrictHostKeyChecking=ask",
"-oEscapeChar=none",
"-oHashKnownHosts=no",
+ "-oStrictHostKeyChecking=ask",
]
for base in ["id_dsa", "id_dsa.pub", "authorized_keys"]]
-def BuildSSHCmd(hostname, user, command, batch=True, ask_key=False):
- """Build an ssh string to execute a command on a remote node.
-
- Args:
- hostname: the target host, string
- user: user to auth as
- command: the command
- batch: if true, ssh will run in batch mode with no prompting
- ask_key: if true, ssh will run with StrictHostKeyChecking=ask, so that
- we can connect to an unknown host (not valid in batch mode)
-
- Returns:
- The ssh call to run 'command' on the remote host.
+class SshRunner:
+ """Wrapper for SSH commands.
"""
- argv = ["ssh", "-q"]
- argv.extend(KNOWN_HOSTS_OPTS)
- if batch:
- # if we are in batch mode, we can't ask the key
- if ask_key:
- raise errors.ProgrammerError("SSH call requested conflicting options")
- argv.extend(BATCH_MODE_OPTS)
- elif ask_key:
- argv.extend(ASK_KEY_OPTS)
- argv.extend(["%s@%s" % (user, hostname), command])
- return argv
-
-
-def SSHCall(hostname, user, command, batch=True, ask_key=False):
- """Execute a command on a remote node.
-
- This method has the same return value as `utils.RunCmd()`, which it
- uses to launch ssh.
-
- Args:
- hostname: the target host, string
- user: user to auth as
- command: the command
- batch: if true, ssh will run in batch mode with no prompting
- ask_key: if true, ssh will run with StrictHostKeyChecking=ask, so that
- we can connect to an unknown host (not valid in batch mode)
-
- Returns:
- `utils.RunResult` as for `utils.RunCmd()`
+ def __init__(self, sstore=None):
+ if sstore is None:
+ self.sstore = ssconf.SimpleStore()
+ else:
+ self.sstore = sstore
+
+ def _GetHostKeyAliasOption(self):
+ return "-oHostKeyAlias=%s" % self.sstore.GetClusterName()
+
+ def BuildCmd(self, hostname, user, command, batch=True, ask_key=False,
+ tty=False):
+ """Build an ssh command to execute a command on a remote node.
+
+ Args:
+ hostname: the target host, string
+ user: user to auth as
+ command: the command
+ batch: if true, ssh will run in batch mode with no prompting
+ ask_key: if true, ssh will run with StrictHostKeyChecking=ask, so that
+ we can connect to an unknown host (not valid in batch mode)
+
+ Returns:
+ The ssh call to run 'command' on the remote host.
+
+ """
+ argv = [constants.SSH, "-q"]
+ argv.extend(KNOWN_HOSTS_OPTS)
+ argv.append(self._GetHostKeyAliasOption())
+ if batch:
+ # if we are in batch mode, we can't ask the key
+ if ask_key:
+ raise errors.ProgrammerError("SSH call requested conflicting options")
+ argv.extend(BATCH_MODE_OPTS)
+ elif ask_key:
+ argv.extend(ASK_KEY_OPTS)
+ if tty:
+ argv.append("-t")
+ argv.extend(["%s@%s" % (user, hostname), command])
+ return argv
+
+ def Run(self, hostname, user, command, batch=True, ask_key=False):
+ """Runs a command on a remote node.
+
+ This method has the same return value as `utils.RunCmd()`, which it
+ uses to launch ssh.
+
+ Args:
+ hostname: the target host, string
+ user: user to auth as
+ command: the command
+ batch: if true, ssh will run in batch mode with no prompting
+ ask_key: if true, ssh will run with StrictHostKeyChecking=ask, so that
+ we can connect to an unknown host (not valid in batch mode)
+
+ Returns:
+ `utils.RunResult` like `utils.RunCmd()`
+
+ """
+ return utils.RunCmd(self.BuildCmd(hostname, user, command, batch=batch,
+ ask_key=ask_key))
+
+ def CopyFileToNode(self, node, filename):
+ """Copy a file to another node with scp.
+
+ Args:
+ node: node in the cluster
+ filename: absolute pathname of a local file
+
+ Returns:
+ success: True/False
- """
- return utils.RunCmd(BuildSSHCmd(hostname, user, command,
- batch=batch, ask_key=ask_key))
+ """
+ if not os.path.isabs(filename):
+ logger.Error("file %s must be an absolute path" % (filename))
+ return False
+ if not os.path.isfile(filename):
+ logger.Error("file %s does not exist" % (filename))
+ return False
-def CopyFileToNode(node, filename):
- """Copy a file to another node with scp.
+ command = [constants.SCP, "-q", "-p"]
+ command.extend(KNOWN_HOSTS_OPTS)
+ command.extend(BATCH_MODE_OPTS)
+ command.append(self._GetHostKeyAliasOption())
+ command.append(filename)
+ command.append("%s:%s" % (node, filename))
- Args:
- node: node in the cluster
- filename: absolute pathname of a local file
+ result = utils.RunCmd(command)
- Returns:
- success: True/False
+ if result.failed:
+ logger.Error("copy to node %s failed (%s) error %s,"
+ " command was %s" %
+ (node, result.fail_reason, result.output, result.cmd))
- """
- if not os.path.isfile(filename):
- logger.Error("file %s does not exist" % (filename))
- return False
+ return not result.failed
- if not os.path.isabs(filename):
- logger.Error("file %s must be an absolute path" % (filename))
- return False
+ def VerifyNodeHostname(self, node):
+ """Verify hostname consistency via SSH.
- command = ["scp", "-q", "-p"]
- command.extend(KNOWN_HOSTS_OPTS)
- command.extend(BATCH_MODE_OPTS)
- command.append(filename)
- command.append("%s:%s" % (node, filename))
+ This functions connects via ssh to a node and compares the hostname
+ reported by the node to the name with have (the one that we
+ connected to).
- result = utils.RunCmd(command)
+ This is used to detect problems in ssh known_hosts files
+ (conflicting known hosts) and incosistencies between dns/hosts
+ entries and local machine names
- if result.failed:
- logger.Error("copy to node %s failed (%s) error %s,"
- " command was %s" %
- (node, result.fail_reason, result.output, result.cmd))
+ Args:
+ node: nodename of a host to check. can be short or full qualified hostname
- return not result.failed
+ Returns:
+ (success, detail)
+ where
+ success: True/False
+ detail: String with details
+ """
+ retval = self.Run(node, 'root', 'hostname')
-def VerifyNodeHostname(node):
- """Verify hostname consistency via SSH.
+ if retval.failed:
+ msg = "ssh problem"
+ output = retval.output
+ if output:
+ msg += ": %s" % output
+ return False, msg
+ remotehostname = retval.stdout.strip()
- This functions connects via ssh to a node and compares the hostname
- reported by the node to the name with have (the one that we
- connected to).
+ if not remotehostname or remotehostname != node:
+ return False, "hostname mismatch, got %s" % remotehostname
- This is used to detect problems in ssh known_hosts files
- (conflicting known hosts) and incosistencies between dns/hosts
- entries and local machine names
+ return True, "host matches"
- Args:
- node: nodename of a host to check. can be short or full qualified hostname
- Returns:
- (success, detail)
- where
- success: True/False
- detail: String with details
+def WriteKnownHostsFile(cfg, sstore, file_name):
+ """Writes the cluster-wide equally known_hosts file.
"""
- retval = SSHCall(node, 'root', 'hostname')
-
- if retval.failed:
- msg = "ssh problem"
- output = retval.output
- if output:
- msg += ": %s" % output
- return False, msg
-
- remotehostname = retval.stdout.strip()
-
- if not remotehostname or remotehostname != node:
- return False, "hostname mismatch, got %s" % remotehostname
-
- return True, "host matches"
+ utils.WriteFile(file_name, mode=0700,
+ data="%s ssh-rsa %s\n" % (sstore.GetClusterName(),
+ cfg.GetHostKey()))