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/pubsubd.c

255 lines
6.6 KiB
C
Raw Normal View History

2016-06-05 20:48:13 +02:00
#include "../lib/pubsubd.h"
2016-05-28 19:34:23 +02:00
#include <stdlib.h>
2016-06-08 17:57:05 +02:00
#include <pthread.h>
2016-05-28 19:34:23 +02:00
2016-09-11 14:37:41 +02:00
#define NB_CLIENTS 3
2016-05-28 19:34:23 +02:00
void
ohshit(int rvalue, const char* str) {
2016-06-04 20:33:44 +02:00
fprintf(stderr, "%s\n", str);
exit(rvalue);
2016-05-28 19:34:23 +02:00
}
// head of the list
LIST_HEAD(workers, worker);
struct workers *my_workers;
// element of the list
// worker : process to handle (threaded)
struct worker {
2016-06-08 17:57:05 +02:00
struct channels *chans;
struct channel *chan;
struct app_list_elm *ale;
LIST_ENTRY(worker) entries;
2016-06-08 17:57:05 +02:00
};
void pubsubd_worker_free (struct worker * w);
struct worker * pubsubd_worker_get (struct workers *wrkrs, struct worker *w);
int pubsubd_worker_eq (const struct worker *w1, const struct worker *w2);
void pubsubd_workers_init (struct workers *wrkrs) { LIST_INIT(wrkrs); }
struct worker *
pubsubd_workers_add (struct workers *wrkrs, const struct worker *w)
{
if (wrkrs == NULL || w == NULL) {
printf ("pubsubd_workers_add: wrkrs == NULL or w == NULL");
return NULL;
}
struct worker *n = malloc (sizeof (struct worker));
memset (n, 0, sizeof (struct worker));
memcpy (n, w, sizeof (struct worker));
if (w->ale != NULL)
n->ale = pubsubd_app_list_elm_copy (w->ale);
LIST_INSERT_HEAD(wrkrs, n, entries);
return n;
}
void pubsubd_worker_del (struct workers *wrkrs, struct worker *w)
{
struct worker *todel = pubsubd_worker_get (wrkrs, w);
if(todel != NULL) {
LIST_REMOVE(todel, entries);
pubsubd_worker_free (todel);
free (todel);
todel = NULL;
}
}
void pubsubd_workers_del_all (struct workers *wrkrs)
{
if (!wrkrs)
return;
struct worker *w = NULL;
while (!LIST_EMPTY(wrkrs)) {
printf ("KILL THE WORKERS : %p\n", w);
w = LIST_FIRST(wrkrs);
LIST_REMOVE(w, entries);
pubsubd_worker_free (w);
free (w);
w = NULL;
}
}
void pubsubd_worker_free (struct worker * w)
{
if (w == NULL)
return;
pubsubd_app_list_elm_free (w->ale);
free (w->ale);
w->ale = NULL;
}
struct worker * pubsubd_worker_get (struct workers *wrkrs, struct worker *w)
{
struct worker * np = NULL;
LIST_FOREACH(np, wrkrs, entries) {
if (pubsubd_worker_eq (np, w))
return np;
}
return NULL;
}
int pubsubd_worker_eq (const struct worker *w1, const struct worker *w2)
{
return w1 == w2; // if it's the same pointer
}
// a thread for each connected process
2016-06-08 17:57:05 +02:00
void * pubsubd_worker_thread (void *params)
{
2016-09-11 14:37:41 +02:00
int s = 0;
s = pthread_setcancelstate(PTHREAD_CANCEL_ENABLE, NULL);
if (s != 0)
printf ("pthread_setcancelstate: %d\n", s);
struct worker *w = (struct worker *) params;
if (w == NULL) {
2016-09-08 13:23:07 +02:00
fprintf (stderr, "error pubsubd_worker_thread : params NULL\n");
return NULL;
}
2016-06-08 17:57:05 +02:00
struct channels *chans = w->chans;
struct channel *chan = w->chan;
struct app_list_elm *ale = w->ale;
2016-09-11 14:37:41 +02:00
// main loop
2016-09-09 01:36:44 +02:00
while (1) {
struct pubsub_msg m;
memset (&m, 0, sizeof (struct pubsub_msg));
2016-09-11 14:37:41 +02:00
pubsubd_msg_recv (ale->p, &m);
2016-09-09 01:36:44 +02:00
if (m.type == PUBSUB_TYPE_DISCONNECT) {
2016-09-11 14:37:41 +02:00
printf ("process %d disconnecting...\n", ale->p->pid);
if ( 0 != pubsubd_subscriber_del (chan->alh, ale)) {
2016-09-09 01:36:44 +02:00
fprintf (stderr, "err : subscriber not registered\n");
}
break;
}
else {
2016-09-11 14:37:41 +02:00
struct channel *ch = pubsubd_channel_search (chans, chan->chan);
if (ch == NULL) {
printf ("CHAN NOT FOUND\n");
2016-09-11 14:37:41 +02:00
}
else {
printf ("what should be sent: ");
2016-09-11 14:37:41 +02:00
pubsubd_msg_print (&m);
printf ("send the message to:\t");
2016-09-11 14:37:41 +02:00
pubsubd_channel_print (ch);
pubsubd_msg_send (ch->alh, &m);
}
2016-09-09 01:36:44 +02:00
}
pubsubd_msg_free (&m);
2016-09-09 01:36:44 +02:00
}
2016-06-08 17:57:05 +02:00
2016-09-11 14:37:41 +02:00
pubsubd_app_list_elm_free (ale);
free (w->ale);
w->ale = NULL;
pubsubd_worker_del (my_workers, w);
2016-06-08 17:57:05 +02:00
pthread_exit (NULL);
}
2016-09-08 13:23:07 +02:00
int
2016-06-12 14:41:25 +02:00
main(int argc, char **argv, char **env)
2016-06-04 20:33:44 +02:00
{
2016-06-05 20:48:13 +02:00
struct service srv;
memset (&srv, 0, sizeof (struct service));
2016-06-13 09:47:19 +02:00
srv_init (argc, argv, env, &srv, PUBSUB_SERVICE_NAME, NULL);
printf ("Listening on %s.\n", srv.spath);
2016-05-28 19:34:23 +02:00
2016-06-04 20:33:44 +02:00
// creates the service named pipe, that listens to client applications
if (srv_create (&srv))
2016-06-04 20:33:44 +02:00
ohshit(1, "service_create error");
2016-05-28 19:34:23 +02:00
2016-06-07 11:45:21 +02:00
// init chans list
2016-06-04 20:33:44 +02:00
struct channels chans;
memset (&chans, 0, sizeof (struct channels));
2016-06-04 20:33:44 +02:00
pubsubd_channels_init (&chans);
2016-05-28 19:34:23 +02:00
2016-09-11 14:37:41 +02:00
pthread_t *thr = NULL;
thr = malloc (sizeof (pthread_t));
memset (thr, 0, sizeof (pthread_t));
2016-09-11 14:37:41 +02:00
my_workers = malloc (sizeof (struct workers));
memset (my_workers, 0, sizeof (struct workers));
pubsubd_workers_init (my_workers);
int i = 0;
// for (i = 0; i < NB_CLIENTS; i++)
for (i = 0; ; i++) {
2016-06-07 11:45:21 +02:00
// for each new process
struct app_list_elm ale;
memset (&ale, 0, sizeof (struct app_list_elm));
struct channel *chan = NULL;
2016-09-09 21:52:56 +02:00
pubsubd_get_new_process (srv.spath, &ale, &chans, &chan);
2016-06-07 17:44:18 +02:00
pubsubd_channels_print (&chans);
// end the application
if (ale.action == PUBSUB_QUIT) {
pubsubd_app_list_elm_free (&ale);
break;
2016-06-07 13:49:23 +02:00
}
// thread to handle multiple clients at a time
struct worker *w = NULL;
w = malloc (sizeof (struct worker));
w->ale = pubsubd_app_list_elm_copy (&ale);
w->chans = &chans;
w->chan = chan;
struct worker *wtmp = pubsubd_workers_add (my_workers, w);
pubsubd_worker_free (w);
free (w);
w = wtmp;
2016-06-07 13:49:23 +02:00
// realloc memory for further workers
pthread_t * tmpthr = realloc (thr, sizeof (pthread_t) * (i+1));
if (tmpthr == NULL) {
fprintf (stderr, "err: can't allocate more thread contexts\n");
pubsubd_app_list_elm_free (&ale);
break;
}
else {
thr = tmpthr;
}
memset (thr + i, 0, sizeof (pthread_t));
pthread_create (thr + i, NULL, pubsubd_worker_thread, w);
2016-09-11 14:37:41 +02:00
pthread_detach (thr[i]);
2016-06-07 11:45:21 +02:00
2016-06-07 17:44:18 +02:00
pubsubd_app_list_elm_free (&ale);
2016-06-04 20:33:44 +02:00
}
2016-05-28 19:34:23 +02:00
printf ("Quitting ...\n");
2016-09-11 14:37:41 +02:00
// stop threads
for (int j = 0 ; j < i ; j++) {
pthread_cancel (thr[j]);
2016-09-11 14:37:41 +02:00
void *ret = NULL;
pthread_join (thr[j], &ret);
2016-09-11 14:37:41 +02:00
if (ret != NULL) {
free (ret);
}
}
free (thr);
pubsubd_channels_del_all (&chans);
pubsubd_workers_del_all (my_workers);
free (my_workers);
2016-09-11 14:37:41 +02:00
2016-06-04 20:33:44 +02:00
// the application will shut down, and remove the service named pipe
if (srv_close (&srv))
2016-06-07 13:49:23 +02:00
ohshit (1, "service_close error");
2016-05-28 19:34:23 +02:00
2016-06-04 20:33:44 +02:00
return EXIT_SUCCESS;
2016-05-28 19:34:23 +02:00
}