From: Youhwan Seol <yh.seol@samsung.com>
Date: Wed, 8 Feb 2023 07:26:01 +0000 (+0900)
Subject: Adding random path generator to fetch helm package within random path
X-Git-Tag: 1.0.0~29^2
X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=commitdiff_plain;h=f6f52b86f033947effa96d271ea3d94c9fadd70a;p=aiml-fw%2Faihp%2Fips%2Fkserve-adapter.git

Adding random path generator to fetch helm package within random path

Change-Id: Ie7471780ffe94f7ddb220af65ad4a0a5dca48986
Signed-off-by: Youhwan Seol <yh.seol@samsung.com>
---

diff --git a/pkg/client/onboard/utils.go b/pkg/client/onboard/utils.go
index aff2cb0..a555345 100644
--- a/pkg/client/onboard/utils.go
+++ b/pkg/client/onboard/utils.go
@@ -23,11 +23,13 @@ import (
 	"archive/tar"
 	"compress/gzip"
 	"io"
+	"math/rand"
 	"net/http"
 	netUrl "net/url"
 	"os"
 	"path/filepath"
 	"strings"
+	"time"
 
 	"gerrit.o-ran-sc.org/r/aiml-fw/aihp/ips/kserve-adapter/pkg/api/commons/url"
 	"github.com/pkg/errors"
@@ -35,6 +37,20 @@ import (
 	"gerrit.o-ran-sc.org/r/aiml-fw/aihp/ips/kserve-adapter/pkg/commons/logger"
 )
 
+const (
+	pathLength    = 20
+	letterBytes   = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
+	letterIdxBits = 6
+	letterIdxMask = 1<<letterIdxBits - 1
+	letterIdxMax  = 63 / letterIdxBits
+)
+
+var randSrc rand.Source
+
+func init() {
+	randSrc = rand.NewSource(time.Now().UnixNano())
+}
+
 func fetchHelmPackageFromOnboard(url string) (path string, err error) {
 	logger.Logging(logger.DEBUG, "IN")
 	defer logger.Logging(logger.DEBUG, "OUT")
@@ -45,7 +61,7 @@ func fetchHelmPackageFromOnboard(url string) (path string, err error) {
 	}
 	defer resp.Body.Close()
 
-	path = "/tmp"
+	path = generateRandPath(pathLength)
 
 	err = Untar(path, resp.Body)
 	if err != nil {
@@ -173,3 +189,24 @@ func mkdirectory(name string) (err error) {
 	}
 	return
 }
+
+func generateRandPath(n int) string {
+	logger.Logging(logger.DEBUG, "IN")
+	defer logger.Logging(logger.DEBUG, "OUT")
+
+	sb := strings.Builder{}
+	sb.Grow(n)
+	for i, cache, remain := n-1, randSrc.Int63(), letterIdxMax; i >= 0; {
+		if remain == 0 {
+			cache, remain = randSrc.Int63(), letterIdxMax
+		}
+		if idx := int(cache & letterIdxMask); idx < len(letterBytes) {
+			sb.WriteByte(letterBytes[idx])
+			i--
+		}
+		cache >>= letterIdxBits
+		remain--
+	}
+
+	return sb.String()
+}