summaryrefslogtreecommitdiff
path: root/libstore/argp.c
blob: 70c05759c512ac9b2611fe495698e89cb0ccf379 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
/* 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 <argz.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[] = "\vIf 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 ERR_RETURN is true, also free the stores in H's
   store vector, and any other return values, otherwise just free the vector
   itself.  */
static void
free_hook (struct store_parse_hook *h, int err_return)
{
  int i;
  if (err_return)
    for (i = 0; i < h->num_stores; i++)
      store_free (h->stores[i]);
  if (h->stores)
    free (h->stores);
  if (err_return && h->params->return_args && h->params->args)
    free (h->params->args);
  free (h);
}

static error_t
open_file (char *name, struct store_parse_hook *h, struct store **s)
{
  error_t err;
  int flags = h->params->flags;
  int open_flags = (flags & STORE_HARD_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, flags, 0, s);
  if (err)
    {
      if (! h->params->no_file_io)
	/* Try making a store that does file io to NODE.  */
	err = store_file_create (node, flags, 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 PERR(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)
	PERR (EINVAL, "--layer and --interleave are exclusive");
      if (h->interleave)
	/* Actually no reason why we couldn't support this.... */
	PERR (EINVAL, "--interleave specified multiple times");

      h->interleave = atoi (arg);
      if (! h->interleave)
	PERR (EINVAL, "%s: Bad value for --interleave", arg);
      break;

    case 'l':
      if (h->interleave)
	PERR (EINVAL, "--layer and --interleave are exclusive");
      h->layer = 1;
      break;

    case ARGP_KEY_ARG:
      /* A store device to use!  */
      if (h->machdev)
	err = store_device_open (arg, h->params->flags, &s);
      else
	err = open_file (arg, h, &s);
      if (! err)
	{
	  struct store **stores = realloc (h->stores, h->num_stores + 1);
	  if (stores)
	    {
	      stores[h->num_stores++] = s;
	      h->stores = stores;
	      if (h->params->return_args)
		err = argz_add (&h->params->args, &h->params->args_len, arg);
	    }
	  else
	    err = ENOMEM;	/* Just fucking lovely */
	}
      if (err)
	{
	  argp_failure (state, 1, err, "%s", arg);
	  return err;
	}
      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;
      if (h->params->return_args)
	/* Initialiaze the returned argument vector.  */
	{
	  h->params->args = 0;
	  h->params->args_len = 0;
	}
      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);
	  PERR (EINVAL, "No store specified");
	}

      if (h->params->return_args)
	{
	  if (h->machdev)
	    err = argz_insert (&h->params->args, &h->params->args_len,
			       h->params->args, "--machdev");
	  if (!err && h->num_stores > 1 && (h->interleave || h->layer))
	    {
	      char buf[40];
	      if (h->interleave)
		snprintf (buf, sizeof buf, "--interleave=%ld", h->interleave);
	      else
		snprintf (buf, sizeof buf, "--layer=%ld", h->layer);
	      err = argz_insert (&h->params->args, &h->params->args_len,
				 h->params->args, buf);
	    }
	}

      if (err)
	/* nothing */;
      else 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,
			       h->params->flags, &s);
      else if (h->layer)
	{
	  free_hook (h, 1);
	  PERR (EINVAL, "--layer not implemented");
	}
      else
	err =
	  store_concat_create (h->stores, h->num_stores, h->params->flags, &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 };