From 3112fd22697f609570e24d9f2bac2b481228b74a Mon Sep 17 00:00:00 2001
From: Mike McCormack <mike_mccormack@looksmart.com.au>
Date: Fri, 29 Sep 2000 00:25:56 +0000
Subject: [PATCH] Implemented SHDeleteEmptyKeyA, SHDeleteKeyA.

---
 dlls/shlwapi/reg.c        | 122 ++++++++++++++++++++++++++++++++++++++++++++--
 dlls/shlwapi/shlwapi.spec |   4 +-
 include/shlwapi.h         |   8 +++
 3 files changed, 128 insertions(+), 6 deletions(-)

diff --git a/dlls/shlwapi/reg.c b/dlls/shlwapi/reg.c
index 9e38a29edfe..1c4acffa535 100644
--- a/dlls/shlwapi/reg.c
+++ b/dlls/shlwapi/reg.c
@@ -223,17 +223,78 @@ HRESULT WINAPI SHQueryValueExW (
 
 /*************************************************************************
  * SHDeleteKeyA   [SHLWAPI.@]
+ *
+ * It appears this function is made available to account for the differences
+ * between the Win9x and WinNT/2k RegDeleteKeyA functions.
+ *
+ * According to docs, Win9x RegDeleteKeyA will delete all subkeys, whereas
+ * WinNt/2k will only delete the key if empty.
  */
 HRESULT WINAPI SHDeleteKeyA(
-	HKEY hkey,
-	LPCSTR pszSubKey)
+	HKEY hKey,
+	LPCSTR lpszSubKey)
 {
-	FIXME("hkey=0x%08x, %s\n", hkey, debugstr_a(pszSubKey));
-	return 0;
+    DWORD r, dwKeyCount, dwSize, i, dwMaxSubkeyLen;
+    HKEY hSubKey;
+    LPSTR lpszName;
+
+    TRACE("hkey=0x%08x, %s\n", hKey, debugstr_a(lpszSubKey));
+
+    hSubKey = 0;
+    r = RegOpenKeyExA(hKey, lpszSubKey, 0, KEY_READ, &hSubKey);
+    if(r != ERROR_SUCCESS)
+        return r;
+
+    /* find how many subkeys there are */
+    dwKeyCount = 0;
+    dwMaxSubkeyLen = 0;
+    r = RegQueryInfoKeyA(hSubKey, NULL, NULL, NULL, &dwKeyCount, 
+                    &dwMaxSubkeyLen, NULL, NULL, NULL, NULL, NULL, NULL);
+    if(r != ERROR_SUCCESS)
+    {
+        RegCloseKey(hSubKey);
+        return r;
+    }
+
+    /* alloc memory for the longest string terminating 0 */
+    dwMaxSubkeyLen++;
+    lpszName = HeapAlloc(GetProcessHeap(), 0, dwMaxSubkeyLen*sizeof(CHAR));
+    if(!lpszName)
+    {
+        RegCloseKey(hSubKey);
+        return ERROR_NOT_ENOUGH_MEMORY;
+    }
+
+    /* recursively delete all the subkeys */
+    for(i=0; i<dwKeyCount; i++)
+    {
+        dwSize = dwMaxSubkeyLen;
+        r = RegEnumKeyExA(hSubKey, i, lpszName, &dwSize, NULL, NULL, NULL, NULL);
+        if(r != ERROR_SUCCESS)
+            break;
+        r = SHDeleteKeyA(hSubKey, lpszName);
+        if(r != ERROR_SUCCESS)
+            break;
+    }
+
+    HeapFree(GetProcessHeap(), 0, lpszName);
+
+    RegCloseKey(hSubKey);
+
+    if(r == ERROR_SUCCESS)
+        r = RegDeleteKeyA(hKey, lpszSubKey);
+
+    return r;
 }
 
 /*************************************************************************
  * SHDeleteKeyW   [SHLWAPI.@]
+ *
+ * It appears this function is made available to account for the differences
+ * between the Win9x and WinNT/2k RegDeleteKeyA functions.
+ *
+ * According to docs, Win9x RegDeleteKeyA will delete all subkeys, whereas
+ * WinNt/2k will only delete the key if empty.
  */
 HRESULT WINAPI SHDeleteKeyW(
 	HKEY hkey,
@@ -242,3 +303,56 @@ HRESULT WINAPI SHDeleteKeyW(
 	FIXME("hkey=0x%08x, %s\n", hkey, debugstr_w(pszSubKey));
 	return 0;
 }
+
+/*************************************************************************
+ * SHDeleteEmptyKeyA   [SHLWAPI.@]
+ *
+ * It appears this function is made available to account for the differences
+ * between the Win9x and WinNT/2k RegDeleteKeyA functions.
+ *
+ * According to docs, Win9x RegDeleteKeyA will delete all subkeys, whereas
+ * WinNt/2k will only delete the key if empty.
+ */
+DWORD WINAPI SHDeleteEmptyKeyA(HKEY hKey, LPCSTR lpszSubKey)
+{
+    DWORD r, dwKeyCount;
+    HKEY hSubKey;
+
+    TRACE("hkey=0x%08x, %s\n", hKey, debugstr_a(lpszSubKey));
+
+    hSubKey = 0;
+    r = RegOpenKeyExA(hKey, lpszSubKey, 0, KEY_READ, &hSubKey);
+    if(r != ERROR_SUCCESS)
+        return r;
+
+    dwKeyCount = 0;
+    r = RegQueryInfoKeyA(hSubKey, NULL, NULL, NULL, &dwKeyCount, 
+                    NULL, NULL, NULL, NULL, NULL, NULL, NULL);
+    if(r != ERROR_SUCCESS)
+        return r;
+
+    RegCloseKey(hSubKey);
+
+    if(dwKeyCount)
+        return ERROR_KEY_HAS_CHILDREN;
+
+    r = RegDeleteKeyA(hKey, lpszSubKey);
+
+    return r;
+}
+
+/*************************************************************************
+ * SHDeleteEmptyKeyW   [SHLWAPI.@]
+ *
+ * It appears this function is made available to account for the differences
+ * between the Win9x and WinNT/2k RegDeleteKeyA functions.
+ *
+ * According to docs, Win9x RegDeleteKeyA will delete all subkeys, whereas
+ * WinNt/2k will only delete the key if empty.
+ */
+DWORD WINAPI SHDeleteEmptyKeyW(HKEY hKey, LPCWSTR lpszSubKey)
+{
+    FIXME("hkey=0x%08x, %s\n", hKey, debugstr_w(lpszSubKey));
+    return 0;
+}
+
diff --git a/dlls/shlwapi/shlwapi.spec b/dlls/shlwapi/shlwapi.spec
index 1ab838e3db7..a71ec6419ff 100644
--- a/dlls/shlwapi/shlwapi.spec
+++ b/dlls/shlwapi/shlwapi.spec
@@ -558,8 +558,8 @@ import kernel32
 @ stdcall PathUnquoteSpacesA (str) PathUnquoteSpacesA
 @ stdcall PathUnquoteSpacesW (wstr) PathUnquoteSpacesW
 @ stdcall SHCreateShellPalette(long)SHCreateShellPalette
-@ stub    SHDeleteEmptyKeyA
-@ stub    SHDeleteEmptyKeyW
+@ stdcall SHDeleteEmptyKeyA(long ptr) SHDeleteEmptyKeyA
+@ stdcall SHDeleteEmptyKeyW(long ptr) SHDeleteEmptyKeyW
 @ stdcall SHDeleteKeyA(long str) SHDeleteKeyA
 @ stdcall SHDeleteKeyW(long wstr) SHDeleteKeyW
 @ stub    SHDeleteOrphanKeyA
diff --git a/include/shlwapi.h b/include/shlwapi.h
index 6d41ebb8e0f..c120cbda293 100644
--- a/include/shlwapi.h
+++ b/include/shlwapi.h
@@ -122,6 +122,14 @@ void WINAPI PathRemoveBlanksW(LPWSTR lpszPath);
 #define  PathRemoveBlanks WINELIB_NAME_AW(PathRemoveBlanks)
 void WINAPI PathRemoveBlanksAW(LPVOID lpszPath);
 
+HRESULT WINAPI SHDeleteKeyA(HKEY hKey, LPCSTR lpszSubKey);
+HRESULT WINAPI SHDeleteKeyW(HKEY hkey, LPCWSTR pszSubKey);
+#define  SHDeleteKey WINELIB_NAME_AW(SHDeleteKey)
+
+DWORD WINAPI SHDeleteEmptyKeyA(HKEY hKey, LPCSTR lpszSubKey);
+DWORD WINAPI SHDeleteEmptyKeyW(HKEY hKey, LPCWSTR lpszSubKey);
+#define  SHDeleteEmptyKey WINELIB_NAME_AW(SHDeleteEmptyKey)
+
 #ifdef __cplusplus
 } /* extern "C" */
 #endif /* defined(__cplusplus) */
-- 
2.11.4.GIT