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 :   return 0;
      25          18 : }
      26             : 
      27             : int
      28             : LLVMFuzzerTestOneInput( uchar const * data,
      29             :                         ulong         data_sz ) {
      30             : 
      31             :   ulong  enc_sz = FD_BASE64_ENC_SZ( data_sz );
      32             :   char * enc    = malloc( enc_sz );
      33             :   assert( enc );
      34             : 
      35             :   ulong enc_res = fd_base64_encode( enc, data, data_sz );
      36             :   assert( enc_res==enc_sz );
      37             : 
      38             :   ulong dec_sz = FD_BASE64_DEC_SZ( enc_sz );
      39             :   assert( dec_sz <= data_sz+3UL );
      40             : 
      41             :   uchar * dec = malloc( dec_sz );
      42             :   assert( dec );
      43             : 
      44             :   long dec_res = fd_base64_decode( dec, enc, enc_sz );
      45             :   assert(        dec_res>=     0L );
      46             :   assert( (ulong)dec_res<= dec_sz );
      47             :   assert( (ulong)dec_res==data_sz );
      48             : 
      49             :   assert( 0==memcmp( dec, data, data_sz ) );
      50             : 
      51             :   free( enc );
      52             :   free( dec );
      53             :   FD_FUZZ_MUST_BE_COVERED;
      54             :   return 0;
      55             : }