mirror of
https://github.com/esp8266/Arduino.git
synced 2025-04-25 20:02:37 +03:00
130 lines
3.3 KiB
C++
130 lines
3.3 KiB
C++
/*
|
|
WiFiClient.h - Library for Arduino Wifi shield.
|
|
Copyright (c) 2011-2014 Arduino. All right reserved.
|
|
|
|
This library is free software; you can redistribute it and/or
|
|
modify it under the terms of the GNU Lesser General Public
|
|
License as published by the Free Software Foundation; either
|
|
version 2.1 of the License, or (at your option) any later version.
|
|
|
|
This library is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
Lesser General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Lesser General Public
|
|
License along with this library; if not, write to the Free Software
|
|
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
|
|
|
|
Modified by Ivan Grokhotkov, December 2014 - esp8266 support
|
|
*/
|
|
|
|
#ifndef wificlient_h
|
|
#define wificlient_h
|
|
#include "Arduino.h"
|
|
#include "Print.h"
|
|
#include "Client.h"
|
|
#include "IPAddress.h"
|
|
#include <memory>
|
|
#include "include/slist.h"
|
|
|
|
class ClientContext;
|
|
class WiFiServer;
|
|
|
|
class WiFiClient : public Client, public SList<WiFiClient> {
|
|
protected:
|
|
WiFiClient(ClientContext* client);
|
|
|
|
public:
|
|
WiFiClient();
|
|
virtual ~WiFiClient();
|
|
WiFiClient(const WiFiClient&);
|
|
WiFiClient& operator=(const WiFiClient&);
|
|
|
|
uint8_t status();
|
|
virtual int connect(IPAddress ip, uint16_t port);
|
|
virtual int connect(const char *host, uint16_t port);
|
|
virtual size_t write(uint8_t);
|
|
virtual size_t write(const uint8_t *buf, size_t size);
|
|
template <typename T>
|
|
size_t write(T& source, size_t unitSize);
|
|
|
|
virtual int available();
|
|
virtual int read();
|
|
virtual int read(uint8_t *buf, size_t size);
|
|
virtual int peek();
|
|
virtual void flush();
|
|
virtual void stop();
|
|
virtual uint8_t connected();
|
|
virtual operator bool();
|
|
|
|
IPAddress remoteIP();
|
|
uint16_t remotePort();
|
|
IPAddress localIP();
|
|
uint16_t localPort();
|
|
bool getNoDelay();
|
|
void setNoDelay(bool nodelay);
|
|
static void setLocalPortStart(uint16_t port) { _localPort = port; }
|
|
|
|
template<typename T> size_t write(T &src){
|
|
uint8_t obuf[1460];
|
|
size_t doneLen = 0;
|
|
size_t sentLen;
|
|
int i;
|
|
|
|
while (src.available() > 1460){
|
|
src.read(obuf, 1460);
|
|
sentLen = write(obuf, 1460);
|
|
doneLen = doneLen + sentLen;
|
|
if(sentLen != 1460){
|
|
return doneLen;
|
|
}
|
|
}
|
|
|
|
uint16_t leftLen = src.available();
|
|
src.read(obuf, leftLen);
|
|
sentLen = write(obuf, leftLen);
|
|
doneLen = doneLen + sentLen;
|
|
return doneLen;
|
|
}
|
|
|
|
friend class WiFiServer;
|
|
|
|
using Print::write;
|
|
|
|
static void stopAll();
|
|
|
|
private:
|
|
|
|
static int8_t _s_connected(void* arg, void* tpcb, int8_t err);
|
|
static void _s_err(void* arg, int8_t err);
|
|
|
|
int8_t _connected(void* tpcb, int8_t err);
|
|
void _err(int8_t err);
|
|
|
|
ClientContext* _client;
|
|
static uint16_t _localPort;
|
|
};
|
|
|
|
|
|
template <typename T>
|
|
inline size_t WiFiClient::write(T& source, size_t unitSize) {
|
|
std::unique_ptr<uint8_t[]> buffer(new uint8_t[unitSize]);
|
|
size_t size_sent = 0;
|
|
while(true) {
|
|
size_t left = source.available();
|
|
if (!left)
|
|
break;
|
|
size_t will_send = (left < unitSize) ? left : unitSize;
|
|
source.read(buffer.get(), will_send);
|
|
size_t cb = write(buffer.get(), will_send);
|
|
size_sent += cb;
|
|
if (cb != will_send) {
|
|
break;
|
|
}
|
|
}
|
|
return size_sent;
|
|
}
|
|
|
|
#endif
|