Home
About
Blog
Products
Forum
Support
Contact
Sunbelt Computer Software
PL/B Language Development and Support
Home
About
Blog
Products
Forum
Support
Contact
tonk/PacketCompression.cpp at master · ForkStorage/tonk · GitHub
Skip to content
Navigation Menu
Sign in
Appearance settings
Platform
AI CODE CREATION
GitHub Copilot
Write better code with AI
GitHub Copilot app
Direct agents from issue to merge
MCP Registry
Integrate external tools
DEVELOPER WORKFLOWS
Actions
Automate any workflow
Codespaces
Instant dev environments
Issues
Plan and track work
Code Review
Manage code changes
Code Quality
Enforce quality at merge
APPLICATION SECURITY
GitHub Advanced Security
Find and fix vulnerabilities
Code security
Secure your code as you build
Secret protection
Stop leaks before they start
EXPLORE
Why GitHub
Documentation
Blog
Changelog
Marketplace
View all features
Solutions
BY COMPANY SIZE
Enterprises
Small and medium teams
Startups
Nonprofits
BY USE CASE
App Modernization
DevSecOps
DevOps
CI/CD
View all use cases
BY INDUSTRY
Healthcare
Financial services
Manufacturing
Government
View all industries
View all solutions
Resources
EXPLORE BY TOPIC
AI
Software Development
DevOps
Security
View all topics
EXPLORE BY TYPE
Customer stories
Events & webinars
Ebooks & reports
Business insights
GitHub Skills
SUPPORT & SERVICES
Documentation
Customer support
Community forum
Trust center
Partners
View all resources
Open Source
COMMUNITY
GitHub Sponsors
Fund open source developers
PROGRAMS
Security Lab
Maintainer Community
GitHub Stars
Archive Program
REPOSITORIES
Topics
Trending
Collections
Enterprise
ENTERPRISE SOLUTIONS
Enterprise platform
AI-powered developer platform
AVAILABLE ADD-ONS
GitHub Advanced Security
Enterprise-grade security features
Copilot for Business
Enterprise-grade AI features
Premium Support
Enterprise-grade 24/7 support
Pricing
Search
/
Sign in
Sign up
Appearance settings
You signed in with another tab or window.
Reload
to refresh your session.
You signed out in another tab or window.
Reload
to refresh your session.
You switched accounts on another tab or window.
Reload
to refresh your session.
Dismiss alert
{{ message }}
Uh oh!
There was an error while loading.
Please reload this page
.
ForkStorage
/
tonk
Public
forked from
catid/tonk
Notifications
You must be signed in to change notification settings
Fork
0
Star
0
Code
Pull requests
0
Actions
Projects
Security and quality
0
Insights
Additional navigation options
Code
Pull requests
Actions
Projects
Security and quality
Insights
Files
Expand file tree
master
Breadcrumbs
tonk
/
PacketCompression.cpp
Copy path
Blame
More file actions
Blame
More file actions
Latest commit
History
History
History
216 lines (171 loc) · 6.55 KB
master
Breadcrumbs
tonk
/
PacketCompression.cpp
Copy path
Top
File metadata and controls
Code
Blame
216 lines (171 loc) · 6.55 KB
Raw
Copy raw file
Download raw file
Open symbols panel
Edit and raw actions
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
/*
* \file
\brief Tonk: Packet Compression
\copyright Copyright (c) 2018 Christopher A. Taylor. All rights reserved.
Redistribution and use in source and binary forms, with or without
modification, are permitted provided that the following conditions are met:
* Redistributions of source code must retain the above copyright notice,
this list of conditions and the following disclaimer.
* Redistributions in binary form must reproduce the above copyright notice,
this list of conditions and the following disclaimer in the documentation
and/or other materials provided with the distribution.
* Neither the name of Tonk nor the names of its contributors may be
used to endorse or promote products derived from this software without
specific prior written permission.
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
POSSIBILITY OF SUCH DAMAGE.
*/
#
include
"
PacketCompression.h
"
namespace
tonk
{
//
------------------------------------------------------------------------------
//
MessageCompressor
Result
MessageCompressor::Initialize
(
unsigned
maxCompressedMessagesBytes)
{
MaxCompressedMessagesBytes = maxCompressedMessagesBytes;
CCtx =
ZSTD_createCCtx
();
if
(!CCtx) {
return
Result
(
"
SessionOutgoing::Initialize
"
,
"
ZSTD_createCCtx failed
"
, ErrorType::Zstd);
}
const
size_t
estimatedPacketSize = MaxCompressedMessagesBytes;
ZSTD_parameters zParams;
zParams.
cParams
=
ZSTD_getCParams
(
kCompressionLevel
,
estimatedPacketSize,
kCompressionDictBytes
);
zParams.
fParams
.
checksumFlag
=
0
;
zParams.
fParams
.
contentSizeFlag
=
0
;
zParams.
fParams
.
noDictIDFlag
=
1
;
const
size_t
icsResult =
ZSTD_compressBegin_advanced
(
CCtx,
nullptr
,
0
,
zParams,
ZSTD_CONTENTSIZE_UNKNOWN
);
if
(
0
!=
ZSTD_isError
(icsResult)) {
TONK_DEBUG_BREAK
();
return
Result
(
"
SessionOutgoing::Initialize
"
,
"
ZSTD_compressBegin_advanced failed
"
, ErrorType::Zstd, icsResult);
}
const
size_t
blockSizeBytes =
ZSTD_getBlockSize
(CCtx);
if
(blockSizeBytes < MaxCompressedMessagesBytes) {
return
Result
(
"
SessionOutgoing::Initialize
"
,
"
Zstd block size is too small
"
, ErrorType::Zstd);
}
return
Result::Success
();
}
MessageCompressor::~MessageCompressor
()
{
if
(CCtx) {
ZSTD_freeCCtx
(CCtx);
}
}
Result
MessageCompressor::Compress
(
const
uint8_t
* data,
unsigned
bytes,
uint8_t
* destBuffer,
unsigned
& writtenBytes)
{
TONK_DEBUG_ASSERT
(bytes >
0
&& destBuffer && data);
writtenBytes =
0
;
//
Insert data into history ring buffer
TONK_DEBUG_ASSERT
(MaxCompressedMessagesBytes >= bytes);
void
* history = History.
Allocate
(MaxCompressedMessagesBytes);
memcpy
(history, data, bytes);
History.
Commit
(bytes);
//
Compress into scratch buffer, leaving room for a frame header
const
size_t
result =
ZSTD_compressBlock
(
CCtx,
destBuffer,
MaxCompressedMessagesBytes,
history,
bytes);
//
If no data to compress, or would require too much space,
//
or did not produce a small enough result:
if
(
0
== result ||
(
size_t
)-ZSTD_error_dstSize_tooSmall == result ||
result >= bytes)
{
//
Note: Input data was accumulated into history ring buffer
return
Result::Success
();
}
//
If compression failed:
if
(
0
!=
ZSTD_isError
(result))
{
std::string reason =
"
ZSTD_compressBlock failed:
"
;
reason +=
ZSTD_getErrorName
(result);
TONK_DEBUG_BREAK
();
return
Result
(
"
SessionOutgoing::compress
"
, reason, ErrorType::Zstd, result);
}
writtenBytes =
static_cast
<
unsigned
>(result);
TONK_DEBUG_ASSERT
(writtenBytes <= MaxCompressedMessagesBytes);
return
Result::Success
();
}
//
------------------------------------------------------------------------------
//
MessageDecompressor
Result
MessageDecompressor::Initialize
(
unsigned
maxCompressedMessagesBytes)
{
MaxCompressedMessagesBytes = maxCompressedMessagesBytes;
DCtx =
ZSTD_createDCtx
();
if
(!DCtx) {
return
Result
(
"
SessionIncoming::Initialize
"
,
"
ZSTD_createDCtx failed
"
, ErrorType::Zstd);
}
const
size_t
beginResult =
ZSTD_decompressBegin
(DCtx);
if
(
0
!=
ZSTD_isError
(beginResult)) {
return
Result
(
"
SessionIncoming::Initialize
"
,
"
ZSTD_decompressBegin failed
"
, ErrorType::Zstd, beginResult);
}
return
Result::Success
();
}
MessageDecompressor::~MessageDecompressor
()
{
if
(DCtx) {
ZSTD_freeDCtx
(DCtx);
}
}
void
MessageDecompressor::InsertUncompressed
(
const
uint8_t
* data,
unsigned
bytes)
{
if
(bytes > MaxCompressedMessagesBytes) {
TONK_DEBUG_BREAK
();
//
Invalid input
return
;
}
void
* history = History.
Allocate
(MaxCompressedMessagesBytes);
memcpy
(history, data, bytes);
ZSTD_insertBlock
(DCtx, history, bytes);
History.
Commit
(bytes);
}
Result
MessageDecompressor::Decompress
(
const
void
* data,
unsigned
bytes,
Decompressed& decompressed)
{
//
Decompress data into history ring buffer
void
* history = History.
Allocate
(MaxCompressedMessagesBytes);
const
size_t
result =
ZSTD_decompressBlock
(
DCtx,
history,
MaxCompressedMessagesBytes,
data,
bytes);
//
If decompression failed:
if
(
0
== result ||
0
!=
ZSTD_isError
(result))
{
std::string reason =
"
ZSTD_decompressBlock failed:
"
;
reason +=
ZSTD_getErrorName
(result);
TONK_DEBUG_BREAK
();
return
Result
(
"
SessionOutgoing::decompress
"
, reason, ErrorType::Zstd, result);
}
const
uint8_t
* datagramData =
reinterpret_cast
<
uint8_t
*>(history);
const
unsigned
datagramBytes =
static_cast
<
unsigned
>(result);
History.
Commit
(datagramBytes);
decompressed.
Data
= datagramData;
decompressed.
Bytes
= datagramBytes;
return
Result::Success
();
}
}
//
namespace tonk
You can’t perform that action at this time.