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?
User | Revision | Line number | New 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 | } |