mirror of
https://github.com/StackExchange/dnscontrol.git
synced 2024-09-20 23:06:23 +08:00
46 lines
1.4 KiB
Go
46 lines
1.4 KiB
Go
package models
|
|
|
|
import (
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
// SetTargetTLSA sets the TLSA fields.
|
|
func (rc *RecordConfig) SetTargetTLSA(usage, selector, matchingtype uint8, target string) error {
|
|
rc.TlsaUsage = usage
|
|
rc.TlsaSelector = selector
|
|
rc.TlsaMatchingType = matchingtype
|
|
rc.SetTarget(target)
|
|
if rc.Type == "" {
|
|
rc.Type = "TLSA"
|
|
}
|
|
if rc.Type != "TLSA" {
|
|
panic("assertion failed: SetTargetTLSA called when .Type is not TLSA")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// SetTargetTLSAStrings is like SetTargetTLSA but accepts strings.
|
|
func (rc *RecordConfig) SetTargetTLSAStrings(usage, selector, matchingtype, target string) (err error) {
|
|
var i64usage, i64selector, i64matchingtype uint64
|
|
if i64usage, err = strconv.ParseUint(usage, 10, 8); err == nil {
|
|
if i64selector, err = strconv.ParseUint(selector, 10, 8); err == nil {
|
|
if i64matchingtype, err = strconv.ParseUint(matchingtype, 10, 8); err == nil {
|
|
return rc.SetTargetTLSA(uint8(i64usage), uint8(i64selector), uint8(i64matchingtype), target)
|
|
}
|
|
}
|
|
}
|
|
return errors.Wrap(err, "TLSA has value that won't fit in field")
|
|
}
|
|
|
|
// SetTargetTLSAString is like SetTargetTLSA but accepts one big string.
|
|
func (rc *RecordConfig) SetTargetTLSAString(s string) error {
|
|
part := strings.Fields(s)
|
|
if len(part) != 4 {
|
|
return errors.Errorf("TLSA value does not contain 4 fields: (%#v)", s)
|
|
}
|
|
return rc.SetTargetTLSAStrings(part[0], part[1], part[2], part[3])
|
|
}
|