ilyakooo0/urbit
1
0
Fork 0
mirror of https://github.com/ilyakooo0/urbit.git synced 2024-12-15 01:52:42 +03:00
Code Issues Packages Projects Releases Wiki Activity

ur: refactors ur_bsr_bytes_any(), fixes edge-case bugs

Browse Source
This commit is contained in:
Joe Bryan 2020-08-31 15:56:55 -07:00
parent bb136b7981
commit c123e9413a
2 changed files with 49 additions and 44 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pkg/urbit/tests/ur_tests.c
View File

@ -1184,6 +1184,8 @@ _test_bsr_bytes_any_loop(const char *cap, uint8_t len, uint8_t val)
}
free(bytes);
free(d);
free(c);
return ret;
}

75
pkg/urbit/ur/bitstream.c
View File

@ -360,6 +360,8 @@ ur_bsr_bytes_any(ur_bsr_t *bsr, uint64_t len, uint8_t *out)
{
uint64_t left = bsr->left;
bsr->bits += len;
if ( !left ) {
return;
}
@ -368,33 +370,24 @@ ur_bsr_bytes_any(ur_bsr_t *bsr, uint64_t len, uint8_t *out)
uint8_t off = bsr->off;
uint64_t len_byt = len >> 3;
uint8_t len_bit = ur_mask_3(len);
uint64_t need = len_byt + !!len_bit;
if ( !off ) {
uint8_t bits = off + len_bit;
uint64_t need = len_byt + (bits >> 3) + !!ur_mask_3(bits);
if ( need > left ) {
memcpy(out, b, left);
left = 0;
bsr->bytes = 0;
bsr->left = 0;
}
else {
memcpy(out, b, len_byt);
off = len_bit;
left -= len_byt;
if ( !left ) {
bsr->bytes = 0;
}
else {
bsr->bytes += len_byt;
}
bsr->left = left;
if ( off ) {
out[len_byt] = b[len_byt] & ((1 << off) - 1);
}
left -= len_byt;
bsr->bytes = ( left ) ? b + len_byt : 0;
}
}
// the most-significant bits from a byte in the stream
@ -403,53 +396,64 @@ ur_bsr_bytes_any(ur_bsr_t *bsr, uint64_t len, uint8_t *out)
else {
uint8_t rest = 8 - off;
uint8_t mask = (1 << off) - 1;
uint8_t byt = b[0];
uint8_t l, m = byt >> off;
ur_bool_t end;
uint64_t i, max;
uint8_t byt, l, m = *b >> off;
uint64_t last = left - 1;
// loop over all the bytes we need (or all that remain)
//
// [l] holds [off] bits
// [m] holds [rest] bits
//
{
uint64_t need = len_byt + !!len_bit;
end = need >= left;
max = end ? (left - 1) : len_byt;
}
uint64_t max = ur_min(last, len_byt);
uint64_t i;
for ( i = 0; i < max; i++ ) {
byt = b[1ULL + i];
byt = *++b;
l = byt & mask;
out[i] = m ^ (l << rest);
m = byt >> off;
}
}
if ( end ) {
if ( len_bit && len_bit < rest ) {
out[max] = m & ((1 << len_bit) - 1);
bsr->bytes += max;
left -= max;
// we're reading into or beyond the last byte [bsr]
//
// [m] holds all the remaining bits in [bsr],
// but we might not need all of it
//
if ( need >= left ) {
uint8_t bits = len - (last << 3);
if ( bits < rest ) {
out[last] = m & ((1 << bits) - 1);
bsr->bytes = b;
left = 1;
off += len_bit;
}
else {
out[max] = m;
out[last] = m;
bsr->bytes = 0;
left = 0;
off = 0;
}
}
// we need less than a byte, but it might span multiple bytes
//
else {
uint8_t bits = off + len_bit;
uint64_t step = max + !!(bits >> 3);
uint8_t step = !!(bits >> 3);
bsr->bytes += step;
left -= step;
bsr->bytes = b + step;
left -= len_byt + step;
off = ur_mask_3(bits);
if ( len_bit ) {
if ( len_bit <= rest ) {
out[max] = m & ((1 << len_bit) - 1);
out[len_byt] = m & ((1 << len_bit) - 1);
}
else {
l = b[1ULL + max] & ((1 << off) - 1);;
out[max] = m ^ (l << rest);
l = *++b & ((1 << off) - 1);
out[len_byt] = m ^ (l << rest);
}
}
}
@ -457,7 +461,6 @@ ur_bsr_bytes_any(ur_bsr_t *bsr, uint64_t len, uint8_t *out)
bsr->off = off;
bsr->left = left;
bsr->bits += len;
}
}