• Home
• org.openjfx
• javafx-media
• 21.0.2
• source code
• HLSConnectionHolder.java

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

com.sun.media.jfxmedia.locator.HLSConnectionHolder Maven / Gradle / Ivy

/*
 * Copyright (c) 2010, 2022, Oracle and/or its affiliates. All rights reserved.
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 *
 * This code is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License version 2 only, as
 * published by the Free Software Foundation.  Oracle designates this
 * particular file as subject to the "Classpath" exception as provided
 * by Oracle in the LICENSE file that accompanied this code.
 *
 * This code is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
 * version 2 for more details (a copy is included in the LICENSE file that
 * accompanied this code).
 *
 * You should have received a copy of the GNU General Public License version
 * 2 along with this work; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
 * or visit www.oracle.com if you need additional information or have any
 * questions.
 */
package com.sun.media.jfxmedia.locator;

import com.sun.media.jfxmedia.MediaError;
import com.sun.media.jfxmediaimpl.MediaUtils;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.net.HttpURLConnection;
import java.net.MalformedURLException;
import java.net.URI;
import java.net.URISyntaxException;
import java.net.URLConnection;
import java.nio.channels.Channels;
import java.nio.channels.ReadableByteChannel;
import java.nio.charset.Charset;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.BlockingQueue;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.Semaphore;

final class HLSConnectionHolder extends ConnectionHolder {

    private URLConnection urlConnection = null;
    private URLConnection headerConnection = null;
    private ReadableByteChannel headerChannel = null;
    private PlaylistThread playlistThread = new PlaylistThread();
    private VariantPlaylist variantPlaylist = null;
    private Playlist currentPlaylist = null;
    private int mediaFileIndex = -1;
    private CountDownLatch readySignal = new CountDownLatch(1);
    private Semaphore liveSemaphore = new Semaphore(0);
    private boolean isPlaylistClosed = false;
    private boolean isBitrateAdjustable = false;
    private long startTime = -1;
    private boolean sendHeader = false;
    private static final long HLS_VALUE_FLOAT_MULTIPLIER = 1000;
    private static final int HLS_PROP_GET_DURATION = 1;
    private static final int HLS_PROP_GET_HLS_MODE = 2;
    private static final int HLS_PROP_GET_MIMETYPE = 3;
    private static final int HLS_PROP_LOAD_SEGMENT = 4;
    private static final int HLS_PROP_SEGMENT_START_TIME = 5;
    private static final int HLS_VALUE_MIMETYPE_UNKNOWN = -1;
    private static final int HLS_VALUE_MIMETYPE_MP2T = 1;
    private static final int HLS_VALUE_MIMETYPE_MP3 = 2;
    private static final int HLS_VALUE_MIMETYPE_FMP4 = 3;
    private static final int HLS_VALUE_MIMETYPE_AAC = 4;
    private static final String CHARSET_UTF_8 = "UTF-8";
    private static final String CHARSET_US_ASCII = "US-ASCII";

    HLSConnectionHolder(URI uri) {
        playlistThread.setPlaylistURI(uri);
        init();
    }

    private void init() {
        playlistThread.putState(PlaylistThread.STATE_INIT);
        playlistThread.start();
    }

    @Override
    public int readNextBlock() throws IOException {
        if (isBitrateAdjustable && startTime == -1) {
            startTime = System.currentTimeMillis();
        }

        if (headerChannel != null) {
            buffer.rewind();
            if (buffer.limit() < buffer.capacity()) {
                buffer.limit(buffer.capacity());
            }
            int read = headerChannel.read(buffer);
            if (read == -1) {
                resetHeaderConnection();
            } else {
                return read;
            }
        }

        int read = super.readNextBlock();
        if (isBitrateAdjustable && read == -1) {
            long readTime = System.currentTimeMillis() - startTime;
            startTime = -1;
            adjustBitrate(readTime);
        }

        return read;
    }

    @Override
    int readBlock(long position, int size) throws IOException {
        throw new IOException();
    }

    @Override
    boolean needBuffer() {
        return true;
    }

    @Override
    boolean isSeekable() {
        return true;
    }

    @Override
    boolean isRandomAccess() {
        return false; // Only by segments
    }

    @Override
    public long seek(long position) {
        try {
            readySignal.await();
        } catch (InterruptedException e) {
            return -1;
        }

        return (long) (currentPlaylist.seek(position) * HLS_VALUE_FLOAT_MULTIPLIER);
    }

    @Override
    public void closeConnection() {
        currentPlaylist.close();
        super.closeConnection();
        resetConnection();
        playlistThread.putState(PlaylistThread.STATE_EXIT);
    }

    @Override
    int property(int prop, int value) {
        try {
            readySignal.await();
        } catch (InterruptedException e) {
            return -1;
        }

        switch (prop) {
            case HLS_PROP_GET_DURATION:
                return (int)(currentPlaylist.getDuration() * HLS_VALUE_FLOAT_MULTIPLIER);
            case HLS_PROP_GET_HLS_MODE:
                return 1;
            case HLS_PROP_GET_MIMETYPE:
                return currentPlaylist.getMimeType();
            case HLS_PROP_LOAD_SEGMENT:
                return loadNextSegment();
            case HLS_PROP_SEGMENT_START_TIME:
                return (int)(currentPlaylist.getMediaFileStartTime() * HLS_VALUE_FLOAT_MULTIPLIER);
            default:
                return -1;
        }
    }

    private void resetConnection() {
        super.closeConnection();

        resetHeaderConnection();

        Locator.closeConnection(urlConnection);
        urlConnection = null;
    }

    private void resetHeaderConnection() {
        try {
            if (headerChannel != null) {
                headerChannel.close();
            }
        } catch (IOException ioex) {}
        finally {
            headerChannel = null;
        }

        Locator.closeConnection(headerConnection);
        headerConnection = null;
    }


    // Returns -1 EOS or critical error
    // Returns positive size of segment if no issues.
    // Returns negative size of segment if discontinuity.
    private int loadNextSegment() {
        resetConnection();

        String mediaFile;
        int headerLength = 0;

        if (sendHeader) {
            mediaFile = currentPlaylist.getHeaderFile();
            if (mediaFile == null) {
                return -1;
            }

            try {
                URI uri = new URI(mediaFile);
                headerConnection = uri.toURL().openConnection();
                headerChannel = openHeaderChannel();
                headerLength = headerConnection.getContentLength();
            } catch (IOException | URISyntaxException e) {
                return -1;
            }
            sendHeader = false;
        }

        mediaFile = currentPlaylist.getNextMediaFile();
        if (mediaFile == null) {
            if (currentPlaylist.isFragmentedMP4()) {
                sendHeader = true;
            }
            return -1;
        }

        try {
            URI uri = new URI(mediaFile);
            urlConnection = uri.toURL().openConnection();
            channel = openChannel();
        } catch (IOException | URISyntaxException e) {
            return -1;
        }

        if (currentPlaylist.isCurrentMediaFileDiscontinuity()) {
            return (-1 * (urlConnection.getContentLength() + headerLength));
        } else {
            return (urlConnection.getContentLength() + headerLength);
        }
    }

    private ReadableByteChannel openChannel() throws IOException {
        return Channels.newChannel(urlConnection.getInputStream());
    }

    private ReadableByteChannel openHeaderChannel() throws IOException {
        return Channels.newChannel(headerConnection.getInputStream());
    }

    private void adjustBitrate(long readTime) {
        int avgBitrate = (int)(((long) urlConnection.getContentLength() * 8 * 1000) / readTime);

        Playlist playlist = variantPlaylist.getPlaylistBasedOnBitrate(avgBitrate);
        if (playlist != null && playlist != currentPlaylist) {
            if (currentPlaylist.isLive()) {
                playlist.update(currentPlaylist.getNextMediaFile());
                playlistThread.setReloadPlaylist(playlist);
            }

            playlist.setForceDiscontinuity(true);
            currentPlaylist = playlist;
            if (currentPlaylist.isFragmentedMP4()) {
                sendHeader = true;
            }
        }
    }

    private static String stripParameters(String mediaFile) {
        int qp = mediaFile.indexOf('?');
        if (qp > 0) {
            mediaFile = mediaFile.substring(0, qp); // Strip all possible http parameters.
        }
        return mediaFile;
    }

    private class PlaylistThread extends Thread {

        public static final int STATE_INIT = 0;
        public static final int STATE_EXIT = 1;
        public static final int STATE_RELOAD_PLAYLIST = 2;
        private final BlockingQueue stateQueue = new LinkedBlockingQueue<>();
        private URI playlistURI = null;
        private Playlist reloadPlaylist = null;
        private final Object reloadLock = new Object();
        private volatile boolean stopped = false;

        private PlaylistThread() {
            setName("JFXMedia HLS Playlist Thread");
            setDaemon(true);
        }

        private void setPlaylistURI(URI playlistURI) {
            this.playlistURI = playlistURI;
        }

        private void setReloadPlaylist(Playlist playlist) {
            synchronized(reloadLock) {
                reloadPlaylist = playlist;
            }
        }

        @Override
        public void run() {
            while (!stopped) {
                try {
                    int state = stateQueue.take();
                    switch (state) {
                        case STATE_INIT:
                            stateInit();
                            break;
                        case STATE_EXIT:
                            stopped = true;
                            break;
                        case STATE_RELOAD_PLAYLIST:
                            stateReloadPlaylist();
                            break;
                        default:
                            break;
                    }
                } catch (InterruptedException e) {
                }
            }
        }

        private void putState(int state) {
            if (stateQueue != null) {
                try {
                    stateQueue.put(state);
                } catch (InterruptedException ex) {
                }
            }
        }

        private void stateInit() {
            if (playlistURI == null) {
                return;
            }

            try {
                PlaylistParser parser = new PlaylistParser();
                parser.load(playlistURI);

                if (parser.isVariantPlaylist()) {
                    variantPlaylist = new VariantPlaylist(playlistURI);

                    while (parser.hasNext()) {
                        variantPlaylist.addPlaylistInfo(parser.getString(), parser.getInteger());
                    }
                } else {
                    if (currentPlaylist == null) {
                        currentPlaylist = new Playlist(parser.isLivePlaylist(), parser.getTargetDuration());
                        currentPlaylist.setPlaylistURI(playlistURI);
                    }

                    if (currentPlaylist.setSequenceNumber(parser.getSequenceNumber())) {
                        while (parser.hasNext()) {
                            currentPlaylist.addMediaFile(parser.getString(), parser.getDouble(), parser.getBoolean());
                        }
                    }

                    if (variantPlaylist != null) {
                        variantPlaylist.addPlaylist(currentPlaylist);
                    }
                }

                // Update variant playlists
                if (variantPlaylist != null) {
                    while (variantPlaylist.hasNext()) {
                        try {
                            currentPlaylist = new Playlist(variantPlaylist.getPlaylistURI());
                            currentPlaylist.update(null);
                            variantPlaylist.addPlaylist(currentPlaylist);
                        } catch (URISyntaxException | MalformedURLException e) {
                        }
                    }
                }

                // Always start with first data playlist
                if (variantPlaylist != null) {
                    currentPlaylist = variantPlaylist.getPlaylist(0);
                    isBitrateAdjustable = true;
                }

                // Start reloading live playlist
                if (currentPlaylist.isLive()) {
                    setReloadPlaylist(currentPlaylist);
                    putState(STATE_RELOAD_PLAYLIST);
                }

                // If we have playlist with fMP4, set flag to add header
                // to first data segment and adjust index to 0
                if (currentPlaylist.isFragmentedMP4()) {
                    sendHeader = true;
                    mediaFileIndex = 0;
                }
            } finally {
                readySignal.countDown();
            }
        }

        private void stateReloadPlaylist() {
            try {
                long timeout;
                synchronized(reloadLock) {
                    timeout = reloadPlaylist.getTargetDuration() / 2;
                }
                Thread.sleep(timeout);
            } catch (InterruptedException ex) {
                return;
            }

            synchronized(reloadLock) {
                reloadPlaylist.update(null);
            }

            putState(STATE_RELOAD_PLAYLIST);
        }
    }

    private static class PlaylistParser {

        private boolean isFirstLine = true;
        private boolean isLineMediaFileURI = false;
        private boolean isEndList = false;
        private boolean isLinePlaylistURI = false;
        private boolean isVariantPlaylist = false;
        private boolean isDiscontinuity = false;
        private int targetDuration = 0;
        private int sequenceNumber = 0;
        private int dataListIndex = -1;
        private List dataListString = new ArrayList<>();
        private List dataListInteger = new ArrayList<>();
        private List dataListDouble = new ArrayList<>();
        private List dataListBoolean = new ArrayList<>();

        private void load(URI uri) {
            HttpURLConnection connection = null;
            BufferedReader reader = null;
            try {
                connection = (HttpURLConnection) uri.toURL().openConnection();
                connection.setRequestMethod("GET");

                if (connection.getResponseCode() != HttpURLConnection.HTTP_OK) {
                    MediaUtils.error(this, MediaError.ERROR_LOCATOR_CONNECTION_LOST.code(), "HTTP responce code: " + connection.getResponseCode(), null);
                }

                Charset charset = getCharset(uri.toURL().toExternalForm(), connection.getContentType());
                if (charset != null) {
                    reader = new BufferedReader(new InputStreamReader(connection.getInputStream(), charset));
                }

                if (reader != null) {
                    boolean result;
                    do {
                        result = parseLine(reader.readLine());
                    } while (result);
                }
            } catch (MalformedURLException e) {
            } catch (IOException e) {
            } finally {
                if (reader != null) {
                    try {
                        reader.close();
                    } catch (IOException e) {}

                    Locator.closeConnection(connection);
                }
            }
        }

        private boolean isVariantPlaylist() {
            return isVariantPlaylist;
        }

        private boolean isLivePlaylist() {
            return !isEndList;
        }

        private int getTargetDuration() {
            return targetDuration;
        }

        private int getSequenceNumber() {
            return sequenceNumber;
        }

        private boolean hasNext() {
            dataListIndex++;
            return dataListString.size() > dataListIndex ||
                    dataListInteger.size() > dataListIndex ||
                    dataListDouble.size() > dataListIndex ||
                    dataListBoolean.size() > dataListIndex;
        }

        private String getString() {
            return dataListString.get(dataListIndex);
        }

        private Integer getInteger() {
            return dataListInteger.get(dataListIndex);
        }

        private Double getDouble() {
            return dataListDouble.get(dataListIndex);
        }

        private Boolean getBoolean() {
            return dataListBoolean.get(dataListIndex);
        }

        private boolean parseLine(String line) {
            if (line == null) {
                return false;
            }

            // First line of playlist must be "#EXTM3U"
            if (isFirstLine) {
                if (line.compareTo("#EXTM3U") != 0) {
                    return false;
                }

                isFirstLine = false;
                return true;
            }

            // Ignore blank lines and comments
            if (line.isEmpty() || (line.startsWith("#") && !line.startsWith("#EXT"))) {
                return true;
            }

            if (line.startsWith("#EXTINF")) { // #EXTINF
                //#EXTINF:,