generic-library/libzmq
1
0
Fork 0
mirror of https://github.com/zeromq/libzmq.git synced 2025-03-01 10:57:59 +01:00
Code Issues Projects Releases Wiki Activity

Problem: radix tree's apply function uses incorrect resizing logic (#3548)

Browse Source 
* Problem: radix tree's apply function uses incorrect resizing logic

Solution: rewrite the function using a vector

The buffer is extended once by 256 bytes, which will not be enough if
the node holds data larger than this number.
This commit is contained in:
Shubham Lagwankar 2019-06-20 02:58:08 -04:00 committed by Luca Boccassi
parent b17e854f15
commit 15dafb1c1c
1 changed files with 17 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
src/radix_tree.cpp
View File

@ -34,6 +34,7 @@
#include <stdlib.h>
#include <string.h>
#include <vector>
node::node (unsigned char *data) : data_ (data)
{
@ -517,35 +518,33 @@ bool zmq::radix_tree::check (const unsigned char *key, size_t size)
static void
visit_keys (node n,
unsigned char **buffer,
size_t buffer_size,
size_t maxbuffer_size,
std::vector<unsigned char> &buffer,
void (*func) (unsigned char *data, size_t size, void *arg),
void *arg)
{
if (buffer_size >= maxbuffer_size) {
maxbuffer_size += 256;
*buffer =
static_cast<unsigned char *> (realloc (*buffer, maxbuffer_size));
zmq_assert (*buffer);
for (size_t i = 0; i < n.prefix_length (); ++i)
buffer.push_back (n.prefix ()[i]);
if (n.refcount () > 0) {
zmq_assert (!buffer.empty ());
func (&buffer[0], buffer.size (), arg);
}
for (size_t i = 0; i < n.prefix_length (); ++i)
(*buffer)[buffer_size++] = n.prefix ()[i];
if (n.refcount () > 0)
func (*buffer, buffer_size, arg);
for (size_t i = 0; i < n.edgecount (); ++i)
visit_keys (n.node_at (i), buffer, buffer_size, maxbuffer_size, func,
arg);
buffer_size -= n.prefix_length ();
visit_keys (n.node_at (i), buffer, func, arg);
for (size_t i = 0; i < n.prefix_length (); ++i)
buffer.pop_back ();
}
void zmq::radix_tree::apply (
void (*func) (unsigned char *data, size_t size, void *arg), void *arg)
{
unsigned char *buffer = NULL;
visit_keys (root_, &buffer, 0, 0, func, arg);
free (buffer);
if (root_.refcount () > 0)
func (NULL, 0, arg); // Root node is always empty.
std::vector<unsigned char> buffer;
for (size_t i = 0; i < root_.edgecount (); ++i)
visit_keys (root_.node_at (i), buffer, func, arg);
}
size_t zmq::radix_tree::size () const