/*

 *

 * (C) 2013-24 - ntop.org

 *

 *

 * This program is free software; you can redistribute it and/or modify

 * it under the terms of the GNU General Public License as published by

 * the Free Software Foundation; either version 3 of the License, or

 * (at your option) any later version.

 *

 * This program 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 General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software Foundation,

 * Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.

 *

 */

#ifndef _ELEPHANT_FLOW_ALERT_H_

#define _ELEPHANT_FLOW_ALERT_H_

#include "ntop_includes.h"

class ElephantFlowAlert : public FlowAlert {

 private:

  u_int64_t l2r_th, r2l_th;

  ndpi_serializer *getAlertJSON(ndpi_serializer *serializer);

 public:

  static FlowAlertType getClassType() {

    return {NDPI_NO_RISK, flow_alert_elephant_flow, alert_category_security};

  }

  static u_int8_t getDefaultScore() { return SCORE_LEVEL_WARNING; };

  inline u_int64_t get_l2r_threshold() { return l2r_th; };

  inline u_int64_t get_r2l_threshold() { return r2l_th; };

  ElephantFlowAlert(FlowCheck *c, Flow *f, u_int64_t _l2r_th, u_int64_t _r2l_th)

      : FlowAlert(c, f) {

    l2r_th = _l2r_th;

    r2l_th = _r2l_th;

    setAlertScore(getDefaultScore());

  };

  ~ElephantFlowAlert(){};

  FlowAlertType getAlertType() const { return getClassType(); }

};

#endif /* _ELEPHANT_FLOW_ALERT_H_ */