Adjust SIP plugin

Author: Zadamsa

process/sip.cpp

@@ -69,33 +69,33 @@ ProcessPlugin *SIPPlugin::copy()
    return new SIPPlugin(*this);
 }
 
-int SIPPlugin::post_create(Flow &rec, const Packet &pkt)
+ProcessPlugin::FlowAction SIPPlugin::post_create(Flow &rec, const Packet &pkt)
 {
    uint16_t msg_type;
 
    msg_type = parse_msg_type(pkt);
    if (msg_type == SIP_MSG_TYPE_INVALID) {
-      return 0;
+      return ProcessPlugin::FlowAction::NO_PROCESS;
    }
 
    RecordExtSIP *sip_data = new RecordExtSIP();
    sip_data->msg_type = msg_type;
    rec.add_extension(sip_data);
    parser_process_sip(pkt, sip_data);
 
-   return 0;
+   return ProcessPlugin::FlowAction::GET_ALL_DATA;
 }
 
-int SIPPlugin::pre_update(Flow &rec, Packet &pkt)
+ProcessPlugin::FlowAction SIPPlugin::pre_update(Flow &rec, Packet &pkt)
 {
    uint16_t msg_type;
 
    msg_type = parse_msg_type(pkt);
    if (msg_type != SIP_MSG_TYPE_INVALID) {
-      return FLOW_FLUSH_WITH_REINSERT;
+      return ProcessPlugin::FlowAction::FLUSH_WITH_REINSERT;
    }
 
-   return 0;
+   return ProcessPlugin::FlowAction::GET_ALL_DATA;
 }
 
 void SIPPlugin::finish(bool print_stats)

process/sip.hpp

@@ -496,8 +496,8 @@ class SIPPlugin : public ProcessPlugin {
    std::string get_name() const { return "sip"; }
    RecordExt *get_ext() const { return new RecordExtSIP(); }
    ProcessPlugin *copy();
-   int post_create(Flow &rec, const Packet &pkt);
-   int pre_update(Flow &rec, Packet &pkt);
+   ProcessPlugin::FlowAction post_create(Flow &rec, const Packet &pkt);
+   ProcessPlugin::FlowAction pre_update(Flow &rec, Packet &pkt);
    void finish(bool print_stats);
 
 private: