bendns' repos / ArduinoStuff
arduino stuffs
Diffstat (limited to 'libraries/Firmata/utility/WiFiServerStream.h')
-rw-r--r--libraries/Firmata/utility/WiFiServerStream.h107
1 files changed, 107 insertions, 0 deletions
diff --git a/libraries/Firmata/utility/WiFiServerStream.h b/libraries/Firmata/utility/WiFiServerStream.h
new file mode 100644
index 0000000..1404b05
--- /dev/null
+++ b/libraries/Firmata/utility/WiFiServerStream.h
@@ -0,0 +1,107 @@
+/*
+ WiFiServerStream.h
+
+ An Arduino Stream extension for a WiFiClient or WiFiServer to be used
+ with legacy Arduino WiFi shield and other boards and shields that
+ are compatible with the Arduino WiFi library.
+
+ Copyright (C) 2016 Jens B. All rights 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.
+
+ See file LICENSE.txt for further informations on licensing terms.
+
+ Parts of this class are based on
+
+ - WiFiStream - Copyright (C) 2015-2016 Jesse Frush. All rights reserved.
+
+ published under the same license.
+
+ Last updated April 23rd, 2016
+ */
+
+#ifndef WIFI_SERVER_STREAM_H
+#define WIFI_SERVER_STREAM_H
+
+#include "WiFiStream.h"
+
+class WiFiServerStream : public WiFiStream
+{
+protected:
+ WiFiServer _server = WiFiServer(3030);
+ bool _listening = false;
+
+ /**
+ * check if TCP client is connected
+ * @return true if connected
+ */
+ virtual inline bool connect_client()
+ {
+ if ( _connected )
+ {
+ if ( _client && _client.connected() ) return true;
+ stop();
+ }
+
+ // passive TCP connect (accept)
+ WiFiClient newClient = _server.available();
+ if ( !newClient ) return false;
+ _client = newClient;
+ _connected = true;
+ if ( _currentHostConnectionCallback )
+ {
+ (*_currentHostConnectionCallback)(HOST_CONNECTION_CONNECTED);
+ }
+
+ return true;
+ }
+
+public:
+ /**
+ * create a WiFi stream with a TCP server
+ */
+ WiFiServerStream(uint16_t server_port) : WiFiStream(server_port) {}
+
+ /**
+ * maintain WiFi and TCP connection
+ * @return true if WiFi and TCP connection are established
+ */
+ virtual inline bool maintain()
+ {
+ if ( connect_client() ) return true;
+
+ stop();
+
+ if ( !_listening && WiFi.status() == WL_CONNECTED )
+ {
+ // start TCP server after first WiFi connect
+ _server = WiFiServer(_port);
+ _server.begin();
+ _listening = true;
+ }
+
+ return false;
+ }
+
+ /**
+ * stop client connection
+ */
+ virtual inline void stop()
+ {
+ if ( _client)
+ {
+ _client.stop();
+ if ( _currentHostConnectionCallback )
+ {
+ (*_currentHostConnectionCallback)(HOST_CONNECTION_DISCONNECTED);
+ }
+ }
+ _connected = false;
+ }
+
+};
+
+#endif //WIFI_SERVER_STREAM_H