From ac99fe75450202bc796950d452920c12037154df Mon Sep 17 00:00:00 2001
From: Sage Weil <sage@redhat.com>
Date: Thu, 3 Oct 2019 16:13:47 -0500
Subject: [PATCH] mgr/ssh: pipe ceph-daemon script to stdin of python3

This avoids any need for the script to be present on the remote host.

We introduce a config option to indicate where the script should be
read from, since the location varies between a vstart environment (source
dir) and a real install (/usr/sbin).

Signed-off-by: Sage Weil <sage@redhat.com>
---
 src/common/options.cc        |  5 ++++
 src/pybind/mgr/ssh/module.py | 48 +++++++++++++++++++++---------------
 src/vstart.sh                |  1 +
 3 files changed, 34 insertions(+), 20 deletions(-)

diff --git a/src/common/options.cc b/src/common/options.cc
index e24c1156ff37b..53d87cba83c39 100644
--- a/src/common/options.cc
+++ b/src/common/options.cc
@@ -5428,6 +5428,11 @@ std::vector<Option> get_global_options() {
     .set_default(false)
     .set_description(""),
 
+    Option("ceph_daemon_path", Option::TYPE_STR, Option::LEVEL_ADVANCED)
+    .set_default("/usr/sbin/ceph-daemon")
+    .add_service("mgr")
+    .set_description("Path to ceph-daemon utility"),
+
     Option("mgr_module_path", Option::TYPE_STR, Option::LEVEL_ADVANCED)
     .set_default(CEPH_DATADIR "/mgr")
     .add_service("mgr")
diff --git a/src/pybind/mgr/ssh/module.py b/src/pybind/mgr/ssh/module.py
index 7144d5062f21d..86519810cc645 100644
--- a/src/pybind/mgr/ssh/module.py
+++ b/src/pybind/mgr/ssh/module.py
@@ -130,6 +130,14 @@ class SSHOrchestrator(MgrModule, orchestrator.Orchestrator):
     def __init__(self, *args, **kwargs):
         super(SSHOrchestrator, self).__init__(*args, **kwargs)
         self._cluster_fsid = None
+
+        path = self.get_ceph_option('ceph_daemon_path')
+        try:
+            with open(path, 'r') as f:
+                self._ceph_daemon = f.read()
+        except:
+            raise RuntimeError("unable to read ceph-daemon at '%s'" % path)
+
         self._worker_pool = multiprocessing.pool.ThreadPool(1)
 
         self._reconfig_ssh()
@@ -345,7 +353,7 @@ class SSHOrchestrator(MgrModule, orchestrator.Orchestrator):
         nodes = [orchestrator.InventoryNode(host_name, []) for host_name in self.inventory_cache]
         return orchestrator.TrivialReadCompletion(nodes)
 
-    def _run_ceph_daemon(self, host, entity, command, args):
+    def _run_ceph_daemon(self, host, entity, command, args, stdin=None):
         """
         Run ceph-daemon on the remote host with the given command + args
         """
@@ -361,14 +369,24 @@ class SSHOrchestrator(MgrModule, orchestrator.Orchestrator):
             image = image.strip()
             self.log.debug('%s container image %s' % (entity, image))
 
-            command = [
-                '/home/sage/src/ceph5/src/ceph-daemon',
+            final_args = [
                 '--image', image,
                 command,
                 '--fsid', self.get('mon_map')['fsid'],
             ] + args
 
-            out, err, code = remoto.process.check(conn, command)
+            script = 'injected_argv = ' + json.dumps(final_args) + '\n'
+            if stdin:
+                script += 'injected_stdin = ' + json.dumps(stdin) + '\n'
+            script += self._ceph_daemon
+            #self.log.debug('script is %s' % script)
+
+            out, err, code = remoto.process.check(
+                conn,
+                ['/usr/bin/python3', '-u'],
+                stdin=script.encode('utf-8'))
+            if code:
+                self.log.debug('code %s, err %s' % (code, err))
             # ceph-daemon combines stdout and stderr, so ignore err.
             self.log.debug('code %s out %s' % (code, out))
             return out, code
@@ -450,25 +468,19 @@ class SSHOrchestrator(MgrModule, orchestrator.Orchestrator):
         })
         self.log.debug('j %s' % j)
 
-        # TODO FIXME
-        conn = self._get_connection(host)
-        conn.remote_module.write_file(
-            '/tmp/foo',
-            j,
-            0o600, None, 0, 0)
-
         devices = drive_group.data_devices.paths
         for device in devices:
             out, code = self._run_ceph_daemon(
                 host, 'osd', 'ceph-volume',
                 [
-                    '--config-and-keyring', '/tmp/foo',
+                    '--config-and-keyring', '-',
                     '--',
                     'lvm', 'prepare',
                     "--cluster-fsid", self._get_cluster_fsid(),
                     "--{}".format(drive_group.objectstore),
                     "--data", device,
-                ])
+                ],
+                stdin=j)
             self.log.debug('ceph-volume prepare: %s' % out)
 
         # check result
@@ -550,17 +562,13 @@ class SSHOrchestrator(MgrModule, orchestrator.Orchestrator):
             })
             self.log.debug('j %s' % j)
 
-            conn.remote_module.write_file(
-                '/tmp/foo',
-                j,
-                0o600, None, 0, 0)
-
             out, code = self._run_ceph_daemon(
                 host, name, 'deploy',
                 [
                     '--name', name,
-                    '--config-and-keyring', '/tmp/foo',
-                ] + extra_args)
+                    '--config-and-keyring', '-',
+                ] + extra_args,
+                stdin=j)
             self.log.debug('create_daemon code %s out %s' % (code, out))
 
             return "Created {} on host '{}'".format(name, host)
diff --git a/src/vstart.sh b/src/vstart.sh
index af046cf6d7057..4b1014460d86d 100755
--- a/src/vstart.sh
+++ b/src/vstart.sh
@@ -692,6 +692,7 @@ $extra_conf
 [mgr]
         mgr data = $CEPH_DEV_DIR/mgr.\$id
         mgr module path = $MGR_PYTHON_PATH
+        ceph daemon path = $CEPH_ROOT/src/ceph-daemon
 $DAEMONOPTS
 $extra_conf
 [osd]
-- 
2.39.5