/src/libpff/libpff/libpff_block_tree_node.c

Source (jump to first uncovered line)
/*
 * Block tree node functions
 *
 * Copyright (C) 2008-2024, Joachim Metz <joachim.metz@gmail.com>
 *
 * Refer to AUTHORS for acknowledgements.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */

#include <common.h>
#include <memory.h>
#include <types.h>

#include "libpff_block_tree_node.h"
#include "libpff_definitions.h"
#include "libpff_libcerror.h"

/* Creates a block tree node
 * Make sure the value block_tree_node is referencing, is set to NULL
 * Returns 1 if successful or -1 on error
 */
int libpff_block_tree_node_initialize(
     libpff_block_tree_node_t **block_tree_node,
     off64_t offset,
     size64_t size,
     size64_t leaf_value_size,
     libcerror_error_t **error )
{
  static char *function  = "libpff_block_tree_node_initialize";
  size64_t sub_node_size = 0;

  if( block_tree_node == NULL )
  {
    libcerror_error_set(
     error,
     LIBCERROR_ERROR_DOMAIN_ARGUMENTS,
     LIBCERROR_ARGUMENT_ERROR_INVALID_VALUE,
     "%s: invalid block tree node.",
     function );

    return( -1 );
  }
  if( *block_tree_node != NULL )
  {
    libcerror_error_set(
     error,
     LIBCERROR_ERROR_DOMAIN_RUNTIME,
     LIBCERROR_RUNTIME_ERROR_VALUE_ALREADY_SET,
     "%s: invalid block tree node value already set.",
     function );

    return( -1 );
  }
  if( offset < 0 )
  {
    libcerror_error_set(
     error,
     LIBCERROR_ERROR_DOMAIN_ARGUMENTS,
     LIBCERROR_ARGUMENT_ERROR_VALUE_OUT_OF_BOUNDS,
     "%s: invalid offset value out of bounds.",
     function );

    return( -1 );
  }
  if( ( size == 0 )
   || ( size > (size64_t) ( INT64_MAX - offset ) ) )
  {
    libcerror_error_set(
     error,
     LIBCERROR_ERROR_DOMAIN_ARGUMENTS,
     LIBCERROR_ARGUMENT_ERROR_VALUE_OUT_OF_BOUNDS,
     "%s: invalid size value out of bounds.",
     function );

    return( -1 );
  }
  if( leaf_value_size == 0 )
  {
    libcerror_error_set(
     error,
     LIBCERROR_ERROR_DOMAIN_ARGUMENTS,
     LIBCERROR_ARGUMENT_ERROR_VALUE_OUT_OF_BOUNDS,
     "%s: invalid leaf value size value out of bounds.",
     function );

    return( -1 );
  }
  *block_tree_node = memory_allocate_structure(
                      libpff_block_tree_node_t );

  if( *block_tree_node == NULL )
  {
    libcerror_error_set(
     error,
     LIBCERROR_ERROR_DOMAIN_MEMORY,
     LIBCERROR_MEMORY_ERROR_INSUFFICIENT,
     "%s: unable to create block tree node.",
     function );

    goto on_error;
  }
  if( memory_set(
       *block_tree_node,
       0,
       sizeof( libpff_block_tree_node_t ) ) == NULL )
  {
    libcerror_error_set(
     error,
     LIBCERROR_ERROR_DOMAIN_MEMORY,
     LIBCERROR_MEMORY_ERROR_SET_FAILED,
     "%s: unable to clear block tree node.",
     function );

    memory_free(
     *block_tree_node );

    *block_tree_node = NULL;

    return( -1 );
  }
  if( libcdata_array_initialize(
       &( ( *block_tree_node )->sub_nodes_array ),
       LIBPFF_BLOCK_TREE_NUMBER_OF_SUB_NODES,
       error ) != 1 )
  {
    libcerror_error_set(
     error,
     LIBCERROR_ERROR_DOMAIN_RUNTIME,
     LIBCERROR_RUNTIME_ERROR_INITIALIZE_FAILED,
     "%s: unable to create sub nodes array.",
     function );

    goto on_error;
  }
  sub_node_size = leaf_value_size;

  while( ( size / sub_node_size ) > LIBPFF_BLOCK_TREE_NUMBER_OF_SUB_NODES )
  {
    sub_node_size *= LIBPFF_BLOCK_TREE_NUMBER_OF_SUB_NODES;
  }
  ( *block_tree_node )->start_offset  = offset;
  ( *block_tree_node )->end_offset    = offset + size;
  ( *block_tree_node )->size          = size;
  ( *block_tree_node )->sub_node_size = sub_node_size;

  if( sub_node_size == leaf_value_size )
  {
    ( *block_tree_node )->is_leaf_node = 1;
  }
  return( 1 );

on_error:
  if( *block_tree_node != NULL )
  {
    memory_free(
     *block_tree_node );

    *block_tree_node = NULL;
  }
  return( -1 );
}

/* Frees a block tree node
 * Returns 1 if successful or -1 on error
 */
int libpff_block_tree_node_free(
     libpff_block_tree_node_t **block_tree_node,
     int (*value_free_function)(
            intptr_t **value,
            libcerror_error_t **error ),
     libcerror_error_t **error )
{
  libpff_block_tree_node_t *sub_block_tree_node = NULL;
  static char *function                         = "libpff_block_tree_node_free";
  int result                                    = 1;
  int sub_node_index                            = 0;

  if( block_tree_node == NULL )
  {
    libcerror_error_set(
     error,
     LIBCERROR_ERROR_DOMAIN_ARGUMENTS,
     LIBCERROR_ARGUMENT_ERROR_INVALID_VALUE,
     "%s: invalid block tree node.",
     function );

    return( -1 );
  }
  if( *block_tree_node != NULL )
  {
    if( ( *block_tree_node )->is_leaf_node != 0 )
    {
      if( libcdata_array_free(
           &( ( *block_tree_node )->sub_nodes_array ),
           value_free_function,
           error ) != 1 )
      {
        libcerror_error_set(
         error,
         LIBCERROR_ERROR_DOMAIN_RUNTIME,
         LIBCERROR_RUNTIME_ERROR_FINALIZE_FAILED,
         "%s: unable to free sub nodes array.",
         function );

        result = -1;
      }
    }
    else
    {
      for( sub_node_index = 0;
           sub_node_index < LIBPFF_BLOCK_TREE_NUMBER_OF_SUB_NODES;
           sub_node_index++ )
      {
        if( libcdata_array_get_entry_by_index(
             ( *block_tree_node )->sub_nodes_array,
             sub_node_index,
             (intptr_t **) &sub_block_tree_node,
             error ) != 1 )
        {
          libcerror_error_set(
           error,
           LIBCERROR_ERROR_DOMAIN_RUNTIME,
           LIBCERROR_RUNTIME_ERROR_GET_FAILED,
           "%s: unable to retrieve sub node: %d.",
           function,
           sub_node_index );

          result = -1;
        }
        if( libpff_block_tree_node_free(
             &sub_block_tree_node,
             value_free_function,
             error ) != 1 )
        {
          libcerror_error_set(
           error,
           LIBCERROR_ERROR_DOMAIN_RUNTIME,
           LIBCERROR_RUNTIME_ERROR_FINALIZE_FAILED,
           "%s: unable to free sub node: %d.",
           function,
           sub_node_index );

          result = -1;
        }
      }
      if( libcdata_array_free(
           &( ( *block_tree_node )->sub_nodes_array ),
           NULL,
           error ) != 1 )
      {
        libcerror_error_set(
         error,
         LIBCERROR_ERROR_DOMAIN_RUNTIME,
         LIBCERROR_RUNTIME_ERROR_FINALIZE_FAILED,
         "%s: unable to free sub nodes array.",
         function );

        result = -1;
      }
    }
    memory_free(
     *block_tree_node );

    *block_tree_node = NULL;
  }
  return( result );
}

/* Retrieves the sub node for a specific offset
 * Returns 1 if successful or -1 on error
 */
int libpff_block_tree_node_get_sub_node_at_offset(
     libpff_block_tree_node_t *block_tree_node,
     off64_t offset,
     libpff_block_tree_node_t **sub_block_tree_node,
     libcerror_error_t **error )
{
  static char *function  = "libpff_block_tree_node_get_sub_node_at_offset";
  off64_t sub_node_index = 0;

  if( block_tree_node == NULL )
  {
    libcerror_error_set(
     error,
     LIBCERROR_ERROR_DOMAIN_ARGUMENTS,
     LIBCERROR_ARGUMENT_ERROR_INVALID_VALUE,
     "%s: invalid block tree node.",
     function );

    return( -1 );
  }
  if( block_tree_node->is_leaf_node != 0 )
  {
    libcerror_error_set(
     error,
     LIBCERROR_ERROR_DOMAIN_ARGUMENTS,
     LIBCERROR_ARGUMENT_ERROR_INVALID_VALUE,
     "%s: invalid block tree node - not a branch node.",
     function );

    return( -1 );
  }
  if( ( offset < block_tree_node->start_offset )
   || ( offset >= block_tree_node->end_offset ) )
  {
    libcerror_error_set(
     error,
     LIBCERROR_ERROR_DOMAIN_ARGUMENTS,
     LIBCERROR_ARGUMENT_ERROR_VALUE_OUT_OF_BOUNDS,
     "%s: invalid offset value out of bounds.",
     function );

    return( -1 );
  }
  sub_node_index = ( offset - block_tree_node->start_offset ) / block_tree_node->sub_node_size;

  if( ( sub_node_index < 0 )
   || ( sub_node_index > (off64_t) INT_MAX ) )
  {
    libcerror_error_set(
     error,
     LIBCERROR_ERROR_DOMAIN_RUNTIME,
     LIBCERROR_RUNTIME_ERROR_VALUE_OUT_OF_BOUNDS,
     "%s: invalid sub node index value out of bounds.",
     function );

    return( -1 );
  }
  if( libcdata_array_get_entry_by_index(
       block_tree_node->sub_nodes_array,
       (int) sub_node_index,
       (intptr_t **) sub_block_tree_node,
       error ) != 1 )
  {
    libcerror_error_set(
     error,
     LIBCERROR_ERROR_DOMAIN_RUNTIME,
     LIBCERROR_RUNTIME_ERROR_GET_FAILED,
     "%s: unable to retrieve sub node: %" PRIi64 ".",
     function,
     sub_node_index );

    return( -1 );
  }
  return( 1 );
}

/* Sets the sub node for a specific offset
 * Returns 1 if successful or -1 on error
 */
int libpff_block_tree_node_set_sub_node_at_offset(
     libpff_block_tree_node_t *block_tree_node,
     off64_t offset,
     libpff_block_tree_node_t *sub_block_tree_node,
     libcerror_error_t **error )
{
  static char *function  = "libpff_block_tree_node_set_sub_node_at_offset";
  off64_t sub_node_index = 0;

  if( block_tree_node == NULL )
  {
    libcerror_error_set(
     error,
     LIBCERROR_ERROR_DOMAIN_ARGUMENTS,
     LIBCERROR_ARGUMENT_ERROR_INVALID_VALUE,
     "%s: invalid block tree node.",
     function );

    return( -1 );
  }
  if( block_tree_node->is_leaf_node != 0 )
  {
    libcerror_error_set(
     error,
     LIBCERROR_ERROR_DOMAIN_ARGUMENTS,
     LIBCERROR_ARGUMENT_ERROR_INVALID_VALUE,
     "%s: invalid block tree node - not a branch node.",
     function );

    return( -1 );
  }
  if( ( offset < block_tree_node->start_offset )
   || ( offset >= block_tree_node->end_offset ) )
  {
    libcerror_error_set(
     error,
     LIBCERROR_ERROR_DOMAIN_ARGUMENTS,
     LIBCERROR_ARGUMENT_ERROR_VALUE_OUT_OF_BOUNDS,
     "%s: invalid offset value out of bounds.",
     function );

    return( -1 );
  }
  sub_node_index = ( offset - block_tree_node->start_offset ) / block_tree_node->sub_node_size;

  if( ( sub_node_index < 0 )
   || ( sub_node_index > (off64_t) INT_MAX ) )
  {
    libcerror_error_set(
     error,
     LIBCERROR_ERROR_DOMAIN_RUNTIME,
     LIBCERROR_RUNTIME_ERROR_VALUE_OUT_OF_BOUNDS,
     "%s: invalid sub node index value out of bounds.",
     function );

    return( -1 );
  }
  if( libcdata_array_set_entry_by_index(
       block_tree_node->sub_nodes_array,
       (int) sub_node_index,
       (intptr_t *) sub_block_tree_node,
       error ) != 1 )
  {
    libcerror_error_set(
     error,
     LIBCERROR_ERROR_DOMAIN_RUNTIME,
     LIBCERROR_RUNTIME_ERROR_SET_FAILED,
     "%s: unable to set sub node: %" PRIi64 ".",
     function,
     sub_node_index );

    return( -1 );
  }
  return( 1 );
}

/* Retrieves the leaf value for a specific offset
 * Returns 1 if successful or -1 on error
 */
int libpff_block_tree_node_get_leaf_value_at_offset(
     libpff_block_tree_node_t *block_tree_node,
     off64_t offset,
     libpff_block_descriptor_t **block_descriptor,
     off64_t *block_offset,
     libcerror_error_t **error )
{
  static char *function     = "libpff_block_tree_node_get_leaf_value_at_offset";
  off64_t leaf_value_index  = 0;
  off64_t safe_block_offset = 0;

  if( block_tree_node == NULL )
  {
    libcerror_error_set(
     error,
     LIBCERROR_ERROR_DOMAIN_ARGUMENTS,
     LIBCERROR_ARGUMENT_ERROR_INVALID_VALUE,
     "%s: invalid block tree node.",
     function );

    return( -1 );
  }
  if( block_tree_node->is_leaf_node == 0 )
  {
    libcerror_error_set(
     error,
     LIBCERROR_ERROR_DOMAIN_ARGUMENTS,
     LIBCERROR_ARGUMENT_ERROR_INVALID_VALUE,
     "%s: invalid block tree node - not a leaf node.",
     function );

    return( -1 );
  }
  if( ( offset < block_tree_node->start_offset )
   || ( offset >= block_tree_node->end_offset ) )
  {
    libcerror_error_set(
     error,
     LIBCERROR_ERROR_DOMAIN_ARGUMENTS,
     LIBCERROR_ARGUMENT_ERROR_VALUE_OUT_OF_BOUNDS,
     "%s: invalid offset value out of bounds.",
     function );

    return( -1 );
  }
  if( block_offset == NULL )
  {
    libcerror_error_set(
     error,
     LIBCERROR_ERROR_DOMAIN_ARGUMENTS,
     LIBCERROR_ARGUMENT_ERROR_INVALID_VALUE,
     "%s: invalid block offset.",
     function );

    return( -1 );
  }
  safe_block_offset = offset - block_tree_node->start_offset;

  leaf_value_index = safe_block_offset / block_tree_node->sub_node_size;

  if( ( leaf_value_index < 0 )
   || ( leaf_value_index > (off64_t) INT_MAX ) )
  {
    libcerror_error_set(
     error,
     LIBCERROR_ERROR_DOMAIN_RUNTIME,
     LIBCERROR_RUNTIME_ERROR_VALUE_OUT_OF_BOUNDS,
     "%s: invalid leaf value index value out of bounds.",
     function );

    return( -1 );
  }
  if( libcdata_array_get_entry_by_index(
       block_tree_node->sub_nodes_array,
       (int) leaf_value_index,
       (intptr_t **) block_descriptor,
       error ) != 1 )
  {
    libcerror_error_set(
     error,
     LIBCERROR_ERROR_DOMAIN_RUNTIME,
     LIBCERROR_RUNTIME_ERROR_GET_FAILED,
     "%s: unable to retrieve leaf value: %" PRIi64 ".",
     function,
     leaf_value_index );

    return( -1 );
  }
  *block_offset = safe_block_offset - ( leaf_value_index * block_tree_node->sub_node_size );

  return( 1 );
}

/* Sets the leaf value for a specific offset
 * Returns 1 if successful or -1 on error
 */
int libpff_block_tree_node_set_leaf_value_at_offset(
     libpff_block_tree_node_t *block_tree_node,
     off64_t offset,
     libpff_block_descriptor_t *block_descriptor,
     libcerror_error_t **error )
{
  static char *function    = "libpff_block_tree_node_set_leaf_value_at_offset";
  off64_t leaf_value_index = 0;

  if( block_tree_node == NULL )
  {
    libcerror_error_set(
     error,
     LIBCERROR_ERROR_DOMAIN_ARGUMENTS,
     LIBCERROR_ARGUMENT_ERROR_INVALID_VALUE,
     "%s: invalid block tree node.",
     function );

    return( -1 );
  }
  if( block_tree_node->is_leaf_node == 0 )
  {
    libcerror_error_set(
     error,
     LIBCERROR_ERROR_DOMAIN_ARGUMENTS,
     LIBCERROR_ARGUMENT_ERROR_INVALID_VALUE,
     "%s: invalid block tree node - not a leaf node.",
     function );

    return( -1 );
  }
  if( ( offset < block_tree_node->start_offset )
   || ( offset >= block_tree_node->end_offset ) )
  {
    libcerror_error_set(
     error,
     LIBCERROR_ERROR_DOMAIN_ARGUMENTS,
     LIBCERROR_ARGUMENT_ERROR_VALUE_OUT_OF_BOUNDS,
     "%s: invalid offset value out of bounds.",
     function );

    return( -1 );
  }
  leaf_value_index = ( offset - block_tree_node->start_offset ) / block_tree_node->sub_node_size;

  if( ( leaf_value_index < 0 )
   || ( leaf_value_index > (off64_t) INT_MAX ) )
  {
    libcerror_error_set(
     error,
     LIBCERROR_ERROR_DOMAIN_RUNTIME,
     LIBCERROR_RUNTIME_ERROR_VALUE_OUT_OF_BOUNDS,
     "%s: invalid leaf value index value out of bounds.",
     function );

    return( -1 );
  }
  if( libcdata_array_set_entry_by_index(
       block_tree_node->sub_nodes_array,
       (int) leaf_value_index,
       (intptr_t *) block_descriptor,
       error ) != 1 )
  {
    libcerror_error_set(
     error,
     LIBCERROR_ERROR_DOMAIN_RUNTIME,
     LIBCERROR_RUNTIME_ERROR_SET_FAILED,
     "%s: unable to set leaf value: %" PRIi64 ".",
     function,
     leaf_value_index );

    return( -1 );
  }
  return( 1 );
}


Coverage Report

Created: 2024-10-02 06:58

Line	Count	Source (jump to first uncovered line)
1		/*
2		* Block tree node functions
3		*
4		* Copyright (C) 2008-2024, Joachim Metz <joachim.metz@gmail.com>
5		*
6		* Refer to AUTHORS for acknowledgements.
7		*
8		* This program is free software: you can redistribute it and/or modify
9		* it under the terms of the GNU Lesser General Public License as published by
10		* the Free Software Foundation, either version 3 of the License, or
11		* (at your option) any later version.
12		*
13		* This program is distributed in the hope that it will be useful,
14		* but WITHOUT ANY WARRANTY; without even the implied warranty of
15		* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16		* GNU General Public License for more details.
17		*
18		* You should have received a copy of the GNU Lesser General Public License
19		* along with this program. If not, see <https://www.gnu.org/licenses/>.
20		*/
21
22		#include <common.h>
23		#include <memory.h>
24		#include <types.h>
25
26		#include "libpff_block_tree_node.h"
27		#include "libpff_definitions.h"
28		#include "libpff_libcerror.h"
29
30		/* Creates a block tree node
31		* Make sure the value block_tree_node is referencing, is set to NULL
32		* Returns 1 if successful or -1 on error
33		*/
34		int libpff_block_tree_node_initialize(
35		libpff_block_tree_node_t **block_tree_node,
36		off64_t offset,
37		size64_t size,
38		size64_t leaf_value_size,
39		libcerror_error_t **error )
40	1.84M	{
41	1.84M	static char *function = "libpff_block_tree_node_initialize";
42	1.84M	size64_t sub_node_size = 0;
43
44	1.84M	if( block_tree_node == NULL )
45	0	{
46	0	libcerror_error_set(
47	0	error,
48	0	LIBCERROR_ERROR_DOMAIN_ARGUMENTS,
49	0	LIBCERROR_ARGUMENT_ERROR_INVALID_VALUE,
50	0	"%s: invalid block tree node.",
51	0	function );
52
53	0	return( -1 );
54	0	}
55	1.84M	if( *block_tree_node != NULL )
56	0	{
57	0	libcerror_error_set(
58	0	error,
59	0	LIBCERROR_ERROR_DOMAIN_RUNTIME,
60	0	LIBCERROR_RUNTIME_ERROR_VALUE_ALREADY_SET,
61	0	"%s: invalid block tree node value already set.",
62	0	function );
63
64	0	return( -1 );
65	0	}
66	1.84M	if( offset < 0 )
67	0	{
68	0	libcerror_error_set(
69	0	error,
70	0	LIBCERROR_ERROR_DOMAIN_ARGUMENTS,
71	0	LIBCERROR_ARGUMENT_ERROR_VALUE_OUT_OF_BOUNDS,
72	0	"%s: invalid offset value out of bounds.",
73	0	function );
74
75	0	return( -1 );
76	0	}
77	1.84M	if( ( size == 0 )
78	1.84M	\|\| ( size > (size64_t) ( INT64_MAX - offset ) ) )
79	21	{
80	21	libcerror_error_set(
81	21	error,
82	21	LIBCERROR_ERROR_DOMAIN_ARGUMENTS,
83	21	LIBCERROR_ARGUMENT_ERROR_VALUE_OUT_OF_BOUNDS,
84	21	"%s: invalid size value out of bounds.",
85	21	function );
86
87	21	return( -1 );
88	21	}
89	1.84M	if( leaf_value_size == 0 )
90	0	{
91	0	libcerror_error_set(
92	0	error,
93	0	LIBCERROR_ERROR_DOMAIN_ARGUMENTS,
94	0	LIBCERROR_ARGUMENT_ERROR_VALUE_OUT_OF_BOUNDS,
95	0	"%s: invalid leaf value size value out of bounds.",
96	0	function );
97
98	0	return( -1 );
99	0	}
100	1.84M	*block_tree_node = memory_allocate_structure(
101	1.84M	libpff_block_tree_node_t );
102
103	1.84M	if( *block_tree_node == NULL )
104	0	{
105	0	libcerror_error_set(
106	0	error,
107	0	LIBCERROR_ERROR_DOMAIN_MEMORY,
108	0	LIBCERROR_MEMORY_ERROR_INSUFFICIENT,
109	0	"%s: unable to create block tree node.",
110	0	function );
111
112	0	goto on_error;
113	0	}
114	1.84M	if( memory_set(
115	1.84M	*block_tree_node,
116	1.84M	0,
117	1.84M	sizeof( libpff_block_tree_node_t ) ) == NULL )
118	0	{
119	0	libcerror_error_set(
120	0	error,
121	0	LIBCERROR_ERROR_DOMAIN_MEMORY,
122	0	LIBCERROR_MEMORY_ERROR_SET_FAILED,
123	0	"%s: unable to clear block tree node.",
124	0	function );
125
126	0	memory_free(
127	0	*block_tree_node );
128
129	0	*block_tree_node = NULL;
130
131	0	return( -1 );
132	0	}
133	1.84M	if( libcdata_array_initialize(
134	1.84M	&( ( *block_tree_node )->sub_nodes_array ),
135	1.84M	LIBPFF_BLOCK_TREE_NUMBER_OF_SUB_NODES,
136	1.84M	error ) != 1 )
137	0	{
138	0	libcerror_error_set(
139	0	error,
140	0	LIBCERROR_ERROR_DOMAIN_RUNTIME,
141	0	LIBCERROR_RUNTIME_ERROR_INITIALIZE_FAILED,
142	0	"%s: unable to create sub nodes array.",
143	0	function );
144
145	0	goto on_error;
146	0	}
147	1.84M	sub_node_size = leaf_value_size;
148
149	3.48M	while( ( size / sub_node_size ) > LIBPFF_BLOCK_TREE_NUMBER_OF_SUB_NODES )
150	1.64M	{
151	1.64M	sub_node_size *= LIBPFF_BLOCK_TREE_NUMBER_OF_SUB_NODES;
152	1.64M	}
153	1.84M	( *block_tree_node )->start_offset = offset;
154	1.84M	( *block_tree_node )->end_offset = offset + size;
155	1.84M	( *block_tree_node )->size = size;
156	1.84M	( *block_tree_node )->sub_node_size = sub_node_size;
157
158	1.84M	if( sub_node_size == leaf_value_size )
159	711k	{
160	711k	( *block_tree_node )->is_leaf_node = 1;
161	711k	}
162	1.84M	return( 1 );
163
164	0	on_error:
165	0	if( *block_tree_node != NULL )
166	0	{
167	0	memory_free(
168	0	*block_tree_node );
169
170	0	*block_tree_node = NULL;
171	0	}
172	0	return( -1 );
173	1.84M	}
174
175		/* Frees a block tree node
176		* Returns 1 if successful or -1 on error
177		*/
178		int libpff_block_tree_node_free(
179		libpff_block_tree_node_t **block_tree_node,
180		int (*value_free_function)(
181		intptr_t **value,
182		libcerror_error_t **error ),
183		libcerror_error_t **error )
184	291M	{
185	291M	libpff_block_tree_node_t *sub_block_tree_node = NULL;
186	291M	static char *function = "libpff_block_tree_node_free";
187	291M	int result = 1;
188	291M	int sub_node_index = 0;
189
190	291M	if( block_tree_node == NULL )
191	0	{
192	0	libcerror_error_set(
193	0	error,
194	0	LIBCERROR_ERROR_DOMAIN_ARGUMENTS,
195	0	LIBCERROR_ARGUMENT_ERROR_INVALID_VALUE,
196	0	"%s: invalid block tree node.",
197	0	function );
198
199	0	return( -1 );
200	0	}
201	291M	if( *block_tree_node != NULL )
202	1.84M	{
203	1.84M	if( ( *block_tree_node )->is_leaf_node != 0 )
204	711k	{
205	711k	if( libcdata_array_free(
206	711k	&( ( *block_tree_node )->sub_nodes_array ),
207	711k	value_free_function,
208	711k	error ) != 1 )
209	0	{
210	0	libcerror_error_set(
211	0	error,
212	0	LIBCERROR_ERROR_DOMAIN_RUNTIME,
213	0	LIBCERROR_RUNTIME_ERROR_FINALIZE_FAILED,
214	0	"%s: unable to free sub nodes array.",
215	0	function );
216
217	0	result = -1;
218	0	}
219	711k	}
220	1.13M	else
221	1.13M	{
222	1.13M	for( sub_node_index = 0;
223	292M	sub_node_index < LIBPFF_BLOCK_TREE_NUMBER_OF_SUB_NODES;
224	290M	sub_node_index++ )
225	290M	{
226	290M	if( libcdata_array_get_entry_by_index(
227	290M	( *block_tree_node )->sub_nodes_array,
228	290M	sub_node_index,
229	290M	(intptr_t **) &sub_block_tree_node,
230	290M	error ) != 1 )
231	0	{
232	0	libcerror_error_set(
233	0	error,
234	0	LIBCERROR_ERROR_DOMAIN_RUNTIME,
235	0	LIBCERROR_RUNTIME_ERROR_GET_FAILED,
236	0	"%s: unable to retrieve sub node: %d.",
237	0	function,
238	0	sub_node_index );
239
240	0	result = -1;
241	0	}
242	290M	if( libpff_block_tree_node_free(
243	290M	&sub_block_tree_node,
244	290M	value_free_function,
245	290M	error ) != 1 )
246	0	{
247	0	libcerror_error_set(
248	0	error,
249	0	LIBCERROR_ERROR_DOMAIN_RUNTIME,
250	0	LIBCERROR_RUNTIME_ERROR_FINALIZE_FAILED,
251	0	"%s: unable to free sub node: %d.",
252	0	function,
253	0	sub_node_index );
254
255	0	result = -1;
256	0	}
257	290M	}
258	1.13M	if( libcdata_array_free(
259	1.13M	&( ( *block_tree_node )->sub_nodes_array ),
260	1.13M	NULL,
261	1.13M	error ) != 1 )
262	0	{
263	0	libcerror_error_set(
264	0	error,
265	0	LIBCERROR_ERROR_DOMAIN_RUNTIME,
266	0	LIBCERROR_RUNTIME_ERROR_FINALIZE_FAILED,
267	0	"%s: unable to free sub nodes array.",
268	0	function );
269
270	0	result = -1;
271	0	}
272	1.13M	}
273	1.84M	memory_free(
274	1.84M	*block_tree_node );
275
276	1.84M	*block_tree_node = NULL;
277	1.84M	}
278	291M	return( result );
279	291M	}
280
281		/* Retrieves the sub node for a specific offset
282		* Returns 1 if successful or -1 on error
283		*/
284		int libpff_block_tree_node_get_sub_node_at_offset(
285		libpff_block_tree_node_t *block_tree_node,
286		off64_t offset,
287		libpff_block_tree_node_t **sub_block_tree_node,
288		libcerror_error_t **error )
289	2.12M	{
290	2.12M	static char *function = "libpff_block_tree_node_get_sub_node_at_offset";
291	2.12M	off64_t sub_node_index = 0;
292
293	2.12M	if( block_tree_node == NULL )
294	0	{
295	0	libcerror_error_set(
296	0	error,
297	0	LIBCERROR_ERROR_DOMAIN_ARGUMENTS,
298	0	LIBCERROR_ARGUMENT_ERROR_INVALID_VALUE,
299	0	"%s: invalid block tree node.",
300	0	function );
301
302	0	return( -1 );
303	0	}
304	2.12M	if( block_tree_node->is_leaf_node != 0 )
305	0	{
306	0	libcerror_error_set(
307	0	error,
308	0	LIBCERROR_ERROR_DOMAIN_ARGUMENTS,
309	0	LIBCERROR_ARGUMENT_ERROR_INVALID_VALUE,
310	0	"%s: invalid block tree node - not a branch node.",
311	0	function );
312
313	0	return( -1 );
314	0	}
315	2.12M	if( ( offset < block_tree_node->start_offset )
316	2.12M	\|\| ( offset >= block_tree_node->end_offset ) )
317	565	{
318	565	libcerror_error_set(
319	565	error,
320	565	LIBCERROR_ERROR_DOMAIN_ARGUMENTS,
321	565	LIBCERROR_ARGUMENT_ERROR_VALUE_OUT_OF_BOUNDS,
322	565	"%s: invalid offset value out of bounds.",
323	565	function );
324
325	565	return( -1 );
326	565	}
327	2.12M	sub_node_index = ( offset - block_tree_node->start_offset ) / block_tree_node->sub_node_size;
328
329	2.12M	if( ( sub_node_index < 0 )
330	2.12M	\|\| ( sub_node_index > (off64_t) INT_MAX ) )
331	0	{
332	0	libcerror_error_set(
333	0	error,
334	0	LIBCERROR_ERROR_DOMAIN_RUNTIME,
335	0	LIBCERROR_RUNTIME_ERROR_VALUE_OUT_OF_BOUNDS,
336	0	"%s: invalid sub node index value out of bounds.",
337	0	function );
338
339	0	return( -1 );
340	0	}
341	2.12M	if( libcdata_array_get_entry_by_index(
342	2.12M	block_tree_node->sub_nodes_array,
343	2.12M	(int) sub_node_index,
344	2.12M	(intptr_t **) sub_block_tree_node,
345	2.12M	error ) != 1 )
346	0	{
347	0	libcerror_error_set(
348	0	error,
349	0	LIBCERROR_ERROR_DOMAIN_RUNTIME,
350	0	LIBCERROR_RUNTIME_ERROR_GET_FAILED,
351	0	"%s: unable to retrieve sub node: %" PRIi64 ".",
352	0	function,
353	0	sub_node_index );
354
355	0	return( -1 );
356	0	}
357	2.12M	return( 1 );
358	2.12M	}
359
360		/* Sets the sub node for a specific offset
361		* Returns 1 if successful or -1 on error
362		*/
363		int libpff_block_tree_node_set_sub_node_at_offset(
364		libpff_block_tree_node_t *block_tree_node,
365		off64_t offset,
366		libpff_block_tree_node_t *sub_block_tree_node,
367		libcerror_error_t **error )
368	1.13M	{
369	1.13M	static char *function = "libpff_block_tree_node_set_sub_node_at_offset";
370	1.13M	off64_t sub_node_index = 0;
371
372	1.13M	if( block_tree_node == NULL )
373	0	{
374	0	libcerror_error_set(
375	0	error,
376	0	LIBCERROR_ERROR_DOMAIN_ARGUMENTS,
377	0	LIBCERROR_ARGUMENT_ERROR_INVALID_VALUE,
378	0	"%s: invalid block tree node.",
379	0	function );
380
381	0	return( -1 );
382	0	}
383	1.13M	if( block_tree_node->is_leaf_node != 0 )
384	0	{
385	0	libcerror_error_set(
386	0	error,
387	0	LIBCERROR_ERROR_DOMAIN_ARGUMENTS,
388	0	LIBCERROR_ARGUMENT_ERROR_INVALID_VALUE,
389	0	"%s: invalid block tree node - not a branch node.",
390	0	function );
391
392	0	return( -1 );
393	0	}
394	1.13M	if( ( offset < block_tree_node->start_offset )
395	1.13M	\|\| ( offset >= block_tree_node->end_offset ) )
396	0	{
397	0	libcerror_error_set(
398	0	error,
399	0	LIBCERROR_ERROR_DOMAIN_ARGUMENTS,
400	0	LIBCERROR_ARGUMENT_ERROR_VALUE_OUT_OF_BOUNDS,
401	0	"%s: invalid offset value out of bounds.",
402	0	function );
403
404	0	return( -1 );
405	0	}
406	1.13M	sub_node_index = ( offset - block_tree_node->start_offset ) / block_tree_node->sub_node_size;
407
408	1.13M	if( ( sub_node_index < 0 )
409	1.13M	\|\| ( sub_node_index > (off64_t) INT_MAX ) )
410	0	{
411	0	libcerror_error_set(
412	0	error,
413	0	LIBCERROR_ERROR_DOMAIN_RUNTIME,
414	0	LIBCERROR_RUNTIME_ERROR_VALUE_OUT_OF_BOUNDS,
415	0	"%s: invalid sub node index value out of bounds.",
416	0	function );
417
418	0	return( -1 );
419	0	}
420	1.13M	if( libcdata_array_set_entry_by_index(
421	1.13M	block_tree_node->sub_nodes_array,
422	1.13M	(int) sub_node_index,
423	1.13M	(intptr_t *) sub_block_tree_node,
424	1.13M	error ) != 1 )
425	0	{
426	0	libcerror_error_set(
427	0	error,
428	0	LIBCERROR_ERROR_DOMAIN_RUNTIME,
429	0	LIBCERROR_RUNTIME_ERROR_SET_FAILED,
430	0	"%s: unable to set sub node: %" PRIi64 ".",
431	0	function,
432	0	sub_node_index );
433
434	0	return( -1 );
435	0	}
436	1.13M	return( 1 );
437	1.13M	}
438
439		/* Retrieves the leaf value for a specific offset
440		* Returns 1 if successful or -1 on error
441		*/
442		int libpff_block_tree_node_get_leaf_value_at_offset(
443		libpff_block_tree_node_t *block_tree_node,
444		off64_t offset,
445		libpff_block_descriptor_t **block_descriptor,
446		off64_t *block_offset,
447		libcerror_error_t **error )
448	1.33M	{
449	1.33M	static char *function = "libpff_block_tree_node_get_leaf_value_at_offset";
450	1.33M	off64_t leaf_value_index = 0;
451	1.33M	off64_t safe_block_offset = 0;
452
453	1.33M	if( block_tree_node == NULL )
454	0	{
455	0	libcerror_error_set(
456	0	error,
457	0	LIBCERROR_ERROR_DOMAIN_ARGUMENTS,
458	0	LIBCERROR_ARGUMENT_ERROR_INVALID_VALUE,
459	0	"%s: invalid block tree node.",
460	0	function );
461
462	0	return( -1 );
463	0	}
464	1.33M	if( block_tree_node->is_leaf_node == 0 )
465	0	{
466	0	libcerror_error_set(
467	0	error,
468	0	LIBCERROR_ERROR_DOMAIN_ARGUMENTS,
469	0	LIBCERROR_ARGUMENT_ERROR_INVALID_VALUE,
470	0	"%s: invalid block tree node - not a leaf node.",
471	0	function );
472
473	0	return( -1 );
474	0	}
475	1.33M	if( ( offset < block_tree_node->start_offset )
476	1.33M	\|\| ( offset >= block_tree_node->end_offset ) )
477	82	{
478	82	libcerror_error_set(
479	82	error,
480	82	LIBCERROR_ERROR_DOMAIN_ARGUMENTS,
481	82	LIBCERROR_ARGUMENT_ERROR_VALUE_OUT_OF_BOUNDS,
482	82	"%s: invalid offset value out of bounds.",
483	82	function );
484
485	82	return( -1 );
486	82	}
487	1.33M	if( block_offset == NULL )
488	0	{
489	0	libcerror_error_set(
490	0	error,
491	0	LIBCERROR_ERROR_DOMAIN_ARGUMENTS,
492	0	LIBCERROR_ARGUMENT_ERROR_INVALID_VALUE,
493	0	"%s: invalid block offset.",
494	0	function );
495
496	0	return( -1 );
497	0	}
498	1.33M	safe_block_offset = offset - block_tree_node->start_offset;
499
500	1.33M	leaf_value_index = safe_block_offset / block_tree_node->sub_node_size;
501
502	1.33M	if( ( leaf_value_index < 0 )
503	1.33M	\|\| ( leaf_value_index > (off64_t) INT_MAX ) )
504	0	{
505	0	libcerror_error_set(
506	0	error,
507	0	LIBCERROR_ERROR_DOMAIN_RUNTIME,
508	0	LIBCERROR_RUNTIME_ERROR_VALUE_OUT_OF_BOUNDS,
509	0	"%s: invalid leaf value index value out of bounds.",
510	0	function );
511
512	0	return( -1 );
513	0	}
514	1.33M	if( libcdata_array_get_entry_by_index(
515	1.33M	block_tree_node->sub_nodes_array,
516	1.33M	(int) leaf_value_index,
517	1.33M	(intptr_t **) block_descriptor,
518	1.33M	error ) != 1 )
519	0	{
520	0	libcerror_error_set(
521	0	error,
522	0	LIBCERROR_ERROR_DOMAIN_RUNTIME,
523	0	LIBCERROR_RUNTIME_ERROR_GET_FAILED,
524	0	"%s: unable to retrieve leaf value: %" PRIi64 ".",
525	0	function,
526	0	leaf_value_index );
527
528	0	return( -1 );
529	0	}
530	1.33M	block_offset = safe_block_offset - ( leaf_value_index block_tree_node->sub_node_size );
531
532	1.33M	return( 1 );
533	1.33M	}
534
535		/* Sets the leaf value for a specific offset
536		* Returns 1 if successful or -1 on error
537		*/
538		int libpff_block_tree_node_set_leaf_value_at_offset(
539		libpff_block_tree_node_t *block_tree_node,
540		off64_t offset,
541		libpff_block_descriptor_t *block_descriptor,
542		libcerror_error_t **error )
543	1.33M	{
544	1.33M	static char *function = "libpff_block_tree_node_set_leaf_value_at_offset";
545	1.33M	off64_t leaf_value_index = 0;
546
547	1.33M	if( block_tree_node == NULL )
548	0	{
549	0	libcerror_error_set(
550	0	error,
551	0	LIBCERROR_ERROR_DOMAIN_ARGUMENTS,
552	0	LIBCERROR_ARGUMENT_ERROR_INVALID_VALUE,
553	0	"%s: invalid block tree node.",
554	0	function );
555
556	0	return( -1 );
557	0	}
558	1.33M	if( block_tree_node->is_leaf_node == 0 )
559	0	{
560	0	libcerror_error_set(
561	0	error,
562	0	LIBCERROR_ERROR_DOMAIN_ARGUMENTS,
563	0	LIBCERROR_ARGUMENT_ERROR_INVALID_VALUE,
564	0	"%s: invalid block tree node - not a leaf node.",
565	0	function );
566
567	0	return( -1 );
568	0	}
569	1.33M	if( ( offset < block_tree_node->start_offset )
570	1.33M	\|\| ( offset >= block_tree_node->end_offset ) )
571	0	{
572	0	libcerror_error_set(
573	0	error,
574	0	LIBCERROR_ERROR_DOMAIN_ARGUMENTS,
575	0	LIBCERROR_ARGUMENT_ERROR_VALUE_OUT_OF_BOUNDS,
576	0	"%s: invalid offset value out of bounds.",
577	0	function );
578
579	0	return( -1 );
580	0	}
581	1.33M	leaf_value_index = ( offset - block_tree_node->start_offset ) / block_tree_node->sub_node_size;
582
583	1.33M	if( ( leaf_value_index < 0 )
584	1.33M	\|\| ( leaf_value_index > (off64_t) INT_MAX ) )
585	0	{
586	0	libcerror_error_set(
587	0	error,
588	0	LIBCERROR_ERROR_DOMAIN_RUNTIME,
589	0	LIBCERROR_RUNTIME_ERROR_VALUE_OUT_OF_BOUNDS,
590	0	"%s: invalid leaf value index value out of bounds.",
591	0	function );
592
593	0	return( -1 );
594	0	}
595	1.33M	if( libcdata_array_set_entry_by_index(
596	1.33M	block_tree_node->sub_nodes_array,
597	1.33M	(int) leaf_value_index,
598	1.33M	(intptr_t *) block_descriptor,
599	1.33M	error ) != 1 )
600	0	{
601	0	libcerror_error_set(
602	0	error,
603	0	LIBCERROR_ERROR_DOMAIN_RUNTIME,
604	0	LIBCERROR_RUNTIME_ERROR_SET_FAILED,
605	0	"%s: unable to set leaf value: %" PRIi64 ".",
606	0	function,
607	0	leaf_value_index );
608
609	0	return( -1 );
610	0	}
611	1.33M	return( 1 );
612	1.33M	}
613