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.

Classes
struct	NotifyData

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 CStringW &szCabName, const CStringW &szCabDir, const CStringW &szOutputDir, EXTRACTCALLBACK Callback, void *Cookie)
BOOL	ExtractFilesFromCab (LPCWSTR FullCabPath, const CStringW &szOutputDir, EXTRACTCALLBACK Callback, void *Cookie)

Typedef Documentation

fnFDICopy

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

Definition at line 242 of file cabinet.cpp.

fnFDICreate

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

Definition at line 240 of file cabinet.cpp.

fnFDIDestroy

typedef BOOL(* fnFDIDestroy) (HFDI)

Definition at line 244 of file cabinet.cpp.

Function Documentation

ExtractFilesFromCab() [1/2]

BOOL ExtractFilesFromCab	(	const CStringW &	szCabName,
		const CStringW &	szCabDir,
		const CStringW &	szOutputDir,
		EXTRACTCALLBACK	Callback,
		void *	Cookie
	)

Definition at line 250 of file cabinet.cpp.

{
    HINSTANCE hCabinetDll;
    HFDI ExtractHandler;
    ERF ExtractErrors;
    ATL::CStringA szCabNameUTF8, szCabDirUTF8, szOutputDirUTF8;
    fnFDICreate pfnFDICreate;
    fnFDICopy pfnFDICopy;
    fnFDIDestroy pfnFDIDestroy;
    BOOL bResult;
    NotifyData nd = { Callback, Cookie };
 
    // 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;
 
        nd.OutputDir = szOutputDirUTF8.GetString();
        bResult = pfnFDICopy(
            ExtractHandler, (LPSTR)szCabNameUTF8.GetString(), (LPSTR)szCabDirUTF8.GetString(), 0, fnNotify, NULL,
            (void FAR *)&nd);
    }
 
    pfnFDIDestroy(ExtractHandler);
    FreeLibrary(hCabinetDll);
    return bResult;
}

Referenced by ExtractCab(), ExtractFilesFromCab(), and CAppDB::UpdateAvailable().

ExtractFilesFromCab() [2/2]

BOOL ExtractFilesFromCab	(	LPCWSTR	FullCabPath,
		const CStringW &	szOutputDir,
		EXTRACTCALLBACK	Callback,
		void *	Cookie
	)

Definition at line 321 of file cabinet.cpp.

{
    CStringW dir, file = SplitFileAndDirectory(FullCabPath, &dir);
    return ExtractFilesFromCab(file, dir, szOutputDir, Callback, Cookie);
}

FNALLOC()

FNALLOC

(

fnMemAlloc

)

Definition at line 65 of file cabinet.cpp.

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

FNCLOSE()

FNCLOSE

(

fnFileClose

)

Definition at line 138 of file cabinet.cpp.

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

FNFDINOTIFY()

FNFDINOTIFY

(

fnNotify

)

Definition at line 150 of file cabinet.cpp.

{
    INT_PTR iResult = 0;
    NotifyData *pND = (NotifyData *)pfdin->pv;
 
    switch (fdint)
    {
        case fdintCOPY_FILE:
        {
            CStringW szExtractDir, szCabFileName;
 
            // Append the destination directory to the file name.
            MultiByteToWide(pND->OutputDir, szExtractDir, CP_UTF8);
            UINT codepage = (pfdin->attribs & _A_NAME_IS_UTF) ? CP_UTF8 : CP_ACP;
            MultiByteToWide(pfdin->psz1, szCabFileName, codepage);
 
            if (!NotifyFileExtractCallback(szCabFileName, pfdin->cb, pfdin->attribs,
                                           pND->Callback, pND->CallerCookie))
            {
                break; // Skip file
            }
 
            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 70 of file cabinet.cpp.

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

FNOPEN()

FNOPEN

(

fnFileOpen

)

Definition at line 75 of file cabinet.cpp.

{
    HANDLE hFile = NULL;
    DWORD dwDesiredAccess = 0;
    DWORD dwCreationDisposition = 0;
    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 114 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 143 of file cabinet.cpp.

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

FNWRITE()

FNWRITE

(

fnFileWrite

)

Definition at line 126 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 42 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(Codepage, 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 26 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().