/*
 Itay Marom
 Hanoch Haim
 Cisco Systems, Inc.
*/

/*
Copyright (c) 2015-2016 Cisco Systems, Inc.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
#include <string.h>

#include "trex_stateless_messaging.h"
#include "trex_stateless_dp_core.h"
#include "trex_stateless_rx_core.h"
#include "trex_streams_compiler.h"
#include "trex_stateless.h"
#include "bp_sim.h"

/*************************
  start traffic message
 ************************/
TrexStatelessDpStart::TrexStatelessDpStart(uint8_t port_id, int event_id, TrexStreamsCompiledObj *obj, double duration) {
    m_port_id = port_id;
    m_event_id = event_id;
    m_obj = obj;
    m_duration = duration;
}


/**
 * clone for DP start message
 *
 */
TrexStatelessCpToDpMsgBase *
TrexStatelessDpStart::clone() {

    TrexStreamsCompiledObj *new_obj = m_obj->clone();

    TrexStatelessCpToDpMsgBase *new_msg = new TrexStatelessDpStart(m_port_id, m_event_id, new_obj, m_duration);

    return new_msg;
}

TrexStatelessDpStart::~TrexStatelessDpStart() {
    if (m_obj) {
        delete m_obj;
    }
}

bool
TrexStatelessDpStart::handle(TrexStatelessDpCore *dp_core) {

    /* staet traffic */
    dp_core->start_traffic(m_obj, m_duration,m_event_id);

    return true;
}

/*************************
  stop traffic message
 ************************/
bool
TrexStatelessDpStop::handle(TrexStatelessDpCore *dp_core) {


    dp_core->stop_traffic(m_port_id,m_stop_only_for_event_id,m_event_id);
    return true;
}


void TrexStatelessDpStop::on_node_remove(){
    if ( m_core ) {
        assert(m_core->m_non_active_nodes>0);
        m_core->m_non_active_nodes--;
    }
}


TrexStatelessCpToDpMsgBase * TrexStatelessDpPause::clone(){

    TrexStatelessDpPause *new_msg = new TrexStatelessDpPause(m_port_id);
    return new_msg;
}


bool TrexStatelessDpPause::handle(TrexStatelessDpCore *dp_core){
    dp_core->pause_traffic(m_port_id);
    return (true);
}



TrexStatelessCpToDpMsgBase * TrexStatelessDpResume::clone(){
    TrexStatelessDpResume *new_msg = new TrexStatelessDpResume(m_port_id);
    return new_msg;
}

bool TrexStatelessDpResume::handle(TrexStatelessDpCore *dp_core){
    dp_core->resume_traffic(m_port_id);
    return (true);
}


/**
 * clone for DP stop message
 *
 */
TrexStatelessCpToDpMsgBase *
TrexStatelessDpStop::clone() {
    TrexStatelessDpStop *new_msg = new TrexStatelessDpStop(m_port_id);

    new_msg->set_event_id(m_event_id);
    new_msg->set_wait_for_event_id(m_stop_only_for_event_id);
    /* set back pointer to master */
    new_msg->set_core_ptr(m_core);

    return new_msg;
}



TrexStatelessCpToDpMsgBase *
TrexStatelessDpQuit::clone(){

    TrexStatelessCpToDpMsgBase *new_msg = new TrexStatelessDpQuit();

    return new_msg;
}


bool TrexStatelessDpQuit::handle(TrexStatelessDpCore *dp_core){

    /* quit  */
    dp_core->quit_main_loop();
    return (true);
}

bool TrexStatelessDpCanQuit::handle(TrexStatelessDpCore *dp_core){

    if ( dp_core->are_all_ports_idle() ){
        /* if all ports are idle quit now */
        set_quit(true);
    }
    return (true);
}

TrexStatelessCpToDpMsgBase *
TrexStatelessDpCanQuit::clone(){

    TrexStatelessCpToDpMsgBase *new_msg = new TrexStatelessDpCanQuit();

    return new_msg;
}

/*************************
  update traffic message
 ************************/
bool
TrexStatelessDpUpdate::handle(TrexStatelessDpCore *dp_core) {
    dp_core->update_traffic(m_port_id, m_factor);

    return true;
}

TrexStatelessCpToDpMsgBase *
TrexStatelessDpUpdate::clone() {
    TrexStatelessCpToDpMsgBase *new_msg = new TrexStatelessDpUpdate(m_port_id, m_factor);

    return new_msg;
}


/*************************
  push PCAP message
 ************************/
bool
TrexStatelessDpPushPCAP::handle(TrexStatelessDpCore *dp_core) {
    dp_core->push_pcap(m_port_id,
                       m_event_id,
                       m_pcap_filename,
                       m_ipg_usec,
                       m_speedup,
                       m_count,
                       m_duration,
                       m_is_dual);
    return true;
}

TrexStatelessCpToDpMsgBase *
TrexStatelessDpPushPCAP::clone() {
    TrexStatelessCpToDpMsgBase *new_msg = new TrexStatelessDpPushPCAP(m_port_id,
                                                                      m_event_id,
                                                                      m_pcap_filename,
                                                                      m_ipg_usec,
                                                                      m_speedup,
                                                                      m_count,
                                                                      m_duration,
                                                                      m_is_dual);

    return new_msg;
}


/*************************
  barrier message
 ************************/

bool
TrexStatelessDpBarrier::handle(TrexStatelessDpCore *dp_core) {
    dp_core->barrier(m_port_id, m_event_id);
    return true;
}

TrexStatelessCpToDpMsgBase *
TrexStatelessDpBarrier::clone() {

    TrexStatelessCpToDpMsgBase *new_msg = new TrexStatelessDpBarrier(m_port_id, m_event_id);

    return new_msg;
}

/************************* messages from DP to CP **********************/
bool
TrexDpPortEventMsg::handle() {
    TrexStatelessPort *port = get_stateless_obj()->get_port_by_id(m_port_id);
    port->get_dp_events().on_core_reporting_in(m_event_id, m_thread_id, get_status());

    return (true);
}

/************************* messages from CP to RX **********************/
bool TrexStatelessRxEnableLatency::handle (CRxCoreStateless *rx_core) {
    rx_core->enable_latency();
    return true;
}

bool TrexStatelessRxDisableLatency::handle (CRxCoreStateless *rx_core) {
    rx_core->disable_latency();
    return true;
}

bool TrexStatelessRxQuit::handle (CRxCoreStateless *rx_core) {
    rx_core->quit();
    return true;
}


TrexStatelessRxSwGetPkts::TrexStatelessRxSwGetPkts(uint8_t port_id, TrexStatelessMsgReply<RxPacketBuffer *> &reply) :  m_reply(reply) {
    m_port_id = port_id;
}


bool
TrexStatelessRxStartCapture::handle(CRxCoreStateless *rx_core) {
    rx_core->start_capture(m_port_id, m_pcap_filename, m_limit, m_shared_counter);

    return true;
}

bool
TrexStatelessRxStopCapture::handle(CRxCoreStateless *rx_core) {
    rx_core->stop_capture(m_port_id);

    return true;
}


bool TrexStatelessRxSwGetPkts::handle(CRxCoreStateless *rx_core) {
    RxPacketBuffer *pkt_buffer = rx_core->get_rx_sw_pkt_buffer(m_port_id);
    assert(pkt_buffer);
    m_reply.set(pkt_buffer);

    return true;
}