Skip to content

Commit 8fbe1a1

Browse files
committed
Adjust SSADetector plugin
1 parent 006f091 commit 8fbe1a1

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

process/ssadetector.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -119,11 +119,11 @@ void SSADetectorPlugin::update_record(RecordExtSSADetector* record, const Packet
119119
}
120120

121121

122-
int SSADetectorPlugin::post_update(Flow& rec, const Packet& pkt)
122+
ProcessPlugin::FlowAction SSADetectorPlugin::post_update(Flow& rec, const Packet& pkt)
123123
{
124124
RecordExtSSADetector *record = nullptr;
125125
if (rec.src_packets + rec.dst_packets < MIN_PKT_IN_FLOW) {
126-
return 0;
126+
return ProcessPlugin::FlowAction::GET_ALL_DATA;
127127
}
128128

129129
record = (RecordExtSSADetector *) rec.get_extension(RecordExtSSADetector::REGISTERED_ID);
@@ -133,7 +133,7 @@ int SSADetectorPlugin::post_update(Flow& rec, const Packet& pkt)
133133
}
134134

135135
update_record(record, pkt);
136-
return 0;
136+
return ProcessPlugin::FlowAction::GET_ALL_DATA;
137137
}
138138

139139
double classes_ratio(uint8_t* syn_pkts, uint8_t size)

process/ssadetector.hpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -174,7 +174,7 @@ class SSADetectorPlugin : public ProcessPlugin {
174174
RecordExt* get_ext() const { return new RecordExtSSADetector(); }
175175
ProcessPlugin* copy();
176176

177-
int post_update(Flow& rec, const Packet& pkt);
177+
ProcessPlugin::FlowAction post_update(Flow& rec, const Packet& pkt);
178178
void pre_export(Flow& rec);
179179
void update_record(RecordExtSSADetector* record, const Packet& pkt);
180180
static inline void

0 commit comments

Comments
 (0)