File content as of revision 139:f8ab852e83e7:

#include <stdio.h>
#include <ctype.h>
#include "PollThread.h"
#include "logging.h"


bool PollThread::handshake()
{
        int l = snprintf(buf2, sizeof(buf2), "%s", "80\r\n");
        l = snprintf(buf, sizeof(buf), fmtSmartRest, uri, l, buf2);
        sock.setBlocking(3000);
        l = sock.sendAndReceive(buf, l, sizeof(buf));
        if (l <= 0) return false;
        const char* p = skipHTTPHeader(buf);
        if (p == NULL) return false;
        size_t i = 0;
        for (; isalnum(*p); ++p, ++i) {
                bayeuxId[i] = *p;
        }
        bayeuxId[i] = 0;
        return bayeuxId[0];
}


bool PollThread::subscribe()
{
        int l = snprintf(buf2, sizeof(buf2), "81,%s,%s\r\n", bayeuxId, chn);
        l = snprintf(buf, sizeof(buf), fmtSmartRest, uri, l, buf2);
        sock.setBlocking(3000);
        l = sock.sendAndReceive(buf, l, sizeof(buf));
        if (l <= 0) return false;
        const char *p = skipHTTPHeader(buf);
        if (p == NULL) return false;
        for (; *p && !isgraph(*p); ++p);
        return *p == 0;
}


bool PollThread::connect()
{
        int l = snprintf(buf2, sizeof(buf2), "83,%s\r\n", bayeuxId);
        l = snprintf(buf, sizeof(buf), fmtSmartRest, uri, l, buf2);
        sock.setBlocking(610000); // Timeout after 10m:10s
//        sock.setBlocking(-1);
//        l = sock.sendAndReceive(buf, l, sizeof(buf));
        int c = sock.connect();
        if (c < 0) {
                sock.close();
                return false;
        }
        c = sock.send(buf, l);
        if (c < 0) {
                sock.close();
                return false;
        }
        // Avoid heartbeat message overflows the buffer
        l = 0;
        for (unsigned short i = 0; i < 300; ++i) {
                int l2 = sock.receive(buf+l, sizeof(buf)-l);
                if (l2 > 0) {
                        l += l2;
                } else {
                        break;
                }
        }
        sock.close();
        buf[l] = 0;
        return l>0;
}


void PollThread::threadFunc()
{
        unsigned short state = 1;
        while (true) {
                switch (state) {
                case 1: if (!handshake()) {
                                aCritical("Poll: handshake fail!\n");
                                break;
                        }
                case 2: if(!subscribe()) {
                                aCritical("Poll: subscribe fail!\n");
                                state = 1;
                                break;
                        }
                case 3: if(!connect()) {
                                aCritical("Poll: connect fail!\n");
                                break;
                        }
                        parser.parse(buf);
                        if (parser.getBayeuxAdvice() == BA_HANDSHAKE)
                                state = 1;
                        else
                                state = 3;
                }
        }
}