text: /*
   * Copyright (c) 2016-2017, Arm Limited and affiliates.
   * SPDX-License-Identifier: Apache-2.0
   *
   * 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 "nsconfig.h"
  
  #include <string.h>
  #include "nsdynmemLIB.h"
  #include "NWK_INTERFACE/Include/protocol.h"
  #include "Service_Libs/mle_service/mle_service_api.h"
  
  #include "net_test_api.h"
  #include "Common_Protocols/tcp.h"
  
  #include "ns_trace.h"
  
  #define TRACE_GROUP "nTES"
  
  bool validate_tcp_state_name(int state)
  {
      switch (state) {
          case TCP_STATE_LISTEN:
          case TCP_STATE_SYN_SENT:
          case TCP_STATE_SYN_RECEIVED:
          case TCP_STATE_ESTABLISHED:
          case TCP_STATE_CLOSE_WAIT:
          case TCP_STATE_LAST_ACK:
          case TCP_STATE_FIN_WAIT_1:
          case TCP_STATE_FIN_WAIT_2:
          case TCP_STATE_CLOSING:
          case TCP_STATE_TIME_WAIT:
              return true;
          default:
              return false;
      }
  }
  
  int8_t arm_nwk_test_tcp_drop_tx(int state, uint8_t count)
  {
      int8_t retcode = -1;
      if (validate_tcp_state_name(state)) {
          if (count > 200) {
              count = 200;
          }
          retcode = tcp_test_drop_tx(state, count);
      }
  
      return retcode;
  }
  
  int8_t arm_nwk_test_tcp_drop_rx(int state, uint8_t count)
  {
      int8_t retcode = -1;
      if (validate_tcp_state_name(state)) {
          if (count > 200) {
              count = 200;
          }
          retcode = tcp_test_drop_rx(state, count);
      }
  
      return retcode;
  }
  
  void arm_nwk_test_tcp_drop_reset() {
      tcp_test_drop_reset();
  }
  
  void arm_nwk_test_mle_receive_filter_set(mle_service_filter_cb *response_filter_cb)
  {
      mle_service_receive_filter_cb_set(response_filter_cb);
  }
  