firehol/netdata

// SPDX-License-Identifier: GPL-3.0-or-later

package chrony

import (
    "fmt"
    "time"
)

const scaleFactor = 1000000000

func (c *Chrony) collect() (map[string]int64, error) {
    if c.client == nil {
        client, err := c.newClient(c.Config)
        if err != nil {
            return nil, err
        }
        c.client = client
    }

    mx := make(map[string]int64)

    if err := c.collectTracking(mx); err != nil {
        return nil, err
    }
    if err := c.collectActivity(mx); err != nil {
        return mx, err
    }
    //if strings.HasPrefix(c.Address, "/") {
    // TODO: Allowed only through the Unix domain socket (requires "_chrony" group membership).
    // See https://github.com/facebook/time/blob/18207c5d8ddc7242e8d4192985898b6dbe66932c/cmd/ntpcheck/checker/chrony.go#L38
    // ^^ For some reason doesn't work, Chrony doesn't respond. Additional configuration needed?
    //if err := c.collectServerStats(mx); err != nil {
    //    return mx, err
    //}
    //}

    return mx, nil
}

const (
    // https://github.com/mlichvar/chrony/blob/7daf34675a5a2487895c74d1578241ca91a4eb70/ntp.h#L70-L75
    leapStatusNormal         = 0
    leapStatusInsertSecond   = 1
    leapStatusDeleteSecond   = 2
    leapStatusUnsynchronised = 3
)

func (c *Chrony) collectTracking(mx map[string]int64) error {
    reply, err := c.client.Tracking()
    if err != nil {
        return fmt.Errorf("error on collecting tracking: %v", err)
    }

    mx["stratum"] = int64(reply.Stratum)
    mx["leap_status_normal"] = boolToInt(reply.LeapStatus == leapStatusNormal)
    mx["leap_status_insert_second"] = boolToInt(reply.LeapStatus == leapStatusInsertSecond)
    mx["leap_status_delete_second"] = boolToInt(reply.LeapStatus == leapStatusDeleteSecond)
    mx["leap_status_unsynchronised"] = boolToInt(reply.LeapStatus == leapStatusUnsynchronised)
    mx["root_delay"] = int64(reply.RootDelay * scaleFactor)
    mx["root_dispersion"] = int64(reply.RootDispersion * scaleFactor)
    mx["skew"] = int64(reply.SkewPPM * scaleFactor)
    mx["last_offset"] = int64(reply.LastOffset * scaleFactor)
    mx["rms_offset"] = int64(reply.RMSOffset * scaleFactor)
    mx["update_interval"] = int64(reply.LastUpdateInterval * scaleFactor)
    // handle chrony restarts
    if reply.RefTime.Year() != 1970 {
        mx["ref_measurement_time"] = time.Now().Unix() - reply.RefTime.Unix()
    }
    mx["residual_frequency"] = int64(reply.ResidFreqPPM * scaleFactor)
    // https://github.com/mlichvar/chrony/blob/5b04f3ca902e5d10aa5948fb7587d30b43941049/client.c#L1706
    mx["current_correction"] = abs(int64(reply.CurrentCorrection * scaleFactor))
    mx["frequency"] = abs(int64(reply.FreqPPM * scaleFactor))

    return nil
}

func (c *Chrony) collectActivity(mx map[string]int64) error {
    reply, err := c.client.Activity()
    if err != nil {
        return fmt.Errorf("error on collecting activity: %v", err)
    }

    mx["online_sources"] = int64(reply.Online)
    mx["offline_sources"] = int64(reply.Offline)
    mx["burst_online_sources"] = int64(reply.BurstOnline)
    mx["burst_offline_sources"] = int64(reply.BurstOffline)
    mx["unresolved_sources"] = int64(reply.Unresolved)

    return nil
}

//func (c *Chrony) collectServerStats(mx map[string]int64) error {
//    stats, err := c.client.ServerStats()
//    if err != nil {
//        return fmt.Errorf("error on collecting server stats: %v", err)
//    }
//
//    switch {
//    case stats.v4 != nil:
//        mx["ntp_packets_received"] = int64(stats.v4.NTPHits)
//        mx["ntp_packets_dropped"] = int64(stats.v4.NTPDrops)
//        mx["command_packets_received"] = int64(stats.v4.CMDHits)
//        mx["command_packets_dropped"] = int64(stats.v4.CMDDrops)
//        mx["client_log_records_dropped"] = int64(stats.v4.LogDrops)
//        mx["nke_connections_accepted"] = int64(stats.v4.NKEHits)
//        mx["nke_connections_dropped"] = int64(stats.v4.NKEDrops)
//        mx["authenticated_ntp_packets"] = int64(stats.v4.NTPAuthHits)
//        mx["interleaved_ntp_packets"] = int64(stats.v4.NTPInterleavedHits)
//    case stats.v3 != nil:
//        mx["ntp_packets_received"] = int64(stats.v3.NTPHits)
//        mx["ntp_packets_dropped"] = int64(stats.v3.NTPDrops)
//        mx["command_packets_received"] = int64(stats.v3.CMDHits)
//        mx["command_packets_dropped"] = int64(stats.v3.CMDDrops)
//        mx["client_log_records_dropped"] = int64(stats.v3.LogDrops)
//        mx["nke_connections_accepted"] = int64(stats.v3.NKEHits)
//        mx["nke_connections_dropped"] = int64(stats.v3.NKEDrops)
//        mx["authenticated_ntp_packets"] = int64(stats.v3.NTPAuthHits)
//        mx["interleaved_ntp_packets"] = int64(stats.v3.NTPInterleavedHits)
//    case stats.v2 != nil:
//        mx["ntp_packets_received"] = int64(stats.v2.NTPHits)
//        mx["ntp_packets_dropped"] = int64(stats.v2.NTPDrops)
//        mx["command_packets_received"] = int64(stats.v2.CMDHits)
//        mx["command_packets_dropped"] = int64(stats.v2.CMDDrops)
//        mx["client_log_records_dropped"] = int64(stats.v2.LogDrops)
//        mx["nke_connections_accepted"] = int64(stats.v2.NKEHits)
//        mx["nke_connections_dropped"] = int64(stats.v2.NKEDrops)
//        mx["authenticated_ntp_packets"] = int64(stats.v2.NTPAuthHits)
//    case stats.v1 != nil:
//        mx["ntp_packets_received"] = int64(stats.v1.NTPHits)
//        mx["ntp_packets_dropped"] = int64(stats.v1.NTPDrops)
//        mx["command_packets_received"] = int64(stats.v1.CMDHits)
//        mx["command_packets_dropped"] = int64(stats.v1.CMDDrops)
//        mx["client_log_records_dropped"] = int64(stats.v1.LogDrops)
//    default:
//        return errors.New("invalid server stats reply")
//    }
//
//    //c.addStatsChartsOnce.Do(func() { c.addServerStatsCharts(stats) })
//
//    return nil
//}

func boolToInt(v bool) int64 {
    if v {
        return 1
    }
    return 0
}

func abs(v int64) int64 {
    if v < 0 {
        return -v
    }
    return v
}