From 7e28cc0981dfcbbea9a08fd622a408f16184347a Mon Sep 17 00:00:00 2001
From: Luca <deri@ntop.org>
Date: Thu, 25 Jun 2015 03:57:50 -0700
Subject: Added MPEG TS protocol Fixed possible decoding loop in RTCP dissector
 Added test pcap for mpegts

---
 src/lib/protocols/mpegts.c | 53 ++++++++++++++++++++++++++++++++++++++++++++++
 src/lib/protocols/quic.c   |  3 ++-
 src/lib/protocols/rtcp.c   |  2 +-
 3 files changed, 56 insertions(+), 2 deletions(-)
 create mode 100644 src/lib/protocols/mpegts.c

(limited to 'src/lib/protocols')

diff --git a/src/lib/protocols/mpegts.c b/src/lib/protocols/mpegts.c
new file mode 100644
index 000000000..f0e81ac5b
--- /dev/null
+++ b/src/lib/protocols/mpegts.c
@@ -0,0 +1,53 @@
+/*
+ * mpegts.c (MPEG Transport Stream)
+ *          https://en.wikipedia.org/wiki/MPEG_transport_stream
+ *
+ * Copyright (C) 2015 - ntop.org
+ *
+ * nDPI is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Lesser General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * nDPI is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with nDPI.  If not, see <http://www.gnu.org/licenses/>.
+ *
+ */
+
+
+#include "ndpi_api.h"
+
+#ifdef NDPI_PROTOCOL_MPEGTS
+
+void ndpi_search_mpegts(struct ndpi_detection_module_struct *ndpi_struct, struct ndpi_flow_struct *flow)
+{
+  struct ndpi_packet_struct *packet = &flow->packet;
+  u_int16_t dport = 0, sport = 0;
+
+  NDPI_LOG(NDPI_PROTOCOL_MPEGTS, ndpi_struct, NDPI_LOG_DEBUG, "search for MPEGTS.\n");
+
+  if((packet->udp != NULL) && ((packet->payload_packet_len % 188) == 0)) {
+    u_int i, num_chunks = packet->payload_packet_len / 188;
+    u_int32_t pkt_id;
+    
+    for(i=0; i<num_chunks; i++) {
+      u_int offset = 188 * i;
+
+      if(packet->payload[offset] != 0x47) goto no_mpegts;
+    }
+
+    /* This looks MPEG TS */
+    ndpi_int_add_connection(ndpi_struct, flow, NDPI_PROTOCOL_MPEGTS, NDPI_REAL_PROTOCOL);
+    return;
+  }    
+
+ no_mpegts:
+  NDPI_LOG(NDPI_PROTOCOL_MPEGTS, ndpi_struct, NDPI_LOG_DEBUG, "Excluded MPEGTS.\n");
+  NDPI_ADD_PROTOCOL_TO_BITMASK(flow->excluded_protocol_bitmask, NDPI_PROTOCOL_MPEGTS);
+}
+#endif
diff --git a/src/lib/protocols/quic.c b/src/lib/protocols/quic.c
index 57cfa0cc0..085ce36b6 100644
--- a/src/lib/protocols/quic.c
+++ b/src/lib/protocols/quic.c
@@ -2,7 +2,8 @@
  * quic.c
  *
  * Andrea Buscarinu - <andrea.buscarinu@gmail.com>
- * Michele Campus - <michelecampus5@gmail.com>
+ * Michele Campus - <campus@ntop.org>
+ *
  * Copyright (C) 2012-15 - ntop.org
  *
  * This module is free software: you can redistribute it and/or modify
diff --git a/src/lib/protocols/rtcp.c b/src/lib/protocols/rtcp.c
index f618261dd..0c84c085f 100644
--- a/src/lib/protocols/rtcp.c
+++ b/src/lib/protocols/rtcp.c
@@ -42,7 +42,7 @@ void ndpi_search_rtcp(struct ndpi_detection_module_struct *ndpi_struct, struct n
       len = packet->payload[2+offset] * 256 + packet->payload[2+offset+1];
       rtcp_section_len = (len + 1) * 4;
       
-      if((offset+rtcp_section_len) > packet->payload_packet_len)
+      if(((offset+rtcp_section_len) > packet->payload_packet_len) || (rtcp_section_len == 0))
 	goto exclude_rtcp;
       else
 	offset += rtcp_section_len;
-- 
cgit v1.2.3