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 FILE_SOURCE_H_
18 
19 #define FILE_SOURCE_H_
20 
21 #include <stdio.h>
22 
23 #include <media/DataSource.h>
24 #include <media/stagefright/MediaErrors.h>
25 #include <utils/threads.h>
26 #include <drm/DrmManagerClient.h>
27 
28 namespace android {
29 
30 class FileSource : public DataSource {
31 public:
32     FileSource(const char *filename);
33     // FileSource takes ownership and will close the fd
34     FileSource(int fd, int64_t offset, int64_t length);
35 
36     virtual status_t initCheck() const;
37 
38     virtual ssize_t readAt(off64_t offset, void *data, size_t size);
39 
40     virtual status_t getSize(off64_t *size);
41 
flags()42     virtual uint32_t flags() {
43         return kIsLocalFileSource;
44     }
45 
46     virtual sp<DecryptHandle> DrmInitialization(const char *mime);
47 
toString()48     virtual String8 toString() {
49         return mName;
50     }
51 
52     static bool requiresDrm(int fd, int64_t offset, int64_t length, const char *mime);
53 
54 protected:
55     virtual ~FileSource();
56 
57 private:
58     int mFd;
59     int64_t mOffset;
60     int64_t mLength;
61     Mutex mLock;
62     String8 mName;
63 
64     /*for DRM*/
65     sp<DecryptHandle> mDecryptHandle;
66     DrmManagerClient *mDrmManagerClient;
67     int64_t mDrmBufOffset;
68     ssize_t mDrmBufSize;
69     unsigned char *mDrmBuf;
70 
71     ssize_t readAtDRM(off64_t offset, void *data, size_t size);
72 
73     FileSource(const FileSource &);
74     FileSource &operator=(const FileSource &);
75 };
76 
77 }  // namespace android
78 
79 #endif  // FILE_SOURCE_H_
80 
81