cabinet.cpp File Reference

#include "rapps.h"
#include <debug.h>
#include <fdi.h>
#include <fcntl.h>

Include dependency graph for cabinet.cpp:

Go to the source code of this file.

Typedefs
typedef HFDI(*	fnFDICreate) (PFNALLOC, PFNFREE, PFNOPEN, PFNREAD, PFNWRITE, PFNCLOSE, PFNSEEK, int, PERF)
typedef BOOL(*	fnFDICopy) (HFDI, LPSTR, LPSTR, INT, PFNFDINOTIFY, PFNFDIDECRYPT, void FAR *pvUser)
typedef BOOL(*	fnFDIDestroy) (HFDI)

Functions
BOOL	WideToMultiByte (const CStringW &szSource, CStringA &szDest, UINT Codepage)
BOOL	MultiByteToWide (const CStringA &szSource, CStringW &szDest, UINT Codepage)
	FNALLOC (fnMemAlloc)
	FNFREE (fnMemFree)
	FNOPEN (fnFileOpen)
	FNREAD (fnFileRead)
	FNWRITE (fnFileWrite)
	FNCLOSE (fnFileClose)
	FNSEEK (fnFileSeek)
	FNFDINOTIFY (fnNotify)
BOOL	ExtractFilesFromCab (const ATL::CStringW &szCabName, const ATL::CStringW &szCabDir, const ATL::CStringW &szOutputDir)

Typedef Documentation

fnFDICopy

typedef BOOL(* fnFDICopy) (HFDI, LPSTR, LPSTR, INT, PFNFDINOTIFY, PFNFDIDECRYPT, void FAR *pvUser)

Definition at line 267 of file cabinet.cpp.

fnFDICreate

typedef HFDI(* fnFDICreate) (PFNALLOC, PFNFREE, PFNOPEN, PFNREAD, PFNWRITE, PFNCLOSE, PFNSEEK, int, PERF)

Definition at line 257 of file cabinet.cpp.

fnFDIDestroy

typedef BOOL(* fnFDIDestroy) (HFDI)

Definition at line 275 of file cabinet.cpp.

Function Documentation

ExtractFilesFromCab()

BOOL ExtractFilesFromCab	(	const ATL::CStringW &	szCabName,
		const ATL::CStringW &	szCabDir,
		const ATL::CStringW &	szOutputDir
	)

Definition at line 281 of file cabinet.cpp.

{
    HINSTANCE hCabinetDll;
    HFDI ExtractHandler;
    ERF ExtractErrors;
    ATL::CStringA szCabNameUTF8, szCabDirUTF8, szOutputDirUTF8;
    fnFDICreate pfnFDICreate;
    fnFDICopy pfnFDICopy;
    fnFDIDestroy pfnFDIDestroy;
    BOOL bResult;

    // Load cabinet.dll and extract needed functions 
    hCabinetDll = LoadLibraryW(L"cabinet.dll");

    if (!hCabinetDll)
    {
        return FALSE;
    }

    pfnFDICreate = (fnFDICreate) GetProcAddress(hCabinetDll, "FDICreate");
    pfnFDICopy = (fnFDICopy) GetProcAddress(hCabinetDll, "FDICopy");
    pfnFDIDestroy = (fnFDIDestroy) GetProcAddress(hCabinetDll, "FDIDestroy");

    if (!pfnFDICreate || !pfnFDICopy || !pfnFDIDestroy)
    {
        FreeLibrary(hCabinetDll);
        return FALSE;
    }

    // Create FDI context
    ExtractHandler = pfnFDICreate(fnMemAlloc,
                                  fnMemFree,
                                  fnFileOpen,
                                  fnFileRead,
                                  fnFileWrite,
                                  fnFileClose,
                                  fnFileSeek,
                                  cpuUNKNOWN,
                                  &ExtractErrors);

    if (!ExtractHandler)
    {
        FreeLibrary(hCabinetDll);
        return FALSE;
    }

    // Create output dir
    bResult = CreateDirectoryW(szOutputDir, NULL);
    
    if (bResult || GetLastError() == ERROR_ALREADY_EXISTS)
    {
        // Convert wide strings to UTF-8
        bResult = WideToMultiByte(szCabName, szCabNameUTF8, CP_UTF8);
        bResult &= WideToMultiByte(szCabDir, szCabDirUTF8, CP_UTF8);
        bResult &= WideToMultiByte(szOutputDir, szOutputDirUTF8, CP_UTF8);
    }

    // Perform extraction
    if (bResult)
    {
        // Add a slash to cab name as required by the api
        szCabNameUTF8 = "\\" + szCabNameUTF8;

        bResult = pfnFDICopy(ExtractHandler,
                             (LPSTR) szCabNameUTF8.GetString(),
                             (LPSTR) szCabDirUTF8.GetString(),
                             0,
                             fnNotify,
                             NULL,
                             (void FAR *) szOutputDirUTF8.GetString());
    }

    pfnFDIDestroy(ExtractHandler);
    FreeLibrary(hCabinetDll);
    return bResult;
}

Referenced by CAvailableApps::UpdateAppsDB().

FNALLOC()

FNALLOC

(

fnMemAlloc

)

Definition at line 84 of file cabinet.cpp.

{
    return HeapAlloc(GetProcessHeap(), NULL, cb);
}

FNCLOSE()

FNCLOSE

(

fnFileClose

)

Definition at line 162 of file cabinet.cpp.

{
    return (CloseHandle((HANDLE) hf) != FALSE) ? 0 : -1;
}

FNFDINOTIFY()

FNFDINOTIFY

(

fnNotify

)

Definition at line 174 of file cabinet.cpp.

{
    INT_PTR iResult = 0;

    switch (fdint)
    {
    case fdintCOPY_FILE:
    {
        CStringW szExtractDir, szCabFileName;

        // Append the destination directory to the file name.
        MultiByteToWide((LPCSTR) pfdin->pv, szExtractDir, CP_UTF8);
        MultiByteToWide(pfdin->psz1, szCabFileName, CP_ACP);

        if (szCabFileName.Find('\\') >= 0)
        {
            CStringW szNewDirName = szExtractDir;
            int nTokenPos = 0;
            // We do not want to interpret the filename as directory,
            // so bail out before the last token!
            while (szCabFileName.Find('\\', nTokenPos) >= 0)
            {
                CStringW token = szCabFileName.Tokenize(L"\\", nTokenPos);
                if (token.IsEmpty())
                    break;

                szNewDirName += L"\\" + token;
                if (!CreateDirectoryW(szNewDirName, NULL))
                {
                    DWORD dwErr = GetLastError();
                    if (dwErr != ERROR_ALREADY_EXISTS)
                    {
                        DPRINT1("ERROR: Unable to create directory %S (err %lu)\n", szNewDirName.GetString(), dwErr);
                    }
                }
            }
        }

        CStringW szNewFileName = szExtractDir + L"\\" + szCabFileName;

        CStringA szFilePathUTF8;
        WideToMultiByte(szNewFileName, szFilePathUTF8, CP_UTF8);

        // Open the file
        iResult = fnFileOpen((LPSTR) szFilePathUTF8.GetString(), 
                             _O_WRONLY | _O_CREAT,
                             0);
    }
    break;

    case fdintCLOSE_FILE_INFO:
        iResult = !fnFileClose(pfdin->hf);
        break;

    case fdintNEXT_CABINET:
        if (pfdin->fdie != FDIERROR_NONE)
        {
            iResult = -1;
        }
        break;

    case fdintPARTIAL_FILE:
        iResult = 0;
        break;

    case fdintCABINET_INFO:
        iResult = 0;
        break;

    case fdintENUMERATE:
        iResult = 0;
        break;

    default:
        iResult = -1;
        break;
    }

    return iResult;
}

FNFREE()

FNFREE

(

fnMemFree

)

Definition at line 89 of file cabinet.cpp.

{
    HeapFree(GetProcessHeap(), NULL, pv);
}

FNOPEN()

FNOPEN

(

fnFileOpen

)

Definition at line 94 of file cabinet.cpp.

{
    HANDLE hFile = NULL;
    DWORD dwDesiredAccess = 0;
    DWORD dwCreationDisposition = 0;
    ATL::CStringW szFileName;

    UNREFERENCED_PARAMETER(pmode);

    if (oflag & _O_RDWR)
    {
        dwDesiredAccess = GENERIC_READ | GENERIC_WRITE;
    }
    else if (oflag & _O_WRONLY)
    {
        dwDesiredAccess = GENERIC_WRITE;
    }
    else
    {
        dwDesiredAccess = GENERIC_READ;
    }

    if (oflag & _O_CREAT)
    {
        dwCreationDisposition = CREATE_ALWAYS;
    }
    else
    {
        dwCreationDisposition = OPEN_EXISTING;
    }

    MultiByteToWide(pszFile, szFileName, CP_UTF8);

    hFile = CreateFileW(szFileName,
                        dwDesiredAccess,
                        FILE_SHARE_READ,
                        NULL,
                        dwCreationDisposition,
                        FILE_ATTRIBUTE_NORMAL,
                        NULL);

    return (INT_PTR) hFile;
}

FNREAD()

FNREAD

(

fnFileRead

)

Definition at line 138 of file cabinet.cpp.

{
    DWORD dwBytesRead = 0;

    if (ReadFile((HANDLE) hf, pv, cb, &dwBytesRead, NULL) == FALSE)
    {
        dwBytesRead = (DWORD) -1L;
    }

    return dwBytesRead;
}

FNSEEK()

FNSEEK

(

fnFileSeek

)

Definition at line 167 of file cabinet.cpp.

{
    return SetFilePointer((HANDLE) hf, dist, NULL, seektype);
}

FNWRITE()

FNWRITE

(

fnFileWrite

)

Definition at line 150 of file cabinet.cpp.

{
    DWORD dwBytesWritten = 0;

    if (WriteFile((HANDLE) hf, pv, cb, &dwBytesWritten, NULL) == FALSE)
    {
        dwBytesWritten = (DWORD) -1;
    }

    return dwBytesWritten;
}

MultiByteToWide()

BOOL MultiByteToWide ( const CStringA &  szSource, CStringW &  szDest, UINT  Codepage  )

inline

Definition at line 56 of file cabinet.cpp.

{
    // determine the needed size
    INT sz = MultiByteToWideChar(Codepage,
                                    0,
                                    szSource,
                                    -1,
                                    NULL,
                                    NULL);
    if (!sz)
        return FALSE;
        
    // do the actual conversion
    sz = MultiByteToWideChar(CP_UTF8,
                                0,
                                szSource,
                                -1,
                                szDest.GetBuffer(sz),
                                sz);

    szDest.ReleaseBuffer();
    return sz != 0;
}

Referenced by FNFDINOTIFY(), and FNOPEN().

WideToMultiByte()

BOOL WideToMultiByte ( const CStringW &  szSource, CStringA &  szDest, UINT  Codepage  )

inline

Definition at line 25 of file cabinet.cpp.

{
    // determine the needed size
    INT sz = WideCharToMultiByte(Codepage,
                                    0,
                                    szSource,
                                    -1,
                                    NULL,
                                    NULL,
                                    NULL,
                                    NULL);
    if (!sz)
        return FALSE;

    // do the actual conversion
    sz = WideCharToMultiByte(Codepage,
                                0,
                                szSource,
                                -1,
                                szDest.GetBuffer(sz),
                                sz,
                                NULL,
                                NULL);

    szDest.ReleaseBuffer();
    return sz != 0;
}

Referenced by ExtractFilesFromCab(), and FNFDINOTIFY().