chromium/components/safe_browsing/content/renderer/phishing_classifier/phishing_model_setter_impl.cc

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

#include "components/safe_browsing/content/renderer/phishing_classifier/phishing_model_setter_impl.h"

#include "components/safe_browsing/content/renderer/phishing_classifier/scorer.h"
#include "third_party/blink/public/common/associated_interfaces/associated_interface_registry.h"

namespace safe_browsing {

std::unique_ptr<Scorer> CreateFlatBufferModelScorer(
    base::ReadOnlySharedMemoryRegion flatbuffer_region,
    base::File tflite_visual_model) { … }

std::unique_ptr<Scorer> CreateScorerWithImageEmbeddingModel(
    base::ReadOnlySharedMemoryRegion flatbuffer_region,
    base::File tflite_visual_model,
    base::File image_embedding_model) { … }

PhishingModelSetterImpl::PhishingModelSetterImpl() = default;
PhishingModelSetterImpl::~PhishingModelSetterImpl() = default;

void PhishingModelSetterImpl::RegisterMojoInterfaces(
    blink::AssociatedInterfaceRegistry* associated_interfaces) { … }

void PhishingModelSetterImpl::UnregisterMojoInterfaces(
    blink::AssociatedInterfaceRegistry* associated_interfaces) { … }

void PhishingModelSetterImpl::SetImageEmbeddingAndPhishingFlatBufferModel(
    base::ReadOnlySharedMemoryRegion flatbuffer_region,
    base::File tflite_visual_model,
    base::File image_embedding_model) { … }

void PhishingModelSetterImpl::SetPhishingFlatBufferModel(
    base::ReadOnlySharedMemoryRegion flatbuffer_region,
    base::File tflite_visual_model) { … }

void PhishingModelSetterImpl::AttachImageEmbeddingModel(
    base::File image_embedding_model) { … }

void PhishingModelSetterImpl::ClearScorer() { … }

void PhishingModelSetterImpl::SetTestObserver(
    mojo::PendingRemote<mojom::PhishingModelSetterTestObserver> observer,
    SetTestObserverCallback callback) { … }

void PhishingModelSetterImpl::OnRendererAssociatedRequest(
    mojo::PendingAssociatedReceiver<mojom::PhishingModelSetter> receiver) { … }

}  // namespace safe_browsing