1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
// Copyright 2018 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef CHROME_BROWSER_UI_WEBUI_ASH_SMB_SHARES_SMB_HANDLER_H_
#define CHROME_BROWSER_UI_WEBUI_ASH_SMB_SHARES_SMB_HANDLER_H_
#include <string>
#include "base/functional/callback_forward.h"
#include "base/memory/raw_ptr.h"
#include "base/memory/weak_ptr.h"
#include "chrome/browser/ash/smb_client/smb_service.h"
#include "content/public/browser/web_ui_message_handler.h"
class Profile;
namespace ash::smb_dialog {
class SmbHandler : public content::WebUIMessageHandler {
public:
using UpdateCredentialsCallback =
base::OnceCallback<void(const std::string& username,
const std::string& password)>;
SmbHandler(Profile* profile, UpdateCredentialsCallback update_cred_callback);
SmbHandler(const SmbHandler&) = delete;
SmbHandler& operator=(const SmbHandler&) = delete;
~SmbHandler() override;
// content::WebUIMessageHandler
void RegisterMessages() override;
void SetSmbServiceForTesting(smb_client::SmbService* smb_service);
protected:
void HandleHasAnySmbMountedBefore(const base::Value::List& args);
private:
friend class TestSmbHandler;
// WebUI call to mount an Smb Filesystem.
void HandleSmbMount(const base::Value::List& args);
// WebUI call to start file share discovery on the network.
void HandleStartDiscovery(const base::Value::List& args);
// WebUI call to update the credentials of a mounted share.
void HandleUpdateCredentials(const base::Value::List& args);
// Callback handler for SmbMount.
void HandleSmbMountResponse(const std::string& callback_id,
smb_client::SmbMountResult result);
// Callback handler for StartDiscovery.
void HandleGatherSharesResponse(
const std::vector<smb_client::SmbUrl>& shares_gathered,
bool done);
// Callback handler that indicates discovery is complete.
void HandleDiscoveryDone();
smb_client::SmbService* GetLocalSmbService();
bool host_discovery_done_ = false;
base::OnceClosure stored_mount_call_;
const raw_ptr<Profile> profile_;
UpdateCredentialsCallback update_cred_callback_;
raw_ptr<smb_client::SmbService> test_smb_service_ = nullptr;
base::WeakPtrFactory<SmbHandler> weak_ptr_factory_{this};
};
} // namespace ash::smb_dialog
#endif // CHROME_BROWSER_UI_WEBUI_ASH_SMB_SHARES_SMB_HANDLER_H_
|
