/*

 * Extent list functions

 *

 * Copyright (C) 2020-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 "libfsxfs_definitions.h"

#include "libfsxfs_extent.h"

#include "libfsxfs_extent_list.h"

#include "libfsxfs_extents.h"

#include "libfsxfs_libcdata.h"

#include "libfsxfs_libcerror.h"

#include "libfsxfs_libcnotify.h"

/* Reads the extent list data

 * Returns 1 if successful or -1 on error

 */

int libfsxfs_extent_list_read_data(

     libcdata_array_t *extents_array,

     uint64_t number_of_blocks,

     uint32_t number_of_extents,

     const uint8_t *data,

     size_t data_size,

     uint8_t add_sparse_extents,

     libcerror_error_t **error )

{

  libfsxfs_extent_t *last_extent   = NULL;

  libfsxfs_extent_t *sparse_extent = NULL;

  static char *function            = "libfsxfs_extent_list_read_data";

  uint64_t logical_block_number    = 0;

  int entry_index                  = 0;

  if( libfsxfs_extents_read_data(

       extents_array,

       number_of_extents,

       data,

       data_size,

       add_sparse_extents,

       error ) != 1 )

  {

    libcerror_error_set(

     error,

     LIBCERROR_ERROR_DOMAIN_IO,

     LIBCERROR_IO_ERROR_READ_FAILED,

     "%s: unable to read data extents.",

     function );

    goto on_error;

  }

  if( libfsxfs_extents_get_last_extent(

       extents_array,

       &last_extent,

       error ) != 1 )

  {

    libcerror_error_set(

     error,

     LIBCERROR_ERROR_DOMAIN_RUNTIME,

     LIBCERROR_RUNTIME_ERROR_GET_FAILED,

     "%s: unable to retrieve last extent.",

     function );

    goto on_error;

  }

  if( last_extent != NULL )

  {

    logical_block_number = last_extent->logical_block_number + last_extent->number_of_blocks;

  }

  if( ( add_sparse_extents != 0 )

   && ( logical_block_number < number_of_blocks ) )

  {

    if( ( last_extent == NULL )

     || ( ( last_extent->range_flags & LIBFSXFS_EXTENT_FLAG_IS_SPARSE ) == 0 ) )

    {

      if( libfsxfs_extent_initialize(

           &sparse_extent,

           error ) != 1 )

      {

        libcerror_error_set(

         error,

         LIBCERROR_ERROR_DOMAIN_RUNTIME,

         LIBCERROR_RUNTIME_ERROR_INITIALIZE_FAILED,

         "%s: unable to create sparse extent.",

         function );

        goto on_error;

      }

      sparse_extent->logical_block_number = logical_block_number;

      sparse_extent->range_flags          = LIBFSXFS_EXTENT_FLAG_IS_SPARSE;

      if( libcdata_array_append_entry(

           extents_array,

           &entry_index,

           (intptr_t *) sparse_extent,

           error ) != 1 )

      {

        libcerror_error_set(

         error,

         LIBCERROR_ERROR_DOMAIN_RUNTIME,

         LIBCERROR_RUNTIME_ERROR_APPEND_FAILED,

         "%s: unable to append sparse extent to array.",

         function );

        goto on_error;

      }

      last_extent   = sparse_extent;

      sparse_extent = NULL;

    }

    last_extent->number_of_blocks += number_of_blocks - logical_block_number;

#if defined( HAVE_DEBUG_OUTPUT )

    if( libcnotify_verbose != 0 )

    {

      libcnotify_printf(

       "%s: logical block number\t\t\t: %" PRIu64 "\n",

       function,

       last_extent->logical_block_number );

      libcnotify_printf(

       "%s: physical block number\t\t\t: %" PRIu64 "\n",

       function,

       last_extent->physical_block_number );

      libcnotify_printf(

       "%s: number of blocks\t\t\t: %" PRIu64 "\n",

       function,

       last_extent->number_of_blocks );

      libcnotify_printf(

       "\n" );

    }

#endif /* defined( HAVE_DEBUG_OUTPUT ) */

  }

  return( 1 );

on_error:

  if( sparse_extent != NULL )

  {

    libfsxfs_extent_free(

     &sparse_extent,

     NULL );

  }

  libcdata_array_empty(

   extents_array,

   (int (*)(intptr_t **, libcerror_error_t **)) &libfsxfs_extent_free,

   NULL );

  return( -1 );

}