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
|
/* Copyright (C) 2011 Vincent Penquerc'h.
This file is part of the Kate codec library.
Written by Vincent Penquerc'h.
Use, distribution and reproduction of this library is governed
by a BSD style source license included with this source in the
file 'COPYING'. Please read these terms before distributing. */
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "kate/kate.h"
#include "../src/kate_bitwise.h"
static kate_uint32_t getrnd(kate_uint32_t *rnd)
{
return *rnd=*rnd*1103515245+12345;
}
static void test_bitwise_size(size_t bits,kate_uint32_t start_rnd)
{
kate_pack_buffer kpb;
size_t data_size=bits*4; /* good margin */
unsigned char *data=alloca(data_size);
size_t n,d;
kate_uint32_t rnd;
int read;
rnd=start_rnd;
d=0;
kate_pack_writeinit(&kpb);
for (n=0;n<bits;) {
kate_uint32_t b=1+getrnd(&rnd)%32;
kate_uint32_t r=getrnd(&rnd);
if (d>=data_size) {
printf("Warning: cut short\n");
break;
}
if (b>bits-n) b=bits-n;
r=r&((1<<b)-1);
data[d++]=r;
kate_pack_write(&kpb,r,b);
n+=b;
}
if (kate_pack_bits(&kpb)!=bits) {
fprintf(stderr,"Expected %zu bits, got %ld\n",bits,kate_pack_bits(&kpb));
exit(1);
}
/* do not reset, reuse allocated buffer as is */
kate_pack_readinit(&kpb,kpb.buffer,kate_pack_bytes(&kpb));
d=0;
rnd=start_rnd;
for (n=0;n<bits;) {
kate_uint32_t b=1+getrnd(&rnd)%32;
kate_uint32_t r=getrnd(&rnd);
if (b>bits-n) b=bits-n;
r=r&((1<<b)-1);
if (d>=data_size) break;
read=kate_pack_read(&kpb,b);
if (r!=read) {
fprintf(stderr,"Read %d from %u bits at bit offset %zu, expected %u\n",read,b,n,r);
exit(1);
}
n+=b;
}
kate_pack_writeclear(&kpb);
}
int main()
{
size_t n,r;
for (n=0;n<4096;++n) {
for (r=0;r<16;++r) {
test_bitwise_size(n,r);
}
}
return 0;
}
|
