aboutsummaryrefslogtreecommitdiff
path: root/src/lib/protocols/ps_vue.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/lib/protocols/ps_vue.c')
-rw-r--r--src/lib/protocols/ps_vue.c8
1 files changed, 5 insertions, 3 deletions
diff --git a/src/lib/protocols/ps_vue.c b/src/lib/protocols/ps_vue.c
index ee91b5f84..a608c96c6 100644
--- a/src/lib/protocols/ps_vue.c
+++ b/src/lib/protocols/ps_vue.c
@@ -38,11 +38,13 @@ static u_int8_t is_ps_vue_flow(struct ndpi_detection_module_struct *ndpi_struct,
struct ndpi_packet_struct *packet = &flow->packet;
if(packet->iph) {
- struct in_addr daddr;
+ struct in_addr saddr, daddr;
+ saddr.s_addr = packet->iph->saddr;
daddr.s_addr = packet->iph->daddr;
- if(ps_vue_ptree_match(ndpi_struct, &daddr)) {
+ if(ps_vue_ptree_match(ndpi_struct, &saddr) ||
+ ps_vue_ptree_match(ndpi_struct, &daddr)) {
return(1);
}
}
@@ -80,7 +82,7 @@ void ndpi_search_ps_vue(struct ndpi_detection_module_struct *ndpi_struct, struct
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_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,