mirror of
https://github.com/Huntereb/Awoo-Installer
synced 2024-11-23 12:13:11 +00:00
161 lines
No EOL
6.1 KiB
C++
Executable file
161 lines
No EOL
6.1 KiB
C++
Executable file
/*
|
|
Copyright (c) 2017-2018 Adubbz
|
|
|
|
Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
of this software and associated documentation files (the "Software"), to deal
|
|
in the Software without restriction, including without limitation the rights
|
|
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
copies of the Software, and to permit persons to whom the Software is
|
|
furnished to do so, subject to the following conditions:
|
|
|
|
The above copyright notice and this permission notice shall be included in all
|
|
copies or substantial portions of the Software.
|
|
|
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
SOFTWARE.
|
|
*/
|
|
|
|
#include "install/http_nsp.hpp"
|
|
|
|
#include <switch.h>
|
|
#include <threads.h>
|
|
#include "data/buffered_placeholder_writer.hpp"
|
|
#include "util/title_util.hpp"
|
|
#include "util/error.hpp"
|
|
#include "util/debug.h"
|
|
#include "nspInstall.hpp"
|
|
#include "util/util.hpp"
|
|
|
|
namespace tin::install::nsp
|
|
{
|
|
HTTPNSP::HTTPNSP(std::string url) :
|
|
m_download(url)
|
|
{
|
|
|
|
}
|
|
|
|
struct StreamFuncArgs
|
|
{
|
|
tin::network::HTTPDownload* download;
|
|
tin::data::BufferedPlaceholderWriter* bufferedPlaceholderWriter;
|
|
u64 pfs0Offset;
|
|
u64 ncaSize;
|
|
};
|
|
|
|
int CurlStreamFunc(void* in)
|
|
{
|
|
StreamFuncArgs* args = reinterpret_cast<StreamFuncArgs*>(in);
|
|
|
|
auto streamFunc = [&](u8* streamBuf, size_t streamBufSize) -> size_t
|
|
{
|
|
while (true)
|
|
{
|
|
if (args->bufferedPlaceholderWriter->CanAppendData(streamBufSize))
|
|
break;
|
|
}
|
|
|
|
args->bufferedPlaceholderWriter->AppendData(streamBuf, streamBufSize);
|
|
return streamBufSize;
|
|
};
|
|
|
|
args->download->StreamDataRange(args->pfs0Offset, args->ncaSize, streamFunc);
|
|
return 0;
|
|
}
|
|
|
|
int PlaceholderWriteFunc(void* in)
|
|
{
|
|
StreamFuncArgs* args = reinterpret_cast<StreamFuncArgs*>(in);
|
|
|
|
while (!args->bufferedPlaceholderWriter->IsPlaceholderComplete())
|
|
{
|
|
if (args->bufferedPlaceholderWriter->CanWriteSegmentToPlaceholder())
|
|
args->bufferedPlaceholderWriter->WriteSegmentToPlaceholder();
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
void HTTPNSP::StreamToPlaceholder(std::shared_ptr<nx::ncm::ContentStorage>& contentStorage, NcmContentId placeholderId)
|
|
{
|
|
const PFS0FileEntry* fileEntry = this->GetFileEntryByNcaId(placeholderId);
|
|
std::string ncaFileName = this->GetFileEntryName(fileEntry);
|
|
|
|
printf("Retrieving %s\n", ncaFileName.c_str());
|
|
size_t ncaSize = fileEntry->fileSize;
|
|
|
|
tin::data::BufferedPlaceholderWriter bufferedPlaceholderWriter(contentStorage, placeholderId, ncaSize);
|
|
StreamFuncArgs args;
|
|
args.download = &m_download;
|
|
args.bufferedPlaceholderWriter = &bufferedPlaceholderWriter;
|
|
args.pfs0Offset = this->GetDataOffset() + fileEntry->dataOffset;
|
|
args.ncaSize = ncaSize;
|
|
thrd_t curlThread;
|
|
thrd_t writeThread;
|
|
|
|
thrd_create(&curlThread, CurlStreamFunc, &args);
|
|
thrd_create(&writeThread, PlaceholderWriteFunc, &args);
|
|
|
|
u64 freq = armGetSystemTickFreq();
|
|
u64 startTime = armGetSystemTick();
|
|
size_t startSizeBuffered = 0;
|
|
double speed = 0.0;
|
|
|
|
//consoleUpdate(NULL);
|
|
|
|
inst::ui::setInstBarPerc(0);
|
|
while (!bufferedPlaceholderWriter.IsBufferDataComplete())
|
|
{
|
|
u64 newTime = armGetSystemTick();
|
|
|
|
if (newTime - startTime >= freq)
|
|
{
|
|
size_t newSizeBuffered = bufferedPlaceholderWriter.GetSizeBuffered();
|
|
double mbBuffered = (newSizeBuffered / 1000000.0) - (startSizeBuffered / 1000000.0);
|
|
double duration = ((double)(newTime - startTime) / (double)freq);
|
|
speed = mbBuffered / duration;
|
|
|
|
startTime = newTime;
|
|
startSizeBuffered = newSizeBuffered;
|
|
}
|
|
|
|
u64 totalSizeMB = bufferedPlaceholderWriter.GetTotalDataSize() / 1000000;
|
|
u64 downloadSizeMB = bufferedPlaceholderWriter.GetSizeBuffered() / 1000000;
|
|
int downloadProgress = (int)(((double)bufferedPlaceholderWriter.GetSizeBuffered() / (double)bufferedPlaceholderWriter.GetTotalDataSize()) * 100.0);
|
|
|
|
printf("> Download Progress: %lu/%lu MB (%i%s) (%.2f MB/s)\r", downloadSizeMB, totalSizeMB, downloadProgress, "%", speed);
|
|
inst::ui::setInstInfoText("Downloading " + inst::util::formatUrlString(ncaFileName) + " at " + std::to_string(speed).substr(0, std::to_string(speed).size()-4) + "MB/s");
|
|
inst::ui::setInstBarPerc((double)downloadProgress);
|
|
//consoleUpdate(NULL);
|
|
}
|
|
inst::ui::setInstBarPerc(100);
|
|
|
|
u64 totalSizeMB = bufferedPlaceholderWriter.GetTotalDataSize() / 1000000;
|
|
|
|
inst::ui::setInstInfoText("Installing " + ncaFileName + "...");
|
|
inst::ui::setInstBarPerc(0);
|
|
while (!bufferedPlaceholderWriter.IsPlaceholderComplete())
|
|
{
|
|
u64 installSizeMB = bufferedPlaceholderWriter.GetSizeWrittenToPlaceholder() / 1000000;
|
|
int installProgress = (int)(((double)bufferedPlaceholderWriter.GetSizeWrittenToPlaceholder() / (double)bufferedPlaceholderWriter.GetTotalDataSize()) * 100.0);
|
|
|
|
printf("> Install Progress: %lu/%lu MB (%i%s)\r", installSizeMB, totalSizeMB, installProgress, "%");
|
|
inst::ui::setInstBarPerc((double)installProgress);
|
|
//consoleUpdate(NULL);
|
|
}
|
|
inst::ui::setInstBarPerc(100);
|
|
|
|
thrd_join(curlThread, NULL);
|
|
thrd_join(writeThread, NULL);
|
|
//consoleUpdate(NULL);
|
|
}
|
|
|
|
void HTTPNSP::BufferData(void* buf, off_t offset, size_t size)
|
|
{
|
|
m_download.BufferDataRange(buf, offset, size, nullptr);
|
|
}
|
|
} |