Ophestra
e599b5583d
This removes the requirement to call fmsg.Exit on every exit path, and enables direct use of the "log" package. However, fmsg.BeforeExit is still encouraged when possible to catch exit on suspended output. Signed-off-by: Ophestra <cat@gensokyo.uk>
108 lines
2.2 KiB
Go
108 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"io"
|
|
"log"
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
"syscall"
|
|
|
|
"git.gensokyo.uk/security/fortify/fst"
|
|
"git.gensokyo.uk/security/fortify/internal/fmsg"
|
|
"git.gensokyo.uk/security/fortify/internal/state"
|
|
)
|
|
|
|
func tryPath(name string) (config *fst.Config) {
|
|
var r io.Reader
|
|
config = new(fst.Config)
|
|
|
|
if name != "-" {
|
|
r = tryFd(name)
|
|
if r == nil {
|
|
fmsg.Verbose("load configuration from file")
|
|
|
|
if f, err := os.Open(name); err != nil {
|
|
log.Fatalf("cannot access configuration file %q: %s", name, err)
|
|
} else {
|
|
// finalizer closes f
|
|
r = f
|
|
}
|
|
} else {
|
|
defer func() {
|
|
if err := r.(io.ReadCloser).Close(); err != nil {
|
|
log.Printf("cannot close config fd: %v", err)
|
|
}
|
|
}()
|
|
}
|
|
} else {
|
|
r = os.Stdin
|
|
}
|
|
|
|
if err := json.NewDecoder(r).Decode(&config); err != nil {
|
|
log.Fatalf("cannot load configuration: %v", err)
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
func tryFd(name string) io.ReadCloser {
|
|
if v, err := strconv.Atoi(name); err != nil {
|
|
fmsg.Verbosef("name cannot be interpreted as int64: %v", err)
|
|
return nil
|
|
} else {
|
|
fd := uintptr(v)
|
|
if _, _, errno := syscall.Syscall(syscall.SYS_FCNTL, fd, syscall.F_GETFD, 0); errno != 0 {
|
|
if errors.Is(errno, syscall.EBADF) {
|
|
return nil
|
|
}
|
|
log.Fatalf("cannot get fd %d: %v", fd, errno)
|
|
}
|
|
return os.NewFile(fd, strconv.Itoa(v))
|
|
}
|
|
}
|
|
|
|
func tryShort(name string) (config *fst.Config, instance *state.State) {
|
|
likePrefix := false
|
|
if len(name) <= 32 {
|
|
likePrefix = true
|
|
for _, c := range name {
|
|
if c >= '0' && c <= '9' {
|
|
continue
|
|
}
|
|
if c >= 'a' && c <= 'f' {
|
|
continue
|
|
}
|
|
likePrefix = false
|
|
break
|
|
}
|
|
}
|
|
|
|
// try to match from state store
|
|
if likePrefix && len(name) >= 8 {
|
|
fmsg.Verbose("argument looks like prefix")
|
|
|
|
s := state.NewMulti(sys.Paths().RunDirPath)
|
|
if entries, err := state.Join(s); err != nil {
|
|
log.Printf("cannot join store: %v", err)
|
|
// drop to fetch from file
|
|
} else {
|
|
for id := range entries {
|
|
v := id.String()
|
|
if strings.HasPrefix(v, name) {
|
|
// match, use config from this state entry
|
|
instance = entries[id]
|
|
config = instance.Config
|
|
break
|
|
}
|
|
|
|
fmsg.Verbosef("instance %s skipped", v)
|
|
}
|
|
}
|
|
}
|
|
|
|
return
|
|
}
|