121 lines
4.4 KiB
C++
121 lines
4.4 KiB
C++
/*
|
|
* Copyright (C) 2010 The Android Open Source Project
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*
|
|
* Copyright (C) 2014 TeamWin - bigbiff and Dees_Troy mtp database conversion to C++
|
|
*/
|
|
|
|
#ifndef _MTP_STORAGE_H
|
|
#define _MTP_STORAGE_H
|
|
|
|
#include "mtp.h"
|
|
#include "MtpObjectInfo.h"
|
|
#include <string>
|
|
#include <deque>
|
|
#include <map>
|
|
#include <libgen.h>
|
|
#include <pthread.h>
|
|
#include "btree.hpp"
|
|
#include "MtpServer.h"
|
|
#include "../tw_atomic.hpp"
|
|
|
|
class MtpDatabase;
|
|
struct inotify_event;
|
|
|
|
class MtpStorage {
|
|
|
|
private:
|
|
MtpStorageID mStorageID;
|
|
MtpString mFilePath;
|
|
MtpString mDescription;
|
|
uint64_t mMaxCapacity;
|
|
uint64_t mMaxFileSize;
|
|
// amount of free space to leave unallocated
|
|
uint64_t mReserveSpace;
|
|
bool mRemovable;
|
|
MtpServer* mServer;
|
|
typedef std::map<int, Tree*> maptree;
|
|
typedef maptree::iterator iter;
|
|
maptree mtpmap;
|
|
std::string mtpstorageparent;
|
|
android::Mutex mMutex;
|
|
|
|
public:
|
|
MtpStorage(MtpStorageID id, const char* filePath,
|
|
const char* description, uint64_t reserveSpace,
|
|
bool removable, uint64_t maxFileSize, MtpServer* refserver);
|
|
virtual ~MtpStorage();
|
|
|
|
inline MtpStorageID getStorageID() const { return mStorageID; }
|
|
int getType() const;
|
|
int getFileSystemType() const;
|
|
int getAccessCapability() const;
|
|
uint64_t getMaxCapacity();
|
|
uint64_t getFreeSpace();
|
|
const char* getDescription() const;
|
|
inline const char* getPath() const { return (const char *)mFilePath; }
|
|
inline bool isRemovable() const { return mRemovable; }
|
|
inline uint64_t getMaxFileSize() const { return mMaxFileSize; }
|
|
|
|
struct PropEntry {
|
|
MtpObjectHandle handle;
|
|
uint16_t property;
|
|
uint16_t datatype;
|
|
uint64_t intvalue;
|
|
std::string strvalue;
|
|
};
|
|
|
|
int readDir(const std::string& path, Tree* tree);
|
|
int createDB();
|
|
MtpObjectHandleList* getObjectList(MtpStorageID storageID, MtpObjectHandle parent);
|
|
int getObjectInfo(MtpObjectHandle handle, MtpObjectInfo& info);
|
|
MtpObjectHandle beginSendObject(const char* path, MtpObjectFormat format, MtpObjectHandle parent, uint64_t size, time_t modified);
|
|
void endSendObject(const char* path, MtpObjectHandle handle, MtpObjectFormat format, bool succeeded);
|
|
int getObjectPropertyList(MtpObjectHandle handle, uint32_t format, uint32_t property, int groupCode, int depth, MtpDataPacket& packet);
|
|
int getObjectFilePath(MtpObjectHandle handle, MtpString& outFilePath, int64_t& outFileLength, MtpObjectFormat& outFormat);
|
|
int deleteFile(MtpObjectHandle handle);
|
|
int renameObject(MtpObjectHandle handle, std::string newName);
|
|
int getObjectPropertyValue(MtpObjectHandle handle, MtpObjectProperty property, PropEntry& prop);
|
|
void lockMutex(int thread_type);
|
|
void unlockMutex(int thread_type);
|
|
|
|
private:
|
|
pthread_t inotify();
|
|
int inotify_t();
|
|
typedef int (MtpStorage::*ThreadPtr)(void);
|
|
typedef void* (*PThreadPtr)(void *);
|
|
std::map<int, Tree*> inotifymap; // inotify wd -> tree
|
|
pthread_t inotify_thread;
|
|
int inotify_fd;
|
|
int addInotify(Tree* tree);
|
|
void handleInotifyEvent(struct inotify_event* event);
|
|
|
|
bool sendEvents;
|
|
MtpObjectHandle handleCurrentlySending;
|
|
|
|
Node* addNewNode(bool isDir, Tree* tree, const std::string& name);
|
|
Node* findNode(MtpObjectHandle handle);
|
|
Node* findNodeByPath(const std::string& path);
|
|
std::string getNodePath(Node* node);
|
|
|
|
void queryNodeProperties(std::vector<PropEntry>& results, Node* node, uint32_t property, int groupCode, MtpStorageID storageID);
|
|
|
|
bool use_mutex;
|
|
pthread_mutex_t inMutex; // inotify mutex
|
|
pthread_mutex_t mtpMutex; // main mtp mutex
|
|
TWAtomicInt inotify_thread_kill;
|
|
};
|
|
|
|
#endif // _MTP_STORAGE_H
|