/*
 * Copyright (C) 2022 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 <android-base/properties.h>
#include <android-base/strings.h>
#include <android/api-level.h>
#include <android/hidl/manager/1.0/IServiceManager.h>
#include <gmock/gmock.h>
#include <hidl/ServiceManagement.h>
#include <vintf/VintfObject.h>

#define __ANDROID_VENDOR_API_24Q2__ 202404

namespace android {
namespace vintf {
namespace testing {

static constexpr int kMaxNumberOfHidlHalsU = 100;

// Tests that the device is not registering any HIDL interfaces.
// HIDL is being deprecated. Only applicable to devices launching with Android
// 14 and later.
class VintfNoHidlTest : public ::testing::Test {};

static std::set<std::string> allHidlManifestInterfaces() {
  std::set<std::string> ret;
  auto setInserter = [&](const vintf::ManifestInstance& i) -> bool {
    if (i.format() != vintf::HalFormat::HIDL) {
      return true;
    }
    ret.insert(i.getFqInstance().getFqNameString());
    return true;
  };
  vintf::VintfObject::GetDeviceHalManifest()->forEachInstance(setInserter);
  vintf::VintfObject::GetFrameworkHalManifest()->forEachInstance(setInserter);
  return ret;
}

// @VsrTest = VSR-3.2-001.001|VSR-3.2-001.002
TEST_F(VintfNoHidlTest, NoHidl) {
  int apiLevel = android::base::GetIntProperty("ro.vendor.api_level", 0);
  if (apiLevel < __ANDROID_API_U__) {
    GTEST_SKIP() << "Not applicable to this device";
    return;
  }
  int maxNumberOfHidlHals;
  std::set<std::string> halInterfaces;
  if (apiLevel == __ANDROID_API_U__) {
    maxNumberOfHidlHals = kMaxNumberOfHidlHalsU;
    sp<hidl::manager::V1_0::IServiceManager> sm =
        ::android::hardware::defaultServiceManager();
    ASSERT_NE(sm, nullptr);
    hardware::Return<void> ret =
        sm->list([&halInterfaces](const auto& interfaces) {
          for (const auto& interface : interfaces) {
            std::vector<std::string> splitInterface =
                android::base::Split(interface, "@");
            ASSERT_GE(splitInterface.size(), 1);
            // We only care about packages, since HIDL HALs typically need to
            // include all of the older minor versions as well as the version
            // they are implementing and we don't want to count those
            halInterfaces.insert(splitInterface[0]);
          }
        });
  } else {
    maxNumberOfHidlHals = 0;
    halInterfaces = allHidlManifestInterfaces();
  }
  if (halInterfaces.size() > maxNumberOfHidlHals) {
    ADD_FAILURE()
        << "There are " << halInterfaces.size()
        << " HIDL interfaces served on the device. "
        << "These must be converted to AIDL as part of HIDL's "
           "deprecation processes.\n"
           "NOTE: vts_treble_vintf_vendor_test should pass before this test. "
           "Make sure the device under test is targeting "
           "the correct Framework Compatibility Matrix with "
           "target-level=\"202404\" or greater. That will cause "
           "the framework/system HIDL services to stop being registered. "
           "If those are still registered because the device is targeting "
           "and older FCM, this test will fail.";
    for (const auto& interface : halInterfaces) {
      ADD_FAILURE() << interface << " registered as a HIDL interface "
                    << "but must be in AIDL";
    }
  }
}

}  // namespace testing
}  // namespace vintf
}  // namespace android