fixed some potentially nasty memory issues

[meshlink] / src / sptps.c

diff --git a/src/sptps.c b/src/sptps.c
index 917b1e209c45cb641e8aa72ef4176b675497199c..49e0a336de6a66436fb245507bb16d0ea869ac94 100644 (file)
--- a/src/sptps.c
+++ b/src/sptps.c
@@ -371,8 +371,11 @@ static bool receive_handshake(sptps_t *s, const char *data, uint16_t len) {
 
 // Check datagram for valid HMAC
 bool sptps_verify_datagram(sptps_t *s, const void *data, size_t len) {
-       if(!s->instate || len < 21)
-               return error(s, EIO, "Received short packet");
+       if (!s->instate)
+               return error(s, EIO, "SPTPS state not ready to verify this datagram");
+
+       if(len < 21)
+               return error(s, EIO, "Received short packet in sptps_verify_datagram");
 
        uint32_t seqno;
        memcpy(&seqno, data, 4);
@@ -389,7 +392,7 @@ static bool sptps_receive_data_datagram(sptps_t *s, const void *vdata, size_t le
        const char *data = vdata;
 
        if(len < (s->instate ? 21 : 5))
-               return error(s, EIO, "Received short packet");
+               return error(s, EIO, "Received short packet in sptps_receive_data_datagram");
 
        uint32_t seqno;
        memcpy(&seqno, data, 4);