/*
 * Copyright (C) 2016 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.
 */

#include "utils.h"

#include <gtest/gtest.h>

#include <android-base/file.h>

#include "environment.h"
#include "get_test_data.h"

using namespace simpleperf;

// @CddTest = 6.1/C-0-2
TEST(utils, ConvertBytesToValue) {
  char buf[8];
  for (int i = 0; i < 8; ++i) {
    buf[i] = i;
  }
  ASSERT_EQ(0x1ULL, ConvertBytesToValue(buf + 1, 1));
  ASSERT_EQ(0x201ULL, ConvertBytesToValue(buf + 1, 2));
  ASSERT_EQ(0x05040302ULL, ConvertBytesToValue(buf + 2, 4));
  ASSERT_EQ(0x0706050403020100ULL, ConvertBytesToValue(buf, 8));
}

// @CddTest = 6.1/C-0-2
TEST(utils, ArchiveHelper) {
  std::unique_ptr<ArchiveHelper> ahelper = ArchiveHelper::CreateInstance(GetTestData(APK_FILE));
  ASSERT_TRUE(ahelper);
  bool found = false;
  ZipEntry lib_entry;
  ASSERT_TRUE(ahelper->IterateEntries([&](ZipEntry& entry, const std::string& name) {
    if (name == NATIVELIB_IN_APK) {
      found = true;
      lib_entry = entry;
      return false;
    }
    return true;
  }));
  ASSERT_TRUE(found);
  ZipEntry entry;
  ASSERT_TRUE(ahelper->FindEntry(NATIVELIB_IN_APK, &entry));
  ASSERT_EQ(entry.offset, lib_entry.offset);
  std::vector<uint8_t> data;
  ASSERT_TRUE(ahelper->GetEntryData(entry, &data));

  // Check reading wrong file formats.
  ASSERT_FALSE(ArchiveHelper::CreateInstance(GetTestData(ELF_FILE)));
  ASSERT_FALSE(ArchiveHelper::CreateInstance("/dev/zero"));
}

// @CddTest = 6.1/C-0-2
TEST(utils, GetCpusFromString) {
  ASSERT_EQ(GetCpusFromString("0-2"), std::make_optional<std::set<int>>({0, 1, 2}));
  ASSERT_EQ(GetCpusFromString("0,2-3"), std::make_optional<std::set<int>>({0, 2, 3}));
  ASSERT_EQ(GetCpusFromString("1,0-3,3,4"), std::make_optional<std::set<int>>({0, 1, 2, 3, 4}));
  ASSERT_EQ(GetCpusFromString("0,1-3, 5, 7-8"),
            std::make_optional<std::set<int>>({0, 1, 2, 3, 5, 7, 8}));
  ASSERT_EQ(GetCpusFromString(""), std::nullopt);
  ASSERT_EQ(GetCpusFromString("-3"), std::nullopt);
  ASSERT_EQ(GetCpusFromString("3,2-1"), std::nullopt);
  ASSERT_EQ("0-2", ToCpuString(std::set<int>({0, 1, 2})));
  ASSERT_EQ("0,2-3", ToCpuString(std::set<int>({0, 2, 3})));
  ASSERT_EQ("0-3,5,7-8", ToCpuString(std::set<int>({0, 1, 2, 3, 5, 7, 8})));
  ASSERT_EQ("", ToCpuString(std::set<int>({})));
}

// @CddTest = 6.1/C-0-2
TEST(utils, GetTidsFromString) {
  ASSERT_EQ(GetTidsFromString("0,12,9", false), std::make_optional(std::set<pid_t>({0, 9, 12})));
  ASSERT_EQ(GetTidsFromString("-2", false), std::nullopt);
}

// @CddTest = 6.1/C-0-2
TEST(utils, GetPidsFromStrings) {
  ASSERT_EQ(GetPidsFromStrings({"0,12", "9"}, false, false),
            std::make_optional(std::set<pid_t>({0, 9, 12})));
  ASSERT_EQ(GetPidsFromStrings({"-2"}, false, false), std::nullopt);
#if defined(__linux__)
  pid_t pid = getpid();
  ASSERT_EQ(GetPidsFromStrings({std::to_string(pid)}, true, false),
            std::make_optional(std::set<pid_t>({pid})));
  std::string process_name = GetCompleteProcessName(pid);
  ASSERT_EQ(GetPidsFromStrings({process_name}, true, true),
            std::make_optional(std::set<pid_t>({pid})));
#endif  // defined(__linux__)
}

// @CddTest = 6.1/C-0-2
TEST(utils, LineReader) {
  TemporaryFile tmpfile;
  close(tmpfile.release());
  ASSERT_TRUE(android::base::WriteStringToFile("line1\nline2", tmpfile.path));
  LineReader reader(tmpfile.path);
  ASSERT_TRUE(reader.Ok());
  std::string* line = reader.ReadLine();
  ASSERT_TRUE(line != nullptr);
  ASSERT_EQ(*line, "line1");
  line = reader.ReadLine();
  ASSERT_TRUE(line != nullptr);
  ASSERT_EQ(*line, "line2");
  ASSERT_TRUE(reader.ReadLine() == nullptr);
}

// @CddTest = 6.1/C-0-2
TEST(utils, OneTimeFreeAllocator) {
  OneTimeFreeAllocator allocator(1024);
  const char* s1 = allocator.AllocateString("hello");
  ASSERT_STREQ(s1, "hello");
  const char* s2 = allocator.AllocateString(std::string(1000, 'a'));
  ASSERT_EQ(strlen(s2), 1000);
  const char* s3 = allocator.AllocateString("world");
  ASSERT_STREQ(s3, "world");
}

// @CddTest = 6.1/C-0-2
TEST(utils, OneTimeFreeAllocator_overflow) {
  OneTimeFreeAllocator allocator(1024);
  // Test with a size that would cause overflow if size_t wraps.
  // We use a large size that is still somewhat manageable or just test the logic.
  // Since we can't easily allocate SIZE_MAX, we check if a large size > unit_size_ works.
  std::string long_string(2000, 'a');
  const char* s = allocator.AllocateString(long_string);
  ASSERT_EQ(strlen(s), 2000);
  ASSERT_EQ(s[0], 'a');
  ASSERT_EQ(s[1999], 'a');
}

// @CddTest = 6.1/C-0-2
TEST(utils, ReadableCount) {
  ASSERT_EQ(ReadableCount(0), "0");
  ASSERT_EQ(ReadableCount(204), "204");
  ASSERT_EQ(ReadableCount(1000), "1,000");
  ASSERT_EQ(ReadableCount(123456789), "123,456,789");
}
