mirror of
https://github.com/Atmosphere-NX/Atmosphere
synced 2024-11-16 01:49:32 +00:00
82 lines
2.9 KiB
C++
82 lines
2.9 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/>.
|
|
*/
|
|
#include <stratosphere.hpp>
|
|
#include "cs_command_impl.hpp"
|
|
|
|
namespace ams::cs {
|
|
|
|
namespace {
|
|
|
|
void SendEmptyData(const CommandDataTakeScreenShot ¶ms, size_t remaining_size) {
|
|
/* Clear the data buffer. */
|
|
std::memset(params.buffer, 0, params.buffer_size);
|
|
|
|
/* Send data until the end. */
|
|
while (remaining_size > 0) {
|
|
/* Send as much as we can. */
|
|
const auto cur_size = std::min(remaining_size, params.buffer_size);
|
|
params.send_data(params.buffer, cur_size);
|
|
|
|
/* Advance. */
|
|
remaining_size -= cur_size;
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
Result DoGetFirmwareVersionCommand(settings::system::FirmwareVersion *out) {
|
|
settings::system::GetFirmwareVersion(out);
|
|
return ResultSuccess();
|
|
}
|
|
|
|
Result DoTakeScreenShotCommand(const CommandDataTakeScreenShot ¶ms) {
|
|
/* Initialize screenshot control. */
|
|
R_TRY(capsrv::InitializeScreenShotControl());
|
|
|
|
/* Finalize screenshot control when we're done. */
|
|
ON_SCOPE_EXIT { capsrv::FinalizeScreenShotControl(); };
|
|
|
|
/* Open screenshot read stream. */
|
|
size_t data_size;
|
|
s32 width, height;
|
|
R_TRY(capsrv::OpenRawScreenShotReadStreamForDevelop(std::addressof(data_size), std::addressof(width), std::addressof(height), params.layer_stack, TimeSpan::FromSeconds(10)));
|
|
|
|
/* Close the screenshot stream when we're done. */
|
|
ON_SCOPE_EXIT { capsrv::CloseRawScreenShotReadStreamForDevelop(); };
|
|
|
|
/* Send the header. */
|
|
params.send_header(static_cast<s32>(data_size), width, height);
|
|
|
|
/* Read and send data. */
|
|
size_t total_read_size = 0;
|
|
auto data_guard = SCOPE_GUARD { SendEmptyData(params, data_size - total_read_size); };
|
|
while (total_read_size < data_size) {
|
|
/* Read data from the stream. */
|
|
size_t read_size;
|
|
R_TRY(capsrv::ReadRawScreenShotReadStreamForDevelop(std::addressof(read_size), params.buffer, params.buffer_size, total_read_size));
|
|
|
|
/* Send the data that was read. */
|
|
params.send_data(params.buffer, read_size);
|
|
|
|
/* Advance. */
|
|
total_read_size += read_size;
|
|
}
|
|
data_guard.Cancel();
|
|
|
|
return ResultSuccess();
|
|
}
|
|
|
|
}
|