|
#
# JupyterMsg: Jupyter kernel using ZeroMQ
#
# Implementations
#
# TODO: Check signature
InstallGlobalFunction( JupyterMsgDecode,
function(kernel, raw)
local result, bindIfBound, sl, ids, tmp;
bindIfBound := function(name, pos)
if IsBound(raw[pos]) then
result.(name) := JsonStringToGap(raw[pos]);
fi;
end;
result := rec();
ids := [];
sl := 1;
while raw[sl] <> "<IDS|MSG>" do
Add(ids, raw[sl]);
sl := sl + 1;
od;
result.hmac := raw[sl + 1];
result.remainder := raw;
bindIfBound("header", sl + 2);
bindIfBound("parent_header", sl + 3);
bindIfBound("metadata", sl + 4);
bindIfBound("content", sl + 5);
tmp := CRYPTING_SHA256_HMAC( kernel!.SessionKey
, Concatenation( raw[sl + 2]
, raw[sl + 3]
, raw[sl + 4]
, raw[sl + 5] ) );
tmp := List(tmp, CRYPTING_HexStringIntPad8);
tmp := LowercaseString(Concatenation(tmp));
result.hmac_verify := tmp;
if result.hmac <> result.hmac_verify then
PrintTo( "*errout*", "HMAC verification for message failed: "
, result.hmac, " <> ", result.hmac_verify, "\n" );
fi;
return result;
end);
InstallGlobalFunction( JupyterMsgEncode,
function(kernel, msg)
local raw, k, bindIfBound, tmp;
bindIfBound := function(pos, name)
if IsBound(msg.(name)) then
raw[pos] := GapToJsonString(msg.(name));
fi;
end;
raw := [];
# TODO: What is the correct behaviour here?
if IsBound(msg.uuid) then
raw[1] := msg.uuid;
else
raw[1] := "";
fi;
raw[2] := "<IDS|MSG>";
raw[3] := msg.hmac;
bindIfBound(4, "header");
bindIfBound(5, "parent_header");
bindIfBound(6, "metadata");
bindIfBound(7, "content");
# TODO: Ugly
if Length(raw) > 3 then
tmp := CRYPTING_SHA256_HMAC(msg.key,
Concatenation( raw[4]
, raw[5]
, raw[6]
, raw[7]));
tmp := List(tmp, CRYPTING_HexStringIntPad8);
tmp := LowercaseString(Concatenation(tmp));
raw[3] := LowercaseString(tmp);
fi;
return raw;
end);
InstallGlobalFunction(JupyterMsgRecv,
function(kernel, sock)
local raw;
raw := ZmqReceiveList(sock);
if IsBound(kernel!.ProtocolLog) then
AppendTo(kernel!.ProtocolLog, raw);
AppendTo(kernel!.ProtocolLog, "\n");
fi;
return JupyterMsgDecode(kernel, raw);
end);
InstallGlobalFunction(JupyterMsgSend,
function(kernel, sock, msg)
local msg2;
if IsBound(kernel!.ProtocolLog) then
msg2 := JupyterMsgEncode(kernel, msg);
AppendTo(kernel!.ProtocolLog, msg2);
AppendTo(kernel!.ProtocolLog, "\n");
fi;
ZmqSend(sock, JupyterMsgEncode(kernel, msg));
end);
# Create a message template with the necessasry fields filled
InstallGlobalFunction(JupyterMsg,
function(kernel, msg_type, parent_header, content, metadata)
return rec( uuid := kernel!.ZmqIdentity
, sep := "<IDS|MSG>" # This could be in JupyterEncode
, hmac := ""
, header := rec( username := kernel!.Username
, session := kernel!.SessionID
, msg_type := msg_type
, version := kernel!.ProtocolVersion
, date := ISO8601Stamp()
, msg_id := HexStringUUID(RandomUUID())
)
, parent_header := parent_header
, metadata := metadata
, content := content
, key := kernel!.SessionKey
# This shouldn't be here as all
# messaging functions
# should just be running in kernel context
);
end);
[ Dauer der Verarbeitung: 0.5 Sekunden
(vorverarbeitet)
]
|
