/*
  This file is Copyright © 1994-1995 Olivier Montanuy,
               Copyright © 1999-2005 André Majorel,
               Copyright © 2006-2017 contributors to the DeuTex project.

  DeuTex incorporates code derived from DEU 5.21 that was put in the
  public domain in 1994 by Raphaël Quinet and Brendon Wyber.

  SPDX-License-Identifier: GPL-2.0+
*/

#include "deutex.h"
#include <errno.h>
#include "tools.h"
#include "endianm.h"
#include "mkwad.h"
#include "texture.h"
#include "ident.h"
#include "lists.h"
#include "merge.h"

#include <fcntl.h>

/*
** old references
**
*/
static const int32_t HDRdirSz = 5 * sizeof(struct WADDIR);
static struct WADDIR HDRdir[6];
/*
**  Take some entries from a WAD
*/
static void HDRplunderWad(struct WADINFO *rwad, struct WADINFO *ewad)
{
    char *data;
    int32_t wsize, sz = 0;
    int32_t ostart, osize;
    int16_t n;
    int32_t oldfilesz;
    /*
     ** calculate required size
     */
    oldfilesz = WADRposition(rwad);     /*old file size */
    /*
     ** copy entries from WAD
     */
    Phase("ME10", "Copying entries from wad (please wait)");
    data = (char *) Malloc(MEMORYCACHE);
    for (n = 0; n < (rwad->ntry); n++) {
        ostart = rwad->dir[n].start;
        osize = rwad->dir[n].size;
        /*detect external entries */
        if (ostart & EXTERNAL) {        /*update new directory */
            WADRalign4(rwad);
            rwad->dir[n].start = WADRposition(rwad);
            /*get entry size */
            if (osize > 0) {    /*copy old entry */
                WADRseek(ewad, ostart & (~EXTERNAL));
                for (wsize = 0; wsize < osize; wsize += sz) {
                    sz = (osize - wsize >
                          MEMORYCACHE) ? MEMORYCACHE : osize - wsize;
                    WADRreadBytes(ewad, data, sz);
                    if (WADRwriteBytes2(rwad, data, sz) < 0) {
                        WADRchsize(rwad, oldfilesz);
                        ProgError("ME13", "Not enough disk space");
                        break;
                    }
                }
            }
        }
    }

    free(data);
}

/*
** Copy a WAD, and link to it's entries
*/
static int32_t HDRinsertWad(struct WADINFO *rwad, struct WADINFO *ewad,
                            int32_t * pesize)
{
    char *data;
    int32_t wsize, sz = 0;
    int32_t estart, esize;
    int16_t n;
    int32_t oldfilesz;
    /*
     ** calculate required size
     */
    oldfilesz = WADRposition(rwad);     /*old file size */
    WADRalign4(rwad);
    estart = WADRposition(rwad);
    WADRseek(ewad, 0);
    esize = ewad->maxpos;
    Phase("ME16", "Inserting wad file into wad");
    data = (char *) Malloc(MEMORYCACHE);
    for (wsize = 0; wsize < esize; wsize += sz) {
        sz = (esize - wsize > MEMORYCACHE) ? MEMORYCACHE : esize - wsize;
        WADRreadBytes(ewad, data, sz);
        if (WADRwriteBytes2(rwad, data, sz) < 0) {
            WADRchsize(rwad, oldfilesz);
            ProgError("ME19", "Not enough disk space");
            break;
        }
        /*if((wsize&0xF000)==0) Phase("."); FIXME need /dev/tty output */
    }
    free(data);
    for (n = 0; n < (rwad->ntry); n++) {        /*detect external entries */
        if ((rwad->dir[n].start) & EXTERNAL) {  /*update new directory */
            rwad->dir[n].start &= (~EXTERNAL);
            rwad->dir[n].start += estart;
        }
    }
    /* Phase("\n"); FIXME need /dev/tty output */
    *pesize = esize;
    return estart;
}

void HDRrestoreWAD(const char *wadres)
{
    struct WADINFO rwad;
    int32_t dirpos, ntry, n;
    int32_t rwadstart = 0, rwadsize = 0;
    int32_t ewadstart = 0, ewadsize = 0;
    static char ewadname[8];
    static char ewadfile[40];
    char *data;
    int32_t size = 0, wsize = 0, sz = 0;
    int32_t time;
    FILE *fp;
    bool Fail;
    Phase("ME22", "Attempting to restore wad %s", fname(wadres));
    /*open DOOM.WAD */
    rwad.ok = 0;
    WADRopenR(&rwad, wadres);

    /*get position of fake directory entry, reference to old dir */
    dirpos = rwad.dirpos - HDRdirSz;
    WADRseek(&rwad, dirpos);
    WADRreadBytes(&rwad, (char *) HDRdir, HDRdirSz);
    Fail = false;
    if (peek_i32_le(&HDRdir[0].start) != 0x24061968L)
        Fail = true;
    if (peek_i32_le(&HDRdir[0].size) != 666L)
        Fail = true;
    if (strncmp(HDRdir[0].name, "IZNOGOOD", 8) != 0)
        Fail = true;
    if (Fail != false) {
        if ((n = WADRfindEntry(&rwad, "_DEUTEX_")) >= 0)
            if (rwad.dir[n].size >= HDRdirSz) {
                dirpos = rwad.dir[n].start;
                WADRseek(&rwad, dirpos);
                WADRreadBytes(&rwad, (char *) HDRdir, HDRdirSz);
                Fail = false;
                if (peek_i32_le(&HDRdir[0].start) != 0x24061968L)
                    Fail = true;
                if (peek_i32_le(&HDRdir[0].size) != 666L)
                    Fail = true;
                if (strncmp(HDRdir[0].name, "IZNOGOOD", 8) != 0)
                    Fail = true;
            }
    }
    if (Fail != false)
        ProgError("ME25", "Not a modified WAD");
    Phase("ME28", "Restoration infos seem correct");
    dirpos = peek_i32_le(&HDRdir[1].start);
    ntry = peek_i32_le(&HDRdir[1].size);
    rwadstart = peek_i32_le(&HDRdir[2].start);
    rwadsize = peek_i32_le(&HDRdir[2].size);
    ewadstart = peek_i32_le(&HDRdir[3].start);
    ewadsize = peek_i32_le(&HDRdir[3].size);
    Normalise(ewadname, HDRdir[3].name);        /*name of WAD inside */
    /*original file time */
    time = peek_i32_le(&HDRdir[4].size);
    if (peek_i32_le(&HDRdir[4].start) != false) {       /*extract the PWAD */
        sprintf(ewadfile, "%.8s.WAD", ewadname);
        ToLowerCase(ewadfile);
        fp = fopen(ewadfile, FOPEN_RB);
        if (fp != NULL) {
            fclose(fp);
            Info("ME31", "%s already exists, internal WAD discarded",
                 fname(ewadfile));
        } else {
            Phase("ME34", "Restoring internal wad %s", fname(ewadfile));
            if ((fp = fopen(ewadfile, FOPEN_WB)) != NULL) {
                data = (char *) Malloc(MEMORYCACHE);
                size = ewadsize;
                WADRseek(&rwad, ewadstart);
                fseek(fp, 0, SEEK_SET);
                for (wsize = 0; wsize < size; wsize += sz) {
                    sz = (size - wsize >
                          MEMORYCACHE) ? MEMORYCACHE : size - wsize;
                    WADRreadBytes(&rwad, data, sz);
                    errno = 0;
                    if (fwrite(data, (size_t) sz, 1, fp) != 1) {
                        Warning("ME37", "%s: %s", fnameofs(ewadfile, (long)
                                                           ewadstart),
                                errno ==
                                0 ? "write error" : strerror(errno));
                        break;
                    }
                }
                free(data);
                fclose(fp);
            } else
                Warning("ME40", "%s: %s", fname(ewadfile),
                        strerror(errno));
        }
    }
    WADRopenA(&rwad, wadres);
    /*correct the directory reference of DOOM.WAD */
    WADRsetDirRef(&rwad, ntry, dirpos);
    /*restore original size */
    WADRchsize(&rwad, rwadstart + rwadsize);
    WADRclose(&rwad);
    Set_File_Time(wadres, time);
    Output("Restoration of %s should be successful\n", fname(wadres));
}

static void HDRsetDir(struct WADINFO *rwad, bool IsIwad, bool Restore,
                      int32_t time, int32_t dirpos, int32_t ntry,
                      int32_t rsize, int32_t estart, int32_t esize,
                      const char *wadext)
{
    static char name[8];
    int32_t pos;
    /*Set the old references */
    Phase("ME43", "Writing new wad directory");
    write_i32_le(&HDRdir[0].start, 0x24061968L);
    write_i32_le(&HDRdir[0].size, 666L);
    Normalise(HDRdir[0].name, "IZNOGOOD");
    /*Set original WAD DIRECTORY */
    write_i32_le(&HDRdir[1].start, dirpos);
    write_i32_le(&HDRdir[1].size, ntry);
    Normalise(HDRdir[1].name, (IsIwad == true) ? "DOOM_DIR" : "PWAD_DIR");
    /*Store original WAD size and start */
    write_i32_le(&HDRdir[2].start, 0);
    write_i32_le(&HDRdir[2].size, rsize);
    Normalise(HDRdir[2].name, "ORIGINAL");
    /*Store external WAD size and start */
    write_i32_le(&HDRdir[3].start, estart);
    write_i32_le(&HDRdir[3].size, esize);
    GetNameOfWAD(name, wadext);
    Normalise(HDRdir[3].name, name);
    /*old file time */
    write_i32_le(&HDRdir[4].size, time);
    write_i32_le(&HDRdir[4].start, Restore);
    Normalise(HDRdir[4].name, "TIME");
    /*save position of old ref if no previous _DEUTEX_ */
    WADRalign4(rwad);
    pos = (int32_t) WADRfindEntry(rwad, "_DEUTEX_");
    if (pos < 0) {
        pos = WADRposition(rwad);       /*BC++ 4.5 bug */
        WADRdirAddEntry(rwad, pos, HDRdirSz, "_DEUTEX_");
    }
    /*write old refs */
    WADRwriteBytes(rwad, (char *) HDRdir, HDRdirSz);
    /*write the directory */
    rwad->dirpos = WADRposition(rwad);
    WADRwriteDir(rwad, 1);
}

/*
** merge WAD
**
*/

void PSTmergeWAD(const char *doomwad, const char *wadin, NTRYB select)
{
    static struct WADINFO iwad;
    static struct WADINFO pwad;
    ENTRY *iiden;               /*identify entry in IWAD */
    ENTRY *piden;               /*identify entry in PWAD */
    int16_t pnm;
    char *Pnam;
    int32_t Pnamsz = 0;
    int32_t dirpos, ntry, isize, pstart, psize, time;
    struct WADDIR *NewDir;
    int32_t NewNtry;
    Phase("ME46", "Attempting to merge iwad %s and pwad %s",
          fname(doomwad), wadin);
    /*open iwad,get iwad directory */
    iwad.ok = 0;
    WADRopenR(&iwad, doomwad);
    /*find PNAMES */
    pnm = WADRfindEntry(&iwad, "PNAMES");
    if (pnm < 0)
        ProgError("ME49", "Can't find PNAMES in iwad");
    Pnam = WADRreadEntry(&iwad, pnm, &Pnamsz);
    /* identify iwad */
    iiden = IDENTentriesIWAD(&iwad, Pnam, Pnamsz, true);
    /* get pwad directory, and identify */
    pwad.ok = 0;
    WADRopenR(&pwad, wadin);
    piden = IDENTentriesPWAD(&pwad, Pnam, Pnamsz);
    free(Pnam);
    /*where to put pwad? at pwadstart */
    /* merge the two directories */
    NewDir =
        LISmergeDir(&NewNtry, false, true, select, &iwad, iiden, 0, &pwad,
                    piden, EXTERNAL);
    /* prepare for append */
    time =
        WADRprepareAppend(doomwad, &iwad, NewDir, NewNtry, &dirpos, &ntry,
                          &isize);
    /* complete IWAD with PWAD, restorable */
    pstart = HDRinsertWad(&iwad, &pwad, &psize);
    /* set directory */
    HDRsetDir(&iwad, true, true, time, dirpos, ntry, isize, pstart, psize,
              wadin);
    /* close */
    WADRclose(&pwad);
    WADRclose(&iwad);
}

/*
** -app complete a PWAD with DOOM entries
**
*/
void ADDappendSpriteFloor(const char *doomwad, const char *wadres,
                          NTRYB select)
{
    struct WADINFO iwad;
    struct WADINFO pwad;
    ENTRY *iiden;               /*identify entry in IWAD */
    ENTRY *piden;               /*identify entry in PWAD */
    int16_t pnm;
    char *Pnam;
    int32_t Pnamsz;
    int32_t dirpos, ntry, psize, time;
    struct WADDIR *NewDir;
    int32_t NewNtry;
    Phase("ME52", "Appending sprites and/or flats");
    Phase("ME52", " from iwad %s", fname(doomwad));
    Phase("ME52", " to   pwad %s", fname(wadres));
    /* get iwad directory, and identify */
    iwad.ok = 0;
    WADRopenR(&iwad, doomwad);
    /*find PNAMES */
    pnm = WADRfindEntry(&iwad, "PNAMES");
    if (pnm < 0)
        ProgError("ME61", "Can't find PNAMES in iwad");
    Pnam = WADRreadEntry(&iwad, pnm, &Pnamsz);
    /* identify iwad */
    iiden = IDENTentriesIWAD(&iwad, Pnam, Pnamsz, true);
    /* get pwad directory, and identify */
    pwad.ok = 0;
    WADRopenR(&pwad, wadres);
    piden = IDENTentriesPWAD(&pwad, Pnam, Pnamsz);
    free(Pnam);
    /* merge the two directories */
    NewDir =
        LISmergeDir(&NewNtry, true, true, select, &iwad, iiden, EXTERNAL,
                    &pwad, piden, 0);
    /* prepare for append */
    time =
        WADRprepareAppend(wadres, &pwad, NewDir, NewNtry, &dirpos, &ntry,
                          &psize);
    /* append DOOM sprites to PWAD */
    HDRplunderWad(&pwad, &iwad);
    /* set dir */
    HDRsetDir(&pwad, false, false, time, dirpos, ntry, psize, -1, -1,
              doomwad);
    /* close */
    WADRclose(&iwad);
    WADRclose(&pwad);
}

/*
** join: complete a PWAD with another PWAD entries
**
*/
void ADDjoinWads(const char *doomwad, const char *wadres, const char
                 *wadext, NTRYB select)
{
    struct WADINFO iwad;
     /*IWAD*/ struct WADINFO ewad;      /*external Wad */
    struct WADINFO rwad;
    ENTRY *eiden;               /*identify entry in IWAD */
    ENTRY *riden;               /*identify entry in PWAD */
    int16_t entry;
    char *Entry;
    int32_t EntrySz;
    int16_t pnm;
    char *Patch;
    int32_t PatchSz;
    int32_t start, size;
    int16_t etexu, rtexu;
    struct WADDIR *NewDir;
    int32_t NewNtry;
    bool TexuMrg = false;
    int32_t dirpos, ntry, rsize, estart, esize, time;
    Phase("ME64", "Merging pwad %s", fname(wadext));
    Phase("ME64", " into pwad %s", fname(wadres));
    /* get iwad directory, and identify */
    iwad.ok = 0;
    WADRopenR(&iwad, doomwad);
    /*find PNAMES */
    entry = WADRfindEntry(&iwad, "PNAMES");
    if (entry < 0)
        ProgError("ME70", "Can't find PNAMES in iwad");
    Entry = WADRreadEntry(&iwad, entry, &EntrySz);
    /* get ewad directory, and identify */
    ewad.ok = 0;
    WADRopenR(&ewad, wadext);
    eiden = IDENTentriesPWAD(&ewad, Entry, EntrySz);
    /* get rwad directory, and identify */
    rwad.ok = 0;
    WADRopenR(&rwad, wadres);
    riden = IDENTentriesPWAD(&rwad, Entry, EntrySz);
    free(Entry);
    /*merge texture1 if needed */
    etexu = WADRfindEntry(&ewad, "TEXTURE1");
    rtexu = WADRfindEntry(&rwad, "TEXTURE1");
    if ((etexu >= 0) && (rtexu >= 0)) {
        TexuMrg = true;
        iwad.ok = 0;
        WADRopenR(&iwad, doomwad);
        /*find PNAMES in IWAD and init */
        pnm = WADRfindEntry(&iwad, "PNAMES");
        if (pnm < 0)
            Bug("ME73", "JnPnm");
        Entry = WADRreadEntry(&iwad, entry, &EntrySz);
        PNMinit(Entry, EntrySz);
        free(Entry);
        /*declare TEXTURE1 from IWAD */
        TXUinit();
    }
    WADRclose(&iwad);
    if (TexuMrg == true) {      /*add TEXTURE1 from rwad */
        Phase("ME76", "  With TEXTURE1 from %s", fname(wadres));
        PatchSz = 0;
        Patch = NULL;
        pnm = WADRfindEntry(&rwad, "PNAMES");
        if (pnm >= 0) {
            Phase("ME79", "  Declaring patches from %s", fname(wadres));
            riden[pnm] = EVOID;
            Patch = WADRreadEntry(&rwad, pnm, &PatchSz);
        }
        Entry = WADRreadEntry(&rwad, rtexu, &EntrySz);
        TXUreadTEXTURE(rwad.dir[pnm].name, Entry, EntrySz, Patch, PatchSz,
                       true);
        if (PatchSz != 0)
            free(Patch);
        free(Entry);
        riden[rtexu] = EVOID;   /* forget r texu */
        /*TEXTURE1 from ewad */
        Phase("ME82", "  And TEXTURE1 from %s", fname(wadext));
        PatchSz = 0;
        Patch = NULL;
        pnm = WADRfindEntry(&ewad, "PNAMES");
        if (pnm >= 0) {
            Phase("ME85", "  Declaring Patches from %s", fname(wadext));
            eiden[pnm] = EVOID;
            Patch = WADRreadEntry(&ewad, pnm, &PatchSz);
        }
        Entry = WADRreadEntry(&ewad, etexu, &EntrySz);
        TXUreadTEXTURE(ewad.dir[pnm].name, Entry, EntrySz, Patch, PatchSz,
                       false);
        if (PatchSz != 0)
            free(Patch);
        free(Entry);
        eiden[etexu] = EVOID;   /* forget e texu */
    }
    /* merge the two directories, all entries */
    NewDir =
        LISmergeDir(&NewNtry, false, false, select, &rwad, riden, 0, &ewad,
                    eiden, EXTERNAL);
    /* prepare for append */
    time =
        WADRprepareAppend(wadres, &rwad, NewDir, NewNtry, &dirpos, &ntry,
                          &rsize);
    /* append PWAD into PWAD, restorable */
    estart = HDRinsertWad(&rwad, &ewad, &esize);
    /* append texu/pname */
    if (TexuMrg == true) {
        WADRalign4(&rwad);
        start = WADRposition(&rwad);
        size = TXUwriteTEXTUREtoWAD(&rwad);
        WADRdirAddEntry(&rwad, start, size, "TEXTURE1");
        TXUfree();
        WADRalign4(&rwad);
        start = WADRposition(&rwad);
        size = PNMwritePNAMEtoWAD(&rwad);
        WADRdirAddEntry(&rwad, start, size, "PNAMES");
        PNMfree();
    }

    /* set directory */
    HDRsetDir(&rwad, false, true, time, dirpos, ntry, rsize, estart, esize,
              wadext);
    /*end */
    WADRclose(&rwad);
    WADRclose(&ewad);
}

/*
** Add sprites and Floors
**
** must delete void entries (old DMADDS files)
** must select SPRITES of FLOORS
**
*/
void ADDallSpriteFloor(const char *wadout, const char *doomwad, const char
                       *wadres, NTRYB select)
{
    struct WADINFO iwad;
    struct WADINFO pwad;
    struct WADINFO rwad;
    int16_t n;
    ENTRY *iiden;               /*identify entry in IWAD */
    ENTRY *piden;               /*identify entry in PWAD */
    int32_t start, size, ostart, osize;
    int16_t pnm;
    char *Pnam;
    int32_t Pnamsz;
    struct WADDIR *NewDir;
    int32_t NewNtry;

    Phase("ME88", "Copying sprites and/or flats");
    Phase("ME88", " from iwad %s", fname(doomwad));
    Phase("ME88", " and  pwad %s", fname(wadres));
    Phase("ME88", " into pwad %s", fname(wadout));
    /* get iwad directory, and identify */
    iwad.ok = 0;
    WADRopenR(&iwad, doomwad);
    /*find PNAMES */
    pnm = WADRfindEntry(&iwad, "PNAMES");
    if (pnm < 0)
        ProgError("ME91", "Can't find PNAMES in main WAD");
    Pnam = WADRreadEntry(&iwad, pnm, &Pnamsz);
    /* identify iwad */
    iiden = IDENTentriesIWAD(&iwad, Pnam, Pnamsz, true);
    /* get pwad directory, and identify */
    pwad.ok = 0;
    WADRopenR(&pwad, wadres);
    piden = IDENTentriesPWAD(&pwad, Pnam, Pnamsz);
    free(Pnam);
    /*where to put pwad? at pwadstart */
    if ((iwad.maxpos | pwad.maxpos) & EXTERNAL)
        Bug("ME94", "AddExt");
    /* merge the two directories */
    NewDir =
        LISmergeDir(&NewNtry, true, true, select, &iwad, iiden, EXTERNAL,
                    &pwad, piden, 0);
    /* create a new PWAD */
    rwad.ok = 0;
    WADRopenW(&rwad, wadout, PWAD, 1);
    for (n = 0; n < NewNtry; n++) {
        ostart = NewDir[n].start;
        osize = NewDir[n].size;
        WADRalign4(&rwad);
        start = WADRposition(&rwad);
        if (ostart & EXTERNAL)
            size =
                WADRwriteWADbytes(&rwad, &iwad, ostart & (~EXTERNAL),
                                  osize);
        else
            size = WADRwriteWADbytes(&rwad, &pwad, ostart, osize);
        WADRdirAddEntry(&rwad, start, size, NewDir[n].name);
    }
    free(NewDir);
    /*close files memory */
    WADRclose(&iwad);
    WADRclose(&pwad);
    WADRwriteDir(&rwad, 1);
    WADRclose(&rwad);
    Phase("ME98", "Addition of sprites and floors is complete");
}
