/**************************************************************************** * 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 "./init.h" #include "./aqhome_data_p.h" #include "aqhome/ipc/endpoint_ipc.h" #include #include #include #include #include #include #ifdef HAVE_SYS_TYPES_H # include #endif #ifdef HAVE_SYS_STAT_H # include #endif #include #include #include #include #include #include /* ------------------------------------------------------------------------------------------------ * defines * ------------------------------------------------------------------------------------------------ */ //#define I18N(msg) msg #define I18S(msg) msg /* ------------------------------------------------------------------------------------------------ * forward declarations * ------------------------------------------------------------------------------------------------ */ static int _setupStorage(AQHOME_DATA *aqh, GWEN_DB_NODE *dbArgs); static void _setupIpc(AQHOME_DATA *aqh, GWEN_DB_NODE *dbArgs); static GWEN_MSG_ENDPOINT *_acceptIpcFn(GWEN_MSG_ENDPOINT *ep, GWEN_SOCKET *sk, const GWEN_INETADDRESS *addr, void *data); static int _readArgs(int argc, char **argv, GWEN_DB_NODE *dbArgs); static int _createPidFile(const char *pidFilename); /* ------------------------------------------------------------------------------------------------ * implementations * ------------------------------------------------------------------------------------------------ */ int AqHomeData_Init(AQHOME_DATA *aqh, int argc, char **argv) { GWEN_DB_NODE *dbArgs; int rv; const char *s; dbArgs=GWEN_DB_Group_new("args"); rv=_readArgs(argc, argv, dbArgs); if (rv<0) { DBG_ERROR(NULL, "Error reading args (%d)", rv); return rv; } aqh->dbArgs=dbArgs; aqh->timeout=GWEN_DB_GetIntValue(dbArgs, "timeout", 0, 0); s=GWEN_DB_GetCharValue(dbArgs, "pidfile", 0, AQHOME_DATA_DEFAULT_PIDFILE); if (s && *s) { free(aqh->pidFile); aqh->pidFile=strdup(s); rv=_createPidFile(s); if (rv<0) { DBG_ERROR(NULL, "Error creating PID file (%d)", rv); return rv; } } rv=_setupStorage(aqh, dbArgs); if (rv<0) { DBG_INFO(NULL, "here (%d)", rv); return rv; } _setupIpc(aqh, dbArgs); return 0; } int _setupStorage(AQHOME_DATA *aqh, GWEN_DB_NODE *dbArgs) { const char *dataFolder; GWEN_BUFFER *nameBuf; AQH_STORAGE *sto; int rv; dataFolder=GWEN_DB_GetCharValue(dbArgs, "dataFolder", 0, AQHOME_DATA_DEFAULT_DATADIR); nameBuf=GWEN_Buffer_new(0, 256, 0, 1); GWEN_Buffer_AppendArgs(nameBuf, "%s%s%s", dataFolder, GWEN_DIR_SEPARATOR_S, AQHOME_DATA_STATEFILENAME); sto=AQH_Storage_new(); AQH_Storage_SetStateFile(sto, GWEN_Buffer_GetStart(nameBuf)); AQH_Storage_SetDataFileFolder(sto, dataFolder); GWEN_Buffer_free(nameBuf); rv=AQH_Storage_Init(sto); if (rv<0) { DBG_INFO(NULL, "here (%d)", rv); AQH_Storage_free(sto); return rv; } aqh->storage=sto; return 0; } void _setupIpc(AQHOME_DATA *aqh, GWEN_DB_NODE *dbArgs) { const char *tcpAddress; int tcpPort; tcpAddress=GWEN_DB_GetCharValue(dbArgs, "tcpAddress", 0, NULL); tcpPort=GWEN_DB_GetIntValue(dbArgs, "tcpPort", 0, AQHOME_DATA_DEFAULT_IPC_PORT); if (tcpAddress && *tcpAddress && tcpPort) { GWEN_MSG_ENDPOINT *ep; DBG_INFO(NULL, "Starting TCP service on \"%s\":%d", tcpAddress, tcpPort); ep=GWEN_TcpdEndpoint_new(tcpAddress, tcpPort, NULL, 0); GWEN_TcpdEndpoint_SetAcceptFn(ep, _acceptIpcFn, aqh); aqh->ipcdEndpoint=ep; } } GWEN_MSG_ENDPOINT *_acceptIpcFn(GWEN_MSG_ENDPOINT *ep, GWEN_SOCKET *sk, const GWEN_INETADDRESS *addr, GWEN_UNUSED void *data) { /* AQHOME_DATA *aqh; * * aqh=(AQHOME_DATA*) data; */ DBG_INFO(NULL, "Incoming IPC connection"); return AQH_IpcEndpoint_CreateIpcTcpServiceForSocket(sk, NULL, 0); } int _createPidFile(const char *pidFilename) { FILE *f; int pidfd; if (remove(pidFilename)==0) { DBG_ERROR(0, "Old PID file existed, removed. (Unclean shutdown?)"); } #ifdef HAVE_SYS_STAT_H pidfd = open(pidFilename, O_EXCL|O_CREAT|O_WRONLY, S_IRUSR|S_IWUSR|S_IRGRP|S_IROTH); if (pidfd < 0) { DBG_ERROR(NULL, "Could not create PID file \"%s\" (%s), aborting.", pidFilename, strerror(errno)); return GWEN_ERROR_IO; } f = fdopen(pidfd, "w"); #else /* HAVE_STAT_H */ f=fopen(pidFilename,"w+"); #endif /* HAVE_STAT_H */ /* write pid */ #ifdef HAVE_GETPID fprintf(f,"%d\n",getpid()); #else fprintf(f,"-1\n"); #endif if (fclose(f)) { DBG_ERROR(0, "Could not close PID file \"%s\" (%s), aborting.", pidFilename, strerror(errno)); return GWEN_ERROR_IO; } return 0; } int _readArgs(int argc, char **argv, GWEN_DB_NODE *dbArgs) { 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 the TCP address to listen on (disabled if missing)"), I18S("Specify the TCP address to listen on (disabled if missing)") }, { 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_Char, /* type */ "datafolder", /* name */ 0, /* minnum */ 1, /* maxnum */ NULL, /* short option */ "datafolder", /* long option */ I18S("Folder where data files are stored"), I18S("Folder where data files are stored") }, { GWEN_ARGS_FLAGS_HAS_ARGUMENT, /* flags */ GWEN_ArgsType_Char, /* type */ "pidfile", /* name */ 0, /* minnum */ 1, /* maxnum */ "p", /* short option */ "pidfile", /* long option */ I18S("Specify the PID file"), I18S("Specify the PID file") }, { GWEN_ARGS_FLAGS_HAS_ARGUMENT, /* flags */ GWEN_ArgsType_Int, /* type */ "timeout", /* name */ 0, /* minnum */ 1, /* maxnum */ "T", /* short option */ "timeout", /* long option */ I18S("Specify timeout in second (default: no timeout)"), I18S("Specify timeout in second (default: no timeout)") }, { GWEN_ARGS_FLAGS_HELP | GWEN_ARGS_FLAGS_LAST, /* flags */ GWEN_ArgsType_Int, /* type */ "help", /* name */ 0, /* minnum */ 0, /* maxnum */ "h", /* short option */ "help", I18S("Show this help screen."), I18S("Show this help screen.") } }; rv=GWEN_Args_Check(argc, argv, 1, 0, args, dbArgs); if (rv==GWEN_ARGS_RESULT_ERROR) { fprintf(stderr, "ERROR: Could not parse arguments main\n"); return GWEN_ERROR_INVALID; } else if (rv==GWEN_ARGS_RESULT_HELP) { GWEN_BUFFER *ubuf; ubuf=GWEN_Buffer_new(0, 1024, 0, 1); GWEN_Buffer_AppendArgs(ubuf, I18N("This is version %s.\nUsage: %s [OPTIONS]\n\nOptions:\n"), AQHOME_VERSION_STRING, argv[0]); if (GWEN_Args_Usage(args, ubuf, GWEN_ArgsOutType_Txt)) { fprintf(stderr, "ERROR: Could not create help string\n"); return 1; } GWEN_Buffer_AppendString(ubuf, "\n"); fprintf(stdout, "%s\n", GWEN_Buffer_GetStart(ubuf)); GWEN_Buffer_free(ubuf); return GWEN_ERROR_CLOSE; } return 0; }