-module(tda). -export([start_client/0, client/1]). -include("acse.hrl"). -include("rose.hrl"). -define(TIMEOUT, 300000). -define(CONNECT_TIMEOUT, 3000). -define(HOST, "192.168.240.20"). associate_request() -> {ok, UI} = acse:encode('ACSEUserInformationForCSTA', {newDefinition, #'NewACSEUserInformationForCSTA'{cSTAVersion = [versionFive]}}), Pdu = #'AARQ-apdu'{ 'protocol-version' = [version1], 'application-context-name' = {1, 3, 12, 0, 218}, 'user-information' = [#'EXTERNAL'{ 'direct-reference' = {1, 3, 12, 0, 285, 200}, encoding = {'single-ASN1-type', UI}}]}, acse:encode('ACSE-apdu', {aarq, Pdu}). start_client() -> register(tda_client, spawn(?MODULE, client, [?HOST])). client(Host) -> case gen_tcp:connect(Host, 33333, [binary, {active, true}, {packet, 2}], ?CONNECT_TIMEOUT) of {ok, Sock} -> {ok, Hello} = associate_request(), gen_tcp:send(Sock, Hello), loop(Sock); {error, Reason} -> {error, Reason} end. loop(Sock) -> io:format("loop~n", []), receive {tcp, Sock, Msg} -> case decode(Sock, Msg) of ok -> loop(Sock); error -> gen_tcp:close(Sock), error end; {tcp_closed, _} -> closed after ?TIMEOUT -> gen_tcp:close(Sock), timeout end. decode(Sock, Msg) -> case dispatch(Msg) of rose -> {ok, Rose} = rose:decode('ROS', Msg), io:format("ROSE> ~p~n", [Rose]), rose_handler(Sock, Rose); acse -> {ok, Acse} = acse:decode('ACSE-apdu', Msg), io:format("ACSE> ~p~n", [Acse]), acse_handler(Acse) end. rose_handler(Sock, {invoke, Rose}) -> gen_tcp:send(Sock, invoke_handler(Rose)), ok. invoke_handler(#'Invoke'{invokeId = Id, opcode = Op}) -> case Op of {_, 211} -> system_status(Id, Op) end. dispatch(<>) -> case Head of 96 -> acse; 97 -> acse; 98 -> acse; 99 -> acse; 100 -> acse; 161 -> rose; 162 -> rose end. system_status(Id, Op) -> RR = #'ReturnResult_result'{opcode = Op, result = <<5,0>>}, R = #'ReturnResult'{invokeId = Id, result = RR}, case rose:encode('ROS', {returnResult, R}) of {ok, Invoke} -> Invoke; {error, Reason} -> Reason end. acse_handler({aarq, _}) -> error; acse_handler({aare, Apdu}) -> case Apdu#'AARE-apdu'.result of accepted -> ok; 'rejected-permanent' -> error; 'rejected-transient' -> error end; acse_handler({rlrq, _}) -> error; acse_handler({rlre, _}) -> error; acse_handler({abrt, _}) -> error.