version: add CmdName func for future use by logpolicy

Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>

Change-Id: I02a7c907844f71242ef06ed097f2a92ece7ae091
This commit is contained in:
Brad Fitzpatrick
2020-02-19 11:22:12 -08:00
parent cf1e386cbd
commit f266e2d1eb
3 changed files with 44 additions and 0 deletions

41
version/cmdname.go Normal file
View File

@@ -0,0 +1,41 @@
// Copyright (c) 2020 Tailscale Inc & AUTHORS All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package version
import (
"os"
"path"
"strings"
"rsc.io/goversion/version"
)
// CmdName returns either the base name of the current binary
// using os.Executable. If os.Executable fails (it sholdn't), then
// "cmd" is returned.
func CmdName() string {
e, err := os.Executable()
if err != nil {
return "cmd"
}
var ret string
v, err := version.ReadExe(e)
if err != nil {
ret = strings.TrimSuffix(strings.ToLower(e), ".exe")
} else {
// v is like:
// "path\ttailscale.com/cmd/tailscale\nmod\ttailscale.com\t(devel)\t\ndep\tgithub.com/apenwarr/fixconsole\tv0.0.0-20191012055117-5a9f6489cc29\th1:muXWUcay7DDy1/hEQWrYlBy+g0EuwT70sBHg65SeUc4=\ndep\tgithub....
for _, line := range strings.Split(v.ModuleInfo, "\n") {
if strings.HasPrefix(line, "path\t") {
ret = path.Base(strings.TrimPrefix(line, "path\t"))
break
}
}
}
if ret == "" {
return "cmd"
}
return ret
}