mirror of
https://github.com/Dark98/threeSD.git
synced 2026-07-03 00:38:58 +00:00
zhupengfei
78 lines
2.3 KiB
C++
78 lines
2.3 KiB
C++
// Copyright 2019 threeSD Project
|
|
// Licensed under GPLv2 or any later version
|
|
// Refer to the license.txt file included.
|
|
|
|
#pragma once
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
#include "common/common_types.h"
|
|
#include "common/file_util.h"
|
|
#include "common/logging/log.h"
|
|
#include "core/quick_decryptor.h"
|
|
|
|
namespace Core {
|
|
|
|
class SDMCDecryptor {
|
|
public:
|
|
/**
|
|
* Initializes the decryptor.
|
|
* @param root_folder Path to the "Nintendo 3DS/<ID0>/<ID1>" folder.
|
|
*/
|
|
explicit SDMCDecryptor(const std::string& root_folder);
|
|
|
|
~SDMCDecryptor();
|
|
|
|
/**
|
|
* Decrypts a file from the SD card and writes it into another file.
|
|
* This function blocks, but can be aborted with the Abort() function (would return false)
|
|
*
|
|
* @param source Path to the file relative to the root folder, starting with "/".
|
|
* @param destination Path to the destination file.
|
|
* @return true on success, false otherwise
|
|
*/
|
|
bool DecryptAndWriteFile(const std::string& source, const std::string& destination,
|
|
const ProgressCallback& callback = [](std::size_t, std::size_t) {});
|
|
|
|
void Abort();
|
|
|
|
/**
|
|
* Decrypts a file and reads it into a vector.
|
|
* @param source Path to the file relative to the root folder, starting with "/".
|
|
*/
|
|
std::vector<u8> DecryptFile(const std::string& source) const;
|
|
|
|
/**
|
|
* Marks the beginning of a new content, resetting imported_size counter, and setting an new
|
|
* total_size for the next content.
|
|
* This doesn't affect at all how the contents will be imported, but will make sure the callback
|
|
* is properly invoked.
|
|
*/
|
|
void Reset(std::size_t total_size);
|
|
|
|
private:
|
|
std::string root_folder;
|
|
QuickDecryptor quick_decryptor;
|
|
};
|
|
|
|
/// Interface for reading an SDMC file like a normal IOFile. This is read-only.
|
|
class SDMCFile : public FileUtil::IOFile {
|
|
public:
|
|
SDMCFile(std::string root_folder, const std::string& filename, const char openmode[],
|
|
int flags = 0);
|
|
|
|
~SDMCFile() override;
|
|
|
|
std::size_t Read(char* data, std::size_t length) override;
|
|
std::size_t Write(const char* data, std::size_t length) override;
|
|
bool Seek(s64 off, int origin) override;
|
|
|
|
private:
|
|
void DecryptData(u8* data, std::size_t size);
|
|
|
|
struct Impl;
|
|
std::unique_ptr<Impl> impl;
|
|
};
|
|
|
|
} // namespace Core
|