/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: t; c-basic-offset: 4 -*- */

/* libwpd

 * Version: MPL 2.0 / LGPLv2.1+

 *

 * This Source Code Form is subject to the terms of the Mozilla Public

 * License, v. 2.0. If a copy of the MPL was not distributed with this

 * file, You can obtain one at http://mozilla.org/MPL/2.0/.

 *

 * Major Contributor(s):

 * Copyright (C) 2005 Fridrich Strba (fridrich.strba@bluewin.ch)

 *

 * For minor contributions see the git repository.

 *

 * Alternatively, the contents of this file may be used under the terms

 * of the GNU Lesser General Public License Version 2.1 or later

 * (LGPLv2.1+), in which case the provisions of the LGPLv2.1+ are

 * applicable instead of those above.

 *

 * For further information visit http://libwpd.sourceforge.net

 */

/* "This product is not manufactured, approved, or supported by

 * Corel Corporation or Corel Corporation Limited."

 */

#include <set>

#include "WP5PrefixData.h"

#include "WP5GeneralPacketIndex.h"

#include "WP5SpecialHeaderIndex.h"

#include "libwpd_internal.h"

WP5PrefixData::WP5PrefixData(librevenge::RVNGInputStream *input, WPXEncryption *encryption) :

  m_generalPacketData()

{

  std::vector<WP5GeneralPacketIndex> prefixIndexVector;

  std::set<long> readBlocks; // guard against a cycle in the block list

  int id = 0;

  while (true)

  {

    readBlocks.insert(input->tell());

    WP5SpecialHeaderIndex shi = WP5SpecialHeaderIndex(input, encryption);

    if ((shi.getType() != 0xfffb) || (shi.getNumOfIndexes() != 5) || (shi.getIndexBlockSize() != 50))

    {

      WPD_DEBUG_MSG(("WordPerfect: detected possible prefix data corruption, ignoring this and all following packets.\n"));

      break;

    }

    bool tmpFoundPossibleCorruption = false;

    for (int i=0; i<(shi.getNumOfIndexes()-1); i++)

    {

      WP5GeneralPacketIndex gpi = WP5GeneralPacketIndex(input, encryption, id);

      if ((gpi.getType() > 0x02ff) && (gpi.getType() < 0xfffb))

      {

        WPD_DEBUG_MSG(("WordPerfect: detected possible prefix data corruption, ignoring this and all following packets.\n"));

        tmpFoundPossibleCorruption = true;

        break;

      }

      if ((gpi.getType() != 0) && (gpi.getType() != 0xffff))

      {

        WPD_DEBUG_MSG(("WordPerfect: constructing general packet index 0x%x\n", id));

        prefixIndexVector.push_back(gpi);

        id++;

      }

    }

    if (tmpFoundPossibleCorruption)

      break;

    if (readBlocks.find(long(shi.getNextBlockOffset())) != readBlocks.end())

    {

      WPD_DEBUG_MSG(("WordPerfect: detected a cycle in the special header index list\n"));

      break;

    }

    if (shi.getNextBlockOffset() != 0)

      input->seek(shi.getNextBlockOffset(), librevenge::RVNG_SEEK_SET);

    else

      break;

  }

  std::vector<WP5GeneralPacketIndex>::iterator gpiIter;

  for (gpiIter = prefixIndexVector.begin(); gpiIter != prefixIndexVector.end(); ++gpiIter)

  {

    WPD_DEBUG_MSG(("WordPerfect: constructing general packet data %i\n", (*gpiIter).getID()));

    std::unique_ptr<WP5GeneralPacketData> generalPacketData{WP5GeneralPacketData::constructGeneralPacketData(input, encryption, &(*gpiIter))};

    if (generalPacketData)

    {

      m_generalPacketData[gpiIter->getType()] = std::move(generalPacketData);

    }

  }

}

WP5PrefixData::~WP5PrefixData()

{

}

const WP5GeneralPacketData *WP5PrefixData::getGeneralPacketData(const int type) const

{

  auto Iter = m_generalPacketData.find(type);

  if (Iter != m_generalPacketData.end())

    return Iter->second.get();

  else

    return nullptr;

}

/* vim:set shiftwidth=4 softtabstop=4 noexpandtab: */