1
0
Fork 0
mirror of https://git.rwth-aachen.de/acs/public/villas/node/ synced 2025-03-09 00:00:00 +01:00

code cleanup for pipe tool

This commit is contained in:
Steffen Vogel 2016-06-26 15:33:04 +02:00
parent fd525e5701
commit def9d0ec6d

View file

@ -17,69 +17,88 @@
#include <signal.h>
#include <pthread.h>
#include "config.h"
#include "cfg.h"
#include "utils.h"
#include "node.h"
#include "msg.h"
#include "timing.h"
#include "pool.h"
#include <villas/cfg.h>
#include <villas/utils.h>
#include <villas/node.h>
#include <villas/msg.h>
#include <villas/timing.h>
#include <villas/pool.h>
#include <villas/kernel/rt.h>
#include "config.h"
struct list nodes; /**< List of all nodes */
struct settings settings; /**< The global configuration */
struct pool recv_pool, send_pool;
pthread_t recv_thread, send_thread;
struct dir {
struct pool pool;
pthread_t thread;
bool enabled;
bool started;
} sendd, recvv;
bool reverse = false;
struct node *node;
pthread_t ptid; /**< Parent thread id */
static void quit(int signal, siginfo_t *sinfo, void *ctx)
{
pthread_cancel(recv_thread);
pthread_cancel(send_thread);
if (recvv.started) {
pthread_cancel(recvv.thread);
pthread_join(recvv.thread, NULL);
}
pthread_join(recv_thread, NULL);
pthread_join(send_thread, NULL);
if (sendd.started) {
pthread_cancel(sendd.thread);
pthread_join(sendd.thread, NULL);
}
pool_destroy(&recvv.pool);
pool_destroy(&sendd.pool);
node_stop(node);
node_deinit(node->_vt);
pool_destroy(&recv_pool);
pool_destroy(&send_pool);
list_destroy(&nodes, (dtor_cb_t) node_destroy, false);
info(GRN("Goodbye!"));
exit(EXIT_SUCCESS);
}
static void usage(char *name)
{
printf("Usage: %s CONFIG [-r] NODE\n", name);
printf("Usage: %s CONFIG NODE [OPTIONS]\n", name);
printf(" CONFIG path to a configuration file\n");
printf(" NODE the name of the node to which samples are sent and received from\n");
printf(" -d LVL set debug log level to LVL\n");
printf(" -x swap read / write endpoints\n");
printf(" -s only read data from stdin and send it to node\n");
printf(" -r only read data from node and write it to stdout\n\n");
printf(" OPTIONS are:\n");
printf(" -d LVL set debug log level to LVL\n");
printf(" -x swap read / write endpoints\n");
printf(" -s only read data from stdin and send it to node\n");
printf(" -r only read data from node and write it to stdout\n\n");
print_copyright();
exit(EXIT_FAILURE);
}
void * send_loop(void *ctx)
static void * send_loop(void *ctx)
{
int ret;
struct sample *smps[node->vectorize];
if (!sendd.enabled)
return NULL;
sendd.started = true;
/* Initialize memory */
ret = pool_init_mmap(&send_pool, SAMPLE_LEN(DEFAULT_VALUES), node->vectorize);
ret = pool_init_mmap(&sendd.pool, SAMPLE_LEN(DEFAULT_VALUES), node->vectorize);
if (ret < 0)
error("Failed to allocate memory for receive pool.");
ret = pool_get_many(&send_pool, (void **) smps, node->vectorize);
ret = pool_get_many(&sendd.pool, (void **) smps, node->vectorize);
if (ret < 0)
error("Failed to get %u samples out of send pool (%d).", node->vectorize, ret);
@ -91,7 +110,7 @@ void * send_loop(void *ctx)
retry: reason = sample_fscan(stdin, s, NULL);
if (reason < 0) {
if (feof(stdin))
return NULL;
goto killme;
else {
warn("Skipped invalid message message: reason=%d", reason);
goto retry;
@ -102,25 +121,33 @@ retry: reason = sample_fscan(stdin, s, NULL);
node_write(node, smps, node->vectorize);
}
killme: pthread_kill(ptid, SIGINT);
return NULL;
}
void * recv_loop(void *ctx)
static void * recv_loop(void *ctx)
{
int ret;
struct sample *smps[node->vectorize];
if (!recvv.enabled)
return NULL;
recvv.started = true;
/* Initialize memory */
ret = pool_init_mmap(&recv_pool, SAMPLE_LEN(DEFAULT_VALUES), node->vectorize);
ret = pool_init_mmap(&recvv.pool, SAMPLE_LEN(DEFAULT_VALUES), node->vectorize);
if (ret < 0)
error("Failed to allocate memory for receive pool.");
ret = pool_get_many(&recv_pool, (void **) smps, node->vectorize);
ret = pool_get_many(&recvv.pool, (void **) smps, node->vectorize);
if (ret < 0)
error("Failed to get %u samples out of receive pool (%d).", node->vectorize, ret);
/* Print header */
fprintf(stdout, "# %-20s\t\t%s\n", "sec.nsec+offset", "data[]");
fflush(stdout);
for (;;) {
int recv = node_read(node, smps, node->vectorize);
@ -137,25 +164,32 @@ void * recv_loop(void *ctx)
int main(int argc, char *argv[])
{
bool send = true, recv = true, reverse = false;
int ret;
char c;
config_t config;
ptid = pthread_self();
log_init();
/* Parse command line arguments */
if (argc < 3)
usage(argv[0]);
log_init();
char c;
/* Default values */
sendd.enabled = true;
recvv.enabled = true;
while ((c = getopt(argc-2, argv+2, "hxrsd:")) != -1) {
switch (c) {
case 'x':
reverse = true;
break;
case 's':
recv = false;
recvv.enabled = false; // send only
break;
case 'r':
send = false;
sendd.enabled = false; // receive only
break;
case 'd':
log_setlevel(atoi(optarg), -1);
@ -177,13 +211,16 @@ int main(int argc, char *argv[])
sigaction(SIGINT, &sa_quit, NULL);
/* Initialize log, configuration.. */
config_t config;
/* Create lists */
list_init(&nodes);
config_init(&config);
config_parse(argv[1], &config, &settings, &nodes, NULL);
info("Parsing configuration");
{ INDENT
config_init(&config);
config_parse(argv[1], &config, &settings, &nodes, NULL);
}
info("Initialize real-time system");
rt_init(settings.affinity, settings.priority);
/* Initialize node */
node = list_lookup(&nodes, argv[2]);
@ -193,16 +230,18 @@ int main(int argc, char *argv[])
if (reverse)
node_reverse(node);
node_init(node->_vt, argc-optind, argv+optind, config_root_setting(&config));
node_start(node);
ret = node_init(node->_vt, argc-optind, argv+optind, config_root_setting(&config));
if (ret)
error("Failed to intialize node: %s", node_name(node));
ret = node_start(node);
if (ret)
error("Failed to start node: %s", node_name(node));
/* Start threads */
if (recv)
pthread_create(&recv_thread, NULL, recv_loop, NULL);
if (send)
pthread_create(&send_thread, NULL, send_loop, NULL);
pthread_create(&recvv.thread, NULL, recv_loop, NULL);
pthread_create(&sendd.thread, NULL, send_loop, NULL);
for (;;)
pause();