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
122
|
/* ISC license. */
#include <unistd.h>
#include <stdint.h>
#include <errno.h>
#include <skalibs/uint32.h>
#include <skalibs/diuint32.h>
#include <skalibs/buffer.h>
#include <skalibs/genalloc.h>
#include <skalibs/cdb.h>
#include <skalibs/cdb_make.h>
int cdb_make_start (struct cdb_make *c, int fd)
{
c->hplist = genalloc_zero ;
c->pos = 2048 ;
buffer_init(&c->b, &buffer_write, fd, c->buf, BUFFER_OUTSIZE) ;
if (lseek(fd, c->pos, SEEK_SET) < 0) return -1 ;
return 0 ;
}
static int posplus (struct cdb_make *c, uint32_t len)
{
uint32_t newpos = c->pos + len ;
if (newpos < len) return (errno = ENOMEM, 0) ;
c->pos = newpos ;
return 1 ;
}
static inline int cdb_make_addend (struct cdb_make *c, unsigned int keylen, unsigned int datalen, uint32_t h)
{
diuint32 blah = { .left = h, .right = c->pos } ;
return genalloc_append(diuint32, &c->hplist, &blah) && posplus(c, 8) && posplus(c, keylen) && posplus(c, datalen) ;
}
static inline ssize_t cdb_make_addbegin (struct cdb_make *c, unsigned int keylen, unsigned int datalen)
{
char buf[8] ;
uint32_pack(buf, (uint32_t)keylen) ;
uint32_pack(buf + 4, (uint32_t)datalen) ;
return buffer_put(&c->b, buf, 8) ;
}
int cdb_make_add (struct cdb_make *c, char const *key, unsigned int keylen, char const *data, unsigned int datalen)
{
if (cdb_make_addbegin(c, keylen, datalen) < 0
|| buffer_put(&c->b, key, keylen) < 0
|| buffer_put(&c->b, data, datalen) < 0
|| !cdb_make_addend(c, keylen, datalen, cdb_hash(key, keylen)))
{
genalloc_free(diuint32, &c->hplist) ;
return -1 ;
}
return 0 ;
}
int cdb_make_finish (struct cdb_make *c)
{
uint32_t count[256] ;
uint32_t start[256] ;
char final[2048] ;
unsigned int size = 1 ;
unsigned int n = genalloc_len(diuint32, &c->hplist) ;
unsigned int i = 0 ;
diuint32 *hp = genalloc_s(diuint32, &c->hplist) ;
for (; i < 256 ; i++) count[i] = 0 ;
for (i = 0 ; i < n ; i++) ++count[hp[i].left & 255] ;
{
uint32_t u = 0 ;
for (i = 0 ; i < 256 ; i++) start[i] = u += count[i] ; /* bounded by n */
for (i = 0 ; i < 256 ; i++)
{
u = count[i] << 1 ;
if (u > size) size = u ;
}
size += n ; /* no overflow possible up to now */
u = 0xffffffffUL ; u /= sizeof(diuint32) ;
if (size > u) return (errno = ENOMEM, -1) ;
}
i = n ;
{
diuint32 split[size] ;
while (i--) split[--start[hp[i].left & 255]] = hp[i] ;
genalloc_free(diuint32, &c->hplist) ;
hp = split + n ;
for (i = 0 ; i < 256 ; ++i)
{
char buf[8] ;
uint32_t k = count[i] ;
uint32_t len = k << 1 ; /* no overflow possible */
diuint32 *p = split + start[i] ;
unsigned int j = 0 ;
uint32_pack(final + (i << 3), c->pos) ;
uint32_pack(final + (i << 3) + 4, len) ;
for (; j < len ; j++) hp[j].left = hp[j].right = 0 ;
for (j = 0 ; j < k ; j++)
{
uint32_t where = (p->left >> 8) % len ;
while (hp[where].right) if (++where == len) where = 0 ;
hp[where] = *p++ ;
}
for (j = 0 ; j < len ; j++)
{
uint32_pack(buf, hp[j].left) ;
uint32_pack(buf + 4, hp[j].right) ;
if (buffer_put(&c->b, buf, 8) < 0) return -1 ;
if (!posplus(c, 8)) return -1 ;
}
}
}
if (!buffer_flush(&c->b)) return -1 ;
if (lseek(buffer_fd(&c->b), 0, SEEK_SET) < 0) return -1 ;
if (buffer_putflush(&c->b, final, 2048) < 0) return -1 ;
return 0 ;
}
|
