From d4b5f6f709af27385b97413b2007ec0af8a3c513 Mon Sep 17 00:00:00 2001
From: kota kanbe <kotakanbe@gmail.com>
Date: Fri, 15 Dec 2017 20:03:30 +0900
Subject: [PATCH] Support Amazon Linux2

---
 scan/redhat.go | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/scan/redhat.go b/scan/redhat.go
index 584e07c97d..a5ffb25d23 100644
--- a/scan/redhat.go
+++ b/scan/redhat.go
@@ -107,9 +107,14 @@ func detectRedhat(c config.ServerInfo) (itsMe bool, red osTypeInterface) {
 		family := config.Amazon
 		release := "unknown"
 		if r := exec(c, "cat /etc/system-release", noSudo); r.isSuccess() {
-			fields := strings.Fields(r.Stdout)
-			if len(fields) == 5 {
-				release = fields[4]
+			if strings.HasPrefix(r.Stdout, "Amazon Linux release 2") {
+				fields := strings.Fields(r.Stdout)
+				release = fmt.Sprintf("%s %s", fields[3], fields[4])
+			} else {
+				fields := strings.Fields(r.Stdout)
+				if len(fields) == 5 {
+					release = fields[4]
+				}
 			}
 		}
 		red.setDistro(family, release)