mirror of
https://github.com/openappsec/openappsec.git
synced 2025-11-15 08:52:16 +03:00
b74957d9d4
* fix ca loading for alpine * fix ca loading for alpine * fix ca loading for alpine * change gzipped detection * change gzipped detection --------- Co-authored-by: Daniel Eisenberg <danielei@checkpoint.com>
116 lines
3.3 KiB
C++
Executable File
116 lines
3.3 KiB
C++
Executable File
// Copyright (C) 2022 Check Point Software Technologies Ltd. All rights reserved.
|
|
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
#include "ParserGzip.h"
|
|
#include "debug.h"
|
|
|
|
USE_DEBUG_FLAG(D_WAAP_PARSER_GZIP);
|
|
|
|
const std::string ParserGzip::m_parserName = "ParserGzip";
|
|
|
|
ParserGzip::ParserGzip(IParserStreamReceiver &receiver, size_t parser_depth)
|
|
:m_receiver(receiver), m_key("gzip"), m_state(s_start), m_stream(nullptr) {
|
|
}
|
|
|
|
ParserGzip::~ParserGzip() {
|
|
if (m_stream != nullptr) {
|
|
finiCompressionStream(m_stream);
|
|
m_stream = nullptr;
|
|
}
|
|
}
|
|
|
|
size_t ParserGzip::push(const char *buf, size_t len) {
|
|
dbgTrace(D_WAAP_PARSER_GZIP) << "len=" << (unsigned long int)len << ")";
|
|
|
|
if (len == 0) {
|
|
dbgTrace(D_WAAP_PARSER_GZIP) << "end of data signal! m_state=" << m_state;
|
|
|
|
// flush
|
|
if (m_state != s_start) { // only emit if at least something was pushed
|
|
if (m_receiver.onKvDone() != 0) {
|
|
m_state = s_error;
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
DecompressionResult res;
|
|
switch (m_state) {
|
|
case s_start:
|
|
dbgTrace(D_WAAP_PARSER_GZIP) << "s_start";
|
|
if (m_receiver.onKey(m_key.data(), m_key.size()) != 0) {
|
|
m_state = s_error;
|
|
return 0;
|
|
}
|
|
m_stream = initCompressionStream();
|
|
m_state = s_forward;
|
|
// fallthrough //
|
|
CP_FALL_THROUGH;
|
|
case s_forward:
|
|
dbgTrace(D_WAAP_PARSER_GZIP) << "s_forward";
|
|
res = decompressData(
|
|
m_stream,
|
|
len,
|
|
reinterpret_cast<const unsigned char *>(buf));
|
|
dbgTrace(D_WAAP_PARSER_GZIP) << "res: " << res.ok
|
|
<< ", size: " << res.num_output_bytes
|
|
<< ", is last: " << res.is_last_chunk;
|
|
|
|
if (!res.ok) {
|
|
m_state = s_error;
|
|
break;
|
|
}
|
|
|
|
if (res.num_output_bytes != 0 &&
|
|
m_receiver.onValue(reinterpret_cast<const char *>(res.output), res.num_output_bytes) != 0) {
|
|
m_state = s_error;
|
|
break;
|
|
}
|
|
|
|
if (res.is_last_chunk) {
|
|
m_state = s_done;
|
|
break;
|
|
}
|
|
break;
|
|
case s_done:
|
|
if (len > 0) {
|
|
dbgTrace(D_WAAP_PARSER_GZIP) << " unexpected data after completion, len=" << len;
|
|
m_state = s_error;
|
|
return 0; // Return 0 to indicate error
|
|
}
|
|
break;
|
|
case s_error:
|
|
dbgTrace(D_WAAP_PARSER_GZIP) << "s_error";
|
|
return 0;
|
|
}
|
|
|
|
return len;
|
|
}
|
|
|
|
void ParserGzip::finish() {
|
|
push(NULL, 0);
|
|
if (m_state != s_done) {
|
|
m_state = s_error;
|
|
return;
|
|
}
|
|
}
|
|
|
|
const std::string &
|
|
ParserGzip::name() const {
|
|
return m_parserName;
|
|
}
|
|
|
|
bool ParserGzip::error() const {
|
|
return m_state == s_error;
|
|
}
|