2 files changed, 183 insertions, 0 deletions
diff --git a/src/blk/zoned/HMSMRDevice.cc b/src/blk/zoned/HMSMRDevice.cc
new file mode 100644
index 000000000..416eae4e4
--- /dev/null
+++ b/src/blk/zoned/HMSMRDevice.cc
@@ -0,0 +1,131 @@
+// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
+// vim: ts=8 sw=2 smarttab
+/*
+ * Ceph - scalable distributed file system
+ *
+ * Copyright (C) 2014 Red Hat
+ * Copyright (C) 2020 Abutalib Aghayev
+ *
+ * This is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License version 2.1, as published by the Free Software
+ * Foundation.  See file COPYING.
+ *
+ */
+
+#include "HMSMRDevice.h"
+extern "C" {
+#include <libzbd/zbd.h>
+}
+#include "common/debug.h"
+#include "common/errno.h"
+
+#define dout_context cct
+#define dout_subsys ceph_subsys_bdev
+#undef dout_prefix
+#define dout_prefix *_dout << "smrbdev(" << this << " " << path << ") "
+
+using namespace std;
+
+HMSMRDevice::HMSMRDevice(CephContext* cct,
+			 aio_callback_t cb,
+			 void *cbpriv,
+			 aio_callback_t d_cb,
+			 void *d_cbpriv)
+  : KernelDevice(cct, cb, cbpriv, d_cb, d_cbpriv)
+{
+}
+
+bool HMSMRDevice::support(const std::string& path)
+{
+  return zbd_device_is_zoned(path.c_str()) == 1;
+}
+
+int HMSMRDevice::_post_open()
+{
+  dout(10) << __func__ << dendl;
+
+  zbd_fd = zbd_open(path.c_str(), O_RDWR | O_DIRECT | O_LARGEFILE, nullptr);
+  int r;
+  if (zbd_fd < 0) {
+    r = errno;
+    derr << __func__ << " zbd_open failed on " << path << ": "
+	 << cpp_strerror(r) << dendl;
+    return -r;
+  }
+
+  unsigned int nr_zones = 0;
+  std::vector<zbd_zone> zones;
+  if (zbd_report_nr_zones(zbd_fd, 0, 0, ZBD_RO_NOT_WP, &nr_zones) != 0) {
+    r = -errno;
+    derr << __func__ << " zbd_report_nr_zones failed on " << path << ": "
+	 << cpp_strerror(r) << dendl;
+    goto fail;
+  }
+
+  zones.resize(nr_zones);
+  if (zbd_report_zones(zbd_fd, 0, 0, ZBD_RO_NOT_WP, zones.data(), &nr_zones) != 0) {
+    r = -errno;
+    derr << __func__ << " zbd_report_zones failed on " << path << dendl;
+    goto fail;
+  }
+
+  zone_size = zbd_zone_len(&zones[0]);
+  conventional_region_size = nr_zones * zone_size;
+
+  dout(10) << __func__ << " setting zone size to " << zone_size
+	   << " and conventional region size to " << conventional_region_size
+           << dendl;
+
+  return 0;
+
+fail:
+  zbd_close(zbd_fd);
+  zbd_fd = -1;
+  return r;
+}
+
+
+void HMSMRDevice::_pre_close()
+{
+  if (zbd_fd >= 0) {
+    zbd_close(zbd_fd);
+    zbd_fd = -1;
+  }
+}
+
+void HMSMRDevice::reset_all_zones()
+{
+  dout(10) << __func__ << dendl;
+  zbd_reset_zones(zbd_fd, conventional_region_size, 0);
+}
+
+void HMSMRDevice::reset_zone(uint64_t zone)
+{
+  dout(10) << __func__ << " zone 0x" << std::hex << zone << std::dec << dendl;
+  if (zbd_reset_zones(zbd_fd, zone * zone_size, zone_size) != 0) {
+    derr << __func__ << " resetting zone failed for zone 0x" << std::hex
+	 << zone << std::dec << dendl;
+    ceph_abort("zbd_reset_zones failed");
+  }
+}
+
+std::vector<uint64_t> HMSMRDevice::get_zones()
+{
+  std::vector<zbd_zone> zones;
+  unsigned int num_zones = size / zone_size;
+  zones.resize(num_zones);
+
+  int r = zbd_report_zones(zbd_fd, 0, 0, ZBD_RO_ALL, zones.data(), &num_zones);
+  if (r != 0) {
+    derr << __func__ << " zbd_report_zones failed on " << path << ": "
+	 << cpp_strerror(errno) << dendl;
+    ceph_abort("zbd_report_zones failed");
+  }
+
+  std::vector<uint64_t> wp(num_zones);
+  for (unsigned i = 0; i < num_zones; ++i) {
+    wp[i] = zones[i].wp;
+  }
+  return wp;
+}
diff --git a/src/blk/zoned/HMSMRDevice.h b/src/blk/zoned/HMSMRDevice.h
new file mode 100644
index 000000000..edf18b5f0
--- /dev/null
+++ b/src/blk/zoned/HMSMRDevice.h
@@ -0,0 +1,52 @@
+// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
+// vim: ts=8 sw=2 smarttab
+/*
+ * Ceph - scalable distributed file system
+ *
+ * Copyright (C) 2014 Red Hat
+ * Copyright (C) 2020 Abutalib Aghayev
+ *
+ * This is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License version 2.1, as published by the Free Software
+ * Foundation.  See file COPYING.
+ *
+ */
+
+#ifndef CEPH_BLK_HMSMRDEVICE_H
+#define CEPH_BLK_HMSMRDEVICE_H
+
+#include <atomic>
+
+#include "include/types.h"
+#include "include/interval_set.h"
+#include "common/Thread.h"
+#include "include/utime.h"
+
+#include "aio/aio.h"
+#include "BlockDevice.h"
+#include "../kernel/KernelDevice.h"
+
+
+class HMSMRDevice final : public KernelDevice {
+  int zbd_fd = -1;	///< fd for the zoned block device
+
+public:
+  HMSMRDevice(CephContext* cct, aio_callback_t cb, void *cbpriv,
+              aio_callback_t d_cb, void *d_cbpriv);
+
+  static bool support(const std::string& path);
+
+  // open/close hooks for libzbd
+  int _post_open() override;
+  void _pre_close() override;
+
+  // smr-specific methods
+  bool is_smr() const final { return true; }
+  void reset_all_zones() override;
+  void reset_zone(uint64_t zone) override;
+  std::vector<uint64_t> get_zones() override;
+
+};
+
+#endif //CEPH_BLK_HMSMRDEVICE_H