1 // -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
2 // vim: ts=8 sw=2 smarttab
4 #include "test/librados_test_stub/TestClassHandler.h"
5 #include "test/librados_test_stub/TestIoCtxImpl.h"
6 #include <boost/algorithm/string/predicate.hpp>
11 #include "common/debug.h"
12 #include "include/assert.h"
14 #define dout_context g_ceph_context
15 #define dout_subsys ceph_subsys_rados
19 TestClassHandler::TestClassHandler() {
22 TestClassHandler::~TestClassHandler() {
23 for (ClassHandles::iterator it = m_class_handles.begin();
24 it != m_class_handles.end(); ++it) {
29 void TestClassHandler::open_class(const std::string& name,
30 const std::string& path) {
31 void *handle = dlopen(path.c_str(), RTLD_NOW);
33 derr << "Failed to load class: " << dlerror() << dendl;
36 m_class_handles.push_back(handle);
39 void (*cls_init)() = reinterpret_cast<void (*)()>(
40 dlsym(handle, "__cls_init"));
46 void TestClassHandler::open_all_classes() {
47 assert(m_class_handles.empty());
49 const char* env = getenv("CEPH_LIB");
50 std::string CEPH_LIB(env ? env : ".libs");
51 DIR *dir = ::opendir(CEPH_LIB.c_str());
56 struct dirent *pde = nullptr;
57 while ((pde = ::readdir(dir))) {
58 std::string name(pde->d_name);
59 if (!boost::algorithm::starts_with(name, "libcls_") ||
60 !boost::algorithm::ends_with(name, ".so")) {
63 std::string class_name = name.substr(7, name.size() - 10);
64 open_class(class_name, CEPH_LIB + "/" + name);
69 int TestClassHandler::create(const std::string &name, cls_handle_t *handle) {
70 if (m_classes.find(name) != m_classes.end()) {
74 SharedClass cls(new Class());
75 m_classes[name] = cls;
76 *handle = reinterpret_cast<cls_handle_t>(cls.get());
80 int TestClassHandler::create_method(cls_handle_t hclass,
82 cls_method_cxx_call_t class_call,
83 cls_method_handle_t *handle) {
84 Class *cls = reinterpret_cast<Class*>(hclass);
85 if (cls->methods.find(name) != cls->methods.end()) {
89 SharedMethod method(new Method());
90 method->class_call = class_call;
91 cls->methods[name] = method;
95 cls_method_cxx_call_t TestClassHandler::get_method(const std::string &cls,
96 const std::string &method) {
97 Classes::iterator c_it = m_classes.find(cls);
98 if (c_it == m_classes.end()) {
102 SharedClass scls = c_it->second;
103 Methods::iterator m_it = scls->methods.find(method);
104 if (m_it == scls->methods.end()) {
107 return m_it->second->class_call;
110 TestClassHandler::SharedMethodContext TestClassHandler::get_method_context(
111 TestIoCtxImpl *io_ctx_impl, const std::string &oid,
112 const SnapContext &snapc) {
113 SharedMethodContext ctx(new MethodContext());
115 // clone to ioctx to provide a firewall for gmock expectations
116 ctx->io_ctx_impl = io_ctx_impl->clone();
122 int TestClassHandler::create_filter(cls_handle_t hclass,
123 const std::string& name,
124 cls_cxx_filter_factory_t fn)
126 Class *cls = reinterpret_cast<Class*>(hclass);
127 if (cls->filters.find(name) != cls->filters.end()) {
130 cls->filters[name] = fn;
134 TestClassHandler::MethodContext::~MethodContext() {
138 } // namespace librados