mirror of
https://github.com/MikaylaFischler/cc-mek-scada.git
synced 2024-08-30 18:22:34 +00:00
462 lines
16 KiB
Lua
462 lines
16 KiB
Lua
local comms = require("scada-common.comms")
|
|
local types = require("scada-common.types")
|
|
|
|
local modbus = {}
|
|
|
|
local MODBUS_FCODE = types.MODBUS_FCODE
|
|
local MODBUS_EXCODE = types.MODBUS_EXCODE
|
|
|
|
-- new modbus comms handler object
|
|
---@nodiscard
|
|
---@param rtu_dev rtu_device|rtu_rs_device RTU device
|
|
---@param use_parallel_read boolean whether or not to use parallel calls when reading
|
|
function modbus.new(rtu_dev, use_parallel_read)
|
|
local insert = table.insert
|
|
|
|
-- read a span of coils (digital outputs)<br>
|
|
-- returns a table of readings or a MODBUS_EXCODE error code
|
|
---@nodiscard
|
|
---@param c_addr_start integer
|
|
---@param count integer
|
|
---@return boolean ok, table|MODBUS_EXCODE readings
|
|
local function _1_read_coils(c_addr_start, count)
|
|
local tasks = {}
|
|
local readings = {} ---@type table|MODBUS_EXCODE
|
|
local access_fault = false
|
|
local _, coils, _, _ = rtu_dev.io_count()
|
|
local return_ok = ((c_addr_start + count) <= (coils + 1)) and (count > 0)
|
|
|
|
if return_ok then
|
|
for i = 1, count do
|
|
local addr = c_addr_start + i - 1
|
|
|
|
if use_parallel_read then
|
|
insert(tasks, function ()
|
|
local reading, fault = rtu_dev.read_coil(addr)
|
|
if fault then access_fault = true else readings[i] = reading end
|
|
end)
|
|
else
|
|
readings[i], access_fault = rtu_dev.read_coil(addr)
|
|
|
|
if access_fault then
|
|
return_ok = false
|
|
readings = MODBUS_EXCODE.SERVER_DEVICE_FAIL
|
|
break
|
|
end
|
|
end
|
|
end
|
|
|
|
-- run parallel tasks if configured
|
|
if use_parallel_read then
|
|
parallel.waitForAll(table.unpack(tasks))
|
|
end
|
|
|
|
if access_fault or #readings ~= count then
|
|
return_ok = false
|
|
readings = MODBUS_EXCODE.SERVER_DEVICE_FAIL
|
|
end
|
|
else
|
|
readings = MODBUS_EXCODE.ILLEGAL_DATA_ADDR
|
|
end
|
|
|
|
return return_ok, readings
|
|
end
|
|
|
|
-- read a span of discrete inputs (digital inputs)<br>
|
|
-- returns a table of readings or a MODBUS_EXCODE error code
|
|
---@nodiscard
|
|
---@param di_addr_start integer
|
|
---@param count integer
|
|
---@return boolean ok, table|MODBUS_EXCODE readings
|
|
local function _2_read_discrete_inputs(di_addr_start, count)
|
|
local tasks = {}
|
|
local readings = {} ---@type table|MODBUS_EXCODE
|
|
local access_fault = false
|
|
local discrete_inputs, _, _, _ = rtu_dev.io_count()
|
|
local return_ok = ((di_addr_start + count) <= (discrete_inputs + 1)) and (count > 0)
|
|
|
|
if return_ok then
|
|
for i = 1, count do
|
|
local addr = di_addr_start + i - 1
|
|
|
|
if use_parallel_read then
|
|
insert(tasks, function ()
|
|
local reading, fault = rtu_dev.read_di(addr)
|
|
if fault then access_fault = true else readings[i] = reading end
|
|
end)
|
|
else
|
|
readings[i], access_fault = rtu_dev.read_di(addr)
|
|
|
|
if access_fault then
|
|
return_ok = false
|
|
readings = MODBUS_EXCODE.SERVER_DEVICE_FAIL
|
|
break
|
|
end
|
|
end
|
|
end
|
|
|
|
-- run parallel tasks if configured
|
|
if use_parallel_read then
|
|
parallel.waitForAll(table.unpack(tasks))
|
|
end
|
|
|
|
if access_fault or #readings ~= count then
|
|
return_ok = false
|
|
readings = MODBUS_EXCODE.SERVER_DEVICE_FAIL
|
|
end
|
|
else
|
|
readings = MODBUS_EXCODE.ILLEGAL_DATA_ADDR
|
|
end
|
|
|
|
return return_ok, readings
|
|
end
|
|
|
|
-- read a span of holding registers (analog outputs)<br>
|
|
-- returns a table of readings or a MODBUS_EXCODE error code
|
|
---@nodiscard
|
|
---@param hr_addr_start integer
|
|
---@param count integer
|
|
---@return boolean ok, table|MODBUS_EXCODE readings
|
|
local function _3_read_multiple_holding_registers(hr_addr_start, count)
|
|
local tasks = {}
|
|
local readings = {} ---@type table|MODBUS_EXCODE
|
|
local access_fault = false
|
|
local _, _, _, hold_regs = rtu_dev.io_count()
|
|
local return_ok = ((hr_addr_start + count) <= (hold_regs + 1)) and (count > 0)
|
|
|
|
if return_ok then
|
|
for i = 1, count do
|
|
local addr = hr_addr_start + i - 1
|
|
|
|
if use_parallel_read then
|
|
insert(tasks, function ()
|
|
local reading, fault = rtu_dev.read_holding_reg(addr)
|
|
if fault then access_fault = true else readings[i] = reading end
|
|
end)
|
|
else
|
|
readings[i], access_fault = rtu_dev.read_holding_reg(addr)
|
|
|
|
if access_fault then
|
|
return_ok = false
|
|
readings = MODBUS_EXCODE.SERVER_DEVICE_FAIL
|
|
break
|
|
end
|
|
end
|
|
end
|
|
|
|
-- run parallel tasks if configured
|
|
if use_parallel_read then
|
|
parallel.waitForAll(table.unpack(tasks))
|
|
end
|
|
|
|
if access_fault or #readings ~= count then
|
|
return_ok = false
|
|
readings = MODBUS_EXCODE.SERVER_DEVICE_FAIL
|
|
end
|
|
else
|
|
readings = MODBUS_EXCODE.ILLEGAL_DATA_ADDR
|
|
end
|
|
|
|
return return_ok, readings
|
|
end
|
|
|
|
-- read a span of input registers (analog inputs)<br>
|
|
-- returns a table of readings or a MODBUS_EXCODE error code
|
|
---@nodiscard
|
|
---@param ir_addr_start integer
|
|
---@param count integer
|
|
---@return boolean ok, table|MODBUS_EXCODE readings
|
|
local function _4_read_input_registers(ir_addr_start, count)
|
|
local tasks = {}
|
|
local readings = {} ---@type table|MODBUS_EXCODE
|
|
local access_fault = false
|
|
local _, _, input_regs, _ = rtu_dev.io_count()
|
|
local return_ok = ((ir_addr_start + count) <= (input_regs + 1)) and (count > 0)
|
|
|
|
if return_ok then
|
|
for i = 1, count do
|
|
local addr = ir_addr_start + i - 1
|
|
|
|
if use_parallel_read then
|
|
insert(tasks, function ()
|
|
local reading, fault = rtu_dev.read_input_reg(addr)
|
|
if fault then access_fault = true else readings[i] = reading end
|
|
end)
|
|
else
|
|
readings[i], access_fault = rtu_dev.read_input_reg(addr)
|
|
|
|
if access_fault then
|
|
return_ok = false
|
|
readings = MODBUS_EXCODE.SERVER_DEVICE_FAIL
|
|
break
|
|
end
|
|
end
|
|
end
|
|
|
|
-- run parallel tasks if configured
|
|
if use_parallel_read then
|
|
parallel.waitForAll(table.unpack(tasks))
|
|
end
|
|
|
|
if access_fault or #readings ~= count then
|
|
return_ok = false
|
|
readings = MODBUS_EXCODE.SERVER_DEVICE_FAIL
|
|
end
|
|
else
|
|
readings = MODBUS_EXCODE.ILLEGAL_DATA_ADDR
|
|
end
|
|
|
|
return return_ok, readings
|
|
end
|
|
|
|
-- write a single coil (digital output)
|
|
---@nodiscard
|
|
---@param c_addr integer
|
|
---@param value any
|
|
---@return boolean ok, MODBUS_EXCODE
|
|
local function _5_write_single_coil(c_addr, value)
|
|
local response = MODBUS_EXCODE.OK
|
|
local _, coils, _, _ = rtu_dev.io_count()
|
|
local return_ok = c_addr <= coils
|
|
|
|
if return_ok then
|
|
local access_fault = rtu_dev.write_coil(c_addr, value)
|
|
|
|
if access_fault then
|
|
return_ok = false
|
|
response = MODBUS_EXCODE.SERVER_DEVICE_FAIL
|
|
end
|
|
else
|
|
response = MODBUS_EXCODE.ILLEGAL_DATA_ADDR
|
|
end
|
|
|
|
return return_ok, response
|
|
end
|
|
|
|
-- write a single holding register (analog output)
|
|
---@nodiscard
|
|
---@param hr_addr integer
|
|
---@param value any
|
|
---@return boolean ok, MODBUS_EXCODE
|
|
local function _6_write_single_holding_register(hr_addr, value)
|
|
local response = MODBUS_EXCODE.OK
|
|
local _, _, _, hold_regs = rtu_dev.io_count()
|
|
local return_ok = hr_addr <= hold_regs
|
|
|
|
if return_ok then
|
|
local access_fault = rtu_dev.write_holding_reg(hr_addr, value)
|
|
|
|
if access_fault then
|
|
return_ok = false
|
|
response = MODBUS_EXCODE.SERVER_DEVICE_FAIL
|
|
end
|
|
else
|
|
response = MODBUS_EXCODE.ILLEGAL_DATA_ADDR
|
|
end
|
|
|
|
return return_ok, response
|
|
end
|
|
|
|
-- write multiple coils (digital outputs)
|
|
---@nodiscard
|
|
---@param c_addr_start integer
|
|
---@param values any
|
|
---@return boolean ok, MODBUS_EXCODE
|
|
local function _15_write_multiple_coils(c_addr_start, values)
|
|
local response = MODBUS_EXCODE.OK
|
|
local _, coils, _, _ = rtu_dev.io_count()
|
|
local count = #values
|
|
local return_ok = ((c_addr_start + count) <= (coils + 1)) and (count > 0)
|
|
|
|
if return_ok then
|
|
for i = 1, count do
|
|
local addr = c_addr_start + i - 1
|
|
local access_fault = rtu_dev.write_coil(addr, values[i])
|
|
|
|
if access_fault then
|
|
return_ok = false
|
|
response = MODBUS_EXCODE.SERVER_DEVICE_FAIL
|
|
break
|
|
end
|
|
end
|
|
else
|
|
response = MODBUS_EXCODE.ILLEGAL_DATA_ADDR
|
|
end
|
|
|
|
return return_ok, response
|
|
end
|
|
|
|
-- write multiple holding registers (analog outputs)
|
|
---@nodiscard
|
|
---@param hr_addr_start integer
|
|
---@param values any
|
|
---@return boolean ok, MODBUS_EXCODE
|
|
local function _16_write_multiple_holding_registers(hr_addr_start, values)
|
|
local response = MODBUS_EXCODE.OK
|
|
local _, _, _, hold_regs = rtu_dev.io_count()
|
|
local count = #values
|
|
local return_ok = ((hr_addr_start + count) <= (hold_regs + 1)) and (count > 0)
|
|
|
|
if return_ok then
|
|
for i = 1, count do
|
|
local addr = hr_addr_start + i - 1
|
|
local access_fault = rtu_dev.write_holding_reg(addr, values[i])
|
|
|
|
if access_fault then
|
|
return_ok = false
|
|
response = MODBUS_EXCODE.SERVER_DEVICE_FAIL
|
|
break
|
|
end
|
|
end
|
|
else
|
|
response = MODBUS_EXCODE.ILLEGAL_DATA_ADDR
|
|
end
|
|
|
|
return return_ok, response
|
|
end
|
|
|
|
---@class modbus
|
|
local public = {}
|
|
|
|
-- validate a request without actually executing it
|
|
---@nodiscard
|
|
---@param packet modbus_frame
|
|
---@return boolean return_code, modbus_packet reply
|
|
function public.check_request(packet)
|
|
local return_code = true
|
|
local response = { MODBUS_EXCODE.ACKNOWLEDGE }
|
|
|
|
if packet.length == 2 then
|
|
-- handle by function code
|
|
if packet.func_code == MODBUS_FCODE.READ_COILS then
|
|
elseif packet.func_code == MODBUS_FCODE.READ_DISCRETE_INPUTS then
|
|
elseif packet.func_code == MODBUS_FCODE.READ_MUL_HOLD_REGS then
|
|
elseif packet.func_code == MODBUS_FCODE.READ_INPUT_REGS then
|
|
elseif packet.func_code == MODBUS_FCODE.WRITE_SINGLE_COIL then
|
|
elseif packet.func_code == MODBUS_FCODE.WRITE_SINGLE_HOLD_REG then
|
|
elseif packet.func_code == MODBUS_FCODE.WRITE_MUL_COILS then
|
|
elseif packet.func_code == MODBUS_FCODE.WRITE_MUL_HOLD_REGS then
|
|
else
|
|
-- unknown function
|
|
return_code = false
|
|
response = { MODBUS_EXCODE.ILLEGAL_FUNCTION }
|
|
end
|
|
else
|
|
-- invalid length
|
|
return_code = false
|
|
response = { MODBUS_EXCODE.NEG_ACKNOWLEDGE }
|
|
end
|
|
|
|
-- default is to echo back<br>
|
|
-- but here we echo back with error flag, on success the "error" will be acknowledgement
|
|
local func_code = bit.bor(packet.func_code, MODBUS_FCODE.ERROR_FLAG)
|
|
|
|
-- create reply
|
|
local reply = comms.modbus_packet()
|
|
reply.make(packet.txn_id, packet.unit_id, func_code, response)
|
|
|
|
return return_code, reply
|
|
end
|
|
|
|
-- handle a MODBUS TCP packet and generate a reply
|
|
---@nodiscard
|
|
---@param packet modbus_frame
|
|
---@return boolean return_code, modbus_packet reply
|
|
function public.handle_packet(packet)
|
|
local return_code ---@type boolean
|
|
local response ---@type table|MODBUS_EXCODE
|
|
|
|
if packet.length >= 2 then
|
|
-- handle by function code
|
|
if packet.func_code == MODBUS_FCODE.READ_COILS then
|
|
return_code, response = _1_read_coils(packet.data[1], packet.data[2])
|
|
elseif packet.func_code == MODBUS_FCODE.READ_DISCRETE_INPUTS then
|
|
return_code, response = _2_read_discrete_inputs(packet.data[1], packet.data[2])
|
|
elseif packet.func_code == MODBUS_FCODE.READ_MUL_HOLD_REGS then
|
|
return_code, response = _3_read_multiple_holding_registers(packet.data[1], packet.data[2])
|
|
elseif packet.func_code == MODBUS_FCODE.READ_INPUT_REGS then
|
|
return_code, response = _4_read_input_registers(packet.data[1], packet.data[2])
|
|
elseif packet.func_code == MODBUS_FCODE.WRITE_SINGLE_COIL then
|
|
return_code, response = _5_write_single_coil(packet.data[1], packet.data[2])
|
|
elseif packet.func_code == MODBUS_FCODE.WRITE_SINGLE_HOLD_REG then
|
|
return_code, response = _6_write_single_holding_register(packet.data[1], packet.data[2])
|
|
elseif packet.func_code == MODBUS_FCODE.WRITE_MUL_COILS then
|
|
return_code, response = _15_write_multiple_coils(packet.data[1], { table.unpack(packet.data, 2, packet.length) })
|
|
elseif packet.func_code == MODBUS_FCODE.WRITE_MUL_HOLD_REGS then
|
|
return_code, response = _16_write_multiple_holding_registers(packet.data[1], { table.unpack(packet.data, 2, packet.length) })
|
|
else
|
|
-- unknown function
|
|
return_code = false
|
|
response = MODBUS_EXCODE.ILLEGAL_FUNCTION
|
|
end
|
|
else
|
|
-- invalid length
|
|
return_code = false
|
|
response = MODBUS_EXCODE.NEG_ACKNOWLEDGE
|
|
end
|
|
|
|
-- default is to echo back
|
|
local func_code = packet.func_code
|
|
if not return_code then
|
|
-- echo back with error flag
|
|
func_code = bit.bor(packet.func_code, MODBUS_FCODE.ERROR_FLAG)
|
|
end
|
|
|
|
if type(response) == "table" then
|
|
elseif response == MODBUS_EXCODE.OK then
|
|
response = {}
|
|
else
|
|
response = { response }
|
|
end
|
|
|
|
-- create reply
|
|
local reply = comms.modbus_packet()
|
|
reply.make(packet.txn_id, packet.unit_id, func_code, response)
|
|
|
|
return return_code, reply
|
|
end
|
|
|
|
return public
|
|
end
|
|
|
|
-- return a SERVER_DEVICE_BUSY error reply
|
|
---@nodiscard
|
|
---@param packet modbus_frame MODBUS packet frame
|
|
---@return modbus_packet reply
|
|
function modbus.reply__srv_device_busy(packet)
|
|
-- reply back with error flag and exception code
|
|
local reply = comms.modbus_packet()
|
|
local fcode = bit.bor(packet.func_code, MODBUS_FCODE.ERROR_FLAG)
|
|
local data = { MODBUS_EXCODE.SERVER_DEVICE_BUSY }
|
|
reply.make(packet.txn_id, packet.unit_id, fcode, data)
|
|
return reply
|
|
end
|
|
|
|
-- return a NEG_ACKNOWLEDGE error reply
|
|
---@nodiscard
|
|
---@param packet modbus_frame MODBUS packet frame
|
|
---@return modbus_packet reply
|
|
function modbus.reply__neg_ack(packet)
|
|
-- reply back with error flag and exception code
|
|
local reply = comms.modbus_packet()
|
|
local fcode = bit.bor(packet.func_code, MODBUS_FCODE.ERROR_FLAG)
|
|
local data = { MODBUS_EXCODE.NEG_ACKNOWLEDGE }
|
|
reply.make(packet.txn_id, packet.unit_id, fcode, data)
|
|
return reply
|
|
end
|
|
|
|
-- return a GATEWAY_PATH_UNAVAILABLE error reply
|
|
---@nodiscard
|
|
---@param packet modbus_frame MODBUS packet frame
|
|
---@return modbus_packet reply
|
|
function modbus.reply__gw_unavailable(packet)
|
|
-- reply back with error flag and exception code
|
|
local reply = comms.modbus_packet()
|
|
local fcode = bit.bor(packet.func_code, MODBUS_FCODE.ERROR_FLAG)
|
|
local data = { MODBUS_EXCODE.GATEWAY_PATH_UNAVAILABLE }
|
|
reply.make(packet.txn_id, packet.unit_id, fcode, data)
|
|
return reply
|
|
end
|
|
|
|
return modbus
|