perf(evaluate): support two new formatted print func

cdk-team · Sep 25, 2022 · 8e5e40d · 8e5e40d
1 parent 950705e
commit 8e5e40d
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 3 deletions.
diff --git a/pkg/evaluate/evaluate.go b/pkg/evaluate/evaluate.go
@@ -44,6 +44,9 @@ func CallBasics() {
 	util.PrintH2("Information Gathering - Sysctl Variables")
 	CheckRouteLocalNetworkValue()
 
+	util.PrintH2("Information Gathering - DNS-Based Service Discovery")
+	DNSBasedServiceDiscovery()
+
 	util.PrintH2("Discovery - K8s API Server")
 	CheckK8sAnonymousLogin()
 
@@ -53,8 +56,8 @@ func CallBasics() {
 	util.PrintH2("Discovery - Cloud Provider Metadata API")
 	CheckCloudMetadataAPI()
 
-	util.PrintH2("Information Gathering - DNS-Based Service Discovery")
-	DNSBasedServiceDiscovery()
+	util.PrintH2("Exploit Pre - Kernel Exploits")
+	kernelExploitSuggester()
 }
 
 func CallAddedFunc() {

diff --git a/pkg/evaluate/evaluate_test.go b/pkg/evaluate/evaluate_test.go
@@ -1,4 +1,3 @@
-
 /*
 Copyright 2022 The Authors of https://github.com/CDK-TEAM/CDK .
 
@@ -20,6 +19,8 @@ package evaluate
 import (
 	"fmt"
 	"testing"
+
+	"github.com/cdk-team/CDK/pkg/util"
 )
 
 func TestDumpCgroup(t *testing.T) {
@@ -31,3 +32,8 @@ func TestFindSidFiles(t *testing.T) {
 	fmt.Printf("\n[Information Gathering - SIDs]\n")
 	FindSidFiles()
 }
+
+func TestKernelExploitSuggester(t *testing.T) {
+	util.PrintH2("Exploit Pre - Kernel Exploits")
+	kernelExploitSuggester()
+}
diff --git a/pkg/util/output.go b/pkg/util/output.go
@@ -46,3 +46,15 @@ func PrintItemValue(value string, color bool) {
 	}
 }
 
+func PrintItemValueWithKeyOneLine(key, value string, color bool) {
+	if color {
+		log.Printf("%s: %s", key, GreenBold.Sprint(value))
+	} else {
+		log.Printf("%s: %s", key, value)
+	}
+}
+
+func PrintOrignal(out string) {
+	fmt.Printf("%s\n", out)
+}
+