Sharing shell_util_unittest code to verify shortcuts

chrome/installer/util/shell_util.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 //
 // This file defines functions that integrate Chrome in Windows shell. These
 // functions can be used by Chrome as well as Chrome installer. All of the
 // work is done by the local functions defined in anonymous namespace in
 // this class.
 
 #include "chrome/installer/util/shell_util.h"
@@ skipping 1639 matching lines @@
       shortcut.c_str(),
       chrome_path.value().c_str(),
       arguments.c_str(),
       description.c_str(),
       icon_path.c_str(),
       icon_index,
       app_id.c_str(),
       ConvertShellUtilShortcutOptionsToFileUtil(options));
 }
 
+ShellUtil::VerifyShortcutStatus ShellUtil::VerifyChromeShortcut(
+    const string16& exe_path, const string16& shortcut,
+    const string16& description, int icon_index) {
+  base::win::ScopedComPtr<IShellLink> i_shell_link;
+  base::win::ScopedComPtr<IPersistFile> i_persist_file;
+  wchar_t long_path[MAX_PATH] = {0};
+  wchar_t short_path[MAX_PATH] = {0};
+  wchar_t file_path[MAX_PATH] = {0};
+  wchar_t icon_path[MAX_PATH] = {0};
+  wchar_t desc[MAX_PATH] = {0};
+  int index = 0;
+
+  // Get the shortcut's properties.
+  if (FAILED(i_shell_link.CreateInstance(CLSID_ShellLink, NULL,
+                                         CLSCTX_INPROC_SERVER)) ||
+      FAILED(i_persist_file.QueryFrom(i_shell_link)) ||
+      FAILED(i_persist_file->Load(shortcut.c_str(), 0)) ||
+      ::GetLongPathName(exe_path.c_str(), long_path, MAX_PATH) == 0 ||
+      ::GetShortPathName(exe_path.c_str(), short_path, MAX_PATH) == 0 ||
+      FAILED(i_shell_link->GetPath(file_path, MAX_PATH, NULL,
+                                   SLGP_UNCPRIORITY)) ||
+      FAILED(i_shell_link->GetIconLocation(icon_path, MAX_PATH, &index)) ||
+      FAILED(i_shell_link->GetDescription(desc, MAX_PATH)) ||
+      FAILED(i_shell_link->GetDescription(desc, MAX_PATH)))
+    return VERIFY_SHORTCUT_FAILURE_UNEXPECTED;
+
+  FilePath path(file_path);
+  if (path != FilePath(long_path) && path != FilePath(short_path))
+    return VERIFY_SHORTCUT_FAILURE_PATH;
+
+  if (string16(desc) != string16(description))
+    return VERIFY_SHORTCUT_FAILURE_DESCRIPTION;
+
+  if (index != icon_index)
+    return VERIFY_SHORTCUT_FAILURE_ICON_INDEX;
+
+  return VERIFY_SHORTCUT_SUCCESS;
+}
+
 bool ShellUtil::GetUserSpecificRegistrySuffix(string16* suffix) {
   // Use a thread-safe cache for the user's suffix.
   static base::LazyInstance<UserSpecificRegistrySuffix>::Leaky suffix_instance =
       LAZY_INSTANCE_INITIALIZER;
   return suffix_instance.Get().GetSuffix(suffix);
 }
 
 bool ShellUtil::GetOldUserSpecificRegistrySuffix(string16* suffix) {
   wchar_t user_name[256];
   DWORD size = arraysize(user_name);
@@ skipping 47 matching lines @@
     // are any left...).
     if (free_bits >= 8 && next_byte_index < size) {
       free_bits -= 8;
       bit_stream += bytes[next_byte_index++] << free_bits;
     }
   }
 
   DCHECK_EQ(ret.length(), encoded_length);
   return ret;
 }