1 /*

     2  * Copyright (c) 2019, Oracle and/or its affiliates. All rights reserved.

     3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

     4  *

     5  * This code is free software; you can redistribute it and/or modify it

     6  * under the terms of the GNU General Public License version 2 only, as

     7  * published by the Free Software Foundation.  Oracle designates this

     8  * particular file as subject to the "Classpath" exception as provided

     9  * by Oracle in the LICENSE file that accompanied this code.

    10  *

    11  * This code is distributed in the hope that it will be useful, but WITHOUT

    12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

    13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

    14  * version 2 for more details (a copy is included in the LICENSE file that

    15  * accompanied this code).

    16  *

    17  * You should have received a copy of the GNU General Public License version

    18  * 2 along with this work; if not, write to the Free Software Foundation,

    19  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

    20  *

    21  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

    22  * or visit www.oracle.com if you need additional information or have any

    23  * questions.

    24  */

    25 

    26 #ifndef RESOURCEEDITOR_H

    27 #define RESOURCEEDITOR_H

    28 

    29 #include <windows.h>

    30 #include <vector>

    31 #include <string>

    32 

    33 

    34 class ResourceEditor {

    35 public:

    36     class FileLock {

    37     public:

    38         FileLock(const std::wstring& binaryPath);

    39         ~FileLock();

    40 

    41         HANDLE get() const {

    42             return h;

    43         }

    44 

    45         void discard(bool v = true) {

    46             theDiscard = v;

    47         }

    48 

    49     private:

    50         FileLock(const FileLock&);

    51         FileLock& operator=(const FileLock&);

    52     private:

    53         HANDLE h;

    54         bool theDiscard;

    55     };

    56 

    57 public:

    58     ResourceEditor();

    59 

    60     /**

    61      * Set the language identifier of the resource to be updated.

    62      */

    63     ResourceEditor& language(unsigned v) {

    64         lang = v;

    65         return *this;

    66     }

    67 

    68     /**

    69      * Set the resource type to be updated.

    70      */

    71     ResourceEditor& type(unsigned v);

    72 

    73     /**

    74      * Set the resource type to be updated.

    75      */

    76     ResourceEditor& type(LPCWSTR v);

    77 

    78     /**

    79      * Set resource ID.

    80      */

    81     ResourceEditor& id(unsigned v);

    82 

    83     /**

    84      * Set resource ID.

    85      */

    86     ResourceEditor& id(LPCWSTR v);

    87 

    88     /**

    89      * Relaces resource configured in the given binary with the given data stream.

    90      */

    91     ResourceEditor& apply(const FileLock& dstBinary, std::istream& srcStream, std::streamsize size=0);

    92 

    93     /**

    94      * Relaces resource configured in the given binary with contents of

    95      * the given binary file.

    96      */

    97     ResourceEditor& apply(const FileLock& dstBinary, const std::wstring& srcFile);

    98 

    99 private:

   100     unsigned lang;

   101     std::wstring theId;

   102     LPCWSTR theIdPtr;

   103     std::wstring theType;

   104     LPCWSTR theTypePtr;

   105 };

   106 

   107 #endif // #ifndef RESOURCEEDITOR_H