github.com/google/syzkaller@v0.0.0-20240517125934-c0f1611a36d6/pkg/osutil/fileutil_test.go

// Copyright 2015 syzkaller project authors. All rights reserved.
// Use of this source code is governed by Apache 2 LICENSE that can be found in the LICENSE file.

package osutil

import (
	"fmt"
	"path/filepath"
	"strconv"
	"sync"
	"testing"
)

func TestProcessTempDir(t *testing.T) {
	for try := 0; try < 10; try++ {
		func() {
			tmp := t.TempDir()
			const P = 16
			// Pre-create half of the instances with stale pid.
			var dirs []string
			for i := 0; i < P/2; i++ {
				dir, err := ProcessTempDir(tmp)
				if err != nil {
					t.Fatalf("failed to create process temp dir")
				}
				dirs = append(dirs, dir)
			}
			for _, dir := range dirs {
				if err := WriteFile(filepath.Join(dir, ".pid"), []byte(strconv.Itoa(999999999))); err != nil {
					t.Fatalf("failed to write pid file: %v", err)
				}
			}
			// Now request a bunch of instances concurrently.
			done := make(chan error)
			allDirs := make(map[string]bool)
			var mu sync.Mutex
			for p := 0; p < P; p++ {
				go func() {
					dir, err := ProcessTempDir(tmp)
					if err != nil {
						done <- fmt.Errorf("failed to create temp dir: %w", err)
						return
					}
					mu.Lock()
					present := allDirs[dir]
					allDirs[dir] = true
					mu.Unlock()
					if present {
						done <- fmt.Errorf("duplicate dir %v", dir)
						return
					}
					done <- nil
				}()
			}
			for p := 0; p < P; p++ {
				if err := <-done; err != nil {
					t.Error(err)
				}
			}
		}()
	}
}