Hiroshi Yamaguchi / SimpleSocket 1.0

A Simple TCP & UDP Socket Library

Dependents:   SimpleSocketExamples 1.0

ClientSocket.cpp@0:3eb1ce25eac4, 2012-06-05 (annotated)

Committer:
yamaguch
Date:
Tue Jun 05 05:56:49 2012 +0000
Revision:
0:3eb1ce25eac4
1.1.3; fixed 0x00 reception bug

Who changed what in which revision?

UserRevisionLine numberNew contents of line
yamaguch 0:3eb1ce25eac4 1 /*
yamaguch 0:3eb1ce25eac4 2 Copyright (c) 2011, Senio Networks, Inc.
yamaguch 0:3eb1ce25eac4 3
yamaguch 0:3eb1ce25eac4 4 Permission is hereby granted, free of charge, to any person obtaining a copy
yamaguch 0:3eb1ce25eac4 5 of this software and associated documentation files (the "Software"), to deal
yamaguch 0:3eb1ce25eac4 6 in the Software without restriction, including without limitation the rights
yamaguch 0:3eb1ce25eac4 7 to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
yamaguch 0:3eb1ce25eac4 8 copies of the Software, and to permit persons to whom the Software is
yamaguch 0:3eb1ce25eac4 9 furnished to do so, subject to the following conditions:
yamaguch 0:3eb1ce25eac4 10
yamaguch 0:3eb1ce25eac4 11 The above copyright notice and this permission notice shall be included in
yamaguch 0:3eb1ce25eac4 12 all copies or substantial portions of the Software.
yamaguch 0:3eb1ce25eac4 13
yamaguch 0:3eb1ce25eac4 14 THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
yamaguch 0:3eb1ce25eac4 15 IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
yamaguch 0:3eb1ce25eac4 16 FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
yamaguch 0:3eb1ce25eac4 17 AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
yamaguch 0:3eb1ce25eac4 18 LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
yamaguch 0:3eb1ce25eac4 19 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
yamaguch 0:3eb1ce25eac4 20 THE SOFTWARE.
yamaguch 0:3eb1ce25eac4 21 */
yamaguch 0:3eb1ce25eac4 22
yamaguch 0:3eb1ce25eac4 23 #include "SimpleSocket.h"
yamaguch 0:3eb1ce25eac4 24 #include <stdarg.h>
yamaguch 0:3eb1ce25eac4 25
yamaguch 0:3eb1ce25eac4 26 ClientSocket::ClientSocket(IpAddr ip, TCPSocket *socket, bool debug)
yamaguch 0:3eb1ce25eac4 27 : ip(ip), socket(socket), readable(true), writable(true), preread(-1),
yamaguch 0:3eb1ce25eac4 28 state(CONNECTED), debug(debug) {
yamaguch 0:3eb1ce25eac4 29 if (socket)
yamaguch 0:3eb1ce25eac4 30 socket->setOnEvent(this, &ClientSocket::onTCPSocketEvent);
yamaguch 0:3eb1ce25eac4 31 }
yamaguch 0:3eb1ce25eac4 32
yamaguch 0:3eb1ce25eac4 33 ClientSocket::ClientSocket(IpAddr ip, int port, float timeout, bool debug)
yamaguch 0:3eb1ce25eac4 34 : readable(true), writable(true), preread(-1), debug(debug) {
yamaguch 0:3eb1ce25eac4 35 createClientSocket(ip, port, timeout);
yamaguch 0:3eb1ce25eac4 36 }
yamaguch 0:3eb1ce25eac4 37
yamaguch 0:3eb1ce25eac4 38 ClientSocket::ClientSocket(char *hostname, int port, float timeout, bool debug)
yamaguch 0:3eb1ce25eac4 39 : readable(true), writable(true), preread(-1), debug(debug) {
yamaguch 0:3eb1ce25eac4 40 Resolver resolver;
yamaguch 0:3eb1ce25eac4 41 createClientSocket(resolver.resolve(hostname, debug), port, timeout);
yamaguch 0:3eb1ce25eac4 42 }
yamaguch 0:3eb1ce25eac4 43
yamaguch 0:3eb1ce25eac4 44 ClientSocket::ClientSocket(const ClientSocket& that) {
yamaguch 0:3eb1ce25eac4 45 *this = that;
yamaguch 0:3eb1ce25eac4 46 ::printf("cc called\n");
yamaguch 0:3eb1ce25eac4 47 }
yamaguch 0:3eb1ce25eac4 48
yamaguch 0:3eb1ce25eac4 49 void ClientSocket::createClientSocket(IpAddr ip, int port, float timeout) {
yamaguch 0:3eb1ce25eac4 50 this->ip = ip;
yamaguch 0:3eb1ce25eac4 51 socket = new TCPSocket();
yamaguch 0:3eb1ce25eac4 52 socket->setOnEvent(this, &ClientSocket::onTCPSocketEvent);
yamaguch 0:3eb1ce25eac4 53 SocketError err = socket->connect(Host(ip, port));
yamaguch 0:3eb1ce25eac4 54 if (!err) {
yamaguch 0:3eb1ce25eac4 55 Timer timer;
yamaguch 0:3eb1ce25eac4 56 timer.start();
yamaguch 0:3eb1ce25eac4 57 state = CONNECTING;
yamaguch 0:3eb1ce25eac4 58 while (state == CONNECTING && timer.read() < timeout) {
yamaguch 0:3eb1ce25eac4 59 Net::poll();
yamaguch 0:3eb1ce25eac4 60 if (state == CONNECTED)
yamaguch 0:3eb1ce25eac4 61 return;
yamaguch 0:3eb1ce25eac4 62 wait_us(100);
yamaguch 0:3eb1ce25eac4 63 }
yamaguch 0:3eb1ce25eac4 64 } else
yamaguch 0:3eb1ce25eac4 65 DBG(SocketError(err));
yamaguch 0:3eb1ce25eac4 66
yamaguch 0:3eb1ce25eac4 67 writable = false;
yamaguch 0:3eb1ce25eac4 68 state = DISCONNECTED;
yamaguch 0:3eb1ce25eac4 69 socket->resetOnEvent();
yamaguch 0:3eb1ce25eac4 70 socket->close();
yamaguch 0:3eb1ce25eac4 71 delete socket;
yamaguch 0:3eb1ce25eac4 72 socket = 0;
yamaguch 0:3eb1ce25eac4 73 }
yamaguch 0:3eb1ce25eac4 74
yamaguch 0:3eb1ce25eac4 75 IpAddr ClientSocket::getIp() {
yamaguch 0:3eb1ce25eac4 76 return ip;
yamaguch 0:3eb1ce25eac4 77 }
yamaguch 0:3eb1ce25eac4 78
yamaguch 0:3eb1ce25eac4 79 bool ClientSocket::connected() {
yamaguch 0:3eb1ce25eac4 80 if (socket)
yamaguch 0:3eb1ce25eac4 81 Net::poll();
yamaguch 0:3eb1ce25eac4 82
yamaguch 0:3eb1ce25eac4 83 if (!readable && state == DISCONNECTED)
yamaguch 0:3eb1ce25eac4 84 close();
yamaguch 0:3eb1ce25eac4 85
yamaguch 0:3eb1ce25eac4 86 return socket != 0;
yamaguch 0:3eb1ce25eac4 87 }
yamaguch 0:3eb1ce25eac4 88
yamaguch 0:3eb1ce25eac4 89 bool ClientSocket::available() {
yamaguch 0:3eb1ce25eac4 90 if (preread != -1)
yamaguch 0:3eb1ce25eac4 91 return true;
yamaguch 0:3eb1ce25eac4 92
yamaguch 0:3eb1ce25eac4 93 if (!readable)
yamaguch 0:3eb1ce25eac4 94 Net::poll();
yamaguch 0:3eb1ce25eac4 95
yamaguch 0:3eb1ce25eac4 96 if (readable) {
yamaguch 0:3eb1ce25eac4 97 char c;
yamaguch 0:3eb1ce25eac4 98 int ret = socket->recv(&c, 1);
yamaguch 0:3eb1ce25eac4 99 if (ret > 0) {
yamaguch 0:3eb1ce25eac4 100 preread = c;
yamaguch 0:3eb1ce25eac4 101 return true;
yamaguch 0:3eb1ce25eac4 102 } else
yamaguch 0:3eb1ce25eac4 103 readable = false;
yamaguch 0:3eb1ce25eac4 104 }
yamaguch 0:3eb1ce25eac4 105
yamaguch 0:3eb1ce25eac4 106 if (!readable && state == DISCONNECTED)
yamaguch 0:3eb1ce25eac4 107 close();
yamaguch 0:3eb1ce25eac4 108
yamaguch 0:3eb1ce25eac4 109 return false;
yamaguch 0:3eb1ce25eac4 110 }
yamaguch 0:3eb1ce25eac4 111
yamaguch 0:3eb1ce25eac4 112 void ClientSocket::onTCPSocketEvent(TCPSocketEvent e) {
yamaguch 0:3eb1ce25eac4 113 DBG(SocketEvent(e));
yamaguch 0:3eb1ce25eac4 114
yamaguch 0:3eb1ce25eac4 115 switch (e) {
yamaguch 0:3eb1ce25eac4 116 case TCPSOCKET_CONNECTED:
yamaguch 0:3eb1ce25eac4 117 state = CONNECTED;
yamaguch 0:3eb1ce25eac4 118 break;
yamaguch 0:3eb1ce25eac4 119 case TCPSOCKET_READABLE:
yamaguch 0:3eb1ce25eac4 120 readable = true;
yamaguch 0:3eb1ce25eac4 121 break;
yamaguch 0:3eb1ce25eac4 122 case TCPSOCKET_WRITEABLE:
yamaguch 0:3eb1ce25eac4 123 writable = true;
yamaguch 0:3eb1ce25eac4 124 break;
yamaguch 0:3eb1ce25eac4 125 case TCPSOCKET_DISCONNECTED:
yamaguch 0:3eb1ce25eac4 126 state = DISCONNECTED;
yamaguch 0:3eb1ce25eac4 127 break;
yamaguch 0:3eb1ce25eac4 128 }
yamaguch 0:3eb1ce25eac4 129 }
yamaguch 0:3eb1ce25eac4 130
yamaguch 0:3eb1ce25eac4 131 int ClientSocket::read() {
yamaguch 0:3eb1ce25eac4 132 char c;
yamaguch 0:3eb1ce25eac4 133
yamaguch 0:3eb1ce25eac4 134 if (preread != -1) {
yamaguch 0:3eb1ce25eac4 135 c = preread & 255;
yamaguch 0:3eb1ce25eac4 136 preread = -1;
yamaguch 0:3eb1ce25eac4 137 return c;
yamaguch 0:3eb1ce25eac4 138 }
yamaguch 0:3eb1ce25eac4 139
yamaguch 0:3eb1ce25eac4 140 int ret = socket->recv(&c, 1);
yamaguch 0:3eb1ce25eac4 141 if (ret > 0) {
yamaguch 0:3eb1ce25eac4 142 return (int) c & 255;
yamaguch 0:3eb1ce25eac4 143 } else {
yamaguch 0:3eb1ce25eac4 144 readable = false;
yamaguch 0:3eb1ce25eac4 145 return -1;
yamaguch 0:3eb1ce25eac4 146 }
yamaguch 0:3eb1ce25eac4 147 }
yamaguch 0:3eb1ce25eac4 148
yamaguch 0:3eb1ce25eac4 149 int ClientSocket::read(char *buf, int size) {
yamaguch 0:3eb1ce25eac4 150 int req = size;
yamaguch 0:3eb1ce25eac4 151
yamaguch 0:3eb1ce25eac4 152 if (preread != -1 && size > 0) {
yamaguch 0:3eb1ce25eac4 153 *buf++ = preread & 255;
yamaguch 0:3eb1ce25eac4 154 preread = -1;
yamaguch 0:3eb1ce25eac4 155 req--;
yamaguch 0:3eb1ce25eac4 156 }
yamaguch 0:3eb1ce25eac4 157
yamaguch 0:3eb1ce25eac4 158 while (req > 0) {
yamaguch 0:3eb1ce25eac4 159 int got = socket->recv(buf, req);
yamaguch 0:3eb1ce25eac4 160 if (got > 0) {
yamaguch 0:3eb1ce25eac4 161 buf += got;
yamaguch 0:3eb1ce25eac4 162 req -= got;
yamaguch 0:3eb1ce25eac4 163 } else {
yamaguch 0:3eb1ce25eac4 164 readable = false;
yamaguch 0:3eb1ce25eac4 165 break;
yamaguch 0:3eb1ce25eac4 166 }
yamaguch 0:3eb1ce25eac4 167 }
yamaguch 0:3eb1ce25eac4 168
yamaguch 0:3eb1ce25eac4 169 return size - req;
yamaguch 0:3eb1ce25eac4 170 }
yamaguch 0:3eb1ce25eac4 171
yamaguch 0:3eb1ce25eac4 172 int ClientSocket::scanf(const char* format, ...) {
yamaguch 0:3eb1ce25eac4 173 va_list argp;
yamaguch 0:3eb1ce25eac4 174 va_start(argp, format);
yamaguch 0:3eb1ce25eac4 175 char buf[256];
yamaguch 0:3eb1ce25eac4 176
yamaguch 0:3eb1ce25eac4 177 int len = read(buf, sizeof(buf) - 1);
yamaguch 0:3eb1ce25eac4 178 if (len <= 0)
yamaguch 0:3eb1ce25eac4 179 return 0;
yamaguch 0:3eb1ce25eac4 180 buf[len] = '\0';
yamaguch 0:3eb1ce25eac4 181
yamaguch 0:3eb1ce25eac4 182 int ret = vsscanf(buf, format, argp);
yamaguch 0:3eb1ce25eac4 183 va_end(argp);
yamaguch 0:3eb1ce25eac4 184 return ret;
yamaguch 0:3eb1ce25eac4 185 }
yamaguch 0:3eb1ce25eac4 186
yamaguch 0:3eb1ce25eac4 187 int ClientSocket::write(char c) {
yamaguch 0:3eb1ce25eac4 188 return write(&c, 1);
yamaguch 0:3eb1ce25eac4 189 }
yamaguch 0:3eb1ce25eac4 190
yamaguch 0:3eb1ce25eac4 191 int ClientSocket::write(char *buf, int size) {
yamaguch 0:3eb1ce25eac4 192 int ret = socket->send(buf, size);
yamaguch 0:3eb1ce25eac4 193 if (ret > 0) {
yamaguch 0:3eb1ce25eac4 194 return ret;
yamaguch 0:3eb1ce25eac4 195 } else {
yamaguch 0:3eb1ce25eac4 196 writable = false;
yamaguch 0:3eb1ce25eac4 197 return -1;
yamaguch 0:3eb1ce25eac4 198 }
yamaguch 0:3eb1ce25eac4 199 }
yamaguch 0:3eb1ce25eac4 200
yamaguch 0:3eb1ce25eac4 201 int ClientSocket::printf(const char* format, ...) {
yamaguch 0:3eb1ce25eac4 202 va_list argp;
yamaguch 0:3eb1ce25eac4 203 va_start(argp, format);
yamaguch 0:3eb1ce25eac4 204 char buf[256];
yamaguch 0:3eb1ce25eac4 205
yamaguch 0:3eb1ce25eac4 206 int len = vsnprintf(buf, sizeof(buf), format, argp);
yamaguch 0:3eb1ce25eac4 207 va_end(argp);
yamaguch 0:3eb1ce25eac4 208 return write(buf, len);
yamaguch 0:3eb1ce25eac4 209 }
yamaguch 0:3eb1ce25eac4 210
yamaguch 0:3eb1ce25eac4 211 void ClientSocket::close() {
yamaguch 0:3eb1ce25eac4 212 if (socket) {
yamaguch 0:3eb1ce25eac4 213 readable = false;
yamaguch 0:3eb1ce25eac4 214 writable = false;
yamaguch 0:3eb1ce25eac4 215 socket->resetOnEvent();
yamaguch 0:3eb1ce25eac4 216 socket->close();
yamaguch 0:3eb1ce25eac4 217 delete socket;
yamaguch 0:3eb1ce25eac4 218 socket = 0;
yamaguch 0:3eb1ce25eac4 219 }
yamaguch 0:3eb1ce25eac4 220 }
yamaguch 0:3eb1ce25eac4 221
yamaguch 0:3eb1ce25eac4 222 ClientSocket::operator bool() {
yamaguch 0:3eb1ce25eac4 223 return connected();
yamaguch 0:3eb1ce25eac4 224 }
yamaguch 0:3eb1ce25eac4 225
yamaguch 0:3eb1ce25eac4 226 void ClientSocket::setDebug(bool debug) {
yamaguch 0:3eb1ce25eac4 227 this->debug = debug;
yamaguch 0:3eb1ce25eac4 228 }