/* * Setupapi file queue routines * * Copyright 2002 Alexandre Julliard for CodeWeavers * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * This library is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this library; if not, write to the Free Software * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA */ #include #include "windef.h" #include "winbase.h" #include "winreg.h" #include "winternl.h" #include "winerror.h" #include "wingdi.h" #include "winuser.h" #include "winnls.h" #include "setupapi.h" #include "wine/unicode.h" #include "setupapi_private.h" #include "winver.h" #include "wine/debug.h" WINE_DEFAULT_DEBUG_CHANNEL(setupapi); /* context structure for the default queue callback */ struct default_callback_context { HWND owner; HWND progress; UINT message; }; struct file_op { struct file_op *next; UINT style; WCHAR *src_root; WCHAR *src_path; WCHAR *src_file; WCHAR *src_descr; WCHAR *src_tag; WCHAR *dst_path; WCHAR *dst_file; }; struct file_op_queue { struct file_op *head; struct file_op *tail; unsigned int count; }; struct file_queue { struct file_op_queue copy_queue; struct file_op_queue delete_queue; struct file_op_queue rename_queue; DWORD flags; }; inline static WCHAR *strdupW( const WCHAR *str ) { WCHAR *ret = NULL; if (str) { int len = (strlenW(str) + 1) * sizeof(WCHAR); if ((ret = HeapAlloc( GetProcessHeap(), 0, len ))) memcpy( ret, str, len ); } return ret; } inline static WCHAR *strdupAtoW( const char *str ) { WCHAR *ret = NULL; if (str) { DWORD len = MultiByteToWideChar( CP_ACP, 0, str, -1, NULL, 0 ); if ((ret = HeapAlloc( GetProcessHeap(), 0, len * sizeof(WCHAR) ))) MultiByteToWideChar( CP_ACP, 0, str, -1, ret, len ); } return ret; } inline static char *strdupWtoA( const WCHAR *str ) { char *ret = NULL; if (str) { DWORD len = WideCharToMultiByte( CP_ACP, 0, str, -1, NULL, 0, NULL, NULL ); if ((ret = HeapAlloc( GetProcessHeap(), 0, len ))) WideCharToMultiByte( CP_ACP, 0, str, -1, ret, len, NULL, NULL ); } return ret; } /* append a file operation to a queue */ inline static void queue_file_op( struct file_op_queue *queue, struct file_op *op ) { op->next = NULL; if (queue->tail) queue->tail->next = op; else queue->head = op; queue->tail = op; queue->count++; } /* free all the file operations on a given queue */ static void free_file_op_queue( struct file_op_queue *queue ) { struct file_op *t, *op = queue->head; while( op ) { HeapFree( GetProcessHeap(), 0, op->src_root ); HeapFree( GetProcessHeap(), 0, op->src_path ); HeapFree( GetProcessHeap(), 0, op->src_file ); HeapFree( GetProcessHeap(), 0, op->src_descr ); HeapFree( GetProcessHeap(), 0, op->src_tag ); HeapFree( GetProcessHeap(), 0, op->dst_path ); if (op->dst_file != op->src_file) HeapFree( GetProcessHeap(), 0, op->dst_file ); t = op; op = op->next; HeapFree( GetProcessHeap(), 0, t ); } } /* concat 3 strings to make a path, handling separators correctly */ static void concat_W( WCHAR *buffer, const WCHAR *src1, const WCHAR *src2, const WCHAR *src3 ) { *buffer = 0; if (src1 && *src1) { strcpyW( buffer, src1 ); buffer += strlenW(buffer ); if (buffer[-1] != '\\') *buffer++ = '\\'; if (src2) while (*src2 == '\\') src2++; } if (src2) { strcpyW( buffer, src2 ); buffer += strlenW(buffer ); if (buffer[-1] != '\\') *buffer++ = '\\'; if (src3) while (*src3 == '\\') src3++; } if (src3) { strcpyW( buffer, src3 ); buffer += strlenW(buffer ); } } /*********************************************************************** * build_filepathsW * * Build a FILEPATHS_W structure for a given file operation. */ static BOOL build_filepathsW( const struct file_op *op, FILEPATHS_W *paths ) { int src_len = 1, dst_len = 1; WCHAR *source = (PWSTR)paths->Source, *target = (PWSTR)paths->Target; if (op->src_root) src_len += strlenW(op->src_root) + 1; if (op->src_path) src_len += strlenW(op->src_path) + 1; if (op->src_file) src_len += strlenW(op->src_file) + 1; if (op->dst_path) dst_len += strlenW(op->dst_path) + 1; if (op->dst_file) dst_len += strlenW(op->dst_file) + 1; src_len *= sizeof(WCHAR); dst_len *= sizeof(WCHAR); if (!source || HeapSize( GetProcessHeap(), 0, source ) < src_len ) { HeapFree( GetProcessHeap(), 0, source ); paths->Source = source = HeapAlloc( GetProcessHeap(), 0, src_len ); } if (!target || HeapSize( GetProcessHeap(), 0, target ) < dst_len ) { HeapFree( GetProcessHeap(), 0, target ); paths->Target = target = HeapAlloc( GetProcessHeap(), 0, dst_len ); } if (!source || !target) return FALSE; concat_W( source, op->src_root, op->src_path, op->src_file ); concat_W( target, NULL, op->dst_path, op->dst_file ); paths->Win32Error = 0; paths->Flags = 0; return TRUE; } /*********************************************************************** * QUEUE_callback_WtoA * * Map a file callback parameters from W to A and call the A callback. */ UINT CALLBACK QUEUE_callback_WtoA( void *context, UINT notification, UINT_PTR param1, UINT_PTR param2 ) { struct callback_WtoA_context *callback_ctx = context; char buffer[MAX_PATH]; UINT ret; UINT_PTR old_param2 = param2; switch(notification) { case SPFILENOTIFY_COPYERROR: param2 = (UINT_PTR)&buffer; /* fall through */ case SPFILENOTIFY_STARTDELETE: case SPFILENOTIFY_ENDDELETE: case SPFILENOTIFY_DELETEERROR: case SPFILENOTIFY_STARTRENAME: case SPFILENOTIFY_ENDRENAME: case SPFILENOTIFY_RENAMEERROR: case SPFILENOTIFY_STARTCOPY: case SPFILENOTIFY_ENDCOPY: { FILEPATHS_W *pathsW = (FILEPATHS_W *)param1; FILEPATHS_A pathsA; pathsA.Source = strdupWtoA( pathsW->Source ); pathsA.Target = strdupWtoA( pathsW->Target ); pathsA.Win32Error = pathsW->Win32Error; pathsA.Flags = pathsW->Flags; ret = callback_ctx->orig_handler( callback_ctx->orig_context, notification, (UINT_PTR)&pathsA, param2 ); HeapFree( GetProcessHeap(), 0, (void *)pathsA.Source ); HeapFree( GetProcessHeap(), 0, (void *)pathsA.Target ); } if (notification == SPFILENOTIFY_COPYERROR) MultiByteToWideChar( CP_ACP, 0, buffer, -1, (WCHAR *)old_param2, MAX_PATH ); break; case SPFILENOTIFY_NEEDMEDIA: case SPFILENOTIFY_QUEUESCAN: FIXME("mapping for %d not implemented\n",notification); case SPFILENOTIFY_STARTQUEUE: case SPFILENOTIFY_ENDQUEUE: case SPFILENOTIFY_STARTSUBQUEUE: case SPFILENOTIFY_ENDSUBQUEUE: default: ret = callback_ctx->orig_handler( callback_ctx->orig_context, notification, param1, param2 ); break; } return ret; } /*********************************************************************** * get_src_file_info * * Retrieve the source file information for a given file. */ static void get_src_file_info( HINF hinf, struct file_op *op ) { static const WCHAR SourceDisksNames[] = {'S','o','u','r','c','e','D','i','s','k','s','N','a','m','e','s',0}; static const WCHAR SourceDisksFiles[] = {'S','o','u','r','c','e','D','i','s','k','s','F','i','l','e','s',0}; INFCONTEXT file_ctx, disk_ctx; INT id, diskid; DWORD len, len2; /* find the SourceDisksFiles entry */ if (!SetupFindFirstLineW( hinf, SourceDisksFiles, op->src_file, &file_ctx )) { const WCHAR *dir; if ((op->style & (SP_COPY_SOURCE_ABSOLUTE|SP_COPY_SOURCEPATH_ABSOLUTE))) return; /* no specific info, use .inf file source directory */ if (!op->src_root && (dir = DIRID_get_string( hinf, DIRID_SRCPATH ))) op->src_root = strdupW( dir ); return; } if (!SetupGetIntField( &file_ctx, 1, &diskid )) return; /* now find the diskid in the SourceDisksNames section */ if (!SetupFindFirstLineW( hinf, SourceDisksNames, NULL, &disk_ctx )) return; for (;;) { if (SetupGetIntField( &disk_ctx, 0, &id ) && (id == diskid)) break; if (!SetupFindNextLine( &disk_ctx, &disk_ctx )) return; } /* and fill in the missing info */ if (!op->src_descr) { if (SetupGetStringFieldW( &disk_ctx, 1, NULL, 0, &len ) && (op->src_descr = HeapAlloc( GetProcessHeap(), 0, len*sizeof(WCHAR) ))) SetupGetStringFieldW( &disk_ctx, 1, op->src_descr, len, NULL ); } if (!op->src_tag) { if (SetupGetStringFieldW( &disk_ctx, 2, NULL, 0, &len ) && (op->src_tag = HeapAlloc( GetProcessHeap(), 0, len*sizeof(WCHAR) ))) SetupGetStringFieldW( &disk_ctx, 2, op->src_tag, len, NULL ); } if (!op->src_path && !(op->style & SP_COPY_SOURCE_ABSOLUTE)) { if (!(op->style & SP_COPY_SOURCEPATH_ABSOLUTE)) { /* retrieve relative path for this disk */ if (!SetupGetStringFieldW( &disk_ctx, 4, NULL, 0, &len )) len = 0; } /* retrieve relative path for this file */ if (!SetupGetStringFieldW( &file_ctx, 2, NULL, 0, &len2 )) len2 = 0; if ((len || len2) && (op->src_path = HeapAlloc( GetProcessHeap(), 0, (len+len2)*sizeof(WCHAR) ))) { WCHAR *ptr = op->src_path; if (len) { SetupGetStringFieldW( &disk_ctx, 4, op->src_path, len, NULL ); ptr = op->src_path + strlenW(op->src_path); if (len2 && ptr > op->src_path && ptr[-1] != '\\') *ptr++ = '\\'; } if (!SetupGetStringFieldW( &disk_ctx, 4, ptr, len2, NULL )) *ptr = 0; } } if (!op->src_root) op->src_root = strdupW( PARSER_get_src_root(hinf) ); } /*********************************************************************** * get_destination_dir * * Retrieve the destination dir for a given section. */ static WCHAR *get_destination_dir( HINF hinf, const WCHAR *section ) { static const WCHAR Dest[] = {'D','e','s','t','i','n','a','t','i','o','n','D','i','r','s',0}; static const WCHAR Def[] = {'D','e','f','a','u','l','t','D','e','s','t','D','i','r',0}; const WCHAR *dir; WCHAR *ptr, *ret; INFCONTEXT context; INT dirid; DWORD len1, len2; if (!SetupFindFirstLineW( hinf, Dest, section, &context ) && !SetupFindFirstLineW( hinf, Dest, Def, &context )) return NULL; if (!SetupGetIntField( &context, 1, &dirid )) return NULL; if (!(dir = DIRID_get_string( hinf, dirid ))) return NULL; len1 = strlenW(dir) + 1; if (!SetupGetStringFieldW( &context, 2, NULL, 0, &len2 )) len2 = 0; if (!(ret = HeapAlloc( GetProcessHeap(), 0, (len1+len2) * sizeof(WCHAR) ))) return NULL; strcpyW( ret, dir ); ptr = ret + strlenW(ret); if (len2 && ptr > ret && ptr[-1] != '\\') *ptr++ = '\\'; if (!SetupGetStringFieldW( &context, 2, ptr, len2, NULL )) *ptr = 0; return ret; } static void (WINAPI *pExtractFiles)( LPSTR, LPSTR, DWORD, DWORD, DWORD, DWORD ); /*********************************************************************** * extract_cabinet_file * * Extract a file from a .cab file. */ static BOOL extract_cabinet_file( const WCHAR *cabinet, const WCHAR *root, const WCHAR *src, const WCHAR *dst ) { static const WCHAR extW[] = {'.','c','a','b',0}; static HMODULE advpack; char *cab_path, *cab_file; int len = strlenW( cabinet ); /* make sure the cabinet file has a .cab extension */ if (len <= 4 || strcmpiW( cabinet + len - 4, extW )) return FALSE; if (!pExtractFiles) { if (!advpack && !(advpack = LoadLibraryA( "advpack.dll" ))) { ERR( "could not load advpack.dll\n" ); return FALSE; } if (!(pExtractFiles = (void *)GetProcAddress( advpack, "ExtractFiles" ))) { ERR( "could not find ExtractFiles in advpack.dll\n" ); return FALSE; } } if (!(cab_path = strdupWtoA( root ))) return FALSE; len = WideCharToMultiByte( CP_ACP, 0, cabinet, -1, NULL, 0, NULL, NULL ); if (!(cab_file = HeapAlloc( GetProcessHeap(), 0, strlen(cab_path) + len + 1 ))) { HeapFree( GetProcessHeap(), 0, cab_path ); return FALSE; } strcpy( cab_file, cab_path ); if (cab_file[0] && cab_file[strlen(cab_file)-1] != '\\') strcat( cab_file, "\\" ); WideCharToMultiByte( CP_ACP, 0, cabinet, -1, cab_file + strlen(cab_file), len, NULL, NULL ); FIXME( "awful hack: extracting cabinet %s\n", debugstr_a(cab_file) ); pExtractFiles( cab_file, cab_path, 0, 0, 0, 0 ); HeapFree( GetProcessHeap(), 0, cab_file ); HeapFree( GetProcessHeap(), 0, cab_path ); return CopyFileW( src, dst, FALSE /*FIXME*/ ); } /*********************************************************************** * SetupOpenFileQueue (SETUPAPI.@) */ HSPFILEQ WINAPI SetupOpenFileQueue(void) { struct file_queue *queue; if (!(queue = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*queue)))) return (HSPFILEQ)INVALID_HANDLE_VALUE; return queue; } /*********************************************************************** * SetupCloseFileQueue (SETUPAPI.@) */ BOOL WINAPI SetupCloseFileQueue( HSPFILEQ handle ) { struct file_queue *queue = handle; free_file_op_queue( &queue->copy_queue ); free_file_op_queue( &queue->rename_queue ); free_file_op_queue( &queue->delete_queue ); HeapFree( GetProcessHeap(), 0, queue ); return TRUE; } /*********************************************************************** * SetupQueueCopyIndirectA (SETUPAPI.@) */ BOOL WINAPI SetupQueueCopyIndirectA( PSP_FILE_COPY_PARAMS_A params ) { struct file_queue *queue = params->QueueHandle; struct file_op *op; if (!(op = HeapAlloc( GetProcessHeap(), 0, sizeof(*op) ))) return FALSE; op->style = params->CopyStyle; op->src_root = strdupAtoW( params->SourceRootPath ); op->src_path = strdupAtoW( params->SourcePath ); op->src_file = strdupAtoW( params->SourceFilename ); op->src_descr = strdupAtoW( params->SourceDescription ); op->src_tag = strdupAtoW( params->SourceTagfile ); op->dst_path = strdupAtoW( params->TargetDirectory ); op->dst_file = strdupAtoW( params->TargetFilename ); /* some defaults */ if (!op->src_file) op->src_file = op->dst_file; if (params->LayoutInf) { get_src_file_info( params->LayoutInf, op ); if (!op->dst_path) op->dst_path = get_destination_dir( params->LayoutInf, op->dst_file ); } TRACE( "root=%s path=%s file=%s -> dir=%s file=%s descr=%s tag=%s\n", debugstr_w(op->src_root), debugstr_w(op->src_path), debugstr_w(op->src_file), debugstr_w(op->dst_path), debugstr_w(op->dst_file), debugstr_w(op->src_descr), debugstr_w(op->src_tag) ); queue_file_op( &queue->copy_queue, op ); return TRUE; } /*********************************************************************** * SetupQueueCopyIndirectW (SETUPAPI.@) */ BOOL WINAPI SetupQueueCopyIndirectW( PSP_FILE_COPY_PARAMS_W params ) { struct file_queue *queue = params->QueueHandle; struct file_op *op; if (!(op = HeapAlloc( GetProcessHeap(), 0, sizeof(*op) ))) return FALSE; op->style = params->CopyStyle; op->src_root = strdupW( params->SourceRootPath ); op->src_path = strdupW( params->SourcePath ); op->src_file = strdupW( params->SourceFilename ); op->src_descr = strdupW( params->SourceDescription ); op->src_tag = strdupW( params->SourceTagfile ); op->dst_path = strdupW( params->TargetDirectory ); op->dst_file = strdupW( params->TargetFilename ); /* some defaults */ if (!op->src_file) op->src_file = op->dst_file; if (params->LayoutInf) { get_src_file_info( params->LayoutInf, op ); if (!op->dst_path) op->dst_path = get_destination_dir( params->LayoutInf, op->dst_file ); } TRACE( "root=%s path=%s file=%s -> dir=%s file=%s descr=%s tag=%s\n", debugstr_w(op->src_root), debugstr_w(op->src_path), debugstr_w(op->src_file), debugstr_w(op->dst_path), debugstr_w(op->dst_file), debugstr_w(op->src_descr), debugstr_w(op->src_tag) ); queue_file_op( &queue->copy_queue, op ); return TRUE; } /*********************************************************************** * SetupQueueCopyA (SETUPAPI.@) */ BOOL WINAPI SetupQueueCopyA( HSPFILEQ queue, PCSTR src_root, PCSTR src_path, PCSTR src_file, PCSTR src_descr, PCSTR src_tag, PCSTR dst_dir, PCSTR dst_file, DWORD style ) { SP_FILE_COPY_PARAMS_A params; params.cbSize = sizeof(params); params.QueueHandle = queue; params.SourceRootPath = src_root; params.SourcePath = src_path; params.SourceFilename = src_file; params.SourceDescription = src_descr; params.SourceTagfile = src_tag; params.TargetDirectory = dst_dir; params.TargetFilename = dst_file; params.CopyStyle = style; params.LayoutInf = 0; params.SecurityDescriptor = NULL; return SetupQueueCopyIndirectA( ¶ms ); } /*********************************************************************** * SetupQueueCopyW (SETUPAPI.@) */ BOOL WINAPI SetupQueueCopyW( HSPFILEQ queue, PCWSTR src_root, PCWSTR src_path, PCWSTR src_file, PCWSTR src_descr, PCWSTR src_tag, PCWSTR dst_dir, PCWSTR dst_file, DWORD style ) { SP_FILE_COPY_PARAMS_W params; params.cbSize = sizeof(params); params.QueueHandle = queue; params.SourceRootPath = src_root; params.SourcePath = src_path; params.SourceFilename = src_file; params.SourceDescription = src_descr; params.SourceTagfile = src_tag; params.TargetDirectory = dst_dir; params.TargetFilename = dst_file; params.CopyStyle = style; params.LayoutInf = 0; params.SecurityDescriptor = NULL; return SetupQueueCopyIndirectW( ¶ms ); } /*********************************************************************** * SetupQueueDefaultCopyA (SETUPAPI.@) */ BOOL WINAPI SetupQueueDefaultCopyA( HSPFILEQ queue, HINF hinf, PCSTR src_root, PCSTR src_file, PCSTR dst_file, DWORD style ) { SP_FILE_COPY_PARAMS_A params; params.cbSize = sizeof(params); params.QueueHandle = queue; params.SourceRootPath = src_root; params.SourcePath = NULL; params.SourceFilename = src_file; params.SourceDescription = NULL; params.SourceTagfile = NULL; params.TargetDirectory = NULL; params.TargetFilename = dst_file; params.CopyStyle = style; params.LayoutInf = hinf; params.SecurityDescriptor = NULL; return SetupQueueCopyIndirectA( ¶ms ); } /*********************************************************************** * SetupQueueDefaultCopyW (SETUPAPI.@) */ BOOL WINAPI SetupQueueDefaultCopyW( HSPFILEQ queue, HINF hinf, PCWSTR src_root, PCWSTR src_file, PCWSTR dst_file, DWORD style ) { SP_FILE_COPY_PARAMS_W params; params.cbSize = sizeof(params); params.QueueHandle = queue; params.SourceRootPath = src_root; params.SourcePath = NULL; params.SourceFilename = src_file; params.SourceDescription = NULL; params.SourceTagfile = NULL; params.TargetDirectory = NULL; params.TargetFilename = dst_file; params.CopyStyle = style; params.LayoutInf = hinf; params.SecurityDescriptor = NULL; return SetupQueueCopyIndirectW( ¶ms ); } /*********************************************************************** * SetupQueueDeleteA (SETUPAPI.@) */ BOOL WINAPI SetupQueueDeleteA( HSPFILEQ handle, PCSTR part1, PCSTR part2 ) { struct file_queue *queue = handle; struct file_op *op; if (!(op = HeapAlloc( GetProcessHeap(), 0, sizeof(*op) ))) return FALSE; op->style = 0; op->src_root = NULL; op->src_path = NULL; op->src_file = NULL; op->src_descr = NULL; op->src_tag = NULL; op->dst_path = strdupAtoW( part1 ); op->dst_file = strdupAtoW( part2 ); queue_file_op( &queue->delete_queue, op ); return TRUE; } /*********************************************************************** * SetupQueueDeleteW (SETUPAPI.@) */ BOOL WINAPI SetupQueueDeleteW( HSPFILEQ handle, PCWSTR part1, PCWSTR part2 ) { struct file_queue *queue = handle; struct file_op *op; if (!(op = HeapAlloc( GetProcessHeap(), 0, sizeof(*op) ))) return FALSE; op->style = 0; op->src_root = NULL; op->src_path = NULL; op->src_file = NULL; op->src_descr = NULL; op->src_tag = NULL; op->dst_path = strdupW( part1 ); op->dst_file = strdupW( part2 ); queue_file_op( &queue->delete_queue, op ); return TRUE; } /*********************************************************************** * SetupQueueRenameA (SETUPAPI.@) */ BOOL WINAPI SetupQueueRenameA( HSPFILEQ handle, PCSTR SourcePath, PCSTR SourceFilename, PCSTR TargetPath, PCSTR TargetFilename ) { struct file_queue *queue = handle; struct file_op *op; if (!(op = HeapAlloc( GetProcessHeap(), 0, sizeof(*op) ))) return FALSE; op->style = 0; op->src_root = NULL; op->src_path = strdupAtoW( SourcePath ); op->src_file = strdupAtoW( SourceFilename ); op->src_descr = NULL; op->src_tag = NULL; op->dst_path = strdupAtoW( TargetPath ); op->dst_file = strdupAtoW( TargetFilename ); queue_file_op( &queue->rename_queue, op ); return TRUE; } /*********************************************************************** * SetupQueueRenameW (SETUPAPI.@) */ BOOL WINAPI SetupQueueRenameW( HSPFILEQ handle, PCWSTR SourcePath, PCWSTR SourceFilename, PCWSTR TargetPath, PCWSTR TargetFilename ) { struct file_queue *queue = handle; struct file_op *op; if (!(op = HeapAlloc( GetProcessHeap(), 0, sizeof(*op) ))) return FALSE; op->style = 0; op->src_root = NULL; op->src_path = strdupW( SourcePath ); op->src_file = strdupW( SourceFilename ); op->src_descr = NULL; op->src_tag = NULL; op->dst_path = strdupW( TargetPath ); op->dst_file = strdupW( TargetFilename ); queue_file_op( &queue->rename_queue, op ); return TRUE; } /*********************************************************************** * SetupQueueCopySectionA (SETUPAPI.@) */ BOOL WINAPI SetupQueueCopySectionA( HSPFILEQ queue, PCSTR src_root, HINF hinf, HINF hlist, PCSTR section, DWORD style ) { UNICODE_STRING sectionW; BOOL ret = FALSE; if (!RtlCreateUnicodeStringFromAsciiz( §ionW, section )) { SetLastError( ERROR_NOT_ENOUGH_MEMORY ); return FALSE; } if (!src_root) ret = SetupQueueCopySectionW( queue, NULL, hinf, hlist, sectionW.Buffer, style ); else { UNICODE_STRING srcW; if (RtlCreateUnicodeStringFromAsciiz( &srcW, src_root )) { ret = SetupQueueCopySectionW( queue, srcW.Buffer, hinf, hlist, sectionW.Buffer, style ); RtlFreeUnicodeString( &srcW ); } else SetLastError( ERROR_NOT_ENOUGH_MEMORY ); } RtlFreeUnicodeString( §ionW ); return ret; } /*********************************************************************** * SetupQueueCopySectionW (SETUPAPI.@) */ BOOL WINAPI SetupQueueCopySectionW( HSPFILEQ queue, PCWSTR src_root, HINF hinf, HINF hlist, PCWSTR section, DWORD style ) { SP_FILE_COPY_PARAMS_W params; INFCONTEXT context; WCHAR dest[MAX_PATH], src[MAX_PATH]; INT flags; TRACE( "hinf=%p/%p section=%s root=%s\n", hinf, hlist, debugstr_w(section), debugstr_w(src_root) ); params.cbSize = sizeof(params); params.QueueHandle = queue; params.SourceRootPath = src_root; params.SourcePath = NULL; params.SourceDescription = NULL; params.SourceTagfile = NULL; params.TargetFilename = dest; params.CopyStyle = style; params.LayoutInf = hinf; params.SecurityDescriptor = NULL; if (!hlist) hlist = hinf; if (!SetupFindFirstLineW( hlist, section, NULL, &context )) return FALSE; if (!(params.TargetDirectory = get_destination_dir( hinf, section ))) return FALSE; do { if (!SetupGetStringFieldW( &context, 1, dest, sizeof(dest)/sizeof(WCHAR), NULL )) return FALSE; if (!SetupGetStringFieldW( &context, 2, src, sizeof(src)/sizeof(WCHAR), NULL )) *src = 0; if (!SetupGetIntField( &context, 4, &flags )) flags = 0; /* FIXME */ params.SourceFilename = *src ? src : NULL; if (!SetupQueueCopyIndirectW( ¶ms )) return FALSE; } while (SetupFindNextLine( &context, &context )); return TRUE; } /*********************************************************************** * SetupQueueDeleteSectionA (SETUPAPI.@) */ BOOL WINAPI SetupQueueDeleteSectionA( HSPFILEQ queue, HINF hinf, HINF hlist, PCSTR section ) { UNICODE_STRING sectionW; BOOL ret = FALSE; if (RtlCreateUnicodeStringFromAsciiz( §ionW, section )) { ret = SetupQueueDeleteSectionW( queue, hinf, hlist, sectionW.Buffer ); RtlFreeUnicodeString( §ionW ); } else SetLastError( ERROR_NOT_ENOUGH_MEMORY ); return ret; } /*********************************************************************** * SetupQueueDeleteSectionW (SETUPAPI.@) */ BOOL WINAPI SetupQueueDeleteSectionW( HSPFILEQ queue, HINF hinf, HINF hlist, PCWSTR section ) { INFCONTEXT context; WCHAR *dest_dir; WCHAR buffer[MAX_PATH]; BOOL ret = FALSE; INT flags; TRACE( "hinf=%p/%p section=%s\n", hinf, hlist, debugstr_w(section) ); if (!hlist) hlist = hinf; if (!SetupFindFirstLineW( hlist, section, NULL, &context )) return FALSE; if (!(dest_dir = get_destination_dir( hinf, section ))) return FALSE; do { if (!SetupGetStringFieldW( &context, 1, buffer, sizeof(buffer)/sizeof(WCHAR), NULL )) goto done; if (!SetupGetIntField( &context, 4, &flags )) flags = 0; if (!SetupQueueDeleteW( queue, dest_dir, buffer )) goto done; } while (SetupFindNextLine( &context, &context )); ret = TRUE; done: HeapFree( GetProcessHeap(), 0, dest_dir ); return ret; } /*********************************************************************** * SetupQueueRenameSectionA (SETUPAPI.@) */ BOOL WINAPI SetupQueueRenameSectionA( HSPFILEQ queue, HINF hinf, HINF hlist, PCSTR section ) { UNICODE_STRING sectionW; BOOL ret = FALSE; if (RtlCreateUnicodeStringFromAsciiz( §ionW, section )) { ret = SetupQueueRenameSectionW( queue, hinf, hlist, sectionW.Buffer ); RtlFreeUnicodeString( §ionW ); } else SetLastError( ERROR_NOT_ENOUGH_MEMORY ); return ret; } /*********************************************************************** * SetupQueueRenameSectionW (SETUPAPI.@) */ BOOL WINAPI SetupQueueRenameSectionW( HSPFILEQ queue, HINF hinf, HINF hlist, PCWSTR section ) { INFCONTEXT context; WCHAR *dest_dir; WCHAR src[MAX_PATH], dst[MAX_PATH]; BOOL ret = FALSE; TRACE( "hinf=%p/%p section=%s\n", hinf, hlist, debugstr_w(section) ); if (!hlist) hlist = hinf; if (!SetupFindFirstLineW( hlist, section, NULL, &context )) return FALSE; if (!(dest_dir = get_destination_dir( hinf, section ))) return FALSE; do { if (!SetupGetStringFieldW( &context, 1, dst, sizeof(dst)/sizeof(WCHAR), NULL )) goto done; if (!SetupGetStringFieldW( &context, 2, src, sizeof(src)/sizeof(WCHAR), NULL )) goto done; if (!SetupQueueRenameW( queue, dest_dir, src, NULL, dst )) goto done; } while (SetupFindNextLine( &context, &context )); ret = TRUE; done: HeapFree( GetProcessHeap(), 0, dest_dir ); return ret; } /*********************************************************************** * SetupCommitFileQueueA (SETUPAPI.@) */ BOOL WINAPI SetupCommitFileQueueA( HWND owner, HSPFILEQ queue, PSP_FILE_CALLBACK_A handler, PVOID context ) { struct callback_WtoA_context ctx; ctx.orig_context = context; ctx.orig_handler = handler; return SetupCommitFileQueueW( owner, queue, QUEUE_callback_WtoA, &ctx ); } /*********************************************************************** * create_full_pathW * * Recursively create all directories in the path. */ static BOOL create_full_pathW(const WCHAR *path) { BOOL ret = TRUE; int len; WCHAR *new_path; new_path = HeapAlloc(GetProcessHeap(), 0, (strlenW(path) + 1) * sizeof(WCHAR)); strcpyW(new_path, path); while((len = strlenW(new_path)) && new_path[len - 1] == '\\') new_path[len - 1] = 0; while(!CreateDirectoryW(new_path, NULL)) { WCHAR *slash; DWORD last_error = GetLastError(); if(last_error == ERROR_ALREADY_EXISTS) break; if(last_error != ERROR_PATH_NOT_FOUND) { ret = FALSE; break; } if(!(slash = strrchrW(new_path, '\\'))) { ret = FALSE; break; } len = slash - new_path; new_path[len] = 0; if(!create_full_pathW(new_path)) { ret = FALSE; break; } new_path[len] = '\\'; } HeapFree(GetProcessHeap(), 0, new_path); return ret; } BOOL static do_file_copyW( LPCWSTR source, LPCWSTR target, DWORD style) { BOOL rc = FALSE; BOOL docopy = TRUE; TRACE("copy %s to %s style 0x%lx\n",debugstr_w(source),debugstr_w(target),style); /* before copy processing */ if (style & SP_COPY_REPLACEONLY) { if (GetFileAttributesW(target) == INVALID_FILE_ATTRIBUTES) docopy = FALSE; } if (style & (SP_COPY_NEWER_OR_SAME | SP_COPY_NEWER_ONLY | SP_COPY_FORCE_NEWER)) { DWORD VersionSizeSource=0; DWORD VersionSizeTarget=0; DWORD zero=0; /* * This is sort of an interesting workaround. You see, calling * GetVersionInfoSize on a builtin dll loads that dll into memory * and we do not properly unload builtin dlls.. so we effectively * lock into memory all the targets we are replacing. This leads * to problems when we try to register the replaced dlls. * * So I will test for the existence of the files first so that * we just basically unconditionally replace the builtin versions. */ if ((GetFileAttributesW(target) != INVALID_FILE_ATTRIBUTES) && (GetFileAttributesW(source) != INVALID_FILE_ATTRIBUTES)) { VersionSizeSource = GetFileVersionInfoSizeW(source,&zero); VersionSizeTarget = GetFileVersionInfoSizeW(target,&zero); } TRACE("SizeTarget %li ... SizeSource %li\n",VersionSizeTarget, VersionSizeSource); if (VersionSizeSource && VersionSizeTarget) { LPVOID VersionSource; LPVOID VersionTarget; VS_FIXEDFILEINFO *TargetInfo; VS_FIXEDFILEINFO *SourceInfo; INT length; WCHAR SubBlock[2]={'\\',0}; DWORD ret; VersionSource = HeapAlloc(GetProcessHeap(),0,VersionSizeSource); VersionTarget = HeapAlloc(GetProcessHeap(),0,VersionSizeTarget); ret = GetFileVersionInfoW(source,0,VersionSizeSource,VersionSource); if (ret) ret = GetFileVersionInfoW(target, 0, VersionSizeTarget, VersionTarget); if (ret) { ret = VerQueryValueW(VersionSource, SubBlock, (LPVOID*)&SourceInfo, &length); if (ret) ret = VerQueryValueW(VersionTarget, SubBlock, (LPVOID*)&TargetInfo, &length); if (ret) { TRACE("Versions: Source %li.%li target %li.%li\n", SourceInfo->dwFileVersionMS, SourceInfo->dwFileVersionLS, TargetInfo->dwFileVersionMS, TargetInfo->dwFileVersionLS); if (TargetInfo->dwFileVersionMS > SourceInfo->dwFileVersionMS) { FIXME("Notify that target version is greater..\n"); docopy = FALSE; } else if ((TargetInfo->dwFileVersionMS == SourceInfo->dwFileVersionMS) && (TargetInfo->dwFileVersionLS > SourceInfo->dwFileVersionLS)) { FIXME("Notify that target version is greater..\n"); docopy = FALSE; } else if ((style & SP_COPY_NEWER_ONLY) && (TargetInfo->dwFileVersionMS == SourceInfo->dwFileVersionMS) &&(TargetInfo->dwFileVersionLS == SourceInfo->dwFileVersionLS)) { FIXME("Notify that target version is greater..\n"); docopy = FALSE; } } HeapFree(GetProcessHeap(),0,VersionSource); HeapFree(GetProcessHeap(),0,VersionTarget); } } } if (style & (SP_COPY_NOOVERWRITE | SP_COPY_FORCE_NOOVERWRITE)) { if (GetFileAttributesW(target) != INVALID_FILE_ATTRIBUTES) { FIXME("Notify user target file exists\n"); docopy = FALSE; } } if (style & (SP_COPY_NODECOMP | SP_COPY_LANGUAGEAWARE | SP_COPY_FORCE_IN_USE | SP_COPY_IN_USE_NEEDS_REBOOT | SP_COPY_NOSKIP | SP_COPY_WARNIFSKIP)) { ERR("Unsupported style(s) 0x%lx\n",style); } if (docopy) { rc = CopyFileW(source,target,FALSE); TRACE("Did copy... rc was %i\n",rc); } /* after copy processing */ if (style & SP_COPY_DELETESOURCE) { if (rc) DeleteFileW(source); } return rc; } /*********************************************************************** * SetupCommitFileQueueW (SETUPAPI.@) */ BOOL WINAPI SetupCommitFileQueueW( HWND owner, HSPFILEQ handle, PSP_FILE_CALLBACK_W handler, PVOID context ) { struct file_queue *queue = handle; struct file_op *op; BOOL result = FALSE; FILEPATHS_W paths; UINT op_result; paths.Source = paths.Target = NULL; if (!queue->copy_queue.count && !queue->delete_queue.count && !queue->rename_queue.count) return TRUE; /* nothing to do */ if (!handler( context, SPFILENOTIFY_STARTQUEUE, (UINT)owner, 0 )) return FALSE; /* perform deletes */ if (queue->delete_queue.count) { if (!(handler( context, SPFILENOTIFY_STARTSUBQUEUE, FILEOP_DELETE, queue->delete_queue.count ))) goto done; for (op = queue->delete_queue.head; op; op = op->next) { build_filepathsW( op, &paths ); op_result = handler( context, SPFILENOTIFY_STARTDELETE, (UINT_PTR)&paths, FILEOP_DELETE); if (op_result == FILEOP_ABORT) goto done; while (op_result == FILEOP_DOIT) { TRACE( "deleting file %s\n", debugstr_w(paths.Target) ); if (DeleteFileW( paths.Target )) break; /* success */ paths.Win32Error = GetLastError(); op_result = handler( context, SPFILENOTIFY_DELETEERROR, (UINT_PTR)&paths, 0 ); if (op_result == FILEOP_ABORT) goto done; } handler( context, SPFILENOTIFY_ENDDELETE, (UINT_PTR)&paths, 0 ); } handler( context, SPFILENOTIFY_ENDSUBQUEUE, FILEOP_DELETE, 0 ); } /* perform renames */ if (queue->rename_queue.count) { if (!(handler( context, SPFILENOTIFY_STARTSUBQUEUE, FILEOP_RENAME, queue->rename_queue.count ))) goto done; for (op = queue->rename_queue.head; op; op = op->next) { build_filepathsW( op, &paths ); op_result = handler( context, SPFILENOTIFY_STARTRENAME, (UINT_PTR)&paths, FILEOP_RENAME); if (op_result == FILEOP_ABORT) goto done; while (op_result == FILEOP_DOIT) { TRACE( "renaming file %s -> %s\n", debugstr_w(paths.Source), debugstr_w(paths.Target) ); if (MoveFileW( paths.Source, paths.Target )) break; /* success */ paths.Win32Error = GetLastError(); op_result = handler( context, SPFILENOTIFY_RENAMEERROR, (UINT_PTR)&paths, 0 ); if (op_result == FILEOP_ABORT) goto done; } handler( context, SPFILENOTIFY_ENDRENAME, (UINT_PTR)&paths, 0 ); } handler( context, SPFILENOTIFY_ENDSUBQUEUE, FILEOP_RENAME, 0 ); } /* perform copies */ if (queue->copy_queue.count) { if (!(handler( context, SPFILENOTIFY_STARTSUBQUEUE, FILEOP_COPY, queue->copy_queue.count ))) goto done; for (op = queue->copy_queue.head; op; op = op->next) { WCHAR newpath[MAX_PATH]; build_filepathsW( op, &paths ); op_result = handler( context, SPFILENOTIFY_STARTCOPY, (UINT_PTR)&paths, FILEOP_COPY ); if (op_result == FILEOP_ABORT) goto done; if (op_result == FILEOP_NEWPATH) op_result = FILEOP_DOIT; while (op_result == FILEOP_DOIT || op_result == FILEOP_NEWPATH) { TRACE( "copying file %s -> %s\n", debugstr_w( op_result == FILEOP_NEWPATH ? newpath : paths.Source ), debugstr_w(paths.Target) ); if (op->dst_path) { if (!create_full_pathW( op->dst_path )) { paths.Win32Error = GetLastError(); op_result = handler( context, SPFILENOTIFY_COPYERROR, (UINT_PTR)&paths, (UINT_PTR)newpath ); if (op_result == FILEOP_ABORT) goto done; } } if (do_file_copyW( op_result == FILEOP_NEWPATH ? newpath : paths.Source, paths.Target, op->style )) break; /* success */ /* try to extract it from the cabinet file */ if (op->src_tag) { if (extract_cabinet_file( op->src_tag, op->src_root, paths.Source, paths.Target )) break; } paths.Win32Error = GetLastError(); op_result = handler( context, SPFILENOTIFY_COPYERROR, (UINT_PTR)&paths, (UINT_PTR)newpath ); if (op_result == FILEOP_ABORT) goto done; } handler( context, SPFILENOTIFY_ENDCOPY, (UINT_PTR)&paths, 0 ); } handler( context, SPFILENOTIFY_ENDSUBQUEUE, FILEOP_COPY, 0 ); } result = TRUE; done: handler( context, SPFILENOTIFY_ENDQUEUE, result, 0 ); HeapFree( GetProcessHeap(), 0, (void *)paths.Source ); HeapFree( GetProcessHeap(), 0, (void *)paths.Target ); return result; } /*********************************************************************** * SetupScanFileQueueA (SETUPAPI.@) */ BOOL WINAPI SetupScanFileQueueA( HSPFILEQ queue, DWORD flags, HWND window, PSP_FILE_CALLBACK_A callback, PVOID context, PDWORD result ) { FIXME("stub\n"); return FALSE; } /*********************************************************************** * SetupScanFileQueueW (SETUPAPI.@) */ BOOL WINAPI SetupScanFileQueueW( HSPFILEQ queue, DWORD flags, HWND window, PSP_FILE_CALLBACK_W callback, PVOID context, PDWORD result ) { FIXME("stub\n"); return FALSE; } /*********************************************************************** * SetupGetFileQueueCount (SETUPAPI.@) */ BOOL WINAPI SetupGetFileQueueCount( HSPFILEQ handle, UINT op, PUINT result ) { struct file_queue *queue = handle; switch(op) { case FILEOP_COPY: *result = queue->copy_queue.count; return TRUE; case FILEOP_RENAME: *result = queue->rename_queue.count; return TRUE; case FILEOP_DELETE: *result = queue->delete_queue.count; return TRUE; } return FALSE; } /*********************************************************************** * SetupGetFileQueueFlags (SETUPAPI.@) */ BOOL WINAPI SetupGetFileQueueFlags( HSPFILEQ handle, PDWORD flags ) { struct file_queue *queue = handle; *flags = queue->flags; return TRUE; } /*********************************************************************** * SetupSetFileQueueFlags (SETUPAPI.@) */ BOOL WINAPI SetupSetFileQueueFlags( HSPFILEQ handle, DWORD mask, DWORD flags ) { struct file_queue *queue = handle; queue->flags = (queue->flags & ~mask) | flags; return TRUE; } /*********************************************************************** * SetupInitDefaultQueueCallback (SETUPAPI.@) */ PVOID WINAPI SetupInitDefaultQueueCallback( HWND owner ) { return SetupInitDefaultQueueCallbackEx( owner, 0, 0, 0, NULL ); } /*********************************************************************** * SetupInitDefaultQueueCallbackEx (SETUPAPI.@) */ PVOID WINAPI SetupInitDefaultQueueCallbackEx( HWND owner, HWND progress, UINT msg, DWORD reserved1, PVOID reserved2 ) { struct default_callback_context *context; if ((context = HeapAlloc( GetProcessHeap(), 0, sizeof(*context) ))) { context->owner = owner; context->progress = progress; context->message = msg; } return context; } /*********************************************************************** * SetupTermDefaultQueueCallback (SETUPAPI.@) */ void WINAPI SetupTermDefaultQueueCallback( PVOID context ) { HeapFree( GetProcessHeap(), 0, context ); } /*********************************************************************** * SetupDefaultQueueCallbackA (SETUPAPI.@) */ UINT WINAPI SetupDefaultQueueCallbackA( PVOID context, UINT notification, UINT_PTR param1, UINT_PTR param2 ) { FILEPATHS_A *paths = (FILEPATHS_A *)param1; switch(notification) { case SPFILENOTIFY_STARTQUEUE: TRACE( "start queue\n" ); return TRUE; case SPFILENOTIFY_ENDQUEUE: TRACE( "end queue\n" ); return 0; case SPFILENOTIFY_STARTSUBQUEUE: TRACE( "start subqueue %d count %d\n", param1, param2 ); return TRUE; case SPFILENOTIFY_ENDSUBQUEUE: TRACE( "end subqueue %d\n", param1 ); return 0; case SPFILENOTIFY_STARTDELETE: TRACE( "start delete %s\n", debugstr_a(paths->Target) ); return FILEOP_DOIT; case SPFILENOTIFY_ENDDELETE: TRACE( "end delete %s\n", debugstr_a(paths->Target) ); return 0; case SPFILENOTIFY_DELETEERROR: ERR( "delete error %d %s\n", paths->Win32Error, debugstr_a(paths->Target) ); return FILEOP_SKIP; case SPFILENOTIFY_STARTRENAME: TRACE( "start rename %s -> %s\n", debugstr_a(paths->Source), debugstr_a(paths->Target) ); return FILEOP_DOIT; case SPFILENOTIFY_ENDRENAME: TRACE( "end rename %s -> %s\n", debugstr_a(paths->Source), debugstr_a(paths->Target) ); return 0; case SPFILENOTIFY_RENAMEERROR: ERR( "rename error %d %s -> %s\n", paths->Win32Error, debugstr_a(paths->Source), debugstr_a(paths->Target) ); return FILEOP_SKIP; case SPFILENOTIFY_STARTCOPY: TRACE( "start copy %s -> %s\n", debugstr_a(paths->Source), debugstr_a(paths->Target) ); return FILEOP_DOIT; case SPFILENOTIFY_ENDCOPY: TRACE( "end copy %s -> %s\n", debugstr_a(paths->Source), debugstr_a(paths->Target) ); return 0; case SPFILENOTIFY_COPYERROR: ERR( "copy error %d %s -> %s\n", paths->Win32Error, debugstr_a(paths->Source), debugstr_a(paths->Target) ); return FILEOP_SKIP; case SPFILENOTIFY_NEEDMEDIA: TRACE( "need media\n" ); return FILEOP_SKIP; default: FIXME( "notification %d params %x,%x\n", notification, param1, param2 ); break; } return 0; } /*********************************************************************** * SetupDefaultQueueCallbackW (SETUPAPI.@) */ UINT WINAPI SetupDefaultQueueCallbackW( PVOID context, UINT notification, UINT_PTR param1, UINT_PTR param2 ) { FIXME( "notification %d params %x,%x\n", notification, param1, param2 ); return FILEOP_SKIP; }