From fd51b52aa6b226136348439c2c95abca4834f480 Mon Sep 17 00:00:00 2001
From: rctl <rasmus@rctl.io>
Date: Sun, 9 Jun 2019 16:05:13 +0100
Subject: [PATCH] Fix WSL support

---
 prodaccess_nix.go | 106 +++++++++++++++++++++++++++++++++++-----------
 1 file changed, 81 insertions(+), 25 deletions(-)

diff --git a/prodaccess_nix.go b/prodaccess_nix.go
index c9cd174..f9640d4 100644
--- a/prodaccess_nix.go
+++ b/prodaccess_nix.go
@@ -10,7 +10,10 @@ import (
 	"log"
 	"os"
 	"os/exec"
+	"path"
 	"strings"
+
+	"golang.org/x/sys/unix"
 )
 
 var (
@@ -108,27 +111,20 @@ func saveVmwareCertificate(c string, k string) {
 	kf.Write([]byte(k))
 	cp := cf.Name()
 	kp := kf.Name()
+	defer os.Remove(cp)	
+	defer os.Remove(kp)
 	cf.Close()
 	kf.Close()
 
 	fp := os.ExpandEnv(*vmwareCertPath)
 	os.Remove(fp)
 	os.OpenFile(fp, os.O_CREATE, 0600)
-	cmd := exec.Command("/usr/bin/env", "openssl", "pkcs12", "-export", "-password", "pass:",
-		"-in", cp, "-inkey", kp, "-out", fp)
-
-	var stdout, stderr bytes.Buffer
-	cmd.Stdout = &stdout
-	cmd.Stderr = &stderr
-
-	err := cmd.Run()
-	if err != nil {
-		log.Printf("Failed to emit VMware certificate: %v", err)
-		log.Printf("Standard output: %s", stdout.String())
-		log.Printf("Error output: %s", stderr.String())
+	executeWithStdout("openssl", "pkcs12", "-export", "-password", "pass:", "-in", cp, "-inkey", kp, "-out", fp)
+	if isWSL() {
+		if err := importCertFromWSL(fp); err != nil {
+			log.Printf("Failed to import certificate: %v", err)
+		}
 	}
-	os.Remove(cp)
-	os.Remove(kp)
 }
 
 func saveBrowserCertificate(c string, k string) {
@@ -138,25 +134,85 @@ func saveBrowserCertificate(c string, k string) {
 	kf.Write([]byte(k))
 	cp := cf.Name()
 	kp := kf.Name()
+	defer os.Remove(cp)
+	defer os.Remove(kp)
 	cf.Close()
 	kf.Close()
 
 	fp := os.ExpandEnv(*browserCertPath)
 	os.Remove(fp)
 	os.OpenFile(fp, os.O_CREATE, 0600)
-	cmd := exec.Command("/usr/bin/env", "openssl", "pkcs12", "-export", "-password", "pass:",
-		"-in", cp, "-inkey", kp, "-out", fp)
+	executeWithStdout("openssl", "pkcs12", "-export", "-password", "pass:", "-in", cp, "-inkey", kp, "-out", fp)
+	if isWSL() {
+		if err := importCertFromWSL(fp); err != nil {
+			log.Printf("Failed to import certificate: %v", err)
+		}
+	}
+}
 
-	var stdout, stderr bytes.Buffer
-	cmd.Stdout = &stdout
-	cmd.Stderr = &stderr
+func isWSL() bool {	
+	u := unix.Utsname{}	
+	_ = unix.Uname(&u)	
+	return strings.Contains(string(u.Release[:]), "Microsoft")	
+}	
+
+func executeWithStdout(cmd ...string) (string, error) {	
+	return executeWithStdoutWithStdin("", cmd...)	
+}
 
-	err := cmd.Run()
+func executeWithStdoutWithStdin(stdin string, cmd ...string) (string, error) {
+	c := exec.Command("/usr/bin/env", cmd...)
+	var stdout, stderr bytes.Buffer
+	si := bytes.NewBufferString(stdin)
+	c.Stdout = &stdout
+	c.Stderr = &stderr
+	c.Stdin = si
+	err := c.Run()
 	if err != nil {
-		log.Printf("Failed to emit Browser certificate: %v", err)
-		log.Printf("Standard output: %s", stdout.String())
-		log.Printf("Error output: %s", stderr.String())
+		log.Printf("Failed to execute %v: %v", cmd, err)
+		return "", err
 	}
-	os.Remove(cp)
-	os.Remove(kp)
+
+	return stdout.String(), nil
+}
+
+func executeWithStdin(stdin string, cmd ...string) error {
+	_, err := executeWithStdoutWithStdin(stdin, cmd...)
+	return err
 }
+
+// If running under WSL invoke PowerShell to import certificate
+func importCertFromWSL(pfx string) error {
+	winpath, err := executeWithStdout("powershell.exe", "echo $env:TEMP")
+	if err != nil {
+		return err
+	}
+
+	winpath = strings.Trim(winpath, "\r\n")
+	winparts := strings.Split(winpath, "\\")
+	drive := strings.ToLower(winparts[0][:1])
+	winparts = winparts[1:]
+	temppath := path.Join("/mnt", drive, path.Join(winparts...), "prodaccess.pfx")
+
+	pfxdata, err := ioutil.ReadFile(pfx)
+	if err != nil {
+		return err
+	}
+
+	err = ioutil.WriteFile(temppath, pfxdata, 0644)
+	if err != nil {
+		return err
+	}
+	defer os.Remove(temppath)
+
+	ps := fmt.Sprintf(psImport, winpath+"\\prodaccess.pfx")
+	fmt.Printf("%s\n", ps)
+	o, err := executeWithStdoutWithStdin(ps, "powershell.exe", "-Command", "-")
+	if err != nil {
+		return err
+	}
+
+	log.Printf("Imported certificate: %s", o)
+
+	return executeWithStdin(psPurge, "powershell.exe", "-Command", "-")
+}
\ No newline at end of file