Changeset 302


Ignore:
Timestamp:
05/02/07 22:31:31 (17 years ago)
Author:
r2d
Message:
  • added some checks in sv8 demuxing
  • corrected a bug in mpc2sv8
Location:
libmpc/trunk
Files:
2 edited

Legend:

Unmodified
Added
Removed
  • libmpc/trunk/libmpcdec/mpc_demux.c

    r301 r302  
    101101        }
    102102
    103         return 0;
     103        return (mpc_uint32_t) -1;
    104104}
    105105
     
    367367{
    368368        mpc_bits_reader r;
     369        int pos = 0;
    369370        if (d->si.stream_version >= 8) {
    370371                i->is_key_frame = MPC_FALSE;
     
    379380                        mpc_demux_fill(d, 11, 0); // max header block size
    380381                        mpc_bits_get_block(&d->bits_reader, &b);
     382                        pos = mpc_demux_pos(d) >> 3;
    381383                        while( memcmp(b.key, "AP", 2) != 0 ) { // scan all blocks until audio
     384                                if (b.key[0] < 65 || b.key[0] > 90 || b.key[1] < 65 || b.key[1] > 90
     385                                                                || b.size > (mpc_uint64_t) DEMUX_BUFFER_SIZE - 11)
     386                                        goto error;
    382387                                if (memcmp(b.key, "SE", 2) == 0) { // end block
    383388                                        i->bits = -1;
    384389                                        return MPC_STATUS_OK;
    385390                                }
    386                                 mpc_demux_fill(d, 11 + (mpc_uint32_t) b.size, 0);
     391                                if (mpc_demux_fill(d, 11 + (mpc_uint32_t) b.size, 0) == 0)
     392                                        goto error;
    387393                                d->bits_reader.buff += b.size;
    388394                                mpc_bits_get_block(&d->bits_reader, &b);
     
    392398                        i->is_key_frame = MPC_TRUE;
    393399                }
    394                 if (d->buffer + d->bytes_total - d->bits_reader.buff <= MAX_FRAME_SIZE)
     400                if (d->buffer + d->bytes_total - d->bits_reader.buff <= MAX_FRAME_SIZE * 2)
    395401                        mpc_demux_fill(d, (d->block_bits >> 3) + 1, 0);
    396402                r = d->bits_reader;
     
    398404                d->block_bits -= ((d->bits_reader.buff - r.buff) << 3) + r.count - d->bits_reader.count;
    399405                d->block_frames--;
    400                 if (d->block_bits < 0 || (d->block_frames == 0 && d->block_bits > 7)) {
    401                         i->bits = -1; // we pretend it's end of file
    402                         return MPC_STATUS_INVALIDSV;
    403                 }
     406                if (d->block_bits < 0 || (d->block_frames == 0 && d->block_bits > 7))
     407                        goto error;
    404408        } else {
    405409                if (d->d->decoded_samples == (d->seek_table_size << d->seek_pwr) * MPC_FRAME_LENGTH) {
     
    412416                r = d->bits_reader;
    413417                mpc_decoder_decode_frame(d->d, &d->bits_reader, i);
    414                 if (i->bits != -1 && d->block_bits != ((d->bits_reader.buff - r.buff) << 3) + r.count - d->bits_reader.count) {
    415                         i->bits = -1; // we pretend it's end of file
    416                         return MPC_STATUS_INVALIDSV;
    417                 }
    418         }
    419         if (d->buffer + d->bytes_total < d->bits_reader.buff + ((8 - d->bits_reader.count) >> 3)) {
     418                if (i->bits != -1 && d->block_bits != ((d->bits_reader.buff - r.buff) << 3) + r.count - d->bits_reader.count)
     419                        goto error;
     420        }
     421        if (d->buffer + d->bytes_total < d->bits_reader.buff + ((8 - d->bits_reader.count) >> 3))
     422                goto error;
     423
     424        return MPC_STATUS_OK;
     425error:
    420426                i->bits = -1; // we pretend it's end of file
    421                 return MPC_STATUS_INVALIDSV;
    422         }
    423         return MPC_STATUS_OK;
     427                // return MPC_STATUS_INVALIDSV;
     428                return pos;
    424429}
    425430
  • libmpc/trunk/mpc2sv8/mpc2sv8.c

    r297 r302  
    177177
    178178        // copy end of file
    179         stream_size = (((mpc_demux_pos(demux) + 7) >> 3) - si.header_position + 3) & ~3;
     179
     180        stream_size = (((mpc_demux_pos(demux) + 7 - 20) >> 3) - si.header_position + 3) & ~3;
    180181        fseek(in_file, si.header_position + stream_size, SEEK_SET);
    181182        while((r_size = fread(buf, 1, TMP_BUF_SIZE, in_file))) {
Note: See TracChangeset for help on using the changeset viewer.