Line data    Source code

       1             : #if !FD_HAS_HOSTED
       2             : #error "This target requires FD_HAS_HOSTED"
       3             : #endif
       4             : 
       5             : #include <assert.h>
       6             : #include <stdio.h>
       7             : #include <stdlib.h>
       8             : 
       9             : #include "../../util/fd_util.h"
      10             : #include "../../util/sanitize/fd_fuzz.h"
      11             : 
      12             : #include "fd_base64.h"
      13             : 
      14             : /* fuzz_base64_enc verifies that decode(encode(x)) is the identity
      15             :    function. */
      16             : 
      17             : int
      18             : LLVMFuzzerInitialize( int  *   argc,
      19          18 :                       char *** argv ) {
      20             :   /* Set up shell without signal handlers */
      21          18 :   putenv( "FD_LOG_BACKTRACE=0" );
      22          18 :   fd_boot( argc, argv );
      23          18 :   atexit( fd_halt );
      24          18 :   fd_log_level_core_set(3); /* crash on warning log */
      25          18 :   return 0;
      26          18 : }
      27             : 
      28             : int
      29             : LLVMFuzzerTestOneInput( uchar const * data,
      30             :                         ulong         data_sz ) {
      31             : 
      32             :   ulong  enc_sz = FD_BASE64_ENC_SZ( data_sz );
      33             :   char * enc    = malloc( enc_sz );
      34             :   assert( enc );
      35             : 
      36             :   ulong enc_res = fd_base64_encode( enc, data, data_sz );
      37             :   assert( enc_res==enc_sz );
      38             : 
      39             :   ulong dec_sz = FD_BASE64_DEC_SZ( enc_sz );
      40             :   assert( dec_sz <= data_sz+3UL );
      41             : 
      42             :   uchar * dec = malloc( dec_sz );
      43             :   assert( dec );
      44             : 
      45             :   long dec_res = fd_base64_decode( dec, enc, enc_sz );
      46             :   assert(        dec_res>=     0L );
      47             :   assert( (ulong)dec_res<= dec_sz );
      48             :   assert( (ulong)dec_res==data_sz );
      49             : 
      50             :   assert( 0==memcmp( dec, data, data_sz ) );
      51             : 
      52             :   free( enc );
      53             :   free( dec );
      54             :   FD_FUZZ_MUST_BE_COVERED;
      55             :   return 0;
      56             : }