Check ping/pong too

2016-04-14 20:02:48 +00:00
parent 23a2cc1e24
commit 60fed031a8
1 changed files with 8 additions and 0 deletions
--- a/mud.c
+++ b/mud.c
@@ -848,6 +848,14 @@ int mud_pull (struct mud *mud)
        if (mud_dt(now, send_time) >= mud->time_tolerance)
            continue;

+        if (pong_packet || (ret == (ssize_t)MUD_PACKET_MIN_SIZE)) {
+            unsigned char tmp[sizeof(packet->data)];
+
+            if (mud_decrypt(mud, NULL, tmp, sizeof(tmp),
+                            packet->data, (size_t)ret, (size_t)ret) == -1)
+                continue;
+        }
+
        mud_unmapv4((struct sockaddr *)&addr);

        struct cmsghdr *cmsg = mud_get_pktinfo(&msg, addr.ss_family);