Function QueryWorkingSet

pkg/util/va/ws.go:269–322

QueryWorkingSet returns working set information for a set of addresses.

(proc windows.Handle, ws []sys.MemoryWorkingSetExInformation)

// QueryWorkingSet returns working set information for a set of addresses.
func QueryWorkingSet(proc windows.Handle, ws []sys.MemoryWorkingSetExInformation) []sys.MemoryWorkingSetExInformation {
	poolOnce.Do(func() {
		var err error
		p, err = newPool(poolSize())
		if err != nil {
			log.Errorf("unable to create working set pool: %v", err)
		}
	})

	if p == nil {
		return nil
	}

	// acquire a worker and don't wait forever if pool is exhausted
	w := p.acquire(50 * time.Millisecond)
	if w == nil {
		return nil // pool exhausted
	}

	// post work to the thread
	w.submit(proc, ws)

	// wait for completion
	start := time.Now()
	wait, err := w.wait()
	if err != nil {
		return nil
	}

	switch wait {
	case windows.WAIT_OBJECT_0:
		workingsetOpsCount.Add(1)
		// feed successful durations back
		w.jitter.record(time.Since(start))
		defer p.release(w, false)

		if atomic.LoadUint32(&w.ctx.done) == 0 {
			return nil
		}
		return w.ctx.ws

	case sys.WaitTimeout:
		workingsetTimeoutCount.Add(1)
		// thread is stalled inside the kernel syscall.
		// Safe to terminate: no user-mode locks held,
		// no Go runtime state to corrupt.
		p.release(w, true)
		return nil

	default:
		p.release(w, true)
		return nil
	}
}