Skip to content

Commit f45a7e7

Browse files
committed
Adjust SSDP plugin
1 parent 8fbe1a1 commit f45a7e7

File tree

2 files changed

+8
-6
lines changed

2 files changed

+8
-6
lines changed

process/ssdp.cpp

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -90,24 +90,26 @@ ProcessPlugin *SSDPPlugin::copy()
9090
return new SSDPPlugin(*this);
9191
}
9292

93-
int SSDPPlugin::post_create(Flow &rec, const Packet &pkt)
93+
ProcessPlugin::FlowAction SSDPPlugin::post_create(Flow &rec, const Packet &pkt)
9494
{
9595
if (pkt.dst_port == 1900) {
9696
record = new RecordExtSSDP();
9797
rec.add_extension(record);
9898
record = nullptr;
9999

100100
parse_ssdp_message(rec, pkt);
101+
return ProcessPlugin::FlowAction::GET_ALL_DATA;
101102
}
102-
return 0;
103+
return ProcessPlugin::FlowAction::NO_PROCESS;
103104
}
104105

105-
int SSDPPlugin::pre_update(Flow &rec, Packet &pkt)
106+
ProcessPlugin::FlowAction SSDPPlugin::pre_update(Flow &rec, Packet &pkt)
106107
{
107108
if (pkt.dst_port == 1900) {
108109
parse_ssdp_message(rec, pkt);
110+
return ProcessPlugin::FlowAction::GET_ALL_DATA;
109111
}
110-
return 0;
112+
return ProcessPlugin::FlowAction::NO_PROCESS;
111113
}
112114

113115
void SSDPPlugin::finish(bool print_stats)

process/ssdp.hpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -181,8 +181,8 @@ class SSDPPlugin : public ProcessPlugin
181181
RecordExt *get_ext() const { return new RecordExtSSDP(); }
182182
ProcessPlugin *copy();
183183

184-
int post_create(Flow &rec, const Packet &pkt);
185-
int pre_update(Flow &rec, Packet &pkt);
184+
ProcessPlugin::FlowAction post_create(Flow &rec, const Packet &pkt);
185+
ProcessPlugin::FlowAction pre_update(Flow &rec, Packet &pkt);
186186
void finish(bool print_stats);
187187

188188
/**

0 commit comments

Comments
 (0)