110 lines
2.7 KiB
C++
110 lines
2.7 KiB
C++
// Copyright 2010-2025 Google LLC
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with 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.
|
|
|
|
#ifndef ORTOOLS_BASE_GZIPSTRING_H_
|
|
#define ORTOOLS_BASE_GZIPSTRING_H_
|
|
|
|
#include <zconf.h>
|
|
#include <zlib.h>
|
|
|
|
#include <string>
|
|
|
|
#include "absl/log/log.h"
|
|
#include "absl/strings/string_view.h"
|
|
|
|
inline bool GunzipString(absl::string_view str, std::string* out) {
|
|
z_stream zs;
|
|
zs.zalloc = Z_NULL;
|
|
zs.zfree = Z_NULL;
|
|
zs.opaque = Z_NULL;
|
|
zs.next_in = Z_NULL;
|
|
zs.avail_in = 0;
|
|
zs.next_out = Z_NULL;
|
|
if (inflateInit2(&zs, /*window_bits=*/15 + 32) != Z_OK) {
|
|
return false;
|
|
}
|
|
|
|
zs.next_in = (Bytef*)str.data();
|
|
zs.avail_in = str.size();
|
|
|
|
int status;
|
|
char buffer[32768];
|
|
|
|
// Decompress string by block.
|
|
do {
|
|
zs.next_out = reinterpret_cast<Bytef*>(buffer);
|
|
zs.avail_out = sizeof(buffer);
|
|
|
|
status = inflate(&zs, 0);
|
|
|
|
if (out->size() < zs.total_out) {
|
|
out->append(buffer, zs.total_out - out->size());
|
|
}
|
|
} while (status == Z_OK);
|
|
|
|
inflateEnd(&zs);
|
|
|
|
if (status != Z_STREAM_END) { // an error occurred that was not EOF
|
|
VLOG(1) << "Exception during zlib decompression: (" << status << ") "
|
|
<< zs.msg;
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
inline bool GzipString(absl::string_view uncompressed,
|
|
std::string* compressed) {
|
|
z_stream zs;
|
|
zs.zalloc = Z_NULL;
|
|
zs.zfree = Z_NULL;
|
|
zs.opaque = Z_NULL;
|
|
zs.next_in = Z_NULL;
|
|
zs.avail_in = 0;
|
|
zs.next_out = Z_NULL;
|
|
|
|
if (deflateInit(&zs, Z_BEST_COMPRESSION) != Z_OK) {
|
|
VLOG(1) << "Cannot initialize zlib compression.";
|
|
return false;
|
|
}
|
|
|
|
zs.next_in = (Bytef*)uncompressed.data();
|
|
zs.avail_in = uncompressed.size(); // set the z_stream's input
|
|
|
|
int status;
|
|
char buffer[32768];
|
|
|
|
// compress block by block.
|
|
do {
|
|
zs.next_out = reinterpret_cast<Bytef*>(buffer);
|
|
zs.avail_out = sizeof(buffer);
|
|
|
|
status = deflate(&zs, Z_FINISH);
|
|
|
|
if (compressed->size() < zs.total_out) {
|
|
compressed->append(buffer, zs.total_out - compressed->size());
|
|
}
|
|
} while (status == Z_OK);
|
|
|
|
deflateEnd(&zs);
|
|
|
|
if (status != Z_STREAM_END) { // an error occurred that was not EOF
|
|
VLOG(1) << "Exception during zlib compression: (" << status << ") "
|
|
<< zs.msg;
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
#endif // ORTOOLS_BASE_GZIPSTRING_H_
|