websockets: mask specification

- Add a FixedLengthValue
- Use it to implement a 4-byte mask specifier
- Adjust docs

1 files changed, 9 insertions, 2 deletions

diff --git a/libpathod/language/websockets.py b/libpathod/language/websockets.py
index a6674988..3abdf9d8 100644
--- a/libpathod/language/websockets.py
+++ b/libpathod/language/websockets.py
@@ -8,8 +8,9 @@ from . import base, generators, actions, message
     wf:c15:r'foo'
     wf:fin:rsv1:rsv2:rsv3:mask
     wf:-fin:-rsv1:-rsv2:-rsv3:-mask
-    wf:p234
-    wf:m"mask"
+
+    wf:k"mask"
+    wf:l234
 """
 
 
@@ -58,6 +59,11 @@ class Mask(base.Boolean):
     name = "mask"
 
 
+class Key(base.FixedLengthValue):
+    preamble = "k"
+    length = 4
+
+
 class WebsocketFrame(message.Message):
     comps = (
         Body,
@@ -72,6 +78,7 @@ class WebsocketFrame(message.Message):
         actions.PauseAt,
         actions.DisconnectAt,
         actions.InjectAt,
+        Key,
 
         Raw,
     )