security/hakurei
6
3
Fork 0
Code Issues 11 Pull Requests Actions Packages Projects Releases 11 Wiki Activity

internal/pkg: implement http artifact
All checks were successful
Test / Create distribution (push) Successful in 44s
Details
Test / Sandbox (push) Successful in 2m29s
Details
Test / ShareFS (push) Successful in 3m39s
Details
Test / Hpkg (push) Successful in 4m30s
Details
Test / Sandbox (race detector) (push) Successful in 4m53s
Details
Test / Hakurei (push) Successful in 2m29s
Details
Test / Hakurei (race detector) (push) Successful in 3m14s
Details
Test / Flake checks (push) Successful in 1m44s
Details

Browse Source 
This is useful for downloading source tarballs from the internet.

Signed-off-by: Ophestra <cat@gensokyo.uk>
This commit is contained in:
Ophestra
2026-01-03 15:26:59 +09:00
parent dd5d792d14
commit ce249d23f1
4 changed files with 343 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

154
internal/pkg/net_test.go Normal file
View File

@@ -0,0 +1,154 @@
package pkg_test
import (
"crypto/sha512"
"encoding/base64"
"net/http"
"reflect"
"testing"
"testing/fstest"
"hakurei.app/container/check"
"hakurei.app/internal/pkg"
)
func TestHTTP(t *testing.T) {
t.Parallel()
const testdata = "\x7f\xe1\x69\xa2\xdd\x63\x96\x26\x83\x79\x61\x8b\xf0\x3f\xd5\x16\x9a\x39\x3a\xdb\xcf\xb1\xbc\x8d\x33\xff\x75\xee\x62\x56\xa9\xf0\x27\xac\x13\x94\x69"
testdataChecksum := func() pkg.Checksum {
h := sha512.New384()
h.Write([]byte(testdata))
return (pkg.Checksum)(h.Sum(nil))
}()
testdataChecksumString := base64.URLEncoding.EncodeToString(testdataChecksum[:])
var transport http.Transport
client := http.Client{Transport: &transport}
transport.RegisterProtocol("file", http.NewFileTransportFS(fstest.MapFS{
"testdata": {Data: []byte(testdata), Mode: 0400},
}))
checkWithCache(t, []cacheTestCase{
{"direct", nil, func(t *testing.T, base *check.Absolute, c *pkg.Cache) {
var got []byte
if f, err := c.NewHTTPGet(&client, "file:///testdata", testdataChecksum); err != nil {
t.Fatalf("NewHTTPGet: error = %v", err)
} else if got, err = f.Data(); err != nil {
t.Fatalf("Data: error = %v", err)
} else if string(got) != testdata {
t.Fatalf("Data: %x, want %x", got, testdata)
} else if gotIdent := f.ID(); gotIdent != testdataChecksum {
t.Fatalf("ID: %x, want %x", gotIdent, testdataChecksum)
}
// check direct validation
wantErrMismatch := &pkg.ChecksumMismatchError{
Got: testdataChecksum,
}
if f, err := c.NewHTTPGet(&client, "file:///testdata", pkg.Checksum{}); err != nil {
t.Fatalf("NewHTTPGet: error = %v", err)
} else if _, err = f.Data(); !reflect.DeepEqual(err, wantErrMismatch) {
t.Fatalf("Data: error = %#v, want %#v", err, wantErrMismatch)
} else if gotIdent := f.ID(); gotIdent != (pkg.Checksum{}) {
t.Fatalf("ID: %x, want %x", gotIdent, pkg.Checksum{})
}
// check direct response error
wantErrNotFound := pkg.ResponseStatusError(http.StatusNotFound)
if f, err := c.NewHTTPGet(&client, "file:///nonexistent", pkg.Checksum{}); err != nil {
t.Fatalf("NewHTTPGet: error = %v", err)
} else if _, err = f.Data(); !reflect.DeepEqual(err, wantErrNotFound) {
t.Fatalf("Data: error = %#v, want %#v", err, wantErrNotFound)
} else if gotIdent := f.ID(); gotIdent != (pkg.Checksum{}) {
t.Fatalf("ID: %x, want %x", gotIdent, pkg.Checksum{})
}
}, pkg.MustDecode("ANVz3GwS4oTcFTOjbc-n_N6MtycCtkELMBJB0ohuRz02PtmWZEJF8v3I51DtM0CY")},
{"load or store", nil, func(t *testing.T, base *check.Absolute, c *pkg.Cache) {
f, err := c.NewHTTPGet(&client, "file:///testdata", testdataChecksum)
if err != nil {
t.Fatalf("NewHTTPGet: error = %v", err)
}
wantPathname := base.Append(
"identifier",
testdataChecksumString,
)
var pathname *check.Absolute
if pathname, err = f.Pathname(); err != nil {
t.Fatalf("Pathname: error = %v", err)
} else if !pathname.Is(wantPathname) {
t.Fatalf("Pathname: %q, want %q", pathname, wantPathname)
}
var checksum pkg.Checksum
if checksum, err = f.Hash(); err != nil {
t.Fatalf("Hash: error = %v", err)
} else if checksum != testdataChecksum {
t.Fatalf("Hash: %x, want %x", checksum, testdataChecksum)
}
var got []byte
if got, err = f.Data(); err != nil {
t.Fatalf("Data: error = %v", err)
} else if string(got) != testdata {
t.Fatalf("Data: %x, want %x", got, testdata)
} else if gotIdent := f.ID(); gotIdent != testdataChecksum {
t.Fatalf("ID: %x, want %x", gotIdent, testdataChecksum)
}
// check load from cache
if f, err = c.NewHTTPGet(&client, "file:///testdata", testdataChecksum); err != nil {
t.Fatalf("NewHTTPGet: error = %v", err)
} else if got, err = f.Data(); err != nil {
t.Fatalf("Data: error = %v", err)
} else if string(got) != testdata {
t.Fatalf("Data: %x, want %x", got, testdata)
} else if gotIdent := f.ID(); gotIdent != testdataChecksum {
t.Fatalf("ID: %x, want %x", gotIdent, testdataChecksum)
}
// check error passthrough
wantErrNotFound := pkg.ResponseStatusError(http.StatusNotFound)
if f, err = c.NewHTTPGet(&client, "file:///nonexistent", pkg.Checksum{}); err != nil {
t.Fatalf("NewHTTPGet: error = %v", err)
} else if _, err = f.Pathname(); !reflect.DeepEqual(err, wantErrNotFound) {
t.Fatalf("Pathname: error = %#v, want %#v", err, wantErrNotFound)
} else if gotIdent := f.ID(); gotIdent != (pkg.Checksum{}) {
t.Fatalf("ID: %x, want %x", gotIdent, pkg.Checksum{})
}
}, pkg.MustDecode("5ns3Ky8-n_pETpwO3UYA88FKKLins6kxtgRQBEfSiGIpZXu6QCBOW2ukm-nWnUwC")},
{"store", nil, func(t *testing.T, base *check.Absolute, c *pkg.Cache) {
var (
got []byte
pathname *check.Absolute
checksum pkg.Checksum
)
wantPathname := base.Append(
"identifier",
testdataChecksumString,
)
if f, err := c.NewHTTPGet(&client, "file:///testdata", testdataChecksum); err != nil {
t.Fatalf("NewHTTPGet: error = %v", err)
} else if got, err = f.Data(); err != nil {
t.Fatalf("Data: error = %v", err)
} else if string(got) != testdata {
t.Fatalf("Data: %x, want %x", got, testdata)
} else if gotIdent := f.ID(); gotIdent != testdataChecksum {
t.Fatalf("ID: %x, want %x", gotIdent, testdataChecksum)
} else if pathname, err = f.Pathname(); err != nil {
t.Fatalf("Pathname: error = %v", err)
} else if !pathname.Is(wantPathname) {
t.Fatalf("Pathname: %q, want %q", pathname, wantPathname)
} else if checksum, err = f.Hash(); err != nil {
t.Fatalf("Hash: error = %v", err)
} else if checksum != testdataChecksum {
t.Fatalf("Hash: %x, want %x", checksum, testdataChecksum)
}
}, pkg.MustDecode("5ns3Ky8-n_pETpwO3UYA88FKKLins6kxtgRQBEfSiGIpZXu6QCBOW2ukm-nWnUwC")},
})
}