Adam Green / EthernetInterface

Deprecated fork of old network stack source from github. Please use official library instead: https://mbed.org/users/mbed_official/code/EthernetInterface/

Socket/TCPSocketConnection.cpp@0:3b00827bb0b7, 2013-09-07 (annotated)

Committer:
AdamGreen
Date:
Sat Sep 07 21:38:42 2013 +0000
Revision:
0:3b00827bb0b7
Sync to latest network stack source from github.

Who changed what in which revision?

UserRevisionLine numberNew contents of line
AdamGreen 0:3b00827bb0b7 1 /* Copyright (C) 2012 mbed.org, MIT License
AdamGreen 0:3b00827bb0b7 2 *
AdamGreen 0:3b00827bb0b7 3 * Permission is hereby granted, free of charge, to any person obtaining a copy of this software
AdamGreen 0:3b00827bb0b7 4 * and associated documentation files (the "Software"), to deal in the Software without restriction,
AdamGreen 0:3b00827bb0b7 5 * including without limitation the rights to use, copy, modify, merge, publish, distribute,
AdamGreen 0:3b00827bb0b7 6 * sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is
AdamGreen 0:3b00827bb0b7 7 * furnished to do so, subject to the following conditions:
AdamGreen 0:3b00827bb0b7 8 *
AdamGreen 0:3b00827bb0b7 9 * The above copyright notice and this permission notice shall be included in all copies or
AdamGreen 0:3b00827bb0b7 10 * substantial portions of the Software.
AdamGreen 0:3b00827bb0b7 11 *
AdamGreen 0:3b00827bb0b7 12 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING
AdamGreen 0:3b00827bb0b7 13 * BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
AdamGreen 0:3b00827bb0b7 14 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
AdamGreen 0:3b00827bb0b7 15 * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
AdamGreen 0:3b00827bb0b7 16 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
AdamGreen 0:3b00827bb0b7 17 */
AdamGreen 0:3b00827bb0b7 18 #include "TCPSocketConnection.h"
AdamGreen 0:3b00827bb0b7 19 #include <cstring>
AdamGreen 0:3b00827bb0b7 20
AdamGreen 0:3b00827bb0b7 21 using std::memset;
AdamGreen 0:3b00827bb0b7 22 using std::memcpy;
AdamGreen 0:3b00827bb0b7 23
AdamGreen 0:3b00827bb0b7 24 TCPSocketConnection::TCPSocketConnection() :
AdamGreen 0:3b00827bb0b7 25 _is_connected(false) {
AdamGreen 0:3b00827bb0b7 26 }
AdamGreen 0:3b00827bb0b7 27
AdamGreen 0:3b00827bb0b7 28 int TCPSocketConnection::connect(const char* host, const int port) {
AdamGreen 0:3b00827bb0b7 29 if (init_socket(SOCK_STREAM) < 0)
AdamGreen 0:3b00827bb0b7 30 return -1;
AdamGreen 0:3b00827bb0b7 31
AdamGreen 0:3b00827bb0b7 32 if (set_address(host, port) != 0)
AdamGreen 0:3b00827bb0b7 33 return -1;
AdamGreen 0:3b00827bb0b7 34
AdamGreen 0:3b00827bb0b7 35 if (lwip_connect(_sock_fd, (const struct sockaddr *) &_remoteHost, sizeof(_remoteHost)) < 0) {
AdamGreen 0:3b00827bb0b7 36 close();
AdamGreen 0:3b00827bb0b7 37 return -1;
AdamGreen 0:3b00827bb0b7 38 }
AdamGreen 0:3b00827bb0b7 39 _is_connected = true;
AdamGreen 0:3b00827bb0b7 40
AdamGreen 0:3b00827bb0b7 41 return 0;
AdamGreen 0:3b00827bb0b7 42 }
AdamGreen 0:3b00827bb0b7 43
AdamGreen 0:3b00827bb0b7 44 bool TCPSocketConnection::is_connected(void) {
AdamGreen 0:3b00827bb0b7 45 return _is_connected;
AdamGreen 0:3b00827bb0b7 46 }
AdamGreen 0:3b00827bb0b7 47
AdamGreen 0:3b00827bb0b7 48 int TCPSocketConnection::send(char* data, int length) {
AdamGreen 0:3b00827bb0b7 49 if ((_sock_fd < 0) || !_is_connected)
AdamGreen 0:3b00827bb0b7 50 return -1;
AdamGreen 0:3b00827bb0b7 51
AdamGreen 0:3b00827bb0b7 52 if (!_blocking) {
AdamGreen 0:3b00827bb0b7 53 TimeInterval timeout(_timeout);
AdamGreen 0:3b00827bb0b7 54 if (wait_writable(timeout) != 0)
AdamGreen 0:3b00827bb0b7 55 return -1;
AdamGreen 0:3b00827bb0b7 56 }
AdamGreen 0:3b00827bb0b7 57
AdamGreen 0:3b00827bb0b7 58 int n = lwip_send(_sock_fd, data, length, 0);
AdamGreen 0:3b00827bb0b7 59 _is_connected = (n != 0);
AdamGreen 0:3b00827bb0b7 60
AdamGreen 0:3b00827bb0b7 61 return n;
AdamGreen 0:3b00827bb0b7 62 }
AdamGreen 0:3b00827bb0b7 63
AdamGreen 0:3b00827bb0b7 64 // -1 if unsuccessful, else number of bytes written
AdamGreen 0:3b00827bb0b7 65 int TCPSocketConnection::send_all(char* data, int length) {
AdamGreen 0:3b00827bb0b7 66 if ((_sock_fd < 0) || !_is_connected)
AdamGreen 0:3b00827bb0b7 67 return -1;
AdamGreen 0:3b00827bb0b7 68
AdamGreen 0:3b00827bb0b7 69 int writtenLen = 0;
AdamGreen 0:3b00827bb0b7 70 TimeInterval timeout(_timeout);
AdamGreen 0:3b00827bb0b7 71 while (writtenLen < length) {
AdamGreen 0:3b00827bb0b7 72 if (!_blocking) {
AdamGreen 0:3b00827bb0b7 73 // Wait for socket to be writeable
AdamGreen 0:3b00827bb0b7 74 if (wait_writable(timeout) != 0)
AdamGreen 0:3b00827bb0b7 75 return writtenLen;
AdamGreen 0:3b00827bb0b7 76 }
AdamGreen 0:3b00827bb0b7 77
AdamGreen 0:3b00827bb0b7 78 int ret = lwip_send(_sock_fd, data + writtenLen, length - writtenLen, 0);
AdamGreen 0:3b00827bb0b7 79 if (ret > 0) {
AdamGreen 0:3b00827bb0b7 80 writtenLen += ret;
AdamGreen 0:3b00827bb0b7 81 continue;
AdamGreen 0:3b00827bb0b7 82 } else if (ret == 0) {
AdamGreen 0:3b00827bb0b7 83 _is_connected = false;
AdamGreen 0:3b00827bb0b7 84 return writtenLen;
AdamGreen 0:3b00827bb0b7 85 } else {
AdamGreen 0:3b00827bb0b7 86 return -1; //Connnection error
AdamGreen 0:3b00827bb0b7 87 }
AdamGreen 0:3b00827bb0b7 88 }
AdamGreen 0:3b00827bb0b7 89 return writtenLen;
AdamGreen 0:3b00827bb0b7 90 }
AdamGreen 0:3b00827bb0b7 91
AdamGreen 0:3b00827bb0b7 92 int TCPSocketConnection::receive(char* data, int length) {
AdamGreen 0:3b00827bb0b7 93 if ((_sock_fd < 0) || !_is_connected)
AdamGreen 0:3b00827bb0b7 94 return -1;
AdamGreen 0:3b00827bb0b7 95
AdamGreen 0:3b00827bb0b7 96 if (!_blocking) {
AdamGreen 0:3b00827bb0b7 97 TimeInterval timeout(_timeout);
AdamGreen 0:3b00827bb0b7 98 if (wait_readable(timeout) != 0)
AdamGreen 0:3b00827bb0b7 99 return -1;
AdamGreen 0:3b00827bb0b7 100 }
AdamGreen 0:3b00827bb0b7 101
AdamGreen 0:3b00827bb0b7 102 int n = lwip_recv(_sock_fd, data, length, 0);
AdamGreen 0:3b00827bb0b7 103 _is_connected = (n != 0);
AdamGreen 0:3b00827bb0b7 104
AdamGreen 0:3b00827bb0b7 105 return n;
AdamGreen 0:3b00827bb0b7 106 }
AdamGreen 0:3b00827bb0b7 107
AdamGreen 0:3b00827bb0b7 108 // -1 if unsuccessful, else number of bytes received
AdamGreen 0:3b00827bb0b7 109 int TCPSocketConnection::receive_all(char* data, int length) {
AdamGreen 0:3b00827bb0b7 110 if ((_sock_fd < 0) || !_is_connected)
AdamGreen 0:3b00827bb0b7 111 return -1;
AdamGreen 0:3b00827bb0b7 112
AdamGreen 0:3b00827bb0b7 113 int readLen = 0;
AdamGreen 0:3b00827bb0b7 114 TimeInterval timeout(_timeout);
AdamGreen 0:3b00827bb0b7 115 while (readLen < length) {
AdamGreen 0:3b00827bb0b7 116 if (!_blocking) {
AdamGreen 0:3b00827bb0b7 117 //Wait for socket to be readable
AdamGreen 0:3b00827bb0b7 118 if (wait_readable(timeout) != 0)
AdamGreen 0:3b00827bb0b7 119 return readLen;
AdamGreen 0:3b00827bb0b7 120 }
AdamGreen 0:3b00827bb0b7 121
AdamGreen 0:3b00827bb0b7 122 int ret = lwip_recv(_sock_fd, data + readLen, length - readLen, 0);
AdamGreen 0:3b00827bb0b7 123 if (ret > 0) {
AdamGreen 0:3b00827bb0b7 124 readLen += ret;
AdamGreen 0:3b00827bb0b7 125 } else if (ret == 0) {
AdamGreen 0:3b00827bb0b7 126 _is_connected = false;
AdamGreen 0:3b00827bb0b7 127 return readLen;
AdamGreen 0:3b00827bb0b7 128 } else {
AdamGreen 0:3b00827bb0b7 129 return -1; //Connnection error
AdamGreen 0:3b00827bb0b7 130 }
AdamGreen 0:3b00827bb0b7 131 }
AdamGreen 0:3b00827bb0b7 132 return readLen;
AdamGreen 0:3b00827bb0b7 133 }