mbed official / WiflyInterface

WiflyInterface

Dependents:   test_over_wifly

Fork of WiflyInterface by mbed official

Socket/TCPSocketConnection.cpp@9:c77799a03294, 2014-01-28 (annotated)

Committer:
Bogdan Marinescu
Date:
Tue Jan 28 11:00:48 2014 +0200
Revision:
9:c77799a03294
Parent:
1:fb4494783863 
Address various potential buffer overflow issues



https://mbed.org/comments/cr/93/4779/#c8499

Who changed what in which revision?

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