mirror of
https://github.com/Atmosphere-NX/Atmosphere.git
synced 2024-12-24 19:26:04 +00:00
148 lines
4.6 KiB
C++
148 lines
4.6 KiB
C++
/*
|
|
* Copyright (c) 2018-2020 Atmosphère-NX
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify it
|
|
* under the terms and conditions of the GNU General Public License,
|
|
* version 2, as published by the Free Software Foundation.
|
|
*
|
|
* This program is distributed in the hope it will be useful, but WITHOUT
|
|
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
|
|
* more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
#if defined(ATMOSPHERE_IS_STRATOSPHERE)
|
|
#include <stratosphere.hpp>
|
|
#elif defined(ATMOSPHERE_IS_MESOSPHERE)
|
|
#include <mesosphere.hpp>
|
|
#elif defined(ATMOSPHERE_IS_EXOSPHERE)
|
|
#include <exosphere.hpp>
|
|
#else
|
|
#include <vapours.hpp>
|
|
#endif
|
|
#include "prfile2_pdm_disk_set.hpp"
|
|
|
|
namespace ams::prfile2::pdm::disk {
|
|
|
|
pdm::Error OpenDisk(InitDisk *init_disk_table, HandleType *out) {
|
|
/* Check the arguments. */
|
|
if (out == nullptr || init_disk_table == nullptr || init_disk_table->function == nullptr) {
|
|
return pdm::Error_InvalidParameter;
|
|
}
|
|
|
|
/* Find a free disk holder. */
|
|
DiskHolder *holder = nullptr;
|
|
for (auto &h : impl::g_disk_set.disk_holders) {
|
|
if (h.disk == nullptr) {
|
|
holder = std::addressof(h);
|
|
break;
|
|
}
|
|
}
|
|
if (holder == nullptr) {
|
|
return pdm::Error_NotExistFreeDiskStruct;
|
|
}
|
|
|
|
/* Find a free disk. */
|
|
Disk *disk = nullptr;
|
|
for (auto &d : impl::g_disk_set.disks) {
|
|
if (!d.IsOpen()) {
|
|
disk = std::addressof(d);
|
|
break;
|
|
}
|
|
}
|
|
if (disk == nullptr) {
|
|
return pdm::Error_NotExistFreeDiskStruct;
|
|
}
|
|
const auto disk_id = disk - impl::g_disk_set.disks;
|
|
|
|
/* Call the disk initialze function. */
|
|
init_disk_table->function(std::addressof(disk->disk_table), init_disk_table->ui_ext);
|
|
|
|
/* Set the disk as open .*/
|
|
disk->SetOpen(true);
|
|
|
|
/* Set the init disk table. */
|
|
disk->init_disk_table = init_disk_table;
|
|
|
|
/* Note the opened disk. */
|
|
++impl::g_disk_set.num_allocated_disks;
|
|
|
|
/* Increment the disk's signature. */
|
|
disk->signature = static_cast<u16>(disk->signature + 1);
|
|
|
|
/* Increment the disk's open count. */
|
|
++disk->open_count;
|
|
|
|
/* Set the disk in the holder. */
|
|
holder->signature = disk->signature;
|
|
holder->disk = disk;
|
|
|
|
/* Set the output handle. */
|
|
*out = ConstructDiskHandle(disk_id, disk->signature);
|
|
|
|
/* If this was the first opening for the disk, initialize it. */
|
|
if (disk->open_count == 1) {
|
|
if (auto err = disk->disk_table.function_table->initialize(*out); err != pdm::Error_Ok) {
|
|
/* Close the disk. */
|
|
--disk->open_count;
|
|
disk->SetOpen(false);
|
|
--impl::g_disk_set.num_allocated_disks;
|
|
|
|
/* Reset the holder. */
|
|
holder->disk = nullptr;
|
|
|
|
return pdm::Error_DriverError;
|
|
}
|
|
}
|
|
|
|
return pdm::Error_Ok;
|
|
}
|
|
|
|
pdm::Error CloseDisk(HandleType handle) {
|
|
/* Get the disk. */
|
|
Disk *disk = GetDisk(handle);
|
|
if (disk == nullptr) {
|
|
return pdm::Error_InvalidParameter;
|
|
}
|
|
|
|
/* Check that the disk is open and unlocked. */
|
|
if (!disk->IsOpen()) {
|
|
return pdm::Error_StateClosed;
|
|
}
|
|
if (disk->IsLocked()) {
|
|
return pdm::Error_StateLocked;
|
|
}
|
|
|
|
/* Get the disk holder. */
|
|
DiskHolder *holder = GetDiskHolder(handle);
|
|
if (holder == nullptr) {
|
|
return pdm::Error_InvalidParameter;
|
|
}
|
|
|
|
/* Close the disk. */
|
|
if (disk->open_count == 1) {
|
|
/* Finalize the disk. */
|
|
if (auto err = disk->disk_table.function_table->finalize(handle); err != pdm::Error_Ok) {
|
|
if (auto part = disk->current_partition_handle; part != InvalidHandle) {
|
|
part::SetDriverErrorCode(part, err);
|
|
}
|
|
return pdm::Error_DriverError;
|
|
}
|
|
|
|
/* Set the disk as closed. */
|
|
disk->SetOpen(false);
|
|
--impl::g_disk_set.num_allocated_disks;
|
|
|
|
/* Clear the disk holder. */
|
|
holder->disk = nullptr;
|
|
}
|
|
|
|
/* Decrement the disk's open count. */
|
|
--disk->open_count;
|
|
|
|
return pdm::Error_Ok;
|
|
}
|
|
|
|
}
|