mirror of
https://github.com/trailofbits/buttercup
synced 2026-06-21 14:11:39 +00:00
2over12
c26a3e223d
* add program model uv * add tool to common * remove oss fuzz tool * add docker * implement build base image * fix ret check * add builder for subimage * remove old tool * hack to avoid dupped kzip files * fixes... need to work out LD hack * fix find * justfile for building kythe * update dockerfile * add upload tool * update protobuf * move into compose * fix merge invocation * fix up invoc * bump kythe * reformat * format * format * remove unused * add lint and change version * fixes * exclude * fix formatting: * dev deps * add dep * exclude gen * fix
62 lines
1.2 KiB
Python
62 lines
1.2 KiB
Python
"""Varint encoder/decoder
|
|
varint from https://github.com/fmoo/python-varint/blob/master/varint.py
|
|
"""
|
|
|
|
from io import BytesIO
|
|
|
|
|
|
import sys
|
|
|
|
if sys.version > "3":
|
|
|
|
def _byte(b):
|
|
return bytes((b,))
|
|
else:
|
|
|
|
def _byte(b):
|
|
return chr(b)
|
|
|
|
|
|
def encode(number):
|
|
"""Pack `number` into varint bytes"""
|
|
buf = b""
|
|
while True:
|
|
towrite = number & 0x7F
|
|
number >>= 7
|
|
if number:
|
|
buf += _byte(towrite | 0x80)
|
|
else:
|
|
buf += _byte(towrite)
|
|
break
|
|
return buf
|
|
|
|
|
|
def decode_stream(stream):
|
|
"""Read a varint from `stream`"""
|
|
shift = 0
|
|
result = 0
|
|
while True:
|
|
i = _read_one(stream)
|
|
result |= (i & 0x7F) << shift
|
|
shift += 7
|
|
if not (i & 0x80):
|
|
break
|
|
|
|
return result
|
|
|
|
|
|
def decode_bytes(buf):
|
|
"""Read a varint from from `buf` bytes"""
|
|
return decode_stream(BytesIO(buf))
|
|
|
|
|
|
def _read_one(stream):
|
|
"""Read a byte from the file (as an integer)
|
|
|
|
raises EOFError if the stream ends while reading bytes.
|
|
"""
|
|
c = stream.read(1)
|
|
if c == b"":
|
|
raise EOFError("Unexpected EOF while reading bytes")
|
|
return ord(c)
|