media/mtp/MtpUtils.cpp - LeafOS-Project/android_frameworks_av - Gitiles

 /*
  * 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.
  */

 #define LOG_TAG "MtpUtils"

 #include <android-base/logging.h>
 #include <android-base/unique_fd.h>
 #include <dirent.h>
 #include <fcntl.h>
 #include <string>
 #include <sys/sendfile.h>
 #include <sys/stat.h>
 #include <sys/types.h>
 #include <stdio.h>
 #include <time.h>
 #include <unistd.h>

 #include "MtpUtils.h"

 using namespace std;

 namespace android {

 constexpr unsigned long FILE_COPY_SIZE = 262144;

 static void access_ok(const char *path) {
     if (access(path, F_OK) == -1) {
         // Ignore. Failure could be common in cases of delete where
         // the metadata was updated through other paths.
     }
 }

 /*
 DateTime strings follow a compatible subset of the definition found in ISO 8601, and
 take the form of a Unicode string formatted as: "YYYYMMDDThhmmss.s". In this
 representation, YYYY shall be replaced by the year, MM replaced by the month (01-12),
 DD replaced by the day (01-31), T is a constant character 'T' delimiting time from date,
 hh is replaced by the hour (00-23), mm is replaced by the minute (00-59), and ss by the
 second (00-59). The ".s" is optional, and represents tenths of a second.
 This is followed by a UTC offset given as "[+-]zzzz" or the literal "Z", meaning UTC.
 */

 bool parseDateTime(const char* dateTime, time_t& outSeconds) {
     int year, month, day, hour, minute, second;
     if (sscanf(dateTime, "%04d%02d%02dT%02d%02d%02d",
                &year, &month, &day, &hour, &minute, &second) != 6)
         return false;

     // skip optional tenth of second
     const char* tail = dateTime + 15;
     if (tail[0] == '.' && tail[1]) tail += 2;

     // FIXME: "Z" means UTC, but non-"Z" doesn't mean local time.
     // It might be that you're in Asia/Seoul on vacation and your Android
     // device has noticed this via the network, but your camera was set to
     // America/Los_Angeles once when you bought it and doesn't know where
     // it is right now, so the camera says "20160106T081700-0800" but we
     // just ignore the "-0800" and assume local time which is actually "+0900".
     // I think to support this (without switching to Java or using icu4c)
     // you'd want to always use timegm(3) and then manually add/subtract
     // the UTC offset parsed from the string (taking care of wrapping).
     // mktime(3) ignores the tm_gmtoff field, so you can't let it do the work.
     bool useUTC = (tail[0] == 'Z');

     struct tm tm = {};
     tm.tm_sec = second;
     tm.tm_min = minute;
     tm.tm_hour = hour;
     tm.tm_mday = day;
     tm.tm_mon = month - 1;  // mktime uses months in 0 - 11 range
     tm.tm_year = year - 1900;
     tm.tm_isdst = -1;
     outSeconds = useUTC ? timegm(&tm) : mktime(&tm);

     return true;
 }

 void formatDateTime(time_t seconds, char* buffer, int bufferLength) {
     struct tm tm;

     localtime_r(&seconds, &tm);
     snprintf(buffer, bufferLength, "%04d%02d%02dT%02d%02d%02d",
         tm.tm_year + 1900,
         tm.tm_mon + 1, // localtime_r uses months in 0 - 11 range
         tm.tm_mday, tm.tm_hour, tm.tm_min, tm.tm_sec);
 }

 int makeFolder(const char *path) {
     mode_t mask = umask(0);
     int ret = mkdir((const char *)path, DIR_PERM);
     umask(mask);
     if (ret && ret != -EEXIST) {
         PLOG(ERROR) << "Failed to create folder " << path;
         ret = -1;
     } else {
         chown((const char *)path, getuid(), FILE_GROUP);
     }
     access_ok(path);
     return ret;
 }

 /**
  * Copies target path and all children to destination path.
  *
  * Returns 0 on success or a negative value indicating number of failures
  */
 int copyRecursive(const char *fromPath, const char *toPath) {
     int ret = 0;
     string fromPathStr(fromPath);
     string toPathStr(toPath);

     DIR* dir = opendir(fromPath);
     if (!dir) {
         PLOG(ERROR) << "opendir " << fromPath << " failed";
         return -1;
     }
     if (fromPathStr[fromPathStr.size()-1] != '/')
         fromPathStr += '/';
     if (toPathStr[toPathStr.size()-1] != '/')
         toPathStr += '/';

     struct dirent* entry;
     while ((entry = readdir(dir))) {
         const char* name = entry->d_name;

         // ignore "." and ".."
         if (name[0] == '.' && (name[1] == 0 || (name[1] == '.' && name[2] == 0))) {
             continue;
         }
         string oldFile = fromPathStr + name;
         string newFile = toPathStr + name;

         if (entry->d_type == DT_DIR) {
             ret += makeFolder(newFile.c_str());
             ret += copyRecursive(oldFile.c_str(), newFile.c_str());
         } else {
             ret += copyFile(oldFile.c_str(), newFile.c_str());
         }
     }
     closedir(dir);
     return ret;
 }

 int copyFile(const char *fromPath, const char *toPath) {
     auto start = std::chrono::steady_clock::now();

     android::base::unique_fd fromFd(open(fromPath, O_RDONLY));
     if (fromFd == -1) {
         PLOG(ERROR) << "Failed to open copy from " << fromPath;
         return -1;
     }
     android::base::unique_fd toFd(open(toPath, O_CREAT | O_WRONLY, FILE_PERM));
     if (toFd == -1) {
         PLOG(ERROR) << "Failed to open copy to " << toPath;
         return -1;
     }
     off_t offset = 0;

     struct stat sstat = {};
     if (stat(fromPath, &sstat) == -1)
         return -1;

     off_t length = sstat.st_size;
     int ret = 0;

     while (offset < length) {
         ssize_t transfer_length = std::min(length - offset, (off_t) FILE_COPY_SIZE);
         ret = sendfile(toFd, fromFd, &offset, transfer_length);
         if (ret != transfer_length) {
             ret = -1;
             PLOG(ERROR) << "Copying failed!";
             break;
         }
     }
     auto end = std::chrono::steady_clock::now();
     std::chrono::duration<double> diff = end - start;
     LOG(DEBUG) << "Copied a file with MTP. Time: " << diff.count() << " s, Size: " << length <<
         ", Rate: " << ((double) length) / diff.count() << " bytes/s";
     chown(toPath, getuid(), FILE_GROUP);
     access_ok(toPath);
     return ret == -1 ? -1 : 0;
 }

 void deleteRecursive(const char* path) {
     string pathStr(path);
     if (pathStr[pathStr.size()-1] != '/') {
         pathStr += '/';
     }

     DIR* dir = opendir(path);
     if (!dir) {
         PLOG(ERROR) << "opendir " << path << " failed";
         return;
     }

     struct dirent* entry;
     while ((entry = readdir(dir))) {
         const char* name = entry->d_name;

         // ignore "." and ".."
         if (name[0] == '.' && (name[1] == 0 || (name[1] == '.' && name[2] == 0))) {
             continue;
         }
         string childPath = pathStr + name;
         int success;
         if (entry->d_type == DT_DIR) {
             deleteRecursive(childPath.c_str());
             success = rmdir(childPath.c_str());
         } else {
             success = unlink(childPath.c_str());
         }
         access_ok(childPath.c_str());
         if (success == -1)
             PLOG(ERROR) << "Deleting path " << childPath << " failed";
     }
     closedir(dir);
 }

 bool deletePath(const char* path) {
     struct stat statbuf;
     int success;
     if (stat(path, &statbuf) == 0) {
         if (S_ISDIR(statbuf.st_mode)) {
             // rmdir will fail if the directory is non empty, so
             // there is no need to keep errors from deleteRecursive
             deleteRecursive(path);
             success = rmdir(path);
         } else {
             success = unlink(path);
         }
     } else {
         PLOG(ERROR) << "deletePath stat failed for " << path;
         return false;
     }
     if (success == -1)
         PLOG(ERROR) << "Deleting path " << path << " failed";
     access_ok(path);
     return success == 0;
 }

 int renameTo(const char *oldPath, const char *newPath) {
     int ret = rename(oldPath, newPath);
     access_ok(oldPath);
     access_ok(newPath);
     return ret;
 }

 // Calls access(2) on the path to update underlying filesystems,
 // then closes the fd.
 void closeObjFd(int fd, const char *path) {
     close(fd);
     access_ok(path);
 }

 }  // namespace android
	/*
	* 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.
	*/

	#define LOG_TAG "MtpUtils"

	#include <android-base/logging.h>
	#include <android-base/unique_fd.h>
	#include <dirent.h>
	#include <fcntl.h>
	#include <string>
	#include <sys/sendfile.h>
	#include <sys/stat.h>
	#include <sys/types.h>
	#include <stdio.h>
	#include <time.h>
	#include <unistd.h>

	#include "MtpUtils.h"

	using namespace std;

	namespace android {

	constexpr unsigned long FILE_COPY_SIZE = 262144;

	static void access_ok(const char *path) {
	if (access(path, F_OK) == -1) {
	// Ignore. Failure could be common in cases of delete where
	// the metadata was updated through other paths.
	}
	}

	/*
	DateTime strings follow a compatible subset of the definition found in ISO 8601, and
	take the form of a Unicode string formatted as: "YYYYMMDDThhmmss.s". In this
	representation, YYYY shall be replaced by the year, MM replaced by the month (01-12),
	DD replaced by the day (01-31), T is a constant character 'T' delimiting time from date,
	hh is replaced by the hour (00-23), mm is replaced by the minute (00-59), and ss by the
	second (00-59). The ".s" is optional, and represents tenths of a second.
	This is followed by a UTC offset given as "[+-]zzzz" or the literal "Z", meaning UTC.
	*/

	bool parseDateTime(const char* dateTime, time_t& outSeconds) {
	int year, month, day, hour, minute, second;
	if (sscanf(dateTime, "%04d%02d%02dT%02d%02d%02d",
	&year, &month, &day, &hour, &minute, &second) != 6)
	return false;

	// skip optional tenth of second
	const char* tail = dateTime + 15;
	if (tail[0] == '.' && tail[1]) tail += 2;

	// FIXME: "Z" means UTC, but non-"Z" doesn't mean local time.
	// It might be that you're in Asia/Seoul on vacation and your Android
	// device has noticed this via the network, but your camera was set to
	// America/Los_Angeles once when you bought it and doesn't know where
	// it is right now, so the camera says "20160106T081700-0800" but we
	// just ignore the "-0800" and assume local time which is actually "+0900".
	// I think to support this (without switching to Java or using icu4c)
	// you'd want to always use timegm(3) and then manually add/subtract
	// the UTC offset parsed from the string (taking care of wrapping).
	// mktime(3) ignores the tm_gmtoff field, so you can't let it do the work.
	bool useUTC = (tail[0] == 'Z');

	struct tm tm = {};
	tm.tm_sec = second;
	tm.tm_min = minute;
	tm.tm_hour = hour;
	tm.tm_mday = day;
	tm.tm_mon = month - 1; // mktime uses months in 0 - 11 range
	tm.tm_year = year - 1900;
	tm.tm_isdst = -1;
	outSeconds = useUTC ? timegm(&tm) : mktime(&tm);

	return true;
	}

	void formatDateTime(time_t seconds, char* buffer, int bufferLength) {
	struct tm tm;

	localtime_r(&seconds, &tm);
	snprintf(buffer, bufferLength, "%04d%02d%02dT%02d%02d%02d",
	tm.tm_year + 1900,
	tm.tm_mon + 1, // localtime_r uses months in 0 - 11 range
	tm.tm_mday, tm.tm_hour, tm.tm_min, tm.tm_sec);
	}

	int makeFolder(const char *path) {
	mode_t mask = umask(0);
	int ret = mkdir((const char *)path, DIR_PERM);
	umask(mask);
	if (ret && ret != -EEXIST) {
	PLOG(ERROR) << "Failed to create folder " << path;
	ret = -1;
	} else {
	chown((const char *)path, getuid(), FILE_GROUP);
	}
	access_ok(path);
	return ret;
	}

	/**
	* Copies target path and all children to destination path.
	*
	* Returns 0 on success or a negative value indicating number of failures
	*/
	int copyRecursive(const char fromPath, const char toPath) {
	int ret = 0;
	string fromPathStr(fromPath);
	string toPathStr(toPath);

	DIR* dir = opendir(fromPath);
	if (!dir) {
	PLOG(ERROR) << "opendir " << fromPath << " failed";
	return -1;
	}
	if (fromPathStr[fromPathStr.size()-1] != '/')
	fromPathStr += '/';
	if (toPathStr[toPathStr.size()-1] != '/')
	toPathStr += '/';

	struct dirent* entry;
	while ((entry = readdir(dir))) {
	const char* name = entry->d_name;

	// ignore "." and ".."
	if (name[0] == '.' && (name[1] == 0 \|\| (name[1] == '.' && name[2] == 0))) {
	continue;
	}
	string oldFile = fromPathStr + name;
	string newFile = toPathStr + name;

	if (entry->d_type == DT_DIR) {
	ret += makeFolder(newFile.c_str());
	ret += copyRecursive(oldFile.c_str(), newFile.c_str());
	} else {
	ret += copyFile(oldFile.c_str(), newFile.c_str());
	}
	}
	closedir(dir);
	return ret;
	}

	int copyFile(const char fromPath, const char toPath) {
	auto start = std::chrono::steady_clock::now();

	android::base::unique_fd fromFd(open(fromPath, O_RDONLY));
	if (fromFd == -1) {
	PLOG(ERROR) << "Failed to open copy from " << fromPath;
	return -1;
	}
	android::base::unique_fd toFd(open(toPath, O_CREAT \| O_WRONLY, FILE_PERM));
	if (toFd == -1) {
	PLOG(ERROR) << "Failed to open copy to " << toPath;
	return -1;
	}
	off_t offset = 0;

	struct stat sstat = {};
	if (stat(fromPath, &sstat) == -1)
	return -1;

	off_t length = sstat.st_size;
	int ret = 0;

	while (offset < length) {
	ssize_t transfer_length = std::min(length - offset, (off_t) FILE_COPY_SIZE);
	ret = sendfile(toFd, fromFd, &offset, transfer_length);
	if (ret != transfer_length) {
	ret = -1;
	PLOG(ERROR) << "Copying failed!";
	break;
	}
	}
	auto end = std::chrono::steady_clock::now();
	std::chrono::duration<double> diff = end - start;
	LOG(DEBUG) << "Copied a file with MTP. Time: " << diff.count() << " s, Size: " << length <<
	", Rate: " << ((double) length) / diff.count() << " bytes/s";
	chown(toPath, getuid(), FILE_GROUP);
	access_ok(toPath);
	return ret == -1 ? -1 : 0;
	}

	void deleteRecursive(const char* path) {
	string pathStr(path);
	if (pathStr[pathStr.size()-1] != '/') {
	pathStr += '/';
	}

	DIR* dir = opendir(path);
	if (!dir) {
	PLOG(ERROR) << "opendir " << path << " failed";
	return;
	}

	struct dirent* entry;
	while ((entry = readdir(dir))) {
	const char* name = entry->d_name;

	// ignore "." and ".."
	if (name[0] == '.' && (name[1] == 0 \|\| (name[1] == '.' && name[2] == 0))) {
	continue;
	}
	string childPath = pathStr + name;
	int success;
	if (entry->d_type == DT_DIR) {
	deleteRecursive(childPath.c_str());
	success = rmdir(childPath.c_str());
	} else {
	success = unlink(childPath.c_str());
	}
	access_ok(childPath.c_str());
	if (success == -1)
	PLOG(ERROR) << "Deleting path " << childPath << " failed";
	}
	closedir(dir);
	}

	bool deletePath(const char* path) {
	struct stat statbuf;
	int success;
	if (stat(path, &statbuf) == 0) {
	if (S_ISDIR(statbuf.st_mode)) {
	// rmdir will fail if the directory is non empty, so
	// there is no need to keep errors from deleteRecursive
	deleteRecursive(path);
	success = rmdir(path);
	} else {
	success = unlink(path);
	}
	} else {
	PLOG(ERROR) << "deletePath stat failed for " << path;
	return false;
	}
	if (success == -1)
	PLOG(ERROR) << "Deleting path " << path << " failed";
	access_ok(path);
	return success == 0;
	}

	int renameTo(const char oldPath, const char newPath) {
	int ret = rename(oldPath, newPath);
	access_ok(oldPath);
	access_ok(newPath);
	return ret;
	}

	// Calls access(2) on the path to update underlying filesystems,
	// then closes the fd.
	void closeObjFd(int fd, const char *path) {
	close(fd);
	access_ok(path);
	}

	} // namespace android