recv.cpp

Go to the documentation of this file.

#include "udp_bc_broker/recv.h"

using namespace std;

namespace udp_bc_broker{

    UdpRecver::UdpRecver(int port)
    {
        setvbuf(stdout, NULL, _IONBF, 0);
        fflush(stdout);

        //bind address
        bzero(&addrto, sizeof(struct sockaddr_in));
        addrto.sin_family = AF_INET;
        addrto.sin_addr.s_addr = htonl(INADDR_ANY);
        addrto.sin_port = htons(port);

        //broadcast address
        bzero(&from, sizeof(struct sockaddr_in));
        from.sin_family = AF_INET;
        from.sin_addr.s_addr = htonl(INADDR_ANY);
        from.sin_port = htons(port);

        if ((sock = socket(AF_INET, SOCK_DGRAM, 0)) == -1) {
            cout<<"socket error!"<<endl;
            exit(-1);
        }

        const int opt = 1;
        int nb = 0;
        nb = setsockopt(sock, SOL_SOCKET, SO_BROADCAST, (char *)&opt, sizeof(opt));
        if(nb == -1) {
            cout<<"set socket error!"<<endl;
            exit(-1);
        }

        if(bind(sock,(struct sockaddr *)&(addrto), sizeof(struct sockaddr_in)) == -1) {
            cout<<"bind error!"<<endl;
            exit(-1);
        }

        len = sizeof(sockaddr_in);
        buf = new char[10000];
    }

    UdpRecver::~UdpRecver()
    {
        process_thread_->interrupt();
        process_thread_->join();
        delete process_thread_;

        delete[] buf;
    }

    void UdpRecver::process_msg()
    {
        while(true) {
            int ret = recvfrom(sock, buf, 10000, 0, (struct sockaddr*)&from, (socklen_t*)&len);
            if(ret <= 0) {
                cout<<"read error...."<<sock<<endl;
            }
            else {
                std::vector<uint8_t> msg_data;
                for(int i = 0; i < ret; i++) {
                    msg_data.push_back(buf[i]);
                }
                callback(msg_data);
            }
            sleep(0.1);
        }
    }

    void UdpRecver::receive(boost::function<void(const vector<uint8_t>&)> callBack)
    {
        callback = callBack;
        process_thread_ = new boost::thread(&UdpRecver::process_msg, this);
    }
};