mirror of
https://github.com/ilyakooo0/urbit.git
synced 2024-12-15 01:52:42 +03:00
ur: refactors ur_bsr_bytes_any(), fixes edge-case bugs
This commit is contained in:
parent
bb136b7981
commit
c123e9413a
@ -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;
|
||||
}
|
||||
|
@ -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;
|
||||
off = len_bit;
|
||||
|
||||
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,61 +396,71 @@ 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;
|
||||
l = byt & mask;
|
||||
out[i] = m ^ (l << rest);
|
||||
m = byt >> off;
|
||||
}
|
||||
}
|
||||
|
||||
for ( i = 0; i < max; i++ ) {
|
||||
byt = b[1ULL + i];
|
||||
l = byt & mask;
|
||||
out[i] = m ^ (l << rest);
|
||||
m = byt >> off;
|
||||
}
|
||||
// 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 ( end ) {
|
||||
if ( len_bit && len_bit < rest ) {
|
||||
out[max] = m & ((1 << len_bit) - 1);
|
||||
bsr->bytes += max;
|
||||
left -= max;
|
||||
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 bits = off + len_bit;
|
||||
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);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
bsr->off = off;
|
||||
bsr->left = left;
|
||||
bsr->bits += len;
|
||||
bsr->off = off;
|
||||
bsr->left = left;
|
||||
}
|
||||
}
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user