Return to daemon.c CVS log

Up to [ELWIX - Embedded LightWeight unIX -] / mqtt / src

Annotation of mqtt/src/daemon.c, revision 1.1.2.8

1.1.2.1   misho       1: #include "global.h"
                      2: 
                      3: 
1.1.2.7   misho       4: extern char cliCmd[], *cliStr[];
                      5: 
                      6: 
1.1.2.5   misho       7: static void *
                      8: startSession(sched_task_t *task)
1.1.2.4   misho       9: {
1.1.2.8 ! misho      10:        static u_char basebuf[USHRT_MAX];
1.1.2.6   misho      11:        mqtt_msg_t *buf;
                     12:        mqtt_cb_t cbs[MQTT_TYPE_MAX + 1] = { 0 };
1.1.2.7   misho      13:        struct mqtthdr *hdr;
1.1.2.8 ! misho      14:        int len, ret = 0;
1.1.2.6   misho      15: 
                     16:        FTRACE(4);
                     17: 
1.1.2.8 ! misho      18:        /*
1.1.2.6   misho      19:        buf = mqtt_msgAlloc(USHRT_MAX);
                     20:        if (!buf) {
                     21:                syslog(LOG_ERR, "Error:: allocate message buf (%d) #%d - %s", (int) TASK_FD(task), 
                     22:                                mqtt_GetErrno(), mqtt_GetError());
                     23:                goto end;
                     24:        }
1.1.2.8 ! misho      25:        */
1.1.2.6   misho      26: 
1.1.2.8 ! misho      27:        if (recv(TASK_FD(task), basebuf, sizeof basebuf, 0) == -1) {
        !            28:                VERB(3) syslog(LOG_ERR, "Error:: recv(%d) #%d - %s", (int) TASK_FD(task), 
1.1.2.6   misho      29:                                errno, strerror(errno));
                     30:                goto end;
1.1.2.8 ! misho      31:        } else
        !            32:                hdr = (struct mqtthdr*) basebuf;
        !            33:        if (hdr->mqtt_msg.type != MQTT_TYPE_CONNECT) {
        !            34:                VERB(2) syslog(LOG_ERR, "Error:: wrong command type for 1st command #%d", 
        !            35:                                hdr->mqtt_msg.type);
        !            36:                goto end;
        !            37:        } else
        !            38:                len = mqtt_decodeLen(hdr->mqtt_len, &ret);
        !            39: 
        !            40:                /*
1.1.2.7   misho      41:                for (ret = i = 0; cliCmd[i] && !(ret = (hdr->mqtt_msg.type == cliCmd[i])); i++);
                     42:                if (!ret) {
1.1.2.8 ! misho      43:                        VERB(2) syslog(LOG_ERR, "Error:: wrong command type #%d %s", 
1.1.2.7   misho      44:                                        hdr->mqtt_msg.type, cliStr[i]);
                     45:                        goto end;
                     46:                }
1.1.2.8 ! misho      47:                */
1.1.2.6   misho      48: 
1.1.2.8 ! misho      49:        /* check online table for user */
        !            50: //     ChkSessPUB(&cfg, );
1.1.2.6   misho      51: 
1.1.2.8 ! misho      52:        ret = mqttDispatcher(cbs, buf);
1.1.2.6   misho      53:        mqtt_msgFree(&buf, 42);
                     54: 
                     55:        switch (ret) {
                     56:                case -1:
                     57:                        syslog(LOG_ERR, "Error:: in MQTT protocol #%d - %s", mqtt_GetErrno(), mqtt_GetError());
                     58:                        break;
                     59:                default:
                     60:                        return NULL;
                     61:        }
                     62: end:   /* close client connection */
1.1.2.5   misho      63:        close(TASK_FD(task));
                     64:        VERB(1) syslog(LOG_DEBUG, "Close client %s with socket=%d", 
                     65:                        (char*) TASK_ARG(task), (int) TASK_FD(task));
                     66:        if (TASK_ARG(task))
                     67:                free(TASK_ARG(task));
1.1.2.6   misho      68:        return NULL;
1.1.2.4   misho      69: }
                     70: 
                     71: /* ----------------------------------------------------------------------- */
                     72: 
1.1.2.5   misho      73: static void *
                     74: thrSched(void *arg __unused)
                     75: {
                     76:        FTRACE(1);
                     77: 
                     78:        schedRun(root, (intptr_t*) &Kill);
                     79:        pthread_exit(NULL);
                     80: }
                     81: 
1.1.2.2   misho      82: int
1.1.2.3   misho      83: Run(int sock)
1.1.2.2   misho      84: {
1.1.2.4   misho      85:        io_sockaddr_t sa;
                     86:        socklen_t sslen = sizeof sa.ss;
                     87:        int cli;
1.1.2.5   misho      88:        char *str = NULL, szAddr[STRSIZ] = { 0 };
                     89:        pthread_t tid;
1.1.2.4   misho      90: 
1.1.2.2   misho      91:        FTRACE(1);
                     92: 
1.1.2.5   misho      93:        if (pthread_create(&tid, NULL, thrSched, NULL)) {
                     94:                syslog(LOG_ERR, "Error:: thread scheduler #%d - %s", errno, strerror(errno));
                     95:                return -1;
                     96:        } else
                     97:                pthread_detach(tid);
                     98:        VERB(2) syslog(LOG_DEBUG, "Run scheduler management thread");
                     99: 
1.1.2.3   misho     100:        if (listen(sock, SOMAXCONN) == -1) {
1.1.2.4   misho     101:                syslog(LOG_ERR, "Error:: listen(%d) #%d - %s\n", sock, errno, strerror(errno));
1.1.2.5   misho     102:                pthread_cancel(tid);
1.1.2.3   misho     103:                return -1;
                    104:        }
                    105: 
1.1.2.4   misho     106:        while (!Kill) {
                    107:                if ((cli = accept(sock, &sa.sa, &sslen)) == -1) {
                    108:                        syslog(LOG_ERR, "Error:: accept() #%d - %s", errno, strerror(errno));
                    109:                        continue;
                    110:                } else
                    111:                        VERB(1) {
                    112:                                switch (sa.sa.sa_family) {
                    113:                                        case AF_INET:
                    114:                                                inet_ntop(AF_INET, &sa.sin.sin_addr, szAddr, sslen);
                    115:                                                snprintf(szAddr, sizeof szAddr, "%s:%d", 
                    116:                                                                szAddr, ntohs(sa.sin.sin_port));
                    117:                                                break;
                    118:                                        case AF_INET6:
                    119:                                                inet_ntop(AF_INET6, &sa.sin6.sin6_addr, szAddr, sslen);
                    120:                                                snprintf(szAddr, sizeof szAddr, "%s:%d", 
                    121:                                                                szAddr, ntohs(sa.sin6.sin6_port));
                    122:                                                break;
                    123:                                        case AF_LOCAL:
                    124:                                                strlcpy(szAddr, sa.sun.sun_path, sizeof szAddr);
                    125:                                                break;
                    126:                                        default:
                    127:                                                close(cli);
                    128:                                                syslog(LOG_ERR, "Error:: unsupported address type %d", 
                    129:                                                                sa.sa.sa_family);
                    130:                                                continue;
                    131:                                }
1.1.2.5   misho     132:                                str = strdup(szAddr);
                    133:                                syslog(LOG_DEBUG, "Connected client with socket=%d from %s", cli, str);
1.1.2.4   misho     134:                        }
                    135: 
1.1.2.5   misho     136:                if (!schedRead(root, startSession, str, cli)) {
1.1.2.4   misho     137:                        close(cli);
                    138:                        VERB(1) syslog(LOG_DEBUG, "Terminated client with socket=%d", cli);
1.1.2.5   misho     139:                        if (str)
                    140:                                free(str);
1.1.2.4   misho     141:                }
                    142:        }
                    143: 
1.1.2.2   misho     144:        return 0;
                    145: }