From 1f35eb165d4b602fc1ceef3bafe656bb1edb24fb Mon Sep 17 00:00:00 2001 From: Mikhail Klementev Date: Mon, 10 Dec 2018 02:32:13 +0000 Subject: [PATCH] if-else -> switch --- pew.go | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/pew.go b/pew.go index 64622dc..f83ca0d 100644 --- a/pew.go +++ b/pew.go @@ -244,7 +244,8 @@ func whatever(swg *sizedwaitgroup.SizedWaitGroup, ka config.Artifact, } } - if ka.Type == config.KernelModule { + switch ka.Type { + case config.KernelModule: // TODO Write insmod log to file or database output, err := q.CopyAndInsmod(outFile) if err != nil { @@ -260,7 +261,7 @@ func whatever(swg *sizedwaitgroup.SizedWaitGroup, ka config.Artifact, return } test_ok = true - } else if ka.Type == config.KernelExploit { + case config.KernelExploit: remoteExploit := fmt.Sprintf("/tmp/exploit_%d", rand.Int()) err = q.CopyFile("user", outFile, remoteExploit) if err != nil { @@ -275,7 +276,7 @@ func whatever(swg *sizedwaitgroup.SizedWaitGroup, ka config.Artifact, } run_ok = true // does not really used test_ok = true - } else { + default: err = errors.New("Unsupported artifact type") } return