Salastil/cerberus
1
0
Fork 0
mirror of https://gitgud.io/yats/cerberus.git synced 2026-05-02 02:32:18 -04:00
Code Issues Packages Projects Releases Wiki Activity

Fix resp.Body not closed when >1 steps

Browse Source
This commit is contained in:
y a t s
2026-02-04 14:52:22 -05:00
parent 6d199a103f
commit 34ceeced8f
2 changed files with 10 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
cerberus_test.go
View File

@@ -42,7 +42,7 @@ func solveTest(ctx context.Context, hc http.Client, host string) error {
if err != nil { if err != nil {
return err return err
} }
log.Printf("Solution hash: %x, nonce: %d, remaining steps: %d\n", s.Hash, s.Nonce, s.Steps) log.Printf("Solution hash: %x, nonce: %d\n", s.Hash, s.Nonce)
resp, err := Submit(ctx, hc, s, "") resp, err := Submit(ctx, hc, s, "")
if err != nil { if err != nil {

16
http.go
View File

@@ -85,21 +85,23 @@ func Submit(ctx context.Context, hc http.Client, s Solution, redirect string) (*
return nil, err return nil, err
} }
if s.Steps > 0 { // This feels gross, but it works.
for s.Steps > 0 {
resp.Body.Close()
c, err := NewChallenge(ctx, hc, s.host.String()) c, err := NewChallenge(ctx, hc, s.host.String())
if err != nil { if err != nil {
return nil, err return nil, err
} }
// maybe useful later. idk. s, err = Solve(ctx, c)
// c.Steps = s.Steps if err != nil {
return nil, err
s, err := Solve(ctx, c) }
resp, err = Submit(ctx, hc, s, redirect)
if err != nil { if err != nil {
return nil, err return nil, err
} }
return Submit(ctx, hc, s, redirect)
} }
return resp, nil return resp, nil