summaryrefslogtreecommitdiffstats
path: root/Tools/perfecthash/perfhash.c
blob: 6e8a8b545ba2a217a5eb650e6991a7ad57098a7d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
#include <Python.h>

static PyObject * hashFunction(PyObject *self, PyObject *args, PyObject *kw)
{
	PyStringObject *a;
	register int len;
	register unsigned char *p;
	register unsigned long x;
	unsigned long ulSeed;
	unsigned long cchSeed;
	unsigned long cHashElements;

	if (!PyArg_ParseTuple(args, "llOl:hash", 
			      &ulSeed, &cchSeed, &a, &cHashElements))
	  return NULL;
	if (!PyString_Check(a))
	{
		PyErr_SetString(PyExc_TypeError, "arg 3 needs to be a string");
		return NULL;
	}
	
	len = a->ob_size;
	p = (unsigned char *) a->ob_sval;
	x = ulSeed;
	while (--len >= 0)
	{
	    /* (1000003 * x) ^ *p++ 
  	     * translated to handle > 32 bit longs 
	     */
	    x = (0xf4243 * x);
	    x = x & 0xFFFFFFFF;
	    x = x ^ *p++;
	}
	x ^= a->ob_size + cchSeed;
	if (x == 0xFFFFFFFF)
	  x = 0xfffffffe;
	if (x & 0x80000000) 
	{
	      /* Emulate Python 32-bit signed (2's complement) 
	       * modulo operation 
	       */
	      x = (~x & 0xFFFFFFFF) + 1;
	      x %= cHashElements;
	      if (x != 0)
	      {
	          x = x + (~cHashElements & 0xFFFFFFFF) + 1;
	          x = (~x & 0xFFFFFFFF) + 1;
              }
	}
	else
	  x %= cHashElements;
	return PyInt_FromLong((long)x);
}

static PyObject * calcSeed(PyObject *self, PyObject *args, PyObject *kw)
{
	PyStringObject *a;
	register int len;
	register unsigned char *p;
	register unsigned long x;

	if (!PyString_Check(args))
	{
		PyErr_SetString(PyExc_TypeError, "arg 1 expected a string, but didn't get it.");
		return NULL;
	}

	a = (PyStringObject *)args;
	
	len = a->ob_size;
	p = (unsigned char *) a->ob_sval;
	x = (*p << 7) & 0xFFFFFFFF;
	while (--len >= 0)
	{
	    /* (1000003 * x) ^ *p++ 
  	     * translated to handle > 32 bit longs 
	     */
	    x = (0xf4243 * x);
	    x = x & 0xFFFFFFFF;
	    x = x ^ *p++;
	}
	return PyInt_FromLong((long)x);
}


static struct PyMethodDef hashMethods[] = {
  { "calcSeed", calcSeed, 0, NULL },
  { "hash", hashFunction, 0, NULL },
  { NULL, NULL, 0, NULL } /* sentinel */
};

#ifdef _MSC_VER
_declspec(dllexport)
#endif
void initperfhash(void)
{
        PyObject *m;

        m = Py_InitModule4("perfhash", hashMethods,
                                           NULL, NULL, PYTHON_API_VERSION);
        if ( m == NULL )
            Py_FatalError("can't initialize module perfhash");
}