forked from TrueCloudLab/distribution
64 lines
1.4 KiB
Go
64 lines
1.4 KiB
Go
// Copyright 2012 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
// +build darwin freebsd
|
|
|
|
package osext
|
|
|
|
import (
|
|
"os"
|
|
"runtime"
|
|
"syscall"
|
|
"unsafe"
|
|
)
|
|
|
|
var startUpcwd, getwdError = os.Getwd()
|
|
|
|
func executable() (string, error) {
|
|
var mib [4]int32
|
|
switch runtime.GOOS {
|
|
case "freebsd":
|
|
mib = [4]int32{1 /* CTL_KERN */, 14 /* KERN_PROC */, 12 /* KERN_PROC_PATHNAME */, -1}
|
|
case "darwin":
|
|
mib = [4]int32{1 /* CTL_KERN */, 38 /* KERN_PROCARGS */, int32(os.Getpid()), -1}
|
|
}
|
|
|
|
n := uintptr(0)
|
|
// get length
|
|
_, _, err := syscall.Syscall6(syscall.SYS___SYSCTL, uintptr(unsafe.Pointer(&mib[0])), 4, 0, uintptr(unsafe.Pointer(&n)), 0, 0)
|
|
if err != 0 {
|
|
return "", err
|
|
}
|
|
if n == 0 { // shouldn't happen
|
|
return "", nil
|
|
}
|
|
buf := make([]byte, n)
|
|
_, _, err = syscall.Syscall6(syscall.SYS___SYSCTL, uintptr(unsafe.Pointer(&mib[0])), 4, uintptr(unsafe.Pointer(&buf[0])), uintptr(unsafe.Pointer(&n)), 0, 0)
|
|
if err != 0 {
|
|
return "", err
|
|
}
|
|
if n == 0 { // shouldn't happen
|
|
return "", nil
|
|
}
|
|
for i, v := range buf {
|
|
if v == 0 {
|
|
buf = buf[:i]
|
|
break
|
|
}
|
|
}
|
|
if buf[0] != '/' {
|
|
if getwdError != nil {
|
|
return string(buf), getwdError
|
|
} else {
|
|
if buf[0] == '.' {
|
|
buf = buf[1:]
|
|
}
|
|
if startUpcwd[len(startUpcwd)-1] != '/' {
|
|
return startUpcwd + "/" + string(buf), nil
|
|
}
|
|
return startUpcwd + string(buf), nil
|
|
}
|
|
}
|
|
return string(buf), nil
|
|
}
|