Decoding: deal with some SPOT6 images that have tiles with a single tile-part with TPsot == 0 and TNsot == 0, and with missing EOC

This commit is contained in:
Even Rouault 2020-11-30 17:36:27 +01:00
parent a2b2980a4b
commit 1aa3c60859
No known key found for this signature in database
GPG Key ID: 33EBBFC47B3DD87D
3 changed files with 32 additions and 15 deletions

View File

@ -9382,6 +9382,7 @@ OPJ_BOOL opj_j2k_read_tile_header(opj_j2k_t * p_j2k,
OPJ_UINT32 l_marker_size; OPJ_UINT32 l_marker_size;
const opj_dec_memory_marker_handler_t * l_marker_handler = 00; const opj_dec_memory_marker_handler_t * l_marker_handler = 00;
opj_tcp_t * l_tcp = NULL; opj_tcp_t * l_tcp = NULL;
const OPJ_UINT32 l_nb_tiles = p_j2k->m_cp.tw * p_j2k->m_cp.th;
/* preconditions */ /* preconditions */
assert(p_stream != 00); assert(p_stream != 00);
@ -9557,7 +9558,6 @@ OPJ_BOOL opj_j2k_read_tile_header(opj_j2k_t * p_j2k,
return OPJ_FALSE; return OPJ_FALSE;
} }
if (l_correction_needed) { if (l_correction_needed) {
OPJ_UINT32 l_nb_tiles = p_j2k->m_cp.tw * p_j2k->m_cp.th;
OPJ_UINT32 l_tile_no; OPJ_UINT32 l_tile_no;
p_j2k->m_specific_param.m_decoder.m_can_decode = 0; p_j2k->m_specific_param.m_decoder.m_can_decode = 0;
@ -9572,27 +9572,42 @@ OPJ_BOOL opj_j2k_read_tile_header(opj_j2k_t * p_j2k,
"Non conformant codestream TPsot==TNsot.\n"); "Non conformant codestream TPsot==TNsot.\n");
} }
} }
if (! p_j2k->m_specific_param.m_decoder.m_can_decode) {
/* Try to read 2 bytes (the next marker ID) from stream and copy them into the buffer */
if (opj_stream_read_data(p_stream,
p_j2k->m_specific_param.m_decoder.m_header_data, 2, p_manager) != 2) {
opj_event_msg(p_manager, EVT_ERROR, "Stream too short\n");
return OPJ_FALSE;
}
/* Read 2 bytes from buffer as the new marker ID */
opj_read_bytes(p_j2k->m_specific_param.m_decoder.m_header_data,
&l_current_marker, 2);
}
} else { } else {
/* Indicate we will try to read a new tile-part header*/ /* Indicate we will try to read a new tile-part header*/
p_j2k->m_specific_param.m_decoder.m_skip_data = 0; p_j2k->m_specific_param.m_decoder.m_skip_data = 0;
p_j2k->m_specific_param.m_decoder.m_can_decode = 0; p_j2k->m_specific_param.m_decoder.m_can_decode = 0;
p_j2k->m_specific_param.m_decoder.m_state = J2K_STATE_TPHSOT; p_j2k->m_specific_param.m_decoder.m_state = J2K_STATE_TPHSOT;
}
if (! p_j2k->m_specific_param.m_decoder.m_can_decode) {
/* Try to read 2 bytes (the next marker ID) from stream and copy them into the buffer */ /* Try to read 2 bytes (the next marker ID) from stream and copy them into the buffer */
if (opj_stream_read_data(p_stream, if (opj_stream_read_data(p_stream,
p_j2k->m_specific_param.m_decoder.m_header_data, 2, p_manager) != 2) { p_j2k->m_specific_param.m_decoder.m_header_data, 2, p_manager) != 2) {
/* Deal with likely non conformant SPOT6 files, where the last */
/* row of tiles have TPsot == 0 and TNsot == 0, and missing EOC, */
/* but no other tile-parts were found. */
if (p_j2k->m_current_tile_number + 1 == l_nb_tiles) {
OPJ_UINT32 l_tile_no;
for (l_tile_no = 0U; l_tile_no < l_nb_tiles; ++l_tile_no) {
if (p_j2k->m_cp.tcps[l_tile_no].m_current_tile_part_number == 0 &&
p_j2k->m_cp.tcps[l_tile_no].m_nb_tile_parts == 0) {
break;
}
}
if (l_tile_no < l_nb_tiles) {
opj_event_msg(p_manager, EVT_INFO,
"Tile %u has TPsot == 0 and TNsot == 0, "
"but no other tile-parts were found. "
"EOC is also missing.\n",
l_tile_no);
p_j2k->m_current_tile_number = l_tile_no;
l_current_marker = J2K_MS_EOC;
p_j2k->m_specific_param.m_decoder.m_state = J2K_STATE_EOC;
break;
}
}
opj_event_msg(p_manager, EVT_ERROR, "Stream too short\n"); opj_event_msg(p_manager, EVT_ERROR, "Stream too short\n");
return OPJ_FALSE; return OPJ_FALSE;
} }
@ -9611,9 +9626,8 @@ OPJ_BOOL opj_j2k_read_tile_header(opj_j2k_t * p_j2k,
} }
} }
/* FIXME DOC ???*/ /* Deal with tiles that have a single tile-part with TPsot == 0 and TNsot == 0 */
if (! p_j2k->m_specific_param.m_decoder.m_can_decode) { if (! p_j2k->m_specific_param.m_decoder.m_can_decode) {
OPJ_UINT32 l_nb_tiles = p_j2k->m_cp.th * p_j2k->m_cp.tw;
l_tcp = p_j2k->m_cp.tcps + p_j2k->m_current_tile_number; l_tcp = p_j2k->m_cp.tcps + p_j2k->m_current_tile_number;
while ((p_j2k->m_current_tile_number < l_nb_tiles) && (l_tcp->m_data == 00)) { while ((p_j2k->m_current_tile_number < l_nb_tiles) && (l_tcp->m_data == 00)) {

View File

@ -392,3 +392,4 @@ b704ad4c0cfefffd78c20a54f5541265 dwt_interleave_h.gsr105.jp2_d_1_1_33_33_0.pgx
1150acbee2c1e33c57592c05c76e565a file1.jp2-c0-t0-r1.png 1150acbee2c1e33c57592c05c76e565a file1.jp2-c0-t0-r1.png
f31bcb01c771f829054cdb013575e86a issue1043.png f31bcb01c771f829054cdb013575e86a issue1043.png
62bc654c830efddf1b23d6e208447dab tnsot_zero.png 62bc654c830efddf1b23d6e208447dab tnsot_zero.png
c34637a0f218e4074936e0c89534c5b5 tnsot_zero_missing_eoc.png

View File

@ -645,3 +645,5 @@ opj_decompress -i @INPUT_NR_PATH@/tnsot_zero.jp2 -o @TEMP_PATH@/tnsot_zero.png
# try to decompress file with repeated PPT marker # try to decompress file with repeated PPT marker
!opj_decompress -i @INPUT_NR_PATH@/oss-fuzz2785.jp2 -o @TEMP_PATH@/oss-fuzz2785.png !opj_decompress -i @INPUT_NR_PATH@/oss-fuzz2785.jp2 -o @TEMP_PATH@/oss-fuzz2785.png
opj_decompress -i @INPUT_NR_PATH@/tnsot_zero_missing_eoc.jp2 -o @TEMP_PATH@/tnsot_zero_missing_eoc.png