/**************************************************************************** * This file is part of the project AqHome. * AqHome (c) by 2023 Martin Preuss, all rights reserved. * * The license for this file can be found in the file COPYING which you * should have received along with this file. ****************************************************************************/ #ifdef HAVE_CONFIG_H # include #endif #include "./ping.h" #include "aqhome/ipc/msg_ipc_setaccmsggrps.h" #include "aqhome/ipc/msg_ipc_ping.h" #include "aqhome/ipc/msg_ipc_forward.h" #include "aqhome/ipc/endpoint_ipc_tcpc.h" #include "aqhome/msg/msg_node.h" #include #include #include #include #include #include #include #define I18S(msg) msg #define I18N(msg) GWEN_I18N_Translate(PACKAGE, msg) static int _doPing(GWEN_DB_NODE *dbArgs); static int _sendAcceptedMsgGroups(GWEN_MSG_ENDPOINT *epTcp); static int _sendPing(GWEN_MSG_ENDPOINT *epTcp, int nodeAddr); static GWEN_MSG *_waitForSpecificNodeMessage(GWEN_MSG_ENDPOINT_MGR *emgr, GWEN_MSG_ENDPOINT *epTcp, int msgCode, int nodeAddr, int timeoutInSeconds); static GWEN_MSG_ENDPOINT *_setupIpcEndpoint(GWEN_MSG_ENDPOINT_MGR *emgr, GWEN_DB_NODE *dbArgs); int AQH_Tool_Ping(GWEN_DB_NODE *dbGlobalArgs, int argc, char **argv) { GWEN_DB_NODE *dbLocalArgs; int rv; const GWEN_ARGS args[]= { { GWEN_ARGS_FLAGS_HAS_ARGUMENT, /* flags */ GWEN_ArgsType_Char, /* type */ "tcpAddress", /* name */ 0, /* minnum */ 1, /* maxnum */ "t", /* short option */ "tcpaddress", /* long option */ I18S("Specify TCP address to connect to (defaults to 127.0.0.1)"), I18S("Specify TCP address to connect to (defaults to 127.0.0.1)") }, { GWEN_ARGS_FLAGS_HAS_ARGUMENT, /* flags */ GWEN_ArgsType_Int, /* type */ "tcpPort", /* name */ 0, /* minnum */ 1, /* maxnum */ "P", /* short option */ "tcpport", /* long option */ I18S("Specify the TCP port to listen on"), I18S("Specify the TCP port to listen on") }, { GWEN_ARGS_FLAGS_HAS_ARGUMENT, /* flags */ GWEN_ArgsType_Int, /* type */ "nodeAddr", /* name */ 1, /* minnum */ 1, /* maxnum */ "n", /* short option */ "nodeaddr", /* long option */ I18S("Specify bus addr of the node to ping"), I18S("Specify bus addr of the node to ping") }, { GWEN_ARGS_FLAGS_HAS_ARGUMENT, /* flags */ GWEN_ArgsType_Int, /* type */ "timeout", /* name */ 0, /* minnum */ 1, /* maxnum */ "T", /* short option */ NULL, /* long option */ I18S("Specify timeout in seconds for PONG response"), I18S("Specify timeout in seconds for PONG response") }, { GWEN_ARGS_FLAGS_HELP | GWEN_ARGS_FLAGS_LAST, /* flags */ GWEN_ArgsType_Int, /* type */ "help", /* name */ 0, /* minnum */ 0, /* maxnum */ "h", /* short option */ "help", /* long option */ "Show this help screen", /* short description */ "Show this help screen" /* long description */ } }; dbLocalArgs=GWEN_DB_GetGroup(dbGlobalArgs, GWEN_DB_FLAGS_DEFAULT, "local"); rv=GWEN_Args_Check(argc, argv, 1, GWEN_ARGS_MODE_ALLOW_FREEPARAM, args, dbLocalArgs); if (rv==GWEN_ARGS_RESULT_ERROR) { fprintf(stderr, "ERROR: Could not parse arguments\n"); return 1; } else if (rv==GWEN_ARGS_RESULT_HELP) { GWEN_BUFFER *ubuf; ubuf=GWEN_Buffer_new(0, 1024, 0, 1); if (GWEN_Args_Usage(args, ubuf, GWEN_ArgsOutType_Txt)) { fprintf(stderr, "ERROR: Could not create help string\n"); return 1; } fprintf(stderr, "%s\n", GWEN_Buffer_GetStart(ubuf)); GWEN_Buffer_free(ubuf); return 0; } return _doPing(dbLocalArgs); } int _doPing(GWEN_DB_NODE *dbArgs) { GWEN_MSG_ENDPOINT_MGR *emgr; GWEN_MSG_ENDPOINT *epTcp; int rv; int nodeAddr; int timeoutInSeconds; GWEN_MSG *msg; emgr=GWEN_MsgEndpointMgr_new(); epTcp=_setupIpcEndpoint(emgr, dbArgs); if (epTcp==NULL) { DBG_ERROR(NULL, "ERROR creating TCP connection"); return 2; } GWEN_MsgEndpointMgr_AddEndpoint(emgr, epTcp); nodeAddr=GWEN_DB_GetIntValue(dbArgs, "nodeAddr", 0, 0); timeoutInSeconds=GWEN_DB_GetIntValue(dbArgs, "timeout", 0, 5); rv=_sendAcceptedMsgGroups(epTcp); if (rv<0) { DBG_INFO(NULL, "here (%d)", rv); return 2; } rv=_sendPing(epTcp, nodeAddr); if (rv<0) { DBG_INFO(NULL, "here (%d)", rv); return 2; } msg=_waitForSpecificNodeMessage(emgr, epTcp, AQH_MSG_TYPE_PONG, nodeAddr, timeoutInSeconds); if (msg==NULL) { DBG_INFO(NULL, "No PONG response received."); GWEN_MsgEndpointMgr_free(emgr); return 2; } DBG_INFO(NULL, "PONG response received"); GWEN_MsgEndpointMgr_free(emgr); return 0; } GWEN_MSG *_waitForSpecificNodeMessage(GWEN_MSG_ENDPOINT_MGR *emgr, GWEN_MSG_ENDPOINT *epTcp, int msgCode, int nodeAddr, int timeoutInSeconds) { time_t startTime; startTime=time(NULL); for (;;) { GWEN_MSG *msg; time_t now; GWEN_MsgEndpointMgr_RunAllEndpoints(emgr); GWEN_MsgEndpointMgr_IoLoopOnce(emgr); msg=GWEN_MsgEndpoint_TakeFirstReceivedMessage(epTcp); if (msg) { if (GWEN_IpcMsg_GetCode(msg)==AQH_MSGTYPE_IPC_FORWARD) { GWEN_MSG *nodeMsg; DBG_INFO(NULL, "Received IPC FORWARD message"); nodeMsg=AQH_ForwardIpcMsg_GetCopyOfNodeMsg(msg); if (nodeMsg) { DBG_ERROR(AQH_LOGDOMAIN, "Received node msg from %d (%d)", AQH_NodeMsg_GetSourceAddress(nodeMsg), AQH_NodeMsg_GetMsgType(nodeMsg)); if (AQH_NodeMsg_GetMsgType(nodeMsg)==msgCode && AQH_NodeMsg_GetSourceAddress(nodeMsg)==nodeAddr) { GWEN_Msg_free(msg); return nodeMsg; } } } GWEN_Msg_free(msg); } now=time(NULL); if (now-startTime>timeoutInSeconds) { DBG_INFO(NULL, "Timeout"); break; } } return NULL; } int _sendAcceptedMsgGroups(GWEN_MSG_ENDPOINT *epTcp) { GWEN_MSG *msgOut; msgOut=AQH_SetAcceptedMsgGroupsIpcMsg_new(AQH_MSGTYPE_IPC_SETACCMSGGRPS, AQH_MSG_TYPEGROUP_ALL); if (msgOut==NULL) { DBG_ERROR(NULL, "Error creating message"); return GWEN_ERROR_GENERIC; } GWEN_MsgEndpoint_AddSendMessage(epTcp, msgOut); return 0; } int _sendPing(GWEN_MSG_ENDPOINT *epTcp, int nodeAddr) { GWEN_MSG *msgOut; msgOut=AQH_PingIpcMsg_new(AQH_MSGTYPE_IPC_PING, nodeAddr); if (msgOut==NULL) { DBG_ERROR(NULL, "Error creating message"); return GWEN_ERROR_GENERIC; } GWEN_MsgEndpoint_AddSendMessage(epTcp, msgOut); return 0; } GWEN_MSG_ENDPOINT *_setupIpcEndpoint(GWEN_MSG_ENDPOINT_MGR *emgr, GWEN_DB_NODE *dbArgs) { GWEN_MSG_ENDPOINT *epTcp; const char *tcpAddress; int tcpPort; tcpAddress=GWEN_DB_GetCharValue(dbArgs, "tcpAddress", 0, "127.0.0.1"); tcpPort=GWEN_DB_GetIntValue(dbArgs, "tcpPort", 0, 45454); DBG_INFO(NULL, "Setup tcp client endpoint to %s:%d", tcpAddress, tcpPort); epTcp=AQH_IpcTcpClientEndpoint_new(tcpAddress, tcpPort, "aqhome-tool-IPC", 0); if (epTcp==NULL) { DBG_ERROR(NULL, "Error creating endpoint TCPc"); return NULL; } return epTcp; }