mirror of
https://github.com/0rangebananaspy/authelia.git
synced 2024-09-14 22:47:21 +07:00
aed9099ce2
* refactor: factorize startup checks * refactor: address linting issues
70 lines
1.8 KiB
Go
70 lines
1.8 KiB
Go
package ntp
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"errors"
|
|
"net"
|
|
"time"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
|
|
"github.com/authelia/authelia/v4/internal/configuration/schema"
|
|
"github.com/authelia/authelia/v4/internal/utils"
|
|
)
|
|
|
|
// NewProvider instantiate a ntp provider given a configuration.
|
|
func NewProvider(config *schema.NTPConfiguration) *Provider {
|
|
return &Provider{config}
|
|
}
|
|
|
|
// StartupCheck implements the startup check provider interface.
|
|
func (p *Provider) StartupCheck(logger *logrus.Logger) (err error) {
|
|
conn, err := net.Dial("udp", p.config.Address)
|
|
if err != nil {
|
|
logger.Warnf("Could not connect to NTP server to validate the system time is properly synchronized: %+v", err)
|
|
|
|
return nil
|
|
}
|
|
|
|
defer conn.Close()
|
|
|
|
if err := conn.SetDeadline(time.Now().Add(5 * time.Second)); err != nil {
|
|
logger.Warnf("Could not connect to NTP server to validate the system time is properly synchronized: %+v", err)
|
|
|
|
return nil
|
|
}
|
|
|
|
version := ntpV4
|
|
if p.config.Version == 3 {
|
|
version = ntpV3
|
|
}
|
|
|
|
req := &ntpPacket{LeapVersionMode: ntpLeapVersionClientMode(false, version)}
|
|
|
|
if err := binary.Write(conn, binary.BigEndian, req); err != nil {
|
|
logger.Warnf("Could not write to the NTP server socket to validate the system time is properly synchronized: %+v", err)
|
|
|
|
return nil
|
|
}
|
|
|
|
now := time.Now()
|
|
|
|
resp := &ntpPacket{}
|
|
|
|
if err := binary.Read(conn, binary.BigEndian, resp); err != nil {
|
|
logger.Warnf("Could not read from the NTP server socket to validate the system time is properly synchronized: %+v", err)
|
|
|
|
return nil
|
|
}
|
|
|
|
maxOffset, _ := utils.ParseDurationString(p.config.MaximumDesync)
|
|
|
|
ntpTime := ntpPacketToTime(resp)
|
|
|
|
if result := ntpIsOffsetTooLarge(maxOffset, now, ntpTime); result {
|
|
return errors.New("the system clock is not synchronized accurately enough with the configured NTP server")
|
|
}
|
|
|
|
return nil
|
|
}
|