working on memory leaks

Author: caffix

engine/dispatcher/dispatcher.go

@@ -117,6 +117,7 @@ func (d *dynamicDispatcher) DispatchEvent(e *et.Event) error {
 
 	err := e.Session.Backlog().Enqueue(e.Entity)
 	if err != nil {
+		_ = d.meta.DeleteSessionEntry(e.Session.ID().String(), e.Entity.ID)
 		return err
 	}
 
@@ -144,6 +145,7 @@ func (d *dynamicDispatcher) ResubmitEvent(e *et.Event) error {
 
 	err := e.Session.Backlog().Enqueue(e.Entity)
 	if err != nil {
+		_ = d.meta.DeleteSessionEntry(e.Session.ID().String(), e.Entity.ID)
 		return err
 	}
 
@@ -235,13 +237,9 @@ func (d *dynamicDispatcher) updateMetaMap() {
 }
 
 func (d *dynamicDispatcher) removeKilledSessions() {
-	sessions := d.mgr.GetSessions()
-	if len(sessions) == 0 {
-		return
-	}
-
 	var sids []string
-	for _, sess := range sessions {
+
+	for _, sess := range d.mgr.GetSessions() {
 		sids = append(sids, sess.ID().String())
 	}
 

engine/engine.go

@@ -7,9 +7,12 @@ package engine
 import (
 	"errors"
 	"log/slog"
+	"net/http"
 	"os"
 	"time"
 
+	_ "net/http/pprof"
+
 	"github.com/owasp-amass/amass/v5/engine/api/server"
 	"github.com/owasp-amass/amass/v5/engine/dispatcher"
 	"github.com/owasp-amass/amass/v5/engine/plugins"
@@ -27,6 +30,10 @@ type Engine struct {
 }
 
 func NewEngine(l *slog.Logger) (*Engine, error) {
+	go func() {
+		_ = http.ListenAndServe("127.0.0.1:6060", nil)
+	}()
+
 	if l == nil {
 		l = slog.New(slog.NewTextHandler(os.Stdout, nil))
 	}