mirror of
https://github.com/GaryOderNichts/drc-fw-patches.git
synced 2024-11-24 11:19:16 +01:00
Initial commit
This commit is contained in:
commit
22562059d8
2
.gitignore
vendored
Normal file
2
.gitignore
vendored
Normal file
@ -0,0 +1,2 @@
|
||||
*drc_fw.bin
|
||||
unpacked
|
339
LICENSE
Normal file
339
LICENSE
Normal file
@ -0,0 +1,339 @@
|
||||
GNU GENERAL PUBLIC LICENSE
|
||||
Version 2, June 1991
|
||||
|
||||
Copyright (C) 1989, 1991 Free Software Foundation, Inc.,
|
||||
51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
||||
Everyone is permitted to copy and distribute verbatim copies
|
||||
of this license document, but changing it is not allowed.
|
||||
|
||||
Preamble
|
||||
|
||||
The licenses for most software are designed to take away your
|
||||
freedom to share and change it. By contrast, the GNU General Public
|
||||
License is intended to guarantee your freedom to share and change free
|
||||
software--to make sure the software is free for all its users. This
|
||||
General Public License applies to most of the Free Software
|
||||
Foundation's software and to any other program whose authors commit to
|
||||
using it. (Some other Free Software Foundation software is covered by
|
||||
the GNU Lesser General Public License instead.) You can apply it to
|
||||
your programs, too.
|
||||
|
||||
When we speak of free software, we are referring to freedom, not
|
||||
price. Our General Public Licenses are designed to make sure that you
|
||||
have the freedom to distribute copies of free software (and charge for
|
||||
this service if you wish), that you receive source code or can get it
|
||||
if you want it, that you can change the software or use pieces of it
|
||||
in new free programs; and that you know you can do these things.
|
||||
|
||||
To protect your rights, we need to make restrictions that forbid
|
||||
anyone to deny you these rights or to ask you to surrender the rights.
|
||||
These restrictions translate to certain responsibilities for you if you
|
||||
distribute copies of the software, or if you modify it.
|
||||
|
||||
For example, if you distribute copies of such a program, whether
|
||||
gratis or for a fee, you must give the recipients all the rights that
|
||||
you have. You must make sure that they, too, receive or can get the
|
||||
source code. And you must show them these terms so they know their
|
||||
rights.
|
||||
|
||||
We protect your rights with two steps: (1) copyright the software, and
|
||||
(2) offer you this license which gives you legal permission to copy,
|
||||
distribute and/or modify the software.
|
||||
|
||||
Also, for each author's protection and ours, we want to make certain
|
||||
that everyone understands that there is no warranty for this free
|
||||
software. If the software is modified by someone else and passed on, we
|
||||
want its recipients to know that what they have is not the original, so
|
||||
that any problems introduced by others will not reflect on the original
|
||||
authors' reputations.
|
||||
|
||||
Finally, any free program is threatened constantly by software
|
||||
patents. We wish to avoid the danger that redistributors of a free
|
||||
program will individually obtain patent licenses, in effect making the
|
||||
program proprietary. To prevent this, we have made it clear that any
|
||||
patent must be licensed for everyone's free use or not licensed at all.
|
||||
|
||||
The precise terms and conditions for copying, distribution and
|
||||
modification follow.
|
||||
|
||||
GNU GENERAL PUBLIC LICENSE
|
||||
TERMS AND CONDITIONS FOR COPYING, DISTRIBUTION AND MODIFICATION
|
||||
|
||||
0. This License applies to any program or other work which contains
|
||||
a notice placed by the copyright holder saying it may be distributed
|
||||
under the terms of this General Public License. The "Program", below,
|
||||
refers to any such program or work, and a "work based on the Program"
|
||||
means either the Program or any derivative work under copyright law:
|
||||
that is to say, a work containing the Program or a portion of it,
|
||||
either verbatim or with modifications and/or translated into another
|
||||
language. (Hereinafter, translation is included without limitation in
|
||||
the term "modification".) Each licensee is addressed as "you".
|
||||
|
||||
Activities other than copying, distribution and modification are not
|
||||
covered by this License; they are outside its scope. The act of
|
||||
running the Program is not restricted, and the output from the Program
|
||||
is covered only if its contents constitute a work based on the
|
||||
Program (independent of having been made by running the Program).
|
||||
Whether that is true depends on what the Program does.
|
||||
|
||||
1. You may copy and distribute verbatim copies of the Program's
|
||||
source code as you receive it, in any medium, provided that you
|
||||
conspicuously and appropriately publish on each copy an appropriate
|
||||
copyright notice and disclaimer of warranty; keep intact all the
|
||||
notices that refer to this License and to the absence of any warranty;
|
||||
and give any other recipients of the Program a copy of this License
|
||||
along with the Program.
|
||||
|
||||
You may charge a fee for the physical act of transferring a copy, and
|
||||
you may at your option offer warranty protection in exchange for a fee.
|
||||
|
||||
2. You may modify your copy or copies of the Program or any portion
|
||||
of it, thus forming a work based on the Program, and copy and
|
||||
distribute such modifications or work under the terms of Section 1
|
||||
above, provided that you also meet all of these conditions:
|
||||
|
||||
a) You must cause the modified files to carry prominent notices
|
||||
stating that you changed the files and the date of any change.
|
||||
|
||||
b) You must cause any work that you distribute or publish, that in
|
||||
whole or in part contains or is derived from the Program or any
|
||||
part thereof, to be licensed as a whole at no charge to all third
|
||||
parties under the terms of this License.
|
||||
|
||||
c) If the modified program normally reads commands interactively
|
||||
when run, you must cause it, when started running for such
|
||||
interactive use in the most ordinary way, to print or display an
|
||||
announcement including an appropriate copyright notice and a
|
||||
notice that there is no warranty (or else, saying that you provide
|
||||
a warranty) and that users may redistribute the program under
|
||||
these conditions, and telling the user how to view a copy of this
|
||||
License. (Exception: if the Program itself is interactive but
|
||||
does not normally print such an announcement, your work based on
|
||||
the Program is not required to print an announcement.)
|
||||
|
||||
These requirements apply to the modified work as a whole. If
|
||||
identifiable sections of that work are not derived from the Program,
|
||||
and can be reasonably considered independent and separate works in
|
||||
themselves, then this License, and its terms, do not apply to those
|
||||
sections when you distribute them as separate works. But when you
|
||||
distribute the same sections as part of a whole which is a work based
|
||||
on the Program, the distribution of the whole must be on the terms of
|
||||
this License, whose permissions for other licensees extend to the
|
||||
entire whole, and thus to each and every part regardless of who wrote it.
|
||||
|
||||
Thus, it is not the intent of this section to claim rights or contest
|
||||
your rights to work written entirely by you; rather, the intent is to
|
||||
exercise the right to control the distribution of derivative or
|
||||
collective works based on the Program.
|
||||
|
||||
In addition, mere aggregation of another work not based on the Program
|
||||
with the Program (or with a work based on the Program) on a volume of
|
||||
a storage or distribution medium does not bring the other work under
|
||||
the scope of this License.
|
||||
|
||||
3. You may copy and distribute the Program (or a work based on it,
|
||||
under Section 2) in object code or executable form under the terms of
|
||||
Sections 1 and 2 above provided that you also do one of the following:
|
||||
|
||||
a) Accompany it with the complete corresponding machine-readable
|
||||
source code, which must be distributed under the terms of Sections
|
||||
1 and 2 above on a medium customarily used for software interchange; or,
|
||||
|
||||
b) Accompany it with a written offer, valid for at least three
|
||||
years, to give any third party, for a charge no more than your
|
||||
cost of physically performing source distribution, a complete
|
||||
machine-readable copy of the corresponding source code, to be
|
||||
distributed under the terms of Sections 1 and 2 above on a medium
|
||||
customarily used for software interchange; or,
|
||||
|
||||
c) Accompany it with the information you received as to the offer
|
||||
to distribute corresponding source code. (This alternative is
|
||||
allowed only for noncommercial distribution and only if you
|
||||
received the program in object code or executable form with such
|
||||
an offer, in accord with Subsection b above.)
|
||||
|
||||
The source code for a work means the preferred form of the work for
|
||||
making modifications to it. For an executable work, complete source
|
||||
code means all the source code for all modules it contains, plus any
|
||||
associated interface definition files, plus the scripts used to
|
||||
control compilation and installation of the executable. However, as a
|
||||
special exception, the source code distributed need not include
|
||||
anything that is normally distributed (in either source or binary
|
||||
form) with the major components (compiler, kernel, and so on) of the
|
||||
operating system on which the executable runs, unless that component
|
||||
itself accompanies the executable.
|
||||
|
||||
If distribution of executable or object code is made by offering
|
||||
access to copy from a designated place, then offering equivalent
|
||||
access to copy the source code from the same place counts as
|
||||
distribution of the source code, even though third parties are not
|
||||
compelled to copy the source along with the object code.
|
||||
|
||||
4. You may not copy, modify, sublicense, or distribute the Program
|
||||
except as expressly provided under this License. Any attempt
|
||||
otherwise to copy, modify, sublicense or distribute the Program is
|
||||
void, and will automatically terminate your rights under this License.
|
||||
However, parties who have received copies, or rights, from you under
|
||||
this License will not have their licenses terminated so long as such
|
||||
parties remain in full compliance.
|
||||
|
||||
5. You are not required to accept this License, since you have not
|
||||
signed it. However, nothing else grants you permission to modify or
|
||||
distribute the Program or its derivative works. These actions are
|
||||
prohibited by law if you do not accept this License. Therefore, by
|
||||
modifying or distributing the Program (or any work based on the
|
||||
Program), you indicate your acceptance of this License to do so, and
|
||||
all its terms and conditions for copying, distributing or modifying
|
||||
the Program or works based on it.
|
||||
|
||||
6. Each time you redistribute the Program (or any work based on the
|
||||
Program), the recipient automatically receives a license from the
|
||||
original licensor to copy, distribute or modify the Program subject to
|
||||
these terms and conditions. You may not impose any further
|
||||
restrictions on the recipients' exercise of the rights granted herein.
|
||||
You are not responsible for enforcing compliance by third parties to
|
||||
this License.
|
||||
|
||||
7. If, as a consequence of a court judgment or allegation of patent
|
||||
infringement or for any other reason (not limited to patent issues),
|
||||
conditions are imposed on you (whether by court order, agreement or
|
||||
otherwise) that contradict the conditions of this License, they do not
|
||||
excuse you from the conditions of this License. If you cannot
|
||||
distribute so as to satisfy simultaneously your obligations under this
|
||||
License and any other pertinent obligations, then as a consequence you
|
||||
may not distribute the Program at all. For example, if a patent
|
||||
license would not permit royalty-free redistribution of the Program by
|
||||
all those who receive copies directly or indirectly through you, then
|
||||
the only way you could satisfy both it and this License would be to
|
||||
refrain entirely from distribution of the Program.
|
||||
|
||||
If any portion of this section is held invalid or unenforceable under
|
||||
any particular circumstance, the balance of the section is intended to
|
||||
apply and the section as a whole is intended to apply in other
|
||||
circumstances.
|
||||
|
||||
It is not the purpose of this section to induce you to infringe any
|
||||
patents or other property right claims or to contest validity of any
|
||||
such claims; this section has the sole purpose of protecting the
|
||||
integrity of the free software distribution system, which is
|
||||
implemented by public license practices. Many people have made
|
||||
generous contributions to the wide range of software distributed
|
||||
through that system in reliance on consistent application of that
|
||||
system; it is up to the author/donor to decide if he or she is willing
|
||||
to distribute software through any other system and a licensee cannot
|
||||
impose that choice.
|
||||
|
||||
This section is intended to make thoroughly clear what is believed to
|
||||
be a consequence of the rest of this License.
|
||||
|
||||
8. If the distribution and/or use of the Program is restricted in
|
||||
certain countries either by patents or by copyrighted interfaces, the
|
||||
original copyright holder who places the Program under this License
|
||||
may add an explicit geographical distribution limitation excluding
|
||||
those countries, so that distribution is permitted only in or among
|
||||
countries not thus excluded. In such case, this License incorporates
|
||||
the limitation as if written in the body of this License.
|
||||
|
||||
9. The Free Software Foundation may publish revised and/or new versions
|
||||
of the General Public License from time to time. Such new versions will
|
||||
be similar in spirit to the present version, but may differ in detail to
|
||||
address new problems or concerns.
|
||||
|
||||
Each version is given a distinguishing version number. If the Program
|
||||
specifies a version number of this License which applies to it and "any
|
||||
later version", you have the option of following the terms and conditions
|
||||
either of that version or of any later version published by the Free
|
||||
Software Foundation. If the Program does not specify a version number of
|
||||
this License, you may choose any version ever published by the Free Software
|
||||
Foundation.
|
||||
|
||||
10. If you wish to incorporate parts of the Program into other free
|
||||
programs whose distribution conditions are different, write to the author
|
||||
to ask for permission. For software which is copyrighted by the Free
|
||||
Software Foundation, write to the Free Software Foundation; we sometimes
|
||||
make exceptions for this. Our decision will be guided by the two goals
|
||||
of preserving the free status of all derivatives of our free software and
|
||||
of promoting the sharing and reuse of software generally.
|
||||
|
||||
NO WARRANTY
|
||||
|
||||
11. BECAUSE THE PROGRAM IS LICENSED FREE OF CHARGE, THERE IS NO WARRANTY
|
||||
FOR THE PROGRAM, TO THE EXTENT PERMITTED BY APPLICABLE LAW. EXCEPT WHEN
|
||||
OTHERWISE STATED IN WRITING THE COPYRIGHT HOLDERS AND/OR OTHER PARTIES
|
||||
PROVIDE THE PROGRAM "AS IS" WITHOUT WARRANTY OF ANY KIND, EITHER EXPRESSED
|
||||
OR IMPLIED, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
|
||||
MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE. THE ENTIRE RISK AS
|
||||
TO THE QUALITY AND PERFORMANCE OF THE PROGRAM IS WITH YOU. SHOULD THE
|
||||
PROGRAM PROVE DEFECTIVE, YOU ASSUME THE COST OF ALL NECESSARY SERVICING,
|
||||
REPAIR OR CORRECTION.
|
||||
|
||||
12. IN NO EVENT UNLESS REQUIRED BY APPLICABLE LAW OR AGREED TO IN WRITING
|
||||
WILL ANY COPYRIGHT HOLDER, OR ANY OTHER PARTY WHO MAY MODIFY AND/OR
|
||||
REDISTRIBUTE THE PROGRAM AS PERMITTED ABOVE, BE LIABLE TO YOU FOR DAMAGES,
|
||||
INCLUDING ANY GENERAL, SPECIAL, INCIDENTAL OR CONSEQUENTIAL DAMAGES ARISING
|
||||
OUT OF THE USE OR INABILITY TO USE THE PROGRAM (INCLUDING BUT NOT LIMITED
|
||||
TO LOSS OF DATA OR DATA BEING RENDERED INACCURATE OR LOSSES SUSTAINED BY
|
||||
YOU OR THIRD PARTIES OR A FAILURE OF THE PROGRAM TO OPERATE WITH ANY OTHER
|
||||
PROGRAMS), EVEN IF SUCH HOLDER OR OTHER PARTY HAS BEEN ADVISED OF THE
|
||||
POSSIBILITY OF SUCH DAMAGES.
|
||||
|
||||
END OF TERMS AND CONDITIONS
|
||||
|
||||
How to Apply These Terms to Your New Programs
|
||||
|
||||
If you develop a new program, and you want it to be of the greatest
|
||||
possible use to the public, the best way to achieve this is to make it
|
||||
free software which everyone can redistribute and change under these terms.
|
||||
|
||||
To do so, attach the following notices to the program. It is safest
|
||||
to attach them to the start of each source file to most effectively
|
||||
convey the exclusion of warranty; and each file should have at least
|
||||
the "copyright" line and a pointer to where the full notice is found.
|
||||
|
||||
<one line to give the program's name and a brief idea of what it does.>
|
||||
Copyright (C) <year> <name of author>
|
||||
|
||||
This program is free software; you can redistribute it and/or modify
|
||||
it under the terms of the GNU General Public License as published by
|
||||
the Free Software Foundation; either version 2 of the License, or
|
||||
(at your option) any later version.
|
||||
|
||||
This program is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU General Public License along
|
||||
with this program; if not, write to the Free Software Foundation, Inc.,
|
||||
51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
||||
|
||||
Also add information on how to contact you by electronic and paper mail.
|
||||
|
||||
If the program is interactive, make it output a short notice like this
|
||||
when it starts in an interactive mode:
|
||||
|
||||
Gnomovision version 69, Copyright (C) year name of author
|
||||
Gnomovision comes with ABSOLUTELY NO WARRANTY; for details type `show w'.
|
||||
This is free software, and you are welcome to redistribute it
|
||||
under certain conditions; type `show c' for details.
|
||||
|
||||
The hypothetical commands `show w' and `show c' should show the appropriate
|
||||
parts of the General Public License. Of course, the commands you use may
|
||||
be called something other than `show w' and `show c'; they could even be
|
||||
mouse-clicks or menu items--whatever suits your program.
|
||||
|
||||
You should also get your employer (if you work as a programmer) or your
|
||||
school, if any, to sign a "copyright disclaimer" for the program, if
|
||||
necessary. Here is a sample; alter the names:
|
||||
|
||||
Yoyodyne, Inc., hereby disclaims all copyright interest in the program
|
||||
`Gnomovision' (which makes passes at compilers) written by James Hacker.
|
||||
|
||||
<signature of Ty Coon>, 1 April 1989
|
||||
Ty Coon, President of Vice
|
||||
|
||||
This General Public License does not permit incorporating your program into
|
||||
proprietary programs. If your program is a subroutine library, you may
|
||||
consider it more useful to permit linking proprietary applications with the
|
||||
library. If this is what you want to do, use the GNU Lesser General
|
||||
Public License instead of this License.
|
21
LVC_.s
Normal file
21
LVC_.s
Normal file
@ -0,0 +1,21 @@
|
||||
.arm.little
|
||||
|
||||
.open "unpacked/LVC_.bin",0
|
||||
|
||||
; Patches configuration
|
||||
ADD_UIC_CONFIG_REGION equ 1
|
||||
|
||||
.if ADD_UIC_CONFIG_REGION
|
||||
.notice "Adding region to UIC config"
|
||||
|
||||
; Patch jumptable to make ID 1 valid
|
||||
.org 0x0003c9c4
|
||||
.d8 0x30
|
||||
|
||||
; Patch config table at offset 1 to insert region
|
||||
.org 0x000b28d0
|
||||
.d32 0x3 ; size
|
||||
.d32 0x103 ; eeprom offset
|
||||
.endif
|
||||
|
||||
.close
|
31
Makefile
Normal file
31
Makefile
Normal file
@ -0,0 +1,31 @@
|
||||
DRC_FW_FILE := drc_fw.bin
|
||||
LVC_SHA256 := "d83be494af0bb35cc4bd3340a3da3adb0823bcf111d6e2fd72092a0532d8df32 unpacked/LVC_.bin"
|
||||
|
||||
.PHONY: unpack patch pack clean
|
||||
|
||||
all: pack
|
||||
@echo "All done!"
|
||||
|
||||
unpack:
|
||||
@echo "Unpacking firmware..."
|
||||
@python3 fwpack.py unpack $(DRC_FW_FILE) unpacked
|
||||
|
||||
@echo "Checking LVC_.bin..."
|
||||
@echo $(LVC_SHA256) | sha256sum -c -
|
||||
|
||||
patch: unpack
|
||||
@echo "Patching LVC_.bin"
|
||||
@armips LVC_.s
|
||||
|
||||
@echo "Patching versions..."
|
||||
# patch version to 254.0.0 (blob_header.bin is big endian, VER_.bin is little endian)
|
||||
@printf '\xfe\x00\x00\x00' | dd of=unpacked/blob_header.bin bs=1 seek=0 conv=notrunc
|
||||
@printf '\x00\x00\x00\xfe' | dd of=unpacked/VER_.bin bs=1 seek=0 conv=notrunc
|
||||
|
||||
pack: patch
|
||||
@echo "Packing firmware..."
|
||||
@python3 fwpack.py pack unpacked patched_$(DRC_FW_FILE)
|
||||
|
||||
clean:
|
||||
@echo "Clean..."
|
||||
@rm -rf unpacked patched_$(DRC_FW_FILE)
|
16
README.md
Normal file
16
README.md
Normal file
@ -0,0 +1,16 @@
|
||||
# drc-fw-patches
|
||||
Patches for the Wii U gamepad firmware.
|
||||
|
||||
## Disclaimer
|
||||
Modifying the DRC firmware can cause permanent damage.
|
||||
No one but yourself is responsible for any sort of damage resulting from using these patches.
|
||||
|
||||
## Usage
|
||||
### Requirements
|
||||
- make
|
||||
- [armips](https://github.com/Kingcom/armips)
|
||||
- python3
|
||||
- python-construct
|
||||
|
||||
Place the original `drc_fw.bin` version `0x190c0117` from your MLC or NUS into this directory.
|
||||
Running `make` will build a `patched_drc_fw.bin` which can be flashed to the gamepad.
|
172
fwpack.py
Normal file
172
fwpack.py
Normal file
@ -0,0 +1,172 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
fwpack - Pack/Unpack DRC/DRH firmware files
|
||||
Created in 2024 by GaryOderNichts
|
||||
<https://github.com/GaryOderNichts/drc-fw-patches>
|
||||
|
||||
Credits to drxtool for the firmware header logic and extracted files structure.
|
||||
"""
|
||||
|
||||
import sys, os
|
||||
import binascii
|
||||
import construct
|
||||
|
||||
class FirmwareType:
|
||||
FIRMWARE_TYPE_DRC = 0x01010000
|
||||
FIRMWARE_TYPE_DRH = 0x00010000
|
||||
|
||||
BlobHeader = construct.Struct(
|
||||
"imageVersion" / construct.Int32ub,
|
||||
"blockSize" / construct.Int32ub,
|
||||
"sequencePerSession" / construct.Int32ub,
|
||||
"imageSize" / construct.Int32ub,
|
||||
)
|
||||
assert(BlobHeader.sizeof() == 0x10)
|
||||
|
||||
FirmwareHeader = construct.Struct(
|
||||
"type" / construct.Int32ul,
|
||||
"superCRCs" / construct.Array(4, construct.Int32ul),
|
||||
construct.Padding(0xFE8),
|
||||
"headerCRC" / construct.Int32ul,
|
||||
"subCRCs" / construct.Array(0x1000, construct.Int32ul),
|
||||
)
|
||||
assert(FirmwareHeader.sizeof() == 0x5000)
|
||||
|
||||
FirmwareSection = construct.Struct(
|
||||
"offset" / construct.Int32ul,
|
||||
"size" / construct.Int32ul,
|
||||
"name" / construct.PaddedString(4, "ascii"),
|
||||
"version" / construct.Int32ul,
|
||||
)
|
||||
assert(FirmwareSection.sizeof() == 0x10)
|
||||
|
||||
FirmwareFile = construct.Struct(
|
||||
"blobHeader" / BlobHeader,
|
||||
"firmwareHeader" / FirmwareHeader,
|
||||
"firmwareData" / construct.Bytes(construct.this.blobHeader.imageSize - FirmwareHeader.sizeof()),
|
||||
)
|
||||
|
||||
# Thanks to drxtool for the crctable logic
|
||||
def verify_firmware_header(fw) -> bool:
|
||||
# Verify header CRC
|
||||
header_crc = binascii.crc32(FirmwareHeader.build(fw.firmwareHeader)[0:0xFFC])
|
||||
if header_crc != fw.firmwareHeader.headerCRC:
|
||||
return False
|
||||
|
||||
# Verify super crcs
|
||||
subcrc_data = construct.Array(0x1000, construct.Int32ul).build(fw.firmwareHeader.subCRCs)
|
||||
for i in range(4):
|
||||
super_crc = binascii.crc32(subcrc_data[i*0x1000:i*0x1000+0x1000])
|
||||
if super_crc != fw.firmwareHeader.superCRCs[i]:
|
||||
return False
|
||||
|
||||
# Verify sub crcs
|
||||
for i in range(len(fw.firmwareData) // 0x1000 + 1):
|
||||
offset = i * 0x1000
|
||||
length = 0x1000
|
||||
if len(fw.firmwareData) - offset < length:
|
||||
length = len(fw.firmwareData) - offset
|
||||
|
||||
sub_crc = binascii.crc32(fw.firmwareData[offset:offset + length])
|
||||
if sub_crc != fw.firmwareHeader.subCRCs[i]:
|
||||
return False
|
||||
|
||||
return True
|
||||
|
||||
def build_firmware_header(blob_type, firmware_data) -> dict:
|
||||
# Calculate CRC for every 0x1000 bytes of firmware data
|
||||
sub_crcs = [0] * 0x1000
|
||||
for i in range(len(firmware_data) // 0x1000 + 1):
|
||||
offset = i * 0x1000
|
||||
length = 0x1000
|
||||
if len(firmware_data) - offset < length:
|
||||
length = len(firmware_data) - offset
|
||||
|
||||
sub_crcs[i] = binascii.crc32(firmware_data[offset:offset + length])
|
||||
|
||||
# Calculate the super CRCs
|
||||
super_crcs = [0] * 4
|
||||
subcrc_data = construct.Array(0x1000, construct.Int32ul).build(sub_crcs)
|
||||
for i in range(4):
|
||||
super_crcs[i] = binascii.crc32(subcrc_data[i*0x1000:i*0x1000+0x1000])
|
||||
|
||||
firmware_header = dict(type=blob_type, superCRCs=super_crcs, headerCRC=0, subCRCs=sub_crcs)
|
||||
|
||||
# Calculate the header CRC
|
||||
firmware_header["headerCRC"] = binascii.crc32(FirmwareHeader.build(firmware_header)[0:0xFFC])
|
||||
|
||||
return firmware_header
|
||||
|
||||
def unpack_firmware(source_file, dest_dir):
|
||||
fw = FirmwareFile.parse_file(source_file)
|
||||
if not verify_firmware_header(fw):
|
||||
print("Firmware header verification failed")
|
||||
sys.exit(1)
|
||||
|
||||
if fw.firmwareHeader.type == FirmwareType.FIRMWARE_TYPE_DRC:
|
||||
print(f"DRC firmware version 0x{fw.blobHeader.imageVersion:08x}")
|
||||
elif fw.firmwareHeader.type == FirmwareType.FIRMWARE_TYPE_DRH:
|
||||
print(f"DRH firmware version 0x{fw.blobHeader.imageVersion:08x}")
|
||||
else:
|
||||
print(f"Unsupported firmware type 0x{fw.firmwareHeader.type:08x}")
|
||||
sys.exit(1)
|
||||
|
||||
if not os.path.isdir(dest_dir):
|
||||
os.mkdir(dest_dir)
|
||||
|
||||
# Write blob header and type
|
||||
BlobHeader.build_file(fw.blobHeader, os.path.join(dest_dir, "blob_header.bin"))
|
||||
construct.Int32ul.build_file(fw.firmwareHeader.type, os.path.join(dest_dir, "blob_type.bin"))
|
||||
|
||||
# Assume first part of the data is the index
|
||||
index = FirmwareSection.parse(fw.firmwareData)
|
||||
|
||||
# Parse sections
|
||||
sections = construct.Array(index.size // FirmwareSection.sizeof(), FirmwareSection).parse(fw.firmwareData)
|
||||
for s in sections:
|
||||
print(f"Saving {s.name} version 0x{s.version:08x} offset 0x{s.offset} size 0x{s.size}")
|
||||
|
||||
# write section to file
|
||||
with open(os.path.join(dest_dir, s.name + ".bin"), "wb") as f:
|
||||
f.write(fw.firmwareData[s.offset:s.offset + s.size])
|
||||
|
||||
def pack_firmware(source_dir, dest_file):
|
||||
# Read blob header and type
|
||||
blob_header = BlobHeader.parse_file(os.path.join(source_dir, "blob_header.bin"))
|
||||
blob_type = construct.Int32ul.parse_file(os.path.join(source_dir, "blob_type.bin"))
|
||||
|
||||
# Parse sections from INDX
|
||||
firmware_data = b""
|
||||
sections = construct.GreedyRange(FirmwareSection).parse_file(os.path.join(source_dir, "INDX.bin"))
|
||||
for s in sections:
|
||||
print(f"Storing {s.name}...")
|
||||
|
||||
# read section data
|
||||
section_data = b""
|
||||
with open(os.path.join(source_dir, s.name + ".bin"), "rb") as f:
|
||||
section_data = f.read()
|
||||
|
||||
# update section
|
||||
s.size = len(section_data)
|
||||
s.offset = len(firmware_data)
|
||||
firmware_data += section_data
|
||||
|
||||
# Update total image size
|
||||
blob_header.imageSize = FirmwareHeader.sizeof() + len(firmware_data)
|
||||
|
||||
# Build the final image size
|
||||
FirmwareFile.build_file(dict(blobHeader=blob_header, firmwareHeader=build_firmware_header(blob_type, firmware_data), firmwareData=firmware_data), dest_file)
|
||||
|
||||
if len(sys.argv) != 4:
|
||||
print(f"Usage:\n{sys.argv[0]} <unpack/pack> <source> <destination>")
|
||||
sys.exit(0)
|
||||
|
||||
if sys.argv[1] == "unpack":
|
||||
print(f"Unpacking {sys.argv[2]}...")
|
||||
unpack_firmware(sys.argv[2], sys.argv[3])
|
||||
elif sys.argv[1] == "pack":
|
||||
print(f"Packing {sys.argv[3]}...")
|
||||
pack_firmware(sys.argv[2], sys.argv[3])
|
||||
else:
|
||||
print(f"Unknown mode \"{sys.argv[1]}\"")
|
||||
sys.exit(1)
|
Loading…
Reference in New Issue
Block a user