1
0
Fork 0
mirror of https://github.com/KingDuckZ/kamokan.git synced 2024-11-23 00:33:44 +00:00

Reject pastie tokens that are not in the a-z range.

This commit is contained in:
King_DuckZ 2017-06-08 10:15:22 +01:00
parent c3609e1768
commit 5b88afb276
4 changed files with 34 additions and 5 deletions

View file

@ -29,6 +29,7 @@ namespace tawashi {
MissingPostVariable,
PastieNotFound,
InvalidContentType,
UnsupportedContentType
UnsupportedContentType,
InvalidToken
)
} //namespace tawashi

View file

@ -52,7 +52,8 @@ namespace tawashi {
"Request is missing a POST variable.",
"Pastie not found.",
"Invalid CONTENT_TYPE.",
"Unsupported CONTENT_TYPE."
"Unsupported CONTENT_TYPE.",
"Invalid pastie token."
};
constexpr const auto lengths = string_lengths(err_descs);
static_assert(err_descs.static_size == lengths.static_size, "Mismatching array sizes between strings and their lengths");

View file

@ -20,6 +20,7 @@
#include "settings_bag.hpp"
#include "escapist.hpp"
#include "cgi_env.hpp"
#include "spdlog.hpp"
#include <ciso646>
#include <srchilite/sourcehighlight.h>
#include <srchilite/langmap.h>
@ -68,6 +69,25 @@ namespace tawashi {
})
);
}
bool is_valid_token (const boost::string_view& parToken) {
if (parToken.empty())
return false;
auto it_mark = std::find(parToken.begin(), parToken.end(), '?');
if (parToken.begin() == it_mark)
return false;
for (auto it_ch = parToken.begin(); it_ch != it_mark; ++it_ch) {
if (*it_ch < 'a' or *it_ch > 'z') {
spdlog::get("statuslog")->info(
"Token's byte {} is invalid; value={}",
it_ch - parToken.begin(),
static_cast<int>(*it_ch)
);
return false;
}
}
return true;
}
} //unnamed namespace
PastieResponse::PastieResponse (
@ -79,14 +99,16 @@ namespace tawashi {
m_langmap_dir(parSettings->as<std::string>("langmap_dir")),
m_plain_text(false),
m_syntax_highlight(true),
m_pastie_not_found(false)
m_pastie_not_found(false),
m_token_invalid(false)
{
}
HttpHeader PastieResponse::on_process() {
if (m_pastie_not_found) {
if (m_pastie_not_found)
return make_error_redirect(ErrorReasons::PastieNotFound);
}
if (m_token_invalid)
return make_error_redirect(ErrorReasons::InvalidToken);
auto get = cgi_env().query_string_split();
const std::string& query_str(cgi_env().query_string());
@ -113,6 +135,10 @@ namespace tawashi {
boost::string_view token = get_pastie_name(cgi_env().request_uri_relative());
boost::optional<std::string> pastie = this->storage().retrieve_pastie(token);
if (not is_valid_token(token)) {
m_token_invalid = true;
return;
}
if (not pastie) {
m_pastie_not_found = true;
return;

View file

@ -43,5 +43,6 @@ namespace tawashi {
bool m_plain_text;
bool m_syntax_highlight;
bool m_pastie_not_found;
bool m_token_invalid;
};
} //namespace tawashi