m3_windows.go

//go:build windows
// +build windows

package shell

import (
	"bufio"
	"bytes"
	"errors"
	"strconv"
	"strings"

	"shell/config"
)

func GetProcessIds(tokens config.ProcessTokens, excludes config.ProcessTokens) (pids map[int]string, err error) {
	arg := "Name != 'WMIC.exe'"
	command, err := M3PS.addDynamicArg(arg)
	if err != nil {
		return
	}
	output, err := CommandCombinedOutput(command)
	if err != nil {
		return
	}
	scanner := bufio.NewScanner(bytes.NewReader(output))
	pids = make(map[int]string)
	if !scanner.Scan() {
		err = errors.New("no line found in the commandline result")
		return
	}
	header := scanner.Text()
	header = strings.TrimSpace(header)
	indexName := strings.Index(header, "Name")
	if indexName < 0 {
		err = errors.New("name in the header line")
		return
	}

Next:
	for scanner.Scan() {
		line := scanner.Text()
		line = strings.TrimSpace(line)
		for _, exclude := range excludes {
			p := strings.Index(line, string(exclude))
			if p >= 0 {
				continue Next
			}
		}
		for _, t := range tokens {
			token := string(t)
			var appName string
			index := strings.Index(token, "$")
			if index >= 0 {
				appName = token[index+1:]
				token = token[:index]
			}

			p := strings.Index(line, token)
			if p >= 0 {
				line = line[indexName:]
				columns := strings.Split(line, " ")
				var col []string
				for _, column := range columns {
					if len(column) <= 0 {
						continue
					}
					col = append(col, column)
					if len(col) <= 1 {
						continue
					}
					id := strings.TrimSpace(col[1])
					pid, err := strconv.Atoi(id)
					if err != nil {
						continue Next
					}
					tokenPid, err := strconv.Atoi(token)
					if err == nil {
						if tokenPid != pid {
							continue Next
						}
					}
					if _, ok := pids[pid]; !ok {
						pids[pid] = appName
					}
					continue Next
				}
			}
		}
	}
	return
}