1

     2

#include "NetworkUDP.hh"

     3

     4

#include <ClanLib/core.h>

     5

#include <cassert>

     6

     7

NetworkUDP::NetworkUDP (void) : 

     8

    socket(CL_Socket::udp, CL_Socket::ipv4) {

     9

    10

    // do not bind

    11

    12

    // connect signal

    13

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

    14

}

    15

    16

NetworkUDP::NetworkUDP (const NetworkAddress &bind_addr) :

    17

    socket(CL_Socket::udp, CL_Socket::ipv4) {

    18

    19

    // bind socket

    20

    socket.bind(bind_addr);

    21

    22

    // connect signal

    23

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

    24

}

    25

    26

void NetworkUDP::on_recv (void) {

    27

    int ret;

    28

    NetworkPacket pkt;

    29

    NetworkAddress src;

    30

    31

    // attempt to recv

    32

    try {

    33

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

    34

    35

    } catch (CL_Error &e) {

    36

        if (errno == EAGAIN)

    37

            return;

    38

        else

    39

            throw;

    40

    }

    41

    42

    // set packet data size

    43

    pkt.set_data_size(ret);

    44

    45

    // handle packet

    46

    _sig_packet(pkt, src);

    47

}

    48

    49

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

    50

    int ret;

    51

    52

    // XXX: shouldn't get trimmed

    53

    try {

    54

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

    55

    56

    } catch (CL_Error &e) {

    57

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

    58

        return false;

    59

    }

    60

    61

    assert(ret > 0);

    62

    63

    // UDP shouldn't trim packets

    64

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

    65

    66

    // good

    67

    return true;

    68

}

    69