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::VerifyShortcuts 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;
+
+  // Get pointer to the IShellLink interface
+  if (FAILED(i_shell_link.CreateInstance(CLSID_ShellLink, NULL,
+                                         CLSCTX_INPROC_SERVER)))
+    return VERIFY_SHORTCUT_FAILURE_CREATE_SHELL_LINK_INSTANCE;
+
+  // Query IShellLink for the IPersistFile interface
+  if (FAILED(i_persist_file.QueryFrom(i_shell_link)))
+    return VERIFY_SHORTCUT_FAILURE_QUERY_FROM_PERSIST;
+
+  if (FAILED(i_persist_file->Load(shortcut.c_str(), 0)))
+    return VERIFY_SHORTCUT_FAILURE_LOAD_FROM_PERSIST;
+
+  wchar_t long_path[MAX_PATH] = {0};
+  wchar_t short_path[MAX_PATH] = {0};
+
+  if (::GetLongPathName(exe_path.c_str(), long_path, MAX_PATH) == 0 ||
+      ::GetShortPathName(exe_path.c_str(), short_path, MAX_PATH) == 0)
+    return VERIFY_SHORTCUT_FAILURE_GET_PATH_NAMES;
+
+  wchar_t file_path[MAX_PATH] = {0};
+  wchar_t desc[MAX_PATH] = {0};
+  wchar_t icon_path[MAX_PATH] = {0};
+  int index = 0;
+
+  if (FAILED(i_shell_link->GetPath(file_path, MAX_PATH, NULL,

[gab, 2012/08/08 20:41:14]

It's the opposite we want.

i.e. the things that should never fail (CreateInstance, Load shortcut, Get*)
should all be grouped in one and return something like
VERIFY_SHORTCUT_FAILURE_UNEXPECTED. Comparisons however should be split and
report on which comparison failed through the enum if any.

[Halli, 2012/08/08 21:06:20]

Done.

+      SLGP_UNCPRIORITY)) || FilePath(file_path) != FilePath(long_path) &&
+      FilePath(file_path) != FilePath(short_path) ||
+      FAILED(i_shell_link->GetDescription(desc, MAX_PATH)) ||
+      string16(desc) != string16(description) ||
+      FAILED(i_shell_link->GetIconLocation(icon_path, MAX_PATH, &index)) ||
+      index != icon_index)
+    return VERIFY_SHORTCUT_FAILURE;
+
+  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;
 }