mirror of
https://github.com/StackExchange/dnscontrol.git
synced 2024-09-20 23:06:23 +08:00
53 lines
1.6 KiB
Go
53 lines
1.6 KiB
Go
package models
|
|
|
|
import (
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
// SetTargetSSHFP sets the SSHFP fields.
|
|
func (rc *RecordConfig) SetTargetSSHFP(algorithm uint8, fingerprint uint8, target string) error {
|
|
rc.SshfpAlgorithm = algorithm
|
|
rc.SshfpFingerprint = fingerprint
|
|
rc.SetTarget(target)
|
|
if rc.Type == "" {
|
|
rc.Type = "SSHFP"
|
|
}
|
|
if rc.Type != "SSHFP" {
|
|
panic("assertion failed: SetTargetSSHFP called when .Type is not SSHFP")
|
|
}
|
|
|
|
if algorithm < 1 && algorithm > 4 {
|
|
return errors.Errorf("SSHFP algorithm (%v) is not one of 1, 2, 3 or 4", algorithm)
|
|
}
|
|
if fingerprint < 1 && fingerprint > 2 {
|
|
return errors.Errorf("SSHFP fingerprint (%v) is not one of 1 or 2", fingerprint)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// SetTargetSSHFPStrings is like SetTargetSSHFP but accepts strings.
|
|
func (rc *RecordConfig) SetTargetSSHFPStrings(algorithm, fingerprint, target string) error {
|
|
i64algorithm, err := strconv.ParseUint(algorithm, 10, 8)
|
|
if err != nil {
|
|
return errors.Wrap(err, "SSHFP algorithm does not fit in 8 bits")
|
|
}
|
|
i64fingerprint, err := strconv.ParseUint(fingerprint, 10, 8)
|
|
if err != nil {
|
|
return errors.Wrap(err, "SSHFP fingerprint does not fit in 8 bits")
|
|
}
|
|
return rc.SetTargetSSHFP(uint8(i64algorithm), uint8(i64fingerprint), target)
|
|
}
|
|
|
|
// SetTargetSSHFPString is like SetTargetSSHFP but accepts one big string.
|
|
func (rc *RecordConfig) SetTargetSSHFPString(s string) error {
|
|
part := strings.Fields(s)
|
|
if len(part) != 3 {
|
|
return errors.Errorf("SSHFP value does not contain 3 fields: (%#v)", s)
|
|
}
|
|
return rc.SetTargetSSHFPStrings(part[0], part[1], StripQuotes(part[2]))
|
|
}
|