Obsolete
/
libipc-old
Archived
3
0
Fork 0
This repository has been archived on 2024-06-18. You can view files and clone it, but cannot push or open issues/pull-requests.
libipc-old/pubsub/app/pubsubc.c

173 lines
4.0 KiB
C

#include "../lib/pubsubd.h"
#include <stdlib.h>
#include <pthread.h>
void
ohshit(int rvalue, const char* str) {
fprintf (stderr, "\033[31merr: %s\033[00m\n", str);
exit (rvalue);
}
void usage (char **argv)
{
printf ( "usage: %s\n", argv[0]);
}
void print_cmd (void) {
printf ("\033[32m>\033[00m ");
fflush (stdout);
}
void * listener (void *params)
{
int s = 0;
s = pthread_setcancelstate(PTHREAD_CANCEL_ENABLE, NULL);
if (s != 0)
printf ("pthread_setcancelstate: %d\n", s);
struct process *p = NULL;
p = (struct process *) params;
if (p == NULL) {
fprintf (stderr, "listener: no process\n");
return NULL;
}
// main loop
while (1) {
struct pubsub_msg m;
memset (&m, 0, sizeof (struct pubsub_msg));
pubsub_msg_recv (p, &m);
printf ("\n\033[31m>\033[00m %s\n", m.data);
print_cmd ();
// if (m.type == PUBSUB_TYPE_DISCONNECT) { }
pubsubd_msg_free (&m);
}
pthread_exit (NULL);
}
void main_loop (int argc, char **argv, char **env
, pid_t pid, int index, int version
, char *cmd, char *chan)
{
printf ("connection : pid %d index %d version %d "
"cmd %s chan %s\n"
, pid, index, version, cmd, chan );
struct service srv;
memset (&srv, 0, sizeof (struct service));
srv_init (argc, argv, env, &srv, PUBSUB_SERVICE_NAME, NULL);
printf ("Writing on %s.\n", srv.spath);
struct process p;
memset (&p, 0, sizeof (struct process));
printf ("app creation\n");
if (app_create (&p, pid, index, version)) // called by the application
ohshit (1, "app_create");
pthread_t thr;
memset (&thr, 0, sizeof (pthread_t));
pthread_create (&thr, NULL, listener, &p);
pthread_detach (thr);
printf ("main_loop\n");
// send a message to warn the service we want to do something
// line : pid index version action chan
enum app_list_elm_action action = PUBSUB_BOTH;
if (strncmp (cmd, "pub", 3) == 0) {
action = PUBSUB_PUB;
}
else if (strncmp (cmd, "sub", 3) == 0) {
action = PUBSUB_SUB;
}
pubsub_connection (&srv, &p, action, chan);
struct pubsub_msg msg;
memset (&msg, 0, sizeof (struct pubsub_msg));
// meta data on the message
msg.type = PUBSUB_TYPE_MESSAGE;
msg.chan = malloc (strlen (chan) + 1);
memset (msg.chan, 0, strlen (chan) + 1);
strncpy ((char *) msg.chan, chan, strlen (chan));
msg.chan[strlen (chan)] = '\0';
msg.chanlen = strlen (chan);
// msg loop
for (;;) {
char buf[BUFSIZ];
memset (buf, 0, BUFSIZ);
print_cmd ();
fflush (stdout);
size_t mlen = read (0, buf, BUFSIZ);
if (mlen > 1) {
mlen--;
}
buf[mlen] = '\0';
// TODO debug
// printf ("data (%ld): %s\n", mlen, buf);
if (strncmp(buf, "quit", strlen ("quit")) == 0) {
break;
}
msg.data = malloc (strlen (buf) + 1);
memset (msg.data, 0, strlen (buf) + 1);
strncpy ((char *) msg.data, buf, strlen (buf) + 1);
msg.datalen = strlen (buf);
// TODO debug
// printf ("send message\n");
pubsub_msg_send (&p, &msg);
free (msg.data);
msg.data = NULL;
msg.datalen = 0;
}
// free everything
pubsubd_msg_free (&msg);
pthread_cancel (thr);
pthread_join (thr, NULL);
printf ("disconnection...\n");
// disconnect from the server
pubsub_disconnect (&p);
printf ("destroying app\n");
// the application will shut down, and remove the application named pipes
if (app_destroy (&p))
ohshit (1, "app_destroy");
}
int main(int argc, char **argv, char **env)
{
if (argc != 1) {
usage (argv);
exit (1);
}
pid_t pid = getpid();
int index = 0;
// don't care about the version
int version = COMMUNICATION_VERSION;
char *cmd = "both";
char *chan = "chan1";
main_loop (argc, argv, env, pid, index, version, cmd, chan);
return EXIT_SUCCESS;
}