You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
107 lines
1.8 KiB
107 lines
1.8 KiB
2 months ago
|
/*
|
||
|
|
* UCW Library: Reading and writing Varints on Fastbuf Streams
|
||
|
|
*
|
||
|
|
* (c) 2013 Tomas Valla <tom@ucw.cz>
|
||
|
|
*
|
||
|
|
* This software may be freely distributed and used according to the terms
|
||
|
|
* of the GNU Lesser General Public License.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include <ucw/lib.h>
|
||
|
|
#include <ucw/fastbuf.h>
|
||
|
|
#include <ucw/varint.h>
|
||
|
|
#include <ucw/ff-varint.h>
|
||
|
|
#include <ucw/ff-binary.h>
|
||
|
|
|
||
|
|
|
||
|
|
u64 bget_varint_slow(struct fastbuf *b, u64 repl)
|
||
|
|
{
|
||
|
|
uint h = bgetc(b);
|
||
|
|
uint l = varint_len(h);
|
||
|
|
byte buf[l];
|
||
|
|
buf[0] = h;
|
||
|
|
l--;
|
||
|
|
if (breadb(b, buf+1, l) < l)
|
||
|
|
return repl;
|
||
|
|
varint_get(buf, &repl);
|
||
|
|
return repl;
|
||
|
|
}
|
||
|
|
|
||
|
|
void bput_varint_slow(struct fastbuf *b, u64 u)
|
||
|
|
{
|
||
|
|
byte buf[9];
|
||
|
|
uint l = varint_put(buf, u);
|
||
|
|
bwrite(b, buf, l);
|
||
|
|
}
|
||
|
|
|
||
|
|
#ifdef TEST
|
||
|
|
|
||
|
|
#include <stdlib.h>
|
||
|
|
#include <stdio.h>
|
||
|
|
|
||
|
|
int main(int argc, char **argv)
|
||
|
|
{
|
||
|
|
#define FUNCS \
|
||
|
|
F(BGET_VARINT) F(BPUT_VARINT)
|
||
|
|
|
||
|
|
enum {
|
||
|
|
#define F(x) FUNC_##x,
|
||
|
|
FUNCS
|
||
|
|
#undef F
|
||
|
|
};
|
||
|
|
char *names[] = {
|
||
|
|
#define F(x) [FUNC_##x] = #x,
|
||
|
|
FUNCS
|
||
|
|
#undef F
|
||
|
|
};
|
||
|
|
|
||
|
|
uint func = ~0U;
|
||
|
|
if (argc > 1)
|
||
|
|
for (uint i = 0; i < ARRAY_SIZE(names); i++)
|
||
|
|
if (!strcasecmp(names[i], argv[1]))
|
||
|
|
func = i;
|
||
|
|
if (!~func) {
|
||
|
|
fprintf(stderr, "Invalid usage!\n");
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
struct fastbuf *b = fbgrow_create(8);
|
||
|
|
switch (func) {
|
||
|
|
uint u;
|
||
|
|
uintmax_t r;
|
||
|
|
int i;
|
||
|
|
case FUNC_BGET_VARINT:
|
||
|
|
while (scanf("%x", &u) == 1)
|
||
|
|
bputc(b, u);
|
||
|
|
fbgrow_rewind(b);
|
||
|
|
while (bpeekc(b) >= 0) {
|
||
|
|
if (btell(b))
|
||
|
|
putchar(' ');
|
||
|
|
r = bget_varint_slow(b, ~0LLU);
|
||
|
|
printf("%jx", r);
|
||
|
|
}
|
||
|
|
putchar('\n');
|
||
|
|
break;
|
||
|
|
|
||
|
|
case FUNC_BPUT_VARINT:
|
||
|
|
i = 0;
|
||
|
|
while (scanf("%jx", &r) == 1)
|
||
|
|
bput_varint_slow(b, r);
|
||
|
|
fbgrow_rewind(b);
|
||
|
|
while (bpeekc(b) >= 0) {
|
||
|
|
if (i++)
|
||
|
|
putchar(' ');
|
||
|
|
printf("%02x", bgetc(b));
|
||
|
|
}
|
||
|
|
putchar('\n');
|
||
|
|
break;
|
||
|
|
default:
|
||
|
|
ASSERT(0);
|
||
|
|
}
|
||
|
|
|
||
|
|
bclose(b);
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
#endif
|