Labels are consistently validated across services
* The combined size of a key/value pair cannot exceed 4096 bytes Signed-off-by: Jess Valarezo <valarezo.jessica@gmail.com>
This commit is contained in:
23
labels/validate.go
Normal file
23
labels/validate.go
Normal file
@@ -0,0 +1,23 @@
|
||||
package labels
|
||||
|
||||
import (
|
||||
"github.com/containerd/containerd/errdefs"
|
||||
"github.com/pkg/errors"
|
||||
)
|
||||
|
||||
const (
|
||||
maxSize = 4096
|
||||
)
|
||||
|
||||
func Validate(k, v string) error {
|
||||
// A label key and value should be under 4096 bytes
|
||||
if (len(k) + len(v)) > maxSize {
|
||||
if len(k) > 10 {
|
||||
k = k[:10]
|
||||
}
|
||||
|
||||
return errors.Wrapf(errdefs.ErrInvalidArgument, "label key and value greater than maximum size (%d bytes), key: %s", maxSize, k)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
37
labels/validate_test.go
Normal file
37
labels/validate_test.go
Normal file
@@ -0,0 +1,37 @@
|
||||
package labels
|
||||
|
||||
import (
|
||||
"strings"
|
||||
"testing"
|
||||
|
||||
"github.com/containerd/containerd/errdefs"
|
||||
)
|
||||
|
||||
func TestValidLabels(t *testing.T) {
|
||||
shortStr := "s"
|
||||
longStr := strings.Repeat("s", maxSize-1)
|
||||
|
||||
for key, value := range map[string]string{
|
||||
"some": "value",
|
||||
shortStr: longStr,
|
||||
} {
|
||||
if err := Validate(key, value); err != nil {
|
||||
t.Fatalf("unexpected error: %v != nil", err)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestInvalidLabels(t *testing.T) {
|
||||
addOneStr := "s"
|
||||
maxSizeStr := strings.Repeat("s", maxSize)
|
||||
|
||||
for key, value := range map[string]string{
|
||||
maxSizeStr: addOneStr,
|
||||
} {
|
||||
if err := Validate(key, value); err == nil {
|
||||
t.Fatal("expected invalid error")
|
||||
} else if !errdefs.IsInvalidArgument(err) {
|
||||
t.Fatal("error should be an invalid label error")
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user