Bismuth/fork.py at master · bismuthfoundation/Bismuth · 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
class Fork():
    def __init__(self):
        self.POW_FORK = 1450000
        self.POW_FORK_TESTNET = 894170
        self.FORK_AHEAD = 5
        self.versions_remove = ['mainnet0020', 'mainnet0019', 'mainnet0018', 'mainnet0017']
        self.FORK_REWARD = None
        self.FORK_REWARD_TESTNET = None
        self.PASSED = False
        self.PASSED_TESTNET = False
        self.REWARD_MAX = 6

        #self.POW_FORK = 1168860 #HACK
        #self.versions_remove = [] #HACK
        #self.REWARD_MAX = 5 #HACK

    def limit_version(self, node):
        for allowed_version in node.version_allow:
            if allowed_version in self.versions_remove:
                node.logger.app_log.warning(f"Beginning to reject old protocol versions - block {node.last_block}")
                node.version_allow.remove(allowed_version)

    def check_postfork_reward(self, db_handler):
        try:
            # Check RAM first
            db_handler.execute_param(db_handler.c, "SELECT reward FROM transactions WHERE block_height = ? AND reward != 0", (self.POW_FORK + 1,))
            result = db_handler.c.fetchone()
            if result:
                self.FORK_REWARD = result[0]
                self.PASSED = self.FORK_REWARD < self.REWARD_MAX
            else:
                # Block not found, assume valid fork
                self.PASSED = True
        except Exception as e:
            try:
                # Fallback to HDD
                db_handler.execute_param(db_handler.h, "SELECT reward FROM transactions WHERE block_height = ? AND reward != 0", (self.POW_FORK + 1,))
                result = db_handler.h.fetchone()
                if result:
                    self.FORK_REWARD = result[0]
                    self.PASSED = self.FORK_REWARD < self.REWARD_MAX
                else:
                    self.PASSED = True  # Block pruned, assume valid
            except:
                self.PASSED = True  # Database error, assume valid
        return self.PASSED

    def check_postfork_reward_testnet(self, db_handler):
        try:
            db_handler.execute_param(db_handler.c, "SELECT reward FROM transactions WHERE block_height = ? AND reward != 0", (self.POW_FORK_TESTNET + 1,))
            result = db_handler.c.fetchone()
            if result:
                self.FORK_REWARD_TESTNET = result[0]
                self.PASSED_TESTNET = self.FORK_REWARD_TESTNET < self.REWARD_MAX
            else:
                self.PASSED_TESTNET = True
        except Exception as e:
            try:
                db_handler.execute_param(db_handler.h, "SELECT reward FROM transactions WHERE block_height = ? AND reward != 0", (self.POW_FORK_TESTNET + 1,))
                result = db_handler.h.fetchone()
                if result:
                    self.FORK_REWARD_TESTNET = result[0]
                    self.PASSED_TESTNET = self.FORK_REWARD_TESTNET < self.REWARD_MAX
                else:
                    self.PASSED_TESTNET = True
            except:
                self.PASSED_TESTNET = True
        return self.PASSED_TESTNET