websocket: interface refactoring

- Separate out FrameHeader. We need to deal with this separately in many circumstances.
- Simpler equality scheme.
- Bits are now specified by truthiness - we don't care about the integer value.
This means lots of validation is not needed any more.

1 files changed, 16 insertions, 0 deletions

diff --git a/netlib/utils.py b/netlib/utils.py
index 66bbdb5e..44bed43a 100644
--- a/netlib/utils.py
+++ b/netlib/utils.py
@@ -49,3 +49,19 @@ def hexdump(s):
             (o, x, cleanBin(part, True))
         )
     return parts
+
+
+def setbit(byte, offset, value):
+    """
+        Set a bit in a byte to 1 if value is truthy, 0 if not.
+    """
+    if value:
+        return byte | (1 << offset)
+    else:
+        return byte & ~(1 << offset)
+
+
+def getbit(byte, offset):
+    mask = 1 << offset
+    if byte & mask:
+        return True