summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorEthan Yonker <dees_troy@teamw.in>2014-09-04 19:59:27 +0200
committerEthan Yonker <dees_troy@teamw.in>2014-09-04 20:05:35 +0200
commit241a3ce4c5ff883c160754e0aebd1e1b0894d332 (patch)
treee39901d3a8b91ad599ec71754dfeba7cdea8dcc1
parentMTP Fix rename in Windows (diff)
downloadandroid_bootable_recovery-241a3ce4c5ff883c160754e0aebd1e1b0894d332.tar
android_bootable_recovery-241a3ce4c5ff883c160754e0aebd1e1b0894d332.tar.gz
android_bootable_recovery-241a3ce4c5ff883c160754e0aebd1e1b0894d332.tar.bz2
android_bootable_recovery-241a3ce4c5ff883c160754e0aebd1e1b0894d332.tar.lz
android_bootable_recovery-241a3ce4c5ff883c160754e0aebd1e1b0894d332.tar.xz
android_bootable_recovery-241a3ce4c5ff883c160754e0aebd1e1b0894d332.tar.zst
android_bootable_recovery-241a3ce4c5ff883c160754e0aebd1e1b0894d332.zip
-rwxr-xr-xmtp/MtpServer.cpp7
-rwxr-xr-xmtp/MtpStorage.cpp45
-rwxr-xr-xmtp/node.cpp21
3 files changed, 47 insertions, 26 deletions
diff --git a/mtp/MtpServer.cpp b/mtp/MtpServer.cpp
index 5dec5801e..c044cad7f 100755
--- a/mtp/MtpServer.cpp
+++ b/mtp/MtpServer.cpp
@@ -1132,7 +1132,12 @@ static void deleteRecursive(const char* path) {
strcpy(fileSpot, name);
int type = entry->d_type;
- if (entry->d_type == DT_DIR) {
+ struct stat st;
+ if (lstat(pathbuf, &st)) {
+ MTPE("Failed to lstat '%s'\n", pathbuf);
+ continue;
+ }
+ if (st.st_mode & S_IFDIR) {
deleteRecursive(pathbuf);
rmdir(pathbuf);
} else {
diff --git a/mtp/MtpStorage.cpp b/mtp/MtpStorage.cpp
index 7531ac44d..57c87745e 100755
--- a/mtp/MtpStorage.cpp
+++ b/mtp/MtpStorage.cpp
@@ -185,7 +185,7 @@ MtpObjectHandleList* MtpStorage::getObjectList(MtpStorageID storageID, MtpObject
int MtpStorage::getObjectInfo(MtpObjectHandle handle, MtpObjectInfo& info) {
struct stat st;
- uint64_t size;
+ uint64_t size = 0;
MTPD("MtpStorage::getObjectInfo handle: %d\n", handle);
for (iter i = mtpmap.begin(); i != mtpmap.end(); i++) {
Node* node = i->second->findNode(handle, i->second->Root());
@@ -196,8 +196,8 @@ int MtpStorage::getObjectInfo(MtpObjectHandle handle, MtpObjectInfo& info) {
MTPD("info.mStorageID: %d\n", info.mStorageID);
info.mParent = node->getMtpParentId();
MTPD("mParent: %d\n", info.mParent);
- lstat(node->getPath().c_str(), &st);
- size = st.st_size;
+ if (lstat(node->getPath().c_str(), &st) == 0)
+ size = st.st_size;
MTPD("size is: %llu\n", size);
info.mCompressedSize = size;//(size > 0xFFFFFFFFLL ? 0xFFFFFFFF : size);
info.mDateModified = st.st_mtime;
@@ -283,8 +283,10 @@ int MtpStorage::getObjectFilePath(MtpObjectHandle handle, MtpString& outFilePath
node = i->second->findNode(handle, i->second->Root());
MTPD("node returned: %d\n", node);
if (node != NULL) {
- lstat(node->getPath().c_str(), &st);
- outFileLength = st.st_size;
+ if (lstat(node->getPath().c_str(), &st) == 0)
+ outFileLength = st.st_size;
+ else
+ outFileLength = 0;
outFilePath = strdup(node->getPath().c_str());
MTPD("outFilePath: %s\n", node->getPath().c_str());
goto end;
@@ -312,11 +314,17 @@ int MtpStorage::readParentDirs(std::string path) {
closedir(d);
}
while ((de = readdir(d)) != NULL) {
- if (de->d_type == DT_DIR && strcmp(de->d_name, ".") == 0)
+ // Because exfat-fuse causes issues with dirent, we will use stat
+ // for some things that dirent should be able to do
+ item = path + "/" + de->d_name;
+ if (lstat(item.c_str(), &st)) {
+ MTPE("Error running lstat on '%s'\n", item.c_str());
+ return -1;
+ }
+ if ((st.st_mode & S_IFDIR) && strcmp(de->d_name, ".") == 0)
continue;
- if (de->d_type == DT_DIR && strcmp(de->d_name, "..") != 0) {
+ if ((st.st_mode & S_IFDIR) && strcmp(de->d_name, "..") != 0) {
// Handle dirs
- item = path + "/" + de->d_name;
MTPD("dir: %s\n", item.c_str());
mtpParentList.push_back(item);
parent = item.substr(0, item.find_last_of('/'));
@@ -540,9 +548,9 @@ int MtpStorage::getObjectPropertyList(MtpObjectHandle handle, uint32_t format, u
Node *node = i->second->findNode(handle, i->second->Root());
if (node != NULL) {
struct stat st;
- uint64_t size;
- lstat(node->getPath().c_str(), &st);
- size = st.st_size;
+ uint64_t size = 0;
+ if (lstat(node->getPath().c_str(), &st) == 0)
+ size = st.st_size;
propertyCodes.push_back(MTP_PROPERTY_OBJECT_SIZE);
longValues.push_back(size);
valueStrs.push_back("");
@@ -705,14 +713,18 @@ int MtpStorage::addInotifyDirs(std::string path) {
}
while ((de = readdir(d)) != NULL) {
- if (de->d_type != DT_DIR || strcmp(de->d_name, ".") == 0 || strcmp(de->d_name, "..") == 0)
- continue;
inotifypath = path + "/" + de->d_name;
+ if (lstat(inotifypath.c_str(), &st)) {
+ MTPE("Error using lstat on '%s'\n", inotifypath.c_str());
+ return -1;
+ }
+ if (!(st.st_mode & S_IFDIR) || strcmp(de->d_name, ".") == 0 || strcmp(de->d_name, "..") == 0)
+ continue;
if (addInotifyDirs(inotifypath)) {
closedir(d);
return -1;
}
- inotify_wd = inotify_add_watch(inotify_fd, inotifypath.c_str(), IN_CREATE | IN_DELETE);
+ inotify_wd = inotify_add_watch(inotify_fd, inotifypath.c_str(), WATCH_FLAGS);
inotifymap[inotify_wd] = inotifypath;
MTPD("added inotify dir: '%s'\n", inotifypath.c_str());
}
@@ -868,8 +880,9 @@ int MtpStorage::inotify_t(void) {
if (node != NULL) {
uint64_t orig_size = node->getIntProperty(MTP_PROPERTY_OBJECT_SIZE);
struct stat st;
- lstat(item.c_str(), &st);
- uint64_t new_size = (uint64_t)st.st_size;
+ uint64_t new_size = 0;
+ if (lstat(item.c_str(), &st) == 0)
+ new_size = (uint64_t)st.st_size;
if (orig_size != new_size) {
MTPD("size changed from %llu to %llu on mtpid: %i\n", orig_size, new_size, node->Mtpid());
node->updateProperty(MTP_PROPERTY_OBJECT_SIZE, new_size, "", MTP_TYPE_UINT64);
diff --git a/mtp/node.cpp b/mtp/node.cpp
index 79936c4a4..1a09c85ae 100755
--- a/mtp/node.cpp
+++ b/mtp/node.cpp
@@ -103,24 +103,27 @@ void Node::addProperties(int storageID, int parent_object) {
struct stat st;
int mFormat = 0;
uint64_t puid;
+ off_t file_size = 0;
+ std::string mtimeStr = "00101T000000";
+ std::string atimeStr = "00101T000000";
std::string mtpidStr = static_cast<std::ostringstream*>( &(std::ostringstream() << mtpid) )->str();
std::string storageIDStr = static_cast<std::ostringstream*>( &(std::ostringstream() << storageID) )->str();
std::string puidStr = storageIDStr + mtpidStr;
if ( ! (std::istringstream(puidStr) >> puid) ) puid = 0;
- stat(getPath().c_str(), &st);
- std::string mtimeStr = static_cast<std::ostringstream*>( &(std::ostringstream() << st.st_mtime) )->str();
- std::string atimeStr = static_cast<std::ostringstream*>( &(std::ostringstream() << st.st_atime) )->str();
- if (S_ISDIR(st.st_mode)) {
- mFormat = MTP_FORMAT_ASSOCIATION; // folder
- }
- else {
- mFormat = MTP_FORMAT_UNDEFINED; // file
+ mFormat = MTP_FORMAT_UNDEFINED; // file
+ if (lstat(getPath().c_str(), &st) == 0) {
+ file_size = st.st_size;
+ if (S_ISDIR(st.st_mode))
+ mFormat = MTP_FORMAT_ASSOCIATION; // folder
+ mtimeStr = static_cast<std::ostringstream*>( &(std::ostringstream() << st.st_mtime) )->str();
+ atimeStr = static_cast<std::ostringstream*>( &(std::ostringstream() << st.st_atime) )->str();
}
+
addProperty(MTP_PROPERTY_STORAGE_ID, storageID, "", MTP_TYPE_UINT32);
addProperty(MTP_PROPERTY_OBJECT_FORMAT, mFormat, "", MTP_TYPE_UINT16);
addProperty(MTP_PROPERTY_PROTECTION_STATUS, 0, "", MTP_TYPE_UINT16);
- addProperty(MTP_PROPERTY_OBJECT_SIZE, st.st_size, "", MTP_TYPE_UINT64);
+ addProperty(MTP_PROPERTY_OBJECT_SIZE, file_size, "", MTP_TYPE_UINT64);
addProperty(MTP_PROPERTY_OBJECT_FILE_NAME, 0, basename(getPath().c_str()), MTP_TYPE_STR);
MTPD("mtpid: %i, filename: '%s', parent object: %i\n", mtpid, basename(getPath().c_str()), parent_object);
addProperty(MTP_PROPERTY_DATE_MODIFIED, 0, mtimeStr, MTP_TYPE_STR);