mirror of
https://github.com/KingDuckZ/kamokan.git
synced 2025-01-13 19:56:40 +00:00
92 lines
2.8 KiB
C++
92 lines
2.8 KiB
C++
/* Copyright 2017, Michele Santullo
|
|
* This file is part of "tawashi".
|
|
*
|
|
* "tawashi" is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* "tawashi" is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with "tawashi". If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "escapist.hpp"
|
|
#include "houdini.h"
|
|
#include <cstddef>
|
|
#include <cassert>
|
|
|
|
namespace tawashi {
|
|
Escapist::Escapist() :
|
|
m_gh_buf(new(&m_gh_buf_mem) gh_buf GH_BUF_INIT)
|
|
{
|
|
assert(reinterpret_cast<uintptr_t>(&m_gh_buf_mem) == reinterpret_cast<uintptr_t>(m_gh_buf));
|
|
static_assert(sizeof(implem::DummyGHBuf) == sizeof(gh_buf), "Dummy struct has the wrong size");
|
|
static_assert(sizeof(gh_buf) == sizeof(m_gh_buf_mem), "Static memory for gh_buf has the wrong size");
|
|
static_assert(alignof(gh_buf) == alignof(m_gh_buf_mem), "Static memory for gh_buf has the wrong alignment");
|
|
}
|
|
|
|
Escapist::~Escapist() noexcept {
|
|
gh_buf_free(static_cast<gh_buf*>(m_gh_buf));
|
|
static_cast<gh_buf*>(m_gh_buf)->~gh_buf();
|
|
}
|
|
|
|
std::string Escapist::unescape_url (const boost::string_ref& parURL) const {
|
|
if (parURL.empty())
|
|
return std::string();
|
|
|
|
assert(m_gh_buf);
|
|
gh_buf* const buf = static_cast<gh_buf*>(m_gh_buf);
|
|
|
|
const int escaped = houdini_unescape_url(
|
|
buf,
|
|
reinterpret_cast<const uint8_t*>(parURL.data()),
|
|
parURL.size()
|
|
);
|
|
if (0 == escaped)
|
|
return std::string(parURL.data(), parURL.size());
|
|
else
|
|
return std::string(buf->ptr, buf->size);
|
|
}
|
|
|
|
std::string Escapist::escape_url (const boost::string_ref& parURL) const {
|
|
if (parURL.empty())
|
|
return std::string();
|
|
|
|
assert(m_gh_buf);
|
|
gh_buf* const buf = static_cast<gh_buf*>(m_gh_buf);
|
|
|
|
const int escaped = houdini_escape_url(
|
|
buf,
|
|
reinterpret_cast<const uint8_t*>(parURL.data()),
|
|
parURL.size()
|
|
);
|
|
if (0 == escaped)
|
|
return std::string(parURL.data(), parURL.size());
|
|
else
|
|
return std::string(buf->ptr, buf->size);
|
|
}
|
|
|
|
std::string Escapist::escape_html (const boost::string_ref& parHtml) const {
|
|
if (parHtml.empty())
|
|
return std::string();
|
|
|
|
assert(m_gh_buf);
|
|
gh_buf* const buf = static_cast<gh_buf*>(m_gh_buf);
|
|
|
|
const int escaped = houdini_escape_html0(
|
|
buf,
|
|
reinterpret_cast<const uint8_t*>(parHtml.data()),
|
|
parHtml.size(),
|
|
1
|
|
);
|
|
if (0 == escaped)
|
|
return std::string(parHtml.data(), parHtml.size());
|
|
else
|
|
return std::string(buf->ptr, buf->size);
|
|
}
|
|
} //namespace tawashi
|