Add UdpContext seek and tell

Author: jonatanolofsson

libraries/ESP8266WiFi/src/include/UdpContext.h

@@ -27,6 +27,7 @@ extern "C" {
 void esp_yield();
 void esp_schedule();
 #include "lwip/init.h" // LWIP_VERSION_
+#include <assert.h>
 }
 
 
@@ -143,6 +144,17 @@ class UdpContext
         return _rx_buf->len - _rx_buf_offset;
     }
 
+    size_t tell() const
+    {
+        return _rx_buf_offset;
+    }
+
+    void seek(const size_t pos)
+    {
+        assert(pos <= _rx_buf->len);
+        _rx_buf_offset = pos;
+    }
+
     uint32_t getRemoteAddress()
     {
         if (!_rx_buf)
@@ -203,7 +215,7 @@ class UdpContext
 
     int read()
     {
-        if (!_rx_buf || _rx_buf_offset == _rx_buf->len)
+        if (!_rx_buf || _rx_buf_offset >= _rx_buf->len)
             return -1;
 
         char c = reinterpret_cast<char*>(_rx_buf->payload)[_rx_buf_offset];
@@ -361,6 +373,9 @@ class UdpContext
     void _consume(size_t size)
     {
         _rx_buf_offset += size;
+        if (_rx_buf_offset > _rx_buf->len) {
+            _rx_buf_offset = _rx_buf->len;
+        }
     }
 
     void _recv(udp_pcb *upcb, pbuf *pb,