blob: 4e9b72685223aa392fead5da572150dd2ec25b60 [file] [log] [blame]
Andreas Huber88572f72012-02-21 11:47:18 -08001/*
2 * Copyright 2012, The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17//#define LOG_NDEBUG 0
18#define LOG_TAG "MediaExtractor-JNI"
19#include <utils/Log.h>
20
21#include "android_media_MediaExtractor.h"
22
23#include "android_media_Utils.h"
24#include "android_runtime/AndroidRuntime.h"
Ruben Brunk87eac992013-09-09 17:44:59 -070025#include "android_runtime/Log.h"
Andreas Huber88572f72012-02-21 11:47:18 -080026#include "jni.h"
27#include "JNIHelp.h"
Chris Watkins4eaa2932015-03-20 10:31:42 -070028#include "android_media_MediaDataSource.h"
Andreas Huber88572f72012-02-21 11:47:18 -080029
Andreas Huberd2506a52014-01-29 10:32:46 -080030#include <media/IMediaHTTPService.h>
Andreas Huberab57d032012-04-18 15:59:43 -070031#include <media/hardware/CryptoAPI.h>
Andreas Huber88572f72012-02-21 11:47:18 -080032#include <media/stagefright/foundation/ABuffer.h>
33#include <media/stagefright/foundation/ADebug.h>
34#include <media/stagefright/foundation/AMessage.h>
35#include <media/stagefright/DataSource.h>
36#include <media/stagefright/MediaErrors.h>
Andreas Huber91befdc2012-04-18 12:19:51 -070037#include <media/stagefright/MetaData.h>
Andreas Huber88572f72012-02-21 11:47:18 -080038#include <media/stagefright/NuMediaExtractor.h>
39
Lajos Molnar7be24522014-07-14 12:24:56 -070040#include <nativehelper/ScopedLocalRef.h>
41
Andreas Huberd2506a52014-01-29 10:32:46 -080042#include "android_util_Binder.h"
43
Andreas Huber88572f72012-02-21 11:47:18 -080044namespace android {
45
46struct fields_t {
47 jfieldID context;
Andreas Huber91befdc2012-04-18 12:19:51 -070048
49 jmethodID cryptoInfoSetID;
Andreas Huber88572f72012-02-21 11:47:18 -080050};
51
52static fields_t gFields;
53
Andreas Huber88572f72012-02-21 11:47:18 -080054JMediaExtractor::JMediaExtractor(JNIEnv *env, jobject thiz)
55 : mClass(NULL),
56 mObject(NULL) {
57 jclass clazz = env->GetObjectClass(thiz);
58 CHECK(clazz != NULL);
59
60 mClass = (jclass)env->NewGlobalRef(clazz);
61 mObject = env->NewWeakGlobalRef(thiz);
62
63 mImpl = new NuMediaExtractor;
64}
65
66JMediaExtractor::~JMediaExtractor() {
67 JNIEnv *env = AndroidRuntime::getJNIEnv();
68
69 env->DeleteWeakGlobalRef(mObject);
70 mObject = NULL;
71 env->DeleteGlobalRef(mClass);
72 mClass = NULL;
73}
74
Andreas Huber07ea4262012-04-11 12:21:20 -070075status_t JMediaExtractor::setDataSource(
Andreas Huberd2506a52014-01-29 10:32:46 -080076 const sp<IMediaHTTPService> &httpService,
77 const char *path,
78 const KeyedVector<String8, String8> *headers) {
79 return mImpl->setDataSource(httpService, path, headers);
Andreas Huber07ea4262012-04-11 12:21:20 -070080}
81
82status_t JMediaExtractor::setDataSource(int fd, off64_t offset, off64_t size) {
83 return mImpl->setDataSource(fd, offset, size);
Andreas Huber88572f72012-02-21 11:47:18 -080084}
85
Marco Nelissenc209a062012-08-24 09:55:44 -070086status_t JMediaExtractor::setDataSource(const sp<DataSource> &datasource) {
87 return mImpl->setDataSource(datasource);
88}
89
Andreas Huber88572f72012-02-21 11:47:18 -080090size_t JMediaExtractor::countTracks() const {
91 return mImpl->countTracks();
92}
93
94status_t JMediaExtractor::getTrackFormat(size_t index, jobject *format) const {
95 sp<AMessage> msg;
96 status_t err;
97 if ((err = mImpl->getTrackFormat(index, &msg)) != OK) {
98 return err;
99 }
100
101 JNIEnv *env = AndroidRuntime::getJNIEnv();
102
103 return ConvertMessageToMap(env, msg, format);
104}
105
Marco Nelissene20a6d52013-04-08 14:28:55 -0700106status_t JMediaExtractor::getFileFormat(jobject *format) const {
107 sp<AMessage> msg;
108 status_t err;
109 if ((err = mImpl->getFileFormat(&msg)) != OK) {
110 return err;
111 }
112
113 JNIEnv *env = AndroidRuntime::getJNIEnv();
114
115 return ConvertMessageToMap(env, msg, format);
116}
117
Andreas Huber88572f72012-02-21 11:47:18 -0800118status_t JMediaExtractor::selectTrack(size_t index) {
119 return mImpl->selectTrack(index);
120}
121
Andreas Huberf2855b32012-04-25 15:57:43 -0700122status_t JMediaExtractor::unselectTrack(size_t index) {
123 return mImpl->unselectTrack(index);
124}
125
126status_t JMediaExtractor::seekTo(
127 int64_t timeUs, MediaSource::ReadOptions::SeekMode mode) {
128 return mImpl->seekTo(timeUs, mode);
Andreas Huber88572f72012-02-21 11:47:18 -0800129}
130
131status_t JMediaExtractor::advance() {
132 return mImpl->advance();
133}
134
135status_t JMediaExtractor::readSampleData(
136 jobject byteBuf, size_t offset, size_t *sampleSize) {
137 JNIEnv *env = AndroidRuntime::getJNIEnv();
138
139 void *dst = env->GetDirectBufferAddress(byteBuf);
140
Ashok Bhat075e9a12014-01-06 13:45:09 +0000141 size_t dstSize;
Andreas Huberc52b9802012-03-12 14:04:01 -0700142 jbyteArray byteArray = NULL;
143
Lajos Molnar7be24522014-07-14 12:24:56 -0700144 ScopedLocalRef<jclass> byteBufClass(env, env->FindClass("java/nio/ByteBuffer"));
145 CHECK(byteBufClass.get() != NULL);
Andreas Huberc52b9802012-03-12 14:04:01 -0700146
Lajos Molnar7be24522014-07-14 12:24:56 -0700147 if (dst == NULL) {
Andreas Huberc52b9802012-03-12 14:04:01 -0700148 jmethodID arrayID =
Lajos Molnar7be24522014-07-14 12:24:56 -0700149 env->GetMethodID(byteBufClass.get(), "array", "()[B");
Andreas Huberc52b9802012-03-12 14:04:01 -0700150 CHECK(arrayID != NULL);
151
152 byteArray =
153 (jbyteArray)env->CallObjectMethod(byteBuf, arrayID);
154
155 if (byteArray == NULL) {
156 return INVALID_OPERATION;
157 }
158
159 jboolean isCopy;
160 dst = env->GetByteArrayElements(byteArray, &isCopy);
161
Ashok Bhat075e9a12014-01-06 13:45:09 +0000162 dstSize = (size_t) env->GetArrayLength(byteArray);
Andreas Huberc52b9802012-03-12 14:04:01 -0700163 } else {
Ashok Bhat075e9a12014-01-06 13:45:09 +0000164 dstSize = (size_t) env->GetDirectBufferCapacity(byteBuf);
Andreas Huber88572f72012-02-21 11:47:18 -0800165 }
166
Andreas Huber88572f72012-02-21 11:47:18 -0800167 if (dstSize < offset) {
Andreas Huberc52b9802012-03-12 14:04:01 -0700168 if (byteArray != NULL) {
169 env->ReleaseByteArrayElements(byteArray, (jbyte *)dst, 0);
170 }
171
Andreas Huber88572f72012-02-21 11:47:18 -0800172 return -ERANGE;
173 }
174
175 sp<ABuffer> buffer = new ABuffer((char *)dst + offset, dstSize - offset);
176
177 status_t err = mImpl->readSampleData(buffer);
178
Andreas Huberc52b9802012-03-12 14:04:01 -0700179 if (byteArray != NULL) {
180 env->ReleaseByteArrayElements(byteArray, (jbyte *)dst, 0);
181 }
182
Andreas Huber88572f72012-02-21 11:47:18 -0800183 if (err != OK) {
184 return err;
185 }
186
187 *sampleSize = buffer->size();
188
Lajos Molnar7be24522014-07-14 12:24:56 -0700189 jmethodID positionID = env->GetMethodID(
190 byteBufClass.get(), "position", "(I)Ljava/nio/Buffer;");
191
192 CHECK(positionID != NULL);
193
194 jmethodID limitID = env->GetMethodID(
195 byteBufClass.get(), "limit", "(I)Ljava/nio/Buffer;");
196
197 CHECK(limitID != NULL);
198
199 jobject me = env->CallObjectMethod(
200 byteBuf, limitID, offset + *sampleSize);
201 env->DeleteLocalRef(me);
202 me = env->CallObjectMethod(
203 byteBuf, positionID, offset);
204 env->DeleteLocalRef(me);
205 me = NULL;
206
Andreas Huber88572f72012-02-21 11:47:18 -0800207 return OK;
208}
209
210status_t JMediaExtractor::getSampleTrackIndex(size_t *trackIndex) {
211 return mImpl->getSampleTrackIndex(trackIndex);
212}
213
214status_t JMediaExtractor::getSampleTime(int64_t *sampleTimeUs) {
215 return mImpl->getSampleTime(sampleTimeUs);
216}
217
Andreas Huber9b8e4962012-03-26 11:13:27 -0700218status_t JMediaExtractor::getSampleFlags(uint32_t *sampleFlags) {
Andreas Huber91befdc2012-04-18 12:19:51 -0700219 *sampleFlags = 0;
220
221 sp<MetaData> meta;
222 status_t err = mImpl->getSampleMeta(&meta);
223
224 if (err != OK) {
225 return err;
226 }
227
228 int32_t val;
229 if (meta->findInt32(kKeyIsSyncFrame, &val) && val != 0) {
230 (*sampleFlags) |= NuMediaExtractor::SAMPLE_FLAG_SYNC;
231 }
232
233 uint32_t type;
234 const void *data;
235 size_t size;
236 if (meta->findData(kKeyEncryptedSizes, &type, &data, &size)) {
237 (*sampleFlags) |= NuMediaExtractor::SAMPLE_FLAG_ENCRYPTED;
238 }
239
240 return OK;
241}
242
243status_t JMediaExtractor::getSampleMeta(sp<MetaData> *sampleMeta) {
244 return mImpl->getSampleMeta(sampleMeta);
Andreas Huber9b8e4962012-03-26 11:13:27 -0700245}
246
Andreas Huber74a78b02012-04-19 16:24:32 -0700247bool JMediaExtractor::getCachedDuration(int64_t *durationUs, bool *eos) const {
248 return mImpl->getCachedDuration(durationUs, eos);
249}
250
Andreas Huber88572f72012-02-21 11:47:18 -0800251} // namespace android
252
253////////////////////////////////////////////////////////////////////////////////
254
255using namespace android;
256
257static sp<JMediaExtractor> setMediaExtractor(
258 JNIEnv *env, jobject thiz, const sp<JMediaExtractor> &extractor) {
259 sp<JMediaExtractor> old =
Ashok Bhat075e9a12014-01-06 13:45:09 +0000260 (JMediaExtractor *)env->GetLongField(thiz, gFields.context);
Andreas Huber88572f72012-02-21 11:47:18 -0800261
262 if (extractor != NULL) {
263 extractor->incStrong(thiz);
264 }
265 if (old != NULL) {
266 old->decStrong(thiz);
267 }
Ashok Bhat075e9a12014-01-06 13:45:09 +0000268 env->SetLongField(thiz, gFields.context, (jlong)extractor.get());
Andreas Huber88572f72012-02-21 11:47:18 -0800269
270 return old;
271}
272
273static sp<JMediaExtractor> getMediaExtractor(JNIEnv *env, jobject thiz) {
Ashok Bhat075e9a12014-01-06 13:45:09 +0000274 return (JMediaExtractor *)env->GetLongField(thiz, gFields.context);
Andreas Huber88572f72012-02-21 11:47:18 -0800275}
276
277static void android_media_MediaExtractor_release(JNIEnv *env, jobject thiz) {
278 setMediaExtractor(env, thiz, NULL);
279}
280
Andreas Huber60d610b2012-05-02 16:06:09 -0700281static jint android_media_MediaExtractor_getTrackCount(
Andreas Huber88572f72012-02-21 11:47:18 -0800282 JNIEnv *env, jobject thiz) {
283 sp<JMediaExtractor> extractor = getMediaExtractor(env, thiz);
284
285 if (extractor == NULL) {
286 jniThrowException(env, "java/lang/IllegalStateException", NULL);
Andreas Huber07ea4262012-04-11 12:21:20 -0700287 return -1;
Andreas Huber88572f72012-02-21 11:47:18 -0800288 }
289
Ashok Bhat075e9a12014-01-06 13:45:09 +0000290 return (jint) extractor->countTracks();
Andreas Huber88572f72012-02-21 11:47:18 -0800291}
292
Andreas Huber60d610b2012-05-02 16:06:09 -0700293static jobject android_media_MediaExtractor_getTrackFormatNative(
Andreas Huber88572f72012-02-21 11:47:18 -0800294 JNIEnv *env, jobject thiz, jint index) {
295 sp<JMediaExtractor> extractor = getMediaExtractor(env, thiz);
296
297 if (extractor == NULL) {
298 jniThrowException(env, "java/lang/IllegalStateException", NULL);
299 return NULL;
300 }
301
302 jobject format;
303 status_t err = extractor->getTrackFormat(index, &format);
304
305 if (err != OK) {
306 jniThrowException(env, "java/lang/IllegalArgumentException", NULL);
307 return NULL;
308 }
309
310 return format;
311}
312
Marco Nelissene20a6d52013-04-08 14:28:55 -0700313static jobject android_media_MediaExtractor_getFileFormatNative(
314 JNIEnv *env, jobject thiz) {
315 sp<JMediaExtractor> extractor = getMediaExtractor(env, thiz);
316
317 if (extractor == NULL) {
318 jniThrowException(env, "java/lang/IllegalStateException", NULL);
319 return NULL;
320 }
321
322 jobject format;
323 status_t err = extractor->getFileFormat(&format);
324
325 if (err != OK) {
326 jniThrowException(env, "java/lang/IllegalArgumentException", NULL);
327 return NULL;
328 }
329
330 return format;
331}
332
Andreas Huber88572f72012-02-21 11:47:18 -0800333static void android_media_MediaExtractor_selectTrack(
334 JNIEnv *env, jobject thiz, jint index) {
335 sp<JMediaExtractor> extractor = getMediaExtractor(env, thiz);
336
337 if (extractor == NULL) {
338 jniThrowException(env, "java/lang/IllegalStateException", NULL);
339 return;
340 }
341
342 status_t err = extractor->selectTrack(index);
343
344 if (err != OK) {
345 jniThrowException(env, "java/lang/IllegalArgumentException", NULL);
346 return;
347 }
348}
349
Andreas Huberf2855b32012-04-25 15:57:43 -0700350static void android_media_MediaExtractor_unselectTrack(
351 JNIEnv *env, jobject thiz, jint index) {
Andreas Huber88572f72012-02-21 11:47:18 -0800352 sp<JMediaExtractor> extractor = getMediaExtractor(env, thiz);
353
354 if (extractor == NULL) {
355 jniThrowException(env, "java/lang/IllegalStateException", NULL);
356 return;
357 }
358
Andreas Huberf2855b32012-04-25 15:57:43 -0700359 status_t err = extractor->unselectTrack(index);
360
361 if (err != OK) {
362 jniThrowException(env, "java/lang/IllegalArgumentException", NULL);
363 return;
364 }
365}
366
367static void android_media_MediaExtractor_seekTo(
368 JNIEnv *env, jobject thiz, jlong timeUs, jint mode) {
369 sp<JMediaExtractor> extractor = getMediaExtractor(env, thiz);
370
371 if (extractor == NULL) {
372 jniThrowException(env, "java/lang/IllegalStateException", NULL);
373 return;
374 }
375
376 if (mode < MediaSource::ReadOptions::SEEK_PREVIOUS_SYNC
Andreas Huberf86374a2012-05-09 11:25:29 -0700377 || mode >= MediaSource::ReadOptions::SEEK_CLOSEST) {
Andreas Huberf2855b32012-04-25 15:57:43 -0700378 jniThrowException(env, "java/lang/IllegalArgumentException", NULL);
379 return;
380 }
381
382 extractor->seekTo(timeUs, (MediaSource::ReadOptions::SeekMode)mode);
Andreas Huber88572f72012-02-21 11:47:18 -0800383}
384
385static jboolean android_media_MediaExtractor_advance(
386 JNIEnv *env, jobject thiz) {
387 sp<JMediaExtractor> extractor = getMediaExtractor(env, thiz);
388
389 if (extractor == NULL) {
390 jniThrowException(env, "java/lang/IllegalStateException", NULL);
Ashok Bhat075e9a12014-01-06 13:45:09 +0000391 return JNI_FALSE;
Andreas Huber88572f72012-02-21 11:47:18 -0800392 }
393
394 status_t err = extractor->advance();
395
396 if (err == ERROR_END_OF_STREAM) {
Ashok Bhat075e9a12014-01-06 13:45:09 +0000397 return JNI_FALSE;
Andreas Huber88572f72012-02-21 11:47:18 -0800398 } else if (err != OK) {
399 jniThrowException(env, "java/lang/IllegalArgumentException", NULL);
Ashok Bhat075e9a12014-01-06 13:45:09 +0000400 return JNI_FALSE;
Andreas Huber88572f72012-02-21 11:47:18 -0800401 }
402
Ashok Bhat075e9a12014-01-06 13:45:09 +0000403 return JNI_TRUE;
Andreas Huber88572f72012-02-21 11:47:18 -0800404}
405
406static jint android_media_MediaExtractor_readSampleData(
407 JNIEnv *env, jobject thiz, jobject byteBuf, jint offset) {
408 sp<JMediaExtractor> extractor = getMediaExtractor(env, thiz);
409
410 if (extractor == NULL) {
411 jniThrowException(env, "java/lang/IllegalStateException", NULL);
412 return -1;
413 }
414
415 size_t sampleSize;
416 status_t err = extractor->readSampleData(byteBuf, offset, &sampleSize);
417
418 if (err == ERROR_END_OF_STREAM) {
419 return -1;
420 } else if (err != OK) {
421 jniThrowException(env, "java/lang/IllegalArgumentException", NULL);
Ashok Bhat075e9a12014-01-06 13:45:09 +0000422 return -1;
Andreas Huber88572f72012-02-21 11:47:18 -0800423 }
424
Ashok Bhat075e9a12014-01-06 13:45:09 +0000425 return (jint) sampleSize;
Andreas Huber88572f72012-02-21 11:47:18 -0800426}
427
428static jint android_media_MediaExtractor_getSampleTrackIndex(
429 JNIEnv *env, jobject thiz) {
430 sp<JMediaExtractor> extractor = getMediaExtractor(env, thiz);
431
432 if (extractor == NULL) {
433 jniThrowException(env, "java/lang/IllegalStateException", NULL);
434 return -1;
435 }
436
437 size_t trackIndex;
438 status_t err = extractor->getSampleTrackIndex(&trackIndex);
439
440 if (err == ERROR_END_OF_STREAM) {
441 return -1;
442 } else if (err != OK) {
443 jniThrowException(env, "java/lang/IllegalArgumentException", NULL);
Ashok Bhat075e9a12014-01-06 13:45:09 +0000444 return -1;
Andreas Huber88572f72012-02-21 11:47:18 -0800445 }
446
Ashok Bhat075e9a12014-01-06 13:45:09 +0000447 return (jint) trackIndex;
Andreas Huber88572f72012-02-21 11:47:18 -0800448}
449
450static jlong android_media_MediaExtractor_getSampleTime(
451 JNIEnv *env, jobject thiz) {
452 sp<JMediaExtractor> extractor = getMediaExtractor(env, thiz);
453
454 if (extractor == NULL) {
455 jniThrowException(env, "java/lang/IllegalStateException", NULL);
456 return -1ll;
457 }
458
459 int64_t sampleTimeUs;
460 status_t err = extractor->getSampleTime(&sampleTimeUs);
461
462 if (err == ERROR_END_OF_STREAM) {
463 return -1ll;
464 } else if (err != OK) {
465 jniThrowException(env, "java/lang/IllegalArgumentException", NULL);
Ashok Bhat075e9a12014-01-06 13:45:09 +0000466 return -1ll;
Andreas Huber88572f72012-02-21 11:47:18 -0800467 }
468
Ashok Bhat075e9a12014-01-06 13:45:09 +0000469 return (jlong) sampleTimeUs;
Andreas Huber88572f72012-02-21 11:47:18 -0800470}
471
Andreas Huber9b8e4962012-03-26 11:13:27 -0700472static jint android_media_MediaExtractor_getSampleFlags(
473 JNIEnv *env, jobject thiz) {
474 sp<JMediaExtractor> extractor = getMediaExtractor(env, thiz);
475
476 if (extractor == NULL) {
477 jniThrowException(env, "java/lang/IllegalStateException", NULL);
Ashok Bhat075e9a12014-01-06 13:45:09 +0000478 return -1;
Andreas Huber9b8e4962012-03-26 11:13:27 -0700479 }
480
481 uint32_t sampleFlags;
482 status_t err = extractor->getSampleFlags(&sampleFlags);
483
484 if (err == ERROR_END_OF_STREAM) {
Ashok Bhat075e9a12014-01-06 13:45:09 +0000485 return -1;
Andreas Huber9b8e4962012-03-26 11:13:27 -0700486 } else if (err != OK) {
487 jniThrowException(env, "java/lang/IllegalArgumentException", NULL);
Ashok Bhat075e9a12014-01-06 13:45:09 +0000488 return -1;
Andreas Huber9b8e4962012-03-26 11:13:27 -0700489 }
490
Ashok Bhat075e9a12014-01-06 13:45:09 +0000491 return (jint) sampleFlags;
Andreas Huber9b8e4962012-03-26 11:13:27 -0700492}
493
Andreas Huber91befdc2012-04-18 12:19:51 -0700494static jboolean android_media_MediaExtractor_getSampleCryptoInfo(
495 JNIEnv *env, jobject thiz, jobject cryptoInfoObj) {
496 sp<JMediaExtractor> extractor = getMediaExtractor(env, thiz);
497
498 if (extractor == NULL) {
499 jniThrowException(env, "java/lang/IllegalStateException", NULL);
Ashok Bhat075e9a12014-01-06 13:45:09 +0000500 return JNI_FALSE;
Andreas Huber91befdc2012-04-18 12:19:51 -0700501 }
502
503 sp<MetaData> meta;
504 status_t err = extractor->getSampleMeta(&meta);
505
506 if (err != OK) {
Ashok Bhat075e9a12014-01-06 13:45:09 +0000507 return JNI_FALSE;
Andreas Huber91befdc2012-04-18 12:19:51 -0700508 }
509
510 uint32_t type;
511 const void *data;
512 size_t size;
513 if (!meta->findData(kKeyEncryptedSizes, &type, &data, &size)) {
Ashok Bhat075e9a12014-01-06 13:45:09 +0000514 return JNI_FALSE;
Andreas Huber91befdc2012-04-18 12:19:51 -0700515 }
516
Ashok Bhate92416d2014-03-05 11:51:15 +0000517 size_t numSubSamples = size / sizeof(int32_t);
Andreas Huber91befdc2012-04-18 12:19:51 -0700518
519 if (numSubSamples == 0) {
Ashok Bhat075e9a12014-01-06 13:45:09 +0000520 return JNI_FALSE;
Andreas Huber91befdc2012-04-18 12:19:51 -0700521 }
522
523 jintArray numBytesOfEncryptedDataObj = env->NewIntArray(numSubSamples);
524 jboolean isCopy;
525 jint *dst = env->GetIntArrayElements(numBytesOfEncryptedDataObj, &isCopy);
526 for (size_t i = 0; i < numSubSamples; ++i) {
Ashok Bhate92416d2014-03-05 11:51:15 +0000527 dst[i] = ((const int32_t *)data)[i];
Andreas Huber91befdc2012-04-18 12:19:51 -0700528 }
529 env->ReleaseIntArrayElements(numBytesOfEncryptedDataObj, dst, 0);
530 dst = NULL;
531
532 size_t encSize = size;
533 jintArray numBytesOfPlainDataObj = NULL;
Andreas Huberab57d032012-04-18 15:59:43 -0700534 if (meta->findData(kKeyPlainSizes, &type, &data, &size)) {
Andreas Huber91befdc2012-04-18 12:19:51 -0700535 if (size != encSize) {
536 // The two must be of the same length.
Ashok Bhat075e9a12014-01-06 13:45:09 +0000537 return JNI_FALSE;
Andreas Huber91befdc2012-04-18 12:19:51 -0700538 }
539
540 numBytesOfPlainDataObj = env->NewIntArray(numSubSamples);
541 jboolean isCopy;
542 jint *dst = env->GetIntArrayElements(numBytesOfPlainDataObj, &isCopy);
543 for (size_t i = 0; i < numSubSamples; ++i) {
Ashok Bhate92416d2014-03-05 11:51:15 +0000544 dst[i] = ((const int32_t *)data)[i];
Andreas Huber91befdc2012-04-18 12:19:51 -0700545 }
546 env->ReleaseIntArrayElements(numBytesOfPlainDataObj, dst, 0);
547 dst = NULL;
548 }
549
550 jbyteArray keyObj = NULL;
551 if (meta->findData(kKeyCryptoKey, &type, &data, &size)) {
552 if (size != 16) {
553 // Keys must be 16 bytes in length.
Ashok Bhat075e9a12014-01-06 13:45:09 +0000554 return JNI_FALSE;
Andreas Huber91befdc2012-04-18 12:19:51 -0700555 }
556
557 keyObj = env->NewByteArray(size);
558 jboolean isCopy;
559 jbyte *dst = env->GetByteArrayElements(keyObj, &isCopy);
560 memcpy(dst, data, size);
561 env->ReleaseByteArrayElements(keyObj, dst, 0);
562 dst = NULL;
563 }
564
565 jbyteArray ivObj = NULL;
566 if (meta->findData(kKeyCryptoIV, &type, &data, &size)) {
567 if (size != 16) {
568 // IVs must be 16 bytes in length.
Ashok Bhat075e9a12014-01-06 13:45:09 +0000569 return JNI_FALSE;
Andreas Huber91befdc2012-04-18 12:19:51 -0700570 }
571
572 ivObj = env->NewByteArray(size);
573 jboolean isCopy;
574 jbyte *dst = env->GetByteArrayElements(ivObj, &isCopy);
575 memcpy(dst, data, size);
576 env->ReleaseByteArrayElements(ivObj, dst, 0);
577 dst = NULL;
578 }
579
580 int32_t mode;
581 if (!meta->findInt32(kKeyCryptoMode, &mode)) {
Andreas Huberab57d032012-04-18 15:59:43 -0700582 mode = CryptoPlugin::kMode_AES_CTR;
Andreas Huber91befdc2012-04-18 12:19:51 -0700583 }
584
585 env->CallVoidMethod(
586 cryptoInfoObj,
587 gFields.cryptoInfoSetID,
Ashok Bhat136c08a2014-03-05 15:23:13 +0000588 (jint)numSubSamples,
Andreas Huber91befdc2012-04-18 12:19:51 -0700589 numBytesOfPlainDataObj,
590 numBytesOfEncryptedDataObj,
591 keyObj,
592 ivObj,
593 mode);
594
Ashok Bhat075e9a12014-01-06 13:45:09 +0000595 return JNI_TRUE;
Andreas Huber91befdc2012-04-18 12:19:51 -0700596}
597
Andreas Huber88572f72012-02-21 11:47:18 -0800598static void android_media_MediaExtractor_native_init(JNIEnv *env) {
599 jclass clazz = env->FindClass("android/media/MediaExtractor");
600 CHECK(clazz != NULL);
601
Ashok Bhat075e9a12014-01-06 13:45:09 +0000602 gFields.context = env->GetFieldID(clazz, "mNativeContext", "J");
Andreas Huber88572f72012-02-21 11:47:18 -0800603 CHECK(gFields.context != NULL);
604
Andreas Huber91befdc2012-04-18 12:19:51 -0700605 clazz = env->FindClass("android/media/MediaCodec$CryptoInfo");
606 CHECK(clazz != NULL);
607
608 gFields.cryptoInfoSetID =
609 env->GetMethodID(clazz, "set", "(I[I[I[B[BI)V");
610
Andreas Huber88572f72012-02-21 11:47:18 -0800611 DataSource::RegisterDefaultSniffers();
612}
613
614static void android_media_MediaExtractor_native_setup(
Andreas Huber07ea4262012-04-11 12:21:20 -0700615 JNIEnv *env, jobject thiz) {
Andreas Huber88572f72012-02-21 11:47:18 -0800616 sp<JMediaExtractor> extractor = new JMediaExtractor(env, thiz);
Andreas Huber07ea4262012-04-11 12:21:20 -0700617 setMediaExtractor(env,thiz, extractor);
618}
Andreas Huber88572f72012-02-21 11:47:18 -0800619
Andreas Huber07ea4262012-04-11 12:21:20 -0700620static void android_media_MediaExtractor_setDataSource(
621 JNIEnv *env, jobject thiz,
Andreas Huberd2506a52014-01-29 10:32:46 -0800622 jobject httpServiceBinderObj,
623 jstring pathObj,
624 jobjectArray keysArray,
625 jobjectArray valuesArray) {
Andreas Huber07ea4262012-04-11 12:21:20 -0700626 sp<JMediaExtractor> extractor = getMediaExtractor(env, thiz);
627
628 if (extractor == NULL) {
629 jniThrowException(env, "java/lang/IllegalStateException", NULL);
630 return;
631 }
632
633 if (pathObj == NULL) {
Andreas Huber88572f72012-02-21 11:47:18 -0800634 jniThrowException(env, "java/lang/IllegalArgumentException", NULL);
635 return;
636 }
637
Andreas Huber07ea4262012-04-11 12:21:20 -0700638 KeyedVector<String8, String8> headers;
639 if (!ConvertKeyValueArraysToKeyedVector(
640 env, keysArray, valuesArray, &headers)) {
Andreas Huber88572f72012-02-21 11:47:18 -0800641 return;
642 }
643
Andreas Huber07ea4262012-04-11 12:21:20 -0700644 const char *path = env->GetStringUTFChars(pathObj, NULL);
Andreas Huber88572f72012-02-21 11:47:18 -0800645
Andreas Huber07ea4262012-04-11 12:21:20 -0700646 if (path == NULL) {
647 return;
648 }
649
Andreas Huberd2506a52014-01-29 10:32:46 -0800650 sp<IMediaHTTPService> httpService;
651 if (httpServiceBinderObj != NULL) {
652 sp<IBinder> binder = ibinderForJavaObject(env, httpServiceBinderObj);
653 httpService = interface_cast<IMediaHTTPService>(binder);
654 }
655
656 status_t err = extractor->setDataSource(httpService, path, &headers);
Andreas Huber07ea4262012-04-11 12:21:20 -0700657
658 env->ReleaseStringUTFChars(pathObj, path);
659 path = NULL;
Andreas Huber88572f72012-02-21 11:47:18 -0800660
661 if (err != OK) {
662 jniThrowException(
663 env,
664 "java/io/IOException",
665 "Failed to instantiate extractor.");
666 return;
667 }
Andreas Huber07ea4262012-04-11 12:21:20 -0700668}
Andreas Huber88572f72012-02-21 11:47:18 -0800669
Andreas Huber07ea4262012-04-11 12:21:20 -0700670static void android_media_MediaExtractor_setDataSourceFd(
671 JNIEnv *env, jobject thiz,
672 jobject fileDescObj, jlong offset, jlong length) {
673 sp<JMediaExtractor> extractor = getMediaExtractor(env, thiz);
674
675 if (extractor == NULL) {
676 jniThrowException(env, "java/lang/IllegalStateException", NULL);
677 return;
678 }
679
680 if (fileDescObj == NULL) {
681 jniThrowException(env, "java/lang/IllegalArgumentException", NULL);
682 return;
683 }
684
685 int fd = jniGetFDFromFileDescriptor(env, fileDescObj);
686
687 status_t err = extractor->setDataSource(fd, offset, length);
688
689 if (err != OK) {
690 jniThrowException(
691 env,
692 "java/io/IOException",
693 "Failed to instantiate extractor.");
694 return;
695 }
Andreas Huber88572f72012-02-21 11:47:18 -0800696}
697
Marco Nelissenc209a062012-08-24 09:55:44 -0700698static void android_media_MediaExtractor_setDataSourceCallback(
699 JNIEnv *env, jobject thiz,
700 jobject callbackObj) {
701 sp<JMediaExtractor> extractor = getMediaExtractor(env, thiz);
702
703 if (extractor == NULL) {
704 jniThrowException(env, "java/lang/IllegalStateException", NULL);
705 return;
706 }
707
708 if (callbackObj == NULL) {
709 jniThrowException(env, "java/lang/IllegalArgumentException", NULL);
710 return;
711 }
712
Chris Watkins4eaa2932015-03-20 10:31:42 -0700713 sp<DataSource> bridge =
714 DataSource::CreateFromIDataSource(new JMediaDataSource(env, callbackObj));
Marco Nelissenc209a062012-08-24 09:55:44 -0700715 status_t err = extractor->setDataSource(bridge);
716
717 if (err != OK) {
Chong Zhangbc101852015-05-14 10:08:08 -0700718 // Clear bridge so that JMediaDataSource::close() is called _before_
719 // we throw the IOException.
720 // Otherwise close() gets called when we go out of scope, it calls
721 // Java with a pending exception and crashes the process.
722 bridge.clear();
Marco Nelissenc209a062012-08-24 09:55:44 -0700723 jniThrowException(
724 env,
725 "java/io/IOException",
726 "Failed to instantiate extractor.");
727 return;
728 }
729}
730
Andreas Huber74a78b02012-04-19 16:24:32 -0700731static jlong android_media_MediaExtractor_getCachedDurationUs(
732 JNIEnv *env, jobject thiz) {
733 sp<JMediaExtractor> extractor = getMediaExtractor(env, thiz);
734
735 if (extractor == NULL) {
736 jniThrowException(env, "java/lang/IllegalStateException", NULL);
737 return -1ll;
738 }
739
740 int64_t cachedDurationUs;
741 bool eos;
742 if (!extractor->getCachedDuration(&cachedDurationUs, &eos)) {
743 return -1ll;
744 }
745
Ashok Bhat075e9a12014-01-06 13:45:09 +0000746 return (jlong) cachedDurationUs;
Andreas Huber74a78b02012-04-19 16:24:32 -0700747}
748
749static jboolean android_media_MediaExtractor_hasCacheReachedEOS(
750 JNIEnv *env, jobject thiz) {
751 sp<JMediaExtractor> extractor = getMediaExtractor(env, thiz);
752
753 if (extractor == NULL) {
754 jniThrowException(env, "java/lang/IllegalStateException", NULL);
Ashok Bhat075e9a12014-01-06 13:45:09 +0000755 return JNI_TRUE;
Andreas Huber74a78b02012-04-19 16:24:32 -0700756 }
757
758 int64_t cachedDurationUs;
759 bool eos;
760 if (!extractor->getCachedDuration(&cachedDurationUs, &eos)) {
Ashok Bhat075e9a12014-01-06 13:45:09 +0000761 return JNI_TRUE;
Andreas Huber74a78b02012-04-19 16:24:32 -0700762 }
763
Ashok Bhat075e9a12014-01-06 13:45:09 +0000764 return eos ? JNI_TRUE : JNI_FALSE;
Andreas Huber74a78b02012-04-19 16:24:32 -0700765}
766
Andreas Huber88572f72012-02-21 11:47:18 -0800767static void android_media_MediaExtractor_native_finalize(
768 JNIEnv *env, jobject thiz) {
769 android_media_MediaExtractor_release(env, thiz);
770}
771
772static JNINativeMethod gMethods[] = {
773 { "release", "()V", (void *)android_media_MediaExtractor_release },
774
Andreas Huber60d610b2012-05-02 16:06:09 -0700775 { "getTrackCount", "()I", (void *)android_media_MediaExtractor_getTrackCount },
Andreas Huber88572f72012-02-21 11:47:18 -0800776
Marco Nelissene20a6d52013-04-08 14:28:55 -0700777 { "getFileFormatNative", "()Ljava/util/Map;",
778 (void *)android_media_MediaExtractor_getFileFormatNative },
779
Andreas Huber60d610b2012-05-02 16:06:09 -0700780 { "getTrackFormatNative", "(I)Ljava/util/Map;",
781 (void *)android_media_MediaExtractor_getTrackFormatNative },
Andreas Huber88572f72012-02-21 11:47:18 -0800782
783 { "selectTrack", "(I)V", (void *)android_media_MediaExtractor_selectTrack },
784
Andreas Huberf2855b32012-04-25 15:57:43 -0700785 { "unselectTrack", "(I)V",
786 (void *)android_media_MediaExtractor_unselectTrack },
787
788 { "seekTo", "(JI)V", (void *)android_media_MediaExtractor_seekTo },
Andreas Huber88572f72012-02-21 11:47:18 -0800789
790 { "advance", "()Z", (void *)android_media_MediaExtractor_advance },
791
792 { "readSampleData", "(Ljava/nio/ByteBuffer;I)I",
793 (void *)android_media_MediaExtractor_readSampleData },
794
795 { "getSampleTrackIndex", "()I",
796 (void *)android_media_MediaExtractor_getSampleTrackIndex },
797
798 { "getSampleTime", "()J",
799 (void *)android_media_MediaExtractor_getSampleTime },
800
Andreas Huber9b8e4962012-03-26 11:13:27 -0700801 { "getSampleFlags", "()I",
802 (void *)android_media_MediaExtractor_getSampleFlags },
803
Andreas Huber91befdc2012-04-18 12:19:51 -0700804 { "getSampleCryptoInfo", "(Landroid/media/MediaCodec$CryptoInfo;)Z",
805 (void *)android_media_MediaExtractor_getSampleCryptoInfo },
806
Andreas Huber88572f72012-02-21 11:47:18 -0800807 { "native_init", "()V", (void *)android_media_MediaExtractor_native_init },
808
Andreas Huber07ea4262012-04-11 12:21:20 -0700809 { "native_setup", "()V",
Andreas Huber88572f72012-02-21 11:47:18 -0800810 (void *)android_media_MediaExtractor_native_setup },
811
812 { "native_finalize", "()V",
813 (void *)android_media_MediaExtractor_native_finalize },
Andreas Huber07ea4262012-04-11 12:21:20 -0700814
Andreas Huberd2506a52014-01-29 10:32:46 -0800815 { "nativeSetDataSource",
816 "(Landroid/os/IBinder;Ljava/lang/String;[Ljava/lang/String;"
817 "[Ljava/lang/String;)V",
Andreas Huber07ea4262012-04-11 12:21:20 -0700818 (void *)android_media_MediaExtractor_setDataSource },
819
820 { "setDataSource", "(Ljava/io/FileDescriptor;JJ)V",
821 (void *)android_media_MediaExtractor_setDataSourceFd },
Andreas Huber74a78b02012-04-19 16:24:32 -0700822
Chris Watkins4eaa2932015-03-20 10:31:42 -0700823 { "setDataSource", "(Landroid/media/MediaDataSource;)V",
Marco Nelissenc209a062012-08-24 09:55:44 -0700824 (void *)android_media_MediaExtractor_setDataSourceCallback },
825
Andreas Huber74a78b02012-04-19 16:24:32 -0700826 { "getCachedDuration", "()J",
827 (void *)android_media_MediaExtractor_getCachedDurationUs },
828
829 { "hasCacheReachedEndOfStream", "()Z",
830 (void *)android_media_MediaExtractor_hasCacheReachedEOS },
Andreas Huber88572f72012-02-21 11:47:18 -0800831};
832
833int register_android_media_MediaExtractor(JNIEnv *env) {
834 return AndroidRuntime::registerNativeMethods(env,
835 "android/media/MediaExtractor", gMethods, NELEM(gMethods));
836}