From c3a5650071b25a2f1eaf8606cd75d3eee28dc503 Mon Sep 17 00:00:00 2001 From: Ravi Kerur Date: Wed, 25 Sep 2019 15:20:19 -0700 Subject: Initial PS_VUE support based on public IP. Signed-off-by: Ravi Kerur --- src/lib/protocols/ps_vue.c | 90 ++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 90 insertions(+) create mode 100644 src/lib/protocols/ps_vue.c (limited to 'src/lib/protocols/ps_vue.c') diff --git a/src/lib/protocols/ps_vue.c b/src/lib/protocols/ps_vue.c new file mode 100644 index 000000000..ee91b5f84 --- /dev/null +++ b/src/lib/protocols/ps_vue.c @@ -0,0 +1,90 @@ +/* + * ps_vue.c + * + * Copyright (C) 2018 by ntop.org + * + * This file is part of nDPI, an open source deep packet inspection + * library based on the OpenDPI and PACE technology by ipoque GmbH + * + * 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 . + * + */ + +#include "ndpi_protocol_ids.h" + +#define NDPI_CURRENT_PROTO NDPI_PROTOCOL_PS_VUE + +#include "ndpi_api.h" + +static u_int8_t ps_vue_ptree_match(struct ndpi_detection_module_struct *ndpi_struct, struct in_addr *pin) { + return((ndpi_network_ptree_match(ndpi_struct, pin) == NDPI_PROTOCOL_PS_VUE) ? 1 : 0); +} + +/* ******************************************* */ + +static u_int8_t is_ps_vue_flow(struct ndpi_detection_module_struct *ndpi_struct, + struct ndpi_flow_struct *flow) { + struct ndpi_packet_struct *packet = &flow->packet; + + if(packet->iph) { + struct in_addr daddr; + + daddr.s_addr = packet->iph->daddr; + + if(ps_vue_ptree_match(ndpi_struct, &daddr)) { + return(1); + } + } + + return(0); +} + +static void ndpi_check_ps_vue(struct ndpi_detection_module_struct *ndpi_struct, + struct ndpi_flow_struct *flow) { + + NDPI_LOG_DBG(ndpi_struct, "search ps_vue video \n"); + + if (is_ps_vue_flow(ndpi_struct, flow)){ + NDPI_LOG_INFO(ndpi_struct, "found ps_vue\n"); + + ndpi_set_detected_protocol(ndpi_struct, flow, NDPI_PROTOCOL_PS_VUE, + NDPI_PROTOCOL_UNKNOWN); + return; + } else { + NDPI_EXCLUDE_PROTO(ndpi_struct, flow); + } +} + +void ndpi_search_ps_vue(struct ndpi_detection_module_struct *ndpi_struct, struct ndpi_flow_struct *flow) +{ + struct ndpi_packet_struct *packet = &flow->packet; + + NDPI_LOG_DBG(ndpi_struct, "search ps_vue\n"); + + /* skip marked packets */ + if(packet->detected_protocol_stack[0] != NDPI_PROTOCOL_PS_VUE) + ndpi_check_ps_vue(ndpi_struct, flow); +} + + +void init_ps_vue_dissector(struct ndpi_detection_module_struct *ndpi_struct, u_int32_t *id, NDPI_PROTOCOL_BITMASK *detection_bitmask) +{ + ndpi_set_bitmask_protocol_detection("PS_Vue", ndpi_struct, detection_bitmask, *id, + NDPI_PROTOCOL_PS_VUE, + ndpi_search_ps_vue, + NDPI_SELECTION_BITMASK_PROTOCOL_TCP_OR_UDP_WITH_PAYLOAD, + SAVE_DETECTION_BITMASK_AS_UNKNOWN, + ADD_TO_DETECTION_BITMASK); + *id += 1; +} -- cgit v1.2.3