proc: a .cmd or .bat is run through cmd.exe with each argument quoted for it, so npm.cmd under C:\Program Files\nodejs (nvm-windows) no longer runs "C:\Program" when magpie updates Claude Code (wztlink1013 on Discord)
Windows starts a batch file as `cmd.exe /c <line>`, and Go builds that line for a program's parser: with more than two quotes cmd.exe drops the first and last, so `"C:\Program Files\nodejs\npm.cmd" install -g --prefix "C:\Program Files\nodejs" …` ran C:\Program. An argument with & or | was also left unquoted and split into a second command. Every command magpie runs is made by proc.Command/CommandContext, so the fix is there: cmd.exe /d /e:ON /v:OFF /s /c "<script> <args>", each argument quoted as Rust's std does for batch files (doubled ", % broken up). TestBatchFileInFolderWithSpace (windows) runs npm.cmd, a .bat and npm via PATH under "Program Files" with the reporter's arguments plus & | % ^ and trailing-\ cases, through a script that hands %* to a real program. On the old code it fails on the Windows box with "'…\Program' 不是内部或外部命令". On the Windows box (Windows 11, zh-CN) magpie's own updater (howInstalled + runUpdate) on a real npm install of Claude Code 2.0.50 under "…\Program Files\nodejs": old code failed with the reporter's error, new code updated it to 2.1.293.
This commit is contained in:
1 parent
811681bc72
commit
9fcd332b6c
4 files changed
+178
No files matched your search
@@ -0,0 +1,114 @@
|
|||||||
|
package proc
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"os"
|
||||||
|
"os/exec"
|
||||||
|
"path/filepath"
|
||||||
|
"strings"
|
||||||
|
"syscall"
|
||||||
|
"unicode"
|
||||||
|
)
|
||||||
|
|
||||||
|
// batch has a .cmd or .bat (npm.cmd, a CLI's npm shim) run by cmd.exe with
|
||||||
|
// a command line cmd.exe reads back as the arguments given. Windows runs a
|
||||||
|
// batch file as `cmd.exe /c <command line>`, and Go builds that line for a
|
||||||
|
// program's own parsing: cmd.exe, seeing more than two quotes, drops the
|
||||||
|
// first and the last, so `"C:\Program Files\nodejs\npm.cmd" … --prefix
|
||||||
|
// "C:\Program Files\nodejs" …` ran C:\Program; and an argument with & or |
|
||||||
|
// in it, which Go leaves unquoted, was split into a second command there.
|
||||||
|
//
|
||||||
|
// Each argument is quoted the way Rust's standard library quotes one for a
|
||||||
|
// batch file: in quotes when it has anything but letters, digits and
|
||||||
|
// #$*+-./:?@\_ in it or a \ at its end, a " doubled, a % broken up so
|
||||||
|
// cmd.exe expands no variable, and a line break refused, as it would end
|
||||||
|
// the command. A doubled " is read back as one by programs built with
|
||||||
|
// Microsoft's C runtime since 2008, node among them; Go's own parser reads
|
||||||
|
// it the older way.
|
||||||
|
func batch(cmd *exec.Cmd) {
|
||||||
|
if cmd.Err != nil || cmd.Path == "" {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
switch strings.ToLower(filepath.Ext(cmd.Path)) {
|
||||||
|
case ".cmd", ".bat":
|
||||||
|
default:
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if strings.ContainsAny(cmd.Path, "\"\r\n") {
|
||||||
|
cmd.Err = errBatchArg
|
||||||
|
return
|
||||||
|
}
|
||||||
|
var b strings.Builder
|
||||||
|
b.WriteString(`/d /e:ON /v:OFF /s /c ""`)
|
||||||
|
b.WriteString(cmd.Path)
|
||||||
|
b.WriteByte('"')
|
||||||
|
for _, a := range cmd.Args[1:] {
|
||||||
|
if strings.ContainsAny(a, "\r\n\x00") {
|
||||||
|
cmd.Err = errBatchArg
|
||||||
|
return
|
||||||
|
}
|
||||||
|
b.WriteByte(' ')
|
||||||
|
quoteBatchArg(&b, a)
|
||||||
|
}
|
||||||
|
b.WriteByte('"')
|
||||||
|
shell := comSpec()
|
||||||
|
if cmd.SysProcAttr == nil {
|
||||||
|
cmd.SysProcAttr = &syscall.SysProcAttr{}
|
||||||
|
}
|
||||||
|
cmd.SysProcAttr.CmdLine = `"` + shell + `" ` + b.String()
|
||||||
|
cmd.Args = append([]string{shell, "/d", "/e:ON", "/v:OFF", "/s", "/c", cmd.Path}, cmd.Args[1:]...)
|
||||||
|
cmd.Path = shell
|
||||||
|
}
|
||||||
|
|
||||||
|
var errBatchArg = errors.New("cmd.exe can't be handed a batch file with a quote in its path, or a line break or NUL in an argument")
|
||||||
|
|
||||||
|
func quoteBatchArg(b *strings.Builder, a string) {
|
||||||
|
// a trailing \ is quoted too, or a script's "%~1" would have it
|
||||||
|
// escape the quote after it
|
||||||
|
quote := a == "" || strings.HasSuffix(a, `\`) || strings.IndexFunc(a, func(r rune) bool {
|
||||||
|
if r < 0x80 {
|
||||||
|
return !(r >= 'a' && r <= 'z' || r >= 'A' && r <= 'Z' || r >= '0' && r <= '9' || strings.ContainsRune(`#$*+-./:?@\_`, r))
|
||||||
|
}
|
||||||
|
return unicode.IsControl(r)
|
||||||
|
}) >= 0
|
||||||
|
if quote {
|
||||||
|
b.WriteByte('"')
|
||||||
|
}
|
||||||
|
backslashes := 0
|
||||||
|
for _, r := range a {
|
||||||
|
switch r {
|
||||||
|
case '\\':
|
||||||
|
backslashes++
|
||||||
|
b.WriteRune(r)
|
||||||
|
continue
|
||||||
|
case '"':
|
||||||
|
// the backslashes before it are doubled and the quote is
|
||||||
|
// doubled, which keeps cmd.exe's idea of what is quoted
|
||||||
|
b.WriteString(strings.Repeat(`\`, backslashes))
|
||||||
|
b.WriteByte('"')
|
||||||
|
case '%':
|
||||||
|
// %cd:~,% is always empty, and it stands between this % and
|
||||||
|
// the name after it
|
||||||
|
b.WriteString("%%cd:~,")
|
||||||
|
}
|
||||||
|
backslashes = 0
|
||||||
|
b.WriteRune(r)
|
||||||
|
}
|
||||||
|
if quote {
|
||||||
|
b.WriteString(strings.Repeat(`\`, backslashes))
|
||||||
|
b.WriteByte('"')
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// comSpec is cmd.exe: %ComSpec% when it names one, else the one in
|
||||||
|
// System32.
|
||||||
|
func comSpec() string {
|
||||||
|
if c := os.Getenv("ComSpec"); filepath.IsAbs(c) && strings.EqualFold(filepath.Base(c), "cmd.exe") {
|
||||||
|
return c
|
||||||
|
}
|
||||||
|
root := os.Getenv("SystemRoot")
|
||||||
|
if root == "" {
|
||||||
|
root = `C:\Windows`
|
||||||
|
}
|
||||||
|
return filepath.Join(root, "System32", "cmd.exe")
|
||||||
|
}
|
||||||
@@ -23,6 +23,7 @@ import (
|
|||||||
// Command is exec.Command, with no window of its own on Windows.
|
// Command is exec.Command, with no window of its own on Windows.
|
||||||
func Command(name string, args ...string) *exec.Cmd {
|
func Command(name string, args ...string) *exec.Cmd {
|
||||||
cmd := exec.Command(name, args...)
|
cmd := exec.Command(name, args...)
|
||||||
|
batch(cmd)
|
||||||
hide(cmd)
|
hide(cmd)
|
||||||
return cmd
|
return cmd
|
||||||
}
|
}
|
||||||
@@ -36,6 +37,7 @@ func CommandContext(ctx context.Context, name string, args ...string) *exec.Cmd
|
|||||||
// Output waited for it however long it took (#123) — its pipes are
|
// Output waited for it however long it took (#123) — its pipes are
|
||||||
// closed a moment after instead
|
// closed a moment after instead
|
||||||
cmd.WaitDelay = waitDelay
|
cmd.WaitDelay = waitDelay
|
||||||
|
batch(cmd)
|
||||||
hide(cmd)
|
hide(cmd)
|
||||||
return cmd
|
return cmd
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -6,3 +6,6 @@ import "os/exec"
|
|||||||
|
|
||||||
// hide has nothing to do: only Windows gives a child a window of its own.
|
// hide has nothing to do: only Windows gives a child a window of its own.
|
||||||
func hide(*exec.Cmd) {}
|
func hide(*exec.Cmd) {}
|
||||||
|
|
||||||
|
// batch has nothing to do: only Windows runs a script through cmd.exe.
|
||||||
|
func batch(*exec.Cmd) {}
|
||||||
@@ -2,6 +2,13 @@ package proc
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
|
"os/exec"
|
||||||
|
"path/filepath"
|
||||||
|
"slices"
|
||||||
|
"strings"
|
||||||
"testing"
|
"testing"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -25,3 +32,55 @@ func TestHideWithoutConsole(t *testing.T) {
|
|||||||
t.Fatalf("with a console the child should share it: %+v", cmd.SysProcAttr)
|
t.Fatalf("with a console the child should share it: %+v", cmd.SysProcAttr)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// A .cmd in a folder with a space, given an argument with a space, hands
|
||||||
|
// the program it runs the arguments as given: npm.cmd under nvm-windows'
|
||||||
|
// C:\Program Files\nodejs, installing into that same folder, had cmd.exe
|
||||||
|
// run "C:\Program" instead (wztlink1013 on Discord). The script passes %*
|
||||||
|
// on, as npm.cmd does to node, to this test binary, which prints what it
|
||||||
|
// was given. A " in an argument isn't among them: it comes back doubled to
|
||||||
|
// Go's parser, which reads "" the way C runtimes did before 2008.
|
||||||
|
func TestBatchFileInFolderWithSpace(t *testing.T) {
|
||||||
|
if os.Getenv("MAGPIE_PROC_ECHO") == "1" {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
self, err := os.Executable()
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
dir := filepath.Join(t.TempDir(), "Program Files", "nodejs")
|
||||||
|
if err := os.MkdirAll(dir, 0o755); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
script := "@echo off\r\n\"" + self + "\" -test.run=TestEchoArgs -- %*\r\n"
|
||||||
|
for _, name := range []string{"npm.cmd", "tool.bat"} {
|
||||||
|
if err := os.WriteFile(filepath.Join(dir, name), []byte(script), 0o755); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
args := []string{"install", "-g", "--prefix", dir, "@anthropic-ai/claude-code@latest",
|
||||||
|
"x&y", "a b|c", "%PATH%", `C:\a dir\`, `C:\b\`, "", "^(<>)!,;=", "中文 路径"}
|
||||||
|
want, _ := json.Marshal(args)
|
||||||
|
t.Setenv("MAGPIE_PROC_ECHO", "1")
|
||||||
|
t.Setenv("PATH", dir+string(os.PathListSeparator)+os.Getenv("PATH"))
|
||||||
|
for _, name := range []string{filepath.Join(dir, "npm.cmd"), filepath.Join(dir, "tool.bat"), "npm"} {
|
||||||
|
for _, cmd := range []*exec.Cmd{Command(name, args...), CommandContext(context.Background(), name, args...)} {
|
||||||
|
out, err := cmd.CombinedOutput()
|
||||||
|
got, _, _ := strings.Cut(string(out), "\n")
|
||||||
|
if err != nil || got != string(want) {
|
||||||
|
t.Errorf("%s: got %q, %v; want %s", name, out, err, want)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestEchoArgs is the program TestBatchFileInFolderWithSpace's scripts run.
|
||||||
|
func TestEchoArgs(t *testing.T) {
|
||||||
|
if os.Getenv("MAGPIE_PROC_ECHO") != "1" {
|
||||||
|
t.Skip("run by TestBatchFileInFolderWithSpace")
|
||||||
|
}
|
||||||
|
i := slices.Index(os.Args, "--")
|
||||||
|
b, _ := json.Marshal(append([]string{}, os.Args[i+1:]...))
|
||||||
|
fmt.Fprintf(os.Stdout, "%s\n", b)
|
||||||
|
os.Exit(0)
|
||||||
|
}
|
||||||
Reference in new issue
Block a user