330 lines
7.6 KiB
C
330 lines
7.6 KiB
C
/*
|
|
* Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
|
|
*
|
|
* This program is free software: you can use, redistribute, and/or modify
|
|
* it under the terms of the GNU Affero General Public License, version 3
|
|
* or later ("AGPL"), as published by the Free Software Foundation.
|
|
*
|
|
* This program is distributed in the hope that it will be useful, but WITHOUT
|
|
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
* FITNESS FOR A PARTICULAR PURPOSE.
|
|
*
|
|
* You should have received a copy of the GNU Affero General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#define _DEFAULT_SOURCE
|
|
#define ALLOW_FORBID_FUNC
|
|
|
|
#include "os.h"
|
|
|
|
#ifdef WINDOWS
|
|
|
|
#include <windows.h>
|
|
|
|
typedef struct TdDirEntry {
|
|
WIN32_FIND_DATA findFileData;
|
|
} TdDirEntry;
|
|
|
|
typedef struct TdDir {
|
|
TdDirEntry dirEntry;
|
|
HANDLE hFind;
|
|
} TdDir;
|
|
|
|
int wordexp(char *words, wordexp_t *pwordexp, int flags) {
|
|
pwordexp->we_offs = 0;
|
|
pwordexp->we_wordc = 1;
|
|
pwordexp->we_wordv[0] = pwordexp->wordPos;
|
|
|
|
memset(pwordexp->wordPos, 0, 1025);
|
|
if (_fullpath(pwordexp->wordPos, words, 1024) == NULL) {
|
|
pwordexp->we_wordv[0] = words;
|
|
printf("failed to parse relative path:%s to abs path\n", words);
|
|
return -1;
|
|
}
|
|
|
|
// printf("parse relative path:%s to abs path:%s\n", words, pwordexp->wordPos);
|
|
return 0;
|
|
}
|
|
|
|
void wordfree(wordexp_t *pwordexp) {}
|
|
|
|
#else
|
|
|
|
#include <dirent.h>
|
|
#include <fcntl.h>
|
|
#include <sys/stat.h>
|
|
#include <unistd.h>
|
|
#include <wordexp.h>
|
|
|
|
typedef struct dirent dirent;
|
|
typedef struct DIR TdDir;
|
|
typedef struct dirent TdDirEntry;
|
|
|
|
#endif
|
|
|
|
void taosRemoveDir(const char *dirname) {
|
|
TdDirPtr pDir = taosOpenDir(dirname);
|
|
if (pDir == NULL) return;
|
|
|
|
TdDirEntryPtr de = NULL;
|
|
while ((de = taosReadDir(pDir)) != NULL) {
|
|
if (strcmp(taosGetDirEntryName(de), ".") == 0 || strcmp(taosGetDirEntryName(de), "..") == 0) continue;
|
|
|
|
char filename[1024] = {0};
|
|
snprintf(filename, sizeof(filename), "%s%s%s", dirname, TD_DIRSEP, taosGetDirEntryName(de));
|
|
if (taosDirEntryIsDir(de)) {
|
|
taosRemoveDir(filename);
|
|
} else {
|
|
(void)taosRemoveFile(filename);
|
|
// printf("file:%s is removed\n", filename);
|
|
}
|
|
}
|
|
|
|
taosCloseDir(&pDir);
|
|
rmdir(dirname);
|
|
|
|
// printf("dir:%s is removed\n", dirname);
|
|
return;
|
|
}
|
|
|
|
bool taosDirExist(const char *dirname) { return taosCheckExistFile(dirname); }
|
|
|
|
int32_t taosMkDir(const char *dirname) {
|
|
int32_t code = mkdir(dirname, 0755);
|
|
if (code < 0 && errno == EEXIST) {
|
|
return 0;
|
|
}
|
|
|
|
return code;
|
|
}
|
|
|
|
int32_t taosMulMkDir(const char *dirname) {
|
|
if (dirname == NULL) return -1;
|
|
char * temp = strdup(dirname);
|
|
char * pos = temp;
|
|
int32_t code = 0;
|
|
|
|
if (strncmp(temp, "/", 1) == 0) {
|
|
pos += 1;
|
|
} else if (strncmp(temp, "./", 2) == 0) {
|
|
pos += 2;
|
|
}
|
|
|
|
for (; *pos != '\0'; pos++) {
|
|
if (*pos == '/') {
|
|
*pos = '\0';
|
|
code = mkdir(temp, 0755);
|
|
if (code < 0 && errno != EEXIST) {
|
|
free(temp);
|
|
return code;
|
|
}
|
|
*pos = '/';
|
|
}
|
|
}
|
|
|
|
if (*(pos - 1) != '/') {
|
|
code = mkdir(temp, 0755);
|
|
if (code < 0 && errno != EEXIST) {
|
|
free(temp);
|
|
return code;
|
|
}
|
|
}
|
|
free(temp);
|
|
|
|
// int32_t code = mkdir(dirname, 0755);
|
|
if (code < 0 && errno == EEXIST) {
|
|
return 0;
|
|
}
|
|
|
|
return code;
|
|
}
|
|
|
|
void taosRemoveOldFiles(const char *dirname, int32_t keepDays) {
|
|
TdDirPtr pDir = taosOpenDir(dirname);
|
|
if (pDir == NULL) return;
|
|
|
|
int64_t sec = taosGetTimestampSec();
|
|
TdDirEntryPtr de = NULL;
|
|
|
|
while ((de = taosReadDir(pDir)) != NULL) {
|
|
if (strcmp(taosGetDirEntryName(de), ".") == 0 || strcmp(taosGetDirEntryName(de), "..") == 0) continue;
|
|
|
|
char filename[1024];
|
|
snprintf(filename, sizeof(filename), "%s/%s", dirname, taosGetDirEntryName(de));
|
|
if (taosDirEntryIsDir(de)) {
|
|
continue;
|
|
} else {
|
|
int32_t len = (int32_t)strlen(filename);
|
|
if (len > 3 && strcmp(filename + len - 3, ".gz") == 0) {
|
|
len -= 3;
|
|
}
|
|
|
|
int64_t fileSec = 0;
|
|
for (int32_t i = len - 1; i >= 0; i--) {
|
|
if (filename[i] == '.') {
|
|
fileSec = atoll(filename + i + 1);
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (fileSec <= 100) continue;
|
|
int32_t days = (int32_t)(TABS(sec - fileSec) / 86400 + 1);
|
|
if (days > keepDays) {
|
|
(void)taosRemoveFile(filename);
|
|
// printf("file:%s is removed, days:%d keepDays:%d", filename, days, keepDays);
|
|
} else {
|
|
// printf("file:%s won't be removed, days:%d keepDays:%d", filename, days, keepDays);
|
|
}
|
|
}
|
|
}
|
|
|
|
taosCloseDir(&pDir);
|
|
rmdir(dirname);
|
|
}
|
|
|
|
int32_t taosExpandDir(const char *dirname, char *outname, int32_t maxlen) {
|
|
wordexp_t full_path;
|
|
if (0 != wordexp(dirname, &full_path, 0)) {
|
|
// printf("failed to expand path:%s since %s", dirname, strerror(errno));
|
|
wordfree(&full_path);
|
|
return -1;
|
|
}
|
|
|
|
if (full_path.we_wordv != NULL && full_path.we_wordv[0] != NULL) {
|
|
strncpy(outname, full_path.we_wordv[0], maxlen);
|
|
}
|
|
|
|
wordfree(&full_path);
|
|
|
|
return 0;
|
|
}
|
|
|
|
int32_t taosRealPath(char *dirname, char *realPath, int32_t maxlen) {
|
|
char tmp[PATH_MAX] = {0};
|
|
#ifdef WINDOWS
|
|
if (_fullpath(tmp, dirname, maxlen) != NULL) {
|
|
#else
|
|
if (realpath(dirname, tmp) != NULL) {
|
|
#endif
|
|
if (realPath == NULL) {
|
|
strncpy(dirname, tmp, maxlen);
|
|
} else {
|
|
strncpy(realPath, tmp, maxlen);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
return -1;
|
|
}
|
|
|
|
bool taosIsDir(const char *dirname) {
|
|
TdDirPtr pDir = taosOpenDir(dirname);
|
|
if (pDir != NULL) {
|
|
taosCloseDir(&pDir);
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
char *taosDirName(char *name) {
|
|
#ifdef WINDOWS
|
|
char Drive1[MAX_PATH], Dir1[MAX_PATH];
|
|
_splitpath(name, Drive1, Dir1, NULL, NULL);
|
|
size_t dirNameLen = strlen(Drive1) + strlen(Dir1);
|
|
if (dirNameLen > 0) {
|
|
name[dirNameLen] = 0;
|
|
}
|
|
return name;
|
|
#else
|
|
return dirname(name);
|
|
#endif
|
|
}
|
|
|
|
char *taosDirEntryBaseName(char *name) {
|
|
#ifdef WINDOWS
|
|
char Filename1[MAX_PATH], Ext1[MAX_PATH];
|
|
_splitpath(name, NULL, NULL, Filename1, Ext1);
|
|
return name + (strlen(name) - strlen(Filename1) - strlen(Ext1));
|
|
#else
|
|
return (char *)basename(name);
|
|
#endif
|
|
}
|
|
|
|
TdDirPtr taosOpenDir(const char *dirname) {
|
|
if (dirname == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
#ifdef WINDOWS
|
|
char szFind[MAX_PATH]; //这是要找的
|
|
HANDLE hFind;
|
|
|
|
TdDirPtr pDir = taosMemoryMalloc(sizeof(TdDir));
|
|
|
|
strcpy(szFind, dirname);
|
|
strcat(szFind, "\\*.*"); //利用通配符找这个目录下的所以文件，包括目录
|
|
|
|
pDir->hFind = FindFirstFile(szFind, &(pDir->dirEntry.findFileData));
|
|
if (INVALID_HANDLE_VALUE == pDir->hFind) {
|
|
taosMemoryFree(pDir);
|
|
return NULL;
|
|
}
|
|
return pDir;
|
|
#else
|
|
return (TdDirPtr)opendir(dirname);
|
|
#endif
|
|
}
|
|
|
|
TdDirEntryPtr taosReadDir(TdDirPtr pDir) {
|
|
if (pDir == NULL) {
|
|
return NULL;
|
|
}
|
|
#ifdef WINDOWS
|
|
if (!FindNextFile(pDir->hFind, &(pDir->dirEntry.findFileData))) {
|
|
return NULL;
|
|
}
|
|
return (TdDirEntryPtr) & (pDir->dirEntry.findFileData);
|
|
#else
|
|
return (TdDirEntryPtr)readdir((DIR *)pDir);
|
|
#endif
|
|
}
|
|
|
|
bool taosDirEntryIsDir(TdDirEntryPtr pDirEntry) {
|
|
if (pDirEntry == NULL) {
|
|
return false;
|
|
}
|
|
#ifdef WINDOWS
|
|
return (pDirEntry->findFileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) != 0;
|
|
#else
|
|
return (((dirent *)pDirEntry)->d_type & DT_DIR) != 0;
|
|
#endif
|
|
}
|
|
|
|
char *taosGetDirEntryName(TdDirEntryPtr pDirEntry) {
|
|
if (pDirEntry == NULL) {
|
|
return NULL;
|
|
}
|
|
#ifdef WINDOWS
|
|
return pDirEntry->findFileData.cFileName;
|
|
#else
|
|
return ((dirent *)pDirEntry)->d_name;
|
|
#endif
|
|
}
|
|
|
|
int32_t taosCloseDir(TdDirPtr *ppDir) {
|
|
if (ppDir == NULL || *ppDir == NULL) {
|
|
return -1;
|
|
}
|
|
#ifdef WINDOWS
|
|
FindClose((*ppDir)->hFind);
|
|
taosMemoryFree(*ppDir);
|
|
*ppDir = NULL;
|
|
return 0;
|
|
#else
|
|
closedir((DIR *)*ppDir);
|
|
*ppDir = NULL;
|
|
return 0;
|
|
#endif
|
|
}
|