/*------------------------------------------------------------------------------
--       Copyright (c) 2015-2017, VeriSilicon Inc. All rights reserved        --
--         Copyright (c) 2011-2014, Google Inc. All rights reserved.          --
--                                                                            --
-- This software is confidential and proprietary and may be used only as      --
--   expressly authorized by VeriSilicon in a written licensing agreement.    --
--                                                                            --
--         This entire notice must be reproduced on all copies                --
--                       and may not be removed.                              --
--                                                                            --
--------------------------------------------------------------------------------
-- Redistribution and use in source and binary forms, with or without         --
-- modification, are permitted provided that the following conditions are met:--
--   * Redistributions of source code must retain the above copyright notice, --
--       this list of conditions and the following disclaimer.                --
--   * Redistributions in binary form must reproduce the above copyright      --
--       notice, this list of conditions and the following disclaimer in the  --
--       documentation and/or other materials provided with the distribution. --
--   * Neither the names of Google nor the names of its contributors may be   --
--       used to endorse or promote products derived from this software       --
--       without specific prior written permission.                           --
--------------------------------------------------------------------------------
-- THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"--
-- AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE  --
-- IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE --
-- ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE  --
-- LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR        --
-- CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF       --
-- SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS   --
-- INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN    --
-- CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)    --
-- ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE --
-- POSSIBILITY OF SUCH DAMAGE.                                                --
--------------------------------------------------------------------------------
------------------------------------------------------------------------------*/

#include "software/test/common/error_simulator.h"

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

#include "software/test/common/swhw/tb_stream_corrupt.h"

#define DEBUG_PRINT(str) printf str

struct ErrorSim {
  u32 headers_decoded;
  Demuxer concrete_demuxer;
  struct ErrorSimulationParams params;
};

ErrorSimulator ErrorSimulatorInject(Demuxer* demuxer,
                                    struct ErrorSimulationParams params) {
  /* Our error simulator injects it's own function in place of the bitstream
     read function. */
  struct ErrorSim* sim = calloc(1, sizeof(struct ErrorSim));
  if (sim == NULL)
    return NULL;
  sim->concrete_demuxer = *demuxer;
  demuxer->GetVideoFormat = ErrorSimulatorIdentifyFormat;
  demuxer->HeadersDecoded = ErrorSimulatorHeadersDecoded;
  demuxer->ReadPacket = ErrorSimulatorReadPacket;
  demuxer->close = ErrorSimulatorClose;
  TBInitializeRandom(params.seed);
  sim->params = params;
  return sim;
}

int ErrorSimulatorIdentifyFormat(ErrorSimulator inst) {
  struct ErrorSim* sim = (struct ErrorSim*)inst;
  return sim->concrete_demuxer.GetVideoFormat(sim->concrete_demuxer.inst);
}


void ErrorSimulatorHeadersDecoded(ErrorSimulator inst) {
  struct ErrorSim* sim = (struct ErrorSim*)inst;
  sim->headers_decoded = 1;
}

int ErrorSimulatorReadPacket(ErrorSimulator inst, u8* buffer, u8* stream[2], i32* size, u8 is_ringbuffer) {
  struct ErrorSim* sim = (struct ErrorSim*)inst;
  u8* orig_buffer = buffer;
  i32 orig_size = *size;
  int rv = sim->concrete_demuxer.ReadPacket(sim->concrete_demuxer.inst, buffer,
           stream, size, is_ringbuffer);
  if (rv < 0) return rv;

  if (sim->params.corrupt_headers ||
      (!sim->params.corrupt_headers && sim->headers_decoded)) {

    if (sim->params.lose_packets) {
      u8 lose_packet = 0;
      if (TBRandomizePacketLoss(sim->params.packet_loss_odds, &lose_packet)) {
        DEBUG_PRINT(("Packet loss simulator error (wrong config?)\n"));
      }
      if (lose_packet) {
        *size = orig_size;
        return ErrorSimulatorReadPacket(inst, orig_buffer, stream, size, is_ringbuffer);
      }
    }
    if (sim->params.swap_bits_in_stream) {
      if (TBRandomizeBitSwapInStream(buffer, *size, sim->params.swap_bit_odds)) {
        DEBUG_PRINT(("Bitswap randomizer error (wrong config?)\n"));
      }
    }
    if (sim->params.truncate_stream) {
      u32 random_size = rv;
      if (TBRandomizeU32(&random_size)) {
        DEBUG_PRINT(("Truncate randomizer error (wrong config?)\n"));
      }
      rv = random_size;
    }
  }
  return rv;
}

void ErrorSimulatorClose(ErrorSimulator inst) {
  struct ErrorSim* sim = (struct ErrorSim*)inst;
  sim->concrete_demuxer.close(sim->concrete_demuxer.inst);
  free(sim);
}
