You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

269 lines
8.3 KiB

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