node-cypress-dfu/otaWriter.js at master · modrobotics/node-cypress-dfu · 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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
var BootLoaderCommands = require('./otaCommands')
var OTAUtil = require('./otaUtil.js')
var debug = require('debug')('cypressDFU:otaWriter')

var OTAFirmwareWrite = function (writeMethod) {
  var BYTE_START_CMD = 0
  var BYTE_CMD_TYPE = 1
  var BYTE_CMD_DATA_SIZE = 2
  var BYTE_CMD_DATA_SIZE_SHIFT = 3
  var BYTE_CHECKSUM = 4
  var BYTE_CHECKSUM_SHIFT = 5
  var BYTE_PACKET_END = 6

  var BYTE_PACKET_END_VER_ROW = 9
  var BYTE_ARRAY_ID = 4
  var BYTE_ROW = 5
  var BYTE_ROW_SHIFT = 6
  var BYTE_CHECKSUM_VER_ROW = 7
  var BYTE_CHECKSUM_VER_ROW_SHIFT = 8

  var RADIX = 16
  var ADDITIVE_OP = 8

  this.OTAEnterBootLoaderCmd = function (checkSumType, callback) {
    var startCommand = 0x01
    var dataLength0 = 0x00
    var dataLength1 = 0x00

    var commandBytes = []
    commandBytes[BYTE_START_CMD] = startCommand
    commandBytes[BYTE_CMD_TYPE] = BootLoaderCommands.ENTER_BOOTLOADER
    commandBytes[BYTE_CMD_DATA_SIZE] = dataLength0
    commandBytes[BYTE_CMD_DATA_SIZE_SHIFT] = dataLength1
    var checksum = OTAUtil.calculateCheckSum(parseInt(checkSumType, RADIX), commandBytes)
    commandBytes[BYTE_CHECKSUM] = checksum
    commandBytes[BYTE_CHECKSUM_SHIFT] = (checksum >> ADDITIVE_OP)
    commandBytes[BYTE_PACKET_END] = BootLoaderCommands.PACKET_END
    debug('OTAEnterBootLoaderCmd')
    writeOTABootLoaderCommand(commandBytes, callback)
  }

  this.OTAGetFlashSizeCmd = function (data, checkSumType, dataLength, callback) {
    debug('OTAGetFlashSizeCmd', data, checkSumType, dataLength)
    var commandBytes = []
    var startCommand = 0x01
    commandBytes[BYTE_START_CMD] = startCommand
    commandBytes[BYTE_CMD_TYPE] = BootLoaderCommands.GET_FLASH_SIZE
    commandBytes[BYTE_CMD_DATA_SIZE] = dataLength
    commandBytes[BYTE_CMD_DATA_SIZE_SHIFT] = (dataLength >> ADDITIVE_OP)
    var dataByteLocationStart = 4
    var datByteLocationEnd
    for (var count = 0; count < dataLength; count++) {
      commandBytes[dataByteLocationStart] = data[count]
      dataByteLocationStart++
    }
    datByteLocationEnd = dataByteLocationStart
    var checksum = OTAUtil.calculateCheckSum(parseInt(checkSumType, RADIX), commandBytes)
    commandBytes[datByteLocationEnd] = checksum
    commandBytes[datByteLocationEnd + 1] = (checksum >> ADDITIVE_OP)
    commandBytes[datByteLocationEnd + 2] = BootLoaderCommands.PACKET_END
    debug('OTAGetFlashSizeCmd')
    writeOTABootLoaderCommand(commandBytes, callback)
  }

  this.OTAProgramRowSendDataCmd = function (data, checksumType, callback) {
    var totalSize = BootLoaderCommands.BASE_CMD_SIZE +
                data.length
    var checksum
    var commandBytes = []
    var startCommand = 0x01

    commandBytes[BYTE_START_CMD] = startCommand
    commandBytes[BYTE_CMD_TYPE] = BootLoaderCommands.SEND_DATA
    commandBytes[BYTE_CMD_DATA_SIZE] = (data.length)
    commandBytes[BYTE_CMD_DATA_SIZE_SHIFT] = (((data.length) >> ADDITIVE_OP))

    debug('OTAProgramRowSendDataCmd', commandBytes)
    debug(data)

    for (var i = 1; i < data.length; i++) { commandBytes[i + 4] = data[i] }
    checksum = OTAUtil.calculateCheckSum(parseInt(checksumType, RADIX), commandBytes)
    commandBytes[totalSize - 3] = checksum
    commandBytes[totalSize - 2] = (checksum >> ADDITIVE_OP)
    commandBytes[totalSize - 1] = BootLoaderCommands.PACKET_END
    debug('OTAProgramRowSendDataCmd Send size--->' + commandBytes.length)
    writeOTABootLoaderCommand(commandBytes, callback)
  }

  this.OTAProgramRowCmd = function (rowMSB, rowLSB, arrayID, data, checkSumType, callback) {
    var COMMAND_DATA_SIZE = 3
    var totalSize = BootLoaderCommands.BASE_CMD_SIZE + COMMAND_DATA_SIZE +
                data.length
    var checksum
    var i
    var commandBytes = []
    var startCommand = 0x01

    commandBytes[BYTE_START_CMD] = startCommand
    commandBytes[BYTE_CMD_TYPE] = BootLoaderCommands.PROGRAM_ROW
    commandBytes[BYTE_CMD_DATA_SIZE] = (data.length + COMMAND_DATA_SIZE)
    commandBytes[BYTE_CMD_DATA_SIZE_SHIFT] = (((data.length + COMMAND_DATA_SIZE) >> ADDITIVE_OP))
    commandBytes[BYTE_ARRAY_ID] = arrayID
    commandBytes[BYTE_ROW] = rowMSB
    commandBytes[6] = rowLSB
    for (i = 0; i < data.length; i++) { commandBytes[i + 7] = data[i] }
    checksum = OTAUtil.calculateCheckSum(parseInt(checkSumType, RADIX), commandBytes)
    commandBytes[totalSize - 3] = checksum
    commandBytes[totalSize - 2] = (checksum >> ADDITIVE_OP)
    commandBytes[totalSize - 1] = BootLoaderCommands.PACKET_END
    debug('OTAProgramRowCmd send size--->' + commandBytes.length)
    writeOTABootLoaderCommand(commandBytes, callback)
  }

  this.OTAVerifyRowCmd = function (rowMSB, rowLSB, model, checkSumType, callback) {
    var COMMAND_DATA_SIZE = 3
    var checksum
    var commandBytes = []
    var startCommand = 0x01

    commandBytes[BYTE_START_CMD] = startCommand
    commandBytes[BYTE_CMD_TYPE] = BootLoaderCommands.VERIFY_ROW
    commandBytes[BYTE_CMD_DATA_SIZE] = (COMMAND_DATA_SIZE)
    commandBytes[BYTE_CMD_DATA_SIZE_SHIFT] = (COMMAND_DATA_SIZE >> ADDITIVE_OP)
    commandBytes[BYTE_ARRAY_ID] = model.arrayID
    commandBytes[BYTE_ROW] = rowMSB
    commandBytes[BYTE_ROW_SHIFT] = rowLSB
    checksum = OTAUtil.calculateCheckSum(parseInt(checkSumType, RADIX), commandBytes)
    commandBytes[BYTE_CHECKSUM_VER_ROW] = checksum
    commandBytes[BYTE_CHECKSUM_VER_ROW_SHIFT] = (checksum >> ADDITIVE_OP)
    commandBytes[BYTE_PACKET_END_VER_ROW] = BootLoaderCommands.PACKET_END
    debug('OTAVerifyRowCmd')
    writeOTABootLoaderCommand(commandBytes, callback)
  }

  this.OTAVerifyCheckSumCmd = function (checkSumType, callback) {
    var checksum
    var commandBytes = []
    var startCommand = 0x01

    commandBytes[BYTE_START_CMD] = startCommand
    commandBytes[BYTE_CMD_TYPE] = BootLoaderCommands.VERIFY_CHECK_SUM
    commandBytes[BYTE_CMD_DATA_SIZE] = (0)
    commandBytes[BYTE_CMD_DATA_SIZE_SHIFT] = (0)
    checksum = OTAUtil.calculateCheckSum(parseInt(checkSumType, RADIX), commandBytes)
    commandBytes[BYTE_CHECKSUM] = checksum
    commandBytes[BYTE_CHECKSUM_SHIFT] = (checksum >> ADDITIVE_OP)
    commandBytes[BYTE_PACKET_END] = BootLoaderCommands.PACKET_END
    debug('OTAVerifyCheckSumCmd')
    writeOTABootLoaderCommand(commandBytes, callback)
  }

  this.OTAExitBootloaderCmd = function (checkSumType, callback) {
    var COMMAND_DATA_SIZE = 0x00
    var checksum
    var commandBytes = []
    var startCommand = 0x01

    commandBytes[BYTE_START_CMD] = startCommand
    commandBytes[BYTE_CMD_TYPE] = BootLoaderCommands.EXIT_BOOTLOADER
    commandBytes[BYTE_CMD_DATA_SIZE] = (COMMAND_DATA_SIZE)
    commandBytes[BYTE_CMD_DATA_SIZE_SHIFT] = (COMMAND_DATA_SIZE >> ADDITIVE_OP)
    checksum = OTAUtil.calculateCheckSum(parseInt(checkSumType, RADIX), commandBytes)
    commandBytes[BYTE_CHECKSUM] = checksum
    commandBytes[BYTE_CHECKSUM_SHIFT] = (checksum >> ADDITIVE_OP)
    commandBytes[BYTE_PACKET_END] = BootLoaderCommands.PACKET_END
    debug('OTAExitBootloaderCmd')
    writeOTABootLoaderCommand(commandBytes, callback)
  }

  function writeOTABootLoaderCommand (commandBytes, callback) {
    writeMethod(commandBytes, callback)
  }
}

module.exports = OTAFirmwareWrite