/*

 * Data stream functions

 *

 * Copyright (C) 2018-2025, 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 <types.h>

#include "libfsapfs_buffer_data_handle.h"

#include "libfsapfs_compressed_data_handle.h"

#include "libfsapfs_data_block_data_handle.h"

#include "libfsapfs_data_stream.h"

#include "libfsapfs_encryption_context.h"

#include "libfsapfs_file_extent.h"

#include "libfsapfs_io_handle.h"

#include "libfsapfs_libcdata.h"

#include "libfsapfs_libfdata.h"

/* Creates data stream from a buffer of data

 * Make sure the value data_stream is referencing, is set to NULL

 * Returns 1 if successful or -1 on error

 */

int libfsapfs_data_stream_initialize_from_data(

     libfdata_stream_t **data_stream,

     const uint8_t *data,

     size_t data_size,

     libcerror_error_t **error )

{

  libfdata_stream_t *safe_data_stream         = NULL;

  libfsapfs_buffer_data_handle_t *data_handle = NULL;

  static char *function                       = "libfsapfs_data_stream_initialize_from_data";

  int segment_index                           = 0;

  if( data_stream == NULL )

  {

    libcerror_error_set(

     error,

     LIBCERROR_ERROR_DOMAIN_ARGUMENTS,

     LIBCERROR_ARGUMENT_ERROR_INVALID_VALUE,

     "%s: invalid data stream.",

     function );

    return( -1 );

  }

  if( libfsapfs_buffer_data_handle_initialize(

       &data_handle,

       data,

       data_size,

       error ) != 1 )

  {

    libcerror_error_set(

     error,

     LIBCERROR_ERROR_DOMAIN_RUNTIME,

     LIBCERROR_RUNTIME_ERROR_INITIALIZE_FAILED,

     "%s: unable to create buffer data handle.",

     function );

    goto on_error;

  }

  if( libfdata_stream_initialize(

       &safe_data_stream,

       (intptr_t *) data_handle,

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

       NULL,

       NULL,

       (ssize_t (*)(intptr_t *, intptr_t *, int, int, uint8_t *, size_t, uint32_t, uint8_t, libcerror_error_t **)) &libfsapfs_buffer_data_handle_read_segment_data,

       NULL,

       (off64_t (*)(intptr_t *, intptr_t *, int, int, off64_t, libcerror_error_t **)) &libfsapfs_buffer_data_handle_seek_segment_offset,

       LIBFDATA_DATA_HANDLE_FLAG_MANAGED,

       error ) != 1 )

  {

    libcerror_error_set(

     error,

     LIBCERROR_ERROR_DOMAIN_RUNTIME,

     LIBCERROR_RUNTIME_ERROR_INITIALIZE_FAILED,

     "%s: unable to create data stream.",

     function );

    goto on_error;

  }

  data_handle = NULL;

  if( libfdata_stream_append_segment(

       safe_data_stream,

       &segment_index,

       0,

       0,

       (size64_t) data_size,

       0,

       error ) != 1 )

  {

    libcerror_error_set(

     error,

     LIBCERROR_ERROR_DOMAIN_RUNTIME,

     LIBCERROR_RUNTIME_ERROR_APPEND_FAILED,

     "%s: unable to append data stream segment.",

     function );

    goto on_error;

  }

  *data_stream = safe_data_stream;

  return( 1 );

on_error:

  if( safe_data_stream != NULL )

  {

    libfdata_stream_free(

     &safe_data_stream,

     NULL );

  }

  if( data_handle != NULL )

  {

    libfsapfs_buffer_data_handle_free(

     &data_handle,

     NULL );

  }

  return( -1 );

}

/* Creates data stream from file extents

 * Make sure the value data_stream is referencing, is set to NULL

 * Returns 1 if successful or -1 on error

 */

int libfsapfs_data_stream_initialize_from_file_extents(

     libfdata_stream_t **data_stream,

     libfsapfs_io_handle_t *io_handle,

     libfsapfs_encryption_context_t *encryption_context,

     libcdata_array_t *file_extents,

     size64_t data_stream_size,

     uint8_t is_sparse,

     libcerror_error_t **error )

{

  libfdata_stream_t *safe_data_stream             = NULL;

  libfsapfs_data_block_data_handle_t *data_handle = NULL;

  static char *function                           = "libfsapfs_data_stream_initialize_from_file_extents";

  int segment_index                               = 0;

  if( data_stream == NULL )

  {

    libcerror_error_set(

     error,

     LIBCERROR_ERROR_DOMAIN_ARGUMENTS,

     LIBCERROR_ARGUMENT_ERROR_INVALID_VALUE,

     "%s: invalid data stream.",

     function );

    return( -1 );

  }

  if( io_handle == NULL )

  {

    libcerror_error_set(

     error,

     LIBCERROR_ERROR_DOMAIN_ARGUMENTS,

     LIBCERROR_ARGUMENT_ERROR_INVALID_VALUE,

     "%s: invalid IO handle.",

     function );

    return( -1 );

  }

  if( libfsapfs_data_block_data_handle_initialize(

       &data_handle,

       io_handle,

       encryption_context,

       file_extents,

       is_sparse,

       error ) != 1 )

  {

    libcerror_error_set(

     error,

     LIBCERROR_ERROR_DOMAIN_RUNTIME,

     LIBCERROR_RUNTIME_ERROR_INITIALIZE_FAILED,

     "%s: unable to create data handle.",

     function );

    goto on_error;

  }

  if( libfdata_stream_initialize(

       &safe_data_stream,

       (intptr_t *) data_handle,

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

       NULL,

       NULL,

       (ssize_t (*)(intptr_t *, intptr_t *, int, int, uint8_t *, size_t, uint32_t, uint8_t, libcerror_error_t **)) &libfsapfs_data_block_data_handle_read_segment_data,

       NULL,

       (off64_t (*)(intptr_t *, intptr_t *, int, int, off64_t, libcerror_error_t **)) &libfsapfs_data_block_data_handle_seek_segment_offset,

       LIBFDATA_DATA_HANDLE_FLAG_MANAGED,

       error ) != 1 )

  {

    libcerror_error_set(

     error,

     LIBCERROR_ERROR_DOMAIN_RUNTIME,

     LIBCERROR_RUNTIME_ERROR_INITIALIZE_FAILED,

     "%s: unable to create data stream.",

     function );

    goto on_error;

  }

  data_handle = NULL;

  if( libfdata_stream_append_segment(

       safe_data_stream,

       &segment_index,

       0,

       0,

       data_stream_size,

       0,

       error ) != 1 )

  {

    libcerror_error_set(

     error,

     LIBCERROR_ERROR_DOMAIN_RUNTIME,

     LIBCERROR_RUNTIME_ERROR_APPEND_FAILED,

     "%s: unable to append data stream segment.",

     function );

    goto on_error;

  }

  *data_stream = safe_data_stream;

  return( 1 );

on_error:

  if( safe_data_stream != NULL )

  {

    libfdata_stream_free(

     &safe_data_stream,

     NULL );

  }

  if( data_handle != NULL )

  {

    libfsapfs_data_block_data_handle_free(

     &data_handle,

     NULL );

  }

  return( -1 );

}

/* Creates data stream from a compressed data stream

 * Make sure the value data_stream is referencing, is set to NULL

 * Returns 1 if successful or -1 on error

 */

int libfsapfs_data_stream_initialize_from_compressed_data_stream(

     libfdata_stream_t **data_stream,

     libfdata_stream_t *compressed_data_stream,

     size64_t uncompressed_data_size,

     int compression_method,

     libcerror_error_t **error )

{

  libfdata_stream_t *safe_data_stream             = NULL;

  libfsapfs_compressed_data_handle_t *data_handle = NULL;

  static char *function                           = "libfsapfs_data_stream_initialize_from_compressed_data_stream";

  int segment_index                               = 0;

  if( data_stream == NULL )

  {

    libcerror_error_set(

     error,

     LIBCERROR_ERROR_DOMAIN_ARGUMENTS,

     LIBCERROR_ARGUMENT_ERROR_INVALID_VALUE,

     "%s: invalid data stream.",

     function );

    return( -1 );

  }

  if( libfsapfs_compressed_data_handle_initialize(

       &data_handle,

       compressed_data_stream,

       uncompressed_data_size,

       compression_method,

       error ) != 1 )

  {

    libcerror_error_set(

     error,

     LIBCERROR_ERROR_DOMAIN_RUNTIME,

     LIBCERROR_RUNTIME_ERROR_INITIALIZE_FAILED,

     "%s: unable to create compressed data handle.",

     function );

    goto on_error;

  }

  if( libfdata_stream_initialize(

       &safe_data_stream,

       (intptr_t *) data_handle,

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

       NULL,

       NULL,

       (ssize_t (*)(intptr_t *, intptr_t *, int, int, uint8_t *, size_t, uint32_t, uint8_t, libcerror_error_t **)) &libfsapfs_compressed_data_handle_read_segment_data,

       NULL,

       (off64_t (*)(intptr_t *, intptr_t *, int, int, off64_t, libcerror_error_t **)) &libfsapfs_compressed_data_handle_seek_segment_offset,

       LIBFDATA_DATA_HANDLE_FLAG_MANAGED,

       error ) != 1 )

  {

    libcerror_error_set(

     error,

     LIBCERROR_ERROR_DOMAIN_RUNTIME,

     LIBCERROR_RUNTIME_ERROR_INITIALIZE_FAILED,

     "%s: unable to create data stream.",

     function );

    goto on_error;

  }

  data_handle = NULL;

  if( libfdata_stream_append_segment(

       safe_data_stream,

       &segment_index,

       0,

       0,

       uncompressed_data_size,

       LIBFDATA_RANGE_FLAG_IS_COMPRESSED,

       error ) != 1 )

  {

    libcerror_error_set(

     error,

     LIBCERROR_ERROR_DOMAIN_RUNTIME,

     LIBCERROR_RUNTIME_ERROR_APPEND_FAILED,

     "%s: unable to append data as data stream segment.",

     function );

    goto on_error;

  }

  *data_stream = safe_data_stream;

  return( 1 );

on_error:

  if( safe_data_stream != NULL )

  {

    libfdata_stream_free(

     &safe_data_stream,

     NULL );

  }

  if( data_handle != NULL )

  {

    libfsapfs_compressed_data_handle_free(

     &data_handle,

     NULL );

  }

  return( -1 );

}