os/bluestore/BlueFS: readdir list dirs, too

Signed-off-by: Sage Weil <sage@redhat.com>

diff --git a/src/os/bluestore/BlueFS.cc b/src/os/bluestore/BlueFS.cc
index 2a0d822ba6e4557c5704b1ee61c8032863a25344..a57b07303f7be10faa1a40326d3f9c88d25f07e1 100644 (file)
--- a/src/os/bluestore/BlueFS.cc
+++ b/src/os/bluestore/BlueFS.cc
@@ -1174,15 +1174,24 @@ int BlueFS::readdir(const string& dirname, vector<string> *ls)
 {
   Mutex::Locker l(lock);
   dout(10) << __func__ << " " << dirname << dendl;
-  map<string,Dir*>::iterator p = dir_map.find(dirname);
-  if (p == dir_map.end()) {
-    dout(20) << __func__ << " dir " << dirname << " not found" << dendl;
-    return -ENOENT;
-  }
-  Dir *dir = p->second;
-  ls->reserve(dir->file_map.size() + 2);
-  for (auto q : dir->file_map) {
-    ls->push_back(q.first);
+  if (dirname.size() == 0) {
+    // list dirs
+    ls->reserve(dir_map.size() + 2);
+    for (auto q : dir_map) {
+      ls->push_back(q.first);
+    }
+  } else {
+    // list files in dir
+    map<string,Dir*>::iterator p = dir_map.find(dirname);
+    if (p == dir_map.end()) {
+      dout(20) << __func__ << " dir " << dirname << " not found" << dendl;
+      return -ENOENT;
+    }
+    Dir *dir = p->second;
+    ls->reserve(dir->file_map.size() + 2);
+    for (auto q : dir->file_map) {
+      ls->push_back(q.first);
+    }
   }
   ls->push_back(".");
   ls->push_back("..");