gops/gops.go

package main

import (
	"fmt"
	"log"

	"hello/gops/internal/objfile"

	ps "github.com/keybase/go-ps"
)

func main() {
	pss, err := ps.Processes()
	if err != nil {
		log.Fatal(err)
	}

	var undetermined int
	for _, pr := range pss {
		name, err := pr.Path()
		if err != nil {
			undetermined++
			continue
		}

		ok, err := isGo(name)
		if err != nil {
			undetermined++
			continue
		}

		if ok {
			fmt.Printf("%d\t%v\n", pr.Pid(), pr.Executable())
		}
	}
}

func isGo(filename string) (ok bool, err error) {
	obj, err := objfile.Open(filename)
	if err != nil {
		return false, err
	}
	defer obj.Close()

	symbols, err := obj.Symbols()
	if err != nil {
		return false, err
	}

	// TODO(jbd): find a faster way to determine Go programs
	for _, s := range symbols {
		if s.Name == "runtime.buildVersion" {
			return true, nil
		}
	}
	return false, nil
}