Daiki Kato / mbed-os-lychee

Dependents:   mbed-os-example-blinky-gr-lychee GR-Boads_Camera_sample GR-Boards_Audio_Recoder GR-Boads_Camera_DisplayApp ... more

Committer:
dkato
Date:
Fri Feb 02 05:42:23 2018 +0000
Revision:
0:f782d9c66c49
mbed-os for GR-LYCHEE

Who changed what in which revision?

UserRevisionLine numberNew contents of line
dkato 0:f782d9c66c49 1 /* Socket
dkato 0:f782d9c66c49 2 * Copyright (c) 2015 ARM Limited
dkato 0:f782d9c66c49 3 *
dkato 0:f782d9c66c49 4 * Licensed under the Apache License, Version 2.0 (the "License");
dkato 0:f782d9c66c49 5 * you may not use this file except in compliance with the License.
dkato 0:f782d9c66c49 6 * You may obtain a copy of the License at
dkato 0:f782d9c66c49 7 *
dkato 0:f782d9c66c49 8 * http://www.apache.org/licenses/LICENSE-2.0
dkato 0:f782d9c66c49 9 *
dkato 0:f782d9c66c49 10 * Unless required by applicable law or agreed to in writing, software
dkato 0:f782d9c66c49 11 * distributed under the License is distributed on an "AS IS" BASIS,
dkato 0:f782d9c66c49 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
dkato 0:f782d9c66c49 13 * See the License for the specific language governing permissions and
dkato 0:f782d9c66c49 14 * limitations under the License.
dkato 0:f782d9c66c49 15 */
dkato 0:f782d9c66c49 16
dkato 0:f782d9c66c49 17 #include "SocketAddress.h"
dkato 0:f782d9c66c49 18 #include "NetworkInterface.h"
dkato 0:f782d9c66c49 19 #include "NetworkStack.h"
dkato 0:f782d9c66c49 20 #include <string.h>
dkato 0:f782d9c66c49 21 #include "mbed.h"
dkato 0:f782d9c66c49 22
dkato 0:f782d9c66c49 23
dkato 0:f782d9c66c49 24 static bool ipv4_is_valid(const char *addr)
dkato 0:f782d9c66c49 25 {
dkato 0:f782d9c66c49 26 int i = 0;
dkato 0:f782d9c66c49 27
dkato 0:f782d9c66c49 28 // Check each digit for [0-9.]
dkato 0:f782d9c66c49 29 for (; addr[i]; i++) {
dkato 0:f782d9c66c49 30 if (!(addr[i] >= '0' && addr[i] <= '9') && addr[i] != '.') {
dkato 0:f782d9c66c49 31 return false;
dkato 0:f782d9c66c49 32 }
dkato 0:f782d9c66c49 33 }
dkato 0:f782d9c66c49 34
dkato 0:f782d9c66c49 35 // Ending with '.' garuntees host
dkato 0:f782d9c66c49 36 if (i > 0 && addr[i-1] == '.') {
dkato 0:f782d9c66c49 37 return false;
dkato 0:f782d9c66c49 38 }
dkato 0:f782d9c66c49 39
dkato 0:f782d9c66c49 40 return true;
dkato 0:f782d9c66c49 41 }
dkato 0:f782d9c66c49 42
dkato 0:f782d9c66c49 43 static bool ipv6_is_valid(const char *addr)
dkato 0:f782d9c66c49 44 {
dkato 0:f782d9c66c49 45 // Check each digit for [0-9a-fA-F:]
dkato 0:f782d9c66c49 46 // Must also have at least 2 colons
dkato 0:f782d9c66c49 47 int colons = 0;
dkato 0:f782d9c66c49 48 for (int i = 0; addr[i]; i++) {
dkato 0:f782d9c66c49 49 if (!(addr[i] >= '0' && addr[i] <= '9') &&
dkato 0:f782d9c66c49 50 !(addr[i] >= 'a' && addr[i] <= 'f') &&
dkato 0:f782d9c66c49 51 !(addr[i] >= 'A' && addr[i] <= 'F') &&
dkato 0:f782d9c66c49 52 addr[i] != ':') {
dkato 0:f782d9c66c49 53 return false;
dkato 0:f782d9c66c49 54 }
dkato 0:f782d9c66c49 55 if (addr[i] == ':') {
dkato 0:f782d9c66c49 56 colons++;
dkato 0:f782d9c66c49 57 }
dkato 0:f782d9c66c49 58 }
dkato 0:f782d9c66c49 59
dkato 0:f782d9c66c49 60 return colons >= 2;
dkato 0:f782d9c66c49 61 }
dkato 0:f782d9c66c49 62
dkato 0:f782d9c66c49 63 static void ipv4_from_address(uint8_t *bytes, const char *addr)
dkato 0:f782d9c66c49 64 {
dkato 0:f782d9c66c49 65 int count = 0;
dkato 0:f782d9c66c49 66 int i = 0;
dkato 0:f782d9c66c49 67
dkato 0:f782d9c66c49 68 for (; count < NSAPI_IPv4_BYTES; count++) {
dkato 0:f782d9c66c49 69 unsigned char b;
dkato 0:f782d9c66c49 70 int scanned = sscanf(&addr[i], "%hhu", &b);
dkato 0:f782d9c66c49 71 if (scanned < 1) {
dkato 0:f782d9c66c49 72 return;
dkato 0:f782d9c66c49 73 }
dkato 0:f782d9c66c49 74
dkato 0:f782d9c66c49 75 bytes[count] = b;
dkato 0:f782d9c66c49 76
dkato 0:f782d9c66c49 77 for (; addr[i] != '.'; i++) {
dkato 0:f782d9c66c49 78 if (!addr[i]) {
dkato 0:f782d9c66c49 79 return;
dkato 0:f782d9c66c49 80 }
dkato 0:f782d9c66c49 81 }
dkato 0:f782d9c66c49 82
dkato 0:f782d9c66c49 83 i++;
dkato 0:f782d9c66c49 84 }
dkato 0:f782d9c66c49 85 }
dkato 0:f782d9c66c49 86
dkato 0:f782d9c66c49 87 static int ipv6_scan_chunk(uint16_t *shorts, const char *chunk) {
dkato 0:f782d9c66c49 88 int count = 0;
dkato 0:f782d9c66c49 89 int i = 0;
dkato 0:f782d9c66c49 90
dkato 0:f782d9c66c49 91 for (; count < NSAPI_IPv6_BYTES/2; count++) {
dkato 0:f782d9c66c49 92 unsigned short s;
dkato 0:f782d9c66c49 93 int scanned = sscanf(&chunk[i], "%hx", &s);
dkato 0:f782d9c66c49 94 if (scanned < 1) {
dkato 0:f782d9c66c49 95 return count;
dkato 0:f782d9c66c49 96 }
dkato 0:f782d9c66c49 97
dkato 0:f782d9c66c49 98 shorts[count] = s;
dkato 0:f782d9c66c49 99
dkato 0:f782d9c66c49 100 for (; chunk[i] != ':'; i++) {
dkato 0:f782d9c66c49 101 if (!chunk[i]) {
dkato 0:f782d9c66c49 102 return count+1;
dkato 0:f782d9c66c49 103 }
dkato 0:f782d9c66c49 104 }
dkato 0:f782d9c66c49 105
dkato 0:f782d9c66c49 106 i++;
dkato 0:f782d9c66c49 107 }
dkato 0:f782d9c66c49 108
dkato 0:f782d9c66c49 109 return count;
dkato 0:f782d9c66c49 110 }
dkato 0:f782d9c66c49 111
dkato 0:f782d9c66c49 112 static void ipv6_from_address(uint8_t *bytes, const char *addr)
dkato 0:f782d9c66c49 113 {
dkato 0:f782d9c66c49 114 // Start with zeroed address
dkato 0:f782d9c66c49 115 uint16_t shorts[NSAPI_IPv6_BYTES/2];
dkato 0:f782d9c66c49 116 int suffix = 0;
dkato 0:f782d9c66c49 117
dkato 0:f782d9c66c49 118 // Find double colons and scan suffix
dkato 0:f782d9c66c49 119 for (int i = 0; addr[i]; i++) {
dkato 0:f782d9c66c49 120 if (addr[i] == ':' && addr[i+1] == ':') {
dkato 0:f782d9c66c49 121 suffix = ipv6_scan_chunk(shorts, &addr[i+2]);
dkato 0:f782d9c66c49 122 break;
dkato 0:f782d9c66c49 123 }
dkato 0:f782d9c66c49 124 }
dkato 0:f782d9c66c49 125
dkato 0:f782d9c66c49 126 // Move suffix to end
dkato 0:f782d9c66c49 127 memmove(&shorts[NSAPI_IPv6_BYTES/2-suffix], &shorts[0],
dkato 0:f782d9c66c49 128 suffix*sizeof(uint16_t));
dkato 0:f782d9c66c49 129 memset(&shorts[0], 0,
dkato 0:f782d9c66c49 130 (NSAPI_IPv6_BYTES/2-suffix)*sizeof(uint16_t));
dkato 0:f782d9c66c49 131
dkato 0:f782d9c66c49 132 // Scan prefix
dkato 0:f782d9c66c49 133 ipv6_scan_chunk(shorts, &addr[0]);
dkato 0:f782d9c66c49 134
dkato 0:f782d9c66c49 135 // Flip bytes
dkato 0:f782d9c66c49 136 for (int i = 0; i < NSAPI_IPv6_BYTES/2; i++) {
dkato 0:f782d9c66c49 137 bytes[2*i+0] = (uint8_t)(shorts[i] >> 8);
dkato 0:f782d9c66c49 138 bytes[2*i+1] = (uint8_t)(shorts[i] >> 0);
dkato 0:f782d9c66c49 139 }
dkato 0:f782d9c66c49 140 }
dkato 0:f782d9c66c49 141
dkato 0:f782d9c66c49 142 static void ipv4_to_address(char *addr, const uint8_t *bytes)
dkato 0:f782d9c66c49 143 {
dkato 0:f782d9c66c49 144 sprintf(addr, "%d.%d.%d.%d", bytes[0], bytes[1], bytes[2], bytes[3]);
dkato 0:f782d9c66c49 145 }
dkato 0:f782d9c66c49 146
dkato 0:f782d9c66c49 147 static void ipv6_to_address(char *addr, const uint8_t *bytes)
dkato 0:f782d9c66c49 148 {
dkato 0:f782d9c66c49 149 for (int i = 0; i < NSAPI_IPv6_BYTES/2; i++) {
dkato 0:f782d9c66c49 150 sprintf(&addr[5*i], "%02x%02x", bytes[2*i], bytes[2*i+1]);
dkato 0:f782d9c66c49 151 addr[5*i+4] = ':';
dkato 0:f782d9c66c49 152 }
dkato 0:f782d9c66c49 153 addr[NSAPI_IPv6_SIZE-1] = '\0';
dkato 0:f782d9c66c49 154 }
dkato 0:f782d9c66c49 155
dkato 0:f782d9c66c49 156
dkato 0:f782d9c66c49 157 SocketAddress::SocketAddress(nsapi_addr_t addr, uint16_t port)
dkato 0:f782d9c66c49 158 {
dkato 0:f782d9c66c49 159 _ip_address[0] = '\0';
dkato 0:f782d9c66c49 160 set_addr(addr);
dkato 0:f782d9c66c49 161 set_port(port);
dkato 0:f782d9c66c49 162 }
dkato 0:f782d9c66c49 163
dkato 0:f782d9c66c49 164 SocketAddress::SocketAddress(const char *addr, uint16_t port)
dkato 0:f782d9c66c49 165 {
dkato 0:f782d9c66c49 166 _ip_address[0] = '\0';
dkato 0:f782d9c66c49 167 set_ip_address(addr);
dkato 0:f782d9c66c49 168 set_port(port);
dkato 0:f782d9c66c49 169 }
dkato 0:f782d9c66c49 170
dkato 0:f782d9c66c49 171 SocketAddress::SocketAddress(const void *bytes, nsapi_version_t version, uint16_t port)
dkato 0:f782d9c66c49 172 {
dkato 0:f782d9c66c49 173 _ip_address[0] = '\0';
dkato 0:f782d9c66c49 174 set_ip_bytes(bytes, version);
dkato 0:f782d9c66c49 175 set_port(port);
dkato 0:f782d9c66c49 176 }
dkato 0:f782d9c66c49 177
dkato 0:f782d9c66c49 178 SocketAddress::SocketAddress(const SocketAddress &addr)
dkato 0:f782d9c66c49 179 {
dkato 0:f782d9c66c49 180 _ip_address[0] = '\0';
dkato 0:f782d9c66c49 181 set_addr(addr.get_addr());
dkato 0:f782d9c66c49 182 set_port(addr.get_port());
dkato 0:f782d9c66c49 183 }
dkato 0:f782d9c66c49 184
dkato 0:f782d9c66c49 185 bool SocketAddress::set_ip_address(const char *addr)
dkato 0:f782d9c66c49 186 {
dkato 0:f782d9c66c49 187 _ip_address[0] = '\0';
dkato 0:f782d9c66c49 188
dkato 0:f782d9c66c49 189 if (addr && ipv4_is_valid(addr)) {
dkato 0:f782d9c66c49 190 _addr.version = NSAPI_IPv4;
dkato 0:f782d9c66c49 191 ipv4_from_address(_addr.bytes, addr);
dkato 0:f782d9c66c49 192 return true;
dkato 0:f782d9c66c49 193 } else if (addr && ipv6_is_valid(addr)) {
dkato 0:f782d9c66c49 194 _addr.version = NSAPI_IPv6;
dkato 0:f782d9c66c49 195 ipv6_from_address(_addr.bytes, addr);
dkato 0:f782d9c66c49 196 return true;
dkato 0:f782d9c66c49 197 } else {
dkato 0:f782d9c66c49 198 _addr = nsapi_addr_t();
dkato 0:f782d9c66c49 199 return false;
dkato 0:f782d9c66c49 200 }
dkato 0:f782d9c66c49 201 }
dkato 0:f782d9c66c49 202
dkato 0:f782d9c66c49 203 void SocketAddress::set_ip_bytes(const void *bytes, nsapi_version_t version)
dkato 0:f782d9c66c49 204 {
dkato 0:f782d9c66c49 205 nsapi_addr_t addr;
dkato 0:f782d9c66c49 206
dkato 0:f782d9c66c49 207 addr = nsapi_addr_t();
dkato 0:f782d9c66c49 208 addr.version = version;
dkato 0:f782d9c66c49 209 if (version == NSAPI_IPv6) {
dkato 0:f782d9c66c49 210 memcpy(addr.bytes, bytes, NSAPI_IPv6_BYTES);
dkato 0:f782d9c66c49 211 } else if (version == NSAPI_IPv4) {
dkato 0:f782d9c66c49 212 memcpy(addr.bytes, bytes, NSAPI_IPv4_BYTES);
dkato 0:f782d9c66c49 213 }
dkato 0:f782d9c66c49 214 set_addr(addr);
dkato 0:f782d9c66c49 215 }
dkato 0:f782d9c66c49 216
dkato 0:f782d9c66c49 217 void SocketAddress::set_addr(nsapi_addr_t addr)
dkato 0:f782d9c66c49 218 {
dkato 0:f782d9c66c49 219 _ip_address[0] = '\0';
dkato 0:f782d9c66c49 220 _addr = addr;
dkato 0:f782d9c66c49 221 }
dkato 0:f782d9c66c49 222
dkato 0:f782d9c66c49 223 void SocketAddress::set_port(uint16_t port)
dkato 0:f782d9c66c49 224 {
dkato 0:f782d9c66c49 225 _port = port;
dkato 0:f782d9c66c49 226 }
dkato 0:f782d9c66c49 227
dkato 0:f782d9c66c49 228 const char *SocketAddress::get_ip_address() const
dkato 0:f782d9c66c49 229 {
dkato 0:f782d9c66c49 230 if (_addr.version == NSAPI_UNSPEC) {
dkato 0:f782d9c66c49 231 return NULL;
dkato 0:f782d9c66c49 232 }
dkato 0:f782d9c66c49 233
dkato 0:f782d9c66c49 234 if (!_ip_address[0]) {
dkato 0:f782d9c66c49 235 if (_addr.version == NSAPI_IPv4) {
dkato 0:f782d9c66c49 236 ipv4_to_address(_ip_address, _addr.bytes);
dkato 0:f782d9c66c49 237 } else if (_addr.version == NSAPI_IPv6) {
dkato 0:f782d9c66c49 238 ipv6_to_address(_ip_address, _addr.bytes);
dkato 0:f782d9c66c49 239 }
dkato 0:f782d9c66c49 240 }
dkato 0:f782d9c66c49 241
dkato 0:f782d9c66c49 242 return _ip_address;
dkato 0:f782d9c66c49 243 }
dkato 0:f782d9c66c49 244
dkato 0:f782d9c66c49 245 const void *SocketAddress::get_ip_bytes() const
dkato 0:f782d9c66c49 246 {
dkato 0:f782d9c66c49 247 return _addr.bytes;
dkato 0:f782d9c66c49 248 }
dkato 0:f782d9c66c49 249
dkato 0:f782d9c66c49 250 nsapi_version_t SocketAddress::get_ip_version() const
dkato 0:f782d9c66c49 251 {
dkato 0:f782d9c66c49 252 return _addr.version;
dkato 0:f782d9c66c49 253 }
dkato 0:f782d9c66c49 254
dkato 0:f782d9c66c49 255 nsapi_addr_t SocketAddress::get_addr() const
dkato 0:f782d9c66c49 256 {
dkato 0:f782d9c66c49 257 return _addr;
dkato 0:f782d9c66c49 258 }
dkato 0:f782d9c66c49 259
dkato 0:f782d9c66c49 260 uint16_t SocketAddress::get_port() const
dkato 0:f782d9c66c49 261 {
dkato 0:f782d9c66c49 262 return _port;
dkato 0:f782d9c66c49 263 }
dkato 0:f782d9c66c49 264
dkato 0:f782d9c66c49 265 SocketAddress::operator bool() const
dkato 0:f782d9c66c49 266 {
dkato 0:f782d9c66c49 267 if (_addr.version == NSAPI_IPv4) {
dkato 0:f782d9c66c49 268 for (int i = 0; i < NSAPI_IPv4_BYTES; i++) {
dkato 0:f782d9c66c49 269 if (_addr.bytes[i]) {
dkato 0:f782d9c66c49 270 return true;
dkato 0:f782d9c66c49 271 }
dkato 0:f782d9c66c49 272 }
dkato 0:f782d9c66c49 273
dkato 0:f782d9c66c49 274 return false;
dkato 0:f782d9c66c49 275 } else if (_addr.version == NSAPI_IPv6) {
dkato 0:f782d9c66c49 276 for (int i = 0; i < NSAPI_IPv6_BYTES; i++) {
dkato 0:f782d9c66c49 277 if (_addr.bytes[i]) {
dkato 0:f782d9c66c49 278 return true;
dkato 0:f782d9c66c49 279 }
dkato 0:f782d9c66c49 280 }
dkato 0:f782d9c66c49 281
dkato 0:f782d9c66c49 282 return false;
dkato 0:f782d9c66c49 283 } else {
dkato 0:f782d9c66c49 284 return false;
dkato 0:f782d9c66c49 285 }
dkato 0:f782d9c66c49 286 }
dkato 0:f782d9c66c49 287
dkato 0:f782d9c66c49 288 bool operator==(const SocketAddress &a, const SocketAddress &b)
dkato 0:f782d9c66c49 289 {
dkato 0:f782d9c66c49 290 if (!a && !b) {
dkato 0:f782d9c66c49 291 return true;
dkato 0:f782d9c66c49 292 } else if (a._addr.version != b._addr.version) {
dkato 0:f782d9c66c49 293 return false;
dkato 0:f782d9c66c49 294 } else if (a._addr.version == NSAPI_IPv4) {
dkato 0:f782d9c66c49 295 return memcmp(a._addr.bytes, b._addr.bytes, NSAPI_IPv4_BYTES) == 0;
dkato 0:f782d9c66c49 296 } else if (a._addr.version == NSAPI_IPv6) {
dkato 0:f782d9c66c49 297 return memcmp(a._addr.bytes, b._addr.bytes, NSAPI_IPv6_BYTES) == 0;
dkato 0:f782d9c66c49 298 }
dkato 0:f782d9c66c49 299
dkato 0:f782d9c66c49 300 MBED_UNREACHABLE;
dkato 0:f782d9c66c49 301 }
dkato 0:f782d9c66c49 302
dkato 0:f782d9c66c49 303 bool operator!=(const SocketAddress &a, const SocketAddress &b)
dkato 0:f782d9c66c49 304 {
dkato 0:f782d9c66c49 305 return !(a == b);
dkato 0:f782d9c66c49 306 }
dkato 0:f782d9c66c49 307
dkato 0:f782d9c66c49 308 void SocketAddress::_SocketAddress(NetworkStack *iface, const char *host, uint16_t port)
dkato 0:f782d9c66c49 309 {
dkato 0:f782d9c66c49 310 _ip_address[0] = '\0';
dkato 0:f782d9c66c49 311
dkato 0:f782d9c66c49 312 // gethostbyname must check for literals, so can call it directly
dkato 0:f782d9c66c49 313 int err = iface->gethostbyname(host, this);
dkato 0:f782d9c66c49 314 _port = port;
dkato 0:f782d9c66c49 315 if (err) {
dkato 0:f782d9c66c49 316 _addr = nsapi_addr_t();
dkato 0:f782d9c66c49 317 _port = 0;
dkato 0:f782d9c66c49 318 }
dkato 0:f782d9c66c49 319 }