/*
 * Copyright (C) 2018 The Android Open Source Project
 *
 * 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.
 */

#ifndef IDMAP2_TESTS_TESTHELPERS_H_
#define IDMAP2_TESTS_TESTHELPERS_H_

#include <stdio.h>
#include <string>
#include <string_view>

#include "gmock/gmock.h"
#include "gtest/gtest.h"

#include "android-base/file.h"

namespace android::idmap2 {

const unsigned char kIdmapRawData[] = {
    // IDMAP HEADER
    // 0x0: magic
    0x49, 0x44, 0x4d, 0x50,

    // 0x4: version
    0x09, 0x00, 0x00, 0x00,

    // 0x8: target crc
    0x34, 0x12, 0x00, 0x00,

    // 0xc: overlay crc
    0x78, 0x56, 0x00, 0x00,

    // 0x10: fulfilled policies
    0x11, 0x00, 0x00, 0x00,

    // 0x14: enforce overlayable
    0x01, 0x00, 0x00, 0x00,

    // 0x18: target path length
    0x0b, 0x00, 0x00, 0x00,

    // 0x1c: target path "targetX.apk"
    0x74, 0x61, 0x72, 0x67, 0x65, 0x74, 0x58, 0x2e, 0x61, 0x70, 0x6b, 0x00,

    // 0x28: overlay path length
    0x0c, 0x00, 0x00, 0x00,

    // 0x2c: overlay path "overlayX.apk"
    0x6f, 0x76, 0x65, 0x72, 0x6c, 0x61, 0x79, 0x58, 0x2e, 0x61, 0x70, 0x6b,

    // 0x38: overlay name length
    0x0b, 0x00, 0x00, 0x00,

    // 0x3c: overlay name "OverlayName"
    0x4f, 0x76, 0x65, 0x72, 0x6c, 0x61, 0x79, 0x4e, 0x61, 0x6D, 0x65, 0x00,

    // 0x48 -> 4c: debug string
    // string length,
    0x05, 0x00, 0x00, 0x00,

    // 0x4c string contents "debug\0\0\0" (padded to word alignment)
    0x64, 0x65, 0x62, 0x75, 0x67, 0x00, 0x00, 0x00,

    // DATA HEADER
    // 0x54: target_entry_count
    0x03, 0x00, 0x00, 0x00,

    // 0x58: target_inline_entry_count
    0x01, 0x00, 0x00, 0x00,

    // 0x5c: target_inline_entry_value_count
    0x01, 0x00, 0x00, 0x00,

    // 0x60: config_count
    0x01, 0x00, 0x00, 0x00,

    // 0x64: overlay_entry_count
    0x03, 0x00, 0x00, 0x00,

    // 0x68: string_pool_offset
    0x00, 0x00, 0x00, 0x00,

    // TARGET ENTRIES
    // 0x6c: target id (0x7f020000)
    0x00, 0x00, 0x02, 0x7f,

    // 0x70: overlay_id (0x7f020000)
    0x00, 0x00, 0x02, 0x7f,

    // 0x74: target id (0x7f030000)
    0x00, 0x00, 0x03, 0x7f,

    // 0x78: overlay_id (0x7f030000)
    0x00, 0x00, 0x03, 0x7f,

    // 0x7c: target id (0x7f030002)
    0x02, 0x00, 0x03, 0x7f,

    // 0x80: overlay_id (0x7f030001)
    0x01, 0x00, 0x03, 0x7f,

    // INLINE TARGET ENTRIES

    // 0x84: target_id
    0x00, 0x00, 0x04, 0x7f,

    // 0x88: start value index
    0x00, 0x00, 0x00, 0x00,

    // 0x8c: value count
    0x01, 0x00, 0x00, 0x00,

    // INLINE TARGET ENTRY VALUES

    // 0x90: config index
    0x00, 0x00, 0x00, 0x00,

    // 0x94: Res_value::size (value ignored by idmap)
    0x08, 0x00,

    // 0x98: Res_value::res0 (value ignored by idmap)
    0x00,

    // 0x9c: Res_value::dataType (TYPE_INT_HEX)
    0x11,

    // 0xa0: Res_value::data
    0x78, 0x56, 0x34, 0x12,

    // CONFIGURATIONS

    // 0xa4: ConfigDescription
    // size
    0x40, 0x00, 0x00, 0x00,
    // 0xa8: imsi
    0x00, 0x00, 0x00, 0x00,
    // 0xac: locale
    0x00, 0x00, 0x00, 0x00,
    // 0xb0: screenType
    0x02, 0x00, 0xe0, 0x01,
    // 0xb4: input
    0x00, 0x00, 0x00, 0x00,
    // 0xb8: screenSize
    0x00, 0x00, 0x00, 0x00,
    // 0xbc: version
    0x07, 0x00, 0x00, 0x00,
    // 0xc0: screenConfig
    0x00, 0x00, 0x00, 0x00,
    // 0xc4: screenSizeDp
    0x00, 0x00, 0x00, 0x00,
    // 0xc8: localeScript
    0x00, 0x00, 0x00, 0x00,
    // 0xcc: localVariant(1)
    0x00, 0x00, 0x00, 0x00,
    // 0xd0: localVariant(2)
    0x00, 0x00, 0x00, 0x00,
    // 0xd4: screenConfig2
    0x00, 0x00, 0x00, 0x00,
    // 0xd8: localeScriptWasComputed
    0x00,
    // 0xd9: localeNumberingSystem(1)
    0x00, 0x00, 0x00, 0x00,
    // 0xdd: localeNumberingSystem(2)
    0x00, 0x00, 0x00, 0x00,

    // 0xe1: padding
    0x00, 0x00, 0x00,


    // OVERLAY ENTRIES
    // 0xe4: 0x7f020000 -> 0x7f020000
    0x00, 0x00, 0x02, 0x7f, 0x00, 0x00, 0x02, 0x7f,

    // 0xec: 0x7f030000 -> 0x7f030000
    0x00, 0x00, 0x03, 0x7f, 0x00, 0x00, 0x03, 0x7f,

    // 0xf4: 0x7f030001 -> 0x7f030002
    0x01, 0x00, 0x03, 0x7f, 0x02, 0x00, 0x03, 0x7f,

    // 0xfc: string pool
    // string length,
    0x04, 0x00, 0x00, 0x00,

    // 0x100 string contents "test"
    0x74, 0x65, 0x73, 0x74};

constexpr unsigned int kIdmapRawDataLen = std::size(kIdmapRawData);
const unsigned int kIdmapRawDataOffset = 0x54;
const unsigned int kIdmapRawDataTargetCrc = 0x1234;
const unsigned int kIdmapRawOverlayCrc = 0x5678;
const unsigned int kIdmapRawDataPolicies = 0x11;
inline const std::string_view kIdmapRawTargetPath = "targetX.apk";
inline const std::string_view kIdmapRawOverlayPath = "overlayX.apk";
inline const std::string_view kIdmapRawOverlayName = "OverlayName";

std::string GetTestDataPath();

class TempFrroFile : public TemporaryFile {
public:
  TempFrroFile() {
    std::string new_path = path;
    new_path += ".frro";
    ::rename(path, new_path.c_str());
    const auto new_len = new_path.copy(path, sizeof(path) - 1);
    path[new_len] = '\0';
  }
};

class Idmap2Tests : public testing::Test {
 protected:
  void SetUp() override {
#ifdef __ANDROID__
    tmp_dir_path_ = "/data/local/tmp/idmap2-tests-XXXXXX";
#else
    tmp_dir_path_ = "/tmp/idmap2-tests-XXXXXX";
#endif
    EXPECT_NE(mkdtemp(const_cast<char*>(tmp_dir_path_.c_str())), nullptr)
        << "Failed to create temporary directory: " << strerror(errno);
    target_apk_path_ = GetTestDataPath() + "/target/target.apk";
    overlay_apk_path_ = GetTestDataPath() + "/overlay/overlay.apk";
    idmap_path_ = tmp_dir_path_ + "/a.idmap";
  }

  void TearDown() override {
    EXPECT_EQ(rmdir(tmp_dir_path_.c_str()), 0)
        << "Failed to remove temporary directory " << tmp_dir_path_ << ": " << strerror(errno);
  }

  const std::string& GetTempDirPath() {
    return tmp_dir_path_;
  }

  const std::string& GetTargetApkPath() {
    return target_apk_path_;
  }

  const std::string& GetOverlayApkPath() {
    return overlay_apk_path_;
  }

  const std::string& GetIdmapPath() {
    return idmap_path_;
  }

 private:
  std::string tmp_dir_path_;
  std::string target_apk_path_;
  std::string overlay_apk_path_;
  std::string idmap_path_;
};

}  // namespace android::idmap2

#endif  // IDMAP2_TESTS_TESTHELPERS_H_