election/assertvalidity.py at master · dubstylee/election · GitHub

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
from shared import mqtt_client, mqtt_topic, send_message, exit_program
from enum import Enum
import sys
import time


class AssertStatus(Enum):
    ERROR = -1
    BEGIN = 0
    HOMEFREE = 1


class FluentStatus(Enum):
    ON = 1
    OFF = 0


# To keep a track of which node we are
# monitoring the assert for
assertParam = 0
assertName = None
assertDescription = ""
fluent = None
systemStatus = AssertStatus.BEGIN


class Fluent():
    status = FluentStatus.OFF
    identifier = 0

    def __init__(self, param):
        self.identifier = param

    def checkFluent(self, message):
        split = message.split(" ")
        if split[3] == "ELECTION":
            sent_id = int(split[5])
            if sent_id == self.identifier:
                self.status = FluentStatus.ON
        elif split[3] == "LEADER":
            recipient = int(split[4])
            elected = int(split[5])
            if recipient == self.identifier or elected == self.identifier:
                self.status = FluentStatus.OFF


def on_message(client, userdata, msg):
    global systemStatus
    message = msg.payload

    # we are only interested in election/leader actions for this assert
    split = message.split(" ")
    if not split[3] in ["ELECTION", "LEADER"]:
        return

    # Check the status of the fluent here after receiving the
    # messages
    fluent.checkFluent(message)
    if fluent.status == FluentStatus.ON:
        systemStatus = AssertStatus.HOMEFREE

    if systemStatus == AssertStatus.HOMEFREE:
        send_message("UPDATEA %s : %s" % (assertName, message))
    elif systemStatus != AssertStatus.HOMEFREE:
        send_message("UPDATEA %s : %s" % (assertName, message))
        split = message.split(" ")
        if "LEADER" == split[3]:
            recipient = int(split[4])
            leader = int(split[5])
            if leader == assertParam and systemStatus == AssertStatus.BEGIN:
                systemStatus = AssertStatus.ERROR
                send_message("UPDATEA %s : ASSERTFAILED" % assertName)


def main():
    global assertParam
    global assertName
    global assertDescription
    global fluent

    if len(sys.argv) < 2:
        print "Invalid number of arguments, usage : " \
              "python assert.py <assert parameter>"
        exit_program()

    assertParam = int(sys.argv[1])
    fluent = Fluent(assertParam)
    mqtt_client.on_message = on_message
    mqtt_client.will_set(mqtt_topic, "Will of Asserter\n\n", 0, False)
    mqtt_client.loop_start()

    assertName = "VALID%d" % assertParam
    assertDescription = "(!send_leader[%d][k:IDS] W send_id[%d][k:IDS])" % \
                 (assertParam, assertParam)
    send_message("LABELA assert %s = %s" % (assertName, assertDescription))

    while True:
        time.sleep(1)


if __name__ == "__main__":
    main()