You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
37 lines
987 B
C
37 lines
987 B
C
#include "parse.h"
|
|
|
|
void bs_init_membuf(struct bytestream* bs,const unsigned char* membuf,size_t len) {
|
|
bs->type = MEMBUF;
|
|
bs->cur = 0;
|
|
bs->max = len;
|
|
bs->u.base=membuf;
|
|
if ((const unsigned char*)(((uintptr_t)membuf) + len) < membuf) {
|
|
bs->cur = 1; // if the buffer + length is implausible, set error immediately
|
|
bs->max = 0;
|
|
}
|
|
}
|
|
|
|
#ifdef UNITTEST
|
|
#include <assert.h>
|
|
|
|
#undef UNITTEST
|
|
#include "buffer/bs_get.c"
|
|
#include "buffer/bs_err.c"
|
|
|
|
ssize_t buffer_getc(buffer* b,char* c) { return -1; }
|
|
|
|
int main() {
|
|
static struct bytestream bs;
|
|
bs_init_membuf(&bs, (const unsigned char*)"fnord\n", 6);
|
|
char buf[7];
|
|
int i;
|
|
for (i=0; i<7; ++i) buf[i]=bs_get(&bs);
|
|
assert(!memcmp(buf,"fnord\n",7)); // we should have gotten everything and then a 0 byte
|
|
assert(bs_err(&bs)); // that should have set the error flag
|
|
|
|
/* see if we catch an impossible buffer size */
|
|
bs_init_membuf(&bs, (const unsigned char*)"fnord\n", (size_t)-1);
|
|
assert(bs_err(&bs));
|
|
}
|
|
#endif
|