/*
 * This file is Copyright Daniel Silverstone <dsilvers@digital-scurf.org> 2006
 *
 * Permission is hereby granted, free of charge, to any person
 * obtaining a copy of this software and associated documentation
 * files (the "Software"), to deal in the Software without
 * restriction, including without limitation the rights to use, copy,
 * modify, merge, publish, distribute, sublicense, and/or sell copies
 * of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be
 * included in all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
 * NONINFRINGEMENT.  IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
 * HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
 * WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
 * DEALINGS IN THE SOFTWARE.
 *
 */

#include "libgfshare.h"

#include <stdio.h>
#include <stdlib.h>
#include <string.h>


int
main( int argc, char **argv )
{
  int ok = 1, i;
  unsigned char* secret = malloc(512);
  unsigned char* share1 = malloc(512);
  unsigned char* share2 = malloc(512);
  unsigned char* share3 = malloc(512);
  unsigned char* recomb = malloc(512);
  unsigned char* sharenrs = (unsigned char*)strdup("012");
  gfshare_ctx *G;
  
  /* Stage 1, make a secret */
  for( i = 0; i < 512; ++i )
    secret[i] = (random() & 0xff00) >> 8;
  /* Stage 2, split it three ways with a threshold of 2 */
  G = gfshare_ctx_init_enc( sharenrs, 3, 2, 512 );
  gfshare_ctx_enc_setsecret( G, secret );
  gfshare_ctx_enc_getshare( G, 0, share1 );
  gfshare_ctx_enc_getshare( G, 1, share2 );
  gfshare_ctx_enc_getshare( G, 2, share3 );
  gfshare_ctx_free( G );
  /* Prep the decode shape */
  G = gfshare_ctx_init_dec( sharenrs, 3, 512 );
  gfshare_ctx_dec_giveshare( G, 0, share1 );
  gfshare_ctx_dec_giveshare( G, 1, share2 );
  gfshare_ctx_dec_giveshare( G, 2, share3 );
  /* Stage 3, attempt a recombination with shares 1 and 2 */
  sharenrs[2] = 0;
  gfshare_ctx_dec_newshares( G, sharenrs );
  gfshare_ctx_dec_extract( G, recomb );
  for( i = 0; i < 512; ++i )
    if( secret[i] != recomb[i] ) 
      ok = 0;
  /* Stage 4, attempt a recombination with shares 1 and 3 */
  sharenrs[2] = '2';
  sharenrs[1] = 0;
  gfshare_ctx_dec_newshares( G, sharenrs );
  gfshare_ctx_dec_extract( G, recomb );
  for( i = 0; i < 512; ++i )
    if( secret[i] != recomb[i] ) 
      ok = 0;
  /* Stage 5, attempt a recombination with shares 2 and 3 */
  sharenrs[0] = 0;
  sharenrs[1] = '1';
  gfshare_ctx_dec_newshares( G, sharenrs );
  gfshare_ctx_dec_extract( G, recomb );
  for( i = 0; i < 512; ++i )
    if( secret[i] != recomb[i] ) 
      ok = 0;
  /* Stage 6, attempt a recombination with shares 1, 2 and 3 */
  sharenrs[0] = '0';
  gfshare_ctx_dec_newshares( G, sharenrs );
  gfshare_ctx_dec_extract( G, recomb );
  for( i = 0; i < 512; ++i )
    if( secret[i] != recomb[i] ) 
      ok = 0;
  gfshare_ctx_free( G );
  return ok!=1;
}