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
|
#include "bootstr.h"
#include <unistr.h>
#include <err.h>
#include <stdio.h>
#include <string.h>
#include <stdint.h>
#define CHUNKSIZE 4096
bool is_ascii(uint32_t c);
const struct bootstr_cfg puny_cfg = {
.base = U"abcdefghijklmnopqrstuvwxyz0123456789",
.baselen = 36,
.delim = U"-",
.is_basic = is_ascii,
.tmin = 1,
.tmax = 26,
.skew = 38,
.damp = 700,
.initial_bias = 72,
.initial_n = 128
};
bool
is_ascii(uint32_t c)
{
return c < 128;
}
uint8_t *
readall(FILE *file, size_t *len)
{
ssize_t nread;
size_t cap;
uint8_t *data;
*len = 0;
cap = CHUNKSIZE + 1;
data = malloc(cap);
if (!data) err(1, "malloc");
while (1) {
if (*len + CHUNKSIZE + 1 > cap) {
cap *= 2;
data = realloc(data, cap);
if (!data) err(1, "realloc");
}
nread = fread(data + *len, 1, CHUNKSIZE, file);
if (nread <= 0) break;
*len += nread;
}
*(data + *len) = '\0';
return data;
}
int
main(int argc, const char **argv)
{
const char **arg;
uint8_t *in, *out;
uint32_t *u_in, *u_out;
size_t inlen, outlen;
size_t u_inlen, u_outlen;
const char *filepath;
bool encode;
char *tok;
FILE *file;
int ret;
encode = true;
filepath = NULL;
for (arg = argv + 1; *arg; arg++) {
if (!strcmp(*arg, "-e")) {
encode = true;
} else if (!strcmp(*arg, "-d")) {
encode = false;
} else if (!filepath) {
filepath = *arg;
} else {
errx(1, "unknown arg %s", *arg);
}
}
out = NULL;
if (filepath) {
file = fopen(filepath, "r");
if (!file) err(1, "fopen %s", filepath);
in = readall(file, &inlen);
fclose(file);
} else {
in = readall(stdin, &inlen);
}
tok = strchr((char *)in, '\n');
if (tok) *tok = '\0';
u_in = u8_to_u32(in, inlen + 1, NULL, &u_inlen);
u_out = NULL;
if (encode) {
ret = bootstr_encode(&puny_cfg, u_in, &u_out);
if (ret) errx(1, "encode: %s", strerror(ret));
} else {
ret = bootstr_decode(&puny_cfg, u_in, &u_out);
if (ret) errx(1, "decode: %s", strerror(ret));
}
out = u32_to_u8(u_out, u32_strlen(u_out) + 1, NULL, &outlen);
printf("%s\n", (char *)out);
free(u_out);
free(u_in);
free(out);
free(in);
}
|
