summaryrefslogtreecommitdiff
path: root/contrib/ntpshmviz
blob: 41e18590f6312775ee3e082a4a34610a643b6980 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
#!/usr/bin/env python
# 
# ntpshmviz - graph the drift of NTP servers
# Written by Keane Wolter <daemoneye2@gmail.com>
#
# To do:
#
# 1. Add exit button so the user does not need to do <Ctrl>-w
#

import sys
# need numpy for float128, normal python floats are too small to hold a timespec
import numpy
import matplotlib.pyplot as PLT
from matplotlib.figure import Figure

class ntpOffset:
    def __init__(self, stream):
        # get the data
        self.read_data(stream)

        # display the data
        self.display()

    def display(self):
        # display the graphs

        # Alert the user that closing the graphs can be done with "Ctrl-w"
        print ("Please note that the graph can be closed with the key combination of <Ctrl-w>")

        PLT.figure()
        subplot_value = 211
        for ntp_item in self.ntp_data:
            # create the subplot for the data
            PLT.subplot(subplot_value)
            # setup and create the vlines graph
            t = numpy.arange(0, self.line_counts[ntp_item], 1)
            PLT.vlines(t, 0, self.ntp_data[ntp_item], color='r')
            # add labels
            PLT.title("NTP drift for " + ntp_item)
            PLT.xlabel('sample')
            PLT.ylabel('drift')
            # increment the subplot by 1.
            # this allows for each data group to have it's own graph
            subplot_value += 1
        # make sure there is no graph or data overlapping each other and display the graph
        PLT.tight_layout()
        PLT.show()

    def read_data(self, stream):
        # Reads data from a ntp log file.  Layout is:
        #
        #   - The keyword "sample"
        #   - The NTP unit from which it was collected.
        #   - Collection time of day, expressed in seconds
        #   - Receiver time of day, expressed in seconds
        #   - Clock time of day, expressed in seconds
        #   - Leep-second notification status
        #   - Source precision (log(2) of source jitter)

        self.ntp_data    = {} # data sets for each ntp unit
        self.line_counts = {} # Count of total lines for each ntp unit
        record           = [] # A single record in the file or data stream
        offset           = 0  # offset value to add to the array 
        self.lines       = 0  # width of graph

        for line in stream:
            if len(line.split(' ')) > 6:
                if line[:1] != '#':
                    line = line.lstrip()
                    record = line.split(' ')
                    offset = numpy.float128(record[3]) - numpy.float128(record[4])

                    # If the NTP unit is in the dictionary
                    # append the offset to the list
                    # Otherwise, create a new list in the dictionary
                    # and add the offset.
                    if record[1] not in self.ntp_data:
                        self.ntp_data[record[1]]  = []
                        self.line_counts[record[1]] = 0

                    self.ntp_data[record[1]].append(offset)
                    self.line_counts[record[1]] += 1
        stream.close()

if __name__ == "__main__":
    if (len(sys.argv) == 2):
        if (sys.argv[1] == "-V" or sys.argv[1] == "--version"):
            print ("Version 2.0")
            sys.exit()

        if (sys.argv[1] == "-H" or sys.argv[1] == "--help"):
            print ("Usage: <input stream> | ntpshmviz")
            print ("Example: cat SAMPLES | ntpshmviz")
            print ("")
            print ("Options:")
            print ("\t-V: version information")
            print ("\t-H: help message")
            print ("")
            print ("Report ntpshmviz bugs to Keane Wolter, daemoneye2@gmail.com")
            print ("gpsd homepage: www.catb.org/gpsd/")
            sys.exit()

    ntpOffset(sys.stdin)
    sys.exit()