#include "config.h"
#if ENABLE(BLOB)
#include "DOMURL.h"
#include "ActiveDOMObject.h"
#include "Blob.h"
#include "BlobURL.h"
#include "KURL.h"
#include "MemoryCache.h"
#include "PublicURLManager.h"
#include "ResourceRequest.h"
#include "ScriptExecutionContext.h"
#include "SecurityOrigin.h"
#include "ThreadableBlobRegistry.h"
#include <wtf/PassOwnPtr.h>
#include <wtf/MainThread.h>
#if ENABLE(MEDIA_SOURCE)
#include "MediaSource.h"
#include "MediaSourceRegistry.h"
#endif
#if ENABLE(MEDIA_STREAM)
#include "MediaStream.h"
#include "MediaStreamRegistry.h"
#endif
namespace WebCore {
#if ENABLE(MEDIA_SOURCE)
String DOMURL::createObjectURL(ScriptExecutionContext* scriptExecutionContext, MediaSource* source)
{
ASSERT(isMainThread());
if (!scriptExecutionContext || !source)
return String();
KURL publicURL = BlobURL::createPublicURL(scriptExecutionContext->securityOrigin());
if (publicURL.isEmpty())
return String();
MediaSourceRegistry::registry().registerMediaSourceURL(publicURL, source);
scriptExecutionContext->publicURLManager().sourceURLs().add(publicURL.string());
return publicURL.string();
}
#endif
#if ENABLE(MEDIA_STREAM)
String DOMURL::createObjectURL(ScriptExecutionContext* scriptExecutionContext, MediaStream* stream)
{
if (!scriptExecutionContext || !stream)
return String();
KURL publicURL = BlobURL::createPublicURL(scriptExecutionContext->securityOrigin());
if (publicURL.isEmpty())
return String();
ASSERT(isMainThread());
MediaStreamRegistry::registry().registerMediaStreamURL(publicURL, stream);
scriptExecutionContext->publicURLManager().streamURLs().add(publicURL.string());
return publicURL.string();
}
#endif
String DOMURL::createObjectURL(ScriptExecutionContext* scriptExecutionContext, Blob* blob)
{
if (!scriptExecutionContext || !blob)
return String();
KURL publicURL = BlobURL::createPublicURL(scriptExecutionContext->securityOrigin());
if (publicURL.isEmpty())
return String();
ThreadableBlobRegistry::registerBlobURL(scriptExecutionContext->securityOrigin(), publicURL, blob->url());
scriptExecutionContext->publicURLManager().blobURLs().add(publicURL.string());
return publicURL.string();
}
void DOMURL::revokeObjectURL(ScriptExecutionContext* scriptExecutionContext, const String& urlString)
{
if (!scriptExecutionContext)
return;
KURL url(KURL(), urlString);
ResourceRequest request(url);
#if ENABLE(CACHE_PARTITIONING)
request.setCachePartition(scriptExecutionContext->topOrigin()->cachePartition());
#endif
MemoryCache::removeRequestFromCache(scriptExecutionContext, request);
HashSet<String>& blobURLs = scriptExecutionContext->publicURLManager().blobURLs();
if (blobURLs.contains(url.string())) {
ThreadableBlobRegistry::unregisterBlobURL(url);
blobURLs.remove(url.string());
}
#if ENABLE(MEDIA_SOURCE)
HashSet<String>& sourceURLs = scriptExecutionContext->publicURLManager().sourceURLs();
if (sourceURLs.contains(url.string())) {
MediaSourceRegistry::registry().unregisterMediaSourceURL(url);
sourceURLs.remove(url.string());
}
#endif
#if ENABLE(MEDIA_STREAM)
HashSet<String>& streamURLs = scriptExecutionContext->publicURLManager().streamURLs();
if (streamURLs.contains(url.string())) {
ASSERT(isMainThread());
MediaStreamRegistry::registry().unregisterMediaStreamURL(url);
streamURLs.remove(url.string());
}
#endif
}
}
#endif // ENABLE(BLOB)