157 lines
7.4 KiB
C
157 lines
7.4 KiB
C
|
#ifndef _BITBUCK_INC
|
||
|
#define _BITBUCK_INC
|
||
|
|
||
|
#include "ids.h"
|
||
|
#include "undo.h"
|
||
|
|
||
|
// whacky #defines
|
||
|
|
||
|
#define DELETEMAX 100000
|
||
|
#define MAX_BITBUCKETS 27
|
||
|
#define MAX_DRIVES 26
|
||
|
#define OPENFILERETRYTIME 500
|
||
|
#define OPENFILERETRYCOUNT 10
|
||
|
#define SERVERDRIVE 26
|
||
|
#define MAX_EMPTY_FILES 100 // if we have MAX_EMPTY_FILES or more then we use the generic "do you want to empty" message.
|
||
|
|
||
|
#define TF_BITBUCKET 0x10000000
|
||
|
// #define TF_BITBUCKET TF_ERROR
|
||
|
|
||
|
//
|
||
|
// NOTE: The on-disk format of the recycle bin should never have to change again.
|
||
|
// If you think you need to change it, then you are wrong.
|
||
|
//
|
||
|
#define BITBUCKET_WIN95_VERSION 0 // (info) Ansi Win95, OSR2 wastebasket
|
||
|
#define BITBUCKET_NT4_VERSION 2 // (info) Unicode NT4 wastebasket
|
||
|
#define BITBUCKET_WIN98IE4INT_VERSION 4 // (info2) win9x+IE4 integrated, and win98 wastebasket
|
||
|
#define BITBUCKET_FINAL_VERSION 5 // (info2) NT4+IE4 integraged, win2k, millenium, every future os wastebasket
|
||
|
|
||
|
#define OPENBBINFO_READ 0x00000000
|
||
|
#define OPENBBINFO_WRITE 0x00000001
|
||
|
#define OPENBBINFO_CREATE 0x00000003
|
||
|
|
||
|
#define IsDeletedEntry(pbbde) (! (((BBDATAENTRYA*)pbbde)->szOriginal[0]) )
|
||
|
#define MarkEntryDeleted(pbbde) ((BBDATAENTRYA*)pbbde)->szOriginal[0] = '\0';
|
||
|
|
||
|
// this is the old (win95) data header. it's maintained in the info file
|
||
|
// but only used for verification. for current stuff look at the driveinfo,
|
||
|
// which is kept in the registry.
|
||
|
typedef struct {
|
||
|
int idVersion;
|
||
|
int cFiles; // the # of items in this drive's recycle bin
|
||
|
int cCurrent; // the current file number.
|
||
|
UINT cbDataEntrySize; // size of each entry
|
||
|
DWORD dwSize; // total size of this recycle bin drive
|
||
|
} BBDATAHEADER;
|
||
|
|
||
|
// The bitbucket datafile (INFO on win95, INFO2 on IE4/NT5, etc...) format is as follows:
|
||
|
//
|
||
|
// (binary writes)
|
||
|
//
|
||
|
// BBDATAHEADER // header
|
||
|
// BBDATAENTRY[X] // array of BBDATAENTRYies
|
||
|
//
|
||
|
|
||
|
typedef struct {
|
||
|
CHAR szOriginal[MAX_PATH]; // original filename (if szOriginal[0] is 0, then it's a deleted entry)
|
||
|
int iIndex; // index (key to name)
|
||
|
int idDrive; // which drive bucket it's currently in
|
||
|
FILETIME ft;
|
||
|
DWORD dwSize;
|
||
|
// shouldn't need file attributes because we did a move file
|
||
|
// which should have preserved them.
|
||
|
} BBDATAENTRYA, *LPBBDATAENTRYA;
|
||
|
|
||
|
typedef struct {
|
||
|
CHAR szShortName[MAX_PATH]; // original filename, shortened (if szOriginal[0] is 0, then it's a deleted entry)
|
||
|
int iIndex; // index (key to name)
|
||
|
int idDrive; // which drive bucket it's currently in
|
||
|
FILETIME ft;
|
||
|
DWORD dwSize;
|
||
|
WCHAR szOriginal[MAX_PATH]; // original filename
|
||
|
} BBDATAENTRYW, *LPBBDATAENTRYW;
|
||
|
|
||
|
typedef BBDATAENTRYA UNALIGNED *PUBBDATAENTRYA;
|
||
|
|
||
|
// On NT5 we are finally going to have cross-process syncrhonization to
|
||
|
// the Recycle Bin. We replaced the global LPBBDRIVEINFO array with an
|
||
|
// array of the following structures:
|
||
|
typedef struct {
|
||
|
BOOL fInited; // is this particular BBSYNCOBJECT fully inited (needed when we race to create it)
|
||
|
HANDLE hgcNextFileNum; // a global counter that garuntees unique deleted file names
|
||
|
HANDLE hgcDirtyCount; // a global counter to tell us if we need to re-read the bitbucket settings from the registry (percent, max size, etc)
|
||
|
LONG lCurrentDirtyCount; // out current dirty count; we compare this to hgcDirtyCount to see if we need to update the settings from the registry
|
||
|
HKEY hkey; // HKLM reg key, under which we store the settings for this specific bucket (iPercent and fNukeOnDelete).
|
||
|
HKEY hkeyPerUser; // HKCU reg key, under which we have volital reg values indicatiog if there is a need to purge or compact this bucket
|
||
|
|
||
|
BOOL fIsUnicode; // is this a bitbucket on a drive whose INFO2 file uses BBDATAENTRYW structs?
|
||
|
int iPercent; // % of the drive to use for the bitbucket
|
||
|
DWORD cbMaxSize; // maximum size of bitbucket (in bytes), NOTE: we use a dword because the biggest the BB can ever grow to is 4 gigabytes.
|
||
|
DWORD dwClusterSize; // cluster size of this volume, needed to round all the file sizes
|
||
|
ULONGLONG qwDiskSize; // total size of the disk - takes into account quotas on NTFS
|
||
|
BOOL fNukeOnDelete; // I love the smell of napalm in the morning.
|
||
|
|
||
|
LPITEMIDLIST pidl; // pidl = bitbucket dir for this drive
|
||
|
int cchBBDir; // # of characters that makes up the bitbucket directory.
|
||
|
|
||
|
} BBSYNCOBJECT;
|
||
|
|
||
|
#define c_szInfo2 TEXT("INFO2") // version 2 of the db file (used in IE4, win98, NT5, ...)
|
||
|
#define c_szInfo TEXT("INFO") // version 1 of the db file (used in win95, osr2, NT4)
|
||
|
#define c_szDStarDotStar TEXT("D*.*")
|
||
|
|
||
|
// globals
|
||
|
|
||
|
EXTERN_C BBSYNCOBJECT *g_pBitBucket[MAX_BITBUCKETS];
|
||
|
EXTERN_C HKEY g_hkBitBucket;
|
||
|
EXTERN_C HANDLE g_hgcNumDeleters;
|
||
|
|
||
|
// prototypes by bitbuck.c, bbckfldr.cpp
|
||
|
|
||
|
STDAPI_(BOOL) InitBBGlobals();
|
||
|
STDAPI_(void) BitBucket_Terminate();
|
||
|
STDAPI_(BOOL) IsBitBucketableDrive(int idDrive);
|
||
|
STDAPI_(int) DriveIDFromBBPath(LPCTSTR pszPath);
|
||
|
STDAPI_(void) UpdateIcon(BOOL fFull);
|
||
|
STDAPI_(void) NukeFileInfoBeforePoint(HANDLE hfile, LPBBDATAENTRYW pbbdew, DWORD dwDataEntrySize);
|
||
|
STDAPI_(BOOL) ReadNextDataEntry(HANDLE hfile, LPBBDATAENTRYW pbbde, BOOL fSkipDeleted, int idDrive);
|
||
|
STDAPI_(void) CloseBBInfoFile(HANDLE hFile, int idDrive);
|
||
|
STDAPI_(HANDLE) OpenBBInfoFile(int idDrive, DWORD dwFlags, int iRetryCount);
|
||
|
STDAPI_(int) BBPathToIndex(LPCTSTR pszPath);
|
||
|
STDAPI BBFileNameToInfo(LPCTSTR pszFileName, int *pidDrive, int *piIndex);
|
||
|
STDAPI_(void) GetDeletedFileName(LPTSTR pszFileName, const BBDATAENTRYW *pbbdew);
|
||
|
STDAPI_(void) DriveIDToBBPath(int idDrive, LPTSTR pszPath);
|
||
|
STDAPI_(void) DriveIDToBBRoot(int idDrive, LPTSTR szPath);
|
||
|
STDAPI_(void) DriveIDToBBVolumeRoot(int idDrive, LPTSTR szPath);
|
||
|
STDAPI_(BOOL) GetNetHomeDir(LPTSTR pszPath);
|
||
|
STDAPI_(BOOL) PersistBBDriveSettings(int idDrive, int iPercent, BOOL fNukeOnDelete);
|
||
|
STDAPI_(BOOL) MakeBitBucket(int idDrive);
|
||
|
STDAPI_(DWORD) PurgeBBFiles(int idDrive);
|
||
|
STDAPI_(BOOL) PersistGlobalSettings(BOOL fUseGlobalSettings, BOOL fNukeOnDelete, int iPercent);
|
||
|
STDAPI_(BOOL) RefreshAllBBDriveSettings();
|
||
|
STDAPI_(BOOL) RefreshBBDriveSettings(int idDrive);
|
||
|
STDAPI_(void) CheckCompactAndPurge();
|
||
|
STDAPI BBPurgeAll(HWND hwndOwner, DWORD dwFlags);
|
||
|
STDAPI_(BOOL) BBDeleteFileInit(LPTSTR pszFile, INT* piRet);
|
||
|
STDAPI_(BOOL) BBDeleteFile(LPTSTR pszFile, INT* piRet, LPUNDOATOM lpua, BOOL fIsDir, HDPA *phdpaDeletedFiles, ULARGE_INTEGER ulSize);
|
||
|
STDAPI_(BOOL) BBFinishDelete(HDPA hdpaDeletedFiles);
|
||
|
STDAPI_(BOOL) IsFileInBitBucket(LPCTSTR pszPath);
|
||
|
STDAPI_(void) UndoBBFileDelete(LPCTSTR pszOriginal, LPCTSTR pszDelFile);
|
||
|
STDAPI_(BOOL) BBWillRecycle(LPCTSTR pszFile, INT* piRet);
|
||
|
STDAPI_(void) BBCheckRestoredFiles(LPCTSTR pszSrc);
|
||
|
STDAPI_(BOOL) BBCheckDeleteFileSize(int idDrive, ULARGE_INTEGER ulSize);
|
||
|
STDAPI_(BOOL) IsFileDeletable(LPCTSTR pszFile);
|
||
|
STDAPI_(BOOL) IsDirectoryDeletable(LPCTSTR pszDir);
|
||
|
STDAPI_(int) BBRecyclePathLength(int idDrive);
|
||
|
|
||
|
|
||
|
STDAPI_(BOOL) IsRecycleBinEmpty();
|
||
|
STDAPI_(void) SHUpdateRecycleBinIcon();
|
||
|
STDAPI_(void) SaveRecycleBinInfo();
|
||
|
|
||
|
STDAPI_(void) SetDateTimeText(HWND hdlg, int id, const FILETIME *pftUTC);
|
||
|
|
||
|
STDAPI_(DWORD) ReadPolicySetting(LPCWSTR pszBaseKey, LPCWSTR pszGroup, LPCWSTR pszRestriction, LPBYTE pbData, DWORD cbData);
|
||
|
|
||
|
#endif // _BITBUCK_INC
|