2021-09-22 23:08:22 +03:00
|
|
|
// This Source Code Form is subject to the terms of the Mozilla Public
|
|
|
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
|
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
|
|
|
|
package server
|
|
|
|
|
|
|
|
import (
|
2021-09-23 17:47:12 +03:00
|
|
|
"time"
|
|
|
|
|
2021-09-22 23:08:22 +03:00
|
|
|
"google.golang.org/grpc/codes"
|
|
|
|
"google.golang.org/grpc/status"
|
|
|
|
|
2021-09-23 17:47:12 +03:00
|
|
|
"github.com/talos-systems/discovery-service/pkg/limits"
|
2021-09-22 23:08:22 +03:00
|
|
|
)
|
|
|
|
|
|
|
|
func validateClusterID(id string) error {
|
|
|
|
if len(id) < 1 {
|
|
|
|
return status.Errorf(codes.InvalidArgument, "cluster ID can't be empty")
|
|
|
|
}
|
|
|
|
|
2021-09-23 17:47:12 +03:00
|
|
|
if len(id) > limits.ClusterIDMax {
|
2021-09-22 23:08:22 +03:00
|
|
|
return status.Errorf(codes.InvalidArgument, "cluster ID is too long")
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func validateAffiliateID(id string) error {
|
|
|
|
if len(id) < 1 {
|
|
|
|
return status.Errorf(codes.InvalidArgument, "affiliate ID can't be empty")
|
|
|
|
}
|
|
|
|
|
2021-09-23 17:47:12 +03:00
|
|
|
if len(id) > limits.AffiliateIDMax {
|
2021-09-22 23:08:22 +03:00
|
|
|
return status.Errorf(codes.InvalidArgument, "affiliate ID is too long")
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
2021-09-23 17:47:12 +03:00
|
|
|
|
|
|
|
func validateAffiliateData(data []byte) error {
|
|
|
|
if len(data) > limits.AffiliateDataMax {
|
|
|
|
return status.Error(codes.InvalidArgument, "affiliate data is too big")
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func validateAffiliateEndpoints(endpoints [][]byte) error {
|
|
|
|
for _, endpoint := range endpoints {
|
|
|
|
if len(endpoint) > limits.AffiliateEndpointMax {
|
|
|
|
return status.Errorf(codes.InvalidArgument, "affiliate endpoint is too big")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func validateTTL(ttl time.Duration) error {
|
|
|
|
if ttl > limits.TTLMax {
|
|
|
|
return status.Errorf(codes.InvalidArgument, "ttl is too large")
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|