utils.h File Reference

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
struct	_msi_table

Macros
#define	ADD_TABLE(x)   {#x".idt", x##_dat, sizeof(x##_dat)}
#define	create_database(name, tables, num_tables)
#define	create_database_template(name, tables, num_tables, version, template)
#define	create_file(name, size)   create_file_data(name, name, size)
#define	create_pf(file, is_file)   create_pf_data(file, file, is_file)

Typedefs
typedef struct _msi_table	msi_table

Functions
BOOL	get_system_dirs (void)
BOOL	get_user_dirs (void)
void	create_database_wordcount (const char *name, const msi_table *tables, int num_tables, int version, int wordcount, const char *template, const char *packagecode)
void	create_cab_file (const char *name, DWORD max_size, const char *files)
void	create_file_data (const char *name, const char *data, DWORD size)
void	create_pf_data (const char *file, const char *data, BOOL is_file)
void	delete_cab_files (void)
BOOL	delete_pf (const char *rel_path, BOOL is_file)
BOOL	file_exists (const char *file)
BOOL	pf_exists (const char *file)
BOOL	is_process_limited (void)
UINT	run_query (MSIHANDLE hdb, MSIHANDLE hrec, const char *query)

Variables
char	PROG_FILES_DIR [MAX_PATH]
char	PROG_FILES_DIR_NATIVE [MAX_PATH]
char	COMMON_FILES_DIR [MAX_PATH]
char	APP_DATA_DIR [MAX_PATH]
char	WINDOWS_DIR [MAX_PATH]
char	CURR_DIR [MAX_PATH]

Macro Definition Documentation

ADD_TABLE

#define ADD_TABLE

(

x

)

{#x".idt", x##_dat, sizeof(x##_dat)}

Definition at line 36 of file utils.h.

create_database

#define create_database	(		name,
			tables,
			num_tables
	)

Value:

    create_database_wordcount(name, tables, num_tables, 100, 0, ";1033", \
                              "{004757CA-5092-49C2-AD20-28E1CE0DF5F2}");

Definition at line 42 of file utils.h.

create_database_template

#define create_database_template	(		name,
			tables,
			num_tables,
			version,
			template
	)

Value:

    create_database_wordcount(name, tables, num_tables, version, 0, template, \
                              "{004757CA-5092-49C2-AD20-28E1CE0DF5F2}");

Definition at line 46 of file utils.h.

create_file

#define create_file	(		name,
			size
	)		create_file_data(name, name, size)

Definition at line 52 of file utils.h.

create_pf

#define create_pf	(		file,
			is_file
	)		create_pf_data(file, file, is_file)

Definition at line 54 of file utils.h.

Typedef Documentation

msi_table

typedef struct _msi_table msi_table

Function Documentation

create_cab_file()

void create_cab_file	(	const char *	name,
		DWORD	max_size,
		const char *	files
	)

Definition at line 2373 of file install.c.

{
    CCAB cabParams;
    LPCSTR ptr;
    HFCI hfci;
    ERF erf;
    BOOL res;
 
    set_cab_parameters(&cabParams, name, max_size);
 
    hfci = FCICreate(&erf, file_placed, mem_alloc, mem_free, fci_open,
                      fci_read, fci_write, fci_close, fci_seek, fci_delete,
                      get_temp_file, &cabParams, NULL);
 
    ok(hfci != NULL, "Failed to create an FCI context\n");
 
    ptr = files;
    while (*ptr)
    {
        res = add_file(hfci, ptr, tcompTYPE_MSZIP);
        ok(res, "Failed to add file: %s\n", ptr);
        ptr += lstrlenA(ptr) + 1;
    }
 
    res = FCIFlushCabinet(hfci, FALSE, get_next_cabinet, progress);
    ok(res, "Failed to flush the cabinet\n");
 
    res = FCIDestroy(hfci);
    ok(res, "Failed to destroy the cabinet\n");
}

create_database_wordcount()

void create_database_wordcount	(	const char *	name,
		const msi_table *	tables,
		int	num_tables,
		int	version,
		int	wordcount,
		const char *	template,
		const char *	packagecode
	)

create_file_data()

void create_file_data	(	const char *	name,
		const char *	data,
		DWORD	size
	)

create_pf_data()

void create_pf_data	(	const char *	file,
		const char *	data,
		BOOL	is_file
	)

delete_cab_files()

void delete_cab_files

(

void

)

Definition at line 3097 of file install.c.

{
    SHFILEOPSTRUCTA shfl;
    CHAR path[MAX_PATH+10];
 
    lstrcpyA(path, CURR_DIR);
    lstrcatA(path, "\\*.cab");
    path[strlen(path) + 1] = '\0';
 
    shfl.hwnd = NULL;
    shfl.wFunc = FO_DELETE;
    shfl.pFrom = path;
    shfl.pTo = NULL;
    shfl.fFlags = FOF_FILESONLY | FOF_NOCONFIRMATION | FOF_NORECURSION | FOF_SILENT;
 
    SHFileOperationA(&shfl);
}

Referenced by test_cabisextracted(), test_caborder(), test_continuouscabs(), test_file_in_use_cab(), test_lastusedsource(), test_readonlyfile_cab(), test_samesequence(), and test_uiLevelFlags().

delete_pf()

BOOL delete_pf	(	const char *	rel_path,
		BOOL	is_file
	)

Definition at line 2499 of file install.c.

{
    CHAR path[MAX_PATH];
 
    lstrcpyA(path, PROG_FILES_DIR);
    lstrcatA(path, "\\");
    lstrcatA(path, rel_path);
 
    if (is_file)
        return DeleteFileA(path);
    else
        return RemoveDirectoryA(path);
}

Referenced by delete_pf_files(), test_admin(), test_adminimage(), test_adminprops(), test_cabisextracted(), test_caborder(), test_continuouscabs(), test_currentworkingdir(), test_customaction51(), test_feature_override(), test_file_in_use(), test_file_in_use_cab(), test_installstate(), test_missingcab(), test_missingcomponent(), test_mixedmedia(), test_package_validation(), test_preselected(), test_propcase(), test_readonlyfile(), test_readonlyfile_cab(), test_samesequence(), test_shortcut(), test_sourcedirprop(), test_sourcefolder(), test_sourcepath(), test_transformprop(), test_uiLevelFlags(), test_upgrade_code(), and test_wow64().

file_exists()

BOOL file_exists

(

const char *

file

)

Definition at line 3615 of file install.c.

{
    return GetFileAttributesA(file) != INVALID_FILE_ATTRIBUTES;
}

Referenced by pf_exists(), and test_icon_table().

get_system_dirs()

BOOL get_system_dirs

(

void

)

Definition at line 2423 of file install.c.

{
    HKEY hkey;
    DWORD type, size;
 
    if (RegOpenKeyExA(HKEY_LOCAL_MACHINE, "Software\\Microsoft\\Windows\\CurrentVersion",
        0, KEY_QUERY_VALUE | KEY_WOW64_64KEY, &hkey))
        return FALSE;
 
    size = MAX_PATH;
    if (RegQueryValueExA(hkey, "ProgramFilesDir (x86)", 0, &type, (LPBYTE)PROG_FILES_DIR, &size) &&
        RegQueryValueExA(hkey, "ProgramFilesDir", 0, &type, (LPBYTE)PROG_FILES_DIR, &size)) {
        RegCloseKey(hkey);
        return FALSE;
    }
 
    size = MAX_PATH;
    if (RegQueryValueExA(hkey, "CommonFilesDir (x86)", 0, &type, (LPBYTE)COMMON_FILES_DIR, &size) &&
        RegQueryValueExA(hkey, "CommonFilesDir", 0, &type, (LPBYTE)COMMON_FILES_DIR, &size)) {
        RegCloseKey(hkey);
        return FALSE;
    }
 
    size = MAX_PATH;
    if (RegQueryValueExA(hkey, "ProgramFilesDir", 0, &type, (LPBYTE)PROG_FILES_DIR_NATIVE, &size)) {
        RegCloseKey(hkey);
        return FALSE;
    }
 
    RegCloseKey(hkey);
 
    if(!GetWindowsDirectoryA(WINDOWS_DIR, MAX_PATH))
        return FALSE;
 
    return TRUE;
}

Referenced by START_TEST().

get_user_dirs()

BOOL get_user_dirs

(

void

)

Definition at line 2404 of file install.c.

{
    HKEY hkey;
    DWORD type, size;
 
    if (RegOpenKeyA(HKEY_CURRENT_USER,
                    "Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\Shell Folders", &hkey))
        return FALSE;
 
    size = MAX_PATH;
    if(RegQueryValueExA(hkey, "AppData", 0, &type, (LPBYTE)APP_DATA_DIR, &size)){
        RegCloseKey(hkey);
        return FALSE;
    }
 
    RegCloseKey(hkey);
    return TRUE;
}

Referenced by START_TEST().

is_process_limited()

BOOL is_process_limited

(

void

)

Definition at line 2254 of file install.c.

{
    SID_IDENTIFIER_AUTHORITY NtAuthority = {SECURITY_NT_AUTHORITY};
    PSID Group = NULL;
    BOOL IsInGroup;
    HANDLE token;
 
    if (!AllocateAndInitializeSid(&NtAuthority, 2, SECURITY_BUILTIN_DOMAIN_RID,
                                  DOMAIN_ALIAS_RID_ADMINS, 0, 0, 0, 0, 0, 0, &Group) ||
        !CheckTokenMembership(NULL, Group, &IsInGroup))
    {
        trace("Could not check if the current user is an administrator\n");
        FreeSid(Group);
        return FALSE;
    }
    FreeSid(Group);
 
    if (!IsInGroup)
    {
        /* Only administrators have enough privileges for these tests */
        return TRUE;
    }
 
    if (OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY, &token))
    {
        BOOL ret;
        TOKEN_ELEVATION_TYPE type = TokenElevationTypeDefault;
        DWORD size;
 
        ret = GetTokenInformation(token, TokenElevationType, &type, sizeof(type), &size);
        CloseHandle(token);
        return (ret && type == TokenElevationTypeLimited);
    }
    return FALSE;
}

Referenced by test_adminimage(), test_adminprops(), test_allusers_prop(), test_cabisextracted(), test_continuouscabs(), test_currentworkingdir(), test_customaction51(), test_feature_override(), test_feature_tree(), test_file_in_use(), test_file_in_use_cab(), test_icon_table(), test_installed_prop(), test_installstate(), test_missingcab(), test_missingcomponent(), test_mixed_package(), test_mixedmedia(), test_MsiInstallProduct(), test_package_validation(), test_preselected(), test_propcase(), test_readonlyfile(), test_readonlyfile_cab(), test_remove_upgrade_code(), test_setdirproperty(), test_shared_component(), test_shortcut(), test_source_resolution(), test_sourcedirprop(), test_sourcefolder(), test_transformprop(), test_upgrade_code(), test_volume_props(), and test_wow64().

pf_exists()

BOOL pf_exists

(

const char *

file

)

Definition at line 3620 of file install.c.

{
    CHAR path[MAX_PATH];
 
    lstrcpyA(path, PROG_FILES_DIR);
    lstrcatA(path, "\\");
    lstrcatA(path, file);
 
    return file_exists(path);
}

Referenced by test_feature_override(), test_missingcomponent(), test_shared_component(), and test_upgrade_code().

run_query()

UINT run_query	(	MSIHANDLE	hdb,
		MSIHANDLE	hrec,
		const char *	query
	)

Definition at line 3653 of file install.c.

{
    MSIHANDLE hview = 0;
    UINT r;
 
    r = MsiDatabaseOpenViewA(hdb, query, &hview);
    if(r != ERROR_SUCCESS)
        return r;
 
    r = MsiViewExecute(hview, hrec);
    if(r == ERROR_SUCCESS)
        r = MsiViewClose(hview);
    MsiCloseHandle(hview);
    return r;
}

Referenced by add_custom_dll(), generate_transform(), test_customaction1(), and test_icon_table().

Variable Documentation

APP_DATA_DIR

char APP_DATA_DIR[MAX_PATH]

extern

Definition at line 69 of file install.c.

Referenced by get_user_dirs(), and test_icon_table().

COMMON_FILES_DIR

char COMMON_FILES_DIR[MAX_PATH]

extern

Definition at line 68 of file install.c.

Referenced by delete_cf(), and get_system_dirs().

CURR_DIR

char CURR_DIR[MAX_PATH]

extern

Definition at line 65 of file install.c.

PROG_FILES_DIR

char PROG_FILES_DIR[MAX_PATH]

extern

Definition at line 66 of file install.c.

Referenced by compare_pf_data(), create_pf_data(), delete_pf(), delete_pfmsitest_files(), get_pf_file_size(), get_system_dirs(), pf_exists(), test_file_in_use(), test_file_in_use_cab(), test_readonlyfile(), and test_readonlyfile_cab().

PROG_FILES_DIR_NATIVE

char PROG_FILES_DIR_NATIVE[MAX_PATH]

extern

Definition at line 67 of file install.c.

Referenced by delete_pf_native(), and get_system_dirs().

WINDOWS_DIR

char WINDOWS_DIR[MAX_PATH]

extern

Definition at line 70 of file install.c.

Referenced by get_system_dirs(), and test_icon_table().