diff options
Diffstat (limited to 'glusterfsd/src/fetch-spec.c')
| -rw-r--r-- | glusterfsd/src/fetch-spec.c | 266 |
1 files changed, 266 insertions, 0 deletions
diff --git a/glusterfsd/src/fetch-spec.c b/glusterfsd/src/fetch-spec.c new file mode 100644 index 00000000000..4009a55b3b9 --- /dev/null +++ b/glusterfsd/src/fetch-spec.c @@ -0,0 +1,266 @@ +/* + Copyright (c) 2007, 2008 Z RESEARCH, Inc. <http://www.zresearch.com> + This file is part of GlusterFS. + + GlusterFS is free software; you can redistribute it and/or modify + it under the terms of the GNU General Public License as published + by the Free Software Foundation; either version 3 of the License, + or (at your option) any later version. + + GlusterFS is distributed in the hope that it will be useful, but + WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + General Public License for more details. + + You should have received a copy of the GNU General Public License + along with this program. If not, see + <http://www.gnu.org/licenses/>. +*/ + +#include <stdio.h> +#include <sys/types.h> +#include <sys/wait.h> +#include <stdlib.h> + +#ifndef _CONFIG_H +#define _CONFIG_H +#include "config.h" +#endif /* _CONFIG_H */ + +#include "glusterfs.h" +#include "stack.h" +#include "dict.h" +#include "transport.h" +#include "event.h" +#include "defaults.h" + + +static int +fetch_cbk (call_frame_t *frame, + void *cookie, + xlator_t *this, + int32_t op_ret, + int32_t op_errno, + char *spec_data) +{ + FILE *spec_fp = NULL; + + spec_fp = frame->local; + + if (op_ret >= 0) { + fwrite (spec_data, strlen (spec_data), 1, spec_fp); + fflush (spec_fp); + fclose (spec_fp); + } + else { + gf_log (frame->this->name, GF_LOG_ERROR, + "GETSPEC from server returned -1 (%s)", + strerror (op_errno)); + } + + frame->local = NULL; + STACK_DESTROY (frame->root); + + /* exit the child process */ + exit (op_ret); +} + + +static int +fetch_notify (xlator_t *this_xl, int event, void *data, ...) +{ + int ret = 0; + call_frame_t *frame = NULL; + + switch (event) + { + case GF_EVENT_CHILD_UP: + frame = create_frame (this_xl, this_xl->ctx->pool); + frame->local = this_xl->private; + + STACK_WIND (frame, fetch_cbk, + this_xl->children->xlator, + this_xl->children->xlator->mops->getspec, + this_xl->ctx->cmd_args.volfile_id, + 0); + break; + case GF_EVENT_CHILD_DOWN: + break; + default: + ret = default_notify (this_xl, event, data); + break; + } + + return ret; +} + + +static int +fetch_init (xlator_t *xl) +{ + return 0; +} + +static xlator_t * +get_shrub (glusterfs_ctx_t *ctx, + const char *remote_host, + const char *transport, + uint32_t remote_port) +{ + int ret = 0; + xlator_t *top = NULL; + xlator_t *trans = NULL; + xlator_list_t *parent = NULL, *tmp = NULL; + + top = CALLOC (1, sizeof (*top)); + ERR_ABORT (top); + trans = CALLOC (1, sizeof (*trans)); + ERR_ABORT (trans); + + top->name = "top"; + top->ctx = ctx; + top->next = trans; + top->init = fetch_init; + top->notify = fetch_notify; + top->children = (void *) CALLOC (1, sizeof (*top->children)); + ERR_ABORT (top->children); + top->children->xlator = trans; + + trans->name = "trans"; + trans->ctx = ctx; + trans->prev = top; + trans->init = fetch_init; + trans->notify = default_notify; + trans->options = get_new_dict (); + + parent = CALLOC (1, sizeof(*parent)); + parent->xlator = top; + if (trans->parents == NULL) + trans->parents = parent; + else { + tmp = trans->parents; + while (tmp->next) + tmp = tmp->next; + tmp->next = parent; + } + + /* TODO: log on failure to set dict */ + if (remote_host) + ret = dict_set_static_ptr (trans->options, "remote-host", + (char *)remote_host); + + if (remote_port) + ret = dict_set_uint32 (trans->options, "remote-port", + remote_port); + + /* 'option remote-subvolume <x>' is needed here even though + * its not used + */ + ret = dict_set_static_ptr (trans->options, "remote-subvolume", + "brick"); + ret = dict_set_static_ptr (trans->options, "disable-handshake", "on"); + ret = dict_set_static_ptr (trans->options, "non-blocking-io", "off"); + + if (transport) { + char *transport_type = CALLOC (1, strlen (transport) + 10); + ERR_ABORT (transport_type); + strcpy(transport_type, transport); + + if (strchr (transport_type, ':')) + *(strchr (transport_type, ':')) = '\0'; + + ret = dict_set_dynstr (trans->options, "transport-type", + transport_type); + } + + xlator_set_type (trans, "protocol/client"); + + if (xlator_tree_init (top) != 0) + return NULL; + + return top; +} + + +static int +_fetch (glusterfs_ctx_t *ctx, + FILE *spec_fp, + const char *remote_host, + const char *transport, + uint32_t remote_port) +{ + xlator_t *this = NULL; + + this = get_shrub (ctx, remote_host, transport, remote_port); + if (this == NULL) + return -1; + + this->private = spec_fp; + + event_dispatch (ctx->event_pool); + + return 0; +} + + +static int +_fork_and_fetch (glusterfs_ctx_t *ctx, + FILE *spec_fp, + const char *remote_host, + const char *transport, + uint32_t remote_port) +{ + int ret; + + ret = fork (); + switch (ret) { + case -1: + perror ("fork()"); + break; + case 0: + /* child */ + ret = _fetch (ctx, spec_fp, remote_host, + transport, remote_port); + if (ret == -1) + exit (ret); + default: + /* parent */ + wait (&ret); + ret = WEXITSTATUS (ret); + } + return ret; +} + +FILE * +fetch_spec (glusterfs_ctx_t *ctx) +{ + char *remote_host = NULL; + char *transport = NULL; + FILE *spec_fp; + int32_t ret; + + spec_fp = tmpfile (); + + if (!spec_fp) { + perror ("tmpfile ()"); + return NULL; + } + + remote_host = ctx->cmd_args.volfile_server; + transport = ctx->cmd_args.volfile_server_transport; + if (!transport) + transport = "socket"; + + ret = _fork_and_fetch (ctx, spec_fp, remote_host, transport, + ctx->cmd_args.volfile_server_port); + + if (!ret) { + fseek (spec_fp, 0, SEEK_SET); + } + else { + fclose (spec_fp); + spec_fp = NULL; + } + + return spec_fp; +} |