diff options
| author | Ophestra <cat@gensokyo.uk> | 2025-06-25 03:59:52 +0900 |
|---|---|---|
| committer | Ophestra <cat@gensokyo.uk> | 2025-06-25 04:57:41 +0900 |
| commit | 87e008d56de974947ebb99c2cc40b25d3c2cf43e (patch) | |
| tree | 31791911e5226d6ec04e3fac7d91b0bf53e63aa5 /internal/fmsg/fmsg.go | |
| parent | 399207321265307bb15f37d867f9370cd51c82a8 (diff) | |
treewide: rename to hakurei
Fortify makes little sense for a container tool.
Signed-off-by: Ophestra <cat@gensokyo.uk>
Diffstat (limited to 'internal/fmsg/fmsg.go')
| -rw-r--r-- | internal/fmsg/fmsg.go | 86 |
1 files changed, 0 insertions, 86 deletions
diff --git a/internal/fmsg/fmsg.go b/internal/fmsg/fmsg.go deleted file mode 100644 index 48538380..00000000 --- a/internal/fmsg/fmsg.go +++ /dev/null @@ -1,86 +0,0 @@ -// Package fmsg provides various functions for output messages. -package fmsg - -import ( - "bytes" - "io" - "log" - "os" - "sync" - "sync/atomic" - "syscall" -) - -const ( - bufSize = 4 * 1024 - bufSizeMax = 16 * 1024 * 1024 -) - -var o = &suspendable{w: os.Stderr} - -// Prepare configures the system logger for [Suspend] and [Resume] to take effect. -func Prepare(prefix string) { log.SetPrefix(prefix + ": "); log.SetFlags(0); log.SetOutput(o) } - -type suspendable struct { - w io.Writer - s atomic.Bool - - buf bytes.Buffer - bufOnce sync.Once - bufMu sync.Mutex - dropped int -} - -func (s *suspendable) Write(p []byte) (n int, err error) { - if !s.s.Load() { - return s.w.Write(p) - } - s.bufOnce.Do(func() { s.prepareBuf() }) - - s.bufMu.Lock() - defer s.bufMu.Unlock() - - if l := len(p); s.buf.Len()+l > bufSizeMax { - s.dropped += l - return 0, syscall.ENOMEM - } - return s.buf.Write(p) -} - -func (s *suspendable) prepareBuf() { s.buf.Grow(bufSize) } -func (s *suspendable) Suspend() bool { return o.s.CompareAndSwap(false, true) } -func (s *suspendable) Resume() (resumed bool, dropped uintptr, n int64, err error) { - if o.s.CompareAndSwap(true, false) { - o.bufMu.Lock() - defer o.bufMu.Unlock() - - resumed = true - dropped = uintptr(o.dropped) - - o.dropped = 0 - n, err = io.Copy(s.w, &s.buf) - s.buf = bytes.Buffer{} - s.prepareBuf() - } - return -} - -func Suspend() bool { return o.Suspend() } -func Resume() bool { - resumed, dropped, _, err := o.Resume() - if err != nil { - // probably going to result in an error as well, - // so this call is as good as unreachable - log.Printf("cannot dump buffer on resume: %v", err) - } - if resumed && dropped > 0 { - log.Fatalf("dropped %d bytes while output is suspended", dropped) - } - return resumed -} - -func BeforeExit() { - if Resume() { - log.Printf("beforeExit reached on suspended output") - } -} |
