From 78e0545774a767c1060ebf9f16d14a04902bc571 Mon Sep 17 00:00:00 2001 From: Oliver Woodman Date: Mon, 10 Jul 2017 12:16:20 +0100 Subject: [PATCH] Re-add RTMP extension classes --- extensions/rtmp/src/main/AndroidManifest.xml | 17 ++++ .../exoplayer2/ext/rtmp/RtmpDataSource.java | 92 +++++++++++++++++++ .../ext/rtmp/RtmpDataSourceFactory.java | 47 ++++++++++ 3 files changed, 156 insertions(+) create mode 100644 extensions/rtmp/src/main/AndroidManifest.xml create mode 100644 extensions/rtmp/src/main/java/com/google/android/exoplayer2/ext/rtmp/RtmpDataSource.java create mode 100644 extensions/rtmp/src/main/java/com/google/android/exoplayer2/ext/rtmp/RtmpDataSourceFactory.java diff --git a/extensions/rtmp/src/main/AndroidManifest.xml b/extensions/rtmp/src/main/AndroidManifest.xml new file mode 100644 index 0000000000..7c5e92c198 --- /dev/null +++ b/extensions/rtmp/src/main/AndroidManifest.xml @@ -0,0 +1,17 @@ + + + + diff --git a/extensions/rtmp/src/main/java/com/google/android/exoplayer2/ext/rtmp/RtmpDataSource.java b/extensions/rtmp/src/main/java/com/google/android/exoplayer2/ext/rtmp/RtmpDataSource.java new file mode 100644 index 0000000000..122c44a39f --- /dev/null +++ b/extensions/rtmp/src/main/java/com/google/android/exoplayer2/ext/rtmp/RtmpDataSource.java @@ -0,0 +1,92 @@ +/* + * Copyright (C) 2017 The Android Open Source Project + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package com.google.android.exoplayer2.ext.rtmp; + +import android.net.Uri; +import android.support.annotation.Nullable; +import com.google.android.exoplayer2.C; +import com.google.android.exoplayer2.upstream.DataSource; +import com.google.android.exoplayer2.upstream.DataSpec; +import com.google.android.exoplayer2.upstream.TransferListener; +import java.io.IOException; +import net.butterflytv.rtmp_client.RtmpClient; +import net.butterflytv.rtmp_client.RtmpClient.RtmpIOException; + +/** + * A Real-Time Messaging Protocol (RTMP) {@link DataSource}. + */ +public final class RtmpDataSource implements DataSource { + + @Nullable private final TransferListener listener; + + private RtmpClient rtmpClient; + private Uri uri; + + public RtmpDataSource() { + this(null); + } + + /** + * @param listener An optional listener. + */ + public RtmpDataSource(@Nullable TransferListener listener) { + this.listener = listener; + } + + @Override + public long open(DataSpec dataSpec) throws RtmpIOException { + rtmpClient = new RtmpClient(); + rtmpClient.open(dataSpec.uri.toString(), false); + + this.uri = dataSpec.uri; + if (listener != null) { + listener.onTransferStart(this, dataSpec); + } + return C.LENGTH_UNSET; + } + + @Override + public int read(byte[] buffer, int offset, int readLength) throws IOException { + int bytesRead = rtmpClient.read(buffer, offset, readLength); + if (bytesRead == -1) { + return C.RESULT_END_OF_INPUT; + } + if (listener != null) { + listener.onBytesTransferred(this, bytesRead); + } + return bytesRead; + } + + @Override + public void close() { + if (uri != null) { + uri = null; + if (listener != null) { + listener.onTransferEnd(this); + } + } + if (rtmpClient != null) { + rtmpClient.close(); + rtmpClient = null; + } + } + + @Override + public Uri getUri() { + return uri; + } + +} diff --git a/extensions/rtmp/src/main/java/com/google/android/exoplayer2/ext/rtmp/RtmpDataSourceFactory.java b/extensions/rtmp/src/main/java/com/google/android/exoplayer2/ext/rtmp/RtmpDataSourceFactory.java new file mode 100644 index 0000000000..0510e9c7da --- /dev/null +++ b/extensions/rtmp/src/main/java/com/google/android/exoplayer2/ext/rtmp/RtmpDataSourceFactory.java @@ -0,0 +1,47 @@ +/* + * Copyright (C) 2017 The Android Open Source Project + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package com.google.android.exoplayer2.ext.rtmp; + +import android.support.annotation.Nullable; +import com.google.android.exoplayer2.upstream.DataSource; +import com.google.android.exoplayer2.upstream.HttpDataSource.Factory; +import com.google.android.exoplayer2.upstream.TransferListener; + +/** + * A {@link Factory} that produces {@link RtmpDataSource}. + */ +public final class RtmpDataSourceFactory implements DataSource.Factory { + + @Nullable + private final TransferListener listener; + + public RtmpDataSourceFactory() { + this(null); + } + + /** + * @param listener An optional listener. + */ + public RtmpDataSourceFactory(@Nullable TransferListener listener) { + this.listener = listener; + } + + @Override + public DataSource createDataSource() { + return new RtmpDataSource(listener); + } + +}