drop UDP message if frag != 0

2015-06-10 14:21:22 +08:00
parent 7c2fe9fd56
commit 1e71fba852
1 changed files with 6 additions and 2 deletions
@@ -194,8 +194,12 @@ class TCPRelayHandler(object):
        #logging.debug("_write_to_sock %s %s %s" % (self._remote_sock, sock, self._remote_udp))
        if self._remote_sock == sock and self._remote_udp:
            try:
-                #TODO
-                data = data[3:]
+                frag = common.ord(data[2])
+                if frag != 0:
+                    logging.warn('drop a message since frag is %d' % (frag,))
+                    return False
+                else:
+                    data = data[3:]
                header_result = parse_header(data)
                if header_result is None:
                    return False