Merge pull request #253 from slimm609/fix_relro

fix: fix relro checks based on gcc and os

README.md

@@ -22,7 +22,7 @@ Updates
       - golang: real    0m0.691s
   - Adds recursive directory support
   TODO:
-  - [ ] Fix Partial RELRO
+  - [X] Fix Partial RELRO
   - [ ] Add fortify file function results
   - [ ] Add fortifyProc
   - [ ] Add ProcLibs

pkg/checksec/fortify.go

@@ -37,14 +37,7 @@ func Fortify(name string, binary *elf.File) *fortify {
 
 	ldd := GetLdd(name)
 
-	if ldd == "none" {
-		res.Output = "N/A"
-		res.Color = "green"
-		res.Fortified = "0"
-		res.Fortifiable = "0"
-		res.LibcSupport = "N/A"
-		return &res
-	} else if ldd == "unk" {
+	if ldd == "none" || ldd == "unk" {
 		res.Output = "N/A"
 		res.Color = "unset"
 		res.Fortified = "0"

pkg/checksec/relro.go

@@ -25,8 +25,14 @@ func RELRO(name string) *relro {
 	}
 	defer file.Close()
 
-	bind, _ := file.DynValue(24)
-	if len(bind) > 0 && bind[0] == 0 {
+	// check both bind and bind_flag.
+	// if DT_BIND_NOW == 0, then it is set
+	// if DT_FLAGS == 8, then DF_BIND_NOW is set
+	// this is depending on the compiler version used.
+	bind, _ := file.DynValue(elf.DT_BIND_NOW)
+	bind_flag, _ := file.DynValue(elf.DT_FLAGS)
+
+	if (len(bind) > 0 && bind[0] == 0) || (len(bind_flag) > 0 && bind_flag[0] == 8) {
 		bindNow = true
 	}
 

pkg/utils/utils.go

@@ -7,14 +7,12 @@ import (
 func PrintLogo() {
 	Red := color.New(color.FgHiRed, color.Bold)
 	asciiLogo := `
-_______           _______  _______  _        _______  _______  _______
-(  ____ \|\     /|(  ____ \(  ____ \| \    /\(  ____ \(  ____ \(  ____ \
-| (    \/| )   ( || (    \/| (    \/|  \  / /| (    \/| (    \/| (    \/
-| |      | (___) || (__    | |      |  (_/ / | (_____ | (__    | |
-| |      |  ___  ||  __)   | |      |   _ (  (_____  )|  __)   | |
-| |      | (   ) || (      | |      |  ( \ \       ) || (      | |
-| (____/\| )   ( || (____/\| (____/\|  /  \ \/\____) || (____/\| (____/\
-(_______/|/     \|(_______/(_______/|_/    \/\_______)(_______/(_______/
+  _____ _    _ ______ _____ _  __ _____ ______ _____
+ / ____| |  | |  ____/ ____| |/ // ____|  ____/ ____|
+| |    | |__| | |__ | |    | ' /| (___ | |__ | |
+| |    |  __  |  __|| |    |  <  \___ \|  __|| |
+| |____| |  | | |___| |____| . \ ____) | |___| |____
+ \_____|_|  |_|______\_____|_|\_\_____/|______\_____|
 `
 	Red.Println(asciiLogo)
 }