/*
   * 2007+ Copyright (c) Evgeniy Polyakov <johnpol@2ka.mipt.ru>
   * All rights reserved.
   *
   * This program 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 of the License, or
   * (at your option) any later version.
   *
   * This program 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.

   */
  
  #include <linux/kernel.h>
  #include <linux/module.h>

  #include <linux/moduleparam.h>

  #include <linux/mod_devicetable.h>
  #include <linux/interrupt.h>
  #include <linux/pci.h>
  #include <linux/slab.h>
  #include <linux/delay.h>
  #include <linux/mm.h>

  #include <linux/dma-mapping.h>
  #include <linux/scatterlist.h>

  #include <linux/highmem.h>
  #include <linux/crypto.h>

  #include <linux/hw_random.h>
  #include <linux/ktime.h>

  
  #include <crypto/algapi.h>

  #include <crypto/des.h>

  static char hifn_pll_ref[sizeof("extNNN")] = "ext";
  module_param_string(hifn_pll_ref, hifn_pll_ref, sizeof(hifn_pll_ref), 0444);
  MODULE_PARM_DESC(hifn_pll_ref,
  		 "PLL reference clock (pci[freq] or ext[freq], default ext)");

  static atomic_t hifn_dev_number;
  
  #define ACRYPTO_OP_DECRYPT	0
  #define ACRYPTO_OP_ENCRYPT	1
  #define ACRYPTO_OP_HMAC		2
  #define ACRYPTO_OP_RNG		3
  
  #define ACRYPTO_MODE_ECB		0
  #define ACRYPTO_MODE_CBC		1
  #define ACRYPTO_MODE_CFB		2
  #define ACRYPTO_MODE_OFB		3
  
  #define ACRYPTO_TYPE_AES_128	0
  #define ACRYPTO_TYPE_AES_192	1
  #define ACRYPTO_TYPE_AES_256	2
  #define ACRYPTO_TYPE_3DES	3
  #define ACRYPTO_TYPE_DES	4
  
  #define PCI_VENDOR_ID_HIFN		0x13A3
  #define PCI_DEVICE_ID_HIFN_7955		0x0020
  #define	PCI_DEVICE_ID_HIFN_7956		0x001d
  
  /* I/O region sizes */
  
  #define HIFN_BAR0_SIZE			0x1000
  #define HIFN_BAR1_SIZE			0x2000
  #define HIFN_BAR2_SIZE			0x8000
  
  /* DMA registres */

  #define HIFN_DMA_CRA			0x0C	/* DMA Command Ring Address */
  #define HIFN_DMA_SDRA			0x1C	/* DMA Source Data Ring Address */

  #define HIFN_DMA_RRA			0x2C	/* DMA Result Ring Address */
  #define HIFN_DMA_DDRA			0x3C	/* DMA Destination Data Ring Address */
  #define HIFN_DMA_STCTL			0x40	/* DMA Status and Control */

  #define HIFN_DMA_INTREN			0x44	/* DMA Interrupt Enable */

  #define HIFN_DMA_CFG1			0x48	/* DMA Configuration #1 */
  #define HIFN_DMA_CFG2			0x6C	/* DMA Configuration #2 */
  #define HIFN_CHIP_ID			0x98	/* Chip ID */
  
  /*
   * Processing Unit Registers (offset from BASEREG0)
   */
  #define	HIFN_0_PUDATA		0x00	/* Processing Unit Data */
  #define	HIFN_0_PUCTRL		0x04	/* Processing Unit Control */
  #define	HIFN_0_PUISR		0x08	/* Processing Unit Interrupt Status */
  #define	HIFN_0_PUCNFG		0x0c	/* Processing Unit Configuration */
  #define	HIFN_0_PUIER		0x10	/* Processing Unit Interrupt Enable */
  #define	HIFN_0_PUSTAT		0x14	/* Processing Unit Status/Chip ID */
  #define	HIFN_0_FIFOSTAT		0x18	/* FIFO Status */
  #define	HIFN_0_FIFOCNFG		0x1c	/* FIFO Configuration */
  #define	HIFN_0_SPACESIZE	0x20	/* Register space size */
  
  /* Processing Unit Control Register (HIFN_0_PUCTRL) */
  #define	HIFN_PUCTRL_CLRSRCFIFO	0x0010	/* clear source fifo */
  #define	HIFN_PUCTRL_STOP	0x0008	/* stop pu */
  #define	HIFN_PUCTRL_LOCKRAM	0x0004	/* lock ram */
  #define	HIFN_PUCTRL_DMAENA	0x0002	/* enable dma */
  #define	HIFN_PUCTRL_RESET	0x0001	/* Reset processing unit */
  
  /* Processing Unit Interrupt Status Register (HIFN_0_PUISR) */
  #define	HIFN_PUISR_CMDINVAL	0x8000	/* Invalid command interrupt */
  #define	HIFN_PUISR_DATAERR	0x4000	/* Data error interrupt */
  #define	HIFN_PUISR_SRCFIFO	0x2000	/* Source FIFO ready interrupt */
  #define	HIFN_PUISR_DSTFIFO	0x1000	/* Destination FIFO ready interrupt */
  #define	HIFN_PUISR_DSTOVER	0x0200	/* Destination overrun interrupt */
  #define	HIFN_PUISR_SRCCMD	0x0080	/* Source command interrupt */
  #define	HIFN_PUISR_SRCCTX	0x0040	/* Source context interrupt */
  #define	HIFN_PUISR_SRCDATA	0x0020	/* Source data interrupt */
  #define	HIFN_PUISR_DSTDATA	0x0010	/* Destination data interrupt */
  #define	HIFN_PUISR_DSTRESULT	0x0004	/* Destination result interrupt */
  
  /* Processing Unit Configuration Register (HIFN_0_PUCNFG) */
  #define	HIFN_PUCNFG_DRAMMASK	0xe000	/* DRAM size mask */
  #define	HIFN_PUCNFG_DSZ_256K	0x0000	/* 256k dram */
  #define	HIFN_PUCNFG_DSZ_512K	0x2000	/* 512k dram */
  #define	HIFN_PUCNFG_DSZ_1M	0x4000	/* 1m dram */
  #define	HIFN_PUCNFG_DSZ_2M	0x6000	/* 2m dram */
  #define	HIFN_PUCNFG_DSZ_4M	0x8000	/* 4m dram */
  #define	HIFN_PUCNFG_DSZ_8M	0xa000	/* 8m dram */
  #define	HIFN_PUNCFG_DSZ_16M	0xc000	/* 16m dram */
  #define	HIFN_PUCNFG_DSZ_32M	0xe000	/* 32m dram */
  #define	HIFN_PUCNFG_DRAMREFRESH	0x1800	/* DRAM refresh rate mask */
  #define	HIFN_PUCNFG_DRFR_512	0x0000	/* 512 divisor of ECLK */
  #define	HIFN_PUCNFG_DRFR_256	0x0800	/* 256 divisor of ECLK */
  #define	HIFN_PUCNFG_DRFR_128	0x1000	/* 128 divisor of ECLK */
  #define	HIFN_PUCNFG_TCALLPHASES	0x0200	/* your guess is as good as mine... */
  #define	HIFN_PUCNFG_TCDRVTOTEM	0x0100	/* your guess is as good as mine... */
  #define	HIFN_PUCNFG_BIGENDIAN	0x0080	/* DMA big endian mode */
  #define	HIFN_PUCNFG_BUS32	0x0040	/* Bus width 32bits */
  #define	HIFN_PUCNFG_BUS16	0x0000	/* Bus width 16 bits */
  #define	HIFN_PUCNFG_CHIPID	0x0020	/* Allow chipid from PUSTAT */
  #define	HIFN_PUCNFG_DRAM	0x0010	/* Context RAM is DRAM */
  #define	HIFN_PUCNFG_SRAM	0x0000	/* Context RAM is SRAM */
  #define	HIFN_PUCNFG_COMPSING	0x0004	/* Enable single compression context */
  #define	HIFN_PUCNFG_ENCCNFG	0x0002	/* Encryption configuration */
  
  /* Processing Unit Interrupt Enable Register (HIFN_0_PUIER) */
  #define	HIFN_PUIER_CMDINVAL	0x8000	/* Invalid command interrupt */
  #define	HIFN_PUIER_DATAERR	0x4000	/* Data error interrupt */
  #define	HIFN_PUIER_SRCFIFO	0x2000	/* Source FIFO ready interrupt */
  #define	HIFN_PUIER_DSTFIFO	0x1000	/* Destination FIFO ready interrupt */
  #define	HIFN_PUIER_DSTOVER	0x0200	/* Destination overrun interrupt */
  #define	HIFN_PUIER_SRCCMD	0x0080	/* Source command interrupt */
  #define	HIFN_PUIER_SRCCTX	0x0040	/* Source context interrupt */
  #define	HIFN_PUIER_SRCDATA	0x0020	/* Source data interrupt */
  #define	HIFN_PUIER_DSTDATA	0x0010	/* Destination data interrupt */
  #define	HIFN_PUIER_DSTRESULT	0x0004	/* Destination result interrupt */
  
  /* Processing Unit Status Register/Chip ID (HIFN_0_PUSTAT) */
  #define	HIFN_PUSTAT_CMDINVAL	0x8000	/* Invalid command interrupt */
  #define	HIFN_PUSTAT_DATAERR	0x4000	/* Data error interrupt */
  #define	HIFN_PUSTAT_SRCFIFO	0x2000	/* Source FIFO ready interrupt */
  #define	HIFN_PUSTAT_DSTFIFO	0x1000	/* Destination FIFO ready interrupt */
  #define	HIFN_PUSTAT_DSTOVER	0x0200	/* Destination overrun interrupt */
  #define	HIFN_PUSTAT_SRCCMD	0x0080	/* Source command interrupt */
  #define	HIFN_PUSTAT_SRCCTX	0x0040	/* Source context interrupt */
  #define	HIFN_PUSTAT_SRCDATA	0x0020	/* Source data interrupt */
  #define	HIFN_PUSTAT_DSTDATA	0x0010	/* Destination data interrupt */
  #define	HIFN_PUSTAT_DSTRESULT	0x0004	/* Destination result interrupt */
  #define	HIFN_PUSTAT_CHIPREV	0x00ff	/* Chip revision mask */
  #define	HIFN_PUSTAT_CHIPENA	0xff00	/* Chip enabled mask */
  #define	HIFN_PUSTAT_ENA_2	0x1100	/* Level 2 enabled */
  #define	HIFN_PUSTAT_ENA_1	0x1000	/* Level 1 enabled */
  #define	HIFN_PUSTAT_ENA_0	0x3000	/* Level 0 enabled */
  #define	HIFN_PUSTAT_REV_2	0x0020	/* 7751 PT6/2 */
  #define	HIFN_PUSTAT_REV_3	0x0030	/* 7751 PT6/3 */
  
  /* FIFO Status Register (HIFN_0_FIFOSTAT) */
  #define	HIFN_FIFOSTAT_SRC	0x7f00	/* Source FIFO available */
  #define	HIFN_FIFOSTAT_DST	0x007f	/* Destination FIFO available */
  
  /* FIFO Configuration Register (HIFN_0_FIFOCNFG) */
  #define	HIFN_FIFOCNFG_THRESHOLD	0x0400	/* must be written as 1 */
  
  /*
   * DMA Interface Registers (offset from BASEREG1)
   */
  #define	HIFN_1_DMA_CRAR		0x0c	/* DMA Command Ring Address */
  #define	HIFN_1_DMA_SRAR		0x1c	/* DMA Source Ring Address */
  #define	HIFN_1_DMA_RRAR		0x2c	/* DMA Result Ring Address */
  #define	HIFN_1_DMA_DRAR		0x3c	/* DMA Destination Ring Address */
  #define	HIFN_1_DMA_CSR		0x40	/* DMA Status and Control */
  #define	HIFN_1_DMA_IER		0x44	/* DMA Interrupt Enable */
  #define	HIFN_1_DMA_CNFG		0x48	/* DMA Configuration */
  #define	HIFN_1_PLL		0x4c	/* 795x: PLL config */
  #define	HIFN_1_7811_RNGENA	0x60	/* 7811: rng enable */
  #define	HIFN_1_7811_RNGCFG	0x64	/* 7811: rng config */
  #define	HIFN_1_7811_RNGDAT	0x68	/* 7811: rng data */
  #define	HIFN_1_7811_RNGSTS	0x6c	/* 7811: rng status */
  #define	HIFN_1_7811_MIPSRST	0x94	/* 7811: MIPS reset */
  #define	HIFN_1_REVID		0x98	/* Revision ID */
  #define	HIFN_1_UNLOCK_SECRET1	0xf4
  #define	HIFN_1_UNLOCK_SECRET2	0xfc
  #define	HIFN_1_PUB_RESET	0x204	/* Public/RNG Reset */
  #define	HIFN_1_PUB_BASE		0x300	/* Public Base Address */
  #define	HIFN_1_PUB_OPLEN	0x304	/* Public Operand Length */
  #define	HIFN_1_PUB_OP		0x308	/* Public Operand */
  #define	HIFN_1_PUB_STATUS	0x30c	/* Public Status */
  #define	HIFN_1_PUB_IEN		0x310	/* Public Interrupt enable */
  #define	HIFN_1_RNG_CONFIG	0x314	/* RNG config */
  #define	HIFN_1_RNG_DATA		0x318	/* RNG data */
  #define	HIFN_1_PUB_MEM		0x400	/* start of Public key memory */
  #define	HIFN_1_PUB_MEMEND	0xbff	/* end of Public key memory */
  
  /* DMA Status and Control Register (HIFN_1_DMA_CSR) */
  #define	HIFN_DMACSR_D_CTRLMASK	0xc0000000	/* Destinition Ring Control */
  #define	HIFN_DMACSR_D_CTRL_NOP	0x00000000	/* Dest. Control: no-op */
  #define	HIFN_DMACSR_D_CTRL_DIS	0x40000000	/* Dest. Control: disable */
  #define	HIFN_DMACSR_D_CTRL_ENA	0x80000000	/* Dest. Control: enable */
  #define	HIFN_DMACSR_D_ABORT	0x20000000	/* Destinition Ring PCIAbort */
  #define	HIFN_DMACSR_D_DONE	0x10000000	/* Destinition Ring Done */
  #define	HIFN_DMACSR_D_LAST	0x08000000	/* Destinition Ring Last */
  #define	HIFN_DMACSR_D_WAIT	0x04000000	/* Destinition Ring Waiting */
  #define	HIFN_DMACSR_D_OVER	0x02000000	/* Destinition Ring Overflow */
  #define	HIFN_DMACSR_R_CTRL	0x00c00000	/* Result Ring Control */
  #define	HIFN_DMACSR_R_CTRL_NOP	0x00000000	/* Result Control: no-op */
  #define	HIFN_DMACSR_R_CTRL_DIS	0x00400000	/* Result Control: disable */
  #define	HIFN_DMACSR_R_CTRL_ENA	0x00800000	/* Result Control: enable */
  #define	HIFN_DMACSR_R_ABORT	0x00200000	/* Result Ring PCI Abort */
  #define	HIFN_DMACSR_R_DONE	0x00100000	/* Result Ring Done */
  #define	HIFN_DMACSR_R_LAST	0x00080000	/* Result Ring Last */
  #define	HIFN_DMACSR_R_WAIT	0x00040000	/* Result Ring Waiting */
  #define	HIFN_DMACSR_R_OVER	0x00020000	/* Result Ring Overflow */
  #define	HIFN_DMACSR_S_CTRL	0x0000c000	/* Source Ring Control */
  #define	HIFN_DMACSR_S_CTRL_NOP	0x00000000	/* Source Control: no-op */
  #define	HIFN_DMACSR_S_CTRL_DIS	0x00004000	/* Source Control: disable */
  #define	HIFN_DMACSR_S_CTRL_ENA	0x00008000	/* Source Control: enable */
  #define	HIFN_DMACSR_S_ABORT	0x00002000	/* Source Ring PCI Abort */
  #define	HIFN_DMACSR_S_DONE	0x00001000	/* Source Ring Done */
  #define	HIFN_DMACSR_S_LAST	0x00000800	/* Source Ring Last */
  #define	HIFN_DMACSR_S_WAIT	0x00000400	/* Source Ring Waiting */
  #define	HIFN_DMACSR_ILLW	0x00000200	/* Illegal write (7811 only) */
  #define	HIFN_DMACSR_ILLR	0x00000100	/* Illegal read (7811 only) */
  #define	HIFN_DMACSR_C_CTRL	0x000000c0	/* Command Ring Control */
  #define	HIFN_DMACSR_C_CTRL_NOP	0x00000000	/* Command Control: no-op */
  #define	HIFN_DMACSR_C_CTRL_DIS	0x00000040	/* Command Control: disable */
  #define	HIFN_DMACSR_C_CTRL_ENA	0x00000080	/* Command Control: enable */
  #define	HIFN_DMACSR_C_ABORT	0x00000020	/* Command Ring PCI Abort */
  #define	HIFN_DMACSR_C_DONE	0x00000010	/* Command Ring Done */
  #define	HIFN_DMACSR_C_LAST	0x00000008	/* Command Ring Last */
  #define	HIFN_DMACSR_C_WAIT	0x00000004	/* Command Ring Waiting */
  #define	HIFN_DMACSR_PUBDONE	0x00000002	/* Public op done (7951 only) */
  #define	HIFN_DMACSR_ENGINE	0x00000001	/* Command Ring Engine IRQ */
  
  /* DMA Interrupt Enable Register (HIFN_1_DMA_IER) */
  #define	HIFN_DMAIER_D_ABORT	0x20000000	/* Destination Ring PCIAbort */
  #define	HIFN_DMAIER_D_DONE	0x10000000	/* Destination Ring Done */
  #define	HIFN_DMAIER_D_LAST	0x08000000	/* Destination Ring Last */
  #define	HIFN_DMAIER_D_WAIT	0x04000000	/* Destination Ring Waiting */
  #define	HIFN_DMAIER_D_OVER	0x02000000	/* Destination Ring Overflow */
  #define	HIFN_DMAIER_R_ABORT	0x00200000	/* Result Ring PCI Abort */
  #define	HIFN_DMAIER_R_DONE	0x00100000	/* Result Ring Done */
  #define	HIFN_DMAIER_R_LAST	0x00080000	/* Result Ring Last */
  #define	HIFN_DMAIER_R_WAIT	0x00040000	/* Result Ring Waiting */
  #define	HIFN_DMAIER_R_OVER	0x00020000	/* Result Ring Overflow */
  #define	HIFN_DMAIER_S_ABORT	0x00002000	/* Source Ring PCI Abort */
  #define	HIFN_DMAIER_S_DONE	0x00001000	/* Source Ring Done */
  #define	HIFN_DMAIER_S_LAST	0x00000800	/* Source Ring Last */
  #define	HIFN_DMAIER_S_WAIT	0x00000400	/* Source Ring Waiting */
  #define	HIFN_DMAIER_ILLW	0x00000200	/* Illegal write (7811 only) */
  #define	HIFN_DMAIER_ILLR	0x00000100	/* Illegal read (7811 only) */
  #define	HIFN_DMAIER_C_ABORT	0x00000020	/* Command Ring PCI Abort */
  #define	HIFN_DMAIER_C_DONE	0x00000010	/* Command Ring Done */
  #define	HIFN_DMAIER_C_LAST	0x00000008	/* Command Ring Last */
  #define	HIFN_DMAIER_C_WAIT	0x00000004	/* Command Ring Waiting */
  #define	HIFN_DMAIER_PUBDONE	0x00000002	/* public op done (7951 only) */
  #define	HIFN_DMAIER_ENGINE	0x00000001	/* Engine IRQ */
  
  /* DMA Configuration Register (HIFN_1_DMA_CNFG) */
  #define	HIFN_DMACNFG_BIGENDIAN	0x10000000	/* big endian mode */
  #define	HIFN_DMACNFG_POLLFREQ	0x00ff0000	/* Poll frequency mask */
  #define	HIFN_DMACNFG_UNLOCK	0x00000800
  #define	HIFN_DMACNFG_POLLINVAL	0x00000700	/* Invalid Poll Scalar */
  #define	HIFN_DMACNFG_LAST	0x00000010	/* Host control LAST bit */
  #define	HIFN_DMACNFG_MODE	0x00000004	/* DMA mode */
  #define	HIFN_DMACNFG_DMARESET	0x00000002	/* DMA Reset # */
  #define	HIFN_DMACNFG_MSTRESET	0x00000001	/* Master Reset # */

  /* PLL configuration register */
  #define HIFN_PLL_REF_CLK_HBI	0x00000000	/* HBI reference clock */
  #define HIFN_PLL_REF_CLK_PLL	0x00000001	/* PLL reference clock */
  #define HIFN_PLL_BP		0x00000002	/* Reference clock bypass */
  #define HIFN_PLL_PK_CLK_HBI	0x00000000	/* PK engine HBI clock */
  #define HIFN_PLL_PK_CLK_PLL	0x00000008	/* PK engine PLL clock */
  #define HIFN_PLL_PE_CLK_HBI	0x00000000	/* PE engine HBI clock */
  #define HIFN_PLL_PE_CLK_PLL	0x00000010	/* PE engine PLL clock */
  #define HIFN_PLL_RESERVED_1	0x00000400	/* Reserved bit, must be 1 */
  #define HIFN_PLL_ND_SHIFT	11		/* Clock multiplier shift */
  #define HIFN_PLL_ND_MULT_2	0x00000000	/* PLL clock multiplier 2 */
  #define HIFN_PLL_ND_MULT_4	0x00000800	/* PLL clock multiplier 4 */
  #define HIFN_PLL_ND_MULT_6	0x00001000	/* PLL clock multiplier 6 */
  #define HIFN_PLL_ND_MULT_8	0x00001800	/* PLL clock multiplier 8 */
  #define HIFN_PLL_ND_MULT_10	0x00002000	/* PLL clock multiplier 10 */
  #define HIFN_PLL_ND_MULT_12	0x00002800	/* PLL clock multiplier 12 */
  #define HIFN_PLL_IS_1_8		0x00000000	/* charge pump (mult. 1-8) */
  #define HIFN_PLL_IS_9_12	0x00010000	/* charge pump (mult. 9-12) */
  
  #define HIFN_PLL_FCK_MAX	266		/* Maximum PLL frequency */

  
  /* Public key reset register (HIFN_1_PUB_RESET) */
  #define	HIFN_PUBRST_RESET	0x00000001	/* reset public/rng unit */
  
  /* Public base address register (HIFN_1_PUB_BASE) */
  #define	HIFN_PUBBASE_ADDR	0x00003fff	/* base address */
  
  /* Public operand length register (HIFN_1_PUB_OPLEN) */
  #define	HIFN_PUBOPLEN_MOD_M	0x0000007f	/* modulus length mask */
  #define	HIFN_PUBOPLEN_MOD_S	0		/* modulus length shift */
  #define	HIFN_PUBOPLEN_EXP_M	0x0003ff80	/* exponent length mask */

  #define	HIFN_PUBOPLEN_EXP_S	7		/* exponent length shift */

  #define	HIFN_PUBOPLEN_RED_M	0x003c0000	/* reducend length mask */
  #define	HIFN_PUBOPLEN_RED_S	18		/* reducend length shift */
  
  /* Public operation register (HIFN_1_PUB_OP) */
  #define	HIFN_PUBOP_AOFFSET_M	0x0000007f	/* A offset mask */
  #define	HIFN_PUBOP_AOFFSET_S	0		/* A offset shift */
  #define	HIFN_PUBOP_BOFFSET_M	0x00000f80	/* B offset mask */
  #define	HIFN_PUBOP_BOFFSET_S	7		/* B offset shift */
  #define	HIFN_PUBOP_MOFFSET_M	0x0003f000	/* M offset mask */
  #define	HIFN_PUBOP_MOFFSET_S	12		/* M offset shift */
  #define	HIFN_PUBOP_OP_MASK	0x003c0000	/* Opcode: */
  #define	HIFN_PUBOP_OP_NOP	0x00000000	/*  NOP */
  #define	HIFN_PUBOP_OP_ADD	0x00040000	/*  ADD */
  #define	HIFN_PUBOP_OP_ADDC	0x00080000	/*  ADD w/carry */
  #define	HIFN_PUBOP_OP_SUB	0x000c0000	/*  SUB */
  #define	HIFN_PUBOP_OP_SUBC	0x00100000	/*  SUB w/carry */
  #define	HIFN_PUBOP_OP_MODADD	0x00140000	/*  Modular ADD */
  #define	HIFN_PUBOP_OP_MODSUB	0x00180000	/*  Modular SUB */
  #define	HIFN_PUBOP_OP_INCA	0x001c0000	/*  INC A */
  #define	HIFN_PUBOP_OP_DECA	0x00200000	/*  DEC A */
  #define	HIFN_PUBOP_OP_MULT	0x00240000	/*  MULT */
  #define	HIFN_PUBOP_OP_MODMULT	0x00280000	/*  Modular MULT */
  #define	HIFN_PUBOP_OP_MODRED	0x002c0000	/*  Modular RED */
  #define	HIFN_PUBOP_OP_MODEXP	0x00300000	/*  Modular EXP */
  
  /* Public status register (HIFN_1_PUB_STATUS) */
  #define	HIFN_PUBSTS_DONE	0x00000001	/* operation done */
  #define	HIFN_PUBSTS_CARRY	0x00000002	/* carry */
  
  /* Public interrupt enable register (HIFN_1_PUB_IEN) */
  #define	HIFN_PUBIEN_DONE	0x00000001	/* operation done interrupt */
  
  /* Random number generator config register (HIFN_1_RNG_CONFIG) */
  #define	HIFN_RNGCFG_ENA		0x00000001	/* enable rng */
  
  #define HIFN_NAMESIZE			32
  #define HIFN_MAX_RESULT_ORDER		5

  #define	HIFN_D_CMD_RSIZE		(24 * 1)
  #define	HIFN_D_SRC_RSIZE		(80 * 1)
  #define	HIFN_D_DST_RSIZE		(80 * 1)
  #define	HIFN_D_RES_RSIZE		(24 * 1)

  #define HIFN_D_DST_DALIGN		4

  #define HIFN_QUEUE_LENGTH		(HIFN_D_CMD_RSIZE - 1)

  
  #define AES_MIN_KEY_SIZE		16
  #define AES_MAX_KEY_SIZE		32
  
  #define HIFN_DES_KEY_LENGTH		8
  #define HIFN_3DES_KEY_LENGTH		24
  #define HIFN_MAX_CRYPT_KEY_LENGTH	AES_MAX_KEY_SIZE
  #define HIFN_IV_LENGTH			8
  #define HIFN_AES_IV_LENGTH		16
  #define	HIFN_MAX_IV_LENGTH		HIFN_AES_IV_LENGTH
  
  #define HIFN_MAC_KEY_LENGTH		64
  #define HIFN_MD5_LENGTH			16
  #define HIFN_SHA1_LENGTH		20
  #define HIFN_MAC_TRUNC_LENGTH		12
  
  #define	HIFN_MAX_COMMAND		(8 + 8 + 8 + 64 + 260)
  #define	HIFN_MAX_RESULT			(8 + 4 + 4 + 20 + 4)
  #define HIFN_USED_RESULT		12

  struct hifn_desc {

  	volatile __le32		l;
  	volatile __le32		p;

  };
  
  struct hifn_dma {

  	struct hifn_desc	cmdr[HIFN_D_CMD_RSIZE + 1];
  	struct hifn_desc	srcr[HIFN_D_SRC_RSIZE + 1];
  	struct hifn_desc	dstr[HIFN_D_DST_RSIZE + 1];
  	struct hifn_desc	resr[HIFN_D_RES_RSIZE + 1];

  
  	u8			command_bufs[HIFN_D_CMD_RSIZE][HIFN_MAX_COMMAND];
  	u8			result_bufs[HIFN_D_CMD_RSIZE][HIFN_MAX_RESULT];

  	/*
  	 *  Our current positions for insertion and removal from the descriptor
  	 *  rings.
  	 */
  	volatile int		cmdi, srci, dsti, resi;
  	volatile int		cmdu, srcu, dstu, resu;
  	int			cmdk, srck, dstk, resk;
  };

  #define HIFN_FLAG_CMD_BUSY	(1 << 0)
  #define HIFN_FLAG_SRC_BUSY	(1 << 1)
  #define HIFN_FLAG_DST_BUSY	(1 << 2)
  #define HIFN_FLAG_RES_BUSY	(1 << 3)
  #define HIFN_FLAG_OLD_KEY	(1 << 4)

  
  #define HIFN_DEFAULT_ACTIVE_NUM	5

  struct hifn_device {

  	char			name[HIFN_NAMESIZE];
  
  	int			irq;
  
  	struct pci_dev		*pdev;
  	void __iomem		*bar[3];

  	void			*desc_virt;
  	dma_addr_t		desc_dma;
  
  	u32			dmareg;

  	void			*sa[HIFN_D_RES_RSIZE];

  
  	spinlock_t		lock;

  	u32			flags;
  	int			active, started;
  	struct delayed_work	work;
  	unsigned long		reset;
  	unsigned long		success;
  	unsigned long		prev_success;
  
  	u8			snum;

  	struct tasklet_struct	tasklet;

  	struct crypto_queue	queue;

  	struct list_head	alg_list;

  
  	unsigned int		pk_clk_freq;

  #ifdef CONFIG_CRYPTO_DEV_HIFN_795X_RNG

  	unsigned int		rng_wait_time;
  	ktime_t			rngtime;
  	struct hwrng		rng;
  #endif

  };
  
  #define	HIFN_D_LENGTH			0x0000ffff
  #define	HIFN_D_NOINVALID		0x01000000
  #define	HIFN_D_MASKDONEIRQ		0x02000000
  #define	HIFN_D_DESTOVER			0x04000000
  #define	HIFN_D_OVER			0x08000000
  #define	HIFN_D_LAST			0x20000000
  #define	HIFN_D_JUMP			0x40000000
  #define	HIFN_D_VALID			0x80000000

  struct hifn_base_command {

  	volatile __le16		masks;
  	volatile __le16		session_num;
  	volatile __le16		total_source_count;
  	volatile __le16		total_dest_count;

  };
  
  #define	HIFN_BASE_CMD_COMP		0x0100	/* enable compression engine */
  #define	HIFN_BASE_CMD_PAD		0x0200	/* enable padding engine */
  #define	HIFN_BASE_CMD_MAC		0x0400	/* enable MAC engine */
  #define	HIFN_BASE_CMD_CRYPT		0x0800	/* enable crypt engine */
  #define	HIFN_BASE_CMD_DECODE		0x2000
  #define	HIFN_BASE_CMD_SRCLEN_M		0xc000
  #define	HIFN_BASE_CMD_SRCLEN_S		14
  #define	HIFN_BASE_CMD_DSTLEN_M		0x3000
  #define	HIFN_BASE_CMD_DSTLEN_S		12
  #define	HIFN_BASE_CMD_LENMASK_HI	0x30000
  #define	HIFN_BASE_CMD_LENMASK_LO	0x0ffff
  
  /*
   * Structure to help build up the command data structure.
   */

  struct hifn_crypt_command {
  	volatile __le16		masks;
  	volatile __le16		header_skip;
  	volatile __le16		source_count;
  	volatile __le16		reserved;

  };
  
  #define	HIFN_CRYPT_CMD_ALG_MASK		0x0003		/* algorithm: */
  #define	HIFN_CRYPT_CMD_ALG_DES		0x0000		/*   DES */
  #define	HIFN_CRYPT_CMD_ALG_3DES		0x0001		/*   3DES */
  #define	HIFN_CRYPT_CMD_ALG_RC4		0x0002		/*   RC4 */
  #define	HIFN_CRYPT_CMD_ALG_AES		0x0003		/*   AES */
  #define	HIFN_CRYPT_CMD_MODE_MASK	0x0018		/* Encrypt mode: */
  #define	HIFN_CRYPT_CMD_MODE_ECB		0x0000		/*   ECB */
  #define	HIFN_CRYPT_CMD_MODE_CBC		0x0008		/*   CBC */
  #define	HIFN_CRYPT_CMD_MODE_CFB		0x0010		/*   CFB */
  #define	HIFN_CRYPT_CMD_MODE_OFB		0x0018		/*   OFB */
  #define	HIFN_CRYPT_CMD_CLR_CTX		0x0040		/* clear context */
  #define	HIFN_CRYPT_CMD_KSZ_MASK		0x0600		/* AES key size: */
  #define	HIFN_CRYPT_CMD_KSZ_128		0x0000		/*  128 bit */
  #define	HIFN_CRYPT_CMD_KSZ_192		0x0200		/*  192 bit */
  #define	HIFN_CRYPT_CMD_KSZ_256		0x0400		/*  256 bit */
  #define	HIFN_CRYPT_CMD_NEW_KEY		0x0800		/* expect new key */
  #define	HIFN_CRYPT_CMD_NEW_IV		0x1000		/* expect new iv */
  #define	HIFN_CRYPT_CMD_SRCLEN_M		0xc000
  #define	HIFN_CRYPT_CMD_SRCLEN_S		14
  
  /*
   * Structure to help build up the command data structure.
   */

  struct hifn_mac_command {
  	volatile __le16	masks;
  	volatile __le16	header_skip;
  	volatile __le16	source_count;
  	volatile __le16	reserved;

  };
  
  #define	HIFN_MAC_CMD_ALG_MASK		0x0001
  #define	HIFN_MAC_CMD_ALG_SHA1		0x0000
  #define	HIFN_MAC_CMD_ALG_MD5		0x0001
  #define	HIFN_MAC_CMD_MODE_MASK		0x000c
  #define	HIFN_MAC_CMD_MODE_HMAC		0x0000
  #define	HIFN_MAC_CMD_MODE_SSL_MAC	0x0004
  #define	HIFN_MAC_CMD_MODE_HASH		0x0008
  #define	HIFN_MAC_CMD_MODE_FULL		0x0004
  #define	HIFN_MAC_CMD_TRUNC		0x0010
  #define	HIFN_MAC_CMD_RESULT		0x0020
  #define	HIFN_MAC_CMD_APPEND		0x0040
  #define	HIFN_MAC_CMD_SRCLEN_M		0xc000
  #define	HIFN_MAC_CMD_SRCLEN_S		14
  
  /*
   * MAC POS IPsec initiates authentication after encryption on encodes
   * and before decryption on decodes.
   */
  #define	HIFN_MAC_CMD_POS_IPSEC		0x0200
  #define	HIFN_MAC_CMD_NEW_KEY		0x0800

  struct hifn_comp_command {
  	volatile __le16		masks;
  	volatile __le16		header_skip;
  	volatile __le16		source_count;
  	volatile __le16		reserved;

  };
  
  #define	HIFN_COMP_CMD_SRCLEN_M		0xc000
  #define	HIFN_COMP_CMD_SRCLEN_S		14
  #define	HIFN_COMP_CMD_ONE		0x0100	/* must be one */
  #define	HIFN_COMP_CMD_CLEARHIST		0x0010	/* clear history */
  #define	HIFN_COMP_CMD_UPDATEHIST	0x0008	/* update history */
  #define	HIFN_COMP_CMD_LZS_STRIP0	0x0004	/* LZS: strip zero */
  #define	HIFN_COMP_CMD_MPPC_RESTART	0x0004	/* MPPC: restart */
  #define	HIFN_COMP_CMD_ALG_MASK		0x0001	/* compression mode: */
  #define	HIFN_COMP_CMD_ALG_MPPC		0x0001	/*   MPPC */
  #define	HIFN_COMP_CMD_ALG_LZS		0x0000	/*   LZS */

  struct hifn_base_result {
  	volatile __le16		flags;
  	volatile __le16		session;
  	volatile __le16		src_cnt;		/* 15:0 of source count */
  	volatile __le16		dst_cnt;		/* 15:0 of dest count */

  };
  
  #define	HIFN_BASE_RES_DSTOVERRUN	0x0200	/* destination overrun */
  #define	HIFN_BASE_RES_SRCLEN_M		0xc000	/* 17:16 of source count */
  #define	HIFN_BASE_RES_SRCLEN_S		14
  #define	HIFN_BASE_RES_DSTLEN_M		0x3000	/* 17:16 of dest count */
  #define	HIFN_BASE_RES_DSTLEN_S		12

  struct hifn_comp_result {

  	volatile __le16		flags;
  	volatile __le16		crc;

  };
  
  #define	HIFN_COMP_RES_LCB_M		0xff00	/* longitudinal check byte */
  #define	HIFN_COMP_RES_LCB_S		8
  #define	HIFN_COMP_RES_RESTART		0x0004	/* MPPC: restart */
  #define	HIFN_COMP_RES_ENDMARKER		0x0002	/* LZS: end marker seen */
  #define	HIFN_COMP_RES_SRC_NOTZERO	0x0001	/* source expired */

  struct hifn_mac_result {
  	volatile __le16		flags;
  	volatile __le16		reserved;

  	/* followed by 0, 6, 8, or 10 u16's of the MAC, then crypt */
  };
  
  #define	HIFN_MAC_RES_MISCOMPARE		0x0002	/* compare failed */
  #define	HIFN_MAC_RES_SRC_NOTZERO	0x0001	/* source expired */

  struct hifn_crypt_result {

  	volatile __le16		flags;
  	volatile __le16		reserved;

  };
  
  #define	HIFN_CRYPT_RES_SRC_NOTZERO	0x0001	/* source expired */
  
  #ifndef HIFN_POLL_FREQUENCY
  #define	HIFN_POLL_FREQUENCY	0x1
  #endif
  
  #ifndef HIFN_POLL_SCALAR
  #define	HIFN_POLL_SCALAR	0x0
  #endif

  #define	HIFN_MAX_SEGLEN		0xffff		/* maximum dma segment len */

  #define	HIFN_MAX_DMALEN		0x3ffff		/* maximum dma length */

  struct hifn_crypto_alg {

  	struct list_head	entry;
  	struct crypto_alg	alg;
  	struct hifn_device	*dev;
  };
  
  #define ASYNC_SCATTERLIST_CACHE	16

  #define ASYNC_FLAGS_MISALIGNED	(1 << 0)

  struct hifn_cipher_walk {

  	struct scatterlist	cache[ASYNC_SCATTERLIST_CACHE];
  	u32			flags;
  	int			num;
  };

  struct hifn_context {

  	u8			key[HIFN_MAX_CRYPT_KEY_LENGTH];

  	struct hifn_device	*dev;

  	unsigned int		keysize;
  };

  struct hifn_request_context {

  	u8			*iv;
  	unsigned int		ivsize;

  	u8			op, type, mode, unused;

  	struct hifn_cipher_walk	walk;

  };

  #define crypto_alg_to_hifn(a)	container_of(a, struct hifn_crypto_alg, alg)

  
  static inline u32 hifn_read_0(struct hifn_device *dev, u32 reg)
  {

  	return readl(dev->bar[0] + reg);

  }
  
  static inline u32 hifn_read_1(struct hifn_device *dev, u32 reg)
  {

  	return readl(dev->bar[1] + reg);

  }
  
  static inline void hifn_write_0(struct hifn_device *dev, u32 reg, u32 val)
  {

  	writel((__force u32)cpu_to_le32(val), dev->bar[0] + reg);

  }
  
  static inline void hifn_write_1(struct hifn_device *dev, u32 reg, u32 val)
  {

  	writel((__force u32)cpu_to_le32(val), dev->bar[1] + reg);

  }
  
  static void hifn_wait_puc(struct hifn_device *dev)
  {
  	int i;
  	u32 ret;

  	for (i = 10000; i > 0; --i) {

  		ret = hifn_read_0(dev, HIFN_0_PUCTRL);
  		if (!(ret & HIFN_PUCTRL_RESET))
  			break;
  
  		udelay(1);
  	}
  
  	if (!i)

  		dev_err(&dev->pdev->dev, "Failed to reset PUC unit.
  ");

  }
  
  static void hifn_reset_puc(struct hifn_device *dev)
  {
  	hifn_write_0(dev, HIFN_0_PUCTRL, HIFN_PUCTRL_DMAENA);
  	hifn_wait_puc(dev);
  }
  
  static void hifn_stop_device(struct hifn_device *dev)
  {
  	hifn_write_1(dev, HIFN_1_DMA_CSR,
  		HIFN_DMACSR_D_CTRL_DIS | HIFN_DMACSR_R_CTRL_DIS |
  		HIFN_DMACSR_S_CTRL_DIS | HIFN_DMACSR_C_CTRL_DIS);
  	hifn_write_0(dev, HIFN_0_PUIER, 0);
  	hifn_write_1(dev, HIFN_1_DMA_IER, 0);
  }
  
  static void hifn_reset_dma(struct hifn_device *dev, int full)
  {
  	hifn_stop_device(dev);
  
  	/*
  	 * Setting poll frequency and others to 0.
  	 */
  	hifn_write_1(dev, HIFN_1_DMA_CNFG, HIFN_DMACNFG_MSTRESET |
  			HIFN_DMACNFG_DMARESET | HIFN_DMACNFG_MODE);
  	mdelay(1);
  
  	/*
  	 * Reset DMA.
  	 */
  	if (full) {
  		hifn_write_1(dev, HIFN_1_DMA_CNFG, HIFN_DMACNFG_MODE);
  		mdelay(1);
  	} else {
  		hifn_write_1(dev, HIFN_1_DMA_CNFG, HIFN_DMACNFG_MODE |
  				HIFN_DMACNFG_MSTRESET);
  		hifn_reset_puc(dev);
  	}
  
  	hifn_write_1(dev, HIFN_1_DMA_CNFG, HIFN_DMACNFG_MSTRESET |
  			HIFN_DMACNFG_DMARESET | HIFN_DMACNFG_MODE);
  
  	hifn_reset_puc(dev);
  }

  static u32 hifn_next_signature(u32 a, u_int cnt)

  {
  	int i;
  	u32 v;
  
  	for (i = 0; i < cnt; i++) {

  		/* get the parity */
  		v = a & 0x80080125;
  		v ^= v >> 16;
  		v ^= v >> 8;
  		v ^= v >> 4;
  		v ^= v >> 2;
  		v ^= v >> 1;
  
  		a = (v & 1) ^ (a << 1);
  	}
  
  	return a;
  }
  
  static struct pci2id {
  	u_short		pci_vendor;
  	u_short		pci_prod;
  	char		card_id[13];
  } pci2id[] = {
  	{
  		PCI_VENDOR_ID_HIFN,
  		PCI_DEVICE_ID_HIFN_7955,
  		{ 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  		  0x00, 0x00, 0x00, 0x00, 0x00 }
  	},
  	{
  		PCI_VENDOR_ID_HIFN,
  		PCI_DEVICE_ID_HIFN_7956,
  		{ 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  		  0x00, 0x00, 0x00, 0x00, 0x00 }
  	}
  };

  #ifdef CONFIG_CRYPTO_DEV_HIFN_795X_RNG

  static int hifn_rng_data_present(struct hwrng *rng, int wait)
  {
  	struct hifn_device *dev = (struct hifn_device *)rng->priv;
  	s64 nsec;
  
  	nsec = ktime_to_ns(ktime_sub(ktime_get(), dev->rngtime));
  	nsec -= dev->rng_wait_time;
  	if (nsec <= 0)
  		return 1;
  	if (!wait)
  		return 0;
  	ndelay(nsec);
  	return 1;
  }
  
  static int hifn_rng_data_read(struct hwrng *rng, u32 *data)
  {
  	struct hifn_device *dev = (struct hifn_device *)rng->priv;
  
  	*data = hifn_read_1(dev, HIFN_1_RNG_DATA);
  	dev->rngtime = ktime_get();
  	return 4;
  }
  
  static int hifn_register_rng(struct hifn_device *dev)
  {
  	/*
  	 * We must wait at least 256 Pk_clk cycles between two reads of the rng.
  	 */

  	dev->rng_wait_time	= DIV_ROUND_UP_ULL(NSEC_PER_SEC,
  						   dev->pk_clk_freq) * 256;

  
  	dev->rng.name		= dev->name;
  	dev->rng.data_present	= hifn_rng_data_present,
  	dev->rng.data_read	= hifn_rng_data_read,
  	dev->rng.priv		= (unsigned long)dev;
  
  	return hwrng_register(&dev->rng);
  }
  
  static void hifn_unregister_rng(struct hifn_device *dev)
  {
  	hwrng_unregister(&dev->rng);
  }
  #else
  #define hifn_register_rng(dev)		0
  #define hifn_unregister_rng(dev)
  #endif

  static int hifn_init_pubrng(struct hifn_device *dev)
  {
  	int i;
  
  	hifn_write_1(dev, HIFN_1_PUB_RESET, hifn_read_1(dev, HIFN_1_PUB_RESET) |
  			HIFN_PUBRST_RESET);

  	for (i = 100; i > 0; --i) {

  		mdelay(1);
  
  		if ((hifn_read_1(dev, HIFN_1_PUB_RESET) & HIFN_PUBRST_RESET) == 0)
  			break;
  	}

  	if (!i) {

  		dev_err(&dev->pdev->dev, "Failed to initialise public key engine.
  ");

  	} else {

  		hifn_write_1(dev, HIFN_1_PUB_IEN, HIFN_PUBIEN_DONE);
  		dev->dmareg |= HIFN_DMAIER_PUBDONE;
  		hifn_write_1(dev, HIFN_1_DMA_IER, dev->dmareg);

  		dev_dbg(&dev->pdev->dev, "Public key engine has been successfully initialised.
  ");

  	}

  	/* Enable RNG engine. */

  
  	hifn_write_1(dev, HIFN_1_RNG_CONFIG,
  			hifn_read_1(dev, HIFN_1_RNG_CONFIG) | HIFN_RNGCFG_ENA);

  	dev_dbg(&dev->pdev->dev, "RNG engine has been successfully initialised.
  ");

  #ifdef CONFIG_CRYPTO_DEV_HIFN_795X_RNG

  	/* First value must be discarded */
  	hifn_read_1(dev, HIFN_1_RNG_DATA);
  	dev->rngtime = ktime_get();
  #endif

  	return 0;
  }
  
  static int hifn_enable_crypto(struct hifn_device *dev)
  {
  	u32 dmacfg, addr;
  	char *offtbl = NULL;
  	int i;

  	for (i = 0; i < ARRAY_SIZE(pci2id); i++) {

  		if (pci2id[i].pci_vendor == dev->pdev->vendor &&
  				pci2id[i].pci_prod == dev->pdev->device) {
  			offtbl = pci2id[i].card_id;
  			break;
  		}
  	}

  	if (!offtbl) {

  		dev_err(&dev->pdev->dev, "Unknown card!
  ");

  		return -ENODEV;
  	}
  
  	dmacfg = hifn_read_1(dev, HIFN_1_DMA_CNFG);
  
  	hifn_write_1(dev, HIFN_1_DMA_CNFG,
  			HIFN_DMACNFG_UNLOCK | HIFN_DMACNFG_MSTRESET |
  			HIFN_DMACNFG_DMARESET | HIFN_DMACNFG_MODE);
  	mdelay(1);
  	addr = hifn_read_1(dev, HIFN_1_UNLOCK_SECRET1);
  	mdelay(1);
  	hifn_write_1(dev, HIFN_1_UNLOCK_SECRET2, 0);
  	mdelay(1);

  	for (i = 0; i < 12; ++i) {

  		addr = hifn_next_signature(addr, offtbl[i] + 0x101);
  		hifn_write_1(dev, HIFN_1_UNLOCK_SECRET2, addr);
  
  		mdelay(1);
  	}
  	hifn_write_1(dev, HIFN_1_DMA_CNFG, dmacfg);

  	dev_dbg(&dev->pdev->dev, "%s %s.
  ", dev->name, pci_name(dev->pdev));

  
  	return 0;
  }
  
  static void hifn_init_dma(struct hifn_device *dev)
  {
  	struct hifn_dma *dma = (struct hifn_dma *)dev->desc_virt;
  	u32 dptr = dev->desc_dma;
  	int i;

  	for (i = 0; i < HIFN_D_CMD_RSIZE; ++i)

  		dma->cmdr[i].p = __cpu_to_le32(dptr +
  				offsetof(struct hifn_dma, command_bufs[i][0]));

  	for (i = 0; i < HIFN_D_RES_RSIZE; ++i)

  		dma->resr[i].p = __cpu_to_le32(dptr +
  				offsetof(struct hifn_dma, result_bufs[i][0]));

  	/* Setup LAST descriptors. */

  	dma->cmdr[HIFN_D_CMD_RSIZE].p = __cpu_to_le32(dptr +
  			offsetof(struct hifn_dma, cmdr[0]));
  	dma->srcr[HIFN_D_SRC_RSIZE].p = __cpu_to_le32(dptr +
  			offsetof(struct hifn_dma, srcr[0]));
  	dma->dstr[HIFN_D_DST_RSIZE].p = __cpu_to_le32(dptr +
  			offsetof(struct hifn_dma, dstr[0]));
  	dma->resr[HIFN_D_RES_RSIZE].p = __cpu_to_le32(dptr +
  			offsetof(struct hifn_dma, resr[0]));
  
  	dma->cmdu = dma->srcu = dma->dstu = dma->resu = 0;
  	dma->cmdi = dma->srci = dma->dsti = dma->resi = 0;
  	dma->cmdk = dma->srck = dma->dstk = dma->resk = 0;
  }

  /*
   * Initialize the PLL. We need to know the frequency of the reference clock
   * to calculate the optimal multiplier. For PCI we assume 66MHz, since that
   * allows us to operate without the risk of overclocking the chip. If it
   * actually uses 33MHz, the chip will operate at half the speed, this can be

   * overridden by specifying the frequency as module parameter (pci33).

   *
   * Unfortunately the PCI clock is not very suitable since the HIFN needs a
   * stable clock and the PCI clock frequency may vary, so the default is the
   * external clock. There is no way to find out its frequency, we default to
   * 66MHz since according to Mike Ham of HiFn, almost every board in existence
   * has an external crystal populated at 66MHz.
   */
  static void hifn_init_pll(struct hifn_device *dev)
  {
  	unsigned int freq, m;
  	u32 pllcfg;
  
  	pllcfg = HIFN_1_PLL | HIFN_PLL_RESERVED_1;
  
  	if (strncmp(hifn_pll_ref, "ext", 3) == 0)
  		pllcfg |= HIFN_PLL_REF_CLK_PLL;
  	else
  		pllcfg |= HIFN_PLL_REF_CLK_HBI;
  
  	if (hifn_pll_ref[3] != '\0')
  		freq = simple_strtoul(hifn_pll_ref + 3, NULL, 10);
  	else {
  		freq = 66;

  		dev_info(&dev->pdev->dev, "assuming %uMHz clock speed, override with hifn_pll_ref=%.3s<frequency>
  ",
  			 freq, hifn_pll_ref);

  	}
  
  	m = HIFN_PLL_FCK_MAX / freq;
  
  	pllcfg |= (m / 2 - 1) << HIFN_PLL_ND_SHIFT;
  	if (m <= 8)
  		pllcfg |= HIFN_PLL_IS_1_8;
  	else
  		pllcfg |= HIFN_PLL_IS_9_12;
  
  	/* Select clock source and enable clock bypass */
  	hifn_write_1(dev, HIFN_1_PLL, pllcfg |
  		     HIFN_PLL_PK_CLK_HBI | HIFN_PLL_PE_CLK_HBI | HIFN_PLL_BP);
  
  	/* Let the chip lock to the input clock */
  	mdelay(10);
  
  	/* Disable clock bypass */
  	hifn_write_1(dev, HIFN_1_PLL, pllcfg |
  		     HIFN_PLL_PK_CLK_HBI | HIFN_PLL_PE_CLK_HBI);
  
  	/* Switch the engines to the PLL */
  	hifn_write_1(dev, HIFN_1_PLL, pllcfg |
  		     HIFN_PLL_PK_CLK_PLL | HIFN_PLL_PE_CLK_PLL);

  
  	/*
  	 * The Fpk_clk runs at half the total speed. Its frequency is needed to
  	 * calculate the minimum time between two reads of the rng. Since 33MHz
  	 * is actually 33.333... we overestimate the frequency here, resulting
  	 * in slightly larger intervals.
  	 */
  	dev->pk_clk_freq = 1000000 * (freq + 1) * m / 2;

  }

  static void hifn_init_registers(struct hifn_device *dev)
  {
  	u32 dptr = dev->desc_dma;
  
  	/* Initialization magic... */
  	hifn_write_0(dev, HIFN_0_PUCTRL, HIFN_PUCTRL_DMAENA);
  	hifn_write_0(dev, HIFN_0_FIFOCNFG, HIFN_FIFOCNFG_THRESHOLD);
  	hifn_write_0(dev, HIFN_0_PUIER, HIFN_PUIER_DSTOVER);
  
  	/* write all 4 ring address registers */

  	hifn_write_1(dev, HIFN_1_DMA_CRAR, dptr +
  				offsetof(struct hifn_dma, cmdr[0]));
  	hifn_write_1(dev, HIFN_1_DMA_SRAR, dptr +
  				offsetof(struct hifn_dma, srcr[0]));
  	hifn_write_1(dev, HIFN_1_DMA_DRAR, dptr +
  				offsetof(struct hifn_dma, dstr[0]));
  	hifn_write_1(dev, HIFN_1_DMA_RRAR, dptr +
  				offsetof(struct hifn_dma, resr[0]));

  
  	mdelay(2);
  #if 0
  	hifn_write_1(dev, HIFN_1_DMA_CSR,
  	    HIFN_DMACSR_D_CTRL_DIS | HIFN_DMACSR_R_CTRL_DIS |
  	    HIFN_DMACSR_S_CTRL_DIS | HIFN_DMACSR_C_CTRL_DIS |
  	    HIFN_DMACSR_D_ABORT | HIFN_DMACSR_D_DONE | HIFN_DMACSR_D_LAST |
  	    HIFN_DMACSR_D_WAIT | HIFN_DMACSR_D_OVER |
  	    HIFN_DMACSR_R_ABORT | HIFN_DMACSR_R_DONE | HIFN_DMACSR_R_LAST |
  	    HIFN_DMACSR_R_WAIT | HIFN_DMACSR_R_OVER |
  	    HIFN_DMACSR_S_ABORT | HIFN_DMACSR_S_DONE | HIFN_DMACSR_S_LAST |
  	    HIFN_DMACSR_S_WAIT |
  	    HIFN_DMACSR_C_ABORT | HIFN_DMACSR_C_DONE | HIFN_DMACSR_C_LAST |
  	    HIFN_DMACSR_C_WAIT |
  	    HIFN_DMACSR_ENGINE |
  	    HIFN_DMACSR_PUBDONE);
  #else
  	hifn_write_1(dev, HIFN_1_DMA_CSR,
  	    HIFN_DMACSR_C_CTRL_ENA | HIFN_DMACSR_S_CTRL_ENA |
  	    HIFN_DMACSR_D_CTRL_ENA | HIFN_DMACSR_R_CTRL_ENA |
  	    HIFN_DMACSR_D_ABORT | HIFN_DMACSR_D_DONE | HIFN_DMACSR_D_LAST |
  	    HIFN_DMACSR_D_WAIT | HIFN_DMACSR_D_OVER |
  	    HIFN_DMACSR_R_ABORT | HIFN_DMACSR_R_DONE | HIFN_DMACSR_R_LAST |
  	    HIFN_DMACSR_R_WAIT | HIFN_DMACSR_R_OVER |
  	    HIFN_DMACSR_S_ABORT | HIFN_DMACSR_S_DONE | HIFN_DMACSR_S_LAST |
  	    HIFN_DMACSR_S_WAIT |
  	    HIFN_DMACSR_C_ABORT | HIFN_DMACSR_C_DONE | HIFN_DMACSR_C_LAST |
  	    HIFN_DMACSR_C_WAIT |
  	    HIFN_DMACSR_ENGINE |
  	    HIFN_DMACSR_PUBDONE);
  #endif
  	hifn_read_1(dev, HIFN_1_DMA_CSR);
  
  	dev->dmareg |= HIFN_DMAIER_R_DONE | HIFN_DMAIER_C_ABORT |
  	    HIFN_DMAIER_D_OVER | HIFN_DMAIER_R_OVER |
  	    HIFN_DMAIER_S_ABORT | HIFN_DMAIER_D_ABORT | HIFN_DMAIER_R_ABORT |
  	    HIFN_DMAIER_ENGINE;
  	dev->dmareg &= ~HIFN_DMAIER_C_WAIT;
  
  	hifn_write_1(dev, HIFN_1_DMA_IER, dev->dmareg);
  	hifn_read_1(dev, HIFN_1_DMA_IER);
  #if 0
  	hifn_write_0(dev, HIFN_0_PUCNFG, HIFN_PUCNFG_ENCCNFG |
  		    HIFN_PUCNFG_DRFR_128 | HIFN_PUCNFG_TCALLPHASES |
  		    HIFN_PUCNFG_TCDRVTOTEM | HIFN_PUCNFG_BUS32 |
  		    HIFN_PUCNFG_DRAM);
  #else
  	hifn_write_0(dev, HIFN_0_PUCNFG, 0x10342);
  #endif

  	hifn_init_pll(dev);

  
  	hifn_write_0(dev, HIFN_0_PUISR, HIFN_PUISR_DSTOVER);
  	hifn_write_1(dev, HIFN_1_DMA_CNFG, HIFN_DMACNFG_MSTRESET |
  	    HIFN_DMACNFG_DMARESET | HIFN_DMACNFG_MODE | HIFN_DMACNFG_LAST |
  	    ((HIFN_POLL_FREQUENCY << 16 ) & HIFN_DMACNFG_POLLFREQ) |
  	    ((HIFN_POLL_SCALAR << 8) & HIFN_DMACNFG_POLLINVAL));
  }
  
  static int hifn_setup_base_command(struct hifn_device *dev, u8 *buf,
  		unsigned dlen, unsigned slen, u16 mask, u8 snum)
  {
  	struct hifn_base_command *base_cmd;
  	u8 *buf_pos = buf;
  
  	base_cmd = (struct hifn_base_command *)buf_pos;
  	base_cmd->masks = __cpu_to_le16(mask);
  	base_cmd->total_source_count =
  		__cpu_to_le16(slen & HIFN_BASE_CMD_LENMASK_LO);
  	base_cmd->total_dest_count =
  		__cpu_to_le16(dlen & HIFN_BASE_CMD_LENMASK_LO);
  
  	dlen >>= 16;
  	slen >>= 16;
  	base_cmd->session_num = __cpu_to_le16(snum |
  	    ((slen << HIFN_BASE_CMD_SRCLEN_S) & HIFN_BASE_CMD_SRCLEN_M) |
  	    ((dlen << HIFN_BASE_CMD_DSTLEN_S) & HIFN_BASE_CMD_DSTLEN_M));
  
  	return sizeof(struct hifn_base_command);
  }
  
  static int hifn_setup_crypto_command(struct hifn_device *dev,
  		u8 *buf, unsigned dlen, unsigned slen,
  		u8 *key, int keylen, u8 *iv, int ivsize, u16 mode)
  {
  	struct hifn_dma *dma = (struct hifn_dma *)dev->desc_virt;
  	struct hifn_crypt_command *cry_cmd;
  	u8 *buf_pos = buf;
  	u16 cmd_len;
  
  	cry_cmd = (struct hifn_crypt_command *)buf_pos;
  
  	cry_cmd->source_count = __cpu_to_le16(dlen & 0xffff);
  	dlen >>= 16;
  	cry_cmd->masks = __cpu_to_le16(mode |
  			((dlen << HIFN_CRYPT_CMD_SRCLEN_S) &
  			 HIFN_CRYPT_CMD_SRCLEN_M));
  	cry_cmd->header_skip = 0;
  	cry_cmd->reserved = 0;
  
  	buf_pos += sizeof(struct hifn_crypt_command);
  
  	dma->cmdu++;
  	if (dma->cmdu > 1) {
  		dev->dmareg |= HIFN_DMAIER_C_WAIT;
  		hifn_write_1(dev, HIFN_1_DMA_IER, dev->dmareg);
  	}
  
  	if (keylen) {
  		memcpy(buf_pos, key, keylen);
  		buf_pos += keylen;
  	}
  	if (ivsize) {
  		memcpy(buf_pos, iv, ivsize);
  		buf_pos += ivsize;
  	}
  
  	cmd_len = buf_pos - buf;
  
  	return cmd_len;
  }

  static int hifn_setup_cmd_desc(struct hifn_device *dev,

  		struct hifn_context *ctx, struct hifn_request_context *rctx,
  		void *priv, unsigned int nbytes)

  {
  	struct hifn_dma *dma = (struct hifn_dma *)dev->desc_virt;
  	int cmd_len, sa_idx;
  	u8 *buf, *buf_pos;
  	u16 mask;

  	sa_idx = dma->cmdi;

  	buf_pos = buf = dma->command_bufs[dma->cmdi];
  
  	mask = 0;

  	switch (rctx->op) {

  	case ACRYPTO_OP_DECRYPT:
  		mask = HIFN_BASE_CMD_CRYPT | HIFN_BASE_CMD_DECODE;
  		break;
  	case ACRYPTO_OP_ENCRYPT:
  		mask = HIFN_BASE_CMD_CRYPT;
  		break;
  	case ACRYPTO_OP_HMAC:
  		mask = HIFN_BASE_CMD_MAC;
  		break;
  	default:
  		goto err_out;

  	}
  
  	buf_pos += hifn_setup_base_command(dev, buf_pos, nbytes,
  			nbytes, mask, dev->snum);

  	if (rctx->op == ACRYPTO_OP_ENCRYPT || rctx->op == ACRYPTO_OP_DECRYPT) {

  		u16 md = 0;
  
  		if (ctx->keysize)
  			md |= HIFN_CRYPT_CMD_NEW_KEY;

  		if (rctx->iv && rctx->mode != ACRYPTO_MODE_ECB)

  			md |= HIFN_CRYPT_CMD_NEW_IV;

  		switch (rctx->mode) {

  		case ACRYPTO_MODE_ECB:
  			md |= HIFN_CRYPT_CMD_MODE_ECB;
  			break;
  		case ACRYPTO_MODE_CBC:
  			md |= HIFN_CRYPT_CMD_MODE_CBC;
  			break;
  		case ACRYPTO_MODE_CFB:
  			md |= HIFN_CRYPT_CMD_MODE_CFB;
  			break;
  		case ACRYPTO_MODE_OFB:
  			md |= HIFN_CRYPT_CMD_MODE_OFB;
  			break;
  		default:
  			goto err_out;

  		}

  		switch (rctx->type) {

  		case ACRYPTO_TYPE_AES_128:
  			if (ctx->keysize != 16)

  				goto err_out;

  			md |= HIFN_CRYPT_CMD_KSZ_128 |
  				HIFN_CRYPT_CMD_ALG_AES;
  			break;
  		case ACRYPTO_TYPE_AES_192:
  			if (ctx->keysize != 24)
  				goto err_out;
  			md |= HIFN_CRYPT_CMD_KSZ_192 |
  				HIFN_CRYPT_CMD_ALG_AES;
  			break;
  		case ACRYPTO_TYPE_AES_256:
  			if (ctx->keysize != 32)
  				goto err_out;
  			md |= HIFN_CRYPT_CMD_KSZ_256 |
  				HIFN_CRYPT_CMD_ALG_AES;
  			break;
  		case ACRYPTO_TYPE_3DES:
  			if (ctx->keysize != 24)
  				goto err_out;
  			md |= HIFN_CRYPT_CMD_ALG_3DES;
  			break;
  		case ACRYPTO_TYPE_DES:
  			if (ctx->keysize != 8)
  				goto err_out;
  			md |= HIFN_CRYPT_CMD_ALG_DES;
  			break;
  		default:
  			goto err_out;

  		}
  
  		buf_pos += hifn_setup_crypto_command(dev, buf_pos,
  				nbytes, nbytes, ctx->key, ctx->keysize,

  				rctx->iv, rctx->ivsize, md);

  	}
  
  	dev->sa[sa_idx] = priv;

  	dev->started++;

  
  	cmd_len = buf_pos - buf;
  	dma->cmdr[dma->cmdi].l = __cpu_to_le32(cmd_len | HIFN_D_VALID |
  			HIFN_D_LAST | HIFN_D_MASKDONEIRQ);
  
  	if (++dma->cmdi == HIFN_D_CMD_RSIZE) {

  		dma->cmdr[dma->cmdi].l = __cpu_to_le32(

  			HIFN_D_VALID | HIFN_D_LAST |
  			HIFN_D_MASKDONEIRQ | HIFN_D_JUMP);
  		dma->cmdi = 0;

  	} else {
  		dma->cmdr[dma->cmdi - 1].l |= __cpu_to_le32(HIFN_D_VALID);
  	}

  
  	if (!(dev->flags & HIFN_FLAG_CMD_BUSY)) {
  		hifn_write_1(dev, HIFN_1_DMA_CSR, HIFN_DMACSR_C_CTRL_ENA);
  		dev->flags |= HIFN_FLAG_CMD_BUSY;
  	}

  	return 0;
  
  err_out:
  	return -EINVAL;
  }

  static int hifn_setup_src_desc(struct hifn_device *dev, struct page *page,

  		unsigned int offset, unsigned int size, int last)

  {
  	struct hifn_dma *dma = (struct hifn_dma *)dev->desc_virt;
  	int idx;
  	dma_addr_t addr;
  
  	addr = pci_map_page(dev->pdev, page, offset, size, PCI_DMA_TODEVICE);
  
  	idx = dma->srci;
  
  	dma->srcr[idx].p = __cpu_to_le32(addr);
  	dma->srcr[idx].l = __cpu_to_le32(size | HIFN_D_VALID |

  			HIFN_D_MASKDONEIRQ | (last ? HIFN_D_LAST : 0));

  
  	if (++idx == HIFN_D_SRC_RSIZE) {
  		dma->srcr[idx].l = __cpu_to_le32(HIFN_D_VALID |

  				HIFN_D_JUMP | HIFN_D_MASKDONEIRQ |
  				(last ? HIFN_D_LAST : 0));

  		idx = 0;
  	}
  
  	dma->srci = idx;
  	dma->srcu++;
  
  	if (!(dev->flags & HIFN_FLAG_SRC_BUSY)) {
  		hifn_write_1(dev, HIFN_1_DMA_CSR, HIFN_DMACSR_S_CTRL_ENA);
  		dev->flags |= HIFN_FLAG_SRC_BUSY;
  	}
  
  	return size;
  }
  
  static void hifn_setup_res_desc(struct hifn_device *dev)
  {
  	struct hifn_dma *dma = (struct hifn_dma *)dev->desc_virt;
  
  	dma->resr[dma->resi].l = __cpu_to_le32(HIFN_USED_RESULT |
  			HIFN_D_VALID | HIFN_D_LAST);
  	/*
  	 * dma->resr[dma->resi].l = __cpu_to_le32(HIFN_MAX_RESULT | HIFN_D_VALID |

  	 *					HIFN_D_LAST);

  	 */
  
  	if (++dma->resi == HIFN_D_RES_RSIZE) {
  		dma->resr[HIFN_D_RES_RSIZE].l = __cpu_to_le32(HIFN_D_VALID |
  				HIFN_D_JUMP | HIFN_D_MASKDONEIRQ | HIFN_D_LAST);
  		dma->resi = 0;
  	}
  
  	dma->resu++;
  
  	if (!(dev->flags & HIFN_FLAG_RES_BUSY)) {
  		hifn_write_1(dev, HIFN_1_DMA_CSR, HIFN_DMACSR_R_CTRL_ENA);
  		dev->flags |= HIFN_FLAG_RES_BUSY;
  	}
  }
  
  static void hifn_setup_dst_desc(struct hifn_device *dev, struct page *page,

  		unsigned offset, unsigned size, int last)

  {
  	struct hifn_dma *dma = (struct hifn_dma *)dev->desc_virt;
  	int idx;
  	dma_addr_t addr;
  
  	addr = pci_map_page(dev->pdev, page, offset, size, PCI_DMA_FROMDEVICE);
  
  	idx = dma->dsti;
  	dma->dstr[idx].p = __cpu_to_le32(addr);
  	dma->dstr[idx].l = __cpu_to_le32(size |	HIFN_D_VALID |

  			HIFN_D_MASKDONEIRQ | (last ? HIFN_D_LAST : 0));

  
  	if (++idx == HIFN_D_DST_RSIZE) {
  		dma->dstr[idx].l = __cpu_to_le32(HIFN_D_VALID |
  				HIFN_D_JUMP | HIFN_D_MASKDONEIRQ |

  				(last ? HIFN_D_LAST : 0));

  		idx = 0;
  	}
  	dma->dsti = idx;
  	dma->dstu++;
  
  	if (!(dev->flags & HIFN_FLAG_DST_BUSY)) {
  		hifn_write_1(dev, HIFN_1_DMA_CSR, HIFN_DMACSR_D_CTRL_ENA);
  		dev->flags |= HIFN_FLAG_DST_BUSY;
  	}
  }

  static int hifn_setup_dma(struct hifn_device *dev,
  		struct hifn_context *ctx, struct hifn_request_context *rctx,

  		struct scatterlist *src, struct scatterlist *dst,
  		unsigned int nbytes, void *priv)

  {

  	struct scatterlist *t;
  	struct page *spage, *dpage;
  	unsigned int soff, doff;
  	unsigned int n, len;

  	n = nbytes;
  	while (n) {
  		spage = sg_page(src);
  		soff = src->offset;
  		len = min(src->length, n);

  		hifn_setup_src_desc(dev, spage, soff, len, n - len == 0);
  
  		src++;
  		n -= len;
  	}

  	t = &rctx->walk.cache[0];

  	n = nbytes;
  	while (n) {

  		if (t->length && rctx->walk.flags & ASYNC_FLAGS_MISALIGNED) {

  			BUG_ON(!sg_page(t));

  			dpage = sg_page(t);
  			doff = 0;

  			len = t->length;
  		} else {

  			BUG_ON(!sg_page(dst));

  			dpage = sg_page(dst);
  			doff = dst->offset;

  			len = dst->length;
  		}
  		len = min(len, n);

  		hifn_setup_dst_desc(dev, dpage, doff, len, n - len == 0);

  		dst++;
  		t++;
  		n -= len;
  	}

  	hifn_setup_cmd_desc(dev, ctx, rctx, priv, nbytes);

  	hifn_setup_res_desc(dev);
  	return 0;
  }

  static int hifn_cipher_walk_init(struct hifn_cipher_walk *w,

  		int num, gfp_t gfp_flags)
  {
  	int i;
  
  	num = min(ASYNC_SCATTERLIST_CACHE, num);
  	sg_init_table(w->cache, num);
  
  	w->num = 0;

  	for (i = 0; i < num; ++i) {

  		struct page *page = alloc_page(gfp_flags);
  		struct scatterlist *s;
  
  		if (!page)
  			break;
  
  		s = &w->cache[i];
  
  		sg_set_page(s, page, PAGE_SIZE, 0);
  		w->num++;
  	}
  
  	return i;
  }

  static void hifn_cipher_walk_exit(struct hifn_cipher_walk *w)

  {
  	int i;

  	for (i = 0; i < w->num; ++i) {

  		struct scatterlist *s = &w->cache[i];
  
  		__free_page(sg_page(s));
  
  		s->length = 0;
  	}
  
  	w->num = 0;
  }

  static int ablkcipher_add(unsigned int *drestp, struct scatterlist *dst,

  		unsigned int size, unsigned int *nbytesp)
  {
  	unsigned int copy, drest = *drestp, nbytes = *nbytesp;
  	int idx = 0;

  
  	if (drest < size || size > nbytes)
  		return -EINVAL;
  
  	while (size) {

  		copy = min3(drest, size, dst->length);

  
  		size -= copy;
  		drest -= copy;
  		nbytes -= copy;

  		pr_debug("%s: copy: %u, size: %u, drest: %u, nbytes: %u.
  ",
  			 __func__, copy, size, drest, nbytes);

  		dst++;

  		idx++;
  	}
  
  	*nbytesp = nbytes;
  	*drestp = drest;
  
  	return idx;
  }

  static int hifn_cipher_walk(struct ablkcipher_request *req,
  		struct hifn_cipher_walk *w)

  {

  	struct scatterlist *dst, *t;

  	unsigned int nbytes = req->nbytes, offset, copy, diff;
  	int idx, tidx, err;
  
  	tidx = idx = 0;
  	offset = 0;
  	while (nbytes) {
  		if (idx >= w->num && (w->flags & ASYNC_FLAGS_MISALIGNED))
  			return -EINVAL;

  		dst = &req->dst[idx];

  		pr_debug("
  %s: dlen: %u, doff: %u, offset: %u, nbytes: %u.
  ",
  			 __func__, dst->length, dst->offset, offset, nbytes);

  		if (!IS_ALIGNED(dst->offset, HIFN_D_DST_DALIGN) ||
  		    !IS_ALIGNED(dst->length, HIFN_D_DST_DALIGN) ||
  		    offset) {

  			unsigned slen = min(dst->length - offset, nbytes);

  			unsigned dlen = PAGE_SIZE;
  
  			t = &w->cache[idx];

  			err = ablkcipher_add(&dlen, dst, slen, &nbytes);

  			if (err < 0)

  				return err;

  
  			idx += err;

  			copy = slen & ~(HIFN_D_DST_DALIGN - 1);
  			diff = slen & (HIFN_D_DST_DALIGN - 1);

  
  			if (dlen < nbytes) {
  				/*
  				 * Destination page does not have enough space
  				 * to put there additional blocksized chunk,
  				 * so we mark that page as containing only
  				 * blocksize aligned chunks:

  				 *	t->length = (slen & ~(HIFN_D_DST_DALIGN - 1));

  				 * and increase number of bytes to be processed
  				 * in next chunk:

  				 *	nbytes += diff;

  				 */
  				nbytes += diff;
  
  				/*
  				 * Temporary of course...
  				 * Kick author if you will catch this one.
  				 */

  				pr_err("%s: dlen: %u, nbytes: %u, slen: %u, offset: %u.
  ",
  				       __func__, dlen, nbytes, slen, offset);
  				pr_err("%s: please contact author to fix this "
  				       "issue, generally you should not catch "
  				       "this path under any condition but who "
  				       "knows how did you use crypto code.
  "
  				       "Thank you.
  ",	__func__);

  				BUG();
  			} else {
  				copy += diff + nbytes;

  				dst = &req->dst[idx];

  				err = ablkcipher_add(&dlen, dst, nbytes, &nbytes);

  				if (err < 0)

  					return err;

  
  				idx += err;
  			}
  
  			t->length = copy;
  			t->offset = offset;

  		} else {

  			nbytes -= min(dst->length, nbytes);

  			idx++;
  		}
  
  		tidx++;
  	}
  
  	return tidx;

  }
  
  static int hifn_setup_session(struct ablkcipher_request *req)
  {
  	struct hifn_context *ctx = crypto_tfm_ctx(req->base.tfm);

  	struct hifn_request_context *rctx = ablkcipher_request_ctx(req);

  	struct hifn_device *dev = ctx->dev;

  	unsigned long dlen, flags;
  	unsigned int nbytes = req->nbytes, idx = 0;

  	int err = -EINVAL, sg_num;

  	struct scatterlist *dst;

  	if (rctx->iv && !rctx->ivsize && rctx->mode != ACRYPTO_MODE_ECB)

  		goto err_out_exit;

  	rctx->walk.flags = 0;

  
  	while (nbytes) {

  		dst = &req->dst[idx];

  		dlen = min(dst->length, nbytes);

  		if (!IS_ALIGNED(dst->offset, HIFN_D_DST_DALIGN) ||

  		    !IS_ALIGNED(dlen, HIFN_D_DST_DALIGN))

  			rctx->walk.flags |= ASYNC_FLAGS_MISALIGNED;

  		nbytes -= dlen;

  		idx++;
  	}

  	if (rctx->walk.flags & ASYNC_FLAGS_MISALIGNED) {

  		err = hifn_cipher_walk_init(&rctx->walk, idx, GFP_ATOMIC);

  		if (err < 0)
  			return err;
  	}

  	sg_num = hifn_cipher_walk(req, &rctx->walk);

  	if (sg_num < 0) {
  		err = sg_num;
  		goto err_out_exit;
  	}

  
  	spin_lock_irqsave(&dev->lock, flags);
  	if (dev->started + sg_num > HIFN_QUEUE_LENGTH) {
  		err = -EAGAIN;
  		goto err_out;
  	}

  	err = hifn_setup_dma(dev, ctx, rctx, req->src, req->dst, req->nbytes, req);

  	if (err)
  		goto err_out;

  	dev->snum++;

  	dev->active = HIFN_DEFAULT_ACTIVE_NUM;
  	spin_unlock_irqrestore(&dev->lock, flags);
  
  	return 0;
  
  err_out:
  	spin_unlock_irqrestore(&dev->lock, flags);
  err_out_exit:

  	if (err) {

  		dev_info(&dev->pdev->dev, "iv: %p [%d], key: %p [%d], mode: %u, op: %u, "
  			 "type: %u, err: %d.
  ",
  			 rctx->iv, rctx->ivsize,
  			 ctx->key, ctx->keysize,
  			 rctx->mode, rctx->op, rctx->type, err);

  	}

  
  	return err;
  }

  static int hifn_start_device(struct hifn_device *dev)
  {
  	int err;

  	dev->started = dev->active = 0;

  	hifn_reset_dma(dev, 1);
  
  	err = hifn_enable_crypto(dev);
  	if (err)
  		return err;
  
  	hifn_reset_puc(dev);
  
  	hifn_init_dma(dev);
  
  	hifn_init_registers(dev);
  
  	hifn_init_pubrng(dev);
  
  	return 0;
  }
  
  static int ablkcipher_get(void *saddr, unsigned int *srestp, unsigned int offset,
  		struct scatterlist *dst, unsigned int size, unsigned int *nbytesp)
  {
  	unsigned int srest = *srestp, nbytes = *nbytesp, copy;
  	void *daddr;
  	int idx = 0;
  
  	if (srest < size || size > nbytes)
  		return -EINVAL;
  
  	while (size) {

  		copy = min3(srest, dst->length, size);

  		daddr = kmap_atomic(sg_page(dst));

  		memcpy(daddr + dst->offset + offset, saddr, copy);

  		kunmap_atomic(daddr);

  
  		nbytes -= copy;
  		size -= copy;
  		srest -= copy;
  		saddr += copy;
  		offset = 0;

  		pr_debug("%s: copy: %u, size: %u, srest: %u, nbytes: %u.
  ",
  			 __func__, copy, size, srest, nbytes);

  
  		dst++;
  		idx++;
  	}
  
  	*nbytesp = nbytes;
  	*srestp = srest;
  
  	return idx;
  }

  static inline void hifn_complete_sa(struct hifn_device *dev, int i)

  {

  	unsigned long flags;

  	spin_lock_irqsave(&dev->lock, flags);
  	dev->sa[i] = NULL;
  	dev->started--;
  	if (dev->started < 0)

  		dev_info(&dev->pdev->dev, "%s: started: %d.
  ", __func__,
  			 dev->started);

  	spin_unlock_irqrestore(&dev->lock, flags);
  	BUG_ON(dev->started < 0);
  }

  static void hifn_process_ready(struct ablkcipher_request *req, int error)
  {
  	struct hifn_request_context *rctx = ablkcipher_request_ctx(req);

  	if (rctx->walk.flags & ASYNC_FLAGS_MISALIGNED) {

  		unsigned int nbytes = req->nbytes;
  		int idx = 0, err;
  		struct scatterlist *dst, *t;
  		void *saddr;

  		while (nbytes) {

  			t = &rctx->walk.cache[idx];

  			dst = &req->dst[idx];

  			pr_debug("
  %s: sg_page(t): %p, t->length: %u, "

  				"sg_page(dst): %p, dst->length: %u, "
  				"nbytes: %u.
  ",
  				__func__, sg_page(t), t->length,
  				sg_page(dst), dst->length, nbytes);

  			if (!t->length) {
  				nbytes -= min(dst->length, nbytes);
  				idx++;
  				continue;
  			}

  			saddr = kmap_atomic(sg_page(t));

  			err = ablkcipher_get(saddr, &t->length, t->offset,
  					dst, nbytes, &nbytes);
  			if (err < 0) {

  				kunmap_atomic(saddr);

  				break;

  			}

  			idx += err;

  			kunmap_atomic(saddr);

  		}

  		hifn_cipher_walk_exit(&rctx->walk);

  	}

  
  	req->base.complete(&req->base, error);

  }

  static void hifn_clear_rings(struct hifn_device *dev, int error)

  {
  	struct hifn_dma *dma = (struct hifn_dma *)dev->desc_virt;
  	int i, u;

  	dev_dbg(&dev->pdev->dev, "ring cleanup 1: i: %d.%d.%d.%d, u: %d.%d.%d.%d, "

  			"k: %d.%d.%d.%d.
  ",

  			dma->cmdi, dma->srci, dma->dsti, dma->resi,
  			dma->cmdu, dma->srcu, dma->dstu, dma->resu,
  			dma->cmdk, dma->srck, dma->dstk, dma->resk);
  
  	i = dma->resk; u = dma->resu;
  	while (u != 0) {
  		if (dma->resr[i].l & __cpu_to_le32(HIFN_D_VALID))
  			break;

  		if (dev->sa[i]) {
  			dev->success++;
  			dev->reset = 0;
  			hifn_process_ready(dev->sa[i], error);
  			hifn_complete_sa(dev, i);
  		}

  		if (++i == HIFN_D_RES_RSIZE)

  			i = 0;

  		u--;

  	}
  	dma->resk = i; dma->resu = u;
  
  	i = dma->srck; u = dma->srcu;
  	while (u != 0) {

  		if (dma->srcr[i].l & __cpu_to_le32(HIFN_D_VALID))
  			break;

  		if (++i == HIFN_D_SRC_RSIZE)
  			i = 0;
  		u--;

  	}
  	dma->srck = i; dma->srcu = u;
  
  	i = dma->cmdk; u = dma->cmdu;
  	while (u != 0) {
  		if (dma->cmdr[i].l & __cpu_to_le32(HIFN_D_VALID))
  			break;

  		if (++i == HIFN_D_CMD_RSIZE)

  			i = 0;

  		u--;

  	}
  	dma->cmdk = i; dma->cmdu = u;
  
  	i = dma->dstk; u = dma->dstu;
  	while (u != 0) {

  		if (dma->dstr[i].l & __cpu_to_le32(HIFN_D_VALID))
  			break;

  		if (++i == HIFN_D_DST_RSIZE)
  			i = 0;
  		u--;

  	}
  	dma->dstk = i; dma->dstu = u;

  	dev_dbg(&dev->pdev->dev, "ring cleanup 2: i: %d.%d.%d.%d, u: %d.%d.%d.%d, "

  			"k: %d.%d.%d.%d.
  ",

  			dma->cmdi, dma->srci, dma->dsti, dma->resi,
  			dma->cmdu, dma->srcu, dma->dstu, dma->resu,
  			dma->cmdk, dma->srck, dma->dstk, dma->resk);
  }
  
  static void hifn_work(struct work_struct *work)
  {

  	struct delayed_work *dw = to_delayed_work(work);

  	struct hifn_device *dev = container_of(dw, struct hifn_device, work);
  	unsigned long flags;
  	int reset = 0;
  	u32 r = 0;
  
  	spin_lock_irqsave(&dev->lock, flags);
  	if (dev->active == 0) {
  		struct hifn_dma *dma = (struct hifn_dma *)dev->desc_virt;
  
  		if (dma->cmdu == 0 && (dev->flags & HIFN_FLAG_CMD_BUSY)) {
  			dev->flags &= ~HIFN_FLAG_CMD_BUSY;
  			r |= HIFN_DMACSR_C_CTRL_DIS;
  		}
  		if (dma->srcu == 0 && (dev->flags & HIFN_FLAG_SRC_BUSY)) {
  			dev->flags &= ~HIFN_FLAG_SRC_BUSY;
  			r |= HIFN_DMACSR_S_CTRL_DIS;
  		}
  		if (dma->dstu == 0 && (dev->flags & HIFN_FLAG_DST_BUSY)) {
  			dev->flags &= ~HIFN_FLAG_DST_BUSY;
  			r |= HIFN_DMACSR_D_CTRL_DIS;
  		}
  		if (dma->resu == 0 && (dev->flags & HIFN_FLAG_RES_BUSY)) {
  			dev->flags &= ~HIFN_FLAG_RES_BUSY;
  			r |= HIFN_DMACSR_R_CTRL_DIS;
  		}
  		if (r)
  			hifn_write_1(dev, HIFN_1_DMA_CSR, r);
  	} else
  		dev->active--;

  	if ((dev->prev_success == dev->success) && dev->started)

  		reset = 1;
  	dev->prev_success = dev->success;
  	spin_unlock_irqrestore(&dev->lock, flags);
  
  	if (reset) {

  		if (++dev->reset >= 5) {

  			int i;
  			struct hifn_dma *dma = (struct hifn_dma *)dev->desc_virt;

  			dev_info(&dev->pdev->dev,
  				 "r: %08x, active: %d, started: %d, "
  				 "success: %lu: qlen: %u/%u, reset: %d.
  ",
  				 r, dev->active, dev->started,
  				 dev->success, dev->queue.qlen, dev->queue.max_qlen,
  				 reset);

  			dev_info(&dev->pdev->dev, "%s: res: ", __func__);

  			for (i = 0; i < HIFN_D_RES_RSIZE; ++i) {

  				pr_info("%x.%p ", dma->resr[i].l, dev->sa[i]);

  				if (dev->sa[i]) {
  					hifn_process_ready(dev->sa[i], -ENODEV);
  					hifn_complete_sa(dev, i);
  				}
  			}

  			pr_info("
  ");

  			hifn_reset_dma(dev, 1);
  			hifn_stop_device(dev);
  			hifn_start_device(dev);
  			dev->reset = 0;
  		}

  		tasklet_schedule(&dev->tasklet);

  	}
  
  	schedule_delayed_work(&dev->work, HZ);
  }
  
  static irqreturn_t hifn_interrupt(int irq, void *data)
  {
  	struct hifn_device *dev = (struct hifn_device *)data;
  	struct hifn_dma *dma = (struct hifn_dma *)dev->desc_virt;
  	u32 dmacsr, restart;
  
  	dmacsr = hifn_read_1(dev, HIFN_1_DMA_CSR);

  	dev_dbg(&dev->pdev->dev, "1 dmacsr: %08x, dmareg: %08x, res: %08x [%d], "

  			"i: %d.%d.%d.%d, u: %d.%d.%d.%d.
  ",

  		dmacsr, dev->dmareg, dmacsr & dev->dmareg, dma->cmdi,

  		dma->cmdi, dma->srci, dma->dsti, dma->resi,
  		dma->cmdu, dma->srcu, dma->dstu, dma->resu);

  
  	if ((dmacsr & dev->dmareg) == 0)
  		return IRQ_NONE;
  
  	hifn_write_1(dev, HIFN_1_DMA_CSR, dmacsr & dev->dmareg);
  
  	if (dmacsr & HIFN_DMACSR_ENGINE)
  		hifn_write_0(dev, HIFN_0_PUISR, hifn_read_0(dev, HIFN_0_PUISR));
  	if (dmacsr & HIFN_DMACSR_PUBDONE)
  		hifn_write_1(dev, HIFN_1_PUB_STATUS,
  			hifn_read_1(dev, HIFN_1_PUB_STATUS) | HIFN_PUBSTS_DONE);
  
  	restart = dmacsr & (HIFN_DMACSR_R_OVER | HIFN_DMACSR_D_OVER);
  	if (restart) {
  		u32 puisr = hifn_read_0(dev, HIFN_0_PUISR);

  		dev_warn(&dev->pdev->dev, "overflow: r: %d, d: %d, puisr: %08x, d: %u.
  ",
  			 !!(dmacsr & HIFN_DMACSR_R_OVER),
  			 !!(dmacsr & HIFN_DMACSR_D_OVER),

  			puisr, !!(puisr & HIFN_PUISR_DSTOVER));

  		if (!!(puisr & HIFN_PUISR_DSTOVER))
  			hifn_write_0(dev, HIFN_0_PUISR, HIFN_PUISR_DSTOVER);
  		hifn_write_1(dev, HIFN_1_DMA_CSR, dmacsr & (HIFN_DMACSR_R_OVER |
  					HIFN_DMACSR_D_OVER));
  	}
  
  	restart = dmacsr & (HIFN_DMACSR_C_ABORT | HIFN_DMACSR_S_ABORT |
  			HIFN_DMACSR_D_ABORT | HIFN_DMACSR_R_ABORT);
  	if (restart) {

  		dev_warn(&dev->pdev->dev, "abort: c: %d, s: %d, d: %d, r: %d.
  ",
  			 !!(dmacsr & HIFN_DMACSR_C_ABORT),
  			 !!(dmacsr & HIFN_DMACSR_S_ABORT),
  			 !!(dmacsr & HIFN_DMACSR_D_ABORT),
  			 !!(dmacsr & HIFN_DMACSR_R_ABORT));

  		hifn_reset_dma(dev, 1);
  		hifn_init_dma(dev);
  		hifn_init_registers(dev);
  	}
  
  	if ((dmacsr & HIFN_DMACSR_C_WAIT) && (dma->cmdu == 0)) {

  		dev_dbg(&dev->pdev->dev, "wait on command.
  ");

  		dev->dmareg &= ~(HIFN_DMAIER_C_WAIT);
  		hifn_write_1(dev, HIFN_1_DMA_IER, dev->dmareg);
  	}

  	tasklet_schedule(&dev->tasklet);

  
  	return IRQ_HANDLED;
  }
  
  static void hifn_flush(struct hifn_device *dev)
  {
  	unsigned long flags;
  	struct crypto_async_request *async_req;

  	struct ablkcipher_request *req;
  	struct hifn_dma *dma = (struct hifn_dma *)dev->desc_virt;
  	int i;

  	for (i = 0; i < HIFN_D_RES_RSIZE; ++i) {

  		struct hifn_desc *d = &dma->resr[i];
  
  		if (dev->sa[i]) {
  			hifn_process_ready(dev->sa[i],

  				(d->l & __cpu_to_le32(HIFN_D_VALID)) ? -ENODEV : 0);

  			hifn_complete_sa(dev, i);

  		}
  	}

  	spin_lock_irqsave(&dev->lock, flags);

  	while ((async_req = crypto_dequeue_request(&dev->queue))) {

  		req = ablkcipher_request_cast(async_req);

  		spin_unlock_irqrestore(&dev->lock, flags);

  
  		hifn_process_ready(req, -ENODEV);

  
  		spin_lock_irqsave(&dev->lock, flags);

  	}
  	spin_unlock_irqrestore(&dev->lock, flags);
  }
  
  static int hifn_setkey(struct crypto_ablkcipher *cipher, const u8 *key,
  		unsigned int len)
  {
  	struct crypto_tfm *tfm = crypto_ablkcipher_tfm(cipher);
  	struct hifn_context *ctx = crypto_tfm_ctx(tfm);
  	struct hifn_device *dev = ctx->dev;
  
  	if (len > HIFN_MAX_CRYPT_KEY_LENGTH) {
  		crypto_ablkcipher_set_flags(cipher, CRYPTO_TFM_RES_BAD_KEY_LEN);
  		return -1;
  	}

  	if (len == HIFN_DES_KEY_LENGTH) {
  		u32 tmp[DES_EXPKEY_WORDS];
  		int ret = des_ekey(tmp, key);

  		if (unlikely(ret == 0) && (tfm->crt_flags & CRYPTO_TFM_REQ_WEAK_KEY)) {
  			tfm->crt_flags |= CRYPTO_TFM_RES_WEAK_KEY;
  			return -EINVAL;
  		}
  	}

  	dev->flags &= ~HIFN_FLAG_OLD_KEY;
  
  	memcpy(ctx->key, key, len);
  	ctx->keysize = len;
  
  	return 0;
  }
  
  static int hifn_handle_req(struct ablkcipher_request *req)
  {
  	struct hifn_context *ctx = crypto_tfm_ctx(req->base.tfm);
  	struct hifn_device *dev = ctx->dev;
  	int err = -EAGAIN;
  
  	if (dev->started + DIV_ROUND_UP(req->nbytes, PAGE_SIZE) <= HIFN_QUEUE_LENGTH)
  		err = hifn_setup_session(req);
  
  	if (err == -EAGAIN) {
  		unsigned long flags;
  
  		spin_lock_irqsave(&dev->lock, flags);
  		err = ablkcipher_enqueue_request(&dev->queue, req);
  		spin_unlock_irqrestore(&dev->lock, flags);
  	}
  
  	return err;
  }
  
  static int hifn_setup_crypto_req(struct ablkcipher_request *req, u8 op,
  		u8 type, u8 mode)
  {
  	struct hifn_context *ctx = crypto_tfm_ctx(req->base.tfm);

  	struct hifn_request_context *rctx = ablkcipher_request_ctx(req);

  	unsigned ivsize;
  
  	ivsize = crypto_ablkcipher_ivsize(crypto_ablkcipher_reqtfm(req));
  
  	if (req->info && mode != ACRYPTO_MODE_ECB) {
  		if (type == ACRYPTO_TYPE_AES_128)
  			ivsize = HIFN_AES_IV_LENGTH;
  		else if (type == ACRYPTO_TYPE_DES)
  			ivsize = HIFN_DES_KEY_LENGTH;
  		else if (type == ACRYPTO_TYPE_3DES)
  			ivsize = HIFN_3DES_KEY_LENGTH;
  	}
  
  	if (ctx->keysize != 16 && type == ACRYPTO_TYPE_AES_128) {
  		if (ctx->keysize == 24)
  			type = ACRYPTO_TYPE_AES_192;
  		else if (ctx->keysize == 32)
  			type = ACRYPTO_TYPE_AES_256;
  	}

  	rctx->op = op;
  	rctx->mode = mode;
  	rctx->type = type;
  	rctx->iv = req->info;
  	rctx->ivsize = ivsize;

  
  	/*
  	 * HEAVY TODO: needs to kick Herbert XU to write documentation.
  	 * HEAVY TODO: needs to kick Herbert XU to write documentation.
  	 * HEAVY TODO: needs to kick Herbert XU to write documentation.
  	 */
  
  	return hifn_handle_req(req);
  }
  
  static int hifn_process_queue(struct hifn_device *dev)
  {

  	struct crypto_async_request *async_req, *backlog;

  	struct ablkcipher_request *req;
  	unsigned long flags;
  	int err = 0;
  
  	while (dev->started < HIFN_QUEUE_LENGTH) {
  		spin_lock_irqsave(&dev->lock, flags);

  		backlog = crypto_get_backlog(&dev->queue);

  		async_req = crypto_dequeue_request(&dev->queue);
  		spin_unlock_irqrestore(&dev->lock, flags);
  
  		if (!async_req)
  			break;

  		if (backlog)
  			backlog->complete(backlog, -EINPROGRESS);

  		req = ablkcipher_request_cast(async_req);

  
  		err = hifn_handle_req(req);
  		if (err)
  			break;
  	}
  
  	return err;
  }
  
  static int hifn_setup_crypto(struct ablkcipher_request *req, u8 op,
  		u8 type, u8 mode)
  {
  	int err;
  	struct hifn_context *ctx = crypto_tfm_ctx(req->base.tfm);
  	struct hifn_device *dev = ctx->dev;
  
  	err = hifn_setup_crypto_req(req, op, type, mode);
  	if (err)
  		return err;
  
  	if (dev->started < HIFN_QUEUE_LENGTH &&	dev->queue.qlen)

  		hifn_process_queue(dev);

  	return -EINPROGRESS;

  }
  
  /*
   * AES ecryption functions.
   */
  static inline int hifn_encrypt_aes_ecb(struct ablkcipher_request *req)
  {
  	return hifn_setup_crypto(req, ACRYPTO_OP_ENCRYPT,
  			ACRYPTO_TYPE_AES_128, ACRYPTO_MODE_ECB);
  }
  static inline int hifn_encrypt_aes_cbc(struct ablkcipher_request *req)
  {
  	return hifn_setup_crypto(req, ACRYPTO_OP_ENCRYPT,
  			ACRYPTO_TYPE_AES_128, ACRYPTO_MODE_CBC);
  }
  static inline int hifn_encrypt_aes_cfb(struct ablkcipher_request *req)
  {
  	return hifn_setup_crypto(req, ACRYPTO_OP_ENCRYPT,
  			ACRYPTO_TYPE_AES_128, ACRYPTO_MODE_CFB);
  }
  static inline int hifn_encrypt_aes_ofb(struct ablkcipher_request *req)
  {
  	return hifn_setup_crypto(req, ACRYPTO_OP_ENCRYPT,
  			ACRYPTO_TYPE_AES_128, ACRYPTO_MODE_OFB);
  }
  
  /*
   * AES decryption functions.
   */
  static inline int hifn_decrypt_aes_ecb(struct ablkcipher_request *req)
  {
  	return hifn_setup_crypto(req, ACRYPTO_OP_DECRYPT,
  			ACRYPTO_TYPE_AES_128, ACRYPTO_MODE_ECB);
  }
  static inline int hifn_decrypt_aes_cbc(struct ablkcipher_request *req)
  {
  	return hifn_setup_crypto(req, ACRYPTO_OP_DECRYPT,
  			ACRYPTO_TYPE_AES_128, ACRYPTO_MODE_CBC);
  }
  static inline int hifn_decrypt_aes_cfb(struct ablkcipher_request *req)
  {
  	return hifn_setup_crypto(req, ACRYPTO_OP_DECRYPT,
  			ACRYPTO_TYPE_AES_128, ACRYPTO_MODE_CFB);
  }
  static inline int hifn_decrypt_aes_ofb(struct ablkcipher_request *req)
  {
  	return hifn_setup_crypto(req, ACRYPTO_OP_DECRYPT,
  			ACRYPTO_TYPE_AES_128, ACRYPTO_MODE_OFB);
  }
  
  /*
   * DES ecryption functions.
   */
  static inline int hifn_encrypt_des_ecb(struct ablkcipher_request *req)
  {
  	return hifn_setup_crypto(req, ACRYPTO_OP_ENCRYPT,
  			ACRYPTO_TYPE_DES, ACRYPTO_MODE_ECB);
  }
  static inline int hifn_encrypt_des_cbc(struct ablkcipher_request *req)
  {
  	return hifn_setup_crypto(req, ACRYPTO_OP_ENCRYPT,
  			ACRYPTO_TYPE_DES, ACRYPTO_MODE_CBC);
  }
  static inline int hifn_encrypt_des_cfb(struct ablkcipher_request *req)
  {
  	return hifn_setup_crypto(req, ACRYPTO_OP_ENCRYPT,
  			ACRYPTO_TYPE_DES, ACRYPTO_MODE_CFB);
  }
  static inline int hifn_encrypt_des_ofb(struct ablkcipher_request *req)
  {
  	return hifn_setup_crypto(req, ACRYPTO_OP_ENCRYPT,
  			ACRYPTO_TYPE_DES, ACRYPTO_MODE_OFB);
  }
  
  /*
   * DES decryption functions.
   */
  static inline int hifn_decrypt_des_ecb(struct ablkcipher_request *req)
  {
  	return hifn_setup_crypto(req, ACRYPTO_OP_DECRYPT,
  			ACRYPTO_TYPE_DES, ACRYPTO_MODE_ECB);
  }
  static inline int hifn_decrypt_des_cbc(struct ablkcipher_request *req)
  {
  	return hifn_setup_crypto(req, ACRYPTO_OP_DECRYPT,
  			ACRYPTO_TYPE_DES, ACRYPTO_MODE_CBC);
  }
  static inline int hifn_decrypt_des_cfb(struct ablkcipher_request *req)
  {
  	return hifn_setup_crypto(req, ACRYPTO_OP_DECRYPT,
  			ACRYPTO_TYPE_DES, ACRYPTO_MODE_CFB);
  }
  static inline int hifn_decrypt_des_ofb(struct ablkcipher_request *req)
  {
  	return hifn_setup_crypto(req, ACRYPTO_OP_DECRYPT,
  			ACRYPTO_TYPE_DES, ACRYPTO_MODE_OFB);
  }
  
  /*
   * 3DES ecryption functions.
   */
  static inline int hifn_encrypt_3des_ecb(struct ablkcipher_request *req)
  {
  	return hifn_setup_crypto(req, ACRYPTO_OP_ENCRYPT,
  			ACRYPTO_TYPE_3DES, ACRYPTO_MODE_ECB);
  }
  static inline int hifn_encrypt_3des_cbc(struct ablkcipher_request *req)
  {
  	return hifn_setup_crypto(req, ACRYPTO_OP_ENCRYPT,
  			ACRYPTO_TYPE_3DES, ACRYPTO_MODE_CBC);
  }
  static inline int hifn_encrypt_3des_cfb(struct ablkcipher_request *req)
  {
  	return hifn_setup_crypto(req, ACRYPTO_OP_ENCRYPT,
  			ACRYPTO_TYPE_3DES, ACRYPTO_MODE_CFB);
  }
  static inline int hifn_encrypt_3des_ofb(struct ablkcipher_request *req)
  {
  	return hifn_setup_crypto(req, ACRYPTO_OP_ENCRYPT,
  			ACRYPTO_TYPE_3DES, ACRYPTO_MODE_OFB);
  }

  /* 3DES decryption functions. */

  static inline int hifn_decrypt_3des_ecb(struct ablkcipher_request *req)
  {
  	return hifn_setup_crypto(req, ACRYPTO_OP_DECRYPT,
  			ACRYPTO_TYPE_3DES, ACRYPTO_MODE_ECB);
  }
  static inline int hifn_decrypt_3des_cbc(struct ablkcipher_request *req)
  {
  	return hifn_setup_crypto(req, ACRYPTO_OP_DECRYPT,
  			ACRYPTO_TYPE_3DES, ACRYPTO_MODE_CBC);
  }
  static inline int hifn_decrypt_3des_cfb(struct ablkcipher_request *req)
  {
  	return hifn_setup_crypto(req, ACRYPTO_OP_DECRYPT,
  			ACRYPTO_TYPE_3DES, ACRYPTO_MODE_CFB);
  }
  static inline int hifn_decrypt_3des_ofb(struct ablkcipher_request *req)
  {
  	return hifn_setup_crypto(req, ACRYPTO_OP_DECRYPT,
  			ACRYPTO_TYPE_3DES, ACRYPTO_MODE_OFB);
  }

  struct hifn_alg_template {

  	char name[CRYPTO_MAX_ALG_NAME];
  	char drv_name[CRYPTO_MAX_ALG_NAME];
  	unsigned int bsize;
  	struct ablkcipher_alg ablkcipher;
  };
  
  static struct hifn_alg_template hifn_alg_templates[] = {
  	/*
  	 * 3DES ECB, CBC, CFB and OFB modes.
  	 */
  	{

  		.name = "cfb(des3_ede)", .drv_name = "cfb-3des", .bsize = 8,

  		.ablkcipher = {
  			.min_keysize	=	HIFN_3DES_KEY_LENGTH,
  			.max_keysize	=	HIFN_3DES_KEY_LENGTH,
  			.setkey		=	hifn_setkey,
  			.encrypt	=	hifn_encrypt_3des_cfb,
  			.decrypt	=	hifn_decrypt_3des_cfb,
  		},
  	},
  	{

  		.name = "ofb(des3_ede)", .drv_name = "ofb-3des", .bsize = 8,

  		.ablkcipher = {
  			.min_keysize	=	HIFN_3DES_KEY_LENGTH,
  			.max_keysize	=	HIFN_3DES_KEY_LENGTH,
  			.setkey		=	hifn_setkey,
  			.encrypt	=	hifn_encrypt_3des_ofb,
  			.decrypt	=	hifn_decrypt_3des_ofb,
  		},
  	},
  	{

  		.name = "cbc(des3_ede)", .drv_name = "cbc-3des", .bsize = 8,

  		.ablkcipher = {

  			.ivsize		=	HIFN_IV_LENGTH,

  			.min_keysize	=	HIFN_3DES_KEY_LENGTH,
  			.max_keysize	=	HIFN_3DES_KEY_LENGTH,
  			.setkey		=	hifn_setkey,
  			.encrypt	=	hifn_encrypt_3des_cbc,
  			.decrypt	=	hifn_decrypt_3des_cbc,
  		},
  	},
  	{

  		.name = "ecb(des3_ede)", .drv_name = "ecb-3des", .bsize = 8,

  		.ablkcipher = {
  			.min_keysize	=	HIFN_3DES_KEY_LENGTH,
  			.max_keysize	=	HIFN_3DES_KEY_LENGTH,
  			.setkey		=	hifn_setkey,
  			.encrypt	=	hifn_encrypt_3des_ecb,
  			.decrypt	=	hifn_decrypt_3des_ecb,
  		},
  	},
  
  	/*
  	 * DES ECB, CBC, CFB and OFB modes.
  	 */
  	{

  		.name = "cfb(des)", .drv_name = "cfb-des", .bsize = 8,

  		.ablkcipher = {
  			.min_keysize	=	HIFN_DES_KEY_LENGTH,
  			.max_keysize	=	HIFN_DES_KEY_LENGTH,
  			.setkey		=	hifn_setkey,
  			.encrypt	=	hifn_encrypt_des_cfb,
  			.decrypt	=	hifn_decrypt_des_cfb,
  		},
  	},
  	{

  		.name = "ofb(des)", .drv_name = "ofb-des", .bsize = 8,

  		.ablkcipher = {
  			.min_keysize	=	HIFN_DES_KEY_LENGTH,
  			.max_keysize	=	HIFN_DES_KEY_LENGTH,
  			.setkey		=	hifn_setkey,
  			.encrypt	=	hifn_encrypt_des_ofb,
  			.decrypt	=	hifn_decrypt_des_ofb,
  		},
  	},
  	{

  		.name = "cbc(des)", .drv_name = "cbc-des", .bsize = 8,

  		.ablkcipher = {

  			.ivsize		=	HIFN_IV_LENGTH,

  			.min_keysize	=	HIFN_DES_KEY_LENGTH,
  			.max_keysize	=	HIFN_DES_KEY_LENGTH,
  			.setkey		=	hifn_setkey,
  			.encrypt	=	hifn_encrypt_des_cbc,
  			.decrypt	=	hifn_decrypt_des_cbc,
  		},
  	},
  	{

  		.name = "ecb(des)", .drv_name = "ecb-des", .bsize = 8,

  		.ablkcipher = {
  			.min_keysize	=	HIFN_DES_KEY_LENGTH,
  			.max_keysize	=	HIFN_DES_KEY_LENGTH,
  			.setkey		=	hifn_setkey,
  			.encrypt	=	hifn_encrypt_des_ecb,
  			.decrypt	=	hifn_decrypt_des_ecb,
  		},
  	},
  
  	/*
  	 * AES ECB, CBC, CFB and OFB modes.
  	 */
  	{

  		.name = "ecb(aes)", .drv_name = "ecb-aes", .bsize = 16,

  		.ablkcipher = {
  			.min_keysize	=	AES_MIN_KEY_SIZE,
  			.max_keysize	=	AES_MAX_KEY_SIZE,
  			.setkey		=	hifn_setkey,
  			.encrypt	=	hifn_encrypt_aes_ecb,
  			.decrypt	=	hifn_decrypt_aes_ecb,
  		},
  	},
  	{

  		.name = "cbc(aes)", .drv_name = "cbc-aes", .bsize = 16,

  		.ablkcipher = {

  			.ivsize		=	HIFN_AES_IV_LENGTH,

  			.min_keysize	=	AES_MIN_KEY_SIZE,
  			.max_keysize	=	AES_MAX_KEY_SIZE,
  			.setkey		=	hifn_setkey,
  			.encrypt	=	hifn_encrypt_aes_cbc,
  			.decrypt	=	hifn_decrypt_aes_cbc,
  		},
  	},
  	{

  		.name = "cfb(aes)", .drv_name = "cfb-aes", .bsize = 16,

  		.ablkcipher = {
  			.min_keysize	=	AES_MIN_KEY_SIZE,
  			.max_keysize	=	AES_MAX_KEY_SIZE,
  			.setkey		=	hifn_setkey,
  			.encrypt	=	hifn_encrypt_aes_cfb,
  			.decrypt	=	hifn_decrypt_aes_cfb,
  		},
  	},
  	{

  		.name = "ofb(aes)", .drv_name = "ofb-aes", .bsize = 16,

  		.ablkcipher = {
  			.min_keysize	=	AES_MIN_KEY_SIZE,
  			.max_keysize	=	AES_MAX_KEY_SIZE,
  			.setkey		=	hifn_setkey,
  			.encrypt	=	hifn_encrypt_aes_ofb,
  			.decrypt	=	hifn_decrypt_aes_ofb,
  		},
  	},
  };
  
  static int hifn_cra_init(struct crypto_tfm *tfm)
  {
  	struct crypto_alg *alg = tfm->__crt_alg;
  	struct hifn_crypto_alg *ha = crypto_alg_to_hifn(alg);
  	struct hifn_context *ctx = crypto_tfm_ctx(tfm);
  
  	ctx->dev = ha->dev;

  	tfm->crt_ablkcipher.reqsize = sizeof(struct hifn_request_context);

  	return 0;
  }
  
  static int hifn_alg_alloc(struct hifn_device *dev, struct hifn_alg_template *t)
  {
  	struct hifn_crypto_alg *alg;
  	int err;

  	alg = kzalloc(sizeof(*alg), GFP_KERNEL);

  	if (!alg)
  		return -ENOMEM;
  
  	snprintf(alg->alg.cra_name, CRYPTO_MAX_ALG_NAME, "%s", t->name);

  	snprintf(alg->alg.cra_driver_name, CRYPTO_MAX_ALG_NAME, "%s-%s",
  		 t->drv_name, dev->name);

  
  	alg->alg.cra_priority = 300;

  	alg->alg.cra_flags = CRYPTO_ALG_TYPE_ABLKCIPHER |
  				CRYPTO_ALG_KERN_DRIVER_ONLY | CRYPTO_ALG_ASYNC;

  	alg->alg.cra_blocksize = t->bsize;
  	alg->alg.cra_ctxsize = sizeof(struct hifn_context);

  	alg->alg.cra_alignmask = 0;

  	alg->alg.cra_type = &crypto_ablkcipher_type;
  	alg->alg.cra_module = THIS_MODULE;
  	alg->alg.cra_u.ablkcipher = t->ablkcipher;
  	alg->alg.cra_init = hifn_cra_init;
  
  	alg->dev = dev;
  
  	list_add_tail(&alg->entry, &dev->alg_list);
  
  	err = crypto_register_alg(&alg->alg);
  	if (err) {
  		list_del(&alg->entry);
  		kfree(alg);
  	}
  
  	return err;
  }
  
  static void hifn_unregister_alg(struct hifn_device *dev)
  {
  	struct hifn_crypto_alg *a, *n;
  
  	list_for_each_entry_safe(a, n, &dev->alg_list, entry) {
  		list_del(&a->entry);
  		crypto_unregister_alg(&a->alg);
  		kfree(a);
  	}
  }
  
  static int hifn_register_alg(struct hifn_device *dev)
  {
  	int i, err;

  	for (i = 0; i < ARRAY_SIZE(hifn_alg_templates); ++i) {

  		err = hifn_alg_alloc(dev, &hifn_alg_templates[i]);
  		if (err)
  			goto err_out_exit;
  	}
  
  	return 0;
  
  err_out_exit:
  	hifn_unregister_alg(dev);
  	return err;
  }

  static void hifn_tasklet_callback(unsigned long data)
  {
  	struct hifn_device *dev = (struct hifn_device *)data;
  
  	/*
  	 * This is ok to call this without lock being held,
  	 * althogh it modifies some parameters used in parallel,
  	 * (like dev->success), but they are used in process
  	 * context or update is atomic (like setting dev->sa[i] to NULL).
  	 */

  	hifn_clear_rings(dev, 0);

  
  	if (dev->started < HIFN_QUEUE_LENGTH &&	dev->queue.qlen)
  		hifn_process_queue(dev);

  }

  static int hifn_probe(struct pci_dev *pdev, const struct pci_device_id *id)

  {
  	int err, i;
  	struct hifn_device *dev;
  	char name[8];
  
  	err = pci_enable_device(pdev);
  	if (err)
  		return err;
  	pci_set_master(pdev);

  	err = pci_set_dma_mask(pdev, DMA_BIT_MASK(32));

  	if (err)
  		goto err_out_disable_pci_device;
  
  	snprintf(name, sizeof(name), "hifn%d",

  			atomic_inc_return(&hifn_dev_number) - 1);

  
  	err = pci_request_regions(pdev, name);
  	if (err)
  		goto err_out_disable_pci_device;
  
  	if (pci_resource_len(pdev, 0) < HIFN_BAR0_SIZE ||
  	    pci_resource_len(pdev, 1) < HIFN_BAR1_SIZE ||
  	    pci_resource_len(pdev, 2) < HIFN_BAR2_SIZE) {

  		dev_err(&pdev->dev, "Broken hardware - I/O regions are too small.
  ");

  		err = -ENODEV;
  		goto err_out_free_regions;
  	}
  
  	dev = kzalloc(sizeof(struct hifn_device) + sizeof(struct crypto_alg),
  			GFP_KERNEL);
  	if (!dev) {
  		err = -ENOMEM;
  		goto err_out_free_regions;
  	}
  
  	INIT_LIST_HEAD(&dev->alg_list);
  
  	snprintf(dev->name, sizeof(dev->name), "%s", name);
  	spin_lock_init(&dev->lock);

  	for (i = 0; i < 3; ++i) {

  		unsigned long addr, size;
  
  		addr = pci_resource_start(pdev, i);
  		size = pci_resource_len(pdev, i);
  
  		dev->bar[i] = ioremap_nocache(addr, size);

  		if (!dev->bar[i]) {
  			err = -ENOMEM;

  			goto err_out_unmap_bars;

  		}

  	}

  	dev->desc_virt = pci_zalloc_consistent(pdev, sizeof(struct hifn_dma),
  					       &dev->desc_dma);

  	if (!dev->desc_virt) {

  		dev_err(&pdev->dev, "Failed to allocate descriptor rings.
  ");

  		err = -ENOMEM;

  		goto err_out_unmap_bars;

  	}

  
  	dev->pdev = pdev;
  	dev->irq = pdev->irq;

  	for (i = 0; i < HIFN_D_RES_RSIZE; ++i)

  		dev->sa[i] = NULL;
  
  	pci_set_drvdata(pdev, dev);

  	tasklet_init(&dev->tasklet, hifn_tasklet_callback, (unsigned long)dev);

  	crypto_init_queue(&dev->queue, 1);
  
  	err = request_irq(dev->irq, hifn_interrupt, IRQF_SHARED, dev->name, dev);
  	if (err) {

  		dev_err(&pdev->dev, "Failed to request IRQ%d: err: %d.
  ",
  			dev->irq, err);

  		dev->irq = 0;
  		goto err_out_free_desc;
  	}
  
  	err = hifn_start_device(dev);
  	if (err)
  		goto err_out_free_irq;

  	err = hifn_register_rng(dev);

  	if (err)
  		goto err_out_stop_device;

  	err = hifn_register_alg(dev);
  	if (err)
  		goto err_out_unregister_rng;

  	INIT_DELAYED_WORK(&dev->work, hifn_work);
  	schedule_delayed_work(&dev->work, HZ);

  	dev_dbg(&pdev->dev, "HIFN crypto accelerator card at %s has been "
  		"successfully registered as %s.
  ",
  		pci_name(pdev), dev->name);

  
  	return 0;

  err_out_unregister_rng:
  	hifn_unregister_rng(dev);

  err_out_stop_device:
  	hifn_reset_dma(dev, 1);
  	hifn_stop_device(dev);
  err_out_free_irq:

  	free_irq(dev->irq, dev);

  	tasklet_kill(&dev->tasklet);

  err_out_free_desc:
  	pci_free_consistent(pdev, sizeof(struct hifn_dma),
  			dev->desc_virt, dev->desc_dma);

  err_out_unmap_bars:

  	for (i = 0; i < 3; ++i)

  		if (dev->bar[i])
  			iounmap(dev->bar[i]);
  
  err_out_free_regions:
  	pci_release_regions(pdev);
  
  err_out_disable_pci_device:
  	pci_disable_device(pdev);
  
  	return err;
  }

  static void hifn_remove(struct pci_dev *pdev)

  {
  	int i;
  	struct hifn_device *dev;
  
  	dev = pci_get_drvdata(pdev);
  
  	if (dev) {

  		cancel_delayed_work_sync(&dev->work);

  		hifn_unregister_rng(dev);

  		hifn_unregister_alg(dev);
  		hifn_reset_dma(dev, 1);
  		hifn_stop_device(dev);

  		free_irq(dev->irq, dev);

  		tasklet_kill(&dev->tasklet);

  
  		hifn_flush(dev);
  
  		pci_free_consistent(pdev, sizeof(struct hifn_dma),
  				dev->desc_virt, dev->desc_dma);

  		for (i = 0; i < 3; ++i)

  			if (dev->bar[i])
  				iounmap(dev->bar[i]);
  
  		kfree(dev);
  	}
  
  	pci_release_regions(pdev);
  	pci_disable_device(pdev);
  }
  
  static struct pci_device_id hifn_pci_tbl[] = {
  	{ PCI_DEVICE(PCI_VENDOR_ID_HIFN, PCI_DEVICE_ID_HIFN_7955) },
  	{ PCI_DEVICE(PCI_VENDOR_ID_HIFN, PCI_DEVICE_ID_HIFN_7956) },
  	{ 0 }
  };
  MODULE_DEVICE_TABLE(pci, hifn_pci_tbl);
  
  static struct pci_driver hifn_pci_driver = {
  	.name     = "hifn795x",
  	.id_table = hifn_pci_tbl,
  	.probe    = hifn_probe,

  	.remove   = hifn_remove,

  };

  static int __init hifn_init(void)

  {

  	unsigned int freq;

  	int err;

  	/* HIFN supports only 32-bit addresses */
  	BUILD_BUG_ON(sizeof(dma_addr_t) != 4);

  	if (strncmp(hifn_pll_ref, "ext", 3) &&
  	    strncmp(hifn_pll_ref, "pci", 3)) {

  		pr_err("hifn795x: invalid hifn_pll_ref clock, must be pci or ext");

  		return -EINVAL;
  	}
  
  	/*
  	 * For the 7955/7956 the reference clock frequency must be in the
  	 * range of 20MHz-100MHz. For the 7954 the upper bound is 66.67MHz,
  	 * but this chip is currently not supported.
  	 */
  	if (hifn_pll_ref[3] != '\0') {
  		freq = simple_strtoul(hifn_pll_ref + 3, NULL, 10);
  		if (freq < 20 || freq > 100) {

  			pr_err("hifn795x: invalid hifn_pll_ref frequency, must"
  			       "be in the range of 20-100");

  			return -EINVAL;
  		}
  	}

  	err = pci_register_driver(&hifn_pci_driver);
  	if (err < 0) {

  		pr_err("Failed to register PCI driver for %s device.
  ",
  		       hifn_pci_driver.name);

  		return -ENODEV;
  	}

  	pr_info("Driver for HIFN 795x crypto accelerator chip "
  		"has been successfully registered.
  ");

  
  	return 0;
  }

  static void __exit hifn_fini(void)

  {
  	pci_unregister_driver(&hifn_pci_driver);

  	pr_info("Driver for HIFN 795x crypto accelerator chip "
  		"has been successfully unregistered.
  ");

  }
  
  module_init(hifn_init);
  module_exit(hifn_fini);
  
  MODULE_LICENSE("GPL");
  MODULE_AUTHOR("Evgeniy Polyakov <johnpol@2ka.mipt.ru>");
  MODULE_DESCRIPTION("Driver for HIFN 795x crypto accelerator chip.");