#include "verpriv.h"
#include <lzexpand.h>
#include "diamondd.h"
#include "mydiam.h"


BOOL    FileInUse(LPWSTR lpszFilePath, LPWSTR lpszFileName);
DWORD   MakeFileName(LPWSTR lpDst, LPWSTR lpDir, LPWSTR lpFile, int cchDst);

typedef struct tagVS_VERSION
  {
    WORD wTotLen;
    WORD wValLen;
    WORD wType;
    WCHAR szSig[16];
    VS_FIXEDFILEINFO vInfo;
  } VS_VERSION;

typedef struct tagLANGANDCP
  {
    WORD wLanguage;
    WORD wCodePage;
  } LANGANDCP;

WCHAR szTrans[] = TEXT("\\VarFileInfo\\Translation");
WCHAR szTempHdr[] = TEXT("temp.");


/* The routines in here will find a file on a path, and an environment
 * variable.  The constants _MAX_PATH and need to be defined
 * by the including module, plus the constant WINDOWS should
 * be defined if this is to be used in Windows, so that lstrcmpi
 * and lstrlen will not be compiled
 */


VOID Ver3IToA(LPWSTR lpwStr, int n)
{
  int nShift;
  WCHAR cTemp;

  for (nShift=8; nShift>=0; nShift-=4, ++lpwStr)
    {
      if ((cTemp = (WCHAR)((n>>nShift)&0x000f)) >= 10)
          *lpwStr = (WCHAR)('A' + cTemp - 10);
      else
          *lpwStr = (WCHAR)('0' + cTemp     );
    }
  *lpwStr = 0;
}


/* Convert a DOS error into an error flag
 */
DWORD FileErrFlag(int err)
{
  switch(err)
    {
      case 0x05:
          return(VIF_ACCESSVIOLATION);

      case 0x20:
          return(VIF_SHARINGVIOLATION);

      default:
          return(0);
    }
}


/* Create the given file with default flags; global nFileErr will
 * receive any DOS error; returns -1 on error, otherwise the DOS
 * file handle.
 */
HANDLE VerCreateFile(LPWSTR lpszFile)
{
  HANDLE hFile;

  hFile = CreateFile(lpszFile, GENERIC_READ | GENERIC_WRITE,
      FILE_SHARE_READ, NULL, CREATE_NEW,
      FILE_ATTRIBUTE_NORMAL, NULL);

  return(hFile);
}


VOID VerClose(HANDLE hW32File)
{
  CloseHandle(hW32File);
}


#define MyAlloc(x) ((WCHAR *)LocalAlloc(LMEM_FIXED, x))
#define MyFree(x) LocalFree((HANDLE)(x))

LPVOID MyGetFileVersionInfo(LPWSTR lpFileName)
{
  WCHAR *pInfo;
  WORD wLen = 2048;

TryAgain:
  if (!(pInfo=MyAlloc(wLen)))
      goto Error1;
  if(!GetFileVersionInfo(lpFileName, 0L, wLen, pInfo))
      goto Error2;
  if (wLen < *(WORD *)pInfo)
    {
      wLen = *(WORD *)pInfo;
      MyFree(pInfo);
      goto TryAgain;
    }
  return(pInfo);

Error2:
  MyFree(pInfo);
Error1:
  return(NULL);
}


DWORD
APIENTRY
VerInstallFileW(
        DWORD wFlags,
        LPWSTR lpszSrcFileName,
        LPWSTR lpszDstFileName,
        LPWSTR lpszSrcDir,
        LPWSTR lpszDstDir,
        LPWSTR lpszCurDir,
        LPWSTR lpszTmpFile,
        PUINT puTmpFileLen
        )
{
  WCHAR szSrcFile[_MAX_PATH];
  WCHAR szDstFile[_MAX_PATH];
  WCHAR szCurFile[_MAX_PATH];
  DWORD dwRetVal = 0L, dwSrcAttr;
  WORD wDirLen;
  LONG lCopy;
  HANDLE hW32Out;
  int i, fIn, fDosOut;
  OFSTRUCT of;
  BOOL DiamondFile;

  LogData("inst", __LINE__, (DWORD)puTmpFileLen);

  /* LZ Open the source for reading
   */
  MakeFileName(szSrcFile, lpszSrcDir, lpszSrcFileName, ARRAYSIZE(szSrcFile));
  InitDiamond();
  if ((fIn=LZOpenFile(szSrcFile, &of, 0)) < 0)
    {
      dwRetVal |= VIF_CANNOTREADSRC;
      goto doReturn;
    }

  LogData("inst", __LINE__, (DWORD)puTmpFileLen);

  DiamondFile = IsDiamondFile(of.szPathName);

  if (DiamondFile) {
      LZClose(fIn);
  }

  LogData("inst", __LINE__, (DWORD)puTmpFileLen);

  /* If the dest file exists and is read only, return immediately;
   * the calling app must change the attributes before continuing
   * In the library version, we assume the file is in use if it exists
   * and we are in a VM; check all other possible errors and then return,
   * so the calling app can override our determination of "in use"
   * on a second call, along with all other problems
   */
  wDirLen = (WORD)MakeFileName(szDstFile, lpszDstDir, lpszDstFileName, ARRAYSIZE(szDstFile));
  lstrcpy(szSrcFile, szDstFile);
  if (!HIWORD(dwSrcAttr=GetFileAttributes(szSrcFile)))
    {
      LogData("inst", __LINE__, (DWORD)puTmpFileLen);
      if (LOWORD(dwSrcAttr)&0x01)
        {
          LogData("inst", __LINE__, (DWORD)puTmpFileLen);
          dwRetVal |= VIF_WRITEPROT;
          goto doCloseSrc;
        }

      if (FileInUse(szSrcFile, lpszDstFileName))
        {
          LogData("inst", __LINE__, (DWORD)puTmpFileLen);
          dwRetVal |= VIF_FILEINUSE;
          goto doCloseSrc;
        }
    }

  /* If this is a force install and there is a temp file name from a
   * previous call to this function, use that as the temp file name
   */
  LogData("inst", __LINE__, (DWORD)puTmpFileLen);
  if ((wFlags&VIFF_FORCEINSTALL) && *lpszTmpFile)
    {
      LogData("inst", __LINE__, (DWORD)puTmpFileLen);
      LogData("fnam", (DWORD)lpszDstDir, (DWORD)lpszTmpFile);
      MakeFileName(szSrcFile, lpszDstDir, lpszTmpFile, ARRAYSIZE(szSrcFile));
      LogData("inst", __LINE__, (DWORD)puTmpFileLen);
      LogData("srcf", (DWORD)szSrcFile, *(LPDWORD)szSrcFile);
      if (!HIWORD(GetFileAttributes(szSrcFile)))
        {
          LogData("inst", __LINE__, (DWORD)puTmpFileLen);
          goto doCheckDstFile;
        }
    }

  /* Determine a file name that is not in use; try names of the form:
   * temp.nnn where nnn is a three digit hex number.  If we get to
   * 0xfff, we have a serious file system problem.  Create the file.
   */
  LogData("inst", __LINE__, (DWORD)puTmpFileLen);
  lstrcpy(szSrcFile+wDirLen, szTempHdr);
  for (i=0; ; ++i)
    {
      Ver3IToA(szSrcFile+wDirLen+lstrlen(szTempHdr), i);
      LogData("inst", __LINE__, (DWORD)puTmpFileLen);
      if (HIWORD(GetFileAttributes(szSrcFile)))
          break;
      if (i > 0xfff)
        {
          dwRetVal |= VIF_CANNOTCREATE;
          goto doCloseSrc;
        }
    }
  /* Copy the file, and fill in appropriate errors
   */

  LogData("inst", __LINE__, (DWORD)puTmpFileLen);

  if (DiamondFile) {
      LZINFO lzi;
      lCopy = ExpandDiamondFile(of.szPathName,  // PARSED name!!
                                szSrcFile,
                                FALSE,
                                &lzi);
      LogData("inst", __LINE__, (DWORD)puTmpFileLen);
  } else {
      if ((hW32Out=VerCreateFile(szSrcFile)) == INVALID_HANDLE_VALUE)
        {
          dwRetVal |= VIF_CANNOTCREATE | FileErrFlag(GetLastError());
          goto doCloseSrc;
        }

      LogData("inst", __LINE__, (DWORD)puTmpFileLen);

      fDosOut = LZInit((INT)hW32Out);
      lCopy = LZCopy(fIn, fDosOut);
      LZClose(fDosOut);
  }

  LogData("inst", __LINE__, (DWORD)puTmpFileLen);

  switch (lCopy)
    {
      case LZERROR_BADINHANDLE:
      case LZERROR_READ:
      case LZERROR_BADVALUE:
      case LZERROR_UNKNOWNALG:
          dwRetVal |= VIF_CANNOTREADSRC;
          goto doDelTempFile;

      case LZERROR_BADOUTHANDLE:
      case LZERROR_WRITE:
          dwRetVal |= VIF_OUTOFSPACE;
          goto doDelTempFile;

      case LZERROR_GLOBALLOC:
      case LZERROR_GLOBLOCK:
          dwRetVal |= VIF_OUTOFMEMORY;
          goto doDelTempFile;

      default:
          break;
    }

  /* If the destination exists, check the versions of the two files,
   * and only copy if the src file is at least as new as the dst, and
   * they are the same type and in the same language and codepage
   */
doCheckDstFile:
  if (!HIWORD(dwSrcAttr))
    {
        VS_VERSION *pSrcVer, *pDstVer;
        LANGANDCP *lpSrcTrans, *lpDstTrans;
        DWORD   dwSrcNum, dwDstNum;
        DWORD   dwSrcTrans, dwDstTrans;

      LogData("inst", __LINE__, (DWORD)puTmpFileLen);
      if (!(wFlags & VIFF_FORCEINSTALL) &&
            (pDstVer=MyGetFileVersionInfo(szDstFile)))
        {
          LogData("inst", __LINE__, (DWORD)puTmpFileLen);
          if (!(pSrcVer=MyGetFileVersionInfo(szSrcFile)))
            {
              dwRetVal |= VIF_MISMATCH | VIF_SRCOLD;
            }
          else
            {
              LogData("inst", __LINE__, (DWORD)puTmpFileLen);
              if (pDstVer->vInfo.dwFileVersionMS>pSrcVer->vInfo.dwFileVersionMS
        || (pDstVer->vInfo.dwFileVersionMS==pSrcVer->vInfo.dwFileVersionMS &&
        pDstVer->vInfo.dwFileVersionLS>pSrcVer->vInfo.dwFileVersionLS))
                  dwRetVal |= VIF_MISMATCH | VIF_SRCOLD;

              if (pDstVer->vInfo.dwFileType!=pSrcVer->vInfo.dwFileType ||
                    pDstVer->vInfo.dwFileSubtype!=pSrcVer->vInfo.dwFileSubtype)
                  dwRetVal |= VIF_MISMATCH | VIF_DIFFTYPE;

              if (VerQueryValueW(pDstVer, szTrans, (LPVOID)&lpDstTrans, &dwDstNum) &&
                  VerQueryValueW(pSrcVer, szTrans, (LPVOID)&lpSrcTrans, &dwSrcNum))
                {
                  dwDstNum /= sizeof(DWORD);
                  dwSrcNum /= sizeof(DWORD);

                  for (dwDstTrans=0; dwDstTrans<dwDstNum; ++dwDstTrans)
                    {
                      for (dwSrcTrans=0; ; ++dwSrcTrans)
                        {
                          if (dwSrcTrans >= dwSrcNum)
                            {
                              dwRetVal |= VIF_MISMATCH | VIF_DIFFLANG;
                              break;
                            }

                          if (lpDstTrans[dwDstTrans].wLanguage
                                == lpSrcTrans[dwSrcTrans].wLanguage)
                            {
                              /* OK if dst is CP0 and src is not UNICODE
                               */
                              if (lpDstTrans[dwDstTrans].wCodePage==0 &&
                                    lpSrcTrans[dwSrcTrans].wCodePage!=1200)
                                  break;
                              if (lpDstTrans[dwDstTrans].wCodePage
                                    == lpSrcTrans[dwSrcTrans].wCodePage)
                                  break;
                            }
                        }
                    }
                }

              LogData("inst", __LINE__, (DWORD)puTmpFileLen);
              MyFree(pSrcVer);
            }
          LogData("inst", __LINE__, (DWORD)puTmpFileLen);
          MyFree(pDstVer);
        }

      /* If there were no errors, delete the currently existing file
       */
      LogData("inst", __LINE__, (DWORD)puTmpFileLen);
      if (!dwRetVal && !DeleteFile(szDstFile))
        {
          dwRetVal |= VIF_CANNOTDELETE | FileErrFlag(GetLastError());
          goto doDelTempFile;
        }
    }

  /* If there were no errors, rename the temp file (any existing file
   * should have been deleted by now).  Otherwise, if we created a valid
   * temp file, then pass along the temp file name.
   */
  LogData("inst", __LINE__, (DWORD)puTmpFileLen);

  if (dwRetVal)
    {
      DWORD wTemp;

      if (*puTmpFileLen > (wTemp=lstrlen(szSrcFile+wDirLen)))
        {
          lstrcpy(lpszTmpFile, szSrcFile+wDirLen);
          dwRetVal |= VIF_TEMPFILE;
        }
      else
        {
          dwRetVal |= VIF_BUFFTOOSMALL;
          DeleteFile(szSrcFile);
        }
      *puTmpFileLen = wTemp + 1;
    }
  else
    {
      /* Delete the currently existing file; this gets done before renaming
       * the temp file in case someone has tried to mess us up with a weird
       * directory name that would allow us to delete the newly installed
       * file.
       */
      if (!(wFlags&VIFF_DONTDELETEOLD) &&
            lpszCurDir && *lpszCurDir && lstrcmpi(lpszCurDir, lpszDstDir))
        {
          MakeFileName(szCurFile, lpszCurDir, lpszDstFileName, ARRAYSIZE(szCurFile));
          if (!HIWORD(GetFileAttributes(szCurFile)) &&
                (FileInUse(szCurFile, lpszDstFileName) ||
                !DeleteFile(szCurFile)))
              dwRetVal |= VIF_CANNOTDELETECUR | FileErrFlag(GetLastError());
        }

      if (!MoveFile(szSrcFile, szDstFile))
        {
          dwRetVal |= VIF_CANNOTRENAME | FileErrFlag(GetLastError());
doDelTempFile:
          DeleteFile(szSrcFile);
        }
    }

doCloseSrc:
  if (!DiamondFile) {
      LZClose(fIn);
  }
doReturn:
  LogData("inst", __LINE__, (DWORD)puTmpFileLen);
  TermDiamond();
  return(dwRetVal);
}