validate ipcidr

multiformats · Jun 12, 2024 · 32ec788 · 32ec788
1 parent 5d767c5
commit 32ec788
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/transcoders.go b/transcoders.go
@@ -56,11 +56,11 @@ func (t twrp) ValidateBytes(b []byte) error {
 var TranscoderIP4 = NewTranscoderFromFunctions(ip4StB, ip4BtS, nil)
 var TranscoderIP6 = NewTranscoderFromFunctions(ip6StB, ip6BtS, nil)
 var TranscoderIP6Zone = NewTranscoderFromFunctions(ip6zoneStB, ip6zoneBtS, ip6zoneVal)
-var TranscoderIPCIDR = NewTranscoderFromFunctions(ipcidrStB, ipcidrBtS, nil)
+var TranscoderIPCIDR = NewTranscoderFromFunctions(ipcidrStB, ipcidrBtS, ipcidrValidate)
 
 func ipcidrBtS(b []byte) (string, error) {
-	if len(b) != 1 {
-		return "", fmt.Errorf("invalid length (should be == 1)")
+	if err := ipcidrValidate(b); err != nil {
+		return "", err
 	}
 	return strconv.Itoa(int(b[0])), nil
 }
@@ -73,6 +73,13 @@ func ipcidrStB(s string) ([]byte, error) {
 	return []byte{byte(uint8(ipMask))}, nil
 }
 
+func ipcidrValidate(b []byte) error {
+	if len(b) != 1 {
+		return fmt.Errorf("invalid length (should be == 1)")
+	}
+	return nil
+}
+
 func ip4StB(s string) ([]byte, error) {
 	i := net.ParseIP(s).To4()
 	if i == nil {