Luis Linares / SimpleSocket

A Simple TCP & UDP Socket Library. An update from that of Hiroshi Yamaguchi.

DatagramSocket.cpp@0:1e4a74c2c16f, 2012-05-23 (annotated)

Committer:
macgyveremir
Date:
Wed May 23 16:26:44 2012 +0000
Revision:
0:1e4a74c2c16f
Fixed dropping of null bytes (0x00) from incoming binary TCP frames

Who changed what in which revision?

UserRevisionLine numberNew contents of line
macgyveremir 0:1e4a74c2c16f 1 /*
macgyveremir 0:1e4a74c2c16f 2 Copyright (c) 2011, Senio Networks, Inc.
macgyveremir 0:1e4a74c2c16f 3
macgyveremir 0:1e4a74c2c16f 4 Permission is hereby granted, free of charge, to any person obtaining a copy
macgyveremir 0:1e4a74c2c16f 5 of this software and associated documentation files (the "Software"), to deal
macgyveremir 0:1e4a74c2c16f 6 in the Software without restriction, including without limitation the rights
macgyveremir 0:1e4a74c2c16f 7 to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
macgyveremir 0:1e4a74c2c16f 8 copies of the Software, and to permit persons to whom the Software is
macgyveremir 0:1e4a74c2c16f 9 furnished to do so, subject to the following conditions:
macgyveremir 0:1e4a74c2c16f 10
macgyveremir 0:1e4a74c2c16f 11 The above copyright notice and this permission notice shall be included in
macgyveremir 0:1e4a74c2c16f 12 all copies or substantial portions of the Software.
macgyveremir 0:1e4a74c2c16f 13
macgyveremir 0:1e4a74c2c16f 14 THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
macgyveremir 0:1e4a74c2c16f 15 IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
macgyveremir 0:1e4a74c2c16f 16 FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
macgyveremir 0:1e4a74c2c16f 17 AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
macgyveremir 0:1e4a74c2c16f 18 LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
macgyveremir 0:1e4a74c2c16f 19 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
macgyveremir 0:1e4a74c2c16f 20 THE SOFTWARE.
macgyveremir 0:1e4a74c2c16f 21 */
macgyveremir 0:1e4a74c2c16f 22
macgyveremir 0:1e4a74c2c16f 23 #include "EthernetNetIf.h"
macgyveremir 0:1e4a74c2c16f 24 #include "SimpleSocket.h"
macgyveremir 0:1e4a74c2c16f 25 #include <stdarg.h>
macgyveremir 0:1e4a74c2c16f 26
macgyveremir 0:1e4a74c2c16f 27 DatagramSocket::DatagramSocket(int port, bool debug) : bufsize(512), debug(debug) {
macgyveremir 0:1e4a74c2c16f 28 host = Host(IpAddr(), port);
macgyveremir 0:1e4a74c2c16f 29 createDatagramSocket();
macgyveremir 0:1e4a74c2c16f 30 }
macgyveremir 0:1e4a74c2c16f 31
macgyveremir 0:1e4a74c2c16f 32 DatagramSocket::DatagramSocket(Host host, bool debug) : host(host), bufsize(512), debug(debug) {
macgyveremir 0:1e4a74c2c16f 33 createDatagramSocket();
macgyveremir 0:1e4a74c2c16f 34 }
macgyveremir 0:1e4a74c2c16f 35
macgyveremir 0:1e4a74c2c16f 36 DatagramSocket::DatagramSocket(IpAddr ip, int port, bool debug) : host(ip, port), bufsize(512), debug(debug) {
macgyveremir 0:1e4a74c2c16f 37 createDatagramSocket();
macgyveremir 0:1e4a74c2c16f 38 }
macgyveremir 0:1e4a74c2c16f 39
macgyveremir 0:1e4a74c2c16f 40 DatagramSocket::~DatagramSocket() {
macgyveremir 0:1e4a74c2c16f 41 delete[] buf;
macgyveremir 0:1e4a74c2c16f 42 }
macgyveremir 0:1e4a74c2c16f 43
macgyveremir 0:1e4a74c2c16f 44 void DatagramSocket::createDatagramSocket() {
macgyveremir 0:1e4a74c2c16f 45 udpSocket.setOnEvent(this, &DatagramSocket::onUDPSocketEvent);
macgyveremir 0:1e4a74c2c16f 46 UDPSocketErr err = udpSocket.bind(host);
macgyveremir 0:1e4a74c2c16f 47 DBG(DatagramError(err));
macgyveremir 0:1e4a74c2c16f 48 buf = new char[bufsize + 1];
macgyveremir 0:1e4a74c2c16f 49 }
macgyveremir 0:1e4a74c2c16f 50
macgyveremir 0:1e4a74c2c16f 51 int DatagramSocket::write(char *buf, int length) {
macgyveremir 0:1e4a74c2c16f 52 if (length > bufsize) length = bufsize;
macgyveremir 0:1e4a74c2c16f 53 this->length = length;
macgyveremir 0:1e4a74c2c16f 54 memcpy(this->buf, buf, length);
macgyveremir 0:1e4a74c2c16f 55 return length;
macgyveremir 0:1e4a74c2c16f 56 }
macgyveremir 0:1e4a74c2c16f 57
macgyveremir 0:1e4a74c2c16f 58 int DatagramSocket::printf(const char* format, ...) {
macgyveremir 0:1e4a74c2c16f 59 va_list argp;
macgyveremir 0:1e4a74c2c16f 60 va_start(argp, format);
macgyveremir 0:1e4a74c2c16f 61 int len = vsnprintf(buf, bufsize, format, argp);
macgyveremir 0:1e4a74c2c16f 62 va_end(argp);
macgyveremir 0:1e4a74c2c16f 63 if (len > 0)
macgyveremir 0:1e4a74c2c16f 64 length = len;
macgyveremir 0:1e4a74c2c16f 65 return len;
macgyveremir 0:1e4a74c2c16f 66 }
macgyveremir 0:1e4a74c2c16f 67
macgyveremir 0:1e4a74c2c16f 68 void DatagramSocket::send(Host& host) {
macgyveremir 0:1e4a74c2c16f 69 DBG("Host = %d.%d.%d.%d:%d\n", host.getIp()[0], host.getIp()[1], host.getIp()[2], host.getIp()[3], host.getPort());
macgyveremir 0:1e4a74c2c16f 70 Net::poll();
macgyveremir 0:1e4a74c2c16f 71 int err = udpSocket.sendto(buf, length, &host);
macgyveremir 0:1e4a74c2c16f 72 DBG(DatagramError((UDPSocketErr) (err < 0 ? err : 0)));
macgyveremir 0:1e4a74c2c16f 73 }
macgyveremir 0:1e4a74c2c16f 74
macgyveremir 0:1e4a74c2c16f 75 void DatagramSocket::send(IpAddr ip, int port) {
macgyveremir 0:1e4a74c2c16f 76 DBG("IP = %d.%d.%d.%d, Port = %d\n", ip[0], ip[1], ip[2], ip[3], port);
macgyveremir 0:1e4a74c2c16f 77 Host host(ip, port);
macgyveremir 0:1e4a74c2c16f 78 send(host);
macgyveremir 0:1e4a74c2c16f 79 }
macgyveremir 0:1e4a74c2c16f 80
macgyveremir 0:1e4a74c2c16f 81 void DatagramSocket::send(char *name, int port) {
macgyveremir 0:1e4a74c2c16f 82 DBG("Name = %s, Port = %d\n", name, port);
macgyveremir 0:1e4a74c2c16f 83 Resolver resolver;
macgyveremir 0:1e4a74c2c16f 84 IpAddr ip = resolver.resolve(name, debug);
macgyveremir 0:1e4a74c2c16f 85 send(ip, port);
macgyveremir 0:1e4a74c2c16f 86 }
macgyveremir 0:1e4a74c2c16f 87
macgyveremir 0:1e4a74c2c16f 88 int DatagramSocket::read(char *buf, int size) {
macgyveremir 0:1e4a74c2c16f 89 int len = length < size ? length : size;
macgyveremir 0:1e4a74c2c16f 90 if (len > 0)
macgyveremir 0:1e4a74c2c16f 91 memcpy(buf, this->buf, len);
macgyveremir 0:1e4a74c2c16f 92 return len;
macgyveremir 0:1e4a74c2c16f 93 }
macgyveremir 0:1e4a74c2c16f 94
macgyveremir 0:1e4a74c2c16f 95 int DatagramSocket::scanf(const char* format, ...) {
macgyveremir 0:1e4a74c2c16f 96 va_list argp;
macgyveremir 0:1e4a74c2c16f 97 va_start(argp, format);
macgyveremir 0:1e4a74c2c16f 98 buf[length] = '\0';
macgyveremir 0:1e4a74c2c16f 99 int ret = vsscanf(buf, format, argp);
macgyveremir 0:1e4a74c2c16f 100 va_end(argp);
macgyveremir 0:1e4a74c2c16f 101 return ret;
macgyveremir 0:1e4a74c2c16f 102 }
macgyveremir 0:1e4a74c2c16f 103
macgyveremir 0:1e4a74c2c16f 104 int DatagramSocket::receive(Host *host, float timeout) {
macgyveremir 0:1e4a74c2c16f 105 timer.reset();
macgyveremir 0:1e4a74c2c16f 106 timer.start();
macgyveremir 0:1e4a74c2c16f 107 readable = false;
macgyveremir 0:1e4a74c2c16f 108 length = 0;
macgyveremir 0:1e4a74c2c16f 109
macgyveremir 0:1e4a74c2c16f 110 do {
macgyveremir 0:1e4a74c2c16f 111 Net::poll();
macgyveremir 0:1e4a74c2c16f 112 if (readable) {
macgyveremir 0:1e4a74c2c16f 113 Host host2;
macgyveremir 0:1e4a74c2c16f 114 length = udpSocket.recvfrom(buf, bufsize, &host2);
macgyveremir 0:1e4a74c2c16f 115 if (length > 0 && host)
macgyveremir 0:1e4a74c2c16f 116 *host = host2;
macgyveremir 0:1e4a74c2c16f 117 break;
macgyveremir 0:1e4a74c2c16f 118 }
macgyveremir 0:1e4a74c2c16f 119 } while (timer.read() < timeout);
macgyveremir 0:1e4a74c2c16f 120
macgyveremir 0:1e4a74c2c16f 121 return length;
macgyveremir 0:1e4a74c2c16f 122 }
macgyveremir 0:1e4a74c2c16f 123
macgyveremir 0:1e4a74c2c16f 124 void DatagramSocket::setDebug(bool debug) {
macgyveremir 0:1e4a74c2c16f 125 this->debug = debug;
macgyveremir 0:1e4a74c2c16f 126 }
macgyveremir 0:1e4a74c2c16f 127
macgyveremir 0:1e4a74c2c16f 128 void DatagramSocket::onUDPSocketEvent(UDPSocketEvent e) {
macgyveremir 0:1e4a74c2c16f 129 DBG(DatagramEvent(e));
macgyveremir 0:1e4a74c2c16f 130
macgyveremir 0:1e4a74c2c16f 131 switch (e) {
macgyveremir 0:1e4a74c2c16f 132 case UDPSOCKET_READABLE: //The only event for now
macgyveremir 0:1e4a74c2c16f 133 readable = true;
macgyveremir 0:1e4a74c2c16f 134 break;
macgyveremir 0:1e4a74c2c16f 135 }
macgyveremir 0:1e4a74c2c16f 136 }