1 files changed, 8 insertions, 4 deletions
diff --git a/media/mtp/MtpClient.h b/media/mtp/MtpClient.h
index 8e2d979..b43a9e0 100644
--- a/media/mtp/MtpClient.h
+++ b/media/mtp/MtpClient.h
@@ -20,12 +20,13 @@
 #include "MtpRequestPacket.h"
 #include "MtpDataPacket.h"
 #include "MtpResponsePacket.h"
-#include "mtp.h"
-
-#include "MtpUtils.h"
+#include "MtpTypes.h"
 
 namespace android {
 
+class MtpDeviceInfo;
+class MtpStorageInfo;
+
 class MtpClient {
 private:
     struct usb_endpoint *mEndpointIn;
@@ -47,9 +48,12 @@ public:
     virtual             ~MtpClient();
 
     bool                openSession();
-    bool                getDeviceInfo();
     bool                closeSession();
 
+    MtpDeviceInfo*      getDeviceInfo();
+    MtpStorageIDList*   getStorageIDs();
+    MtpStorageInfo*     getStorageInfo(MtpStorageID storageID);
+
 private:
     bool                sendRequest(MtpOperationCode operation);
     bool                sendData(MtpOperationCode operation);