libipc-old/pubsub/pubsubd.c

#include "../lib/pubsubd.h"
#include <stdlib.h>
#include <pthread.h>

void
ohshit(int rvalue, const char* str) {
    fprintf(stderr, "%s\n", str);
    exit(rvalue);
}

// give this structure to the thread worker function
struct worker_params {
    struct channels *chans;
    struct channel *chan;
    struct app_list_elm *ale;
};

void * pubsubd_worker_thread (void *params)
{
    struct worker_params *wp = (struct worker_params *) params;

    // each chan has a list of subscribers
    // someone who only push a msg doesn't need to be registered
    if (wp->ale->action == PUBSUB_BOTH || wp->ale->action == PUBSUB_PUB) {
        // TODO add it to the application to follow
        // TODO publish a message
        printf ("publish or publish and subscribe to something\n");

        struct pubsub_msg m;
        bzero (&m, sizeof (struct pubsub_msg));
        pubsubd_msg_recv (wp->ale->p, &m);

        pubsubd_msg_print (&m);

        if (m.type == PUBSUB_TYPE_DISCONNECT) {
            // TODO remove the application from the subscribers
        }
        else {
            struct channel *chan = pubsubd_channel_get (wp->chans, wp->chan);
            pubsubd_msg_send (chan->alh, &m);
        }
    }
    else if (wp->ale->action == PUBSUB_SUB) {
        // TODO
        printf ("subscribe to something\n");
    }
    else {
        printf ("\033[31mdo not know what you want to do\033[00m\n");
        printf ("\tale->p : %p\n", wp->ale->p);
    }

    pubsubd_app_list_elm_free (wp->ale);

    pthread_exit (NULL);
}

    int
main(int argc, char* argv[])
{
    struct service srv;
    bzero (&srv, sizeof (struct service));
    srv_init (&srv, PUBSUB_SERVICE_NAME);
    printf ("Listening on %s.\n", srv.spath);

    // creates the service named pipe, that listens to client applications
    if (srv_create (&srv))
        ohshit(1, "service_create error");

    // init chans list
    struct channels chans;
    bzero (&chans, sizeof (struct channels));
    pubsubd_channels_init (&chans);

    for (;;) {
        // for each new process
        struct app_list_elm ale;
        bzero (&ale, sizeof (struct app_list_elm));
        struct channel *chan;
        pubsubd_get_new_process (&srv, &ale, &chans, &chan);
        pubsubd_channels_print (&chans);

        // end the application
        if (ale.action == PUBSUB_QUIT) {
            printf ("Quitting ...\n");

            pubsubd_channels_del_all (&chans);
            srv_close (&srv);
            
            // TODO end the threads

            exit (0);
        }

        // TODO thread to handle multiple clients at a time
        struct worker_params *wp;
        wp = malloc (sizeof (struct worker_params));
        wp->ale = pubsubd_app_list_elm_copy (&ale);
        wp->chans = &chans;
        wp->chan = chan;

        pthread_t thr;

        pthread_create (&thr, NULL, pubsubd_worker_thread, wp);
        pthread_detach (thr);

        pubsubd_app_list_elm_free (&ale);
    }

    // the application will shut down, and remove the service named pipe
    if (srv_close (&srv))
        ohshit (1, "service_close error");

    return EXIT_SUCCESS;
}

#if 0
void main_loop (const struct service *srv)
{
    int ret;
    struct process proc;

    int cnt = 10;

    while (cnt--) {
        // -1 : error, 0 = no new process, 1 = new process
        ret = srv_get_new_process (&proc, srv);
        if (ret == -1) {
            fprintf (stderr, "error service_get_new_process\n");
            continue;
        } else if (ret == 0) { // that should not happen
            continue;
        }

        // printf ("before print\n");
        process_print (&proc);
        // printf ("after print\n");

        // about the message
        size_t msize = BUFSIZ;
        char buf[BUFSIZ];
        bzero(buf, BUFSIZ);

        // printf ("before read\n");
        if ((ret = srv_read (&proc, &buf, &msize))) {
            fprintf(stdout, "error service_read %d\n", ret);
            continue;
        }
        // printf ("after read\n");
        printf ("read, size %ld : %s\n", msize, buf);

        // printf ("before proc write\n");
        if ((ret = srv_write (&proc, &buf, msize))) {
            fprintf(stdout, "error service_write %d\n", ret);
            continue;
        }

        // printf ("after proc write\n");
        printf ("\033[32mStill \033[31m%d\033[32m applications to serve\n",cnt);
    }
}
#endif