ziggi/ociclient
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

working commit

Browse Source
This commit is contained in:
Олег Бородин
2026-02-26 18:15:10 +02:00
commit 19b173357a
9 changed files with 328 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
authbas.go Normal file
View File

@@ -0,0 +1,24 @@
package client
import (
"encoding/base64"
)
type Authenticator interface {
MakeHeader(user, pass string) (key, value string, err error)
}
type BasicAuthenticator struct {}
func NewBasicAuthenticator() *BasicAuthenticator{
return &BasicAuthenticator{}
}
func (auth *BasicAuthenticator) MakeHeader(user, pass string) (string, string, error){
pair := base64.StdEncoding.EncodeToString([]byte(user + ":" + pass))
return "Autentification", "Basic " + pair, nil
}

32
client.go Normal file
View File

@@ -0,0 +1,32 @@
package client
import (
"crypto/tls"
"net/http"
)
type Client struct {
httpClient *http.Client
authenticator Authenticator
userAgent string
}
func NewClient(skipTLSVerify bool) *Client {
transport := &http.Transport{
TLSClientConfig: &tls.Config{
InsecureSkipVerify: skipTLSVerify,
},
}
httpClient := &http.Client{
Transport: transport,
}
return &Client{
httpClient: httpClient,
userAgent: "ociClient/1.0",
}
}
func (cli *Client) SetAuthenticator(auth Authenticator) {
cli.authenticator = auth
}

50
digest.go Normal file
View File

@@ -0,0 +1,50 @@
package client
import (
"crypto/sha256"
"crypto/sha512"
"encoding/hex"
"strings"
)
func SHA256Digest(src []byte) string {
hasher := sha256.New()
hasher.Write(src)
sum := hasher.Sum(nil)
return "sha256:" + hex.EncodeToString(sum)
}
func SHA512Digest(src []byte) string {
hasher := sha512.New()
hasher.Write(src)
sum := hasher.Sum(nil)
return "sha512:" + hex.EncodeToString(sum)
}
const (
Undefined int = iota
SHA256
SHA512
)
func DigestType(digest string) int {
var err error
var typ int
digest = strings.ToLower(digest)
digest = strings.TrimPrefix(digest, "sha256:")
digest = strings.TrimPrefix(digest, "sha512:")
decoded, err := hex.DecodeString(digest)
if err != nil {
return Undefined
}
switch (len(decoded)) {
case 64:
typ = SHA256
case 128:
typ = SHA512
default:
typ = Undefined
}
return typ
}

62
getman.go Normal file
View File

@@ -0,0 +1,62 @@
package client
import (
"bytes"
"context"
"fmt"
"io"
"net/http"
"strconv"
)
func (cli *Client) GetManifest(ctx context.Context, rawref string) (bool, string, []byte, error) {
var err error
var exist bool
var mime string
var man []byte
ref, err := NewReference(rawref)
if err != nil {
return exist, mime, man, err
}
uri := ref.Manifest()
req, err := http.NewRequestWithContext(ctx, http.MethodGet, uri, nil)
if err != nil {
return exist, mime, man, err
}
req.Header.Set("User-Agent", cli.userAgent)
req.Header.Set("Accept", "*/*")
resp, err := cli.httpClient.Do(req)
if err != nil {
return exist, mime, man, err
}
defer resp.Body.Close()
if resp.StatusCode == http.StatusNotFound {
return exist, mime, man, err
}
if resp.StatusCode != http.StatusOK {
err := fmt.Errorf("Unxected response code %s", resp.Status)
return exist, mime, man, err
}
contentLength := resp.Header.Get("Content-Length")
manSize, err := strconv.ParseInt(contentLength, 10, 64)
if err != nil {
return exist, mime, man, err
}
mime = resp.Header.Get("Content-Type")
if mime == "" {
err := fmt.Errorf("Empty MIME type declaration")
return exist, mime, man, err
}
buffer := bytes.NewBuffer(nil)
recSize, err := io.Copy(buffer, resp.Body)
if manSize != recSize {
err := fmt.Errorf("Mismatch declared and actual body size")
return exist, mime, man, err
}
man = buffer.Bytes()
exist = true
return exist, mime, man, err
}

34
getman_test.go Normal file
View File

@@ -0,0 +1,34 @@
package client
import (
"github.com/stretchr/testify/require"
"testing"
"fmt"
"time"
"context"
"encoding/json"
"bytes"
)
func TestClientGetManigest(t *testing.T) {
rawrefs := []string{
"mirror.gcr.io/alpine:3.20.0",
"mirror.gcr.io/alpine:sha256:29e5ba63e79337818e6c63cfcc68e2ab4e9ca483853b2de303bfbfba9372426c",
}
for _, rawref := range rawrefs {
cli := NewClient(true)
ctx, _ := context.WithTimeout(context.Background(), 10 * time.Second)
exist, mime, man, err := cli.GetManifest(ctx, rawref)
require.NoError(t, err)
require.True(t, exist)
fmt.Printf("MIME: %s\n", mime)
buffer := bytes.NewBuffer(nil)
err = json.Indent(buffer, man, " ", " ")
require.NoError(t, err)
fmt.Printf("%s\n", buffer.String())
}
}

11
go.mod Normal file
View File

@@ -0,0 +1,11 @@
module client
go 1.25.0
require github.com/stretchr/testify v1.11.1
require (
github.com/davecgh/go-spew v1.1.1 // indirect
github.com/pmezard/go-difflib v1.0.0 // indirect
gopkg.in/yaml.v3 v3.0.1 // indirect
)

10
go.sum Normal file
View File

@@ -0,0 +1,10 @@
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
github.com/stretchr/testify v1.11.1 h1:7s2iGBzp5EwR7/aIZr8ao5+dra3wiQyKjjFuvgVKu7U=
github.com/stretchr/testify v1.11.1/go.mod h1:wZwfW3scLgRK+23gO65QZefKpKQRnfz6sD981Nm4B6U=
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=

50
putman.go Normal file
View File

@@ -0,0 +1,50 @@
package client
import (
"bytes"
"context"
"fmt"
"net/http"
)
func (cli *Client) PutManifest(ctx context.Context, rawref string, man []byte, mime string) error {
var err error
ref, err := NewReference(rawref)
if err != nil {
return err
}
uri := ref.Manifest()
user, pass := ref.Userinfo()
buffer := bytes.NewBuffer(man)
req, err := http.NewRequestWithContext(ctx, http.MethodPut, uri, buffer)
if err != nil {
return err
}
req.Header.Set("User-Agent", cli.userAgent)
req.Header.Set("Docker-Content-Digest", SHA256Digest(man))
req.Header.Set("Content-Type", mime)
if cli.authenticator != nil {
authHeader, authKey, err := cli.authenticator.MakeHeader(user, pass)
if err != nil {
return err
}
req.Header.Set(authHeader, authKey)
}
resp, err := cli.httpClient.Do(req)
if err != nil {
return err
}
resp.Body.Close()
if resp.StatusCode != http.StatusAccepted {
err = fmt.Errorf("Manifest not accepted, code %d", resp.StatusCode)
return err
}
loc := resp.Header.Get("Location")
if loc == "" {
err := fmt.Errorf("Empty manifest location declaration")
return err
}
return err
}

55
refer.go Normal file
View File

@@ -0,0 +1,55 @@
package client
import (
"errors"
"net/url"
"strings"
)
type Reference struct {
urlobj *url.URL
user, pass string
repo, tag string
}
func NewReference(rawref string) (*Reference, error) {
ref := &Reference{}
if !strings.Contains(rawref, "://") {
rawref = "https://" + rawref
}
urlobj, err := url.Parse(rawref)
if err != nil {
return ref, err
}
if urlobj.User != nil {
ref.user = urlobj.User.Username()
ref.pass, _ = urlobj.User.Password()
urlobj.User = nil
}
repotag := strings.SplitN(urlobj.Path, ":", 2)
if len(repotag) != 2 {
err = errors.New("Incorrect repo")
return ref, err
}
ref.urlobj = urlobj
ref.urlobj.Path = ""
ref.repo = repotag[0]
ref.tag = repotag[1]
ref.urlobj = urlobj
return ref, err
}
func (ref *Reference) Manifest() string {
curl := ref.urlobj.JoinPath("/v2", ref.repo, "/manifests", ref.tag)
return curl.String()
}
func (ref *Reference) Blob(digest string) string {
curl := ref.urlobj.JoinPath("/v2", ref.repo, "/blobs", digest)
return curl.String()
}
func (ref *Reference) Userinfo() (string, string) {
return ref.user, ref.pass
}