Merge pull request #11197 from hashicorp/f-add-cloudflare-record-validation
provider/cloudflare: Add validation for record types and record content
This commit is contained in:
commit
fd20f16a06
|
@ -34,9 +34,10 @@ func resourceCloudFlareRecord() *schema.Resource {
|
|||
},
|
||||
|
||||
"type": &schema.Schema{
|
||||
Type: schema.TypeString,
|
||||
Required: true,
|
||||
ForceNew: true,
|
||||
Type: schema.TypeString,
|
||||
Required: true,
|
||||
ForceNew: true,
|
||||
ValidateFunc: validateRecordType,
|
||||
},
|
||||
|
||||
"value": &schema.Schema{
|
||||
|
@ -88,6 +89,11 @@ func resourceCloudFlareRecordCreate(d *schema.ResourceData, meta interface{}) er
|
|||
newRecord.TTL = ttl.(int)
|
||||
}
|
||||
|
||||
// Validate value based on type
|
||||
if err := validateRecordName(newRecord.Type, newRecord.Content); err != nil {
|
||||
return fmt.Errorf("Error validating record name %q: %s", newRecord.Name, err)
|
||||
}
|
||||
|
||||
zoneId, err := client.ZoneIDByName(newRecord.ZoneName)
|
||||
if err != nil {
|
||||
return fmt.Errorf("Error finding zone %q: %s", newRecord.ZoneName, err)
|
||||
|
|
|
@ -0,0 +1,51 @@
|
|||
package cloudflare
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"net"
|
||||
"strings"
|
||||
)
|
||||
|
||||
// validateRecordType ensures that the cloudflare record type is valid
|
||||
func validateRecordType(v interface{}, k string) (ws []string, errors []error) {
|
||||
value := v.(string)
|
||||
|
||||
validTypes := map[string]struct{}{
|
||||
"A": {},
|
||||
"AAAA": {},
|
||||
"CNAME": {},
|
||||
"TXT": {},
|
||||
"SRV": {},
|
||||
"LOC": {},
|
||||
"MX": {},
|
||||
"NS": {},
|
||||
"SPF": {},
|
||||
}
|
||||
|
||||
if _, ok := validTypes[value]; !ok {
|
||||
errors = append(errors, fmt.Errorf(
|
||||
`%q contains an invalid type %q. Valid types are "A", "AAAA", "CNAME", "TXT", "SRV", "LOC", "MX", "NS" or "SPF"`, k, value))
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// validateRecordName ensures that based on supplied record type, the name content matches
|
||||
// Currently only validates A and AAAA types
|
||||
func validateRecordName(t string, value string) error {
|
||||
switch t {
|
||||
case "A":
|
||||
// Must be ipv4 addr
|
||||
addr := net.ParseIP(value)
|
||||
if addr == nil || !strings.Contains(value, ".") {
|
||||
return fmt.Errorf("A record must be a valid IPv4 address, got: %q", value)
|
||||
}
|
||||
case "AAAA":
|
||||
// Must be ipv6 addr
|
||||
addr := net.ParseIP(value)
|
||||
if addr == nil || !strings.Contains(value, ":") {
|
||||
return fmt.Errorf("AAAA record must be a valid IPv6 address, got: %q", value)
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
|
@ -0,0 +1,61 @@
|
|||
package cloudflare
|
||||
|
||||
import "testing"
|
||||
|
||||
func TestValidateRecordType(t *testing.T) {
|
||||
validTypes := []string{
|
||||
"A",
|
||||
"AAAA",
|
||||
"CNAME",
|
||||
"TXT",
|
||||
"SRV",
|
||||
"LOC",
|
||||
"MX",
|
||||
"NS",
|
||||
"SPF",
|
||||
}
|
||||
for _, v := range validTypes {
|
||||
_, errors := validateRecordType(v, "type")
|
||||
if len(errors) != 0 {
|
||||
t.Fatalf("%q should be a valid record type: %q", v, errors)
|
||||
}
|
||||
}
|
||||
|
||||
invalidTypes := []string{
|
||||
"a",
|
||||
"cName",
|
||||
"txt",
|
||||
"SRv",
|
||||
"foo",
|
||||
"bar",
|
||||
}
|
||||
for _, v := range invalidTypes {
|
||||
_, errors := validateRecordType(v, "type")
|
||||
if len(errors) == 0 {
|
||||
t.Fatalf("%q should be an invalid record type", v)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestValidateRecordName(t *testing.T) {
|
||||
validNames := map[string]string{
|
||||
"A": "192.168.0.1",
|
||||
"AAAA": "2001:0db8:0000:0042:0000:8a2e:0370:7334",
|
||||
}
|
||||
|
||||
for k, v := range validNames {
|
||||
if err := validateRecordName(k, v); err != nil {
|
||||
t.Fatalf("%q should be a valid name for type %q: %v", v, k, err)
|
||||
}
|
||||
}
|
||||
|
||||
invalidNames := map[string]string{
|
||||
"A": "terraform.io",
|
||||
"AAAA": "192.168.0.1",
|
||||
}
|
||||
for k, v := range invalidNames {
|
||||
if err := validateRecordName(k, v); err == nil {
|
||||
t.Fatalf("%q should be an invalid name for type %q", v, k)
|
||||
}
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue