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
115
116
117
118
119
120
121
|
# Copyright (C) 2009 Canonical Ltd
#
# 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.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
cdef extern from *:
void sprintf(char *, char *, ...)
cdef extern from "Python.h":
struct _PyObject:
pass
ctypedef _PyObject PyObject
int PyTuple_CheckExact(object p)
Py_ssize_t PyTuple_GET_SIZE(object t)
char *PyString_AS_STRING(object s)
PyObject * PyTuple_GET_ITEM_ptr "PyTuple_GET_ITEM" (object t,
Py_ssize_t offset)
int PyString_CheckExact_ptr "PyString_CheckExact" (PyObject *p)
Py_ssize_t PyString_GET_SIZE_ptr "PyTuple_GET_SIZE" (PyObject *s)
char *PyString_AS_STRING_ptr "PyString_AS_STRING" (PyObject *s)
object PyString_FromStringAndSize(char*, Py_ssize_t)
cdef extern from "zlib.h":
ctypedef unsigned long uLong
ctypedef unsigned int uInt
ctypedef unsigned char Bytef
uLong crc32(uLong crc, Bytef *buf, uInt len)
def _search_key_16(key):
"""See chk_map._search_key_16."""
cdef Py_ssize_t num_bits
cdef Py_ssize_t i, j
cdef Py_ssize_t num_out_bytes
cdef Bytef *c_bit
cdef uLong c_len
cdef uInt crc_val
cdef Py_ssize_t out_off
cdef char *c_out
cdef PyObject *bit
if not PyTuple_CheckExact(key):
raise TypeError('key %r is not a tuple' % (key,))
num_bits = PyTuple_GET_SIZE(key)
# 4 bytes per crc32, and another 1 byte between bits
num_out_bytes = (9 * num_bits) - 1
out = PyString_FromStringAndSize(NULL, num_out_bytes)
c_out = PyString_AS_STRING(out)
for i from 0 <= i < num_bits:
if i > 0:
c_out[0] = c'\x00'
c_out = c_out + 1
# We use the _ptr variant, because GET_ITEM returns a borrowed
# reference, and Pyrex assumes that returned 'object' are a new
# reference
bit = PyTuple_GET_ITEM_ptr(key, i)
if not PyString_CheckExact_ptr(bit):
raise TypeError('Bit %d of %r is not a string' % (i, key))
c_bit = <Bytef *>PyString_AS_STRING_ptr(bit)
c_len = PyString_GET_SIZE_ptr(bit)
crc_val = crc32(0, c_bit, c_len)
# Hex(val) order
sprintf(c_out, '%08X', crc_val)
c_out = c_out + 8
return out
def _search_key_255(key):
"""See chk_map._search_key_255."""
cdef Py_ssize_t num_bits
cdef Py_ssize_t i, j
cdef Py_ssize_t num_out_bytes
cdef Bytef *c_bit
cdef uLong c_len
cdef uInt crc_val
cdef Py_ssize_t out_off
cdef char *c_out
cdef PyObject *bit
if not PyTuple_CheckExact(key):
raise TypeError('key %r is not a tuple' % (key,))
num_bits = PyTuple_GET_SIZE(key)
# 4 bytes per crc32, and another 1 byte between bits
num_out_bytes = (5 * num_bits) - 1
out = PyString_FromStringAndSize(NULL, num_out_bytes)
c_out = PyString_AS_STRING(out)
for i from 0 <= i < num_bits:
if i > 0:
c_out[0] = c'\x00'
c_out = c_out + 1
bit = PyTuple_GET_ITEM_ptr(key, i)
if not PyString_CheckExact_ptr(bit):
raise TypeError('Bit %d of %r is not a string: %r' % (i, key,
<object>bit))
c_bit = <Bytef *>PyString_AS_STRING_ptr(bit)
c_len = PyString_GET_SIZE_ptr(bit)
crc_val = crc32(0, c_bit, c_len)
# MSB order
c_out[0] = (crc_val >> 24) & 0xFF
c_out[1] = (crc_val >> 16) & 0xFF
c_out[2] = (crc_val >> 8) & 0xFF
c_out[3] = (crc_val >> 0) & 0xFF
for j from 0 <= j < 4:
if c_out[j] == c'\n':
c_out[j] = c'_'
c_out = c_out + 4
return out
|