/*

 * Task functions

 *

 * Copyright (C) 2011-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 <byte_stream.h>

#include <memory.h>

#include <narrow_string.h>

#include <system_string.h>

#include <types.h>

#include <wide_string.h>

#include "libfwevt_debug.h"

#include "libfwevt_libcerror.h"

#include "libfwevt_libcnotify.h"

#include "libfwevt_libfguid.h"

#include "libfwevt_libuna.h"

#include "libfwevt_task.h"

#include "libfwevt_types.h"

#include "fwevt_template.h"

/* Creates a task

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

 * Returns 1 if successful or -1 on error

 */

int libfwevt_task_initialize(

     libfwevt_task_t **task,

     libcerror_error_t **error )

{

  libfwevt_internal_task_t *internal_task = NULL;

  static char *function                   = "libfwevt_task_initialize";

  if( task == NULL )

  {

    libcerror_error_set(

     error,

     LIBCERROR_ERROR_DOMAIN_ARGUMENTS,

     LIBCERROR_ARGUMENT_ERROR_INVALID_VALUE,

     "%s: invalid task.",

     function );

    return( -1 );

  }

  if( *task != NULL )

  {

    libcerror_error_set(

     error,

     LIBCERROR_ERROR_DOMAIN_RUNTIME,

     LIBCERROR_RUNTIME_ERROR_VALUE_ALREADY_SET,

     "%s: invalid task value already set.",

     function );

    return( -1 );

  }

  internal_task = memory_allocate_structure(

                   libfwevt_internal_task_t );

  if( internal_task == NULL )

  {

    libcerror_error_set(

     error,

     LIBCERROR_ERROR_DOMAIN_MEMORY,

     LIBCERROR_MEMORY_ERROR_INSUFFICIENT,

     "%s: unable to create task.",

     function );

    goto on_error;

  }

  if( memory_set(

       internal_task,

       0,

       sizeof( libfwevt_internal_task_t ) ) == NULL )

  {

    libcerror_error_set(

     error,

     LIBCERROR_ERROR_DOMAIN_MEMORY,

     LIBCERROR_MEMORY_ERROR_SET_FAILED,

     "%s: unable to clear task.",

     function );

    goto on_error;

  }

  *task = (libfwevt_task_t *) internal_task;

  return( 1 );

on_error:

  if( internal_task != NULL )

  {

    memory_free(

     internal_task );

  }

  return( -1 );

}

/* Frees a task

 * Returns 1 if successful or -1 on error

 */

int libfwevt_task_free(

     libfwevt_task_t **task,

     libcerror_error_t **error )

{

  static char *function = "libfwevt_task_free";

  if( task == NULL )

  {

    libcerror_error_set(

     error,

     LIBCERROR_ERROR_DOMAIN_ARGUMENTS,

     LIBCERROR_ARGUMENT_ERROR_INVALID_VALUE,

     "%s: invalid task.",

     function );

    return( -1 );

  }

  if( *task != NULL )

  {

    *task = NULL;

  }

  return( 1 );

}

/* Frees a task

 * Returns 1 if successful or -1 on error

 */

int libfwevt_internal_task_free(

     libfwevt_internal_task_t **internal_task,

     libcerror_error_t **error )

{

  static char *function = "libfwevt_internal_task_free";

  if( internal_task == NULL )

  {

    libcerror_error_set(

     error,

     LIBCERROR_ERROR_DOMAIN_ARGUMENTS,

     LIBCERROR_ARGUMENT_ERROR_INVALID_VALUE,

     "%s: invalid task.",

     function );

    return( -1 );

  }

  if( *internal_task != NULL )

  {

    memory_free(

     *internal_task );

    *internal_task = NULL;

  }

  return( 1 );

}

/* Reads the task

 * Returns 1 if successful or -1 on error

 */

int libfwevt_task_read_data(

     libfwevt_task_t *task,

     const uint8_t *data,

     size_t data_size,

     size_t data_offset,

     libcerror_error_t **error )

{

  libfwevt_internal_task_t *internal_task = NULL;

  fwevt_template_task_t *wevt_task        = NULL;

  static char *function                   = "libfwevt_task_read_data";

  uint32_t task_data_offset               = 0;

  uint32_t task_data_size                 = 0;

#if defined( HAVE_DEBUG_OUTPUT )

  uint32_t value_32bit                    = 0;

#endif

  if( task == NULL )

  {

    libcerror_error_set(

     error,

     LIBCERROR_ERROR_DOMAIN_ARGUMENTS,

     LIBCERROR_ARGUMENT_ERROR_INVALID_VALUE,

     "%s: invalid task.",

     function );

    return( -1 );

  }

  internal_task = (libfwevt_internal_task_t *) task;

  if( data == NULL )

  {

    libcerror_error_set(

     error,

     LIBCERROR_ERROR_DOMAIN_ARGUMENTS,

     LIBCERROR_ARGUMENT_ERROR_INVALID_VALUE,

     "%s: invalid data.",

     function );

    return( -1 );

  }

  if( data_size > (size_t) SSIZE_MAX )

  {

    libcerror_error_set(

     error,

     LIBCERROR_ERROR_DOMAIN_ARGUMENTS,

     LIBCERROR_ARGUMENT_ERROR_VALUE_EXCEEDS_MAXIMUM,

     "%s: invalid data size value exceeds maximum.",

     function );

    return( -1 );

  }

  if( data_offset >= data_size )

  {

    libcerror_error_set(

     error,

     LIBCERROR_ERROR_DOMAIN_ARGUMENTS,

     LIBCERROR_ARGUMENT_ERROR_VALUE_OUT_OF_BOUNDS,

     "%s: invalid data offset value out of bounds.",

     function );

    return( -1 );

  }

  if( ( data_size < sizeof( fwevt_template_task_t ) )

   || ( data_offset > ( data_size - sizeof( fwevt_template_task_t ) ) ) )

  {

    libcerror_error_set(

     error,

     LIBCERROR_ERROR_DOMAIN_ARGUMENTS,

     LIBCERROR_ARGUMENT_ERROR_VALUE_TOO_SMALL,

     "%s: invalid data value too small.",

     function );

    return( -1 );

  }

  wevt_task = (fwevt_template_task_t *) &( data[ data_offset ] );

#if defined( HAVE_DEBUG_OUTPUT )

  if( libcnotify_verbose != 0 )

  {

    libcnotify_printf(

     "%s: task data:\n",

     function );

    libcnotify_print_data(

     (uint8_t *) wevt_task,

     sizeof( fwevt_template_task_t ),

     0 );

  }

#endif

  byte_stream_copy_to_uint32_little_endian(

   wevt_task->identifier,

   internal_task->identifier );

  byte_stream_copy_to_uint32_little_endian(

   wevt_task->data_offset,

   task_data_offset );

#if defined( HAVE_DEBUG_OUTPUT )

  if( libcnotify_verbose != 0 )

  {

    libcnotify_printf(

     "%s: identifier\t\t\t\t\t: %" PRIu32 "\n",

     function,

     internal_task->identifier );

    byte_stream_copy_to_uint32_little_endian(

     wevt_task->message_identifier,

     value_32bit );

    libcnotify_printf(

     "%s: message identifier\t\t\t\t: 0x%08" PRIx32 "\n",

     function,

     value_32bit );

    if( libfwevt_debug_print_guid_value(

         function,

         "unknown1\t\t\t\t\t",

         wevt_task->unknown1,

         16,

         LIBFGUID_ENDIAN_LITTLE,

         LIBFGUID_STRING_FORMAT_FLAG_USE_LOWER_CASE,

         error ) != 1 )

    {

      libcerror_error_set(

       error,

       LIBCERROR_ERROR_DOMAIN_RUNTIME,

       LIBCERROR_RUNTIME_ERROR_PRINT_FAILED,

       "%s: unable to print GUID value.",

       function );

      return( -1 );

    }

    libcnotify_printf(

     "%s: data offset\t\t\t\t\t: 0x%08" PRIx32 "\n",

     function,

     task_data_offset );

  }

#endif /* defined( HAVE_DEBUG_OUTPUT ) */

  if( task_data_offset > 0 )

  {

    if( task_data_offset >= ( data_size - 4 ) )

    {

      libcerror_error_set(

       error,

       LIBCERROR_ERROR_DOMAIN_RUNTIME,

       LIBCERROR_RUNTIME_ERROR_VALUE_OUT_OF_BOUNDS,

       "%s: invalid task data offset value out of bounds.",

       function );

      return( -1 );

    }

    byte_stream_copy_to_uint32_little_endian(

     &( data[ task_data_offset ] ),

     task_data_size );

    if( ( data_size < task_data_size )

     || ( task_data_offset > ( data_size - task_data_size ) ) )

    {

      libcerror_error_set(

       error,

       LIBCERROR_ERROR_DOMAIN_RUNTIME,

       LIBCERROR_RUNTIME_ERROR_VALUE_OUT_OF_BOUNDS,

       "%s: invalid task data size value out of bounds.",

       function );

      return( -1 );

    }

#if defined( HAVE_DEBUG_OUTPUT )

    if( libcnotify_verbose != 0 )

    {

      libcnotify_printf(

       "%s: data:\n",

       function );

      libcnotify_print_data(

       &( data[ task_data_offset ] ),

       task_data_size,

       0 );

    }

#endif /* defined( HAVE_DEBUG_OUTPUT ) */

#if defined( HAVE_DEBUG_OUTPUT )

    if( libcnotify_verbose != 0 )

    {

      libcnotify_printf(

       "%s: data size\t\t\t\t\t: %" PRIu32 "\n",

       function,

       task_data_size );

    }

#endif /* defined( HAVE_DEBUG_OUTPUT ) */

    if( task_data_size >= 4 )

    {

      task_data_offset += 4;

      task_data_size   -= 4;

#if defined( HAVE_DEBUG_OUTPUT )

      if( libcnotify_verbose != 0 )

      {

        if( libfwevt_debug_print_utf16_string_value(

             function,

             "name\t\t\t\t\t\t",

             &( data[ task_data_offset ] ),

             task_data_size,

             LIBUNA_ENDIAN_LITTLE,

             error ) != 1 )

        {

          libcerror_error_set(

           error,

           LIBCERROR_ERROR_DOMAIN_RUNTIME,

           LIBCERROR_RUNTIME_ERROR_PRINT_FAILED,

           "%s: unable to print UTF-16 string value.",

           function );

          return( -1 );

        }

      }

#endif /* defined( HAVE_DEBUG_OUTPUT ) */

    }

  }

#if defined( HAVE_DEBUG_OUTPUT )

  if( libcnotify_verbose != 0 )

  {

    libcnotify_printf(

     "\n" );

  }

#endif

  return( 1 );

}