/* Copyright (c) 2008-2012 Red Hat, Inc. This file is part of GlusterFS. This file is licensed to you under your choice of the GNU Lesser General Public License, version 3 or any later version (LGPLv3 or later), or the GNU General Public License, version 2 (GPLv2), in all cases as published by the Free Software Foundation. */ #include "quota.h" #include "quotad-helpers.h" #include "quotad-aggregator.h" struct rpcsvc_program quotad_aggregator_prog; struct iobuf * quotad_serialize_reply (rpcsvc_request_t *req, void *arg, struct iovec *outmsg, xdrproc_t xdrproc) { struct iobuf *iob = NULL; ssize_t retlen = 0; ssize_t xdr_size = 0; GF_VALIDATE_OR_GOTO ("server", req, ret); /* First, get the io buffer into which the reply in arg will * be serialized. */ if (arg && xdrproc) { xdr_size = xdr_sizeof (xdrproc, arg); iob = iobuf_get2 (req->svc->ctx->iobuf_pool, xdr_size); if (!iob) { gf_log_callingfn (THIS->name, GF_LOG_ERROR, "Failed to get iobuf"); goto ret; }; iobuf_to_iovec (iob, outmsg); /* Use the given serializer to translate the give C structure in arg * to XDR format which will be written into the buffer in outmsg. */ /* retlen is used to received the error since size_t is unsigned and we * need -1 for error notification during encoding. */ retlen = xdr_serialize_generic (*outmsg, arg, xdrproc); if (retlen == -1) { /* Failed to Encode 'GlusterFS' msg in RPC is not exactly failure of RPC return values.. client should get notified about this, so there are no missing frames */ gf_log_callingfn ("", GF_LOG_ERROR, "Failed to encode message"); req->rpc_err = GARBAGE_ARGS; retlen = 0; } } outmsg->iov_len = retlen; ret: if (retlen == -1) { iobuf_unref (iob); iob = NULL; } return iob; } int quotad_aggregator_submit_reply (call_frame_t *frame, rpcsvc_request_t *req, void *arg, struct iovec *payload, int payloadcount, struct iobref *iobref, xdrproc_t xdrproc) { struct iobuf *iob = NULL; int ret = -1; struct iovec rsp = {0,}; quotad_aggregator_state_t *state = NULL; char new_iobref = 0; GF_VALIDATE_OR_GOTO ("server", req, ret); if (frame) { state = frame->root->state; frame->local = NULL; } if (!iobref) { iobref = iobref_new (); if (!iobref) { goto ret; } new_iobref = 1; } iob = quotad_serialize_reply (req, arg, &rsp, xdrproc); if (!iob) { gf_log ("", GF_LOG_ERROR, "Failed to serialize reply"); goto ret; } iobref_add (iobref, iob); ret = rpcsvc_submit_generic (req, &rsp, 1, payload, payloadcount, iobref); iobuf_unref (iob); ret = 0; ret: if (state) { quotad_aggregator_free_state (state); } if (frame) { if (frame->root->trans) rpc_transport_unref (frame->root->trans); STACK_DESTROY (frame->root); } if (new_iobref) { iobref_unref (iobref); } return ret; } int quotad_aggregator_lookup_cbk (xlator_t *this, call_frame_t *frame, struct gfs3_lookup_rsp *rsp) { quotad_aggregator_submit_reply (frame, frame->local, rsp, NULL, 0, NULL, (xdrproc_t)xdr_gfs3_lookup_rsp); return 0; } int quotad_aggregator_lookup (rpcsvc_request_t *req) { call_frame_t *frame = NULL; gfs3_lookup_req args = {{0,},}; int ret = -1, op_errno = 0; gfs3_lookup_rsp rsp = {0,}; quotad_aggregator_state_t *state = NULL; xlator_t *this = NULL; GF_VALIDATE_OR_GOTO ("quotad-aggregator", req, err); this = THIS; args.bname = alloca (req->msg[0].iov_len); args.xdata.xdata_val = alloca (req->msg[0].iov_len); ret = xdr_to_generic (req->msg[0], &args, (xdrproc_t)xdr_gfs3_lookup_req); if (ret < 0) { rsp.op_errno = EINVAL; goto err; } frame = quotad_aggregator_get_frame_from_req (req); if (frame == NULL) { rsp.op_errno = ENOMEM; goto err; } state = frame->root->state; GF_PROTOCOL_DICT_UNSERIALIZE (this, state->xdata, (args.xdata.xdata_val), (args.xdata.xdata_len), ret, op_errno, err); ret = qd_nameless_lookup (this, frame, &args, state->xdata, quotad_aggregator_lookup_cbk); if (ret) { rsp.op_errno = ret; goto err; } return ret; err: rsp.op_ret = -1; rsp.op_errno = op_errno; quotad_aggregator_lookup_cbk (this, frame, &rsp); return ret; } int quotad_aggregator_rpc_notify (rpcsvc_t *rpc, void *xl, rpcsvc_event_t event, void *data) { if (!xl || !data) { gf_log_callingfn ("server", GF_LOG_WARNING, "Calling rpc_notify without initializing"); goto out; } switch (event) { case RPCSVC_EVENT_ACCEPT: break; case RPCSVC_EVENT_DISCONNECT: break; default: break; } out: return 0; } int quotad_aggregator_init (xlator_t *this) { quota_priv_t *priv = NULL; int ret = -1; priv = this->private; ret = dict_set_str (this->options, "transport.address-family", "unix"); if (ret) goto out; ret = dict_set_str (this->options, "transport-type", "socket"); if (ret) goto out; ret = dict_set_str (this->options, "transport.socket.listen-path", "/tmp/quotad.socket"); if (ret) goto out; /* RPC related */ priv->rpcsvc = rpcsvc_init (this, this->ctx, this->options, 0); if (priv->rpcsvc == NULL) { gf_log (this->name, GF_LOG_WARNING, "creation of rpcsvc failed"); ret = -1; goto out; } ret = rpcsvc_create_listeners (priv->rpcsvc, this->options, this->name); if (ret < 1) { gf_log (this->name, GF_LOG_WARNING, "creation of listener failed"); ret = -1; goto out; } priv->quotad_aggregator = "ad_aggregator_prog; quotad_aggregator_prog.options = this->options; ret = rpcsvc_program_register (priv->rpcsvc, "ad_aggregator_prog); if (ret) { gf_log (this->name, GF_LOG_WARNING, "registration of program (name:%s, prognum:%d, " "progver:%d) failed", quotad_aggregator_prog.progname, quotad_aggregator_prog.prognum, quotad_aggregator_prog.progver); goto out; } ret = 0; out: return ret; } rpcsvc_actor_t quotad_aggregator_actors[] = { [GF_AGGREGATOR_NULL] = {"NULL", GF_AGGREGATOR_NULL, NULL, NULL, 0, DRC_NA}, [GF_AGGREGATOR_LOOKUP] = {"LOOKUP", GF_AGGREGATOR_NULL, quotad_aggregator_lookup, NULL, 0, DRC_NA}, }; struct rpcsvc_program quotad_aggregator_prog = { .progname = "GlusterFS 3.3", .prognum = GLUSTER_AGGREGATOR_PROGRAM, .progver = GLUSTER_AGGREGATOR_VERSION, .numactors = GF_AGGREGATOR_MAXVALUE, .actors = quotad_aggregator_actors };