diff --git a/include/media/MidiIoWrapper.h b/include/media/MidiIoWrapper.h
index caf1d75..e6f8cf7 100644
--- a/include/media/MidiIoWrapper.h
+++ b/include/media/MidiIoWrapper.h
@@ -19,12 +19,15 @@
 
 #include <libsonivox/eas_types.h>
 
+#include "media/stagefright/DataSource.h"
+
 namespace android {
 
 class MidiIoWrapper : public RefBase {
 public:
     MidiIoWrapper(const char *path);
     MidiIoWrapper(int fd, off64_t offset, int64_t size);
+    MidiIoWrapper(const sp<DataSource> &source);
 
     ~MidiIoWrapper();
 
@@ -37,6 +40,7 @@
     int mFd;
     off64_t mBase;
     int64_t  mLength;
+    sp<DataSource> mDataSource;
     EAS_FILE mEasFile;
 };