+++ /dev/null
-// -*- mode:C; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
-// vim: ts=8 sw=2 smarttab
-
-#include <errno.h>
-
-#include "objclass/objclass.h"
-#include "cls/refcount/cls_refcount_ops.h"
-
-#include "include/compat.h"
-
-CLS_VER(1,0)
-CLS_NAME(refcount)
-
-
-#define REFCOUNT_ATTR "refcount"
-
-struct obj_refcount {
- map<string, bool> refs;
- set<string> retired_refs;
-
- obj_refcount() {}
-
- void encode(bufferlist& bl) const {
- ENCODE_START(2, 1, bl);
- ::encode(refs, bl);
- ::encode(retired_refs, bl);
- ENCODE_FINISH(bl);
- }
-
- void decode(bufferlist::iterator& bl) {
- DECODE_START(2, bl);
- ::decode(refs, bl);
- if (struct_v >= 2) {
- ::decode(retired_refs, bl);
- }
- DECODE_FINISH(bl);
- }
-};
-WRITE_CLASS_ENCODER(obj_refcount)
-
-static string wildcard_tag;
-
-static int read_refcount(cls_method_context_t hctx, bool implicit_ref, obj_refcount *objr)
-{
- bufferlist bl;
- objr->refs.clear();
- int ret = cls_cxx_getxattr(hctx, REFCOUNT_ATTR, &bl);
- if (ret == -ENODATA) {
- if (implicit_ref) {
- objr->refs[wildcard_tag] = true;
- }
- return 0;
- }
- if (ret < 0)
- return ret;
-
- try {
- bufferlist::iterator iter = bl.begin();
- ::decode(*objr, iter);
- } catch (buffer::error& err) {
- CLS_LOG(0, "ERROR: read_refcount(): failed to decode refcount entry\n");
- return -EIO;
- }
-
- return 0;
-}
-
-static int set_refcount(cls_method_context_t hctx, const struct obj_refcount& objr)
-{
- bufferlist bl;
-
- ::encode(objr, bl);
-
- int ret = cls_cxx_setxattr(hctx, REFCOUNT_ATTR, &bl);
- if (ret < 0)
- return ret;
-
- return 0;
-}
-
-static int cls_rc_refcount_get(cls_method_context_t hctx, bufferlist *in, bufferlist *out)
-{
- bufferlist::iterator in_iter = in->begin();
-
- cls_refcount_get_op op;
- try {
- ::decode(op, in_iter);
- } catch (buffer::error& err) {
- CLS_LOG(1, "ERROR: cls_rc_refcount_get(): failed to decode entry\n");
- return -EINVAL;
- }
-
- obj_refcount objr;
- int ret = read_refcount(hctx, op.implicit_ref, &objr);
- if (ret < 0)
- return ret;
-
- CLS_LOG(10, "cls_rc_refcount_get() tag=%s\n", op.tag.c_str());
-
- objr.refs[op.tag] = true;
-
- ret = set_refcount(hctx, objr);
- if (ret < 0)
- return ret;
-
- return 0;
-}
-
-static int cls_rc_refcount_put(cls_method_context_t hctx, bufferlist *in, bufferlist *out)
-{
- bufferlist::iterator in_iter = in->begin();
-
- cls_refcount_put_op op;
- try {
- ::decode(op, in_iter);
- } catch (buffer::error& err) {
- CLS_LOG(1, "ERROR: cls_rc_refcount_put(): failed to decode entry\n");
- return -EINVAL;
- }
-
- obj_refcount objr;
- int ret = read_refcount(hctx, op.implicit_ref, &objr);
- if (ret < 0)
- return ret;
-
- if (objr.refs.empty()) {// shouldn't happen!
- CLS_LOG(0, "ERROR: cls_rc_refcount_put() was called without any references!\n");
- return -EINVAL;
- }
-
- CLS_LOG(10, "cls_rc_refcount_put() tag=%s\n", op.tag.c_str());
-
- bool found = false;
- map<string, bool>::iterator iter = objr.refs.find(op.tag);
- if (iter != objr.refs.end()) {
- found = true;
- } else if (op.implicit_ref) {
- iter = objr.refs.find(wildcard_tag);
- if (iter != objr.refs.end()) {
- found = true;
- }
- }
-
- if (!found ||
- objr.retired_refs.find(op.tag) != objr.retired_refs.end())
- return 0;
-
- objr.retired_refs.insert(op.tag);
- objr.refs.erase(iter);
-
- if (objr.refs.empty()) {
- return cls_cxx_remove(hctx);
- }
-
- ret = set_refcount(hctx, objr);
- if (ret < 0)
- return ret;
-
- return 0;
-}
-
-static int cls_rc_refcount_set(cls_method_context_t hctx, bufferlist *in, bufferlist *out)
-{
- bufferlist::iterator in_iter = in->begin();
-
- cls_refcount_set_op op;
- try {
- ::decode(op, in_iter);
- } catch (buffer::error& err) {
- CLS_LOG(1, "ERROR: cls_refcount_set(): failed to decode entry\n");
- return -EINVAL;
- }
-
- if (!op.refs.size()) {
- return cls_cxx_remove(hctx);
- }
-
- obj_refcount objr;
- list<string>::iterator iter;
- for (iter = op.refs.begin(); iter != op.refs.end(); ++iter) {
- objr.refs[*iter] = true;
- }
-
- int ret = set_refcount(hctx, objr);
- if (ret < 0)
- return ret;
-
- return 0;
-}
-
-static int cls_rc_refcount_read(cls_method_context_t hctx, bufferlist *in, bufferlist *out)
-{
- bufferlist::iterator in_iter = in->begin();
-
- cls_refcount_read_op op;
- try {
- ::decode(op, in_iter);
- } catch (buffer::error& err) {
- CLS_LOG(1, "ERROR: cls_rc_refcount_read(): failed to decode entry\n");
- return -EINVAL;
- }
-
- obj_refcount objr;
-
- cls_refcount_read_ret read_ret;
- int ret = read_refcount(hctx, op.implicit_ref, &objr);
- if (ret < 0)
- return ret;
-
- map<string, bool>::iterator iter;
- for (iter = objr.refs.begin(); iter != objr.refs.end(); ++iter) {
- read_ret.refs.push_back(iter->first);
- }
-
- ::encode(read_ret, *out);
-
- return 0;
-}
-
-CLS_INIT(refcount)
-{
- CLS_LOG(1, "Loaded refcount class!");
-
- cls_handle_t h_class;
- cls_method_handle_t h_refcount_get;
- cls_method_handle_t h_refcount_put;
- cls_method_handle_t h_refcount_set;
- cls_method_handle_t h_refcount_read;
-
- cls_register("refcount", &h_class);
-
- /* refcount */
- cls_register_cxx_method(h_class, "get", CLS_METHOD_RD | CLS_METHOD_WR, cls_rc_refcount_get, &h_refcount_get);
- cls_register_cxx_method(h_class, "put", CLS_METHOD_RD | CLS_METHOD_WR, cls_rc_refcount_put, &h_refcount_put);
- cls_register_cxx_method(h_class, "set", CLS_METHOD_RD | CLS_METHOD_WR, cls_rc_refcount_set, &h_refcount_set);
- cls_register_cxx_method(h_class, "read", CLS_METHOD_RD, cls_rc_refcount_read, &h_refcount_read);
-
- return;
-}
-