Andrew DeMaria
5 years ago
commit
3d0d0482f5
@ -0,0 +1,11 @@ |
||||
package org.airsonic.player.domain; |
||||
|
||||
public class InternetRadioSource { |
||||
private String streamUrl; |
||||
|
||||
public InternetRadioSource(String streamUrl) { |
||||
this.streamUrl = streamUrl; |
||||
} |
||||
|
||||
public String getStreamUrl() { return streamUrl; } |
||||
} |
||||
@ -0,0 +1,302 @@ |
||||
package org.airsonic.player.service; |
||||
|
||||
import chameleon.playlist.*; |
||||
import org.airsonic.player.domain.InternetRadio; |
||||
import org.airsonic.player.domain.InternetRadioSource; |
||||
import org.apache.commons.io.input.BoundedInputStream; |
||||
import org.slf4j.Logger; |
||||
import org.slf4j.LoggerFactory; |
||||
import org.springframework.stereotype.Service; |
||||
|
||||
import java.io.IOException; |
||||
import java.io.InputStream; |
||||
import java.net.HttpURLConnection; |
||||
import java.net.URL; |
||||
import java.util.*; |
||||
|
||||
@Service |
||||
public class InternetRadioService { |
||||
|
||||
private static final Logger LOG = LoggerFactory.getLogger(InternetRadioService.class); |
||||
|
||||
/** |
||||
* The maximum number of source URLs in a remote playlist. |
||||
*/ |
||||
private static final int PLAYLIST_REMOTE_MAX_LENGTH = 250; |
||||
|
||||
/** |
||||
* The maximum size, in bytes, for a remote playlist response. |
||||
*/ |
||||
private static final long PLAYLIST_REMOTE_MAX_BYTE_SIZE = 100 * 1024; // 100 kB
|
||||
|
||||
/** |
||||
* The maximum number of redirects for a remote playlist response. |
||||
*/ |
||||
private static final int PLAYLIST_REMOTE_MAX_REDIRECTS = 20; |
||||
|
||||
/** |
||||
* A list of cached source URLs for remote playlists. |
||||
*/ |
||||
private final Map<Integer, List<InternetRadioSource>> cachedSources; |
||||
|
||||
/** |
||||
* Generic exception class for playlists. |
||||
*/ |
||||
private class PlaylistException extends Exception { |
||||
public PlaylistException(String message) { super(message); } |
||||
} |
||||
|
||||
/** |
||||
* Exception thrown when the remote playlist is too large to be parsed completely. |
||||
*/ |
||||
private class PlaylistTooLarge extends PlaylistException { |
||||
public PlaylistTooLarge(String message) { super(message); } |
||||
} |
||||
|
||||
/** |
||||
* Exception thrown when the remote playlist format cannot be determined. |
||||
*/ |
||||
private class PlaylistFormatUnsupported extends PlaylistException { |
||||
public PlaylistFormatUnsupported(String message) { |
||||
super(message); |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* Exception thrown when too many redirects occurred when retrieving a remote playlist. |
||||
*/ |
||||
private class PlaylistHasTooManyRedirects extends PlaylistException { |
||||
public PlaylistHasTooManyRedirects(String message) { super(message); } |
||||
} |
||||
|
||||
public InternetRadioService() { |
||||
this.cachedSources = new HashMap<>(); |
||||
} |
||||
|
||||
/** |
||||
* Clear the radio source cache. |
||||
*/ |
||||
public void clearInternetRadioSourceCache() { |
||||
cachedSources.clear(); |
||||
} |
||||
|
||||
/** |
||||
* Clear the radio source cache for the given radio id |
||||
* @param internetRadioId a radio id |
||||
*/ |
||||
public void clearInternetRadioSourceCache(Integer internetRadioId) { |
||||
if (internetRadioId != null) { |
||||
cachedSources.remove(internetRadioId); |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* Retrieve a list of sources for the given internet radio. |
||||
* |
||||
* This method caches the sources using the InternetRadio.getId |
||||
* method as a key, until clearInternetRadioSourceCache is called. |
||||
* |
||||
* @param radio an internet radio |
||||
* @return a list of internet radio sources |
||||
*/ |
||||
public List<InternetRadioSource> getInternetRadioSources(InternetRadio radio) { |
||||
List<InternetRadioSource> sources; |
||||
if (cachedSources.containsKey(radio.getId())) { |
||||
LOG.debug("Got cached sources for internet radio {}!", radio.getStreamUrl()); |
||||
sources = cachedSources.get(radio.getId()); |
||||
} else { |
||||
LOG.debug("Retrieving sources for internet radio {}...", radio.getStreamUrl()); |
||||
try { |
||||
sources = retrieveInternetRadioSources(radio); |
||||
if (sources.isEmpty()) { |
||||
LOG.warn("No entries found for internet radio {}.", radio.getStreamUrl()); |
||||
} else { |
||||
LOG.info("Retrieved playlist for internet radio {}, got {} sources.", radio.getStreamUrl(), sources.size()); |
||||
} |
||||
} catch (Exception e) { |
||||
LOG.error("Failed to retrieve sources for internet radio {}.", radio.getStreamUrl(), e); |
||||
sources = new ArrayList<>(); |
||||
} |
||||
cachedSources.put(radio.getId(), sources); |
||||
} |
||||
return sources; |
||||
} |
||||
|
||||
/** |
||||
* Retrieve a list of sources from the given internet radio |
||||
* |
||||
* This method uses a default maximum limit of PLAYLIST_REMOTE_MAX_LENGTH sources. |
||||
* |
||||
* @param radio an internet radio |
||||
* @return a list of internet radio sources |
||||
* @throws Exception |
||||
*/ |
||||
private List<InternetRadioSource> retrieveInternetRadioSources(InternetRadio radio) throws Exception { |
||||
return retrieveInternetRadioSources( |
||||
radio, |
||||
PLAYLIST_REMOTE_MAX_LENGTH, |
||||
PLAYLIST_REMOTE_MAX_BYTE_SIZE, |
||||
PLAYLIST_REMOTE_MAX_REDIRECTS |
||||
); |
||||
} |
||||
|
||||
/** |
||||
* Retrieve a list of sources from the given internet radio. |
||||
* |
||||
* @param radio an internet radio |
||||
* @param maxCount the maximum number of items to read from the remote playlist, or 0 if unlimited |
||||
* @param maxByteSize maximum size of the response, in bytes, or 0 if unlimited |
||||
* @param maxRedirects maximum number of redirects, or 0 if unlimited |
||||
* @return a list of internet radio sources |
||||
* @throws Exception |
||||
*/ |
||||
private List<InternetRadioSource> retrieveInternetRadioSources(InternetRadio radio, int maxCount, long maxByteSize, int maxRedirects) throws Exception { |
||||
// Retrieve the remote playlist
|
||||
String playlistUrl = radio.getStreamUrl(); |
||||
LOG.debug("Parsing internet radio playlist at {}...", playlistUrl); |
||||
SpecificPlaylist inputPlaylist = retrievePlaylist(new URL(playlistUrl), maxByteSize, maxRedirects); |
||||
|
||||
// Retrieve stream URLs
|
||||
List<InternetRadioSource> entries = new ArrayList<>(); |
||||
try { |
||||
inputPlaylist.toPlaylist().acceptDown(new PlaylistVisitor() { |
||||
@Override |
||||
public void beginVisitPlaylist(Playlist playlist) throws Exception { |
||||
|
||||
} |
||||
|
||||
@Override |
||||
public void endVisitPlaylist(Playlist playlist) throws Exception { |
||||
|
||||
} |
||||
|
||||
@Override |
||||
public void beginVisitParallel(Parallel parallel) throws Exception { |
||||
|
||||
} |
||||
|
||||
@Override |
||||
public void endVisitParallel(Parallel parallel) throws Exception { |
||||
|
||||
} |
||||
|
||||
@Override |
||||
public void beginVisitSequence(Sequence sequence) throws Exception { |
||||
|
||||
} |
||||
|
||||
@Override |
||||
public void endVisitSequence(Sequence sequence) throws Exception { |
||||
|
||||
} |
||||
|
||||
@Override |
||||
public void beginVisitMedia(Media media) throws Exception { |
||||
// Since we're dealing with remote content, we place a hard
|
||||
// limit on the maximum number of items to load from the playlist,
|
||||
// in order to avoid parsing erroneous data.
|
||||
if (maxCount > 0 && entries.size() >= maxCount) { |
||||
throw new PlaylistTooLarge("Remote playlist has too many sources (maximum " + maxCount + ")"); |
||||
} |
||||
String streamUrl = media.getSource().getURI().toString(); |
||||
LOG.debug("Got source media at {}", streamUrl); |
||||
entries.add(new InternetRadioSource(streamUrl)); |
||||
} |
||||
|
||||
@Override |
||||
public void endVisitMedia(Media media) throws Exception { |
||||
|
||||
} |
||||
}); |
||||
} catch (PlaylistTooLarge e) { |
||||
// Ignore if playlist is too large, but truncate the rest and log a warning.
|
||||
LOG.warn(e.getMessage()); |
||||
} |
||||
|
||||
return entries; |
||||
} |
||||
|
||||
/** |
||||
* Retrieve playlist data from a given URL. |
||||
* |
||||
* @param url URL to the remote playlist |
||||
* @param maxByteSize maximum size of the response, in bytes, or 0 if unlimited |
||||
* @param maxRedirects maximum number of redirects, or 0 if unlimited |
||||
* @return the remote playlist data |
||||
*/ |
||||
protected SpecificPlaylist retrievePlaylist(URL url, long maxByteSize, int maxRedirects) throws IOException, PlaylistException { |
||||
|
||||
SpecificPlaylist playlist; |
||||
HttpURLConnection urlConnection = connectToURLWithRedirects(url, maxRedirects); |
||||
try (InputStream in = urlConnection.getInputStream()) { |
||||
String contentEncoding = urlConnection.getContentEncoding(); |
||||
if (maxByteSize > 0) { |
||||
playlist = SpecificPlaylistFactory.getInstance().readFrom(new BoundedInputStream(in, maxByteSize), contentEncoding); |
||||
} else { |
||||
playlist = SpecificPlaylistFactory.getInstance().readFrom(in, contentEncoding); |
||||
} |
||||
} |
||||
finally { |
||||
urlConnection.disconnect(); |
||||
} |
||||
if (playlist == null) { |
||||
throw new PlaylistFormatUnsupported("Unsupported playlist format " + url.toString()); |
||||
} |
||||
return playlist; |
||||
} |
||||
|
||||
/** |
||||
* Start a new connection to a remote URL, and follow redirects. |
||||
* |
||||
* @param url the remote URL |
||||
* @param maxRedirects maximum number of redirects, or 0 if unlimited |
||||
* @return an open connection |
||||
*/ |
||||
protected HttpURLConnection connectToURLWithRedirects(URL url, int maxRedirects) throws IOException, PlaylistException { |
||||
|
||||
int redirectCount = 0; |
||||
URL currentURL = url; |
||||
|
||||
// Start a first connection.
|
||||
HttpURLConnection connection = connectToURL(currentURL); |
||||
|
||||
// While it redirects, follow redirects in new connections.
|
||||
while (connection.getResponseCode() == HttpURLConnection.HTTP_MOVED_PERM || |
||||
connection.getResponseCode() == HttpURLConnection.HTTP_MOVED_TEMP || |
||||
connection.getResponseCode() == HttpURLConnection.HTTP_SEE_OTHER) { |
||||
|
||||
// Check if redirect count is not too large.
|
||||
redirectCount += 1; |
||||
if (maxRedirects > 0 && redirectCount > maxRedirects) { |
||||
connection.disconnect(); |
||||
throw new PlaylistHasTooManyRedirects(String.format("Too many redirects (%d) for URL %s", redirectCount, url)); |
||||
} |
||||
|
||||
// Reconnect to the new URL.
|
||||
currentURL = new URL(connection.getHeaderField("Location")); |
||||
connection.disconnect(); |
||||
connection = connectToURL(currentURL); |
||||
} |
||||
|
||||
// Return the last connection that did not redirect.
|
||||
return connection; |
||||
} |
||||
|
||||
/** |
||||
* Start a new connection to a remote URL. |
||||
* |
||||
* @param url the remote URL |
||||
* @return an open connection |
||||
*/ |
||||
protected HttpURLConnection connectToURL(URL url) throws IOException { |
||||
HttpURLConnection urlConnection = (HttpURLConnection) url.openConnection(); |
||||
urlConnection.setAllowUserInteraction(false); |
||||
urlConnection.setConnectTimeout(10000); |
||||
urlConnection.setDoInput(true); |
||||
urlConnection.setDoOutput(false); |
||||
urlConnection.setReadTimeout(60000); |
||||
urlConnection.setUseCaches(true); |
||||
urlConnection.connect(); |
||||
return urlConnection; |
||||
} |
||||
} |
||||
@ -0,0 +1,177 @@ |
||||
package org.airsonic.player.service; |
||||
|
||||
import org.airsonic.player.domain.InternetRadio; |
||||
import org.airsonic.player.domain.InternetRadioSource; |
||||
import org.junit.Assert; |
||||
import org.junit.Before; |
||||
import org.junit.Test; |
||||
import org.junit.runner.RunWith; |
||||
import org.mockito.*; |
||||
import org.mockito.runners.MockitoJUnitRunner; |
||||
|
||||
import java.io.ByteArrayInputStream; |
||||
import java.io.IOException; |
||||
import java.io.InputStream; |
||||
import java.net.HttpURLConnection; |
||||
import java.net.URL; |
||||
import java.util.Date; |
||||
import java.util.List; |
||||
|
||||
import static org.mockito.Matchers.any; |
||||
import static org.mockito.Mockito.*; |
||||
|
||||
@RunWith(MockitoJUnitRunner.class) |
||||
public class InternetRadioServiceTest { |
||||
|
||||
String TEST_RADIO_NAME = "Test Radio"; |
||||
String TEST_RADIO_HOMEPAGE = "http://example.com"; |
||||
String TEST_PLAYLIST_URL_MOVE = "http://example.com/stream_move.m3u"; |
||||
String TEST_PLAYLIST_URL_MOVE_LOOP = "http://example.com/stream_infinity_move.m3u"; |
||||
String TEST_PLAYLIST_URL_LARGE = "http://example.com/stream_infinity_repeat.m3u"; |
||||
String TEST_PLAYLIST_URL_LARGE_2 = "http://example.com/stream_infinity_big.m3u"; |
||||
String TEST_PLAYLIST_URL_1 = "http://example.com/stream1.m3u"; |
||||
String TEST_PLAYLIST_URL_2 = "http://example.com/stream2.m3u"; |
||||
String TEST_STREAM_URL_1 = "http://example.com/stream1"; |
||||
String TEST_STREAM_URL_2 = "http://example.com/stream2"; |
||||
String TEST_STREAM_URL_3 = "http://example.com/stream3"; |
||||
String TEST_STREAM_URL_4 = "http://example.com/stream4"; |
||||
String TEST_STREAM_PLAYLIST_CONTENTS_1 = ( |
||||
"http://example.com/stream1\n" + |
||||
"http://example.com/stream2\n" |
||||
); |
||||
String TEST_STREAM_PLAYLIST_CONTENTS_2 = ( |
||||
"#EXTM3U\n" + |
||||
"#EXTINF:123, Sample artist - Sample title\n" + |
||||
"http://example.com/stream3\n" + |
||||
"#EXTINF:321,Example Artist - Example title\n" + |
||||
"http://example.com/stream4\n" |
||||
); |
||||
|
||||
InternetRadio radio1; |
||||
InternetRadio radio2; |
||||
InternetRadio radioMove; |
||||
InternetRadio radioMoveLoop; |
||||
InternetRadio radioLarge; |
||||
InternetRadio radioLarge2; |
||||
|
||||
@Spy |
||||
InternetRadioService internetRadioService; |
||||
|
||||
@Before |
||||
public void setup() throws Exception { |
||||
|
||||
// Prepare a mock InternetRadio object
|
||||
radio1 = new InternetRadio(1, TEST_RADIO_NAME, TEST_PLAYLIST_URL_1, TEST_RADIO_HOMEPAGE, true, new Date()); |
||||
radio2 = new InternetRadio(2, TEST_RADIO_NAME, TEST_PLAYLIST_URL_2, TEST_RADIO_HOMEPAGE, true, new Date()); |
||||
radioMove = new InternetRadio(3, TEST_RADIO_NAME, TEST_PLAYLIST_URL_MOVE, TEST_RADIO_HOMEPAGE, true, new Date()); |
||||
radioMoveLoop = new InternetRadio(3, TEST_RADIO_NAME, TEST_PLAYLIST_URL_MOVE_LOOP, TEST_RADIO_HOMEPAGE, true, new Date()); |
||||
radioLarge = new InternetRadio(4, TEST_RADIO_NAME, TEST_PLAYLIST_URL_LARGE, TEST_RADIO_HOMEPAGE, true, new Date()); |
||||
radioLarge2 = new InternetRadio(5, TEST_RADIO_NAME, TEST_PLAYLIST_URL_LARGE_2, TEST_RADIO_HOMEPAGE, true, new Date()); |
||||
|
||||
// Prepare the mocked URL connection for the simple playlist
|
||||
HttpURLConnection mockURLConnection1 = Mockito.mock(HttpURLConnection.class); |
||||
InputStream mockURLInputStream1 = new ByteArrayInputStream(TEST_STREAM_PLAYLIST_CONTENTS_1.getBytes()); |
||||
doReturn(mockURLInputStream1).when(mockURLConnection1).getInputStream(); |
||||
doReturn(HttpURLConnection.HTTP_OK).when(mockURLConnection1).getResponseCode(); |
||||
|
||||
// Prepare the mocked URL connection for the second simple playlist
|
||||
HttpURLConnection mockURLConnection2 = Mockito.mock(HttpURLConnection.class); |
||||
InputStream mockURLInputStream2 = new ByteArrayInputStream(TEST_STREAM_PLAYLIST_CONTENTS_2.getBytes()); |
||||
doReturn(mockURLInputStream2).when(mockURLConnection2).getInputStream(); |
||||
doReturn(HttpURLConnection.HTTP_OK).when(mockURLConnection2).getResponseCode(); |
||||
|
||||
// Prepare the mocked URL connection for the redirection to simple playlist
|
||||
HttpURLConnection mockURLConnectionMove = Mockito.mock(HttpURLConnection.class); |
||||
InputStream mockURLInputStreamMove = new ByteArrayInputStream("".getBytes()); |
||||
doReturn(mockURLInputStreamMove).when(mockURLConnectionMove).getInputStream(); |
||||
doReturn(HttpURLConnection.HTTP_MOVED_PERM).when(mockURLConnectionMove).getResponseCode(); |
||||
doReturn(TEST_PLAYLIST_URL_2).when(mockURLConnectionMove).getHeaderField(eq("Location")); |
||||
|
||||
// Prepare the mocked URL connection for the redirection loop
|
||||
HttpURLConnection mockURLConnectionMoveLoop = Mockito.mock(HttpURLConnection.class); |
||||
InputStream mockURLInputStreamMoveLoop = new ByteArrayInputStream("".getBytes()); |
||||
doReturn(mockURLInputStreamMoveLoop).when(mockURLConnectionMoveLoop).getInputStream(); |
||||
doReturn(HttpURLConnection.HTTP_MOVED_PERM).when(mockURLConnectionMoveLoop).getResponseCode(); |
||||
doReturn(TEST_PLAYLIST_URL_MOVE_LOOP).when(mockURLConnectionMoveLoop).getHeaderField(eq("Location")); |
||||
|
||||
// Prepare the mocked URL connection for the 'content too large' test
|
||||
HttpURLConnection mockURLConnectionLarge = Mockito.mock(HttpURLConnection.class); |
||||
InputStream mockURLInputStreamLarge = new InputStream() { |
||||
private long pos = 0; |
||||
@Override |
||||
public int read() throws IOException { |
||||
return TEST_STREAM_PLAYLIST_CONTENTS_2.charAt((int)(pos++ % TEST_STREAM_PLAYLIST_CONTENTS_2.length())); |
||||
} |
||||
}; |
||||
doReturn(mockURLInputStreamLarge).when(mockURLConnectionLarge).getInputStream(); |
||||
doReturn(HttpURLConnection.HTTP_OK).when(mockURLConnectionLarge).getResponseCode(); |
||||
|
||||
// Prepare the mocked URL connection for the 'content too large' test
|
||||
// (return a single entry with 'aaaa...' running infinitely long).
|
||||
HttpURLConnection mockURLConnectionLarge2 = Mockito.mock(HttpURLConnection.class); |
||||
InputStream mockURLInputStreamLarge2 = new InputStream() { |
||||
private long pos = 0; |
||||
@Override |
||||
public int read() throws IOException { |
||||
return 0x41; |
||||
} |
||||
}; |
||||
doReturn(mockURLInputStreamLarge2).when(mockURLConnectionLarge2).getInputStream(); |
||||
doReturn(HttpURLConnection.HTTP_OK).when(mockURLConnectionLarge2).getResponseCode(); |
||||
|
||||
// Prepare the mock 'connectToURL' method
|
||||
doReturn(mockURLConnection1).when(internetRadioService).connectToURL(eq(new URL(TEST_PLAYLIST_URL_1))); |
||||
doReturn(mockURLConnection2).when(internetRadioService).connectToURL(eq(new URL(TEST_PLAYLIST_URL_2))); |
||||
doReturn(mockURLConnectionMove).when(internetRadioService).connectToURL(eq(new URL(TEST_PLAYLIST_URL_MOVE))); |
||||
doReturn(mockURLConnectionMoveLoop).when(internetRadioService).connectToURL(eq(new URL(TEST_PLAYLIST_URL_MOVE_LOOP))); |
||||
doReturn(mockURLConnectionLarge).when(internetRadioService).connectToURL(eq(new URL(TEST_PLAYLIST_URL_LARGE))); |
||||
doReturn(mockURLConnectionLarge2).when(internetRadioService).connectToURL(eq(new URL(TEST_PLAYLIST_URL_LARGE_2))); |
||||
} |
||||
|
||||
@Test |
||||
public void testParseSimplePlaylist() throws Exception { |
||||
List<InternetRadioSource> radioSources = internetRadioService.getInternetRadioSources(radio1); |
||||
|
||||
Assert.assertEquals(2, radioSources.size()); |
||||
Assert.assertEquals(TEST_STREAM_URL_1, radioSources.get(0).getStreamUrl()); |
||||
Assert.assertEquals(TEST_STREAM_URL_2, radioSources.get(1).getStreamUrl()); |
||||
} |
||||
|
||||
@Test |
||||
public void testRedirects() throws Exception { |
||||
List<InternetRadioSource> radioSources = internetRadioService.getInternetRadioSources(radioMove); |
||||
|
||||
Assert.assertEquals(2, radioSources.size()); |
||||
Assert.assertEquals(TEST_STREAM_URL_3, radioSources.get(0).getStreamUrl()); |
||||
Assert.assertEquals(TEST_STREAM_URL_4, radioSources.get(1).getStreamUrl()); |
||||
} |
||||
|
||||
@Test |
||||
public void testLargeInput() throws Exception { |
||||
List<InternetRadioSource> radioSources = internetRadioService.getInternetRadioSources(radioLarge); |
||||
|
||||
// A PlaylistTooLarge exception is thrown internally, and the
|
||||
// `getInternetRadioSources` method logs it and returns a
|
||||
// limited number of sources.
|
||||
Assert.assertEquals(250, radioSources.size()); |
||||
} |
||||
|
||||
@Test |
||||
public void testLargeInputURL() throws Exception { |
||||
List<InternetRadioSource> radioSources = internetRadioService.getInternetRadioSources(radioLarge2); |
||||
|
||||
// A PlaylistTooLarge exception is thrown internally, and the
|
||||
// `getInternetRadioSources` method logs it and returns a
|
||||
// limited number of bytes from the input.
|
||||
Assert.assertEquals(1, radioSources.size()); |
||||
} |
||||
|
||||
@Test |
||||
public void testRedirectLoop() throws Exception { |
||||
List<InternetRadioSource> radioSources = internetRadioService.getInternetRadioSources(radioMoveLoop); |
||||
|
||||
// A PlaylistHasTooManyRedirects exception is thrown internally,
|
||||
// and the `getInternetRadioSources` method logs it and returns 0 sources.
|
||||
Assert.assertEquals(0, radioSources.size()); |
||||
} |
||||
} |
||||
Loading…
Reference in new issue