summaryrefslogtreecommitdiff
path: root/test-cli/test/flashers/flasheeprom.py
blob: e427b877c14845dd928d7c3cde7fbb7842d0f6aa (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
import os
import binascii


def flash_eeprom(eeprompath, boarduuid, mac0, mac1=None):
    print("Start programming Eeprom...")
    # check if eeprompath is correct
    if os.path.isfile(eeprompath):
        # create u-boot data struct
        data = bytearray()
        data += (2029785358).to_bytes(4, 'big')  # magicid --> 0x78FC110E
        data += bytearray([0, 0, 0, 0])  # crc32 = 0
        data += bytearray(boarduuid + "\0",
                          'ascii')  # uuid --> 'c0846c8a-5fa5-11ea-8576-f8b156ac62d7' and \0 at the end
        data += binascii.unhexlify(mac0.replace(':', ''))  # mac0 --> 'f8:b1:56:ac:62:d7'
        if mac1 is not None:
            data += binascii.unhexlify(mac1.replace(':', ''))  # mac1 --> 'f8:b1:56:ac:62:d7'
        else:
            data += bytearray([0, 0, 0, 0, 0, 0])  # mac1 --> 0:0:0:0:0:0
        # calculate crc
        crc = (binascii.crc32(data, 0)).to_bytes(4, 'big')
        data[4:8] = crc
        # write into eeprom and read back
        f = open(eeprompath, "r+b")
        f.write(data)
        f.seek(0)
        data_rx = f.read(57)
        for i in range(57):
            if data_rx[i] != data[i]:
                print("Error while programming eeprom memory.")
                return 1
        print("Eeprom programmed succesfully.")
        return 0
    else:
        print("eeprom memory not found.")
        return 1