From 8a20c82e2db729e0b37522e75d743419e988c34a Mon Sep 17 00:00:00 2001
From: Vincent Povirk <vincent@codeweavers.com>
Date: Wed, 22 Oct 2008 17:19:14 -0500
Subject: [PATCH] setupapi: add minimal implementation of ProfileItems

---
 dlls/setupapi/Makefile.in     |    2 +-
 dlls/setupapi/install.c       |  136 ++++++++++++++++++++++++++++++++++++++++-
 dlls/setupapi/tests/install.c |    4 +-
 3 files changed, 138 insertions(+), 4 deletions(-)

diff --git a/dlls/setupapi/Makefile.in b/dlls/setupapi/Makefile.in
index 1f633a4..91bc87b 100644
--- a/dlls/setupapi/Makefile.in
+++ b/dlls/setupapi/Makefile.in
@@ -6,7 +6,7 @@ VPATH     = @srcdir@
 MODULE    = setupapi.dll
 IMPORTLIB = setupapi
 IMPORTS   = uuid user32 version advapi32 rpcrt4 kernel32 ntdll
-DELAYIMPORTS = shell32 wintrust
+DELAYIMPORTS = shell32 wintrust ole32
 
 C_SRCS = \
 	devinst.c \
diff --git a/dlls/setupapi/install.c b/dlls/setupapi/install.c
index 321d4d4..aa0c6cb 100644
--- a/dlls/setupapi/install.c
+++ b/dlls/setupapi/install.c
@@ -20,6 +20,8 @@
 
 #include <stdarg.h>
 
+#define COBJMACROS
+
 #include "windef.h"
 #include "winbase.h"
 #include "winreg.h"
@@ -29,6 +31,9 @@
 #include "winuser.h"
 #include "winnls.h"
 #include "winsvc.h"
+#include "shlobj.h"
+#include "objidl.h"
+#include "objbase.h"
 #include "setupapi.h"
 #include "setupapi_private.h"
 #include "wine/unicode.h"
@@ -79,6 +84,8 @@ static const WCHAR UpdateIniFields[] = {'U','p','d','a','t','e','I','n','i','F',
 static const WCHAR RegisterDlls[]    = {'R','e','g','i','s','t','e','r','D','l','l','s',0};
 static const WCHAR UnregisterDlls[]  = {'U','n','r','e','g','i','s','t','e','r','D','l','l','s',0};
 static const WCHAR ProfileItems[]    = {'P','r','o','f','i','l','e','I','t','e','m','s',0};
+static const WCHAR Name[]            = {'N','a','m','e',0};
+static const WCHAR CmdLine[]         = {'C','m','d','L','i','n','e',0};
 static const WCHAR WineFakeDlls[]    = {'W','i','n','e','F','a','k','e','D','l','l','s',0};
 static const WCHAR DisplayName[]     = {'D','i','s','p','l','a','y','N','a','m','e',0};
 static const WCHAR Description[]     = {'D','e','s','c','r','i','p','t','i','o','n',0};
@@ -773,9 +780,136 @@ static BOOL bitreg_callback( HINF hinf, PCWSTR field, void *arg )
     return TRUE;
 }
 
+static LPCWSTR pathsepW( LPCWSTR path )
+{
+    static const WCHAR backslash[] = {'\\',0};
+
+    if (path && path[0] && path[strlenW(path)-1] != '\\')
+        return backslash;
+    else
+        return NULL;
+}
+
+static LPWSTR join_stringsW(DWORD count, ...)
+{
+    size_t len=0;
+    va_list va;
+    int i;
+    LPWSTR result, end;
+
+    va_start(va, count);
+    for (i=0; i<count; i++)
+    {
+        LPCWSTR str = va_arg(va, LPCWSTR);
+        if (str) len += strlenW(str);
+    }
+    va_end(va);
+
+    end = result = HeapAlloc(GetProcessHeap(), 0, (len+1)*sizeof(WCHAR));
+    if (!result)
+        return NULL;
+
+    va_start(va, count);
+    for (i=0; i<count; i++)
+    {
+        LPCWSTR str = va_arg(va, LPCWSTR);
+
+        if (str)
+        {
+            strcpyW(end, str);
+            end = end + strlenW(str);
+        }
+    }
+    va_end(va);
+
+    return result;
+}
+
 static BOOL profile_items_callback( HINF hinf, PCWSTR field, void *arg )
 {
-    FIXME( "should do profile items %s\n", debugstr_w(field) );
+    LPWSTR name=NULL, lnkpath=NULL, cmdline=NULL;
+    WCHAR lnkdir[MAX_PATH];
+    INFCONTEXT context;
+    IShellLinkW* shelllink=NULL;
+    IPersistFile* persistfile=NULL;
+    int attrs=0;
+
+    static const WCHAR dotlnk[] = {'.','l','n','k',0};
+
+    TRACE( "(%s)\n", debugstr_w(field) );
+
+    if (SetupFindFirstLineW( hinf, field, Name, &context ))
+    {
+        name = dup_section_line_field( hinf, field, Name, 1 );
+
+        SetupGetIntField( &context, 2, &attrs );
+
+        if (attrs)
+            FIXME( "unhandled attributes: %x\n", attrs );
+    }
+    
+    if (!name)
+    {
+        ERR( "missing Name entry\n" );
+
+        return TRUE;
+    }
+
+    if (SetupFindFirstLineW( hinf, field, CmdLine, &context ))
+    {
+        LPWSTR dir=NULL, filename=NULL;
+
+        if ((dir = PARSER_get_dest_dir( &context )) &&
+            (filename = dup_section_line_field( hinf, field, CmdLine, 3 )))
+        {
+            cmdline = join_stringsW( 3, dir, pathsepW(dir), filename );
+        }
+
+        HeapFree( GetProcessHeap(), 0, dir );
+        HeapFree( GetProcessHeap(), 0, filename );
+    }
+
+    if (!cmdline)
+    {
+        ERR( "missing CmdLine entry\n" );
+
+        HeapFree( GetProcessHeap(), 0, cmdline );
+        return TRUE;
+    }
+
+    TRACE( "name: %s\n", debugstr_w(name) );
+    TRACE( "cmdline: %s\n", debugstr_w(cmdline) );
+
+    CoInitialize(NULL);
+
+    if (!SUCCEEDED(CoCreateInstance( &CLSID_ShellLink, NULL,
+                        CLSCTX_INPROC_SERVER, &IID_IShellLinkW, (LPVOID*)&shelllink)))
+    {
+        ERR( "failed to create IShellLink instance\n" );
+        return TRUE;
+    }
+
+    IShellLinkW_SetPath( shelllink, cmdline );
+
+    SHGetFolderPathW( NULL, CSIDL_COMMON_PROGRAMS, NULL, SHGFP_TYPE_CURRENT, lnkdir );
+
+    lnkpath = join_stringsW( 4, lnkdir, pathsepW(lnkdir), name, dotlnk );
+
+    if (SUCCEEDED(IShellLinkW_QueryInterface( shelllink, &IID_IPersistFile, (LPVOID*)&persistfile)))
+    {
+        TRACE( "writing link: %s\n", debugstr_w(lnkpath) );
+
+        IPersistFile_Save( persistfile, lnkpath, FALSE );
+
+        IPersistFile_Release( persistfile );
+    }
+
+    HeapFree( GetProcessHeap(), 0, name );
+    HeapFree( GetProcessHeap(), 0, cmdline );
+    HeapFree( GetProcessHeap(), 0, lnkpath );
+
+    IShellLinkW_Release( shelllink );
+
     return TRUE;
 }
 
diff --git a/dlls/setupapi/tests/install.c b/dlls/setupapi/tests/install.c
index d2e6ddf..daa4661 100644
--- a/dlls/setupapi/tests/install.c
+++ b/dlls/setupapi/tests/install.c
@@ -429,9 +429,9 @@ static void test_profile_items(void)
     else
     {
         snprintf(path, MAX_PATH, "%s\\TestDir", commonprogs);
-        ok(INVALID_FILE_ATTRIBUTES != GetFileAttributes(path), "directory not created\n");
+        todo_wine ok(INVALID_FILE_ATTRIBUTES != GetFileAttributes(path), "directory not created\n");
         snprintf(path, MAX_PATH, "%s\\TestDir\\TestItem2.lnk", commonprogs);
-        ok(INVALID_FILE_ATTRIBUTES != GetFileAttributes(path), "link not created\n");
+        todo_wine ok(INVALID_FILE_ATTRIBUTES != GetFileAttributes(path), "link not created\n");
     }
 
     snprintf(path, MAX_PATH, "%s\\TestItem.lnk", commonprogs);
-- 
1.5.6.3