2014-05-22 05:48:51 +08:00
|
|
|
#include <algorithm>
|
|
|
|
#include <locale>
|
2014-05-21 22:20:30 +08:00
|
|
|
#include <sstream>
|
|
|
|
|
2014-08-14 06:56:34 +08:00
|
|
|
#include <xlnt/worksheet/sheet_protection.hpp>
|
2014-05-21 22:20:30 +08:00
|
|
|
|
|
|
|
namespace xlnt {
|
|
|
|
|
|
|
|
void sheet_protection::set_password(const std::string &password)
|
|
|
|
{
|
|
|
|
hashed_password_ = hash_password(password);
|
|
|
|
}
|
|
|
|
|
2015-11-01 22:43:01 +08:00
|
|
|
template <typename T>
|
2014-05-21 22:20:30 +08:00
|
|
|
std::string int_to_hex(T i)
|
|
|
|
{
|
|
|
|
std::stringstream stream;
|
|
|
|
stream << std::hex << i;
|
|
|
|
return stream.str();
|
|
|
|
}
|
|
|
|
|
|
|
|
std::string sheet_protection::hash_password(const std::string &plaintext_password)
|
|
|
|
{
|
|
|
|
int password = 0x0000;
|
|
|
|
int i = 1;
|
2015-11-01 22:43:01 +08:00
|
|
|
|
|
|
|
for (auto character : plaintext_password)
|
2014-05-21 22:20:30 +08:00
|
|
|
{
|
|
|
|
int value = character << i;
|
|
|
|
int rotated_bits = value >> 15;
|
|
|
|
value &= 0x7fff;
|
|
|
|
password ^= (value | rotated_bits);
|
|
|
|
i++;
|
|
|
|
}
|
2015-11-01 22:43:01 +08:00
|
|
|
|
2014-05-21 22:20:30 +08:00
|
|
|
password ^= plaintext_password.size();
|
|
|
|
password ^= 0xCE4B;
|
2015-11-01 22:43:01 +08:00
|
|
|
|
2014-05-21 22:20:30 +08:00
|
|
|
std::string hashed = int_to_hex(password);
|
2015-11-01 22:43:01 +08:00
|
|
|
std::transform(hashed.begin(), hashed.end(), hashed.begin(),
|
|
|
|
[](char c) { return std::toupper(c, std::locale::classic()); });
|
2014-05-21 22:20:30 +08:00
|
|
|
return hashed;
|
|
|
|
}
|
2014-06-06 04:19:31 +08:00
|
|
|
}
|