// Copyright 2017 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "base/android/jni_array.h"
#include "base/android/library_loader/library_loader_hooks.h"
#include "base/android/memory_pressure_listener_android.h"
#include "base/android/unguessable_token_android.h"
#include "base/check.h"
#include "base/command_line.h"
#include "base/no_destructor.h"
#include "base/unguessable_token.h"
#include "components/input/android/input_token_forwarder.h"
#include "content/app/android/content_main_android.h"
#include "content/child/child_thread_impl.h"
#include "content/common/android/surface_wrapper.h"
#include "content/common/shared_file_util.h"
#include "content/public/common/content_descriptors.h"
#include "content/public/common/content_switches.h"
#include "gpu/command_buffer/service/texture_owner.h"
#include "gpu/ipc/common/gpu_surface_lookup.h"
#include "ui/gl/android/scoped_java_surface.h"
#include "ui/gl/android/scoped_java_surface_control.h"
#include "ui/gl/android/surface_texture.h"

// Must come after all headers that specialize FromJniType() / ToJniType().
#include "content/public/android/content_app_jni/ContentChildProcessServiceDelegate_jni.h"

using base::android::AttachCurrentThread;
using base::android::JavaParamRef;

namespace content {

namespace {

// TODO(sievers): Use two different implementations of this depending on if
// we're in a renderer or gpu process.
class ChildProcessSurfaceManager : public gpu::GpuSurfaceLookup,
                                   public input::InputTokenForwarder {
 public:
  ChildProcessSurfaceManager() = default;

  ChildProcessSurfaceManager(const ChildProcessSurfaceManager&) = delete;
  ChildProcessSurfaceManager& operator=(const ChildProcessSurfaceManager&) =
      delete;

  ~ChildProcessSurfaceManager() override = default;

  // |service_impl| is the instance of
  // org.chromium.content.app.ChildProcessService.
  void SetServiceImpl(const base::android::JavaRef<jobject>& service_impl) {
    service_impl_.Reset(service_impl);
  }

  // Overridden from GpuSurfaceLookup:
  gpu::SurfaceRecord AcquireJavaSurface(int surface_id) override {
    JNIEnv* env = base::android::AttachCurrentThread();
    base::android::ScopedJavaLocalRef<jobject> surface_wrapper =
        content::Java_ContentChildProcessServiceDelegate_getViewSurface(
            env, service_impl_, surface_id);
    if (!surface_wrapper)
      return gpu::SurfaceRecord(gl::ScopedJavaSurface(),
                                /*can_be_used_with_surface_control=*/false);

    bool can_be_used_with_surface_control =
        JNI_SurfaceWrapper_canBeUsedWithSurfaceControl(env, surface_wrapper);
    bool wraps_surface =
        JNI_SurfaceWrapper_getWrapsSurface(env, surface_wrapper);

    if (wraps_surface) {
      gl::ScopedJavaSurface surface(
          content::JNI_SurfaceWrapper_takeSurface(env, surface_wrapper),
          /*auto_release=*/true);
      DCHECK(!surface.j_surface().is_null());
      return gpu::SurfaceRecord(
          std::move(surface), can_be_used_with_surface_control,
          content::JNI_SurfaceWrapper_getBrowserInputToken(env,
                                                           surface_wrapper));
    } else {
      gl::ScopedJavaSurfaceControl surface_control(
          JNI_SurfaceWrapper_takeSurfaceControl(env, surface_wrapper),
          /*release_on_destroy=*/true);
      return gpu::SurfaceRecord(std::move(surface_control));
    }
  }

  // input::InputTokenForwarder overrides.
  void ForwardVizInputTransferToken(
      int surface_id,
      const jni_zero::JavaRef<>& viz_input_token) override {
    JNIEnv* env = base::android::AttachCurrentThread();
    content::Java_ContentChildProcessServiceDelegate_forwardInputTransferToken(
        env, service_impl_, surface_id, viz_input_token);
  }

 private:
  // The instance of org.chromium.content.app.ChildProcessService.
  base::android::ScopedJavaGlobalRef<jobject> service_impl_;
};

ChildProcessSurfaceManager* GetChildProcessSurfaceManager() {
  static base::NoDestructor<ChildProcessSurfaceManager> manager;
  return manager.get();
}

// Chrome actually uses the renderer code path for all of its child
// processes such as renderers, plugins, etc.
static void JNI_ContentChildProcessServiceDelegate_InternalInitChildProcess(
    JNIEnv* env,
    const JavaParamRef<jobject>& service_impl,
    jint cpu_count,
    jlong cpu_features) {
  InitChildProcessCommon(cpu_count, cpu_features);

  GetChildProcessSurfaceManager()->SetServiceImpl(service_impl);

  gpu::GpuSurfaceLookup::InitInstance(GetChildProcessSurfaceManager());
  input::InputTokenForwarder::SetInstance(GetChildProcessSurfaceManager());
}

}  // namespace

static void JNI_ContentChildProcessServiceDelegate_InitChildProcess(
    JNIEnv* env,
    const JavaParamRef<jobject>& obj,
    jint cpu_count,
    jlong cpu_features) {
  JNI_ContentChildProcessServiceDelegate_InternalInitChildProcess(
      env, obj, cpu_count, cpu_features);
}

static void JNI_ContentChildProcessServiceDelegate_InitMemoryPressureListener(
    JNIEnv* env) {
  base::android::MemoryPressureListenerAndroid::Initialize(env);
}

static void
JNI_ContentChildProcessServiceDelegate_RetrieveFileDescriptorsIdsToKeys(
    JNIEnv* env,
    const JavaParamRef<jobject>& obj) {
  std::map<int, std::string> ids_to_keys;
  std::string file_switch_value =
      base::CommandLine::ForCurrentProcess()->GetSwitchValueASCII(
          switches::kSharedFiles);

  std::vector<int> ids;
  std::vector<std::string> keys;
  if (!file_switch_value.empty()) {
    std::optional<std::map<int, std::string>> ids_to_keys_from_command_line =
        ParseSharedFileSwitchValue(file_switch_value);
    if (ids_to_keys_from_command_line) {
      for (auto iter : *ids_to_keys_from_command_line) {
        ids.push_back(iter.first);
        keys.push_back(iter.second);
      }
    }
  }

  Java_ContentChildProcessServiceDelegate_setFileDescriptorsIdsToKeys(
      env, obj, base::android::ToJavaIntArray(env, ids),
      base::android::ToJavaArrayOfStrings(env, keys));
}

}  // namespace content

DEFINE_JNI(ContentChildProcessServiceDelegate)