Belle II Software development
rfctl.cc
1/**************************************************************************
2 * basf2 (Belle II Analysis Software Framework) *
3 * Author: The Belle II Collaboration *
4 * *
5 * See git log for contributors and copyright holders. *
6 * This file is licensed under LGPL-3.0, see LICENSE.md. *
7 **************************************************************************/
8// ----------------------------------------------------------------------
9// master.cc
10//
11// A simple-minded NSM2 master program which does:
12// - text-base command line input with readline
13// - log message storage in a file
14// (no screen output as it conflicts with readline)
15// ----------------------------------------------------------------------
16#include <stdio.h>
17#include <stdlib.h>
18#include <string.h>
19#include <ctype.h>
20#include <stdarg.h>
21#include <time.h>
22#include <sys/time.h>
23#include <sys/stat.h>
24#include "nsm2/belle2nsm.h"
25
26extern "C" {
27#ifdef USE_READLINE
28#include <readline/readline.h>
29#undef CTRL /* also defined in sys/ioctl.h in solaris */
30 extern int add_history(char*);
31#endif
32}
33
34// -- global variables --------------------------------------------------
35const char* master_logdir = ".";
36FILE* master_logfp = 0;
37
38// -- xreopenlog ----------------------------------------------------
39// log directory
40// ----------------------------------------------------------------------
41void
42xreopenlog()
43{
44 static char logprefix[1024];
45 char logfile[2048];
46 time_t now = time(0);
47 const tm* cur = localtime(&now);
48 struct stat statbuf;
49
50 if (! logprefix[0]) {
51 strcpy(logprefix, master_logdir);
52 if (! logprefix[0]) strcpy(logprefix, ".");
53 if (logprefix[strlen(logprefix) - 1] == '/') {
54 logprefix[strlen(logprefix) - 1] = 0;
55 }
56 if (stat(logprefix, &statbuf) >= 0 && S_ISDIR(statbuf.st_mode)) {
57 strcat(logprefix, "/master");
58 }
59 }
60
61 sprintf(logfile, "%s.%04d%02d%02d.log", logprefix,
62 cur->tm_year + 1900, cur->tm_mon + 1, cur->tm_mday);
63 if (master_logfp != stdout) {
64 if (master_logfp) fclose(master_logfp);
65 if (!(master_logfp = fopen(logfile, "a"))) {
66 printf("cannot open logfile %s\n", logfile);
67 exit(1);
68 }
69 }
70}
71// -- xprintlog ---------------------------------------------------------
72// printlog
73// ----------------------------------------------------------------------
74void
75xprintlog(const char* fmt, ...)
76{
77 void xlogtime(char* buf);
78
79 if (master_logfp) {
80 va_list ap;
81 va_start(ap, fmt);
82 char datebuf[32];
83 xlogtime(datebuf);
84 fputs(datebuf, master_logfp);
85 vfprintf(master_logfp, fmt, ap);
86 fputc('\n', master_logfp);
87 va_end(ap);
88
89 fflush(master_logfp);
90 }
91}
92// -- xlogtime ----------------------------------------------------------
93// logtime
94// ----------------------------------------------------------------------
95void
96xlogtime(char* buf)
97{
98 timeval now;
99 const tm* cur;
100
101 gettimeofday(&now, 0);
102 cur = localtime(static_cast<time_t*>(&now.tv_sec));
103 /*
104 static int lastday = -1;
105 if (lastday != cur->tm_yday) {
106 char datebuf[128];
107 lastday = cur->tm_yday;
108 xreopenlog();
109 xprintlog(
110 "------------------------------------------------------------------");
111 sprintf(datebuf, "Date: %04d.%02d.%02d",
112 cur->tm_year + 1900, cur->tm_mon + 1, cur->tm_mday);
113 xprintlog(datebuf);
114 }
115 */
116 sprintf(buf, "%02d:%02d:%02d.%03d ",
117 cur->tm_hour, cur->tm_min, cur->tm_sec, (int)now.tv_usec / 1000);
118}
119// -- xgetargs ----------------------------------------------------------
120int
121xgetargs(char* buf, char* argv[])
122{
123 int argc = 0, inword = 0, inquote = 0;
124 char* p = buf;
125
126 while (*p) {
127 if (*p == '\\') {
128 if (*(p + 1) == 'n') {
129 memmove(p, p + 1, strlen(p));
130 *p = '\n';
131 } else {
132 memmove(p, p + 1, strlen(p));
133 }
134 } else if (*p == '"') {
135 inquote = !inquote;
136 memmove(p, p + 1, strlen(p));
137 if (! inword) {
138 argc++;
139 *argv++ = p;
140 inword = 1;
141 }
142 p--;
143 } else if (isspace(*p) && !inquote && inword) {
144 *p = 0;
145 inword = 0;
146 } else if (! isspace(*p) && !inword) {
147 inword = 1;
148 argc++;
149 *argv++ = p;
150 }
151 p++;
152 }
153 if (inquote) return -1;
154 return argc;
155}
156// -- ok_handler --------------------------------------------------------
157//
158// ----------------------------------------------------------------------
159void
160ok_handler(NSMmsg* /*msg*/, NSMcontext*)
161{
162 xprintlog("OK received");
163}
164// -- error_handler -----------------------------------------------------
165//
166// ----------------------------------------------------------------------
167void
168error_handler(NSMmsg* /*msg*/, NSMcontext*)
169{
170 xprintlog("ERROR received");
171}
172
173// -- msg_handler -----------------------------------------------------
174//
175// ----------------------------------------------------------------------
176void
177msg_handler(NSMmsg* /*msg*/, NSMcontext*)
178{
179 xprintlog("MSG received");
180}
181// -- main --------------------------------------------------------------
182// main does everything except callback functions
183// ----------------------------------------------------------------------
184int
185main(int argc, const char** argv)
186{
187 const char* program = argv[0];
188 const char* nodename = argv[1]; // need to check before using
189 const char myname[] = "rfctl";
190
191 char* buf = 0;
192 char* av[128];
193 char* prompt;
194
195 // xreopenlog();
196 master_logfp = stdout;
197
198 // ARGV check
199 if (argc < 2) {
200 nodename = myname;
201 }
202
203 // INIT
204 if (! b2nsm_init(nodename)) {
205 xprintlog("%s: INIT %s", program, b2nsm_strerror());
206 return 1;
207 }
208 xprintlog("%s: INIT", program);
209 b2nsm_logging(master_logfp);
210
211 // handlers
212 if (b2nsm_callback("OK", ok_handler) < 0) {
213 xprintlog("%s: CALLBACK(OK) %s", program, b2nsm_strerror());
214 return 1;
215 }
216 if (b2nsm_callback("ERROR", ok_handler) < 0) {
217 xprintlog("%s: CALLBACK(ERROR) %s", program, b2nsm_strerror());
218 return 1;
219 }
220
221 /*
222 if ( b2nsm_callback("MSG", msg_handler ) < 0) {
223 xprintlog("%s: CALLBACK(MSG) %s", program, b2nsm_strerror());
224 }
225 */
226 // prompt
227 prompt = static_cast<char*>(malloc(strlen(nodename) + 2));
228 if (prompt == NULL) {
229 fprintf(stderr, "rfctl: out of memory\n");
230 exit(-1);
231 }
232 strcpy(prompt, nodename);
233 strcat(prompt, ">");
234
235
236#ifdef USE_READLINE
237 char* input = 0;
238 char* prev = 0;
239#endif
240
241 // infinite loop
242 while (1) {
243#ifdef USE_READLINE
244 /* On solaris, fgets returns without changing buf when it receives
245 a network message (it doesn't happen on linux). If the buffer
246 is not initialized, it will produce a faulty input.
247 */
248 if (buf) free(buf);
249 if (input) free(input);
250 if (!(input = readline(prompt)))
251 exit(1);
252 buf = (char*)malloc(strlen(input) + 1);
253 strcpy(buf, input);
254
255 if (!prev || (*input && strcmp(prev, input) != 0)) {
256 add_history(input);
257 if (prev) free(prev);
258 prev = (char*)malloc(strlen(input) + 1);
259 strcpy(prev, input);
260 }
261#else
262 static char buffer[1024];
263 buffer[0] = 0;
264 printf("%s", prompt);
265 do {
266 fgets(buffer, sizeof(buffer), stdin);
267 } while (! buffer[0]); /* to cope with EINTR */
268#endif
269
270 int ac;
271 if ((ac = xgetargs(buf, av)) <= 0) //TODO should there be buf or buffer?
272 continue;
273 printf("ac=%d av[0]=%s\n", ac, av[0]);
274
275 if (strcasecmp(av[0], "q") == 0 || strcasecmp(av[0], "quit") == 0) {
276 break;
277 } else if (strcasecmp(av[0], "start") == 0) {
278 if (ac < 4 || ! isdigit(av[2][0])) {
279 printf("usage: start <node> <exp-number> <run-number>\n");
280 } else {
281 int pars[2];
282 pars[0] = atoi(av[2]);
283 pars[1] = atoi(av[3]);
284 b2nsm_sendreq(av[1], "RC_START", 2, pars);
285 }
286 } else if (strcasecmp(av[0], "stop") == 0) {
287 if (ac < 4) {
288 printf("usage: start <node> <exp-number> <run-number>\n");
289 } else {
290 int pars[10];
291 pars[0] = atoi(av[2]);
292 pars[1] = atoi(av[3]);
293 b2nsm_sendreq(av[1], "RC_STOP", 2, pars);
294 }
295 } else if (strcasecmp(av[0], "config") == 0) {
296 if (ac < 2) {
297 printf("usage: config <node>\n");
298 } else {
299 // b2nsm_sendreq(av[1], "RC_LOAD", 0, 0);
300 char scrfile[256];
301 strcpy(scrfile, "run_processor.py");
302 b2nsm_sendany(av[1], "RC_LOAD", 0, 0, sizeof(scrfile) + 1, scrfile, "sendreq");
303 }
304 } else if (strcasecmp(av[0], "unconfig") == 0) {
305 if (ac < 2) {
306 printf("usage: unconfig <node>\n");
307 } else {
308 b2nsm_sendreq(av[1], "RC_ABORT", 0, 0);
309 }
310 } else if (strcasecmp(av[0], "restart") == 0) {
311 if (ac < 2) {
312 printf("usage: restart <node>\n");
313 } else {
314 b2nsm_sendreq(av[1], "RC_RECOVER", 0, 0);
315 }
316 } else {
317 printf("unknown request %s\n", av[0]);
318 }
319 }
320
321 return 0; // never reached
322}
323// ----------------------------------------------------------------------
324// -- (emacs outline mode setup)
325// Local Variables: ***
326// mode:outline-minor ***
327// outline-regexp:"^// --[+ ]" ***
328// End: ***
NSMmsg
Definition nsm2.h:224