Home | History | Annotate | Download | only in libstagefright 
      1 /*
      2  * Copyright (C) 2017 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 //#define LOG_NDEBUG 0
     17 #define LOG_TAG "DataSource"
     18 
     19 #include "include/HTTPBase.h"
     20 #include "include/NuCachedSource2.h"
     21 
     22 #include <media/MediaHTTPConnection.h>
     23 #include <media/MediaHTTPService.h>
     24 #include <media/stagefright/DataSourceFactory.h>
     25 #include <media/stagefright/DataURISource.h>
     26 #include <media/stagefright/FileSource.h>
     27 #include <media/stagefright/MediaHTTP.h>
     28 #include <utils/String8.h>
     29 
     30 namespace android {
     31 
     32 // static
     33 sp<DataSource> DataSourceFactory::CreateFromURI(
     34         const sp<MediaHTTPService> &httpService,
     35         const char *uri,
     36         const KeyedVector<String8, String8> *headers,
     37         String8 *contentType,
     38         HTTPBase *httpSource) {
     39     if (contentType != NULL) {
     40         *contentType = "";
     41     }
     42 
     43     sp<DataSource> source;
     44     if (!strncasecmp("file://", uri, 7)) {
     45         source = new FileSource(uri + 7);
     46     } else if (!strncasecmp("http://", uri, 7) || !strncasecmp("https://", uri, 8)) {
     47         if (httpService == NULL) {
     48             ALOGE("Invalid http service!");
     49             return NULL;
     50         }
     51 
     52         if (httpSource == NULL) {
     53             sp<MediaHTTPConnection> conn = httpService->makeHTTPConnection();
     54             if (conn == NULL) {
     55                 ALOGE("Failed to make http connection from http service!");
     56                 return NULL;
     57             }
     58             httpSource = new MediaHTTP(conn);
     59         }
     60 
     61         String8 cacheConfig;
     62         bool disconnectAtHighwatermark = false;
     63         KeyedVector<String8, String8> nonCacheSpecificHeaders;
     64         if (headers != NULL) {
     65             nonCacheSpecificHeaders = *headers;
     66             NuCachedSource2::RemoveCacheSpecificHeaders(
     67                     &nonCacheSpecificHeaders,
     68                     &cacheConfig,
     69                     &disconnectAtHighwatermark);
     70         }
     71 
     72         if (httpSource->connect(uri, &nonCacheSpecificHeaders) != OK) {
     73             ALOGE("Failed to connect http source!");
     74             return NULL;
     75         }
     76 
     77         if (contentType != NULL) {
     78             *contentType = httpSource->getMIMEType();
     79         }
     80 
     81         source = NuCachedSource2::Create(
     82                 httpSource,
     83                 cacheConfig.isEmpty() ? NULL : cacheConfig.string(),
     84                 disconnectAtHighwatermark);
     85     } else if (!strncasecmp("data:", uri, 5)) {
     86         source = DataURISource::Create(uri);
     87     } else {
     88         // Assume it's a filename.
     89         source = new FileSource(uri);
     90     }
     91 
     92     if (source == NULL || source->initCheck() != OK) {
     93         return NULL;
     94     }
     95 
     96     return source;
     97 }
     98 
     99 sp<DataSource> DataSourceFactory::CreateFromFd(int fd, int64_t offset, int64_t length) {
    100     sp<FileSource> source = new FileSource(fd, offset, length);
    101     return source->initCheck() != OK ? nullptr : source;
    102 }
    103 
    104 sp<DataSource> DataSourceFactory::CreateMediaHTTP(const sp<MediaHTTPService> &httpService) {
    105     if (httpService == NULL) {
    106         return NULL;
    107     }
    108 
    109     sp<MediaHTTPConnection> conn = httpService->makeHTTPConnection();
    110     if (conn == NULL) {
    111         return NULL;
    112     } else {
    113         return new MediaHTTP(conn);
    114     }
    115 }
    116 
    117 }  // namespace android
    118