From 1d64e99f599561264f9a4a6d59fe850c62481319 Mon Sep 17 00:00:00 2001
From: ntruchsess <norbert.truchsess@t-online.de>
Date: Wed, 27 Nov 2013 23:17:01 +0100
Subject: [PATCH] add operator==, localPort, remotePort and remoteIP to
 UIPClient, new example AdvancedChatServer.ino

---
 UIPClient.cpp                                 |  24 ++++
 UIPClient.h                                   |   4 +
 .../AdvancedChatServer/AdvancedChatServer.ino | 110 ++++++++++++++++++
 3 files changed, 138 insertions(+)
 create mode 100644 examples/AdvancedChatServer/AdvancedChatServer.ino

diff --git a/UIPClient.cpp b/UIPClient.cpp
index 4919303..665fdca 100644
--- a/UIPClient.cpp
+++ b/UIPClient.cpp
@@ -135,6 +135,30 @@ UIPClient::connected()
   return ((_uip_conn && (_uip_conn->tcpstateflags & UIP_TS_MASK) == UIP_ESTABLISHED) || available() > 0) ? 1 : 0;
 }
 
+bool
+UIPClient::operator==(const UIPClient& rhs) {
+  return _uip_conn && rhs._uip_conn && _uip_conn == rhs._uip_conn;
+}
+
+uint16_t
+UIPClient::localPort() {
+  if (!_uip_conn) return 0;
+  return htons(_uip_conn->lport);
+}
+
+IPAddress
+UIPClient::remoteIP() {
+  if (!_uip_conn) return IPAddress();
+  return ip_addr_uip(_uip_conn->ripaddr);
+}
+
+uint16_t
+UIPClient::remotePort() {
+  if (!_uip_conn) return 0;
+  return htons(_uip_conn->rport);
+}
+
+
 UIPClient::operator bool()
 {
   UIPEthernet.tick();
diff --git a/UIPClient.h b/UIPClient.h
index cae8f97..f2f0a8b 100644
--- a/UIPClient.h
+++ b/UIPClient.h
@@ -63,6 +63,10 @@ class UIPClient : public Client {
   void stop();
   uint8_t connected();
   operator bool();
+  virtual bool operator==(const EthernetClient&);
+  virtual uint16_t localPort();
+  virtual IPAddress remoteIP();
+  virtual uint16_t remotePort();
 
   size_t write(uint8_t);
   size_t write(const uint8_t *buf, size_t size);
diff --git a/examples/AdvancedChatServer/AdvancedChatServer.ino b/examples/AdvancedChatServer/AdvancedChatServer.ino
new file mode 100644
index 0000000..5a77c90
--- /dev/null
+++ b/examples/AdvancedChatServer/AdvancedChatServer.ino
@@ -0,0 +1,110 @@
+/*
+ Advanced Chat Server
+
+ A simple server that distributes any incoming messages to all
+ connected clients but the client the message comes from.
+ To use telnet to  your device's IP address and type.
+ You can see the client's input in the serial monitor as well.
+ Using an Arduino Wiznet Ethernet shield.
+
+ Circuit:
+ * Ethernet shield attached to pins 10, 11, 12, 13
+ * Analog inputs attached to pins A0 through A5 (optional)
+
+ created 18 Dec 2009
+ by David A. Mellis
+ modified 9 Apr 2012
+ by Tom Igoe
+ redesigned to make use of operator== 25 Nov 2013
+ by Norbert Truchsess
+
+ */
+
+#include <UIPEthernet.h>
+
+// Enter a MAC address and IP address for your controller below.
+// The IP address will be dependent on your local network.
+
+byte mac[] = { 0xDE, 0xAD, 0xBE, 0xEF, 0xFE, 0xED };
+IPAddress ip(192,168,0,6);
+
+// telnet defaults to port 23
+EthernetServer server(23);
+
+EthernetClient clients[4];
+
+void setup() {
+  // initialize the ethernet device
+  Ethernet.begin(mac, ip);
+  // start listening for clients
+  server.begin();
+ // Open serial communications and wait for port to open:
+  Serial.begin(9600);
+   while (!Serial) {
+    ; // wait for serial port to connect. Needed for Leonardo only
+  }
+
+
+  Serial.print("Chat server address:");
+  Serial.println(Ethernet.localIP());
+}
+
+void loop() {
+  // wait for a new client:
+  EthernetClient client = server.available();
+
+  if (client) {
+
+    boolean newClient = true;
+    for (byte i=0;i<4;i++) {
+      //check whether this client refers to the same socket as one of the existing instances:
+      if (clients[i]==client) {
+        newClient = false;
+        break;
+      }
+    }
+
+    if (newClient) {
+      //check which of the existing clients can be overridden:
+      for (byte i=0;i<4;i++) {
+        if (!clients[i] && clients[i]!=client) {
+          clients[i] = client;
+          break;
+        }
+      }
+
+      // clead out the input buffer:
+      client.flush();
+      Serial.println("We have a new client");
+      client.println("Hello, client!");
+      client.print("my IP: ");
+      client.println(Ethernet.localIP());
+      client.print("my port: ");
+      client.println(client.localPort());
+      client.print("your IP: ");
+      client.println(client.remoteIP());
+      client.print("your port: ");
+      client.println(client.remotePort());
+    }
+
+    if (client.available() > 0) {
+      // read the bytes incoming from the client:
+      char thisChar = client.read();
+      // echo the bytes back to all other connected clients:
+      for (byte i=0;i<4;i++) {
+        if (!clients[i] || (clients[i]==client)) {
+          continue;
+        }
+        clients[i].write(thisChar);
+      }
+      // echo the bytes to the server as well:
+      Serial.write(thisChar);
+    }
+  }
+  for (byte i=0;i<4;i++) {
+    if (!(clients[i].connected())) {
+      // client.stop() invalidates the internal socket-descriptor, so next use of == will allways return false;
+      clients[i].stop();
+    }
+  }
+}