/*
 *  Copyright 2017 The WebRTC Project Authors. All rights reserved.
 *
 *  Use of this source code is governed by a BSD-style license
 *  that can be found in the LICENSE file in the root of the source
 *  tree. An additional intellectual property rights grant can be found
 *  in the file PATENTS.  All contributing project authors may
 *  be found in the AUTHORS file in the root of the source tree.
 */

#ifndef P2P_TEST_TEST_TURN_CUSTOMIZER_H_
#define P2P_TEST_TEST_TURN_CUSTOMIZER_H_

#include <cstddef>
#include <memory>

#include "api/transport/stun.h"
#include "api/turn_customizer.h"
#include "p2p/base/port_interface.h"
#include "test/gtest.h"

namespace webrtc {

class TestTurnCustomizer : public TurnCustomizer {
 public:
  TestTurnCustomizer() {}
  ~TestTurnCustomizer() override {}

  enum TestTurnAttributeExtensions {
    // Test only attribute
    STUN_ATTR_COUNTER = 0xFF02  // Number
  };

  void MaybeModifyOutgoingStunMessage(PortInterface* port,
                                      StunMessage* message) override {
    modify_cnt_++;

    ASSERT_NE(0, message->type());
    if (add_counter_) {
      // TODO: crbug.com/504567957 - redo authentication so that
      // attribute will not be ignored, or move caller to do
      // modification before authentication.
      message->AddAttribute(std::make_unique<StunUInt32Attribute>(
          STUN_ATTR_COUNTER, modify_cnt_));
    }
    return;
  }

  bool AllowChannelData(PortInterface* port,
                        const void* data,
                        size_t size,
                        bool payload) override {
    allow_channel_data_cnt_++;
    return allow_channel_data_;
  }

  bool add_counter_ = false;
  bool allow_channel_data_ = true;
  unsigned int modify_cnt_ = 0;
  unsigned int allow_channel_data_cnt_ = 0;
};

}  //  namespace webrtc


#endif  // P2P_TEST_TEST_TURN_CUSTOMIZER_H_