The Tor Browser: xulrunner/tools/redit/redit.cpp@b8a032363ba2 (annotated)

xulrunner/tools/redit/redit.cpp@b8a032363ba2 (annotated)

xulrunner/tools/redit/redit.cpp

Thu, 22 Jan 2015 13:21:57 +0100

author: Michael Schloh von Bennewitz <michael@schloh.com>
date: Thu, 22 Jan 2015 13:21:57 +0100
branch: TOR_BUG_9701
changeset 15: b8a032363ba2
permissions: -rw-r--r--

Incorporate requested changes from Mozilla in review:
https://bugzilla.mozilla.org/show_bug.cgi?id=1123480#c6

 /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
 /* This Source Code Form is subject to the terms of the Mozilla Public
  * License, v. 2.0. If a copy of the MPL was not distributed with this file,
  * You can obtain one at http://mozilla.org/MPL/2.0/. */
 // System headers (alphabetical)
 #include <fcntl.h>
 #include <io.h>
 #include <share.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <sys/stat.h>
 #include <windows.h>
 // Mozilla headers (alphabetical)
 #include "mozilla/FileUtils.h"  // ScopedClose
 #include "nsAutoPtr.h"          // nsAutoArrayPtr
 /*
 Icon files are made up of:
 IconHeader
 IconDirEntry1
 IconDirEntry2
 ...
 IconDirEntryN
 IconData1
 IconData2
 ...
 IconDataN
 Each IconData must be added as a new RT_ICON resource to the exe. Then
 an RT_GROUP_ICON resource must be added that contains an equivalent
 header:
 IconHeader
 IconResEntry1
 IconResEntry2
 ...
 IconResEntryN
 */
 #pragma pack(push, 2)
 typedef struct
 {
   WORD Reserved;
   WORD ResourceType;
   WORD ImageCount;
 } IconHeader;
 typedef struct
 {
   BYTE Width;
   BYTE Height;
   BYTE Colors;
   BYTE Reserved;
   WORD Planes;
   WORD BitsPerPixel;
   DWORD ImageSize;
   DWORD ImageOffset;
 } IconDirEntry;
 typedef struct
 {
   BYTE Width;
   BYTE Height;
   BYTE Colors;
   BYTE Reserved;
   WORD Planes;
   WORD BitsPerPixel;
   DWORD ImageSize;
   WORD ResourceID;    // This field is the one difference to above
 } IconResEntry;
 #pragma pack(pop)
 namespace {
   /**
    * ScopedResourceUpdate is a RAII wrapper for Windows resource updating
    *
    * Instances |EndUpdateResourceW()| their handles when they go out of scope.
    * They pass |TRUE| as the second argument to |EndUpdateResourceW()|, which
    * causes the resource update to be aborted (changes are discarded).
    */
   struct ScopedResourceUpdateTraits
   {
     typedef HANDLE type;
     static type empty() { return nullptr; }
     static void release(type handle) {
       if(nullptr != handle) {
         EndUpdateResourceW(handle, TRUE); // Discard changes
       }
     }
   };
   typedef mozilla::Scoped<ScopedResourceUpdateTraits> ScopedResourceUpdate;
 };
 #ifdef __MINGW32__
 extern "C"
 #endif
 int
 wmain(int argc, wchar_t** argv)
 {
   if (argc != 3) {
     printf("Usage: redit <exe file> <icon file>\n");
     return 1;
   }
   mozilla::ScopedClose file;
   if (0 != _wsopen_s(&file.rwget(),
                      argv[2],
                      _O_BINARY | _O_RDONLY,
                      _SH_DENYWR,
                      _S_IREAD)
   || (-1 == file)) {
     fprintf(stderr, "Unable to open icon file.\n");
     return 1;
   }
   // Load all the data from the icon file
   long filesize = _filelength(file);
   nsAutoArrayPtr<BYTE> data(new BYTE[filesize]);
   if(!data) {
     fprintf(stderr, "Failed to allocate memory for icon file.\n");
     return 1;
   }
   _read(file, data, filesize);
   IconHeader* header = reinterpret_cast<IconHeader*>(data.get());
   // Open the target library for updating
   ScopedResourceUpdate updateRes(BeginUpdateResourceW(argv[1], FALSE));
   if (nullptr == updateRes) {
     fprintf(stderr, "Unable to open library for modification.\n");
     return 1;
   }
   // Allocate the group resource entry
   long groupSize = sizeof(IconHeader)
                  + header->ImageCount * sizeof(IconResEntry);
   nsAutoArrayPtr<BYTE> group(new BYTE[groupSize]);
   if(!group) {
     fprintf(stderr, "Failed to allocate memory for new images.\n");
     return 1;
   }
   memcpy(group, data, sizeof(IconHeader));
   IconDirEntry* sourceIcon =
                     reinterpret_cast<IconDirEntry*>(data
                                                   + sizeof(IconHeader));
   IconResEntry* targetIcon =
                     reinterpret_cast<IconResEntry*>(group
                                                   + sizeof(IconHeader));
   for (int id = 1; id <= header->ImageCount; id++) {
     // Add the individual icon
     if (!UpdateResourceW(updateRes, RT_ICON, MAKEINTRESOURCE(id),
                          MAKELANGID(LANG_NEUTRAL, SUBLANG_NEUTRAL),
                          data + sourceIcon->ImageOffset,
                          sourceIcon->ImageSize)) {
       fprintf(stderr, "Unable to update resource (RT_ICON).\n");
       return 1;
     }
     // Copy the data for this icon
     // (note that the structs have different sizes)
     memcpy(targetIcon, sourceIcon, sizeof(IconResEntry));
     targetIcon->ResourceID = id;
     sourceIcon++;
     targetIcon++;
   }
   if (!UpdateResourceW(updateRes, RT_GROUP_ICON, L"MAINICON",
                        MAKELANGID(LANG_NEUTRAL, SUBLANG_NEUTRAL),
                        group, groupSize)) {
     fprintf(stderr, "Unable to update resource (RT_GROUP_ICON).\n");
     return 1;
   }
   // Save the modifications
   if(!EndUpdateResourceW(updateRes.forget(), FALSE)) {
     fprintf(stderr, "Unable to write changes to library.\n");
     return 1;
   }
   return 0;
 }

The Tor Browser / annotate

xulrunner/tools/redit/redit.cpp@b8a032363ba2 (annotated)

xulrunner/tools/redit/redit.cpp