some refactoring

This commit is contained in:
loki
2021-06-08 22:10:42 +02:00
parent c53564dd31
commit 61b195e9f4
4 changed files with 49 additions and 55 deletions
+18 -17
View File
@@ -331,7 +331,7 @@ void savePassword(resp_https_t response, req_https_t request) {
std::stringstream configStream; std::stringstream configStream;
ss << request->content.rdbuf(); ss << request->content.rdbuf();
pt::ptree inputTree,outputTree,fileTree; pt::ptree inputTree, outputTree, fileTree;
auto g = util::fail_guard([&]() { auto g = util::fail_guard([&]() {
std::ostringstream data; std::ostringstream data;
@@ -347,23 +347,24 @@ void savePassword(resp_https_t response, req_https_t request) {
std::string password = inputTree.get<std::string>("currentPassword"); std::string password = inputTree.get<std::string>("currentPassword");
std::string newPassword = inputTree.get<std::string>("newPassword"); std::string newPassword = inputTree.get<std::string>("newPassword");
std::string confirmPassword = inputTree.get<std::string>("confirmNewPassword"); std::string confirmPassword = inputTree.get<std::string>("confirmNewPassword");
if(newUsername.length() == 0) newUsername = username; if(newUsername.length() == 0) newUsername = username;
std::string hash = util::hex(crypto::hash(password + config::sunshine.salt)).to_string(); std::string hash = util::hex(crypto::hash(password + config::sunshine.salt)).to_string();
if(username == config::sunshine.username && hash == config::sunshine.password){ if(username == config::sunshine.username && hash == config::sunshine.password) {
if(newPassword != confirmPassword){ if(newPassword != confirmPassword) {
outputTree.put("status",false); outputTree.put("status", false);
outputTree.put("error","Password Mismatch"); outputTree.put("error", "Password Mismatch");
} }
fileTree.put("username",newUsername); fileTree.put("username", newUsername);
fileTree.put("password",util::hex(crypto::hash(newPassword + config::sunshine.salt)).to_string()); fileTree.put("password", util::hex(crypto::hash(newPassword + config::sunshine.salt)).to_string());
fileTree.put("salt",config::sunshine.salt); fileTree.put("salt", config::sunshine.salt);
pt::write_json(config::sunshine.credentials_file,fileTree); pt::write_json(config::sunshine.credentials_file, fileTree);
http::reload_user_creds(config::sunshine.credentials_file); http::reload_user_creds(config::sunshine.credentials_file);
outputTree.put("status",true); outputTree.put("status", true);
} else { }
outputTree.put("status",false); else {
outputTree.put("error","Invalid Current Credentials"); outputTree.put("status", false);
outputTree.put("error", "Invalid Current Credentials");
} }
} }
catch(std::exception &e) { catch(std::exception &e) {
@@ -374,13 +375,13 @@ void savePassword(resp_https_t response, req_https_t request) {
} }
} }
void savePin(resp_https_t response, req_https_t request){ void savePin(resp_https_t response, req_https_t request) {
if(!authenticate(response, request)) return; if(!authenticate(response, request)) return;
std::stringstream ss; std::stringstream ss;
ss << request->content.rdbuf(); ss << request->content.rdbuf();
pt::ptree inputTree,outputTree; pt::ptree inputTree, outputTree;
auto g = util::fail_guard([&]() { auto g = util::fail_guard([&]() {
std::ostringstream data; std::ostringstream data;
@@ -392,7 +393,7 @@ void savePin(resp_https_t response, req_https_t request){
//TODO: Input Validation //TODO: Input Validation
pt::read_json(ss, inputTree); pt::read_json(ss, inputTree);
std::string pin = inputTree.get<std::string>("pin"); std::string pin = inputTree.get<std::string>("pin");
outputTree.put("status",nvhttp::pin(pin)); outputTree.put("status", nvhttp::pin(pin));
} }
catch(std::exception &e) { catch(std::exception &e) {
BOOST_LOG(warning) << e.what(); BOOST_LOG(warning) << e.what();
+3 -11
View File
@@ -340,21 +340,13 @@ void md_ctx_destroy(EVP_MD_CTX *ctx) {
EVP_MD_CTX_destroy(ctx); EVP_MD_CTX_destroy(ctx);
} }
std::string rand_string(std::size_t bytes) { std::string rand_alphabet(std::size_t bytes, const std::string_view &alphabet) {
std::string alphabet = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789!%&()=-"; auto value = rand(bytes);
std::string value = rand(bytes);
for(std::size_t i = 0; i != value.size(); ++i) { for(std::size_t i = 0; i != value.size(); ++i) {
value[i] = alphabet[value[i] % alphabet.length()]; value[i] = alphabet[value[i] % alphabet.length()];
} }
return value; return value;
} }
std::string hash_hexstr(const std::string_view &plaintext) {
sha256_t hashBytes = crypto::hash(plaintext);
std::ostringstream hashStream;
hashStream << std::hex << std::setfill('0');
std::for_each(hashBytes.cbegin(), hashBytes.cend(), [&](int c) { hashStream << std::setw(2) << c; });
std::string hashString = hashStream.str();
return hashString;
}
} // namespace crypto } // namespace crypto
+3 -2
View File
@@ -36,7 +36,7 @@ using bio_t = util::safe_ptr<BIO, BIO_free_all>;
using pkey_t = util::safe_ptr<EVP_PKEY, EVP_PKEY_free>; using pkey_t = util::safe_ptr<EVP_PKEY, EVP_PKEY_free>;
sha256_t hash(const std::string_view &plaintext); sha256_t hash(const std::string_view &plaintext);
std::string hash_hexstr(const std::string_view &plaintext);
aes_t gen_aes_key(const std::array<uint8_t, 16> &salt, const std::string_view &pin); aes_t gen_aes_key(const std::array<uint8_t, 16> &salt, const std::string_view &pin);
x509_t x509(const std::string_view &x); x509_t x509(const std::string_view &x);
@@ -52,7 +52,8 @@ creds_t gen_creds(const std::string_view &cn, std::uint32_t key_bits);
std::string_view signature(const x509_t &x); std::string_view signature(const x509_t &x);
std::string rand(std::size_t bytes); std::string rand(std::size_t bytes);
std::string rand_string(std::size_t bytes); std::string rand_alphabet(std::size_t bytes,
const std::string_view &alphabet = std::string_view { "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789!%&()=-" });
class cert_chain_t { class cert_chain_t {
public: public:
+2 -2
View File
@@ -68,8 +68,8 @@ int generate_user_creds(const std::string &file) {
pt::ptree outputTree; pt::ptree outputTree;
try { try {
std::string username = "sunshine"; std::string username = "sunshine";
std::string plainPassword = crypto::rand_string(16); std::string plainPassword = crypto::rand_alphabet(16);
std::string salt = crypto::rand_string(16); std::string salt = crypto::rand_alphabet(16);
outputTree.put("username", "sunshine"); outputTree.put("username", "sunshine");
outputTree.put("salt", salt); outputTree.put("salt", salt);
outputTree.put("password", util::hex(crypto::hash(plainPassword + salt)).to_string()); outputTree.put("password", util::hex(crypto::hash(plainPassword + salt)).to_string());