1

     2

#include "NetworkUDP.hh"

     3

     4

NetworkUDP::NetworkUDP (void) : 

     5

    socket(udp, ipv4) {

     6

     7

    // do not bind

     8

     9

    // connect signal

    10

    slots.connect(socket.sig_read_triggered(), this, &NetworkUDP::on_recv);

    11

}

    12

    13

NetworkUDP::NetworkUDP (const NetworkAddress &bind_addr) :

    14

    socket(udp, ipv4) {

    15

    16

    // bind socket

    17

    socket.bind(bind_addr);

    18

    19

    // connect signal

    20

    slots.connect(socket.sig_read_triggered(), this, &NetworkUDP::on_recv);

    21

}

    22

    23

void NetworkUDP::on_recv (void) {

    24

    int ret;

    25

    NetworkPacket pkt;

    26

    NetworkAddress src;

    27

    28

    // attempt to recv

    29

    try {

    30

        ret = socket.recv(pkt.get_buf(), pkt.get_buf_size());

    31

    32

    } catch (CL_Error &e) {

    33

        if (errno == EAGAIN)

    34

            return;

    35

        else

    36

            throw;

    37

    }

    38

    39

    // set packet data size

    40

    pkt.set_data_size(ret);

    41

    42

    // handle packet

    43

    sig_packet(pkt, src);

    44

}

    45

    46

bool NetworkUDP::sendto (const NetworkPacket &packet, const NetworkAddress &dst) {

    47

    int ret;

    48

    49

    // XXX: shouldn't get trimmed

    50

    try {

    51

        ret =socket.send(packet.get_buf(), packet.get_data_size(), dst);

    52

    53

    } catch (CL_Error &e) {

    54

        // XXX: catch some errors, but not others?

    55

        return false;

    56

    }

    57

    58

    // UDP shouldn't trim packets

    59

    assert(ret == packet.get_data_size());

    60

    61

    // good

    62

    return true;

    63

}

    64