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 {
|
|
|
|
|
2015-11-05 07:45:03 +08:00
|
|
|
void sheet_protection::set_password(const string &password)
|
2014-05-21 22:20:30 +08:00
|
|
|
{
|
|
|
|
hashed_password_ = hash_password(password);
|
|
|
|
}
|
|
|
|
|
2015-11-01 22:43:01 +08:00
|
|
|
template <typename T>
|
2015-11-05 07:45:03 +08:00
|
|
|
string int_to_hex(T i)
|
2014-05-21 22:20:30 +08:00
|
|
|
{
|
|
|
|
std::stringstream stream;
|
|
|
|
stream << std::hex << i;
|
2015-11-05 07:45:03 +08:00
|
|
|
return stream.str().c_str();
|
2014-05-21 22:20:30 +08:00
|
|
|
}
|
|
|
|
|
2015-11-05 07:45:03 +08:00
|
|
|
string sheet_protection::hash_password(const string &plaintext_password)
|
2014-05-21 22:20:30 +08:00
|
|
|
{
|
|
|
|
int password = 0x0000;
|
2015-11-01 22:43:01 +08:00
|
|
|
|
2015-11-05 07:45:03 +08:00
|
|
|
for (int i = 1; i <= plaintext_password.bytes(); i++)
|
2014-05-21 22:20:30 +08:00
|
|
|
{
|
2015-11-05 07:45:03 +08:00
|
|
|
char character = plaintext_password.data()[i - 1];
|
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
|
|
|
|
2015-11-05 07:45:03 +08:00
|
|
|
password ^= plaintext_password.bytes();
|
2014-05-21 22:20:30 +08:00
|
|
|
password ^= 0xCE4B;
|
2015-11-01 22:43:01 +08:00
|
|
|
|
2015-11-05 07:45:03 +08:00
|
|
|
string hashed = int_to_hex(password);
|
|
|
|
auto iter = hashed.data();
|
|
|
|
|
|
|
|
while (iter != hashed.data() + hashed.bytes())
|
|
|
|
{
|
|
|
|
*iter = std::toupper(*iter, std::locale::classic());
|
|
|
|
}
|
|
|
|
|
2014-05-21 22:20:30 +08:00
|
|
|
return hashed;
|
|
|
|
}
|
2014-06-06 04:19:31 +08:00
|
|
|
}
|