#8 work in progress on RTU sessions and added unit object

This commit is contained in:
Mikayla Fischler 2022-05-09 09:35:39 -04:00
parent 469ee29b5a
commit 679d98c8bf
2 changed files with 262 additions and 0 deletions

View File

@ -1,3 +1,183 @@
local comms = require("scada-common.comms")
local log = require("scada-common.log")
local mqueue = require("scada-common.mqueue")
local util = require("scada-common.util")
local rtu = {}
local PROTOCOLS = comms.PROTOCOLS
local RPLC_TYPES = comms.RPLC_TYPES
local SCADA_MGMT_TYPES = comms.SCADA_MGMT_TYPES
local print = util.print
local println = util.println
local print_ts = util.print_ts
local println_ts = util.println_ts
rtu.new_session = function (id, in_queue, out_queue)
local log_header = "rtu_session(" .. id .. "): "
local self = {
id = id,
in_q = in_queue,
out_q = out_queue,
commanded_state = false,
commanded_burn_rate = 0.0,
ramping_rate = false,
-- connection properties
seq_num = 0,
r_seq_num = nil,
connected = true,
received_struct = false,
received_status_cache = false,
rtu_conn_watchdog = util.new_watchdog(3),
last_rtt = 0
}
-- send a MODBUS TCP packet
local send_modbus = function (m_pkt)
local s_pkt = comms.scada_packet()
s_pkt.make(self.seq_num, PROTOCOLS.MODBUS_TCP, m_pkt.raw_sendable())
self.modem.transmit(self.s_port, self.l_port, s_pkt.raw_sendable())
self.seq_num = self.seq_num + 1
end
-- send a SCADA management packet
local _send_mgmt = function (msg_type, msg)
local s_pkt = comms.scada_packet()
local m_pkt = comms.mgmt_packet()
m_pkt.make(msg_type, msg)
s_pkt.make(self.seq_num, PROTOCOLS.SCADA_MGMT, m_pkt.raw_sendable())
self.out_q.push_packet(s_pkt)
self.seq_num = self.seq_num + 1
end
-- handle a packet
local _handle_packet = function (pkt)
-- check sequence number
if self.r_seq_num == nil then
self.r_seq_num = pkt.scada_frame.seq_num()
elseif self.r_seq_num >= pkt.scada_frame.seq_num() then
log.warning(log_header .. "sequence out-of-order: last = " .. self.r_seq_num .. ", new = " .. pkt.scada_frame.seq_num())
return
else
self.r_seq_num = pkt.scada_frame.seq_num()
end
-- process packet
if pkt.scada_frame.protocol() == PROTOCOLS.MODBUS_TCP then
-- feed watchdog
self.rtu_conn_watchdog.feed()
elseif pkt.scada_frame.protocol() == PROTOCOLS.SCADA_MGMT then
-- feed watchdog
self.rtu_conn_watchdog.feed()
if pkt.type == SCADA_MGMT_TYPES.CLOSE then
-- close the session
self.connected = false
elseif pkt.type == SCADA_MGMT_TYPES.RTU_ADVERT then
-- RTU unit advertisement
for i = 1, packet.length do
local unit = packet.data[i]
unit
end
elseif pkt.type == SCADA_MGMT_TYPES.RTU_HEARTBEAT then
-- periodic RTU heartbeat
else
log.debug(log_header .. "handler received unsupported SCADA_MGMT packet type " .. pkt.type)
end
end
end
-- PUBLIC FUNCTIONS --
-- get the session ID
local get_id = function () return self.id end
-- check if a timer matches this session's watchdog
local check_wd = function (timer)
return timer == self.rtu_conn_watchdog.get_timer()
end
-- close the connection
local close = function ()
self.rtu_conn_watchdog.cancel()
self.connected = false
_send_mgmt(SCADA_MGMT_TYPES.CLOSE, {})
println(log_header .. "connection to RTU closed by server")
log.info(log_header .. "session closed by server")
end
-- iterate the session
local iterate = function ()
if self.connected then
------------------
-- handle queue --
------------------
local handle_start = util.time()
while self.in_q.ready() and self.connected do
-- get a new message to process
local message = self.in_q.pop()
if message.qtype == mqueue.TYPE.PACKET then
-- handle a packet
_handle_packet(message.message)
elseif message.qtype == mqueue.TYPE.COMMAND then
-- handle instruction
local cmd = message.message
elseif message.qtype == mqueue.TYPE.DATA then
-- instruction with body
local cmd = message.message
end
-- max 100ms spent processing queue
if util.time() - handle_start > 100 then
log.warning(log_header .. "exceeded 100ms queue process limit")
break
end
end
-- exit if connection was closed
if not self.connected then
self.rtu_conn_watchdog.cancel()
println(log_header .. "connection to RTU closed by remote host")
log.info(log_header .. "session closed by remote host")
return self.connected
end
----------------------
-- update periodics --
----------------------
local elapsed = util.time() - self.periodics.last_update
local periodics = self.periodics
-- keep alive
periodics.keep_alive = periodics.keep_alive + elapsed
if periodics.keep_alive >= PERIODICS.KEEP_ALIVE then
-- _send(RPLC_TYPES.KEEP_ALIVE, { util.time() })
periodics.keep_alive = 0
end
self.periodics.last_update = util.time()
end
return self.connected
end
return {
get_id = get_id,
check_wd = check_wd,
close = close,
iterate = iterate
}
end
return rtu

82
supervisor/unit.lua Normal file
View File

@ -0,0 +1,82 @@
local unit = {}
unit.new = function (for_reactor)
local public = {}
local self = {
r_id = for_reactor,
plc_s = nil,
turbines = {},
boilers = {},
energy_storage = {},
redstone = {},
db = {
annunciator = {
-- RPS
-- reactor
PLCOnline = false,
ReactorTrip = false,
ManualReactorTrip = false,
RCPTrip = false,
RCSFlowLow = false,
ReactorTempHigh = false,
ReactorHighDeltaT = false,
ReactorOverPower = false,
HighStartupRate = false,
-- boiler
BoilerOnline = false,
HeatingRateLow = false,
CoolantFeedMismatch = false,
-- turbine
TurbineOnline = false,
SteamFeedMismatch = false,
SteamDumpOpen = false,
TurbineTrip = false,
TurbineOverUnderSpeed = false
}
}
}
public.link_plc_session = function (plc_session)
self.plc_s = plc_session
end
public.add_turbine = function (turbine)
table.insert(self.turbines, turbine)
end
public.add_boiler = function (turbine)
table.insert(self.boilers, boiler)
end
public.add_redstone = function (field, accessor)
-- ensure field exists
if redstone[field] == nil then
redstone[field] = {}
end
-- insert into list
table.insert(redstone[field], accessor)
end
local _update_annunciator = function ()
self.db.annunciator.PLCOnline = (self.plc_s ~= nil) and (self.plc_s.open)
self.db.annunciator.ReactorTrip = false
end
public.update = function ()
-- unlink PLC if session was closed
if not self.plc_s.open then
self.plc_s = nil
end
-- update annunciator logic
_update_annunciator()
end
public.get_annunciator = function () return self.db.annunciator end
return public
end
return unit