From b989b6a26c7ad20991313e844ad6179f13d1b65b Mon Sep 17 00:00:00 2001
From: Federico Di Pierro <nierro92@gmail.com>
Date: Mon, 6 May 2024 10:55:09 +0200
Subject: [PATCH] chore: remove non printable chars from gh token.

See https://github.com/falcosecurity/peribolos-syncer/pull/17.

Signed-off-by: Federico Di Pierro <nierro92@gmail.com>
---
 main.go | 27 ++++++++++++++++++++++++---
 1 file changed, 24 insertions(+), 3 deletions(-)

diff --git a/main.go b/main.go
index dff2b70..8fad49c 100644
--- a/main.go
+++ b/main.go
@@ -17,8 +17,10 @@ package main
 import (
 	"context"
 	"flag"
+	"github.com/pkg/errors"
 	"os"
 	"strings"
+	"unicode"
 
 	"github.com/falcosecurity/pigeon/pkg/pigeon"
 	"github.com/google/go-github/v50/github"
@@ -41,6 +43,26 @@ func init() {
 	flag.BoolVar(&verbose, "verbose", false, "enable verbose logging")
 }
 
+func getTokenFromFile(path string) (string, error) {
+	token, err := os.ReadFile(path)
+	if err != nil {
+		return "", errors.Wrap(err, "error reading token file")
+	}
+
+	return removeNonPrintableChars(string(token)), nil
+}
+
+func removeNonPrintableChars(s string) string {
+	return strings.Map(func(r rune) rune {
+		switch {
+		case unicode.IsPrint(r):
+			return r
+		default:
+			return -1
+		}
+	}, s)
+}
+
 func initOpts() {
 	flag.Parse()
 
@@ -55,12 +77,11 @@ func initOpts() {
 			logrus.Fatal(`Github token must be provided either through "gh-token" flag, or "GITHUB_AUTH_TOKEN" env."`)
 		}
 	}
-	ghTokBytes, err := os.ReadFile(ghToken)
+	var err error
+	ghToken, err = getTokenFromFile(ghToken)
 	if err != nil {
 		logrus.Fatal(err)
 	}
-	ghToken = string(ghTokBytes)
-	ghToken = strings.Trim(ghToken, "\n")
 
 	if confFile == "" {
 		logrus.Fatal(`"conf" flag must be set`)