mirrors/ctrl
mirrors/ctrl
1
0
Fork 0
mirror of https://github.com/postmannen/ctrl.git synced 2024-12-14 12:37:31 +00:00
Code Activity

added a push update when deleting keys

Browse source
This commit is contained in:
postmannen 2022-06-02 09:36:09 +02:00
parent 7e8ecc3397
commit bf14ad4d2a
1 changed files with 116 additions and 79 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

69
requests_keys.go
View file

@ -304,7 +304,28 @@ func (m methodREQKeysAllow) handler(proc process, message Message, node string)
//
// If there are errors we will return from the function, and send no
// updates.
err := func() error {
err := pushKeys(proc, message, []Node{})
if err != nil {
proc.errorKernel.errSend(proc, message, err)
return
}
}
}()
ackMsg := []byte("confirmed from: " + node + ": " + fmt.Sprint(message.ID))
return ackMsg, nil
}
// pushKeys will try to push the the current keys and hashes to each node once.
// As input arguments it takes the current process, the current message, and a
// []Node in nodes.
// The nodes are used when a node or nodes have been deleted from the current
// nodesAcked map since it will contain the nodes that were deleted so we are
// also able to send an update to them as well.
func pushKeys(proc process, message Message, nodes []Node) error {
fmt.Printf(" \n\n\n ************** beginning of pushKeys, nodes=%v\n", nodes)
var knh []byte
err := func() error {
@ -331,6 +352,7 @@ func (m methodREQKeysAllow) handler(proc process, message Message, node string)
// For all nodes that is not ack'ed we try to send an update once.
for n := range proc.centralAuth.pki.nodeNotAckedPublicKeys.KeyMap {
fmt.Printf("\n\n\n ************** DEBUG: node to send REQKeysDeliverUpdate to:%v\n ", n)
msg := Message{
ToNode: n,
Method: REQKeysDeliverUpdate,
@ -357,8 +379,23 @@ func (m methodREQKeysAllow) handler(proc process, message Message, node string)
proc.errorKernel.errSend(proc, message, er)
}
// For all nodes that is ack'ed we try to send an update once.
proc.centralAuth.pki.nodesAcked.mu.Lock()
defer proc.centralAuth.pki.nodesAcked.mu.Unlock()
// Concatenate the current nodes in the keysAndHash map and the nodes
// we got from the function argument when this function was called.
nodeMap := make(map[Node]struct{})
for n := range proc.centralAuth.pki.nodesAcked.keysAndHash.Keys {
nodeMap[n] = struct{}{}
}
for _, n := range nodes {
nodeMap[n] = struct{}{}
}
// For all nodes that is ack'ed we try to send an update once.
for n := range nodeMap {
fmt.Printf("\n\n\n ************** DEBUG: node to send REQKeysDeliverUpdate to:%v\n ", n)
msg := Message{
ToNode: n,
Method: REQKeysDeliverUpdate,
@ -380,18 +417,6 @@ func (m methodREQKeysAllow) handler(proc process, message Message, node string)
return nil
}()
if err != nil {
proc.errorKernel.errSend(proc, message, err)
return
}
}
}()
ackMsg := []byte("confirmed from: " + node + ": " + fmt.Sprint(message.ID))
return ackMsg, nil
}
type methodREQKeysDelete struct {
@ -440,6 +465,19 @@ func (m methodREQKeysDelete) handler(proc process, message Message, node string)
proc.centralAuth.updateHash(proc, message)
log.Printf(" * DEBUG updated hash for public keys\n")
var nodes []Node
for _, n := range message.MethodArgs {
nodes = append(nodes, Node(n))
}
err := pushKeys(proc, message, nodes)
if err != nil {
proc.errorKernel.errSend(proc, message, err)
return
}
outString := fmt.Sprintf("deleted public keys for the nodes=%v\n", message.MethodArgs)
out := []byte(outString)
@ -460,8 +498,7 @@ func (m methodREQKeysDelete) handler(proc process, message Message, node string)
proc.errorKernel.errSend(proc, message, er)
case out := <-outCh:
// Prepare and queue for sending a new message with the output
// of the action executed.
newReplyMessage(proc, message, out)
}