diff options
| -rw-r--r-- | libstore/argp.c | 229 |
1 files changed, 229 insertions, 0 deletions
diff --git a/libstore/argp.c b/libstore/argp.c new file mode 100644 index 00000000..dc0d2aec --- /dev/null +++ b/libstore/argp.c @@ -0,0 +1,229 @@ +/* Store argument parsing + + Copyright (C) 1996 Free Software Foundation, Inc. + + Written by Miles Bader <miles@gnu.ai.mit.edu> + + This file is part of the GNU Hurd. + + The GNU Hurd 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 2, or (at + your option) any later version. + + The GNU Hurd 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, write to the Free Software Foundation, Inc., + 675 Mass Ave, Cambridge, MA 02139, USA. */ + +#include <stdlib.h> +#include <string.h> +#include <fcntl.h> +#include <hurd.h> +#include <argp.h> + +#include "store.h" + +static const struct argp_option options[] = { + {"machdev", 'm', 0, 0, "DEVICE is a mach device, not a file"}, + {"interleave", 'i', "BLOCKS", 0, "Interleave in runs of length BLOCKS"}, + {"layer", 'l', 0, 0, "Layer multiple devices for redundancy"}, + {0} +}; + +static const char args_doc[] = "DEVICE..."; +static const char doc[] = "If multiple DEVICEs are specified, they are" +" concatenated unless either --interleave or --layer is specified (mutually" +" exlusive)."; + +/* Used to hold data during argument parsing. */ +struct store_parse_hook +{ + /* A malloced vector of stores specified on the command line, NUM_STORES + long. */ + struct store **stores; + size_t num_stores; + + /* Pointer to params struct passed in by user. */ + struct store_argp_params *params; + + off_t interleave; /* --interleave value */ + int machdev : 1; /* --machdev specified */ + int layer : 1; /* --layer specified */ +}; + +/* Free the parse hook H. If FREE_STORES is true, also free the stores in + H's store vector, otherwise just free the vector itself. */ +static void +free_hook (struct store_parse_hook *h, int free_stores) +{ + int i; + if (free_stores) + for (i = 0; i < h->num_stores; i++) + store_free (h->stores[i]); + if (h->stores) + free (h->stores); + free (h); +} + +static error_t +open_machdev (char *name, struct store_parse_hook *h, struct store **s) +{ + device_t dev_master, device; + int open_flags = (h->params->readonly ? 0 : D_WRITE) | D_READ; + error_t err = get_privileged_ports (0, &dev_master); + + if (err) + return err; + + err = device_open (dev_master, open_flags, name, &device); + + mach_port_deallocate (mach_task_self (), dev_master); + + err = store_device_create (device, s); + if (err) + mach_port_deallocate (mach_task_self (), device); + + return err; +} + +static error_t +open_file (char *name, struct store_parse_hook *h, struct store **s) +{ + error_t err; + int open_flags = h->params->readonly ? O_RDONLY : O_RDWR; + file_t node = file_name_lookup (name, open_flags, 0); + + if (node == MACH_PORT_NULL) + return errno; + + err = store_create (node, s); + if (err) + { + if (! h->params->no_file_io) + /* Try making a store that does file io to NODE. */ + err = store_file_create (node, s); + if (err) + mach_port_deallocate (mach_task_self (), node); + } + + return err; +} + +static error_t +parse_opt (int opt, char *arg, struct argp_state *state) +{ + error_t err = 0; + struct store_parse_hook *h = state->hook; + + /* Print a parsing error message and (if exiting is turned off) return the + error code ERR. */ +#define ERR(err, fmt, args...) \ + do { argp_error (state, fmt , ##args); return err; } while (0) + + switch (opt) + { + struct store *s; + + case 'm': + h->machdev = 1; break; + + case 'i': + if (h->layer) + ERR (EINVAL, "--layer and --interleave are exclusive"); + if (h->interleave) + /* Actually no reason why we couldn't support this.... */ + ERR (EINVAL, "--interleave specified multiple times"); + + h->interleave = atoi (arg); + if (! h->interleave) + ERR (EINVAL, "%s: Bad value for --interleave", arg); + break; + + case 'l': + if (h->interleave) + ERR (EINVAL, "--layer and --interleave are exclusive"); + h->layer = 1; + break; + + case ARGP_KEY_ARG: + /* A store device to use! */ + if (h->machdev) + err = open_machdev (arg, h, &s); + else + err = open_file (arg, h, &s); + if (err) + ERR (err, "%s: %s", arg, strerror (err)); + else + { + struct store **stores = realloc (h->stores, h->num_stores + 1); + if (stores) + { + stores[h->num_stores++] = s; + h->stores = stores; + } + else + return ENOMEM; /* Just fucking lovely */ + } + break; + + case ARGP_KEY_INIT: + /* Initialize our parsing state. */ + if (! state->input) + return EINVAL; /* Need at least a way to return a result. */ + h = malloc (sizeof (struct store_parse_hook)); + if (! h) + return ENOMEM; + bzero (h, sizeof (struct store_parse_hook)); + h->params = state->input; + state->hook = h; + break; + + case ARGP_KEY_ERROR: + /* Parsing error occured, free everything. */ + free_hook (h, 1); break; + + case ARGP_KEY_SUCCESS: + /* Successfully finished parsing, return a result. */ + + if (h->num_stores == 0) + { + free_hook (h, 1); + ERR (EINVAL, "No store specified"); + } + + if (state->input == 0) + /* No way to return a value! */ + err = EINVAL; + else if (h->num_stores == 1) + s = h->stores[0]; /* Just a single store. */ + else if (h->interleave) + err = + store_ileave_create (h->stores, h->num_stores, h->interleave, &s); + else if (h->layer) + { + free_hook (h, 1); + ERR (EINVAL, "--layer not implemented"); + } + else + err = store_concat_create (h->stores, h->num_stores, &s); + + free_hook (h, err); + if (! err) + h->params->result = s; + + break; + + default: + return ARGP_ERR_UNKNOWN; + } + + return 0; +} + +struct argp +store_argp = { options, parse_opt, args_doc, doc }; |