From 50cd57e902fe508f98f63fea30626780b07561d9 Mon Sep 17 00:00:00 2001
From: David Zafman <david.zafman@inktank.com>
Date: Mon, 19 May 2014 20:47:14 -0700
Subject: [PATCH] ceph_filestore_dump: Add set-omap object command

Signed-off-by: David Zafman <david.zafman@inktank.com>
---
 src/tools/ceph_filestore_dump.cc | 44 ++++++++++++++++++++++++++++++++
 1 file changed, 44 insertions(+)

diff --git a/src/tools/ceph_filestore_dump.cc b/src/tools/ceph_filestore_dump.cc
index 548ba1976cc9e..23e748437723a 100644
--- a/src/tools/ceph_filestore_dump.cc
+++ b/src/tools/ceph_filestore_dump.cc
@@ -1394,6 +1394,29 @@ int do_get_omap(ObjectStore *store, coll_t coll, ghobject_t &ghobj, string key)
   return 0;
 }
 
+int do_set_omap(ObjectStore *store, coll_t coll, ghobject_t &ghobj, string key, int fd)
+{
+  ObjectStore::Transaction tran;
+  ObjectStore::Transaction *t = &tran;
+  map<string, bufferlist> attrset;
+  bufferlist valbl;
+
+  if (debug)
+    cerr << "Set_omap " << ghobj << std::endl;
+
+  if (get_fd_data(fd, valbl))
+    return 1;
+
+  attrset.insert(pair<string, bufferlist>(key, valbl));
+
+  t->touch(coll, ghobj);
+
+  t->omap_setkeys(coll, ghobj, attrset);
+
+  store->apply_transaction(*t);
+  return 0;
+}
+
 void usage(po::options_description &desc)
 {
     cerr << std::endl;
@@ -1856,6 +1879,27 @@ int main(int argc, char **argv)
 	if (r)
 	  ret = 1;
         goto out;
+      } else if (objcmd == "set-omap") {
+	if (vm.count("arg1") == 0)
+	  usage(desc);
+
+	int fd;
+	if (vm.count("arg2") == 0 || arg2 == "-") {
+	  fd = STDIN_FILENO;
+	} else {
+	  fd = open(arg2.c_str(), O_RDONLY|O_LARGEFILE, 0666);
+	  if (fd == -1) {
+	    cerr << "open " << arg2 << " " << cpp_strerror(errno) << std::endl;
+	    ret = 1;
+	    goto out;
+	  }
+	}
+	r = do_set_omap(fs, coll, ghobj, arg1, fd);
+	if (fd != STDIN_FILENO)
+	  close(fd);
+	if (r)
+	  ret = 1;
+        goto out;
       }
       cerr << "Unknown object command '" << objcmd << "'" << std::endl;
       usage(desc);
-- 
2.39.5