SaraP
513f08e691
- aggiornamento versione - utilizzate le nostre librerie esterne libzip e zlib e non quelle incluse in lib3mf.
71 lines
2.6 KiB
C++
71 lines
2.6 KiB
C++
|
|
|
|
#include "Common/Platform/NMR_EncryptionHeader.h"
|
|
#include "Common/NMR_ErrorConst.h"
|
|
#include "Common/NMR_Exception.h"
|
|
|
|
#include "Common/Platform/NMR_ImportStream.h"
|
|
#include "Common/Platform/NMR_ExportStream.h"
|
|
|
|
#include "Common/NMR_Architecture_Utils.h"
|
|
#include <cstdint>
|
|
#include <cstring>
|
|
|
|
#define LIB3MF_MAXENCRYPTIONHEADERSIZE (1UL << 31)
|
|
|
|
namespace NMR {
|
|
|
|
CEncryptionHeader::CEncryptionHeader(std::vector<nfByte> const & additionalData)
|
|
:m_rgAdditionalData(additionalData)
|
|
{
|
|
}
|
|
|
|
size_t CEncryptionHeader::readFrom(PImportStream from) {
|
|
uEncryptedFileHeader header = { { 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 } };
|
|
from->readIntoBuffer(header.bytes, sizeof(header), true);
|
|
if (isBigEndian()) {
|
|
header.Header.Length.length = swapBytes(header.Header.Length.length);
|
|
}
|
|
constexpr size_t sigSize = sizeof(header.Header.Signature.bytes);
|
|
if (strncmp((char *)header.Header.Signature.bytes, "%3McF", sigSize) != 0)
|
|
throw CNMRException(NMR_ERROR_COULDNOTREADENCRYPTEDSTREAM);
|
|
if (header.Header.majorVersion != 0 || header.Header.minorVersion != 0)
|
|
throw CNMRException(NMR_ERROR_ENCRYPTEDCONTENTVERSIONUNSUPPORTED);
|
|
constexpr size_t headerSize = sizeof(header);
|
|
|
|
if (header.Header.Length.length < (uint64_t) headerSize)
|
|
throw CNMRException(NMR_ERROR_COULDNOTREADENCRYPTEDSTREAM);
|
|
if (header.Header.Length.length > LIB3MF_MAXENCRYPTIONHEADERSIZE)
|
|
throw CNMRException(NMR_ERROR_COULDNOTREADENCRYPTEDSTREAM);
|
|
|
|
nfUint32 remainingBytes = header.Header.Length.length - headerSize;
|
|
if (remainingBytes > 0) {
|
|
m_rgAdditionalData.resize(remainingBytes, 0);
|
|
from->readIntoBuffer(m_rgAdditionalData.data(), remainingBytes, true);
|
|
}
|
|
m_nfHeaderSize = (size_t) header.Header.Length.length;
|
|
return (size_t) m_nfHeaderSize;
|
|
}
|
|
size_t CEncryptionHeader::writeTo(PExportStream to) {
|
|
uEncryptedFileHeader header = { { '%', '3', 'M', 'c', 'F', 0, 0, 0, 0, 0, 0, 0 } };
|
|
constexpr size_t headerSize = sizeof(header);
|
|
m_nfHeaderSize = (nfUint32)(headerSize + m_rgAdditionalData.size());
|
|
if (m_nfHeaderSize > LIB3MF_MAXENCRYPTIONHEADERSIZE)
|
|
throw CNMRException(NMR_ERROR_COULDNOTWRITESTREAM);
|
|
|
|
header.Header.Length.length = (nfUint32)m_nfHeaderSize;
|
|
if (isBigEndian()) {
|
|
header.Header.Length.length = swapBytes(header.Header.Length.length);
|
|
}
|
|
to->writeBuffer(header.bytes, headerSize);
|
|
if (m_rgAdditionalData.size() > 0)
|
|
to->writeBuffer(m_rgAdditionalData.data(), m_rgAdditionalData.size());
|
|
return (size_t)m_nfHeaderSize;
|
|
}
|
|
std::vector<nfByte> const & CEncryptionHeader::additionalData() const {
|
|
return m_rgAdditionalData;
|
|
}
|
|
nfUint64 CEncryptionHeader::headerSize() const {
|
|
return m_nfHeaderSize;
|
|
}
|
|
} |