DataSource.h revision 62ad9071cd61fc56355d3c2d25895382003ea362 
1/*
2 * Copyright (C) 2009 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
17#ifndef DATA_SOURCE_H_
18
19#define DATA_SOURCE_H_
20
21#include <sys/types.h>
22
23#include <media/stagefright/MediaErrors.h>
24#include <utils/Errors.h>
25#include <utils/KeyedVector.h>
26#include <utils/List.h>
27#include <utils/RefBase.h>
28#include <utils/threads.h>
29#include <drm/DrmManagerClient.h>
30
31namespace android {
32
33struct AMessage;
34class String8;
35
36class DataSource : public RefBase {
37public:
38    enum Flags {
39        kWantsPrefetching      = 1,
40        kStreamedFromLocalHost = 2,
41        kIsCachingDataSource   = 4,
42        kIsHTTPBasedSource     = 8,
43    };
44
45    static sp<DataSource> CreateFromURI(
46            const char *uri,
47            const KeyedVector<String8, String8> *headers = NULL);
48
49    DataSource() {}
50
51    virtual status_t initCheck() const = 0;
52
53    virtual ssize_t readAt(off64_t offset, void *data, size_t size) = 0;
54
55    // Convenience methods:
56    bool getUInt16(off64_t offset, uint16_t *x);
57    bool getUInt32(off64_t offset, uint32_t *x);
58    bool getUInt64(off64_t offset, uint64_t *x);
59
60    // May return ERROR_UNSUPPORTED.
61    virtual status_t getSize(off64_t *size);
62
63    virtual uint32_t flags() {
64        return 0;
65    }
66
67    virtual status_t reconnectAtOffset(off64_t offset) {
68        return ERROR_UNSUPPORTED;
69    }
70
71    ////////////////////////////////////////////////////////////////////////////
72
73    bool sniff(String8 *mimeType, float *confidence, sp<AMessage> *meta);
74
75    // The sniffer can optionally fill in "meta" with an AMessage containing
76    // a dictionary of values that helps the corresponding extractor initialize
77    // its state without duplicating effort already exerted by the sniffer.
78    typedef bool (*SnifferFunc)(
79            const sp<DataSource> &source, String8 *mimeType,
80            float *confidence, sp<AMessage> *meta);
81
82    static void RegisterSniffer(SnifferFunc func);
83    static void RegisterDefaultSniffers();
84
85    // for DRM
86    virtual sp<DecryptHandle> DrmInitialization(const char *mime = NULL) {
87        return NULL;
88    }
89    virtual void getDrmInfo(sp<DecryptHandle> &handle, DrmManagerClient **client) {};
90
91    virtual String8 getUri() {
92        return String8();
93    }
94
95    virtual String8 getMIMEType() const;
96
97protected:
98    virtual ~DataSource() {}
99
100private:
101    static Mutex gSnifferMutex;
102    static List<SnifferFunc> gSniffers;
103
104    DataSource(const DataSource &);
105    DataSource &operator=(const DataSource &);
106};
107
108}  // namespace android
109
110#endif  // DATA_SOURCE_H_
111