2019-12-09 11:57:37 +00:00
/*
2021-10-04 20:59:10 +01:00
* Copyright ( c ) Atmosphère - NX
2019-12-09 11:57:37 +00:00
*
* This program is free software ; you can redistribute it and / or modify it
* under the terms and conditions of the GNU General Public License ,
* version 2 , as published by the Free Software Foundation .
*
* This program is distributed in the hope 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 , see < http : //www.gnu.org/licenses/>.
*/
2020-05-11 23:02:10 +01:00
# include <stratosphere.hpp>
2019-12-09 11:57:37 +00:00
# include "updater_bis_management.hpp"
namespace ams : : updater {
2020-11-11 00:53:19 +00:00
namespace {
/* Recognize special public key (https://gist.github.com/SciresM/16b63ac1d80494522bdba2c57995257c). */
/* P = 19 */
/* Q = 1696986749729493925354392349339746171297507422986462747526968361144447230710192316397327889522451749459854070558277878297255552508603806832852079596337539247651161831569525505882103311631577368514276343192042634740927726070847704397913856975832811679847928433261678072951551065705680482548543833651752439700272736498378724153330763357721354498194000536297732323628263256733931353143625854828275237159155585342783077681713929284136658773985266864804093157854331138230313706015557050002740810464618031715670281442110238274404626065924786185264268216336867948322976979393032640085259926883014490947373494538254895109731 */
/* N = 0xFF696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696969696959 */
/* E = 0x10001 */
/* D = 6512128715229088976470211610075969347035078304643231077895577077900787352712063823560162578441773733649014439616165727455431015055675770987914713980812453585413988983206576233689754710500864883529402371292948326392791238474661859182717295176679567362482790015587820446999760239570255254879359445627372805817473978644067558931078225451477635089763009580492462097185005355990612929951162366081631888011031830742459571000203341001926135389508196521518687349554188686396554248868403128728646457247197407637887195043486221295751496667162366700477934591694110831002874992896076061627516220934290742867397720103040314639313 */
constexpr const u8 CustomPublicKey [ 0x100 ] = {
0x59 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 ,
0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 ,
0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 ,
0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 ,
0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 ,
0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 ,
0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 ,
0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 ,
0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 ,
0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 ,
0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 ,
0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 ,
0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 ,
0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 ,
0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 ,
0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0x69 , 0xFF ,
} ;
}
2019-12-09 11:57:37 +00:00
Result BisAccessor : : Initialize ( ) {
2020-03-08 08:06:23 +00:00
return fs : : OpenBisPartition ( std : : addressof ( this - > storage ) , this - > partition_id ) ;
2019-12-09 11:57:37 +00:00
}
void BisAccessor : : Finalize ( ) {
2020-03-08 08:06:23 +00:00
/* ... */
2019-12-09 11:57:37 +00:00
}
Result BisAccessor : : Read ( void * dst , size_t size , u64 offset ) {
2020-02-23 07:05:14 +00:00
AMS_ABORT_UNLESS ( ( offset % SectorAlignment ) = = 0 ) ;
2020-03-08 08:06:23 +00:00
return this - > storage - > Read ( static_cast < u32 > ( offset ) , dst , size ) ;
2019-12-09 11:57:37 +00:00
}
Result BisAccessor : : Write ( u64 offset , const void * src , size_t size ) {
2020-02-23 07:05:14 +00:00
AMS_ABORT_UNLESS ( ( offset % SectorAlignment ) = = 0 ) ;
2020-03-08 08:06:23 +00:00
return this - > storage - > Write ( static_cast < u32 > ( offset ) , src , size ) ;
2019-12-09 11:57:37 +00:00
}
Result BisAccessor : : Write ( u64 offset , size_t size , const char * bip_path , void * work_buffer , size_t work_buffer_size ) {
2020-02-23 07:05:14 +00:00
AMS_ABORT_UNLESS ( ( offset % SectorAlignment ) = = 0 ) ;
AMS_ABORT_UNLESS ( ( work_buffer_size % SectorAlignment ) = = 0 ) ;
2019-12-09 11:57:37 +00:00
2020-03-08 08:06:23 +00:00
fs : : FileHandle file ;
R_TRY_CATCH ( fs : : OpenFile ( std : : addressof ( file ) , bip_path , fs : : OpenMode_Read ) ) {
2021-10-09 18:36:21 +01:00
R_CONVERT ( fs : : ResultPathNotFound , updater : : ResultInvalidBootImagePackage ( ) )
2020-03-08 08:06:23 +00:00
} R_END_TRY_CATCH ;
ON_SCOPE_EXIT { fs : : CloseFile ( file ) ; } ;
2019-12-09 11:57:37 +00:00
size_t written = 0 ;
while ( true ) {
std : : memset ( work_buffer , 0 , work_buffer_size ) ;
2020-03-08 08:06:23 +00:00
size_t read_size ;
R_TRY ( fs : : ReadFile ( std : : addressof ( read_size ) , file , written , work_buffer , work_buffer_size , fs : : ReadOption ( ) ) ) ;
2020-02-23 07:05:14 +00:00
AMS_ABORT_UNLESS ( written + read_size < = size ) ;
2019-12-09 11:57:37 +00:00
size_t aligned_size = ( ( read_size + SectorAlignment - 1 ) / SectorAlignment ) * SectorAlignment ;
R_TRY ( this - > Write ( offset + written , work_buffer , aligned_size ) ) ;
written + = read_size ;
if ( read_size ! = work_buffer_size ) {
break ;
}
}
return ResultSuccess ( ) ;
}
Result BisAccessor : : Clear ( u64 offset , u64 size , void * work_buffer , size_t work_buffer_size ) {
2020-02-23 07:05:14 +00:00
AMS_ABORT_UNLESS ( ( offset % SectorAlignment ) = = 0 ) ;
AMS_ABORT_UNLESS ( ( work_buffer_size % SectorAlignment ) = = 0 ) ;
2019-12-09 11:57:37 +00:00
std : : memset ( work_buffer , 0 , work_buffer_size ) ;
size_t written = 0 ;
while ( written < size ) {
size_t cur_write_size = std : : min ( work_buffer_size , size - written ) ;
R_TRY ( this - > Write ( offset + written , work_buffer , cur_write_size ) ) ;
written + = cur_write_size ;
}
return ResultSuccess ( ) ;
}
Result BisAccessor : : GetHash ( void * dst , u64 offset , u64 size , u64 hash_size , void * work_buffer , size_t work_buffer_size ) {
2020-02-23 07:05:14 +00:00
AMS_ABORT_UNLESS ( ( offset % SectorAlignment ) = = 0 ) ;
AMS_ABORT_UNLESS ( ( work_buffer_size % SectorAlignment ) = = 0 ) ;
2019-12-09 11:57:37 +00:00
2020-03-08 08:06:23 +00:00
crypto : : Sha256Generator generator ;
generator . Initialize ( ) ;
2019-12-09 11:57:37 +00:00
size_t total_read = 0 ;
while ( total_read < hash_size ) {
size_t cur_read_size = std : : min ( work_buffer_size , size - total_read ) ;
size_t cur_update_size = std : : min ( cur_read_size , hash_size - total_read ) ;
R_TRY ( this - > Read ( work_buffer , cur_read_size , offset + total_read ) ) ;
2020-03-08 08:06:23 +00:00
generator . Update ( work_buffer , cur_update_size ) ;
2019-12-09 11:57:37 +00:00
total_read + = cur_read_size ;
}
2020-03-08 08:06:23 +00:00
generator . GetHash ( dst , hash_size ) ;
2019-12-09 11:57:37 +00:00
return ResultSuccess ( ) ;
}
size_t Boot0Accessor : : GetBootloaderVersion ( void * bct ) {
u32 version = * reinterpret_cast < u32 * > ( reinterpret_cast < uintptr_t > ( bct ) + BctVersionOffset ) ;
2020-02-23 07:05:14 +00:00
AMS_ABORT_UNLESS ( version < = BctVersionMax ) ;
2019-12-09 11:57:37 +00:00
return static_cast < size_t > ( version ) ;
}
size_t Boot0Accessor : : GetEksIndex ( size_t bootloader_version ) {
2020-02-23 07:05:14 +00:00
AMS_ABORT_UNLESS ( bootloader_version < = BctVersionMax ) ;
2019-12-09 11:57:37 +00:00
return ( bootloader_version > 0 ) ? bootloader_version - 1 : 0 ;
}
void Boot0Accessor : : CopyEks ( void * dst_bct , const void * src_eks , size_t eks_index ) {
std : : memcpy ( reinterpret_cast < u8 * > ( dst_bct ) + BctEksOffset , reinterpret_cast < const u8 * > ( src_eks ) + eks_index * EksEntrySize , EksBlobSize ) ;
}
Result Boot0Accessor : : UpdateEks ( void * dst_bct , void * eks_work_buffer ) {
size_t read_size ;
R_TRY ( this - > Read ( & read_size , eks_work_buffer , EksSize , Boot0Partition : : Eks ) ) ;
return this - > UpdateEksManually ( dst_bct , eks_work_buffer ) ;
}
Result Boot0Accessor : : UpdateEksManually ( void * dst_bct , const void * src_eks ) {
this - > CopyEks ( dst_bct , src_eks , GetEksIndex ( GetBootloaderVersion ( dst_bct ) ) ) ;
return ResultSuccess ( ) ;
}
Result Boot0Accessor : : PreserveAutoRcm ( void * dst_bct , void * work_buffer , Boot0Partition which ) {
std : : memset ( work_buffer , 0 , BctSize ) ;
size_t read_size ;
R_TRY ( this - > Read ( & read_size , work_buffer , BctSize , which ) ) ;
2020-11-11 00:53:19 +00:00
/* NOTE: AutoRcm is only viable on Erista, so hardcode erista offsets. */
void * dst_pubk = reinterpret_cast < void * > ( reinterpret_cast < uintptr_t > ( dst_bct ) + BctPubkOffsetErista ) ;
void * src_pubk = reinterpret_cast < void * > ( reinterpret_cast < uintptr_t > ( work_buffer ) + BctPubkOffsetErista ) ;
2019-12-09 11:57:37 +00:00
std : : memcpy ( dst_pubk , src_pubk , BctPubkSize ) ;
2020-03-08 08:06:23 +00:00
2019-12-09 11:57:37 +00:00
return ResultSuccess ( ) ;
}
2020-11-11 00:53:19 +00:00
Result Boot0Accessor : : DetectCustomPublicKey ( bool * out , void * work_buffer , BootImageUpdateType boot_image_update_type ) {
std : : memset ( work_buffer , 0 , BctSize ) ;
const size_t pubk_offset = GetBctPubkOffset ( boot_image_update_type ) ;
size_t read_size ;
R_TRY ( this - > Read ( & read_size , work_buffer , BctSize , Boot0Partition : : BctNormalMain ) ) ;
if ( std : : memcmp ( reinterpret_cast < void * > ( reinterpret_cast < uintptr_t > ( work_buffer ) + pubk_offset ) , CustomPublicKey , sizeof ( CustomPublicKey ) ) ! = 0 ) {
* out = false ;
return ResultSuccess ( ) ;
}
R_TRY ( this - > Read ( & read_size , work_buffer , BctSize , Boot0Partition : : BctSafeMain ) ) ;
if ( std : : memcmp ( reinterpret_cast < void * > ( reinterpret_cast < uintptr_t > ( work_buffer ) + pubk_offset ) , CustomPublicKey , sizeof ( CustomPublicKey ) ) ! = 0 ) {
* out = false ;
return ResultSuccess ( ) ;
}
* out = true ;
return ResultSuccess ( ) ;
}
2019-12-09 11:57:37 +00:00
}