/****************************************************************************
** Copyright (C) 2001-2010 Klaralvdalens Datakonsult AB.  All rights reserved.
**
** This file is part of the KD Tools library.
**
** Licensees holding valid commercial KD Tools licenses may use this file in
** accordance with the KD Tools Commercial License Agreement provided with
** the Software.
**
**
** This file may be distributed and/or modified under the terms of the
** GNU Lesser General Public License version 2 and version 3 as published by the
** Free Software Foundation and appearing in the file LICENSE.LGPL included.
**
** This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
** WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
**
** Contact info@kdab.com if any conditions of this licensing are not
** clear to you.
**
**********************************************************************/

#include "kdupdaterufcompresscommon_p.h"

#include <QCryptographicHash>
#include <QDataStream>

using namespace KDUpdater;

bool UFHeader::isValid() const
{
    return magic == QLatin1String(KD_UPDATER_UF_HEADER_MAGIC)
            && fileList.count() == permList.count()
            && fileList.count() == isDirList.count();
}

void UFHeader::addToHash(QCryptographicHash &hash) const
{
    QByteArray data;
    QDataStream stream(&data, QIODevice::WriteOnly);
    stream << *this;
    hash.addData(data);
}

UFEntry::UFEntry() 
    : permissions(0)
{}

bool UFEntry::isValid() const
{
    return !fileName.isEmpty();
}

void UFEntry::addToHash(QCryptographicHash &hash) const
{
    QByteArray data;
    QDataStream stream(&data, QIODevice::WriteOnly);
    stream.setVersion(QDataStream::Qt_4_2);
    stream << *this;
    hash.addData(data);
}

namespace KDUpdater {

QDataStream &operator<<(QDataStream &stream, const UFHeader &hdr)
{
    stream << hdr.magic;
    stream << hdr.fileList;
    stream << hdr.permList;
    stream << hdr.isDirList;
    return stream;
}

QDataStream &operator>>(QDataStream &stream, UFHeader &hdr)
{
    const QDataStream::Status oldStatus = stream.status();
    stream >> hdr.magic;
    if (stream.status() == QDataStream::Ok && hdr.magic != QLatin1String(KD_UPDATER_UF_HEADER_MAGIC))
        stream.setStatus(QDataStream::ReadCorruptData);
    
    if (stream.status() == QDataStream::Ok)
        stream >> hdr.fileList;

    if (stream.status() == QDataStream::Ok)
        stream >> hdr.permList;
    
    if (stream.status() == QDataStream::Ok)
        stream >> hdr.isDirList;
    
    if (stream.status() == QDataStream::Ok
            && (hdr.fileList.count() != hdr.permList.count() || hdr.permList.count() != hdr.isDirList.count()))
        stream.setStatus(QDataStream::ReadCorruptData);
   
    if( stream.status() != QDataStream::Ok)
        hdr = UFHeader();

    if (oldStatus != QDataStream::Ok)
        stream.setStatus(oldStatus);

    return stream;
}

QDataStream &operator<<(QDataStream &stream, const UFEntry &entry)
{
    stream << entry.fileName;
    stream << entry.permissions;
    stream << entry.fileData;
    return stream;
}

QDataStream &operator>>(QDataStream &stream, UFEntry &entry)
{
    const QDataStream::Status oldStatus = stream.status();
    if (stream.status() == QDataStream::Ok)
        stream >> entry.fileName;
    if (stream.status() == QDataStream::Ok)
        stream >> entry.permissions;
    if (stream.status() == QDataStream::Ok)
        stream >> entry.fileData;

    if (stream.status() != QDataStream::Ok)
        entry = UFEntry();

    if (oldStatus != QDataStream::Ok)
        stream.setStatus(oldStatus);

    return stream;
}

} // namespace KDUpdater