mirror of
https://github.com/tfussell/xlnt.git
synced 2024-03-22 13:11:17 +08:00
50 lines
1.0 KiB
C++
50 lines
1.0 KiB
C++
#include <algorithm>
|
|
#include <locale>
|
|
#include <sstream>
|
|
|
|
#include <xlnt/worksheet/sheet_protection.hpp>
|
|
|
|
namespace xlnt {
|
|
|
|
void sheet_protection::set_password(const string &password)
|
|
{
|
|
hashed_password_ = hash_password(password);
|
|
}
|
|
|
|
template <typename T>
|
|
string int_to_hex(T i)
|
|
{
|
|
std::stringstream stream;
|
|
stream << std::hex << i;
|
|
return stream.str().c_str();
|
|
}
|
|
|
|
string sheet_protection::hash_password(const string &plaintext_password)
|
|
{
|
|
int password = 0x0000;
|
|
|
|
for (int i = 1; i <= plaintext_password.bytes(); i++)
|
|
{
|
|
char character = plaintext_password.data()[i - 1];
|
|
int value = character << i;
|
|
int rotated_bits = value >> 15;
|
|
value &= 0x7fff;
|
|
password ^= (value | rotated_bits);
|
|
i++;
|
|
}
|
|
|
|
password ^= plaintext_password.bytes();
|
|
password ^= 0xCE4B;
|
|
|
|
string hashed = int_to_hex(password);
|
|
auto iter = hashed.data();
|
|
|
|
while (iter != hashed.data() + hashed.bytes())
|
|
{
|
|
*iter = std::toupper(*iter, std::locale::classic());
|
|
}
|
|
|
|
return hashed;
|
|
}
|
|
}
|