118 lines
4.5 KiB
C++
118 lines
4.5 KiB
C++
// Copyright (c) 2022 Microsoft, Inc.
|
|
// Use of this source code is governed by the MIT license that can be
|
|
// found in the LICENSE file.
|
|
|
|
#ifndef ELECTRON_SHELL_BROWSER_USB_ELECTRON_USB_DELEGATE_H_
|
|
#define ELECTRON_SHELL_BROWSER_USB_ELECTRON_USB_DELEGATE_H_
|
|
|
|
#include <memory>
|
|
#include <string>
|
|
#include <unordered_map>
|
|
#include <vector>
|
|
|
|
#include "base/containers/span.h"
|
|
#include "base/memory/weak_ptr.h"
|
|
#include "content/public/browser/usb_chooser.h"
|
|
#include "content/public/browser/usb_delegate.h"
|
|
#include "mojo/public/cpp/bindings/pending_receiver.h"
|
|
#include "mojo/public/cpp/bindings/pending_remote.h"
|
|
#include "services/device/public/mojom/usb_device.mojom-forward.h"
|
|
#include "services/device/public/mojom/usb_enumeration_options.mojom-forward.h"
|
|
#include "services/device/public/mojom/usb_manager.mojom-forward.h"
|
|
#include "third_party/blink/public/mojom/usb/web_usb_service.mojom.h"
|
|
#include "url/origin.h"
|
|
|
|
namespace content {
|
|
class BrowserContext;
|
|
class RenderFrameHost;
|
|
} // namespace content
|
|
|
|
namespace electron {
|
|
|
|
class UsbChooserController;
|
|
|
|
class ElectronUsbDelegate : public content::UsbDelegate {
|
|
public:
|
|
ElectronUsbDelegate();
|
|
ElectronUsbDelegate(ElectronUsbDelegate&&) = delete;
|
|
ElectronUsbDelegate& operator=(ElectronUsbDelegate&) = delete;
|
|
~ElectronUsbDelegate() override;
|
|
|
|
// content::UsbDelegate:
|
|
void AdjustProtectedInterfaceClasses(content::BrowserContext* browser_context,
|
|
const url::Origin& origin,
|
|
content::RenderFrameHost* frame,
|
|
std::vector<uint8_t>& classes) override;
|
|
std::unique_ptr<content::UsbChooser> RunChooser(
|
|
content::RenderFrameHost& frame,
|
|
blink::mojom::WebUsbRequestDeviceOptionsPtr options,
|
|
blink::mojom::WebUsbService::GetPermissionCallback callback) override;
|
|
bool CanRequestDevicePermission(content::BrowserContext* browser_context,
|
|
const url::Origin& origin) override;
|
|
void RevokeDevicePermissionWebInitiated(
|
|
content::BrowserContext* browser_context,
|
|
const url::Origin& origin,
|
|
const device::mojom::UsbDeviceInfo& device) override;
|
|
const device::mojom::UsbDeviceInfo* GetDeviceInfo(
|
|
content::BrowserContext* browser_context,
|
|
const std::string& guid) override;
|
|
bool HasDevicePermission(
|
|
content::BrowserContext* browser_context,
|
|
content::RenderFrameHost* frame,
|
|
const url::Origin& origin,
|
|
const device::mojom::UsbDeviceInfo& device_info) override;
|
|
void GetDevices(
|
|
content::BrowserContext* browser_context,
|
|
blink::mojom::WebUsbService::GetDevicesCallback callback) override;
|
|
void GetDevice(
|
|
content::BrowserContext* browser_context,
|
|
const std::string& guid,
|
|
base::span<const uint8_t> blocked_interface_classes,
|
|
mojo::PendingReceiver<device::mojom::UsbDevice> device_receiver,
|
|
mojo::PendingRemote<device::mojom::UsbDeviceClient> device_client)
|
|
override;
|
|
void AddObserver(content::BrowserContext* browser_context,
|
|
Observer* observer) override;
|
|
void RemoveObserver(content::BrowserContext* browser_context,
|
|
Observer* observer) override;
|
|
|
|
// TODO: See if we can separate these from Profiles upstream.
|
|
void IncrementConnectionCount(content::BrowserContext* browser_context,
|
|
const url::Origin& origin) override {}
|
|
|
|
void DecrementConnectionCount(content::BrowserContext* browser_context,
|
|
const url::Origin& origin) override {}
|
|
|
|
bool IsServiceWorkerAllowedForOrigin(const url::Origin& origin) override;
|
|
|
|
void DeleteControllerForFrame(content::RenderFrameHost* render_frame_host);
|
|
|
|
bool PageMayUseUsb(content::Page& page) override;
|
|
|
|
private:
|
|
UsbChooserController* ControllerForFrame(
|
|
content::RenderFrameHost* render_frame_host);
|
|
|
|
UsbChooserController* AddControllerForFrame(
|
|
content::RenderFrameHost* render_frame_host,
|
|
blink::mojom::WebUsbRequestDeviceOptionsPtr options,
|
|
blink::mojom::WebUsbService::GetPermissionCallback callback);
|
|
|
|
class ContextObservation;
|
|
|
|
ContextObservation* GetContextObserver(
|
|
content::BrowserContext* browser_context);
|
|
|
|
base::flat_map<content::BrowserContext*, std::unique_ptr<ContextObservation>>
|
|
observations_;
|
|
|
|
std::unordered_map<content::RenderFrameHost*,
|
|
std::unique_ptr<UsbChooserController>>
|
|
controller_map_;
|
|
|
|
base::WeakPtrFactory<ElectronUsbDelegate> weak_factory_{this};
|
|
};
|
|
|
|
} // namespace electron
|
|
|
|
#endif // ELECTRON_SHELL_BROWSER_USB_ELECTRON_USB_DELEGATE_H_
|