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 1706 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;
 }
+
+bool 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 false;
+
+  // Query IShellLink for the IPersistFile interface
+  if (FAILED(i_persist_file.QueryFrom(i_shell_link)))
+    return false;
+
+  if (FAILED(i_persist_file->Load(shortcut.c_str(), 0)))
+    return false;
+
+  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 false;
+
+  wchar_t file_path[MAX_PATH] = {0};
+
+  // File path did not match exe path

[gab, 2012/08/08 19:17:08]

nit: remove this comment as well

+  if (FAILED(i_shell_link->GetPath(file_path, MAX_PATH, NULL,
+      SLGP_UNCPRIORITY)) || FilePath(file_path) != FilePath(long_path) &&

[gab, 2012/08/08 19:17:08]

align SLGP_UNCPRIORITY with file_path above and put the next condition on the
next line

+      FilePath(file_path) != FilePath(short_path))
+    return false;
+
+  wchar_t desc[MAX_PATH] = {0};
+
+  if (FAILED(i_shell_link->GetDescription(desc, MAX_PATH)) ||
+      string16(desc) != string16(description))
+    return false;
+
+  wchar_t icon_path[MAX_PATH] = {0};
+  int index = 0;
+
+  if (FAILED(i_shell_link->GetIconLocation(icon_path, MAX_PATH, &index)) ||
+      FilePath(file_path) != FilePath(long_path) &&
+      FilePath(file_path) != FilePath(short_path) || index != icon_index)
+    return false;
+
+  return true;
+}