feat: hook gateway simulator within client messages

31333337 · Sep 20, 2023 · 263e5f9 · 263e5f9
1 parent e5d55f7
commit 263e5f9
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/client/clients.go b/client/clients.go
@@ -11,6 +11,7 @@ import (
 	"runtime"
 	"sync"
 
+	"github.com/simonlangowski/lightning1/cmd/xtrellis/gateway"
 	"github.com/simonlangowski/lightning1/config"
 	coord "github.com/simonlangowski/lightning1/coordinator/messages"
 	"github.com/simonlangowski/lightning1/crypto/token"
@@ -125,6 +126,12 @@ func (c *ClientRunner) ClientStart(_ context.Context, i *coord.RoundInfo) (*coor
 					}
 					m := make([]byte, i.MessageSize)
 					binary.LittleEndian.PutUint64(m, uint64(id))
+
+					// gateway simulation
+					if gateway.Enable {
+						m, _ = gateway.GetMessageForClient(i, id)
+					}
+
 					done <- cli.SendLightningMessage(c.Caller, cli.PathKeys, m)
 				}
 			}(id)

diff --git a/coordinator/coordinator.go b/coordinator/coordinator.go
@@ -11,6 +11,8 @@ import (
 	"sync"
 	"time"
 
+	"github.com/simonlangowski/lightning1/cmd/xtrellis/gateway"
+	"github.com/simonlangowski/lightning1/cmd/xtrellis/utils"
 	"github.com/simonlangowski/lightning1/config"
 	coord "github.com/simonlangowski/lightning1/coordinator/messages"
 	"github.com/simonlangowski/lightning1/crypto"
@@ -145,6 +147,7 @@ func (c *Coordinator) DoAction(exp *Experiment) error {
 				return err
 			}
 		}
+		utils.DebugLog("exp.Info. PathEstablishment=%t; ReceiptLayer=%d; Round=%d", exp.Info.PathEstablishment, exp.Info.ReceiptLayer, exp.Info.Round)
 		if exp.Info.PathEstablishment {
 			if exp.Info.ReceiptLayer == 0 && exp.Info.Round != 0 {
 				err := c.Net.CheckClientReceipt(exp.Info, exp.NumMessages)
@@ -180,7 +183,15 @@ func (c *Coordinator) DoAction(exp *Experiment) error {
 				log.Printf("Get messages")
 				return err
 			}
-			exp.Passed = c.Check(messages, exp.NumMessages)
+
+			if gateway.Enable {
+				// give final messages to gateway and check them
+				exp.Passed = gateway.CheckFinalMessages(messages, exp.NumMessages)
+				utils.DebugLog("exp.Passed = gateway.CheckFinalMessages = %t", exp.Passed)
+			} else {
+				exp.Passed = c.Check(messages, exp.NumMessages)
+			}
+
 		}
 		endTime := time.Now()
 		exp.ServerRoundTime = endTime.Sub(roundStartTime)