mptyl/ThothII
1
0
Fork 0
Code Issues 25 Pull Requests Actions Packages Projects Releases 1 Wiki Activity

fix(auth): harden interactive diagnostic lifecycle

Browse Source
This commit is contained in:
marcopan
2026-08-17 17:57:10 +02:00
parent ef244ab56d
commit af16464e68
9 changed files with 576 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tools/tht/internal/compose/runner.go
+66 -16
View File
@@ -55,6 +55,10 @@ type boundedRunner interface {
RunBounded(context.Context, []string, io.Reader, CaptureLimits) (Result, error)
}
type boundedStreamingRunner interface {
RunBoundedStreaming(context.Context, []string, io.Reader, CaptureLimits, func([]byte)) (Result, error)
}
// execRunner executes the Docker CLI. It never invokes a shell.
type execRunner struct {
binary string
@@ -78,10 +82,16 @@ func (r execRunner) Run(ctx context.Context, args []string, stdin io.Reader) (Re
// RunBounded invokes Docker while enforcing both stream limits during capture.
func (r execRunner) RunBounded(ctx context.Context, args []string, stdin io.Reader, limits CaptureLimits) (Result, error) {
return r.runBounded(ctx, args, stdin, limits, true)
return r.runBounded(ctx, args, stdin, limits, true, nil)
}
func (r execRunner) runBounded(ctx context.Context, args []string, stdin io.Reader, limits CaptureLimits, manageOneShot bool) (Result, error) {
// RunBoundedStreaming retains bounded stderr capture while synchronously observing only the
// retained prefix. It is used for the one validated interactive device-flow prompt.
func (r execRunner) RunBoundedStreaming(ctx context.Context, args []string, stdin io.Reader, limits CaptureLimits, stderrObserver func([]byte)) (Result, error) {
return r.runBounded(ctx, args, stdin, limits, true, stderrObserver)
}
func (r execRunner) runBounded(ctx context.Context, args []string, stdin io.Reader, limits CaptureLimits, manageOneShot bool, stderrObserver func([]byte)) (Result, error) {
if err := validCaptureLimits(limits); err != nil {
return Result{}, err
}
@@ -101,8 +111,8 @@ func (r execRunner) runBounded(ctx context.Context, args []string, stdin io.Read
configureProcess(command)
command.Stdin = stdin
overflow := make(chan struct{}, 1)
stdout := newCappedBuffer(limits.StdoutBytes, overflow)
stderr := newCappedBuffer(limits.StderrBytes, overflow)
stdout := newCappedBuffer(limits.StdoutBytes, overflow, nil)
stderr := newCappedBuffer(limits.StderrBytes, overflow, stderrObserver)
command.Stdout = stdout
command.Stderr = stderr
if err := command.Start(); err != nil {
@@ -128,7 +138,7 @@ func (r execRunner) runBounded(ctx context.Context, args []string, stdin io.Read
lifecycleErr = errors.Join(lifecycleErr, ErrProcessReap)
}
}
if interrupted && containerName != "" {
if containerName != "" {
if err := r.cleanupOneShotContainer(containerName); err != nil {
lifecycleErr = errors.Join(lifecycleErr, ErrContainerCleanup)
}
@@ -138,20 +148,27 @@ func (r execRunner) runBounded(ctx context.Context, args []string, stdin io.Read
result.ExitCode = command.ProcessState.ExitCode()
}
if stdout.Overflowed() || stderr.Overflowed() {
return result, errors.Join(ErrOutputLimit, lifecycleErr)
return result, joinLifecycleError(ErrOutputLimit, lifecycleErr)
}
if interrupted {
return result, errors.Join(processErr, lifecycleErr)
return result, joinLifecycleError(processErr, lifecycleErr)
}
if processErr == nil {
return result, nil
return result, lifecycleErr
}
var exitError *exec.ExitError
if errors.As(processErr, &exitError) {
result.ExitCode = exitError.ExitCode()
return result, processErr
return result, joinLifecycleError(processErr, lifecycleErr)
}
return result, processErr
return result, joinLifecycleError(processErr, lifecycleErr)
}
func joinLifecycleError(processErr, lifecycleErr error) error {
if lifecycleErr == nil {
return processErr
}
return errors.Join(processErr, lifecycleErr)
}
// NewOneShotContainerName returns a Docker-safe, cross-process unique name with a bounded prefix.
@@ -253,7 +270,7 @@ func (r execRunner) cleanupOneShotContainer(name string) error {
return nil
}
limits := CaptureLimits{StdoutBytes: cleanupCaptureBytes, StderrBytes: cleanupCaptureBytes}
if _, err := r.runBounded(ctx, []string{"container", "rm", "-f", name}, nil, limits, false); err != nil {
if _, err := r.runBounded(ctx, []string{"container", "rm", "-f", name}, nil, limits, false, nil); err != nil {
return ErrContainerCleanup
}
exists, err = r.oneShotContainerExists(ctx, name)
@@ -267,7 +284,7 @@ func (r execRunner) oneShotContainerExists(ctx context.Context, name string) (bo
limits := CaptureLimits{StdoutBytes: cleanupCaptureBytes, StderrBytes: cleanupCaptureBytes}
result, err := r.runBounded(ctx, []string{
"container", "ls", "--all", "--quiet", "--filter", "name=^/" + name + "$",
}, nil, limits, false)
}, nil, limits, false, nil)
if err != nil {
return false, ErrContainerCleanup
}
@@ -294,6 +311,22 @@ func RunBounded(runner Runner, ctx context.Context, args []string, stdin io.Read
return result, err
}
// RunBoundedStreaming uses the production runner's during-capture observer. Compatibility
// runners are observed only after their already-bounded result returns.
func RunBoundedStreaming(runner Runner, ctx context.Context, args []string, stdin io.Reader, limits CaptureLimits, stderrObserver func([]byte)) (Result, error) {
if err := validCaptureLimits(limits); err != nil {
return Result{}, err
}
if streaming, ok := runner.(boundedStreamingRunner); ok {
return streaming.RunBoundedStreaming(ctx, args, stdin, limits, stderrObserver)
}
result, err := RunBounded(runner, ctx, args, stdin, limits)
if stderrObserver != nil && result.Stderr != "" {
stderrObserver([]byte(result.Stderr))
}
return result, err
}
func validCaptureLimits(limits CaptureLimits) error {
if limits.StdoutBytes < 1 || limits.StderrBytes < 1 ||
limits.StdoutBytes > maximumCaptureBytes || limits.StderrBytes > maximumCaptureBytes {
@@ -323,23 +356,24 @@ type cappedBuffer struct {
contents []byte
maximum int
overflow chan<- struct{}
observer func([]byte)
exceeded bool
}
func newCappedBuffer(maximum int, overflow chan<- struct{}) *cappedBuffer {
func newCappedBuffer(maximum int, overflow chan<- struct{}, observer func([]byte)) *cappedBuffer {
capacity := maximum
if capacity > 4096 {
capacity = 4096
}
return &cappedBuffer{contents: make([]byte, 0, capacity), maximum: maximum, overflow: overflow}
return &cappedBuffer{contents: make([]byte, 0, capacity), maximum: maximum, overflow: overflow, observer: observer}
}
func (b *cappedBuffer) Write(value []byte) (int, error) {
b.mu.Lock()
defer b.mu.Unlock()
remaining := b.maximum - len(b.contents)
kept := 0
if remaining > 0 {
kept := len(value)
kept = len(value)
if kept > remaining {
kept = remaining
}
@@ -352,6 +386,14 @@ func (b *cappedBuffer) Write(value []byte) (int, error) {
default:
}
}
var observed []byte
if kept > 0 && b.observer != nil {
observed = append([]byte(nil), value[:kept]...)
}
b.mu.Unlock()
if len(observed) > 0 {
b.observer(observed)
}
return len(value), nil
}
@@ -397,3 +439,11 @@ func (r InstallationRunner) RunBounded(ctx context.Context, args []string, stdin
}
return RunBounded(r.Runner, ctx, args, stdin, limits)
}
// RunBoundedStreaming preserves the stderr observer across installation argument injection.
func (r InstallationRunner) RunBoundedStreaming(ctx context.Context, args []string, stdin io.Reader, limits CaptureLimits, stderrObserver func([]byte)) (Result, error) {
if len(args) > 0 && args[0] == "compose" {
return RunBoundedStreaming(r.Runner, ctx, r.Installation.ComposeArgs(args[1:]...), stdin, limits, stderrObserver)
}
return RunBoundedStreaming(r.Runner, ctx, args, stdin, limits, stderrObserver)
}