xufulong
5 years ago
parent
dca13944d5
commit
5dd3f0eb8a
@ -0,0 +1,660 @@ |
||||
//
|
||||
// Created by frank on 2019-12-06.
|
||||
//
|
||||
|
||||
#include <stdio.h> |
||||
#include <stdlib.h> |
||||
#include <inttypes.h> |
||||
#include <string.h> |
||||
#include <limits.h> |
||||
#include <jni.h> |
||||
#include <ffmpeg_jni_define.h> |
||||
|
||||
#ifdef __MINGW32__ |
||||
#undef fseeko |
||||
#define fseeko(x, y, z) fseeko64(x, y, z) |
||||
#undef ftello |
||||
#define ftello(x) ftello64(x) |
||||
#elif defined(_WIN32) |
||||
#undef fseeko |
||||
#define fseeko(x, y, z) _fseeki64(x, y, z) |
||||
#undef ftello |
||||
#define ftello(x) _ftelli64(x) |
||||
#endif |
||||
|
||||
#define MIN(a,b) ((a) > (b) ? (b) : (a)) |
||||
|
||||
#define BE_32(x) (((uint32_t)(((uint8_t*)(x))[0]) << 24) | \ |
||||
(((uint8_t*)(x))[1] << 16) | \
|
||||
(((uint8_t*)(x))[2] << 8) | \
|
||||
((uint8_t*)(x))[3]) |
||||
|
||||
#define BE_64(x) (((uint64_t)(((uint8_t*)(x))[0]) << 56) | \ |
||||
((uint64_t)(((uint8_t*)(x))[1]) << 48) | \
|
||||
((uint64_t)(((uint8_t*)(x))[2]) << 40) | \
|
||||
((uint64_t)(((uint8_t*)(x))[3]) << 32) | \
|
||||
((uint64_t)(((uint8_t*)(x))[4]) << 24) | \
|
||||
((uint64_t)(((uint8_t*)(x))[5]) << 16) | \
|
||||
((uint64_t)(((uint8_t*)(x))[6]) << 8) | \
|
||||
((uint64_t)( (uint8_t*)(x))[7])) |
||||
|
||||
#define AV_WB32(p, val) { \ |
||||
((uint8_t*)(p))[0] = ((val) >> 24) & 0xff; \
|
||||
((uint8_t*)(p))[1] = ((val) >> 16) & 0xff; \
|
||||
((uint8_t*)(p))[2] = ((val) >> 8) & 0xff; \
|
||||
((uint8_t*)(p))[3] = (val) & 0xff; \
|
||||
} |
||||
|
||||
#define AV_WB64(p, val) { \ |
||||
AV_WB32(p, (val) >> 32) \
|
||||
AV_WB32(p + 4, val) \
|
||||
} |
||||
|
||||
#define BE_FOURCC(ch0, ch1, ch2, ch3) \ |
||||
( (uint32_t)(unsigned char)(ch3) | \
|
||||
((uint32_t)(unsigned char)(ch2) << 8) | \
|
||||
((uint32_t)(unsigned char)(ch1) << 16) | \
|
||||
((uint32_t)(unsigned char)(ch0) << 24) ) |
||||
|
||||
#define QT_ATOM BE_FOURCC |
||||
/* top level atoms */ |
||||
#define FREE_ATOM QT_ATOM('f', 'r', 'e', 'e') |
||||
#define JUNK_ATOM QT_ATOM('j', 'u', 'n', 'k') |
||||
#define MDAT_ATOM QT_ATOM('m', 'd', 'a', 't') |
||||
#define MOOV_ATOM QT_ATOM('m', 'o', 'o', 'v') |
||||
#define PNOT_ATOM QT_ATOM('p', 'n', 'o', 't') |
||||
#define SKIP_ATOM QT_ATOM('s', 'k', 'i', 'p') |
||||
#define WIDE_ATOM QT_ATOM('w', 'i', 'd', 'e') |
||||
#define PICT_ATOM QT_ATOM('P', 'I', 'C', 'T') |
||||
#define FTYP_ATOM QT_ATOM('f', 't', 'y', 'p') |
||||
#define UUID_ATOM QT_ATOM('u', 'u', 'i', 'd') |
||||
|
||||
#define CMOV_ATOM QT_ATOM('c', 'm', 'o', 'v') |
||||
#define TRAK_ATOM QT_ATOM('t', 'r', 'a', 'k') |
||||
#define MDIA_ATOM QT_ATOM('m', 'd', 'i', 'a') |
||||
#define MINF_ATOM QT_ATOM('m', 'i', 'n', 'f') |
||||
#define STBL_ATOM QT_ATOM('s', 't', 'b', 'l') |
||||
#define STCO_ATOM QT_ATOM('s', 't', 'c', 'o') |
||||
#define CO64_ATOM QT_ATOM('c', 'o', '6', '4') |
||||
|
||||
#define ATOM_PREAMBLE_SIZE 8 |
||||
#define COPY_BUFFER_SIZE 33554432 |
||||
#define MAX_FTYP_ATOM_SIZE 1048576 |
||||
|
||||
typedef struct { |
||||
uint32_t type; |
||||
uint32_t header_size; |
||||
uint64_t size; |
||||
unsigned char *data; |
||||
} atom_t; |
||||
|
||||
typedef struct { |
||||
uint64_t moov_atom_size; |
||||
uint64_t stco_offset_count; |
||||
uint64_t stco_data_size; |
||||
int stco_overflow; |
||||
uint32_t depth; |
||||
} update_chunk_offsets_context_t; |
||||
|
||||
typedef struct { |
||||
unsigned char *dest; |
||||
uint64_t original_moov_size; |
||||
uint64_t new_moov_size; |
||||
} upgrade_stco_context_t; |
||||
|
||||
typedef int (*parse_atoms_callback_t)(void *context, atom_t *atom); |
||||
|
||||
static int parse_atoms( |
||||
unsigned char *buf, |
||||
uint64_t size, |
||||
parse_atoms_callback_t callback, |
||||
void *context) |
||||
{ |
||||
unsigned char *pos = buf; |
||||
unsigned char *end = pos + size; |
||||
atom_t atom; |
||||
int ret; |
||||
|
||||
while (end - pos >= ATOM_PREAMBLE_SIZE) { |
||||
atom.size = BE_32(pos); |
||||
atom.type = BE_32(pos + 4); |
||||
pos += ATOM_PREAMBLE_SIZE; |
||||
atom.header_size = ATOM_PREAMBLE_SIZE; |
||||
|
||||
switch (atom.size) { |
||||
case 1: |
||||
if (end - pos < 8) { |
||||
fprintf(stderr, "not enough room for 64 bit atom size\n"); |
||||
return -1; |
||||
} |
||||
|
||||
atom.size = BE_64(pos); |
||||
pos += 8; |
||||
atom.header_size = ATOM_PREAMBLE_SIZE + 8; |
||||
break; |
||||
|
||||
case 0: |
||||
atom.size = ATOM_PREAMBLE_SIZE + end - pos; |
||||
break; |
||||
default: |
||||
break; |
||||
} |
||||
|
||||
if (atom.size < atom.header_size) { |
||||
fprintf(stderr, "atom size %"PRIu64" too small\n", atom.size); |
||||
return -1; |
||||
} |
||||
|
||||
atom.size -= atom.header_size; |
||||
|
||||
if (atom.size > end - pos) { |
||||
fprintf(stderr, "atom size %"PRIu64" too big\n", atom.size); |
||||
return -1; |
||||
} |
||||
|
||||
atom.data = pos; |
||||
ret = callback(context, &atom); |
||||
if (ret < 0) { |
||||
return ret; |
||||
} |
||||
|
||||
pos += atom.size; |
||||
} |
||||
|
||||
return 0; |
||||
} |
||||
|
||||
static int update_stco_offsets(update_chunk_offsets_context_t *context, atom_t *atom) |
||||
{ |
||||
uint32_t current_offset; |
||||
uint32_t offset_count; |
||||
unsigned char *pos; |
||||
unsigned char *end; |
||||
|
||||
printf(" patching stco atom...\n"); |
||||
if (atom->size < 8) { |
||||
fprintf(stderr, "stco atom size %"PRIu64" too small\n", atom->size); |
||||
return -1; |
||||
} |
||||
|
||||
offset_count = BE_32(atom->data + 4); |
||||
if (offset_count > (atom->size - 8) / 4) { |
||||
fprintf(stderr, "stco offset count %"PRIu32" too big\n", offset_count); |
||||
return -1; |
||||
} |
||||
|
||||
context->stco_offset_count += offset_count; |
||||
context->stco_data_size += atom->size - 8; |
||||
|
||||
for (pos = atom->data + 8, end = pos + offset_count * 4; |
||||
pos < end; |
||||
pos += 4) { |
||||
current_offset = BE_32(pos); |
||||
if (current_offset > UINT_MAX - context->moov_atom_size) { |
||||
context->stco_overflow = 1; |
||||
} |
||||
current_offset += context->moov_atom_size; |
||||
AV_WB32(pos, current_offset); |
||||
} |
||||
|
||||
return 0; |
||||
} |
||||
|
||||
static int update_co64_offsets(update_chunk_offsets_context_t *context, atom_t *atom) |
||||
{ |
||||
uint64_t current_offset; |
||||
uint32_t offset_count; |
||||
unsigned char *pos; |
||||
unsigned char *end; |
||||
|
||||
printf(" patching co64 atom...\n"); |
||||
if (atom->size < 8) { |
||||
fprintf(stderr, "co64 atom size %"PRIu64" too small\n", atom->size); |
||||
return -1; |
||||
} |
||||
|
||||
offset_count = BE_32(atom->data + 4); |
||||
if (offset_count > (atom->size - 8) / 8) { |
||||
fprintf(stderr, "co64 offset count %"PRIu32" too big\n", offset_count); |
||||
return -1; |
||||
} |
||||
|
||||
for (pos = atom->data + 8, end = pos + offset_count * 8; |
||||
pos < end; |
||||
pos += 8) { |
||||
current_offset = BE_64(pos); |
||||
current_offset += context->moov_atom_size; |
||||
AV_WB64(pos, current_offset); |
||||
} |
||||
|
||||
return 0; |
||||
} |
||||
|
||||
static int update_chunk_offsets_callback(void *ctx, atom_t *atom) |
||||
{ |
||||
update_chunk_offsets_context_t *context = ctx; |
||||
int ret; |
||||
|
||||
switch (atom->type) { |
||||
case STCO_ATOM: |
||||
return update_stco_offsets(context, atom); |
||||
|
||||
case CO64_ATOM: |
||||
return update_co64_offsets(context, atom); |
||||
|
||||
case MOOV_ATOM: |
||||
case TRAK_ATOM: |
||||
case MDIA_ATOM: |
||||
case MINF_ATOM: |
||||
case STBL_ATOM: |
||||
context->depth++; |
||||
if (context->depth > 10) { |
||||
fprintf(stderr, "atoms too deeply nested\n"); |
||||
return -1; |
||||
} |
||||
|
||||
ret = parse_atoms( |
||||
atom->data, |
||||
atom->size, |
||||
update_chunk_offsets_callback, |
||||
context); |
||||
context->depth--; |
||||
return ret; |
||||
default: |
||||
break; |
||||
} |
||||
|
||||
return 0; |
||||
} |
||||
|
||||
static void set_atom_size(unsigned char *header, uint32_t header_size, uint64_t size) |
||||
{ |
||||
switch (header_size) { |
||||
case 8: |
||||
AV_WB32(header, size); |
||||
break; |
||||
|
||||
case 16: |
||||
AV_WB64(header + 8, size); |
||||
break; |
||||
default: |
||||
break; |
||||
} |
||||
} |
||||
|
||||
static void upgrade_stco_atom(upgrade_stco_context_t *context, atom_t *atom) |
||||
{ |
||||
unsigned char *pos; |
||||
unsigned char *end; |
||||
uint64_t new_offset; |
||||
uint32_t offset_count; |
||||
uint32_t original_offset; |
||||
|
||||
/* Note: not performing validations since they were performed on the first pass */ |
||||
|
||||
offset_count = BE_32(atom->data + 4); |
||||
|
||||
/* write the header */ |
||||
memcpy(context->dest, atom->data - atom->header_size, atom->header_size + 8); |
||||
AV_WB32(context->dest + 4, CO64_ATOM); |
||||
set_atom_size(context->dest, atom->header_size, atom->header_size + 8 + offset_count * 8); |
||||
context->dest += atom->header_size + 8; |
||||
|
||||
/* write the data */ |
||||
for (pos = atom->data + 8, end = pos + offset_count * 4; |
||||
pos < end; |
||||
pos += 4) { |
||||
original_offset = BE_32(pos) - context->original_moov_size; |
||||
new_offset = (uint64_t)original_offset + context->new_moov_size; |
||||
AV_WB64(context->dest, new_offset); |
||||
context->dest += 8; |
||||
} |
||||
} |
||||
|
||||
static int upgrade_stco_callback(void *ctx, atom_t *atom) |
||||
{ |
||||
upgrade_stco_context_t *context = ctx; |
||||
unsigned char *start_pos; |
||||
uint64_t copy_size; |
||||
|
||||
switch (atom->type) { |
||||
case STCO_ATOM: |
||||
upgrade_stco_atom(context, atom); |
||||
break; |
||||
|
||||
case MOOV_ATOM: |
||||
case TRAK_ATOM: |
||||
case MDIA_ATOM: |
||||
case MINF_ATOM: |
||||
case STBL_ATOM: |
||||
/* write the atom header */ |
||||
memcpy(context->dest, atom->data - atom->header_size, atom->header_size); |
||||
start_pos = context->dest; |
||||
context->dest += atom->header_size; |
||||
|
||||
/* parse internal atoms*/ |
||||
if (parse_atoms( |
||||
atom->data, |
||||
atom->size, |
||||
upgrade_stco_callback, |
||||
context) < 0) { |
||||
return -1; |
||||
} |
||||
|
||||
/* update the atom size */ |
||||
set_atom_size(start_pos, atom->header_size, context->dest - start_pos); |
||||
break; |
||||
|
||||
default: |
||||
copy_size = atom->header_size + atom->size; |
||||
memcpy(context->dest, atom->data - atom->header_size, copy_size); |
||||
context->dest += copy_size; |
||||
break; |
||||
} |
||||
|
||||
return 0; |
||||
} |
||||
|
||||
static int update_moov_atom( |
||||
unsigned char **moov_atom, |
||||
uint64_t *moov_atom_size) |
||||
{ |
||||
update_chunk_offsets_context_t update_context = { 0 }; |
||||
upgrade_stco_context_t upgrade_context; |
||||
unsigned char *new_moov_atom; |
||||
|
||||
update_context.moov_atom_size = *moov_atom_size; |
||||
|
||||
if (parse_atoms( |
||||
*moov_atom, |
||||
*moov_atom_size, |
||||
update_chunk_offsets_callback, |
||||
&update_context) < 0) { |
||||
return -1; |
||||
} |
||||
|
||||
if (!update_context.stco_overflow) { |
||||
return 0; |
||||
} |
||||
|
||||
printf(" upgrading stco atoms to co64...\n"); |
||||
upgrade_context.new_moov_size = *moov_atom_size + |
||||
update_context.stco_offset_count * 8 - |
||||
update_context.stco_data_size; |
||||
|
||||
new_moov_atom = malloc(upgrade_context.new_moov_size); |
||||
if (new_moov_atom == NULL) { |
||||
fprintf(stderr, "could not allocate %"PRIu64" bytes for updated moov atom\n", |
||||
upgrade_context.new_moov_size); |
||||
return -1; |
||||
} |
||||
|
||||
upgrade_context.original_moov_size = *moov_atom_size; |
||||
upgrade_context.dest = new_moov_atom; |
||||
|
||||
if (parse_atoms( |
||||
*moov_atom, |
||||
*moov_atom_size, |
||||
upgrade_stco_callback, |
||||
&upgrade_context) < 0) { |
||||
free(new_moov_atom); |
||||
return -1; |
||||
} |
||||
|
||||
free(*moov_atom); |
||||
*moov_atom = new_moov_atom; |
||||
*moov_atom_size = upgrade_context.new_moov_size; |
||||
|
||||
if (upgrade_context.dest != *moov_atom + *moov_atom_size) { |
||||
fprintf(stderr, "unexpected - wrong number of moov bytes written\n"); |
||||
return -1; |
||||
} |
||||
|
||||
return 0; |
||||
} |
||||
|
||||
FFMPEG_FUNC(jint, fastStart, jstring inputFile, jstring outputFile) { |
||||
FILE *infile = NULL; |
||||
FILE *outfile = NULL; |
||||
unsigned char atom_bytes[ATOM_PREAMBLE_SIZE]; |
||||
uint32_t atom_type = 0; |
||||
uint64_t atom_size = 0; |
||||
uint64_t atom_offset = 0; |
||||
int64_t last_offset; |
||||
unsigned char *moov_atom = NULL; |
||||
unsigned char *ftyp_atom = NULL; |
||||
uint64_t moov_atom_size; |
||||
uint64_t ftyp_atom_size = 0; |
||||
int64_t start_offset = 0; |
||||
unsigned char *copy_buffer = NULL; |
||||
int bytes_to_copy; |
||||
uint64_t free_size = 0; |
||||
uint64_t moov_size = 0; |
||||
|
||||
|
||||
printf("Usage: qt-faststart <infile.mp4> <outfile.mp4>\n" |
||||
"Note: alternatively you can use -movflags +faststart in ffmpeg\n"); |
||||
|
||||
const char *input_file = (*env)->GetStringUTFChars(env, inputFile, JNI_FALSE); |
||||
const char *output_file = (*env)->GetStringUTFChars(env, outputFile, JNI_FALSE); |
||||
|
||||
infile = fopen(input_file, "rb"); |
||||
if (!infile) { |
||||
perror(input_file); |
||||
goto error_out; |
||||
} |
||||
|
||||
/* traverse through the atoms in the file to make sure that 'moov' is
|
||||
* at the end */ |
||||
while (!feof(infile)) { |
||||
if (fread(atom_bytes, ATOM_PREAMBLE_SIZE, 1, infile) != 1) { |
||||
break; |
||||
} |
||||
atom_size = BE_32(&atom_bytes[0]); |
||||
atom_type = BE_32(&atom_bytes[4]); |
||||
|
||||
/* keep ftyp atom */ |
||||
if (atom_type == FTYP_ATOM) { |
||||
if (atom_size > MAX_FTYP_ATOM_SIZE) { |
||||
fprintf(stderr, "ftyp atom size %"PRIu64" too big\n", |
||||
atom_size); |
||||
goto error_out; |
||||
} |
||||
ftyp_atom_size = atom_size; |
||||
free(ftyp_atom); |
||||
ftyp_atom = malloc((size_t) ftyp_atom_size); |
||||
if (!ftyp_atom) { |
||||
fprintf(stderr, "could not allocate %"PRIu64" bytes for ftyp atom\n", |
||||
atom_size); |
||||
goto error_out; |
||||
} |
||||
if (fseeko(infile, -ATOM_PREAMBLE_SIZE, SEEK_CUR) || |
||||
fread(ftyp_atom, (size_t) atom_size, 1, infile) != 1 || |
||||
(start_offset = ftello(infile)) < 0) { |
||||
perror(input_file); |
||||
goto error_out; |
||||
} |
||||
} else { |
||||
int ret; |
||||
/* 64-bit special case */ |
||||
if (atom_size == 1) { |
||||
if (fread(atom_bytes, ATOM_PREAMBLE_SIZE, 1, infile) != 1) { |
||||
break; |
||||
} |
||||
atom_size = BE_64(&atom_bytes[0]); |
||||
ret = fseeko(infile, atom_size - ATOM_PREAMBLE_SIZE * 2, SEEK_CUR); |
||||
} else { |
||||
ret = fseeko(infile, atom_size - ATOM_PREAMBLE_SIZE, SEEK_CUR); |
||||
} |
||||
if (ret) { |
||||
perror(input_file); |
||||
goto error_out; |
||||
} |
||||
} |
||||
printf("%c%c%c%c %10"PRIu64" %"PRIu64"\n", |
||||
(atom_type >> 24) & 255, |
||||
(atom_type >> 16) & 255, |
||||
(atom_type >> 8) & 255, |
||||
(atom_type >> 0) & 255, |
||||
atom_offset, |
||||
atom_size); |
||||
if ((atom_type != FREE_ATOM) && |
||||
(atom_type != JUNK_ATOM) && |
||||
(atom_type != MDAT_ATOM) && |
||||
(atom_type != MOOV_ATOM) && |
||||
(atom_type != PNOT_ATOM) && |
||||
(atom_type != SKIP_ATOM) && |
||||
(atom_type != WIDE_ATOM) && |
||||
(atom_type != PICT_ATOM) && |
||||
(atom_type != UUID_ATOM) && |
||||
(atom_type != FTYP_ATOM)) { |
||||
fprintf(stderr, "encountered non-QT top-level atom (is this a QuickTime file?)\n"); |
||||
break; |
||||
} |
||||
atom_offset += atom_size; |
||||
|
||||
/* The atom header is 8 (or 16 bytes), if the atom size (which
|
||||
* includes these 8 or 16 bytes) is less than that, we won't be |
||||
* able to continue scanning sensibly after this atom, so break. */ |
||||
if (atom_size < 8) |
||||
break; |
||||
|
||||
if (atom_type == MOOV_ATOM) |
||||
moov_size = atom_size; |
||||
|
||||
if (moov_size && atom_type == FREE_ATOM) { |
||||
free_size += atom_size; |
||||
atom_type = MOOV_ATOM; |
||||
atom_size = moov_size; |
||||
} |
||||
} |
||||
|
||||
if (atom_type != MOOV_ATOM) { |
||||
printf("last atom in file was not a moov atom\n"); |
||||
free(ftyp_atom); |
||||
fclose(infile); |
||||
return 0; |
||||
} |
||||
|
||||
if (atom_size < 16) { |
||||
fprintf(stderr, "bad moov atom size\n"); |
||||
goto error_out; |
||||
} |
||||
|
||||
/* moov atom was, in fact, the last atom in the chunk; load the whole
|
||||
* moov atom */ |
||||
if (fseeko(infile, -(atom_size + free_size), SEEK_END)) { |
||||
perror(input_file); |
||||
goto error_out; |
||||
} |
||||
last_offset = ftello(infile); |
||||
if (last_offset < 0) { |
||||
perror(input_file); |
||||
goto error_out; |
||||
} |
||||
moov_atom_size = atom_size; |
||||
moov_atom = malloc((size_t) moov_atom_size); |
||||
if (!moov_atom) { |
||||
fprintf(stderr, "could not allocate %"PRIu64" bytes for moov atom\n", atom_size); |
||||
goto error_out; |
||||
} |
||||
if (fread(moov_atom, (size_t) atom_size, 1, infile) != 1) { |
||||
perror(input_file); |
||||
goto error_out; |
||||
} |
||||
|
||||
/* this utility does not support compressed atoms yet, so disqualify
|
||||
* files with compressed QT atoms */ |
||||
if (BE_32(&moov_atom[12]) == CMOV_ATOM) { |
||||
fprintf(stderr, "this utility does not support compressed moov atoms yet\n"); |
||||
goto error_out; |
||||
} |
||||
|
||||
/* close; will be re-opened later */ |
||||
fclose(infile); |
||||
infile = NULL; |
||||
|
||||
if (update_moov_atom(&moov_atom, &moov_atom_size) < 0) { |
||||
goto error_out; |
||||
} |
||||
|
||||
/* re-open the input file and open the output file */ |
||||
infile = fopen(input_file, "rb"); |
||||
if (!infile) { |
||||
perror(input_file); |
||||
goto error_out; |
||||
} |
||||
|
||||
if (start_offset > 0) { /* seek after ftyp atom */ |
||||
if (fseeko(infile, start_offset, SEEK_SET)) { |
||||
perror(input_file); |
||||
goto error_out; |
||||
} |
||||
|
||||
last_offset -= start_offset; |
||||
} |
||||
|
||||
outfile = fopen(output_file, "wb"); |
||||
if (!outfile) { |
||||
perror(output_file); |
||||
goto error_out; |
||||
} |
||||
|
||||
/* dump the same ftyp atom */ |
||||
if (ftyp_atom_size > 0) { |
||||
printf(" writing ftyp atom...\n"); |
||||
if (fwrite(ftyp_atom, (size_t) ftyp_atom_size, 1, outfile) != 1) { |
||||
perror(output_file); |
||||
goto error_out; |
||||
} |
||||
} |
||||
|
||||
/* dump the new moov atom */ |
||||
printf(" writing moov atom...\n"); |
||||
if (fwrite(moov_atom, (size_t) moov_atom_size, 1, outfile) != 1) { |
||||
perror(output_file); |
||||
goto error_out; |
||||
} |
||||
|
||||
/* copy the remainder of the infile, from offset 0 -> last_offset - 1 */ |
||||
bytes_to_copy = (int) MIN(COPY_BUFFER_SIZE, last_offset); |
||||
copy_buffer = malloc(bytes_to_copy); |
||||
if (!copy_buffer) { |
||||
fprintf(stderr, "could not allocate %d bytes for copy_buffer\n", bytes_to_copy); |
||||
goto error_out; |
||||
} |
||||
printf(" copying rest of file...\n"); |
||||
while (last_offset) { |
||||
bytes_to_copy = (int) MIN(bytes_to_copy, last_offset); |
||||
|
||||
if (fread(copy_buffer, (size_t) bytes_to_copy, 1, infile) != 1) { |
||||
perror(input_file); |
||||
goto error_out; |
||||
} |
||||
if (fwrite(copy_buffer, (size_t) bytes_to_copy, 1, outfile) != 1) { |
||||
perror(output_file); |
||||
goto error_out; |
||||
} |
||||
last_offset -= bytes_to_copy; |
||||
} |
||||
|
||||
fclose(infile); |
||||
fclose(outfile); |
||||
free(moov_atom); |
||||
free(ftyp_atom); |
||||
free(copy_buffer); |
||||
(*env)->ReleaseStringUTFChars(env, inputFile, input_file); |
||||
(*env)->ReleaseStringUTFChars(env, outputFile, output_file); |
||||
|
||||
return 0; |
||||
|
||||
error_out: |
||||
if (infile) |
||||
fclose(infile); |
||||
if (outfile) |
||||
fclose(outfile); |
||||
free(moov_atom); |
||||
free(ftyp_atom); |
||||
free(copy_buffer); |
||||
return 1; |
||||
} |
||||
Loading…
Reference in new issue