Marcelo Rebonatto / Socket

Modify changes to test TCP socket.

Dependents:   EthernetInterface

Fork of Socket by mbed official

TCPSocketConnection.cpp@20:b06c58defd65, 2013-12-13 (annotated)

Committer:
rebonatto
Date:
Fri Dec 13 11:36:01 2013 +0000
Revision:
20:b06c58defd65
Parent:
19:434906b5b977 
Modify

Who changed what in which revision?

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