1

     2

#include "UDP.hh"

     3

#include "../Engine.hh"

     4

     5

#include <ClanLib/core.h>

     6

#include <cassert>

     7

     8

NetworkUDP::NetworkUDP (void) : 

     9

    socket(AF_UNSPEC, SOCK_DGRAM) 

    10

{

    11

    // do not bind

    12

    13

    // connect signal

    14

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

    15

    16

    // nonblocking

    17

    socket.set_nonblocking(true);

    21

}

    22

    23

NetworkUDP::NetworkUDP (const NetworkAddress &bind_addr) :

    24

    socket(AF_UNSPEC, SOCK_DGRAM) {

    25

    26

    // bind socket

    27

    socket.bind(bind_addr);

    28

    29

    // connect signal

    30

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

    31

    32

    // nonblocking

    33

    socket.set_nonblocking(true);

    37

}

    38

    39

void NetworkUDP::on_recv (void) {

    40

    size_t ret;

    41

    NetworkPacket pkt;

    42

    NetworkAddress src;

    43

    44

    // receieve as many packets as possible

    45

    do {    

    46

        // attempt to recv a packet

    47

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

    48

    49

        // no more packets?

    50

        if (ret == 0)

    51

            return;

    52

    53

        // set packet data size

    54

        pkt.set_data_size(ret);

    55

    56

        // handle packet

    57

        _sig_packet(pkt, src);

    58

    59

    } while (true);

    60

}

    61

    62

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

    63

    size_t ret;

    64

    65

    try {

    66

        // try and send

    67

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

    68

    69

    } catch (NetworkSocketError &e) {

    70

        // catch and log errors

    71

        Engine::log(WARN, "udp.sendto") << "socket->send raised error: " << e.what();

    72

        return false;

    73

    }

    74

    75

    // weird packet size?

    76

    if (ret != packet.get_data_size()) {

    77

        Engine::log(ERROR, "udp.sendto") << "socket->send returned weird length: " << ret << ", packet was " << packet.get_data_size();

    78

        return false;

    79

    80

    } else {

    81

        // sent

    82

        return true;

    83

    }

    84

}

    85