File size: 2,807 Bytes
e36aeda | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 | // Copyright 2023 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
//go:build windows
package os_test
import (
"internal/testenv"
"io"
. "os"
"path/filepath"
"sync"
"syscall"
"testing"
"time"
)
func TestRemoveAllWithExecutedProcess(t *testing.T) {
// Regression test for golang.org/issue/25965.
if testing.Short() {
t.Skip("slow test; skipping")
}
testenv.MustHaveExec(t)
name, err := Executable()
if err != nil {
t.Fatal(err)
}
r, err := Open(name)
if err != nil {
t.Fatal(err)
}
defer r.Close()
const n = 100
var execs [n]string
// First create n executables.
for i := 0; i < n; i++ {
// Rewind r.
if _, err := r.Seek(0, io.SeekStart); err != nil {
t.Fatal(err)
}
name := filepath.Join(t.TempDir(), "test.exe")
execs[i] = name
w, err := Create(name)
if err != nil {
t.Fatal(err)
}
if _, err = io.Copy(w, r); err != nil {
w.Close()
t.Fatal(err)
}
if err := w.Sync(); err != nil {
w.Close()
t.Fatal(err)
}
if err = w.Close(); err != nil {
t.Fatal(err)
}
}
// Then run each executable and remove its directory.
// Run each executable in a separate goroutine to add some load
// and increase the chance of triggering the bug.
var wg sync.WaitGroup
wg.Add(n)
for i := 0; i < n; i++ {
go func(i int) {
defer wg.Done()
name := execs[i]
dir := filepath.Dir(name)
// Run test.exe without executing any test, just to make it do something.
cmd := testenv.Command(t, name, "-test.run=^$")
if err := cmd.Run(); err != nil {
t.Errorf("exec failed: %v", err)
}
// Remove dir and check that it doesn't return `ERROR_ACCESS_DENIED`.
err = RemoveAll(dir)
if err != nil {
t.Errorf("RemoveAll failed: %v", err)
}
}(i)
}
wg.Wait()
}
func TestProcessWithHandleWindows(t *testing.T) {
const envVar = "OSTEST_PROCESS_WITH_HANDLE"
if Getenv(envVar) != "" {
time.Sleep(1 * time.Minute)
return
}
cmd := testenv.CommandContext(t, t.Context(), testenv.Executable(t), "-test.run=^"+t.Name()+"$")
cmd = testenv.CleanCmdEnv(cmd)
cmd.Env = append(cmd.Env, envVar+"=1")
if err := cmd.Start(); err != nil {
t.Fatal(err)
}
defer func() {
cmd.Process.Kill()
cmd.Wait()
}()
called := false
err := cmd.Process.WithHandle(func(handle uintptr) {
called = true
// Check the handle is valid.
var u syscall.Rusage
e := syscall.GetProcessTimes(syscall.Handle(handle), &u.CreationTime, &u.ExitTime, &u.KernelTime, &u.UserTime)
if e != nil {
t.Errorf("Using process handle failed: %v", NewSyscallError("GetProcessTimes", e))
}
})
if err != nil {
t.Fatalf("WithHandle: got error %v, want nil", err)
}
if !called {
t.Fatal("WithHandle did not call the callback function")
}
}
|