#ifndef _LINUX_JHASH_H
#define _LINUX_JHASH_H

#include <stdint.h>

/**
 * rol32 - rotate a 32-bit value left
 * @word: value to rotate
 * @shift: bits to roll
 */
static inline uint32_t rol32(uint32_t word, unsigned int shift)
{
	return (word << shift) | (word >> (32 - shift));
}

/**
 * ror32 - rotate a 32-bit value right
 * @word: value to rotate
 * @shift: bits to roll
 */
static inline uint32_t ror32(uint32_t word, unsigned int shift)
{
	return (word >> shift) | (word << (32 - shift));
}

/* jhash.h: Jenkins hash support.
 *
 * Copyright (C) 2006. Bob Jenkins (bob_jenkins@burtleburtle.net)
 *
 * http://burtleburtle.net/bob/hash/
 *
 * These are the credits from Bob's sources:
 *
 * lookup3.c, by Bob Jenkins, May 2006, Public Domain.
 *
 * These are functions for producing 32-bit hashes for hash table lookup.
 * hashword(), hashlittle(), hashlittle2(), hashbig(), mix(), and final() 
 * are externally useful functions.  Routines to test the hash are included 
 * if SELF_TEST is defined.  You can use this free for any purpose.  It's in
 * the public domain.  It has no warranty.
 *
 * Copyright (C) 2009 Jozsef Kadlecsik (kadlec@blackhole.kfki.hu)
 *
 * I've modified Bob's hash to be useful in the Linux kernel, and
 * any bugs present are my fault.  Jozsef
 */

/* __jhash_mix - mix 3 32-bit values reversibly. */
#define __jhash_mix(a,b,c) \
{ \
  a -= c;  a ^= rol32(c, 4);  c += b; \
  b -= a;  b ^= rol32(a, 6);  a += c; \
  c -= b;  c ^= rol32(b, 8);  b += a; \
  a -= c;  a ^= rol32(c,16);  c += b; \
  b -= a;  b ^= rol32(a,19);  a += c; \
  c -= b;  c ^= rol32(b, 4);  b += a; \
}

/* __jhash_final - final mixing of 3 32-bit values (a,b,c) into c */
#define __jhash_final(a,b,c) \
{ \
  c ^= b; c -= rol32(b,14); \
  a ^= c; a -= rol32(c,11); \
  b ^= a; b -= rol32(a,25); \
  c ^= b; c -= rol32(b,16); \
  a ^= c; a -= rol32(c,4);  \
  b ^= a; b -= rol32(a,14); \
  c ^= b; c -= rol32(b,24); \
}

#define QUARTERROUND(__a,__b,__c,__d) \
  __a += __b; __d = rol32(__d ^ __a, 16); \
  __c += __d; __b = rol32(__b ^ __c, 12); \
  __a += __b; __d = rol32(__d ^ __a,  8); \
  __c += __d; __b = rol32(__b ^ __c,  7);

/* __jhash_final4 - final mixing of 4 32-bit values (a,b,c,d) into d */
#define __jhash_final4(a,b,c,d) \
{ \
  QUARTERROUND(a,b,c,d) \
  QUARTERROUND(b,c,d,a) \
  QUARTERROUND(c,d,a,b) \
  QUARTERROUND(d,a,b,c) \
  a += b; d = rol32(d ^ a, 9); \
}

/* An arbitrary initial parameter */
#define JHASH_INIT_PARAM	0xDEADBEEF

/* The most generic version, hashes an arbitrary sequence
 * of bytes.  No alignment or length assumptions are made about
 * the input key. The result depends on endianness.
 */
static inline uint32_t jhash(const void *key, uint32_t length, uint32_t initval)
{
	uint32_t a,b,c;
	const uint8_t *k = key;

	/* Set up the internal state */
	a = b = c = JHASH_INIT_PARAM + length + initval;

	/* all but the last block: affect some 32 bits of (a,b,c) */
	while (length > 12) {
    		a += (k[0] + ((uint32_t)k[1]<<8) + ((uint32_t)k[2]<<16) + ((uint32_t)k[3]<<24));
		b += (k[4] + ((uint32_t)k[5]<<8) + ((uint32_t)k[6]<<16) + ((uint32_t)k[7]<<24));
		c += (k[8] + ((uint32_t)k[9]<<8) + ((uint32_t)k[10]<<16) + ((uint32_t)k[11]<<24));
		__jhash_mix(a, b, c);
		length -= 12;
		k += 12;
	}

	/* last block: affect all 32 bits of (c) */
	/* all the case statements fall through */
	switch (length) {
	case 12: c += (uint32_t)k[11]<<24;
	case 11: c += (uint32_t)k[10]<<16;
	case 10: c += (uint32_t)k[9]<<8;
	case 9 : c += k[8];
	case 8 : b += (uint32_t)k[7]<<24;
	case 7 : b += (uint32_t)k[6]<<16;
	case 6 : b += (uint32_t)k[5]<<8;
	case 5 : b += k[4];
	case 4 : a += (uint32_t)k[3]<<24;
	case 3 : a += (uint32_t)k[2]<<16;
	case 2 : a += (uint32_t)k[1]<<8;
	case 1 : a += k[0];
		__jhash_final(a, b, c);
	case 0 :
		break;
	}

	return c;
}

/* A special optimized version that handles 1 or more of u32s.
 * The length parameter here is the number of u32s in the key.
 */
static inline uint32_t jhash2(const uint32_t *k, uint32_t length, uint32_t initval)
{
	uint32_t a, b, c;

	/* Set up the internal state */
	a = b = c = JHASH_INIT_PARAM + (length<<2) + initval;

	/* handle most of the key */
	while (length > 3) {
		a += k[0];
		b += k[1];
		c += k[2];
		__jhash_mix(a, b, c);
		length -= 3;
		k += 3;
	}

	/* handle the last 3 uint32_t's */
	/* all the case statements fall through */ 
	switch (length) {
	case 3: c += k[2];
	case 2: b += k[1];
	case 1: a += k[0];
		__jhash_final(a, b, c);
	case 0:     /* case 0: nothing left to add */
		break;
	}

	return c;
}

/* A special ultra-optimized versions that knows they are hashing exactly
 * 4, 3, 2 or 1 word(s).
 */
static inline uint32_t jhash_4words(uint32_t a, uint32_t b, uint32_t c, uint32_t d, uint32_t initval)
{
	a += JHASH_INIT_PARAM + rol32(initval, 3);
	b += JHASH_INIT_PARAM + rol32(initval, 7);
	c += JHASH_INIT_PARAM + rol32(initval, 11);
	d += JHASH_INIT_PARAM + rol32(initval, 13);

	__jhash_final4(a, b, c, d);

	return d;
}

static inline uint32_t jhash_3words(uint32_t a, uint32_t b, uint32_t c, uint32_t initval)
{
	a += JHASH_INIT_PARAM + rol32(initval, 7);
	b += JHASH_INIT_PARAM + rol32(initval, 11);
	c += JHASH_INIT_PARAM + rol32(initval, 13);

	__jhash_final(a, b, c);

	return c;
}

static inline uint32_t jhash_2words(uint32_t a, uint32_t b, uint32_t initval)
{
	return jhash_3words(0, a, b, initval);
}

static inline uint32_t jhash_1word(uint32_t a, uint32_t initval)
{
	return jhash_3words(0, 0, a, initval);
}

#endif /* _LINUX_JHASH_H */