external/ton
1
0
Fork 0
mirror of https://github.com/ton-blockchain/ton synced 2025-03-09 15:40:10 +00:00
Code Issues Releases Wiki Activity

Added udp msg size check

Browse source
This commit is contained in:
Oleg Baranov 2025-02-25 21:51:26 +04:00
parent 723de3cbad
commit 9ef4d929df
1 changed files with 7 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

10
tdnet/td/net/UdpServer.cpp
View file

@ -46,7 +46,7 @@ class UdpServerTunnelImpl : public UdpServer {
private:
td::Promise<td::IPAddress> on_ready_;
uint8_t out_buf_[(16+2+1500)*300];
uint8_t out_buf_[(16+2+1500)*100];
size_t out_buf_offset_ = 0;
size_t out_buf_msg_num_ = 0;
size_t tunnel_index_;
@ -75,13 +75,17 @@ void UdpServerTunnelImpl::send(td::UdpMessage &&message) {
out_buf_[out_buf_offset_] = static_cast<uint8_t>(sz >> 8);
out_buf_[out_buf_offset_ + 1] = static_cast<uint8_t>(sz & 0xff);
if (sz > 1500) {
LOG(WARNING) << "udp message is too big, dropping";
return;
}
memcpy(out_buf_ + out_buf_offset_ + 2, message.data.data(), sz);
out_buf_offset_ += 2 + sz;
out_buf_msg_num_++;
if (out_buf_msg_num_ >= 100) {
td::Timer timer;
if (out_buf_msg_num_ == 100) {
WriteTunnel(tunnel_index_, out_buf_, out_buf_msg_num_);
out_buf_offset_ = 0;