-
Notifications
You must be signed in to change notification settings - Fork 587
/
Base32.h
149 lines (132 loc) · 4.76 KB
/
Base32.h
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
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
/* vim: set expandtab ts=4 sw=4: */
/*
* You may redistribute this program and/or modify it under the terms of
* the GNU General Public License as published by the Free Software Foundation,
* either version 3 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, see <http://www.gnu.org/licenses/>.
*/
#ifndef Base32_H
#define Base32_H
#include <stdint.h>
/*
* Base32 encode or decode a byte array using the format defined in:
* http://dnscurve.org/in-implement.html
*/
/** Returned by Base32_decode() if the input is not valid base32. */
#define Base32_BAD_INPUT -1
/** Returned by Base32_decode() or Base32_encode() if the output buffer is too small. */
#define Base32_TOO_BIG -2
/**
* Decode a base32 encoded number.
*
* @param output the place to put the decoded bytes.
* This may be modified even if there is a decoding error.
* @param outLength the length of the output array, if the decoded output is longer,
* Base32_TOO_BIG will be returned.
* @param in the buffer holding the base32 encoded number.
* @param inputLength the length of the input buffer.
* @return the length of the output if all goes well, Base32_BAD_INPUT if the input
* is not valid base32, or Base32_TOO_BIG if the output buffer is not large
* enough to handle the output.
*/
static inline int Base32_decode(uint8_t* output,
const uint32_t outLength,
const uint8_t* in,
const uint32_t inputLength)
{
// Maps ascii character inputs to the numbers
// Invalid characters are represented by 99
static const uint8_t numForAscii[] =
{
99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,
99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,
99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,
0, 1, 2, 3, 4, 5, 6, 7, 8, 9,99,99,99,99,99,99,
99,99,10,11,12,99,13,14,15,99,16,17,18,19,20,99,
21,22,23,24,25,26,27,28,29,30,31,99,99,99,99,99,
99,99,10,11,12,99,13,14,15,99,16,17,18,19,20,99,
21,22,23,24,25,26,27,28,29,30,31,99,99,99,99,99
};
uint32_t outIndex = 0;
uint32_t inputIndex = 0;
uint32_t nextByte = 0;
uint32_t bits = 0;
while (inputIndex < inputLength) {
if (in[inputIndex] & 0x80) {
return Base32_BAD_INPUT;
}
const uint8_t b = numForAscii[in[inputIndex++]];
if (b > 31) {
return Base32_BAD_INPUT;
}
nextByte |= ((unsigned) b) << bits;
bits += 5;
if (bits >= 8) {
if (outIndex >= outLength) {
return Base32_TOO_BIG;
}
output[outIndex++] = nextByte;
bits -= 8;
nextByte >>= 8;
}
}
if (bits >= 5 || nextByte) {
return Base32_BAD_INPUT;
}
return outIndex;
}
/**
* Base32 encode a number.
*
* @param output the place to put the base32 encoded output.
* This may be modified even if there is a encoding error.
* @param outLength the length of the output array, if the encoded output is longer,
* Base32_TOO_BIG will be returned.
* @param in the buffer holding the bytes to encode.
* @param inputLength the length of the input buffer.
* @return the length of the output if all goes well,
* or Base32_TOO_BIG if the output buffer is not large enough to handle the output.
*/
static inline int Base32_encode(uint8_t* output,
const uint32_t outputLength,
const uint8_t* in,
const uint32_t inputLength)
{
uint32_t outIndex = 0;
uint32_t inIndex = 0;
uint32_t work = 0;
uint32_t bits = 0;
static const uint8_t* kChars = (uint8_t*) "0123456789bcdfghjklmnpqrstuvwxyz";
while (inIndex < inputLength) {
work |= ((unsigned) in[inIndex++]) << bits;
bits += 8;
while (bits >= 5) {
if (outIndex >= outputLength) {
return Base32_TOO_BIG;
}
output[outIndex++] = kChars[work & 31];
bits -= 5;
work >>= 5;
}
}
if (bits) {
if (outIndex >= outputLength) {
return Base32_TOO_BIG;
}
output[outIndex++] = kChars[work & 31];
bits -= 5;
work >>= 5;
}
if (outIndex < outputLength) {
output[outIndex] = '\0';
}
return outIndex;
}
#endif