forked from security/hakurei
Ophestra
c74c269b66
This is a more reliable form of pathname to self and also cheaper than os.Executable. Signed-off-by: Ophestra <cat@gensokyo.uk>
38 lines
683 B
Go
38 lines
683 B
Go
package container
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"sync"
|
|
|
|
"hakurei.app/message"
|
|
)
|
|
|
|
var (
|
|
executable string
|
|
executableOnce sync.Once
|
|
)
|
|
|
|
func copyExecutable(msg message.Msg) {
|
|
if name, err := os.Executable(); err != nil {
|
|
m := fmt.Sprintf("cannot read executable path: %v", err)
|
|
if msg != nil {
|
|
msg.BeforeExit()
|
|
msg.GetLogger().Fatal(m)
|
|
} else {
|
|
log.Fatal(m)
|
|
}
|
|
} else {
|
|
executable = name
|
|
}
|
|
}
|
|
|
|
// MustExecutable calls [os.Executable] and terminates the process on error.
|
|
//
|
|
// Deprecated: This is no longer used and will be removed in 0.4.
|
|
func MustExecutable(msg message.Msg) string {
|
|
executableOnce.Do(func() { copyExecutable(msg) })
|
|
return executable
|
|
}
|