rbd: verify the specified namespace before set

Signed-off-by: songweibin <song.weibin@zte.com.cn>

diff --git a/src/tools/rbd/Utils.cc b/src/tools/rbd/Utils.cc
index e436a876ab6aff406593fb3a1b812c5ca5f70b91..722f4ba626159001b39f5e00e29b9e860295bc5e 100644 (file)
--- a/src/tools/rbd/Utils.cc
+++ b/src/tools/rbd/Utils.cc
@@ -665,6 +665,21 @@ int init_io_ctx(librados::Rados &rados, const std::string &pool_name,
     return r;
   }
 
+  if (!namespace_name.empty()) {
+    librbd::RBD rbd;
+    bool exists = false;
+    r = rbd.namespace_exists(*io_ctx, namespace_name.c_str(), &exists);
+    if (r < 0) {
+      std::cerr << "rbd: error asserting namespace: "
+                << cpp_strerror(r) << std::endl;
+      return r;
+    }
+    if (!exists) {
+      std::cerr << "rbd: namespace '" << namespace_name << "' does not exist."
+                << std::endl;
+      return -ENOENT;
+    }
+  }
   io_ctx->set_namespace(namespace_name);
   return 0;
 }