Files
epics-base/modules/libcom/src/iocsh/atInit.c

100 lines
2.4 KiB
C

/*************************************************************************\
* Copyright (C) 2020 Dirk Zimoch
* Copyright (C) 2020-2025 European Spallation Source, ERIC
* SPDX-License-Identifier: EPICS
* EPICS BASE is distributed subject to a Software License Agreement found
* in file LICENSE that is included with this distribution.
\*************************************************************************/
#include <cantProceed.h>
#include <ellLib.h>
#include <errlog.h>
#include <initHooks.h>
#include <iocsh.h>
#include <string.h>
#include "atInit.h"
static const char helpMessage[] =
"Allows you to define commands to be run after the iocInit\n"
"Example commands:\n"
" atInit \"dbpf <PV> <VAL>\"\n"
" atInit \"date\"\n";
struct cmditem {
ELLNODE node;
char *cmd;
};
static ELLLIST s_cmdlist = ELLLIST_INIT;
static int s_initendflag = 0; // Defines the end of the initialization
static void atInitHook(initHookState state)
{
if (state != initHookAfterIocRunning)
return;
struct cmditem *item = NULL;
while ((item = (struct cmditem *)ellGet(&s_cmdlist))) {
printf("%s\n", item->cmd);
if (iocshCmd(item->cmd))
printf(ERL_ERROR " atInit: "
"command '%s' failed to run\n",
item->cmd);
free(item);
}
s_initendflag = 1;
}
static struct cmditem *newItem(const char *cmd)
{
const size_t cmd_len = strlen(cmd) + 1;
struct cmditem *item = mallocMustSucceed(sizeof(struct cmditem) + cmd_len, "atInit");
item->cmd = (char *)(item + 1);
memcpy(item->cmd, cmd, cmd_len);
ellAdd(&s_cmdlist, &item->node);
return item;
}
static const iocshFuncDef atInitDef = {
"atInit",
1,
(const iocshArg *[]){&(iocshArg){"command (before iocInit)", iocshArgString}},
helpMessage};
static void atInitFunc(const iocshArgBuf *args)
{
char *cmd = args[0].sval;
if (s_initendflag) {
printf(ERL_WARNING " atInit: "
"can only be used before iocInit (check help)\n");
return;
}
if (!cmd || !cmd[0]) {
printf(ERL_WARNING " atInit: "
"received an empty argument (check help)\n");
return;
}
newItem(cmd);
}
void atInitRegister(void)
{
static int first_time = 1;
if (first_time) {
first_time = 0;
iocshRegister(&atInitDef, atInitFunc);
initHookRegister(atInitHook);
}
}