Missed a few calls in the refactoring

lkarlslund · Dec 15, 2023 · b8f39b4 · b8f39b4
1 parent 53ea2e3
commit b8f39b4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/modules/analyze/webservicefuncs.go b/modules/analyze/webservicefuncs.go
@@ -40,7 +40,7 @@ func analysisfuncs(ws *webservice) {
 		}
 		var results returnobject
 
-		for _, edge := range engine.AllEdgesSlice() {
+		for _, edge := range engine.Edges() {
 			if !edge.IsHidden() {
 				results.Methods = append(results.Methods, filterinfo{
 					Name:            edge.String(),
@@ -282,7 +282,7 @@ func analysisfuncs(ws *webservice) {
 			if strings.HasPrefix(potentialfilter, "pwn_") {
 				prefix := potentialfilter[4 : len(potentialfilter)-2]
 				suffix := potentialfilter[len(potentialfilter)-2:]
-				edge := engine.E(prefix)
+				edge := engine.LookupEdge(prefix)
 				if edge == engine.NonExistingEdgeType {
 					continue
 				}

diff --git a/modules/query/ldapparser.go b/modules/query/ldapparser.go
@@ -246,7 +246,7 @@ valueloop:
 			if edgename == "*" {
 				edge = engine.AnyEdgeType
 			} else {
-				edge = engine.E(edgename)
+				edge = engine.LookupEdge(edgename)
 				if edge == engine.NonExistingEdgeType {
 					return nil, nil, fmt.Errorf("Could not convert value %v to edge", edgename)
 				}