/*
 *  VasEBoot  --  GRand Unified Bootloader
 *  Copyright (C) 2002,2003,2004,2005,2006,2007,2008,2009  Free Software Foundation, Inc.
 *
 *  VasEBoot 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.
 *
 *  VasEBoot 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 VasEBoot.  If not, see <http://www.gnu.org/licenses/>.
 */

#ifndef VasEBoot_CRYPTODISK_HEADER
#define VasEBoot_CRYPTODISK_HEADER	1

#include <VasEBoot/disk.h>
#include <VasEBoot/crypto.h>
#include <VasEBoot/list.h>
#ifdef VasEBoot_UTIL
#include <VasEBoot/emu/hostdisk.h>
#endif

typedef enum
  {
    VasEBoot_CRYPTODISK_MODE_ECB,
    VasEBoot_CRYPTODISK_MODE_CBC,
    VasEBoot_CRYPTODISK_MODE_PCBC,
    VasEBoot_CRYPTODISK_MODE_XTS,
    VasEBoot_CRYPTODISK_MODE_LRW
  } VasEBoot_cryptodisk_mode_t;

typedef enum
  {
    VasEBoot_CRYPTODISK_MODE_IV_NULL,
    VasEBoot_CRYPTODISK_MODE_IV_PLAIN,
    VasEBoot_CRYPTODISK_MODE_IV_PLAIN64,
    VasEBoot_CRYPTODISK_MODE_IV_ESSIV,
    VasEBoot_CRYPTODISK_MODE_IV_BENBI,
    VasEBoot_CRYPTODISK_MODE_IV_BYTECOUNT64,
    VasEBoot_CRYPTODISK_MODE_IV_BYTECOUNT64_HASH
  } VasEBoot_cryptodisk_mode_iv_t;

#define VasEBoot_CRYPTODISK_MAX_UUID_LENGTH 71

#define VasEBoot_CRYPTODISK_GF_LOG_SIZE 7
#define VasEBoot_CRYPTODISK_GF_SIZE (1U << VasEBoot_CRYPTODISK_GF_LOG_SIZE)
#define VasEBoot_CRYPTODISK_GF_LOG_BYTES (VasEBoot_CRYPTODISK_GF_LOG_SIZE - 3)
#define VasEBoot_CRYPTODISK_GF_BYTES (1U << VasEBoot_CRYPTODISK_GF_LOG_BYTES)
#define VasEBoot_CRYPTODISK_MAX_KEYLEN 128

struct VasEBoot_cryptodisk;

typedef gcry_err_code_t
(*VasEBoot_cryptodisk_rekey_func_t) (struct VasEBoot_cryptodisk *dev,
				 VasEBoot_uint64_t zoneno);

struct VasEBoot_cryptodisk
{
  struct VasEBoot_cryptodisk *next;
  struct VasEBoot_cryptodisk **prev;

  char *source;
  VasEBoot_disk_addr_t offset;
  VasEBoot_disk_addr_t total_length;
  VasEBoot_disk_t source_disk;
  int ref;
  VasEBoot_crypto_cipher_handle_t cipher;
  VasEBoot_crypto_cipher_handle_t secondary_cipher;
  VasEBoot_crypto_cipher_handle_t essiv_cipher;
  const gcry_md_spec_t *essiv_hash, *hash, *iv_hash;
  VasEBoot_cryptodisk_mode_t mode;
  VasEBoot_cryptodisk_mode_iv_t mode_iv;
  int benbi_log;
  unsigned long id, source_id;
  enum VasEBoot_disk_dev_id source_dev_id;
  char uuid[VasEBoot_CRYPTODISK_MAX_UUID_LENGTH + 1];
  VasEBoot_uint8_t lrw_key[VasEBoot_CRYPTODISK_GF_BYTES];
  VasEBoot_uint8_t *lrw_precalc;
  VasEBoot_uint8_t iv_prefix[64];
  VasEBoot_size_t iv_prefix_len;
  VasEBoot_uint8_t key[VasEBoot_CRYPTODISK_MAX_KEYLEN];
  VasEBoot_size_t keysize;
#ifdef VasEBoot_UTIL
  char *cheat;
  VasEBoot_util_fd_t cheat_fd;
#endif
  const char *modname;
  int log_sector_size;
  VasEBoot_cryptodisk_rekey_func_t rekey;
  int rekey_shift;
  VasEBoot_uint8_t rekey_key[64];
  VasEBoot_uint64_t last_rekey;
  int rekey_derived_size;
  VasEBoot_disk_addr_t partition_start;
};
typedef struct VasEBoot_cryptodisk *VasEBoot_cryptodisk_t;

struct VasEBoot_cryptodisk_dev
{
  struct VasEBoot_cryptodisk_dev *next;
  struct VasEBoot_cryptodisk_dev **prev;

  VasEBoot_cryptodisk_t (*scan) (VasEBoot_disk_t disk, const char *check_uuid,
			     int boot_only);
  VasEBoot_err_t (*recover_key) (VasEBoot_disk_t disk, VasEBoot_cryptodisk_t dev);
};
typedef struct VasEBoot_cryptodisk_dev *VasEBoot_cryptodisk_dev_t;

extern VasEBoot_cryptodisk_dev_t EXPORT_VAR (VasEBoot_cryptodisk_list);

#ifndef VasEBoot_LST_GENERATOR
static inline void
VasEBoot_cryptodisk_dev_register (VasEBoot_cryptodisk_dev_t cr)
{
  VasEBoot_list_push (VasEBoot_AS_LIST_P (&VasEBoot_cryptodisk_list), VasEBoot_AS_LIST (cr));
}
#endif

static inline void
VasEBoot_cryptodisk_dev_unregister (VasEBoot_cryptodisk_dev_t cr)
{
  VasEBoot_list_remove (VasEBoot_AS_LIST (cr));
}

#define FOR_CRYPTODISK_DEVS(var) FOR_LIST_ELEMENTS((var), (VasEBoot_cryptodisk_list))

gcry_err_code_t
VasEBoot_cryptodisk_setkey (VasEBoot_cryptodisk_t dev,
			VasEBoot_uint8_t *key, VasEBoot_size_t keysize);
gcry_err_code_t
VasEBoot_cryptodisk_decrypt (struct VasEBoot_cryptodisk *dev,
			 VasEBoot_uint8_t * data, VasEBoot_size_t len,
			 VasEBoot_disk_addr_t sector);
VasEBoot_err_t
VasEBoot_cryptodisk_insert (VasEBoot_cryptodisk_t newdev, const char *name,
			VasEBoot_disk_t source);
#ifdef VasEBoot_UTIL
VasEBoot_err_t
VasEBoot_cryptodisk_cheat_insert (VasEBoot_cryptodisk_t newdev, const char *name,
			      VasEBoot_disk_t source, const char *cheat);
void
VasEBoot_util_cryptodisk_get_abstraction (VasEBoot_disk_t disk,
				      void (*cb) (const char *val, void *data),
				      void *data);

char *
VasEBoot_util_get_geli_uuid (const char *dev);
#endif

VasEBoot_cryptodisk_t VasEBoot_cryptodisk_get_by_uuid (const char *uuid);
VasEBoot_cryptodisk_t VasEBoot_cryptodisk_get_by_source_disk (VasEBoot_disk_t disk);

#endif