Merge pull request #363 from TrekkieCoder/main

PR - egress hooks enable or disable based on runtime args
loxilb-io · Jul 26, 2023 · 1bb014d · 1bb014d
2 parents f0438f8 + e1ed228
commit 1bb014d
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 3 deletions.
diff --git a/loxilb-ebpf b/loxilb-ebpf
diff --git a/loxinet/dpebpf_linux.go b/loxinet/dpebpf_linux.go
@@ -232,14 +232,20 @@ func DpEbpfSetLogLevel(logLevel tk.LogLevelT) {
 }
 
 // DpEbpfInit - initialize the ebpf dp subsystem
-func DpEbpfInit(clusterEn bool, nodeNum int, rssEn bool, logLevel tk.LogLevelT) *DpEbpfH {
+func DpEbpfInit(clusterEn bool, nodeNum int, rssEn bool, egrHooks bool, logLevel tk.LogLevelT) *DpEbpfH {
  var cfg C.struct_ebpfcfg
 
  if clusterEn {
  cfg.have_mtrace = 1
  } else {
  cfg.have_mtrace = 0
  }
+ if egrHooks {
+ cfg.egr_hooks = 1
+ } else {
+ cfg.egr_hooks = 0
+ }
+
  cfg.nodenum = C.int(nodeNum)
  cfg.loglevel = 1
  cfg.no_loader = 0

diff --git a/loxinet/loxinet.go b/loxinet/loxinet.go
@@ -76,6 +76,7 @@ type loxiNetH struct {
  ready bool
  self int
  rssEn bool
+ eHooks bool
  pFile *os.File
 }
 
@@ -226,6 +227,7 @@ func loxiNetInit() {
 
  mh.self = opts.Opts.ClusterSelf
  mh.rssEn = opts.Opts.RssEnable
+ mh.eHooks = opts.Opts.EgrHooks
  mh.sumDis = opts.Opts.CSumDisable
  mh.pProbe = opts.Opts.PassiveEPProbe
  mh.sigCh = make(chan os.Signal, 5)
@@ -247,7 +249,7 @@ func loxiNetInit() {
  }
 
  // Initialize the ebpf datapath subsystem
- mh.dpEbpf = DpEbpfInit(clusterMode, mh.self, mh.rssEn, -1)
+ mh.dpEbpf = DpEbpfInit(clusterMode, mh.self, mh.rssEn, mh.eHooks, -1)
  mh.dp = DpBrokerInit(mh.dpEbpf)
 
  // Initialize the security zone subsystem

diff --git a/options/options.go b/options/options.go
@@ -24,4 +24,5 @@ var Opts struct {
  CSumDisable bool `long:"disable-csum" description:"Disable checksum update(experimental)"`
  PassiveEPProbe bool `long:"passive-probe" description:"Enable passive liveness probes(experimental)"`
  RssEnable bool `long:"rss-enable" description:"Enable rss optimization(experimental)"`
+ EgrHooks bool `long:"egr-hooks" description:"Enable eBPF egress hooks(experimental)"`
 }