Method Poll

LibLemon/src/ipc/interfaceuds.cpp:55–126

    }

    std::shared_ptr<LemonMessageInfo> MessageServer::Poll(){
    retry:
        int fd = 0;
        while((fd = accept(sock.fd, nullptr, nullptr)) > 0){
            fds.push_back({ .fd = fd, .events = POLLIN, .revents = 0 });
        }

        if(queue.size() > 0){
            auto element = queue.front();
            queue.pop_front();
            return element;
        }

        if(!fds.size()) {
            return std::shared_ptr<LemonMessageInfo>(nullptr);
        }

        int evCount = poll(fds.data(), fds.size(), 0);
        if(evCount > 0){
            for(size_t i = 0; i < fds.size(); i++){
                if(fds[i].revents & (POLLNVAL | POLLHUP)){
                    int fd = fds[i].fd;
                    fds.erase(fds.begin() + i);
                    
                    std::shared_ptr<LemonMessageInfo> newMsg = std::shared_ptr<LemonMessageInfo>((LemonMessageInfo*)malloc(sizeof(LemonMessageInfo)));
                    newMsg->msg.protocol = 0; // Disconnected
                    newMsg->clientFd = fd;

                    return newMsg;
                }

                if(!(fds[i].revents & POLLIN)) continue; // We only care about POLLIN
                LemonMessage msg;

                ssize_t len = recv(fds[i].fd, &msg, sizeof(LemonMessage), 0);
                if(len < static_cast<ssize_t>(sizeof(LemonMessage))){
                    printf("invalid length: %ld\n", len);
                    continue;
                }

                if(msg.magic != LEMON_MESSAGE_MAGIC){
                    printf("Invalid magic: %x, discarding data.\n", msg.magic);
                    while(msg.magic != LEMON_MESSAGE_MAGIC && len >= static_cast<ssize_t>(sizeof(LemonMessage))){
                        len = recv(fds[i].fd, &msg, sizeof(msg), MSG_DONTWAIT); // Discard everything until we find a message
                    }
                }

                if(msg.magic != LEMON_MESSAGE_MAGIC){
                    printf("invalid magic: %x\n", msg.magic);
                    continue;
                }
                
                std::shared_ptr<LemonMessageInfo> newMsg((LemonMessageInfo*)malloc(sizeof(LemonMessageInfo) + msg.length));
                newMsg->msg = msg;
                newMsg->clientFd = fds[i].fd;
                len = recv(fds[i].fd, newMsg->msg.data, msg.length, 0);

                if(len < msg.length){