/* Striped store backend

   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 "store.h"

struct stripe_info
{
  struct store **stripes;
  int dealloc : 1;
};

/* Return ADDR adjust for any block size difference between STORE and
   STRIPE.  We assume that STORE's block size is no less than STRIPE's.  */
static inline off_t
addr_adj (off_t addr, struct store *store, struct store *stripe)
{
  unsigned common_bs = store->block_shift;
  unsigned stripe_bs = stripe->block_shift;
  if (common_bs == stripe_bs)
    return addr;
  else
    return addr << (common_bs - stripe_bs);
}

static error_t
stripe_read (struct store *store,
	     off_t addr, size_t index, mach_msg_type_number_t amount,
	     char **buf, mach_msg_type_number_t *len)
{
  struct stripe_info *info = store->hook;
  struct store *stripe = info->stripes[index];
  return store_read (stripe, addr_adj (addr, store, stripe), amount, buf, len);
}

static error_t
stripe_write (struct store *store,
	      off_t addr, size_t index, char *buf, mach_msg_type_number_t len,
	      mach_msg_type_number_t *amount)
{
  struct stripe_info *info = store->hook;
  struct store *stripe = info->stripes[index];
  return
    store_write (stripe, addr_adj (addr, store, stripe), buf, len, amount);
}

static struct store_meths
stripe_meths = {stripe_read, stripe_write};

/* Return a new store in STORE that interleaves all the stores in STRIPES
   (NUM_STRIPES of them) every INTERLEAVE bytes; INTERLEAVE must be an
   integer multiple of each stripe's block size.  If DEALLOC is true, then
   the striped stores are freed when this store is (in any case, the array
   STRIPES is copied, and so should be freed by the caller).  */
error_t
store_ileave_create (struct store **stripes, size_t num_stripes, int dealloc,
		     off_t interleave, struct store **store)
{
  size_t i;
  error_t err = EINVAL;		/* default error */
  off_t block_size = 1, min_end = 0;
  off_t runs[num_stripes * 2];
  struct stripe_info *info = malloc (sizeof (struct stripe_info));

  if (info == 0)
    return ENOMEM;

  info->stripes = malloc (sizeof (struct store *) * num_stripes);
  info->dealloc = dealloc;

  if (info->stripes == 0)
    {
      free (info);
      return ENOMEM;
    }

  /* Find a common block size.  */
  for (i = 0; i < num_stripes; i++)
    block_size = lcm (block_size, stripes[i]->block_size);

  if (interleave < block_size || (interleave % block_size) != 0)
    goto barf;

  interleave /= block_size;

  for (i = 0; i < num_stripes; i++)
    {
       /* The stripe's end adjusted to the common block size.  */
      off_t end = stripes[i]->end;

      runs[i * 2] = 0;
      runs[i * 2 + 1] = interleave;

      if (stripes[i]->block_size != block_size)
	end /= (block_size / stripes[i]->block_size);
  
      if (min_end < 0)
	min_end = end;
      else if (min_end > end)
	/* Only use as much space as the smallest stripe has.  */
	min_end = end;
    }

  *store = _make_store (0, &stripe_meths, MACH_PORT_NULL, block_size,
			runs, num_stripes * 2, min_end);
  if (! *store)
    {
      err = ENOMEM;
      goto barf;
    }

  (*store)->wrap_dst = interleave;
  (*store)->hook = info;
  bcopy (stripes, info->stripes, num_stripes * sizeof *stripes);

  return 0;

 barf:
  free (info->stripes);
  free (info);
  return err;
}

/* Return a new store in STORE that concatenates all the stores in STORES
   (NUM_STORES of them).  If DEALLOC is true, then the sub-stores are freed
   when this store is (in any case, the array STORES is copied, and so should
   be freed by the caller).  */
error_t
store_concat_create (struct store **stores, size_t num_stores, int dealloc,
		     struct store **store)
{
  size_t i;
  error_t err = EINVAL;		/* default error */
  off_t block_size = 1;
  off_t runs[num_stores * 2];
  struct stripe_info *info = malloc (sizeof (struct stripe_info));

  if (info == 0)
    return ENOMEM;

  info->stripes = malloc (sizeof (struct store *) * num_stores);
  info->dealloc = dealloc;

  if (info->stripes == 0)
    {
      free (info);
      return ENOMEM;
    }

  /* Find a common block size.  */
  for (i = 0; i < num_stripes; i++)
    block_size = lcm (block_size, stripes[i]->block_size);

  for (i = 0; i < num_stores; i++)
    {
      runs[i * 2] = 0;
      runs[i * 2 + 1] = stores[i]->end;
    }

  *store = _make_store (0, &stripe_meths, MACH_PORT_NULL, block_size,
			runs, num_stores * 2, 0);
  if (! *store)
    {
      err = ENOMEM;
      goto barf;
    }

  (*store)->hook = info;
  bcopy (stores, info->stripes, num_stores * sizeof *stores);

  return 0;

 barf:
  free (info->stripes);
  free (info);
  return err;
}