Used PwnMethodBitmap functionality rather than duplicating it

modules/engine/analyzeobjects.go

@@ -159,19 +159,11 @@ func AnalyzeObjects(opts AnalyzeObjectsOptions) (pg PwnGraph) {
 					}
 				}
 
-				maxprobability := Probability(-128)
-				for i := 0; i < len(pwnnums); i++ {
-					if detectedmethods.IsSet(PwnMethod(i)) {
-						var probability Probability
-						if forward {
-							probability = CalculateProbability(pwntarget, object, PwnMethod(i))
-						} else {
-							probability = CalculateProbability(object, pwntarget, PwnMethod(i))
-						}
-						if probability > maxprobability {
-							maxprobability = probability
-						}
-					}
+				var maxprobability Probability
+				if forward {
+					maxprobability = detectedmethods.MaxProbability(pwntarget, object)
+				} else {
+					maxprobability = detectedmethods.MaxProbability(object, pwntarget)
 				}
 				if maxprobability < Probability(opts.MinProbability) {
 					// Too unlikeliy, so we skip it

modules/engine/pwn.go

@@ -20,6 +20,11 @@ const MAXPWNMETHODPOSSIBLE = PMBSIZE * 64
 type PwnMethodBitmap [PMBSIZE]uint64
 type Probability int8
 
+const (
+	MINPROBABILITY Probability = -1
+	MAXPROBABILITY Probability = 100
+)
+
 type PwnInfo struct {
 	Target      *Object
 	Method      PwnMethod
@@ -220,12 +225,12 @@ func (m PwnMethodBitmap) IsSet(method PwnMethod) bool {
 	return (m[method/64] & (1 << (method % 64))) != 0 // Uuuuh, nasty and unreadable
 }
 
-func (m PwnMethodBitmap) MaxProbabiltity(source, target *Object) Probability {
+func (m PwnMethodBitmap) MaxProbability(source, target *Object) Probability {
 	var max Probability
 	for i := 0; i < len(pwnnums); i++ {
 		if m.IsSet(PwnMethod(i)) {
-			prob := CalculateProbability(source, target, PwnMethod(i))
-			if prob == 100 {
+			prob := PwnMethod(i).Probability(source, target)
+			if prob == MAXPROBABILITY {
 				return prob
 			}
 			if prob > max {