blob: 2008f8dfdc7e712d6803729724f9d3d1a8ba2236 [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"
Ray Essickba5ce2c2017-02-07 09:50:47 -080022#include "android_media_MediaMetricsJNI.h"
Andreas Huber88572f72012-02-21 11:47:18 -080023
24#include "android_media_Utils.h"
25#include "android_runtime/AndroidRuntime.h"
Ruben Brunk87eac992013-09-09 17:44:59 -070026#include "android_runtime/Log.h"
Andreas Huber88572f72012-02-21 11:47:18 -080027#include "jni.h"
28#include "JNIHelp.h"
Chris Watkins4eaa2932015-03-20 10:31:42 -070029#include "android_media_MediaDataSource.h"
Andreas Huber88572f72012-02-21 11:47:18 -080030
Andreas Huberd2506a52014-01-29 10:32:46 -080031#include <media/IMediaHTTPService.h>
Andreas Huberab57d032012-04-18 15:59:43 -070032#include <media/hardware/CryptoAPI.h>
Andreas Huber88572f72012-02-21 11:47:18 -080033#include <media/stagefright/foundation/ABuffer.h>
34#include <media/stagefright/foundation/ADebug.h>
35#include <media/stagefright/foundation/AMessage.h>
36#include <media/stagefright/DataSource.h>
37#include <media/stagefright/MediaErrors.h>
Andreas Huber91befdc2012-04-18 12:19:51 -070038#include <media/stagefright/MetaData.h>
Andreas Huber88572f72012-02-21 11:47:18 -080039#include <media/stagefright/NuMediaExtractor.h>
40
Lajos Molnar7be24522014-07-14 12:24:56 -070041#include <nativehelper/ScopedLocalRef.h>
42
Andreas Huberd2506a52014-01-29 10:32:46 -080043#include "android_util_Binder.h"
44
Andreas Huber88572f72012-02-21 11:47:18 -080045namespace android {
46
47struct fields_t {
48 jfieldID context;
Andreas Huber91befdc2012-04-18 12:19:51 -070049
50 jmethodID cryptoInfoSetID;
Andreas Huber88572f72012-02-21 11:47:18 -080051};
52
53static fields_t gFields;
54
Andreas Huber88572f72012-02-21 11:47:18 -080055JMediaExtractor::JMediaExtractor(JNIEnv *env, jobject thiz)
56 : mClass(NULL),
57 mObject(NULL) {
58 jclass clazz = env->GetObjectClass(thiz);
59 CHECK(clazz != NULL);
60
61 mClass = (jclass)env->NewGlobalRef(clazz);
62 mObject = env->NewWeakGlobalRef(thiz);
63
64 mImpl = new NuMediaExtractor;
65}
66
67JMediaExtractor::~JMediaExtractor() {
68 JNIEnv *env = AndroidRuntime::getJNIEnv();
69
70 env->DeleteWeakGlobalRef(mObject);
71 mObject = NULL;
72 env->DeleteGlobalRef(mClass);
73 mClass = NULL;
74}
75
Andreas Huber07ea4262012-04-11 12:21:20 -070076status_t JMediaExtractor::setDataSource(
Andreas Huberd2506a52014-01-29 10:32:46 -080077 const sp<IMediaHTTPService> &httpService,
78 const char *path,
79 const KeyedVector<String8, String8> *headers) {
80 return mImpl->setDataSource(httpService, path, headers);
Andreas Huber07ea4262012-04-11 12:21:20 -070081}
82
83status_t JMediaExtractor::setDataSource(int fd, off64_t offset, off64_t size) {
84 return mImpl->setDataSource(fd, offset, size);
Andreas Huber88572f72012-02-21 11:47:18 -080085}
86
Marco Nelissenc209a062012-08-24 09:55:44 -070087status_t JMediaExtractor::setDataSource(const sp<DataSource> &datasource) {
88 return mImpl->setDataSource(datasource);
89}
90
Andreas Huber88572f72012-02-21 11:47:18 -080091size_t JMediaExtractor::countTracks() const {
92 return mImpl->countTracks();
93}
94
95status_t JMediaExtractor::getTrackFormat(size_t index, jobject *format) const {
96 sp<AMessage> msg;
97 status_t err;
98 if ((err = mImpl->getTrackFormat(index, &msg)) != OK) {
99 return err;
100 }
101
102 JNIEnv *env = AndroidRuntime::getJNIEnv();
103
104 return ConvertMessageToMap(env, msg, format);
105}
106
Marco Nelissene20a6d52013-04-08 14:28:55 -0700107status_t JMediaExtractor::getFileFormat(jobject *format) const {
108 sp<AMessage> msg;
109 status_t err;
110 if ((err = mImpl->getFileFormat(&msg)) != OK) {
111 return err;
112 }
113
114 JNIEnv *env = AndroidRuntime::getJNIEnv();
115
116 return ConvertMessageToMap(env, msg, format);
117}
118
Andreas Huber88572f72012-02-21 11:47:18 -0800119status_t JMediaExtractor::selectTrack(size_t index) {
120 return mImpl->selectTrack(index);
121}
122
Andreas Huberf2855b32012-04-25 15:57:43 -0700123status_t JMediaExtractor::unselectTrack(size_t index) {
124 return mImpl->unselectTrack(index);
125}
126
127status_t JMediaExtractor::seekTo(
128 int64_t timeUs, MediaSource::ReadOptions::SeekMode mode) {
129 return mImpl->seekTo(timeUs, mode);
Andreas Huber88572f72012-02-21 11:47:18 -0800130}
131
132status_t JMediaExtractor::advance() {
133 return mImpl->advance();
134}
135
136status_t JMediaExtractor::readSampleData(
137 jobject byteBuf, size_t offset, size_t *sampleSize) {
138 JNIEnv *env = AndroidRuntime::getJNIEnv();
139
140 void *dst = env->GetDirectBufferAddress(byteBuf);
141
Ashok Bhat075e9a12014-01-06 13:45:09 +0000142 size_t dstSize;
Andreas Huberc52b9802012-03-12 14:04:01 -0700143 jbyteArray byteArray = NULL;
144
Lajos Molnar7be24522014-07-14 12:24:56 -0700145 ScopedLocalRef<jclass> byteBufClass(env, env->FindClass("java/nio/ByteBuffer"));
146 CHECK(byteBufClass.get() != NULL);
Andreas Huberc52b9802012-03-12 14:04:01 -0700147
Lajos Molnar7be24522014-07-14 12:24:56 -0700148 if (dst == NULL) {
Andreas Huberc52b9802012-03-12 14:04:01 -0700149 jmethodID arrayID =
Lajos Molnar7be24522014-07-14 12:24:56 -0700150 env->GetMethodID(byteBufClass.get(), "array", "()[B");
Andreas Huberc52b9802012-03-12 14:04:01 -0700151 CHECK(arrayID != NULL);
152
153 byteArray =
154 (jbyteArray)env->CallObjectMethod(byteBuf, arrayID);
155
156 if (byteArray == NULL) {
157 return INVALID_OPERATION;
158 }
159
160 jboolean isCopy;
161 dst = env->GetByteArrayElements(byteArray, &isCopy);
162
Ashok Bhat075e9a12014-01-06 13:45:09 +0000163 dstSize = (size_t) env->GetArrayLength(byteArray);
Andreas Huberc52b9802012-03-12 14:04:01 -0700164 } else {
Ashok Bhat075e9a12014-01-06 13:45:09 +0000165 dstSize = (size_t) env->GetDirectBufferCapacity(byteBuf);
Andreas Huber88572f72012-02-21 11:47:18 -0800166 }
167
Andreas Huber88572f72012-02-21 11:47:18 -0800168 if (dstSize < offset) {
Andreas Huberc52b9802012-03-12 14:04:01 -0700169 if (byteArray != NULL) {
170 env->ReleaseByteArrayElements(byteArray, (jbyte *)dst, 0);
171 }
172
Andreas Huber88572f72012-02-21 11:47:18 -0800173 return -ERANGE;
174 }
175
176 sp<ABuffer> buffer = new ABuffer((char *)dst + offset, dstSize - offset);
177
178 status_t err = mImpl->readSampleData(buffer);
179
Andreas Huberc52b9802012-03-12 14:04:01 -0700180 if (byteArray != NULL) {
181 env->ReleaseByteArrayElements(byteArray, (jbyte *)dst, 0);
182 }
183
Andreas Huber88572f72012-02-21 11:47:18 -0800184 if (err != OK) {
185 return err;
186 }
187
188 *sampleSize = buffer->size();
189
Lajos Molnar7be24522014-07-14 12:24:56 -0700190 jmethodID positionID = env->GetMethodID(
191 byteBufClass.get(), "position", "(I)Ljava/nio/Buffer;");
192
193 CHECK(positionID != NULL);
194
195 jmethodID limitID = env->GetMethodID(
196 byteBufClass.get(), "limit", "(I)Ljava/nio/Buffer;");
197
198 CHECK(limitID != NULL);
199
200 jobject me = env->CallObjectMethod(
201 byteBuf, limitID, offset + *sampleSize);
202 env->DeleteLocalRef(me);
203 me = env->CallObjectMethod(
204 byteBuf, positionID, offset);
205 env->DeleteLocalRef(me);
206 me = NULL;
207
Andreas Huber88572f72012-02-21 11:47:18 -0800208 return OK;
209}
210
211status_t JMediaExtractor::getSampleTrackIndex(size_t *trackIndex) {
212 return mImpl->getSampleTrackIndex(trackIndex);
213}
214
215status_t JMediaExtractor::getSampleTime(int64_t *sampleTimeUs) {
216 return mImpl->getSampleTime(sampleTimeUs);
217}
218
Andreas Huber9b8e4962012-03-26 11:13:27 -0700219status_t JMediaExtractor::getSampleFlags(uint32_t *sampleFlags) {
Andreas Huber91befdc2012-04-18 12:19:51 -0700220 *sampleFlags = 0;
221
222 sp<MetaData> meta;
223 status_t err = mImpl->getSampleMeta(&meta);
224
225 if (err != OK) {
226 return err;
227 }
228
229 int32_t val;
230 if (meta->findInt32(kKeyIsSyncFrame, &val) && val != 0) {
231 (*sampleFlags) |= NuMediaExtractor::SAMPLE_FLAG_SYNC;
232 }
233
234 uint32_t type;
235 const void *data;
236 size_t size;
237 if (meta->findData(kKeyEncryptedSizes, &type, &data, &size)) {
238 (*sampleFlags) |= NuMediaExtractor::SAMPLE_FLAG_ENCRYPTED;
239 }
240
241 return OK;
242}
243
Ray Essickba5ce2c2017-02-07 09:50:47 -0800244status_t JMediaExtractor::getMetrics(Parcel *reply) const {
245
246 status_t status = mImpl->getMetrics(reply);
247 return status;
248}
249
250
Andreas Huber91befdc2012-04-18 12:19:51 -0700251status_t JMediaExtractor::getSampleMeta(sp<MetaData> *sampleMeta) {
252 return mImpl->getSampleMeta(sampleMeta);
Andreas Huber9b8e4962012-03-26 11:13:27 -0700253}
254
Andreas Huber74a78b02012-04-19 16:24:32 -0700255bool JMediaExtractor::getCachedDuration(int64_t *durationUs, bool *eos) const {
256 return mImpl->getCachedDuration(durationUs, eos);
257}
258
Andreas Huber88572f72012-02-21 11:47:18 -0800259} // namespace android
260
261////////////////////////////////////////////////////////////////////////////////
262
263using namespace android;
264
265static sp<JMediaExtractor> setMediaExtractor(
266 JNIEnv *env, jobject thiz, const sp<JMediaExtractor> &extractor) {
267 sp<JMediaExtractor> old =
Ashok Bhat075e9a12014-01-06 13:45:09 +0000268 (JMediaExtractor *)env->GetLongField(thiz, gFields.context);
Andreas Huber88572f72012-02-21 11:47:18 -0800269
270 if (extractor != NULL) {
271 extractor->incStrong(thiz);
272 }
273 if (old != NULL) {
274 old->decStrong(thiz);
275 }
Ashok Bhat075e9a12014-01-06 13:45:09 +0000276 env->SetLongField(thiz, gFields.context, (jlong)extractor.get());
Andreas Huber88572f72012-02-21 11:47:18 -0800277
278 return old;
279}
280
281static sp<JMediaExtractor> getMediaExtractor(JNIEnv *env, jobject thiz) {
Ashok Bhat075e9a12014-01-06 13:45:09 +0000282 return (JMediaExtractor *)env->GetLongField(thiz, gFields.context);
Andreas Huber88572f72012-02-21 11:47:18 -0800283}
284
285static void android_media_MediaExtractor_release(JNIEnv *env, jobject thiz) {
286 setMediaExtractor(env, thiz, NULL);
287}
288
Andreas Huber60d610b2012-05-02 16:06:09 -0700289static jint android_media_MediaExtractor_getTrackCount(
Andreas Huber88572f72012-02-21 11:47:18 -0800290 JNIEnv *env, jobject thiz) {
291 sp<JMediaExtractor> extractor = getMediaExtractor(env, thiz);
292
293 if (extractor == NULL) {
294 jniThrowException(env, "java/lang/IllegalStateException", NULL);
Andreas Huber07ea4262012-04-11 12:21:20 -0700295 return -1;
Andreas Huber88572f72012-02-21 11:47:18 -0800296 }
297
Ashok Bhat075e9a12014-01-06 13:45:09 +0000298 return (jint) extractor->countTracks();
Andreas Huber88572f72012-02-21 11:47:18 -0800299}
300
Andreas Huber60d610b2012-05-02 16:06:09 -0700301static jobject android_media_MediaExtractor_getTrackFormatNative(
Andreas Huber88572f72012-02-21 11:47:18 -0800302 JNIEnv *env, jobject thiz, jint index) {
303 sp<JMediaExtractor> extractor = getMediaExtractor(env, thiz);
304
305 if (extractor == NULL) {
306 jniThrowException(env, "java/lang/IllegalStateException", NULL);
307 return NULL;
308 }
309
310 jobject format;
311 status_t err = extractor->getTrackFormat(index, &format);
312
313 if (err != OK) {
314 jniThrowException(env, "java/lang/IllegalArgumentException", NULL);
315 return NULL;
316 }
317
318 return format;
319}
320
Marco Nelissene20a6d52013-04-08 14:28:55 -0700321static jobject android_media_MediaExtractor_getFileFormatNative(
322 JNIEnv *env, jobject thiz) {
323 sp<JMediaExtractor> extractor = getMediaExtractor(env, thiz);
324
325 if (extractor == NULL) {
326 jniThrowException(env, "java/lang/IllegalStateException", NULL);
327 return NULL;
328 }
329
330 jobject format;
331 status_t err = extractor->getFileFormat(&format);
332
333 if (err != OK) {
334 jniThrowException(env, "java/lang/IllegalArgumentException", NULL);
335 return NULL;
336 }
337
338 return format;
339}
340
Andreas Huber88572f72012-02-21 11:47:18 -0800341static void android_media_MediaExtractor_selectTrack(
342 JNIEnv *env, jobject thiz, jint index) {
343 sp<JMediaExtractor> extractor = getMediaExtractor(env, thiz);
344
345 if (extractor == NULL) {
346 jniThrowException(env, "java/lang/IllegalStateException", NULL);
347 return;
348 }
349
350 status_t err = extractor->selectTrack(index);
351
352 if (err != OK) {
353 jniThrowException(env, "java/lang/IllegalArgumentException", NULL);
354 return;
355 }
356}
357
Andreas Huberf2855b32012-04-25 15:57:43 -0700358static void android_media_MediaExtractor_unselectTrack(
359 JNIEnv *env, jobject thiz, jint index) {
Andreas Huber88572f72012-02-21 11:47:18 -0800360 sp<JMediaExtractor> extractor = getMediaExtractor(env, thiz);
361
362 if (extractor == NULL) {
363 jniThrowException(env, "java/lang/IllegalStateException", NULL);
364 return;
365 }
366
Andreas Huberf2855b32012-04-25 15:57:43 -0700367 status_t err = extractor->unselectTrack(index);
368
369 if (err != OK) {
370 jniThrowException(env, "java/lang/IllegalArgumentException", NULL);
371 return;
372 }
373}
374
375static void android_media_MediaExtractor_seekTo(
376 JNIEnv *env, jobject thiz, jlong timeUs, jint mode) {
377 sp<JMediaExtractor> extractor = getMediaExtractor(env, thiz);
378
379 if (extractor == NULL) {
380 jniThrowException(env, "java/lang/IllegalStateException", NULL);
381 return;
382 }
383
384 if (mode < MediaSource::ReadOptions::SEEK_PREVIOUS_SYNC
Andreas Huberf86374a2012-05-09 11:25:29 -0700385 || mode >= MediaSource::ReadOptions::SEEK_CLOSEST) {
Andreas Huberf2855b32012-04-25 15:57:43 -0700386 jniThrowException(env, "java/lang/IllegalArgumentException", NULL);
387 return;
388 }
389
390 extractor->seekTo(timeUs, (MediaSource::ReadOptions::SeekMode)mode);
Andreas Huber88572f72012-02-21 11:47:18 -0800391}
392
393static jboolean android_media_MediaExtractor_advance(
394 JNIEnv *env, jobject thiz) {
395 sp<JMediaExtractor> extractor = getMediaExtractor(env, thiz);
396
397 if (extractor == NULL) {
398 jniThrowException(env, "java/lang/IllegalStateException", NULL);
Ashok Bhat075e9a12014-01-06 13:45:09 +0000399 return JNI_FALSE;
Andreas Huber88572f72012-02-21 11:47:18 -0800400 }
401
402 status_t err = extractor->advance();
403
404 if (err == ERROR_END_OF_STREAM) {
Ashok Bhat075e9a12014-01-06 13:45:09 +0000405 return JNI_FALSE;
Andreas Huber88572f72012-02-21 11:47:18 -0800406 } else if (err != OK) {
407 jniThrowException(env, "java/lang/IllegalArgumentException", NULL);
Ashok Bhat075e9a12014-01-06 13:45:09 +0000408 return JNI_FALSE;
Andreas Huber88572f72012-02-21 11:47:18 -0800409 }
410
Ashok Bhat075e9a12014-01-06 13:45:09 +0000411 return JNI_TRUE;
Andreas Huber88572f72012-02-21 11:47:18 -0800412}
413
414static jint android_media_MediaExtractor_readSampleData(
415 JNIEnv *env, jobject thiz, jobject byteBuf, jint offset) {
416 sp<JMediaExtractor> extractor = getMediaExtractor(env, thiz);
417
418 if (extractor == NULL) {
419 jniThrowException(env, "java/lang/IllegalStateException", NULL);
420 return -1;
421 }
422
423 size_t sampleSize;
424 status_t err = extractor->readSampleData(byteBuf, offset, &sampleSize);
425
426 if (err == ERROR_END_OF_STREAM) {
427 return -1;
428 } else if (err != OK) {
429 jniThrowException(env, "java/lang/IllegalArgumentException", NULL);
Ashok Bhat075e9a12014-01-06 13:45:09 +0000430 return -1;
Andreas Huber88572f72012-02-21 11:47:18 -0800431 }
432
Ashok Bhat075e9a12014-01-06 13:45:09 +0000433 return (jint) sampleSize;
Andreas Huber88572f72012-02-21 11:47:18 -0800434}
435
436static jint android_media_MediaExtractor_getSampleTrackIndex(
437 JNIEnv *env, jobject thiz) {
438 sp<JMediaExtractor> extractor = getMediaExtractor(env, thiz);
439
440 if (extractor == NULL) {
441 jniThrowException(env, "java/lang/IllegalStateException", NULL);
442 return -1;
443 }
444
445 size_t trackIndex;
446 status_t err = extractor->getSampleTrackIndex(&trackIndex);
447
448 if (err == ERROR_END_OF_STREAM) {
449 return -1;
450 } else if (err != OK) {
451 jniThrowException(env, "java/lang/IllegalArgumentException", NULL);
Ashok Bhat075e9a12014-01-06 13:45:09 +0000452 return -1;
Andreas Huber88572f72012-02-21 11:47:18 -0800453 }
454
Ashok Bhat075e9a12014-01-06 13:45:09 +0000455 return (jint) trackIndex;
Andreas Huber88572f72012-02-21 11:47:18 -0800456}
457
458static jlong android_media_MediaExtractor_getSampleTime(
459 JNIEnv *env, jobject thiz) {
460 sp<JMediaExtractor> extractor = getMediaExtractor(env, thiz);
461
462 if (extractor == NULL) {
463 jniThrowException(env, "java/lang/IllegalStateException", NULL);
464 return -1ll;
465 }
466
467 int64_t sampleTimeUs;
468 status_t err = extractor->getSampleTime(&sampleTimeUs);
469
470 if (err == ERROR_END_OF_STREAM) {
471 return -1ll;
472 } else if (err != OK) {
473 jniThrowException(env, "java/lang/IllegalArgumentException", NULL);
Ashok Bhat075e9a12014-01-06 13:45:09 +0000474 return -1ll;
Andreas Huber88572f72012-02-21 11:47:18 -0800475 }
476
Ashok Bhat075e9a12014-01-06 13:45:09 +0000477 return (jlong) sampleTimeUs;
Andreas Huber88572f72012-02-21 11:47:18 -0800478}
479
Andreas Huber9b8e4962012-03-26 11:13:27 -0700480static jint android_media_MediaExtractor_getSampleFlags(
481 JNIEnv *env, jobject thiz) {
482 sp<JMediaExtractor> extractor = getMediaExtractor(env, thiz);
483
484 if (extractor == NULL) {
485 jniThrowException(env, "java/lang/IllegalStateException", NULL);
Ashok Bhat075e9a12014-01-06 13:45:09 +0000486 return -1;
Andreas Huber9b8e4962012-03-26 11:13:27 -0700487 }
488
489 uint32_t sampleFlags;
490 status_t err = extractor->getSampleFlags(&sampleFlags);
491
492 if (err == ERROR_END_OF_STREAM) {
Ashok Bhat075e9a12014-01-06 13:45:09 +0000493 return -1;
Andreas Huber9b8e4962012-03-26 11:13:27 -0700494 } else if (err != OK) {
495 jniThrowException(env, "java/lang/IllegalArgumentException", NULL);
Ashok Bhat075e9a12014-01-06 13:45:09 +0000496 return -1;
Andreas Huber9b8e4962012-03-26 11:13:27 -0700497 }
498
Ashok Bhat075e9a12014-01-06 13:45:09 +0000499 return (jint) sampleFlags;
Andreas Huber9b8e4962012-03-26 11:13:27 -0700500}
501
Andreas Huber91befdc2012-04-18 12:19:51 -0700502static jboolean android_media_MediaExtractor_getSampleCryptoInfo(
503 JNIEnv *env, jobject thiz, jobject cryptoInfoObj) {
504 sp<JMediaExtractor> extractor = getMediaExtractor(env, thiz);
505
506 if (extractor == NULL) {
507 jniThrowException(env, "java/lang/IllegalStateException", NULL);
Ashok Bhat075e9a12014-01-06 13:45:09 +0000508 return JNI_FALSE;
Andreas Huber91befdc2012-04-18 12:19:51 -0700509 }
510
511 sp<MetaData> meta;
512 status_t err = extractor->getSampleMeta(&meta);
513
514 if (err != OK) {
Ashok Bhat075e9a12014-01-06 13:45:09 +0000515 return JNI_FALSE;
Andreas Huber91befdc2012-04-18 12:19:51 -0700516 }
517
518 uint32_t type;
519 const void *data;
520 size_t size;
521 if (!meta->findData(kKeyEncryptedSizes, &type, &data, &size)) {
Ashok Bhat075e9a12014-01-06 13:45:09 +0000522 return JNI_FALSE;
Andreas Huber91befdc2012-04-18 12:19:51 -0700523 }
524
Ashok Bhate92416d2014-03-05 11:51:15 +0000525 size_t numSubSamples = size / sizeof(int32_t);
Andreas Huber91befdc2012-04-18 12:19:51 -0700526
527 if (numSubSamples == 0) {
Ashok Bhat075e9a12014-01-06 13:45:09 +0000528 return JNI_FALSE;
Andreas Huber91befdc2012-04-18 12:19:51 -0700529 }
530
531 jintArray numBytesOfEncryptedDataObj = env->NewIntArray(numSubSamples);
532 jboolean isCopy;
533 jint *dst = env->GetIntArrayElements(numBytesOfEncryptedDataObj, &isCopy);
534 for (size_t i = 0; i < numSubSamples; ++i) {
Ashok Bhate92416d2014-03-05 11:51:15 +0000535 dst[i] = ((const int32_t *)data)[i];
Andreas Huber91befdc2012-04-18 12:19:51 -0700536 }
537 env->ReleaseIntArrayElements(numBytesOfEncryptedDataObj, dst, 0);
538 dst = NULL;
539
540 size_t encSize = size;
541 jintArray numBytesOfPlainDataObj = NULL;
Andreas Huberab57d032012-04-18 15:59:43 -0700542 if (meta->findData(kKeyPlainSizes, &type, &data, &size)) {
Andreas Huber91befdc2012-04-18 12:19:51 -0700543 if (size != encSize) {
544 // The two must be of the same length.
Ashok Bhat075e9a12014-01-06 13:45:09 +0000545 return JNI_FALSE;
Andreas Huber91befdc2012-04-18 12:19:51 -0700546 }
547
548 numBytesOfPlainDataObj = env->NewIntArray(numSubSamples);
549 jboolean isCopy;
550 jint *dst = env->GetIntArrayElements(numBytesOfPlainDataObj, &isCopy);
551 for (size_t i = 0; i < numSubSamples; ++i) {
Ashok Bhate92416d2014-03-05 11:51:15 +0000552 dst[i] = ((const int32_t *)data)[i];
Andreas Huber91befdc2012-04-18 12:19:51 -0700553 }
554 env->ReleaseIntArrayElements(numBytesOfPlainDataObj, dst, 0);
555 dst = NULL;
556 }
557
558 jbyteArray keyObj = NULL;
559 if (meta->findData(kKeyCryptoKey, &type, &data, &size)) {
560 if (size != 16) {
561 // Keys must be 16 bytes in length.
Ashok Bhat075e9a12014-01-06 13:45:09 +0000562 return JNI_FALSE;
Andreas Huber91befdc2012-04-18 12:19:51 -0700563 }
564
565 keyObj = env->NewByteArray(size);
566 jboolean isCopy;
567 jbyte *dst = env->GetByteArrayElements(keyObj, &isCopy);
568 memcpy(dst, data, size);
569 env->ReleaseByteArrayElements(keyObj, dst, 0);
570 dst = NULL;
571 }
572
573 jbyteArray ivObj = NULL;
574 if (meta->findData(kKeyCryptoIV, &type, &data, &size)) {
575 if (size != 16) {
576 // IVs must be 16 bytes in length.
Ashok Bhat075e9a12014-01-06 13:45:09 +0000577 return JNI_FALSE;
Andreas Huber91befdc2012-04-18 12:19:51 -0700578 }
579
580 ivObj = env->NewByteArray(size);
581 jboolean isCopy;
582 jbyte *dst = env->GetByteArrayElements(ivObj, &isCopy);
583 memcpy(dst, data, size);
584 env->ReleaseByteArrayElements(ivObj, dst, 0);
585 dst = NULL;
586 }
587
588 int32_t mode;
589 if (!meta->findInt32(kKeyCryptoMode, &mode)) {
Andreas Huberab57d032012-04-18 15:59:43 -0700590 mode = CryptoPlugin::kMode_AES_CTR;
Andreas Huber91befdc2012-04-18 12:19:51 -0700591 }
592
593 env->CallVoidMethod(
594 cryptoInfoObj,
595 gFields.cryptoInfoSetID,
Ashok Bhat136c08a2014-03-05 15:23:13 +0000596 (jint)numSubSamples,
Andreas Huber91befdc2012-04-18 12:19:51 -0700597 numBytesOfPlainDataObj,
598 numBytesOfEncryptedDataObj,
599 keyObj,
600 ivObj,
601 mode);
602
Ashok Bhat075e9a12014-01-06 13:45:09 +0000603 return JNI_TRUE;
Andreas Huber91befdc2012-04-18 12:19:51 -0700604}
605
Andreas Huber88572f72012-02-21 11:47:18 -0800606static void android_media_MediaExtractor_native_init(JNIEnv *env) {
607 jclass clazz = env->FindClass("android/media/MediaExtractor");
608 CHECK(clazz != NULL);
609
Ashok Bhat075e9a12014-01-06 13:45:09 +0000610 gFields.context = env->GetFieldID(clazz, "mNativeContext", "J");
Andreas Huber88572f72012-02-21 11:47:18 -0800611 CHECK(gFields.context != NULL);
612
Andreas Huber91befdc2012-04-18 12:19:51 -0700613 clazz = env->FindClass("android/media/MediaCodec$CryptoInfo");
614 CHECK(clazz != NULL);
615
616 gFields.cryptoInfoSetID =
617 env->GetMethodID(clazz, "set", "(I[I[I[B[BI)V");
Andreas Huber88572f72012-02-21 11:47:18 -0800618}
619
620static void android_media_MediaExtractor_native_setup(
Andreas Huber07ea4262012-04-11 12:21:20 -0700621 JNIEnv *env, jobject thiz) {
Andreas Huber88572f72012-02-21 11:47:18 -0800622 sp<JMediaExtractor> extractor = new JMediaExtractor(env, thiz);
Andreas Huber07ea4262012-04-11 12:21:20 -0700623 setMediaExtractor(env,thiz, extractor);
624}
Andreas Huber88572f72012-02-21 11:47:18 -0800625
Andreas Huber07ea4262012-04-11 12:21:20 -0700626static void android_media_MediaExtractor_setDataSource(
627 JNIEnv *env, jobject thiz,
Andreas Huberd2506a52014-01-29 10:32:46 -0800628 jobject httpServiceBinderObj,
629 jstring pathObj,
630 jobjectArray keysArray,
631 jobjectArray valuesArray) {
Andreas Huber07ea4262012-04-11 12:21:20 -0700632 sp<JMediaExtractor> extractor = getMediaExtractor(env, thiz);
633
634 if (extractor == NULL) {
635 jniThrowException(env, "java/lang/IllegalStateException", NULL);
636 return;
637 }
638
639 if (pathObj == NULL) {
Andreas Huber88572f72012-02-21 11:47:18 -0800640 jniThrowException(env, "java/lang/IllegalArgumentException", NULL);
641 return;
642 }
643
Andreas Huber07ea4262012-04-11 12:21:20 -0700644 KeyedVector<String8, String8> headers;
645 if (!ConvertKeyValueArraysToKeyedVector(
646 env, keysArray, valuesArray, &headers)) {
Andreas Huber88572f72012-02-21 11:47:18 -0800647 return;
648 }
649
Andreas Huber07ea4262012-04-11 12:21:20 -0700650 const char *path = env->GetStringUTFChars(pathObj, NULL);
Andreas Huber88572f72012-02-21 11:47:18 -0800651
Andreas Huber07ea4262012-04-11 12:21:20 -0700652 if (path == NULL) {
653 return;
654 }
655
Andreas Huberd2506a52014-01-29 10:32:46 -0800656 sp<IMediaHTTPService> httpService;
657 if (httpServiceBinderObj != NULL) {
658 sp<IBinder> binder = ibinderForJavaObject(env, httpServiceBinderObj);
659 httpService = interface_cast<IMediaHTTPService>(binder);
660 }
661
662 status_t err = extractor->setDataSource(httpService, path, &headers);
Andreas Huber07ea4262012-04-11 12:21:20 -0700663
664 env->ReleaseStringUTFChars(pathObj, path);
665 path = NULL;
Andreas Huber88572f72012-02-21 11:47:18 -0800666
667 if (err != OK) {
668 jniThrowException(
669 env,
670 "java/io/IOException",
671 "Failed to instantiate extractor.");
672 return;
673 }
Andreas Huber07ea4262012-04-11 12:21:20 -0700674}
Andreas Huber88572f72012-02-21 11:47:18 -0800675
Andreas Huber07ea4262012-04-11 12:21:20 -0700676static void android_media_MediaExtractor_setDataSourceFd(
677 JNIEnv *env, jobject thiz,
678 jobject fileDescObj, jlong offset, jlong length) {
679 sp<JMediaExtractor> extractor = getMediaExtractor(env, thiz);
680
681 if (extractor == NULL) {
682 jniThrowException(env, "java/lang/IllegalStateException", NULL);
683 return;
684 }
685
686 if (fileDescObj == NULL) {
687 jniThrowException(env, "java/lang/IllegalArgumentException", NULL);
688 return;
689 }
690
691 int fd = jniGetFDFromFileDescriptor(env, fileDescObj);
692
693 status_t err = extractor->setDataSource(fd, offset, length);
694
695 if (err != OK) {
696 jniThrowException(
697 env,
698 "java/io/IOException",
699 "Failed to instantiate extractor.");
700 return;
701 }
Andreas Huber88572f72012-02-21 11:47:18 -0800702}
703
Marco Nelissenc209a062012-08-24 09:55:44 -0700704static void android_media_MediaExtractor_setDataSourceCallback(
705 JNIEnv *env, jobject thiz,
706 jobject callbackObj) {
707 sp<JMediaExtractor> extractor = getMediaExtractor(env, thiz);
708
709 if (extractor == NULL) {
710 jniThrowException(env, "java/lang/IllegalStateException", NULL);
711 return;
712 }
713
714 if (callbackObj == NULL) {
715 jniThrowException(env, "java/lang/IllegalArgumentException", NULL);
716 return;
717 }
718
Chris Watkins4eaa2932015-03-20 10:31:42 -0700719 sp<DataSource> bridge =
720 DataSource::CreateFromIDataSource(new JMediaDataSource(env, callbackObj));
Marco Nelissenc209a062012-08-24 09:55:44 -0700721 status_t err = extractor->setDataSource(bridge);
722
723 if (err != OK) {
Chong Zhangbc101852015-05-14 10:08:08 -0700724 // Clear bridge so that JMediaDataSource::close() is called _before_
725 // we throw the IOException.
726 // Otherwise close() gets called when we go out of scope, it calls
727 // Java with a pending exception and crashes the process.
728 bridge.clear();
Marco Nelissenc209a062012-08-24 09:55:44 -0700729 jniThrowException(
730 env,
731 "java/io/IOException",
732 "Failed to instantiate extractor.");
733 return;
734 }
735}
736
Andreas Huber74a78b02012-04-19 16:24:32 -0700737static jlong android_media_MediaExtractor_getCachedDurationUs(
738 JNIEnv *env, jobject thiz) {
739 sp<JMediaExtractor> extractor = getMediaExtractor(env, thiz);
740
741 if (extractor == NULL) {
742 jniThrowException(env, "java/lang/IllegalStateException", NULL);
743 return -1ll;
744 }
745
746 int64_t cachedDurationUs;
747 bool eos;
748 if (!extractor->getCachedDuration(&cachedDurationUs, &eos)) {
749 return -1ll;
750 }
751
Ashok Bhat075e9a12014-01-06 13:45:09 +0000752 return (jlong) cachedDurationUs;
Andreas Huber74a78b02012-04-19 16:24:32 -0700753}
754
755static jboolean android_media_MediaExtractor_hasCacheReachedEOS(
756 JNIEnv *env, jobject thiz) {
757 sp<JMediaExtractor> extractor = getMediaExtractor(env, thiz);
758
759 if (extractor == NULL) {
760 jniThrowException(env, "java/lang/IllegalStateException", NULL);
Ashok Bhat075e9a12014-01-06 13:45:09 +0000761 return JNI_TRUE;
Andreas Huber74a78b02012-04-19 16:24:32 -0700762 }
763
764 int64_t cachedDurationUs;
765 bool eos;
766 if (!extractor->getCachedDuration(&cachedDurationUs, &eos)) {
Ashok Bhat075e9a12014-01-06 13:45:09 +0000767 return JNI_TRUE;
Andreas Huber74a78b02012-04-19 16:24:32 -0700768 }
769
Ashok Bhat075e9a12014-01-06 13:45:09 +0000770 return eos ? JNI_TRUE : JNI_FALSE;
Andreas Huber74a78b02012-04-19 16:24:32 -0700771}
772
Andreas Huber88572f72012-02-21 11:47:18 -0800773static void android_media_MediaExtractor_native_finalize(
774 JNIEnv *env, jobject thiz) {
775 android_media_MediaExtractor_release(env, thiz);
776}
777
Ray Essickba5ce2c2017-02-07 09:50:47 -0800778static jobject
779android_media_MediaExtractor_getMetrics(JNIEnv * env, jobject thiz)
780{
781 ALOGV("android_media_MediaExtractor_getMetrics");
782
783 sp<JMediaExtractor> extractor = getMediaExtractor(env, thiz);
784 if (extractor == NULL ) {
785 jniThrowException(env, "java/lang/IllegalStateException", NULL);
786 return NULL;
787 }
788
789 // get what we have for the metrics from the codec
790 Parcel reply;
791 status_t err = extractor->getMetrics(&reply);
792 if (err != OK) {
793 ALOGE("getMetrics failed");
794 return (jobject) NULL;
795 }
796
797 // build and return the Bundle
798 MediaAnalyticsItem *item = new MediaAnalyticsItem;
799 item->readFromParcel(reply);
800 jobject mybundle = MediaMetricsJNI::writeMetricsToBundle(env, item, NULL);
801
802 // housekeeping
803 delete item;
804 item = NULL;
805
806 return mybundle;
807}
808
809
Daniel Micay76f6a862015-09-19 17:31:01 -0400810static const JNINativeMethod gMethods[] = {
Andreas Huber88572f72012-02-21 11:47:18 -0800811 { "release", "()V", (void *)android_media_MediaExtractor_release },
812
Andreas Huber60d610b2012-05-02 16:06:09 -0700813 { "getTrackCount", "()I", (void *)android_media_MediaExtractor_getTrackCount },
Andreas Huber88572f72012-02-21 11:47:18 -0800814
Marco Nelissene20a6d52013-04-08 14:28:55 -0700815 { "getFileFormatNative", "()Ljava/util/Map;",
816 (void *)android_media_MediaExtractor_getFileFormatNative },
817
Andreas Huber60d610b2012-05-02 16:06:09 -0700818 { "getTrackFormatNative", "(I)Ljava/util/Map;",
819 (void *)android_media_MediaExtractor_getTrackFormatNative },
Andreas Huber88572f72012-02-21 11:47:18 -0800820
821 { "selectTrack", "(I)V", (void *)android_media_MediaExtractor_selectTrack },
822
Andreas Huberf2855b32012-04-25 15:57:43 -0700823 { "unselectTrack", "(I)V",
824 (void *)android_media_MediaExtractor_unselectTrack },
825
826 { "seekTo", "(JI)V", (void *)android_media_MediaExtractor_seekTo },
Andreas Huber88572f72012-02-21 11:47:18 -0800827
828 { "advance", "()Z", (void *)android_media_MediaExtractor_advance },
829
830 { "readSampleData", "(Ljava/nio/ByteBuffer;I)I",
831 (void *)android_media_MediaExtractor_readSampleData },
832
833 { "getSampleTrackIndex", "()I",
834 (void *)android_media_MediaExtractor_getSampleTrackIndex },
835
836 { "getSampleTime", "()J",
837 (void *)android_media_MediaExtractor_getSampleTime },
838
Andreas Huber9b8e4962012-03-26 11:13:27 -0700839 { "getSampleFlags", "()I",
840 (void *)android_media_MediaExtractor_getSampleFlags },
841
Andreas Huber91befdc2012-04-18 12:19:51 -0700842 { "getSampleCryptoInfo", "(Landroid/media/MediaCodec$CryptoInfo;)Z",
843 (void *)android_media_MediaExtractor_getSampleCryptoInfo },
844
Andreas Huber88572f72012-02-21 11:47:18 -0800845 { "native_init", "()V", (void *)android_media_MediaExtractor_native_init },
846
Andreas Huber07ea4262012-04-11 12:21:20 -0700847 { "native_setup", "()V",
Andreas Huber88572f72012-02-21 11:47:18 -0800848 (void *)android_media_MediaExtractor_native_setup },
849
850 { "native_finalize", "()V",
851 (void *)android_media_MediaExtractor_native_finalize },
Andreas Huber07ea4262012-04-11 12:21:20 -0700852
Andreas Huberd2506a52014-01-29 10:32:46 -0800853 { "nativeSetDataSource",
854 "(Landroid/os/IBinder;Ljava/lang/String;[Ljava/lang/String;"
855 "[Ljava/lang/String;)V",
Andreas Huber07ea4262012-04-11 12:21:20 -0700856 (void *)android_media_MediaExtractor_setDataSource },
857
858 { "setDataSource", "(Ljava/io/FileDescriptor;JJ)V",
859 (void *)android_media_MediaExtractor_setDataSourceFd },
Andreas Huber74a78b02012-04-19 16:24:32 -0700860
Chris Watkins4eaa2932015-03-20 10:31:42 -0700861 { "setDataSource", "(Landroid/media/MediaDataSource;)V",
Marco Nelissenc209a062012-08-24 09:55:44 -0700862 (void *)android_media_MediaExtractor_setDataSourceCallback },
863
Andreas Huber74a78b02012-04-19 16:24:32 -0700864 { "getCachedDuration", "()J",
865 (void *)android_media_MediaExtractor_getCachedDurationUs },
866
867 { "hasCacheReachedEndOfStream", "()Z",
868 (void *)android_media_MediaExtractor_hasCacheReachedEOS },
Ray Essickba5ce2c2017-02-07 09:50:47 -0800869
870 {"getMetrics", "()Landroid/os/Bundle;",
871 (void *)android_media_MediaExtractor_getMetrics},
Andreas Huber88572f72012-02-21 11:47:18 -0800872};
873
874int register_android_media_MediaExtractor(JNIEnv *env) {
875 return AndroidRuntime::registerNativeMethods(env,
876 "android/media/MediaExtractor", gMethods, NELEM(gMethods));
877}