diff options
Diffstat (limited to 'cmd/mbf/daemon.go')
| -rw-r--r-- | cmd/mbf/daemon.go | 34 |
1 files changed, 21 insertions, 13 deletions
diff --git a/cmd/mbf/daemon.go b/cmd/mbf/daemon.go index d64643d2..4a5f7d6d 100644 --- a/cmd/mbf/daemon.go +++ b/cmd/mbf/daemon.go @@ -35,21 +35,10 @@ const ( // cureFromIR services an IR curing request. func cureFromIR( - ctx context.Context, cache *pkg.Cache, conn net.Conn, + flags uint64, ) (pkg.Artifact, error) { - go func() { <-ctx.Done(); _ = conn.SetDeadline(time.Now()) }() - if err := conn.SetReadDeadline(daemonDeadline()); err != nil { - return nil, errors.Join(err, conn.Close()) - } - - var flagsWire [8]byte - if _, err := io.ReadFull(conn, flagsWire[:]); err != nil { - return nil, errors.Join(err, conn.Close()) - } - flags := binary.LittleEndian.Uint64(flagsWire[:]) - a, decodeErr := cache.NewDecoder(conn).Decode() if decodeErr != nil { _, err := conn.Write([]byte("\x00" + decodeErr.Error())) @@ -105,7 +94,26 @@ func serve( continue } wg.Go(func() { - if a, _err := cureFromIR(ctx, cm.c, conn); _err != nil { + go func() { <-ctx.Done(); _ = conn.SetDeadline(time.Now()) }() + if _err := conn.SetReadDeadline(daemonDeadline()); _err != nil { + log.Println(_err) + if _err = conn.Close(); _err != nil { + log.Println(_err) + } + return + } + + var flagsWire [8]byte + if _, _err := io.ReadFull(conn, flagsWire[:]); _err != nil { + log.Println(_err) + if _err = conn.Close(); _err != nil { + log.Println(_err) + } + return + } + flags := binary.LittleEndian.Uint64(flagsWire[:]) + + if a, _err := cureFromIR(cm.c, conn, flags); _err != nil { log.Println(_err) } else { log.Printf( |