Atmosphere/libraries/libstratosphere/source/powctl/powctl_session_api.cpp

89 lines
2.9 KiB
C++
Raw Normal View History

/*
* 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/>.
*/
#include <stratosphere.hpp>
#include "impl/powctl_device_management.hpp"
namespace ams::powctl {
namespace {
ddsf::AccessMode SanitizeAccessMode(ddsf::AccessMode access_mode) {
switch (access_mode) {
case ddsf::AccessMode_Read:
case ddsf::AccessMode_Write:
case ddsf::AccessMode_ReadWrite:
case ddsf::AccessMode_WriteShared:
case ddsf::AccessMode_ReadWriteShared:
return access_mode;
default:
return ddsf::AccessMode_None;
}
}
impl::SessionImpl &GetSessionImpl(Session &session) {
return GetReference(session.impl_storage);
}
void DestroySession(Session &session) {
GetSessionImpl(session).~SessionImpl();
session.has_session = false;
}
void DestroySessionIfNecessary(Session &session) {
if (session.has_session) {
DestroySession(session);
}
}
void CloseSessionIfOpen(Session &session) {
if (session.has_session && GetSessionImpl(session).IsOpen()) {
DestroySession(session);
}
}
}
Result OpenSession(Session *out, DeviceCode device_code, ddsf::AccessMode access_mode) {
/* Validate input. */
AMS_ASSERT(out != nullptr);
access_mode = SanitizeAccessMode(access_mode);
/* Find the target device. */
impl::IDevice *device = nullptr;
R_TRY(impl::FindDevice(std::addressof(device), device_code));
/* Clean up the session if we have one. */
DestroySessionIfNecessary(*out);
/* Construct the session. */
new (std::addressof(GetSessionImpl(*out))) impl::SessionImpl;
auto guard = SCOPE_GUARD { DestroySessionIfNecessary(*out); };
/* Try to open the session. */
R_TRY(ddsf::OpenSession(device, std::addressof(GetSessionImpl(*out)), access_mode));
/* We opened the session! */
guard.Cancel();
return ResultSuccess();
}
void CloseSession(Session &session) {
/* This seems extremely unnecessary/duplicate, but it's what Nintendo does. */
CloseSessionIfOpen(session);
DestroySession(session);
}
}