From 2d64461959bbc368d40e028b8282ad7999b69b03 Mon Sep 17 00:00:00 2001
From: Josh Durgin <josh.durgin@dreamhost.com>
Date: Thu, 9 Jun 2011 11:47:52 -0700
Subject: [PATCH] Add rbd -> /dev mapping method.

---
 teuthology/task/rbd.py | 96 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 96 insertions(+)

diff --git a/teuthology/task/rbd.py b/teuthology/task/rbd.py
index 0bae7f9accd49..092e342eea7e3 100644
--- a/teuthology/task/rbd.py
+++ b/teuthology/task/rbd.py
@@ -1,6 +1,9 @@
 import contextlib
 import logging
 
+from orchestra import run
+from teuthology import misc as teuthology
+
 log = logging.getLogger(__name__)
 
 def default_image_name(role):
@@ -108,6 +111,99 @@ def modprobe(ctx, config):
                     ],
                 )
 
+@contextlib.contextmanager
+def dev_create(ctx, config):
+    """
+    Map block devices to rbd images.
+
+    For example::
+
+        tasks:
+        - ceph:
+        - rbd.create_image: [client.0]
+        - rbd.modprobe: [client.0]
+        - rbd.dev_create:
+            client.0: testimage.client.0
+    """
+    assert isinstance(config, dict) or isinstance(config, list), \
+        "task dev_create only supports a list or dictionary for configuration"
+
+    if isinstance(config, dict):
+        role_images = config.items()
+    else:
+        role_images = [(role, None) for role in config]
+
+    for role, image in role_images:
+        if image is None:
+            image = default_image_name(role)
+        (remote,) = ctx.cluster.only(role).remotes.keys()
+
+        # add udev rule for creating /dev/rbd/pool/image
+        remote.run(
+            args=[
+                'echo',
+                'KERNEL=="rbd[0-9]*", PROGRAM="/tmp/cephtest/binary/usr/local/bin/crbdnamer %n", SYMLINK+="rbd/%c{1}/%c{2}"',
+                run.Raw('>'),
+                '/tmp/cephtest/51-rbd.rules',
+                ],
+            )
+        remote.run(
+            args=[
+                'sudo',
+                'mv',
+                '/tmp/cephtest/51-rbd.rules',
+                '/etc/udev/rules.d',
+                ],
+            )
+
+        secretfile = '/tmp/cephtest/data/{role}.secret'.format(role=role)
+        teuthology.write_secret_file(remote, role, secretfile)
+
+        remote.run(
+            args=[
+                'sudo',
+                'LD_LIBRARY_PATH=/tmp/cephtest/binary/usr/local/lib',
+                '/tmp/cephtest/binary/usr/local/bin/ceph-coverage',
+                '/tmp/cephtest/archive/coverage',
+                '/tmp/cephtest/binary/usr/local/bin/rbd',
+                '-c', '/tmp/cephtest/ceph.conf',
+                '--user', role.rsplit('.')[-1],
+                '--secret', secretfile,
+                '-p', 'rbd',
+                'map',
+                image,
+                ],
+            )
+    try:
+        yield
+    finally:
+        log.info('Unmapping rbd devices...')
+        for role, image in role_images:
+            if image is None:
+                image = default_image_name(role)
+            (remote,) = ctx.cluster.only(role).remotes.keys()
+            remote.run(
+                args=[
+                    'sudo',
+                    'LD_LIBRARY_PATH=/tmp/cephtest/binary/usr/local/lib',
+                    '/tmp/cephtest/binary/usr/local/bin/ceph-coverage',
+                    '/tmp/cephtest/archive/coverage',
+                    '/tmp/cephtest/binary/usr/local/bin/rbd',
+                    '-c', '/tmp/cephtest/ceph.conf',
+                    '-p', 'rbd',
+                    'unmap',
+                    '/dev/rbd/rbd/{imgname}'.format(imgname=image),
+                    ],
+                )
+            remote.run(
+                args=[
+                    'sudo',
+                    'rm',
+                    '/etc/udev/rules.d/51-rbd.rules',
+                    ],
+                wait=False,
+                )
+
 @contextlib.contextmanager
 def task(ctx, config):
     create_image(ctx, config)
-- 
2.39.5