platform_system_core/fastboot/storage.cpp
Dmitrii Merkurev 0ee285aa68 fastboot: Introduce connect / disconnect for network-connected devices
Use introduced FileLock and network-connected devices storage entities
to introduce fastboot connect / disconnect commands

Test: everything works like discussed here go/fastboot-connect-disconnect on windows/linux
Bug: 267507450
Bug: 267506875
Change-Id: I2d6495ad567a3ddadd471a89b82d78c8c36a3d52
Signed-off-by: Dmitrii Merkurev <dimorinny@google.com>
2023-02-11 00:02:22 +00:00

67 lines
No EOL
2.5 KiB
C++

/*
* Copyright (C) 2023 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/file.h>
#include <android-base/logging.h>
#include <fstream>
#include "storage.h"
#include "util.h"
ConnectedDevicesStorage::ConnectedDevicesStorage() {
const std::string home_path = GetHomeDirPath();
if (home_path.empty()) {
return;
}
const std::string home_fastboot_path = home_path + kPathSeparator + ".fastboot";
if (!EnsureDirectoryExists(home_fastboot_path)) {
LOG(FATAL) << "Cannot create directory: " << home_fastboot_path;
}
// We're using a separate file for locking because the Windows LockFileEx does not
// permit opening a file stream for the locked file, even within the same process. So,
// we have to use fd or handle API to manipulate the storage files, which makes it
// nearly impossible to fully rewrite a file content without having to recreate it.
// Unfortunately, this is not an option during holding a lock.
devices_path_ = home_fastboot_path + kPathSeparator + "devices";
devices_lock_path_ = home_fastboot_path + kPathSeparator + "devices.lock";
}
void ConnectedDevicesStorage::WriteDevices(const FileLock&, const std::set<std::string>& devices) {
std::ofstream devices_stream(devices_path_);
std::copy(devices.begin(), devices.end(),
std::ostream_iterator<std::string>(devices_stream, "\n"));
}
std::set<std::string> ConnectedDevicesStorage::ReadDevices(const FileLock&) {
std::ifstream devices_stream(devices_path_);
std::istream_iterator<std::string> start(devices_stream), end;
std::set<std::string> devices(start, end);
return devices;
}
void ConnectedDevicesStorage::Clear(const FileLock&) {
if (!android::base::RemoveFileIfExists(devices_path_)) {
LOG(FATAL) << "Failed to clear connected device list: " << devices_path_;
}
}
FileLock ConnectedDevicesStorage::Lock() const {
return FileLock(devices_lock_path_);
}