This repository has been archived by the owner on Feb 24, 2020. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 887
/
decode_cgo.go
121 lines (97 loc) · 2.28 KB
/
decode_cgo.go
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 2014 The zappy Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
// Copyright 2011 The Snappy-Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the SNAPPY-GO-LICENSE file.
// +build cgo,!purego
package zappy
/*
#include <stdint.h>
#include <string.h>
// supports only uint32 encoded values
int uvarint(unsigned int* n, uint8_t* src, int len) {
int r = 0;
unsigned int v = 0;
unsigned int s = 0;
while ((len-- != 0) && (++r <= 5)) {
uint8_t b = *src++;
v = v | ((b&0x7f)<<s);
if (b < 0x80) {
*n = v;
return r;
}
s += 7;
}
return -1;
}
int varint(int* n, uint8_t* src, int len) {
unsigned int u;
int i = uvarint(&u, src, len);
int x = u>>1;
if ((u&1) != 0)
x = ~x;
*n = x;
return i;
}
int decode(int s, int len_src, uint8_t* src, int len_dst, uint8_t* dst) {
int d = 0;
int length;
while (s < len_src) {
int n, i = varint(&n, src+s, len_src-s);
if (i <= 0) {
return -1;
}
s += i;
if (n >= 0) {
length = n+1;
if ((length > len_dst-d) || (length > len_src-s))
return -1;
memcpy(dst+d, src+s, length);
d += length;
s += length;
continue;
}
length = -n;
int offset;
i = uvarint((unsigned int*)(&offset), src+s, len_src-s);
if (i <= 0)
return -1;
s += i;
if (s > len_src)
return -1;
int end = d+length;
if ((offset > d) || (end > len_dst))
return -1;
for( ; d < end; d++)
*(dst+d) = *(dst+d-offset);
}
return d;
}
*/
import "C"
func puregoDecode() bool { return false }
// Decode returns the decoded form of src. The returned slice may be a sub-
// slice of buf if buf was large enough to hold the entire decoded block.
// Otherwise, a newly allocated slice will be returned.
// It is valid to pass a nil buf.
func Decode(buf, src []byte) ([]byte, error) {
dLen, s, err := decodedLen(src)
if err != nil {
return nil, err
}
if dLen == 0 {
if len(src) == 1 {
return nil, nil
}
return nil, ErrCorrupt
}
if len(buf) < dLen {
buf = make([]byte, dLen)
}
d := int(C.decode(C.int(s), C.int(len(src)), (*C.uint8_t)(&src[0]), C.int(len(buf)), (*C.uint8_t)(&buf[0])))
if d != dLen {
return nil, ErrCorrupt
}
return buf[:d], nil
}