mirror of
https://github.com/projectacrn/acrn-hypervisor.git
synced 2025-06-28 00:06:55 +00:00
tools: acrn-crashlog: add more functions to file system utils
More file system operations for acrn-crashlog. Signed-off-by: Liu Xinwu <xinwu.liu@intel.com> Signed-off-by: CHEN Gang <gang.c.chen@intel.com> Reviewed-by: Zhang Yanmin <yanmin.zhang@intel.com> Reviewed-by: Liu Chuansheng <chuansheng.liu@intel.com> Reviewed-by: Zhao Yakui <yakui.zhao@intel.com> Reviewed-by: Geoffroy Van Cutsem <geoffroy.vancutsem@intel.com> Acked-by: Eddie Dong <Eddie.dong@intel.com>
This commit is contained in:
Liu Xinwu
lijinxia
parent
87b8968314
commit
543cdce6c6
@ -624,3 +624,534 @@ int file_update_int(const char *filename, unsigned int current,
|
|||||||
/* Close file */
|
/* Close file */
|
||||||
return close_file(filename, fp);
|
return close_file(filename, fp);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Copy file in a read/write loop.
|
||||||
|
*
|
||||||
|
* @param src Path of source file.
|
||||||
|
* @param dest Path of destin file.
|
||||||
|
*
|
||||||
|
* @return 0 if successful, or a negative value if not.
|
||||||
|
*/
|
||||||
|
int do_copy_eof(const char *src, const char *des)
|
||||||
|
{
|
||||||
|
char buffer[CPBUFFERSIZE];
|
||||||
|
int rc = 0;
|
||||||
|
int fd1 = -1, fd2 = -1;
|
||||||
|
struct stat info;
|
||||||
|
int r_count, w_count = 0;
|
||||||
|
|
||||||
|
if (src == NULL || des == NULL)
|
||||||
|
return -EINVAL;
|
||||||
|
|
||||||
|
if (stat(src, &info) < 0) {
|
||||||
|
LOGE("can not open file: %s\n", src);
|
||||||
|
return -errno;
|
||||||
|
}
|
||||||
|
|
||||||
|
fd1 = open(src, O_RDONLY);
|
||||||
|
if (fd1 < 0)
|
||||||
|
return -errno;
|
||||||
|
|
||||||
|
fd2 = open(des, O_WRONLY | O_CREAT | O_TRUNC, 0660);
|
||||||
|
if (fd2 < 0) {
|
||||||
|
LOGE("can not open file: %s\n", des);
|
||||||
|
close(fd1);
|
||||||
|
return -errno;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Start copy loop */
|
||||||
|
while (1) {
|
||||||
|
/* Read data from src */
|
||||||
|
r_count = read(fd1, buffer, CPBUFFERSIZE);
|
||||||
|
if (r_count < 0) {
|
||||||
|
LOGE("read failed, err:%s\n", strerror(errno));
|
||||||
|
rc = -1;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (r_count == 0)
|
||||||
|
break;
|
||||||
|
|
||||||
|
/* Copy data to des */
|
||||||
|
w_count = write(fd2, buffer, r_count);
|
||||||
|
if (w_count < 0) {
|
||||||
|
LOGE("write failed, err:%s\n", strerror(errno));
|
||||||
|
rc = -1;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
if (r_count != w_count) {
|
||||||
|
LOGE("write failed, r_count:%d w_count:%d\n",
|
||||||
|
r_count, w_count);
|
||||||
|
rc = -1;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (fd1 >= 0)
|
||||||
|
close(fd1);
|
||||||
|
if (fd2 >= 0)
|
||||||
|
close(fd2);
|
||||||
|
|
||||||
|
return rc;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Check the storage space.
|
||||||
|
*
|
||||||
|
* @param path Path to the file system.
|
||||||
|
* @param quota Threshold value.
|
||||||
|
* @return 1 if the percentage of using space is lower than the specified quota.
|
||||||
|
* or 0 if not.
|
||||||
|
*/
|
||||||
|
int space_available(char *path, int quota)
|
||||||
|
{
|
||||||
|
struct statfs diskInfo;
|
||||||
|
unsigned long long totalBlocks;
|
||||||
|
unsigned long long totalSize;
|
||||||
|
unsigned long long freeDisk;
|
||||||
|
int mbTotalsize;
|
||||||
|
int mbFreedisk;
|
||||||
|
int ret;
|
||||||
|
|
||||||
|
/* quota is valid in range (0, 100) */
|
||||||
|
if (quota <= 0 || quota > 100)
|
||||||
|
return 0;
|
||||||
|
|
||||||
|
ret = statfs(path, &diskInfo);
|
||||||
|
if (ret < 0) {
|
||||||
|
LOGE("statfs (%s) failed, error (%s)\n", path,
|
||||||
|
strerror(errno));
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
totalBlocks = diskInfo.f_bsize;
|
||||||
|
totalSize = totalBlocks * diskInfo.f_blocks;
|
||||||
|
freeDisk = diskInfo.f_bfree * totalBlocks;
|
||||||
|
mbTotalsize = totalSize >> 20;
|
||||||
|
mbFreedisk = freeDisk >> 20;
|
||||||
|
if ((float)mbFreedisk / (float)mbTotalsize >
|
||||||
|
1 - ((float)quota / (float)100))
|
||||||
|
return 1;
|
||||||
|
|
||||||
|
LOGE("space meet quota[%d] total=%dMB, free=%dMB\n", quota,
|
||||||
|
mbTotalsize, mbFreedisk);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Count file lines.
|
||||||
|
*
|
||||||
|
* @param filename Path of file to count lines.
|
||||||
|
*
|
||||||
|
* @return lines of file is successful, or a negative errno-style value if not.
|
||||||
|
*/
|
||||||
|
int count_lines_in_file(const char *filename)
|
||||||
|
{
|
||||||
|
struct mm_file_t *f;
|
||||||
|
int ret;
|
||||||
|
|
||||||
|
f = mmap_file(filename);
|
||||||
|
if (!f)
|
||||||
|
return -errno;
|
||||||
|
|
||||||
|
ret = mm_count_lines(f);
|
||||||
|
|
||||||
|
unmap_file(f);
|
||||||
|
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Read binary file.
|
||||||
|
*
|
||||||
|
* @param path File path to read.
|
||||||
|
* @param[out] File size being read.
|
||||||
|
* @param[out] data File content.
|
||||||
|
*
|
||||||
|
* @return 0 if successful, or -1 if not.
|
||||||
|
*/
|
||||||
|
int read_full_binary_file(const char *path, unsigned long *size, void **data)
|
||||||
|
{
|
||||||
|
FILE *f;
|
||||||
|
long _size;
|
||||||
|
void *buf;
|
||||||
|
int err;
|
||||||
|
|
||||||
|
if (!path || !data || !size) {
|
||||||
|
errno = EINVAL;
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
f = fopen(path, "rb");
|
||||||
|
if (!f)
|
||||||
|
return -1;
|
||||||
|
|
||||||
|
if (fseek(f, 0, SEEK_END) == -1) {
|
||||||
|
err = errno;
|
||||||
|
goto close;
|
||||||
|
}
|
||||||
|
|
||||||
|
_size = ftell(f);
|
||||||
|
if (_size == -1) {
|
||||||
|
err = errno;
|
||||||
|
goto close;
|
||||||
|
} else if (!_size) {
|
||||||
|
err = ERANGE;
|
||||||
|
goto close;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (fseek(f, 0, SEEK_SET) == -1) {
|
||||||
|
err = errno;
|
||||||
|
goto close;
|
||||||
|
}
|
||||||
|
|
||||||
|
buf = malloc(_size + 10);
|
||||||
|
if (!buf) {
|
||||||
|
err = ENOMEM;
|
||||||
|
goto close;
|
||||||
|
}
|
||||||
|
|
||||||
|
memset(buf, 0, _size + 10);
|
||||||
|
if (fread(buf, 1, _size, f) != (unsigned int)_size) {
|
||||||
|
err = EBADF;
|
||||||
|
goto free;
|
||||||
|
}
|
||||||
|
|
||||||
|
close_file(path, f);
|
||||||
|
|
||||||
|
*data = buf;
|
||||||
|
*size = (unsigned int)_size;
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
|
||||||
|
free:
|
||||||
|
free(buf);
|
||||||
|
close:
|
||||||
|
close_file(path, f);
|
||||||
|
errno = err;
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int _file_read_key_value(char *path, char op, char *key, char *value)
|
||||||
|
{
|
||||||
|
int fd;
|
||||||
|
int size;
|
||||||
|
int len;
|
||||||
|
char *data;
|
||||||
|
char *msg = NULL;
|
||||||
|
char *end, *start;
|
||||||
|
|
||||||
|
if (!key || !path) {
|
||||||
|
errno = EINVAL;
|
||||||
|
return -errno;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (op != 'l' && op != 'r') {
|
||||||
|
errno = EINVAL;
|
||||||
|
return -errno;
|
||||||
|
}
|
||||||
|
|
||||||
|
size = get_file_size(path);
|
||||||
|
if (size < 0)
|
||||||
|
return size;
|
||||||
|
|
||||||
|
fd = open(path, O_RDONLY);
|
||||||
|
if (fd < 0)
|
||||||
|
return -errno;
|
||||||
|
|
||||||
|
data = mmap(NULL, size, PROT_READ, MAP_PRIVATE, fd, 0);
|
||||||
|
if (data == MAP_FAILED)
|
||||||
|
goto close;
|
||||||
|
|
||||||
|
if (op == 'l')
|
||||||
|
msg = strstr(data, key);
|
||||||
|
else if (op == 'r')
|
||||||
|
msg = strrstr(data, key);
|
||||||
|
if (!msg) {
|
||||||
|
errno = ENOMSG;
|
||||||
|
goto unmap;
|
||||||
|
}
|
||||||
|
end = strchr(msg, '\n');
|
||||||
|
if (end == NULL)
|
||||||
|
end = data + size;
|
||||||
|
|
||||||
|
start = msg + strlen(key);
|
||||||
|
len = end - start;
|
||||||
|
memcpy(value, start, len);
|
||||||
|
*(value + len) = 0;
|
||||||
|
|
||||||
|
munmap(data, size);
|
||||||
|
close(fd);
|
||||||
|
|
||||||
|
return len;
|
||||||
|
|
||||||
|
unmap:
|
||||||
|
munmap(data, size);
|
||||||
|
close:
|
||||||
|
close(fd);
|
||||||
|
return -errno;
|
||||||
|
}
|
||||||
|
|
||||||
|
int file_read_key_value(char *path, char *key, char *value)
|
||||||
|
{
|
||||||
|
return _file_read_key_value(path, 'l', key, value);
|
||||||
|
}
|
||||||
|
|
||||||
|
int file_read_key_value_r(char *path, char *key, char *value)
|
||||||
|
{
|
||||||
|
return _file_read_key_value(path, 'r', key, value);
|
||||||
|
}
|
||||||
|
|
||||||
|
int dir_contains(const char *dir, const char *filename, int exact,
|
||||||
|
char *fullname)
|
||||||
|
{
|
||||||
|
int ret, count = 0;
|
||||||
|
struct dirent **filelist;
|
||||||
|
char *name;
|
||||||
|
|
||||||
|
if (!dir || !filename)
|
||||||
|
return -EINVAL;
|
||||||
|
|
||||||
|
/* pass parameters to scandir's filter is not convenience, so we use
|
||||||
|
* this implementation.
|
||||||
|
*/
|
||||||
|
ret = scandir(dir, &filelist, 0, 0);
|
||||||
|
if (ret < 0)
|
||||||
|
return -errno;
|
||||||
|
|
||||||
|
while (ret--) {
|
||||||
|
name = filelist[ret]->d_name;
|
||||||
|
if (exact) {
|
||||||
|
if (!strcmp(name, filename))
|
||||||
|
count++;
|
||||||
|
} else {
|
||||||
|
if (strstr(name, filename)) {
|
||||||
|
count++;
|
||||||
|
if (fullname)
|
||||||
|
strcpy(fullname, name);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
free(filelist[ret]);
|
||||||
|
}
|
||||||
|
|
||||||
|
free(filelist);
|
||||||
|
|
||||||
|
return count;
|
||||||
|
}
|
||||||
|
|
||||||
|
int lsdir(const char *dir, char *fullname[], int limit)
|
||||||
|
{
|
||||||
|
int ret, num, count = 0;
|
||||||
|
int i;
|
||||||
|
struct dirent **filelist;
|
||||||
|
char *name;
|
||||||
|
|
||||||
|
if (!dir || !fullname)
|
||||||
|
return -EINVAL;
|
||||||
|
|
||||||
|
num = scandir(dir, &filelist, 0, 0);
|
||||||
|
if (num < 0)
|
||||||
|
return -errno;
|
||||||
|
if (num > limit) {
|
||||||
|
LOGE("(%s) contains (%d) files, meet limit (%d)\n",
|
||||||
|
dir, num, limit);
|
||||||
|
count = -EINVAL;
|
||||||
|
goto free_list;
|
||||||
|
}
|
||||||
|
|
||||||
|
for (i = 0; i < num; i++) {
|
||||||
|
name = filelist[i]->d_name;
|
||||||
|
ret = asprintf(&fullname[count++], "%s/%s", dir, name);
|
||||||
|
if (ret < 0) {
|
||||||
|
LOGE("compute string failed, out of memory\n");
|
||||||
|
count = -ENOMEM;
|
||||||
|
goto free_fname;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for (i = 0; i < num; i++)
|
||||||
|
free(filelist[i]);
|
||||||
|
free(filelist);
|
||||||
|
|
||||||
|
return count;
|
||||||
|
|
||||||
|
free_fname:
|
||||||
|
for (i = 0; i < count; i++)
|
||||||
|
free(fullname[i]);
|
||||||
|
|
||||||
|
free_list:
|
||||||
|
for (i = 0; i < num; i++)
|
||||||
|
free(filelist[i]);
|
||||||
|
free(filelist);
|
||||||
|
|
||||||
|
return count;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int is_subdir(const struct dirent *entry)
|
||||||
|
{
|
||||||
|
return (entry->d_type == DT_DIR) &&
|
||||||
|
strcmp(entry->d_name, ".") &&
|
||||||
|
strcmp(entry->d_name, "..");
|
||||||
|
}
|
||||||
|
|
||||||
|
static void expand_dir(char *_dirs[], int *count, int depth, int max_depth)
|
||||||
|
{
|
||||||
|
int files;
|
||||||
|
int i;
|
||||||
|
int res;
|
||||||
|
struct dirent **filelist;
|
||||||
|
char *subdir;
|
||||||
|
char *name;
|
||||||
|
char *current_dir = _dirs[*count - 1];
|
||||||
|
|
||||||
|
if (depth > max_depth)
|
||||||
|
return;
|
||||||
|
|
||||||
|
|
||||||
|
files = scandir(current_dir, &filelist, is_subdir, 0);
|
||||||
|
if (files < 0) {
|
||||||
|
LOGE("lsdir failed, error (%s)\n", strerror(-files));
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
for (i = 0; i < files; i++) {
|
||||||
|
if (*count >= MAX_SEARCH_DIRS) {
|
||||||
|
LOGE("too many dirs(%d) under %s\n",
|
||||||
|
*count, _dirs[0]);
|
||||||
|
goto free;
|
||||||
|
}
|
||||||
|
|
||||||
|
name = filelist[i]->d_name;
|
||||||
|
res = asprintf(&subdir, "%s/%s", current_dir, name);
|
||||||
|
if (res < 0) {
|
||||||
|
LOGE("compute string failed, out of memory\n");
|
||||||
|
goto free;
|
||||||
|
}
|
||||||
|
|
||||||
|
_dirs[*count] = subdir;
|
||||||
|
(*count)++;
|
||||||
|
expand_dir(_dirs, count, depth++, max_depth);
|
||||||
|
}
|
||||||
|
free:
|
||||||
|
for (i = 0; i < files; i++)
|
||||||
|
free(filelist[i]);
|
||||||
|
free(filelist);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Find target file in specified dir.
|
||||||
|
*
|
||||||
|
* @param dir Where to start search.
|
||||||
|
* @param target_file Target file to search.
|
||||||
|
* @param depth File's depth in the directory tree.
|
||||||
|
* @param path[out] Searched file path in given dir.
|
||||||
|
* @param limit The number of files uplayer want to get.
|
||||||
|
*
|
||||||
|
* @return the count of searched files if successful, or a negative
|
||||||
|
* errno-style value if not.
|
||||||
|
*/
|
||||||
|
int find_file(char *dir, char *target_file, int depth, char *path[], int limit)
|
||||||
|
{
|
||||||
|
int i, ret;
|
||||||
|
int count = 0;
|
||||||
|
char *_dirs[MAX_SEARCH_DIRS];
|
||||||
|
int dirs;
|
||||||
|
|
||||||
|
if (depth < 1 || !dir || !target_file || !path || limit <= 0)
|
||||||
|
return -EINVAL;
|
||||||
|
|
||||||
|
ret = asprintf(&_dirs[0], "%s", dir);
|
||||||
|
if (ret < 0) {
|
||||||
|
LOGE("compute string failed, out of memory\n");
|
||||||
|
return -ENOMEM;
|
||||||
|
}
|
||||||
|
dirs = 1;
|
||||||
|
|
||||||
|
/* expand all dirs */
|
||||||
|
expand_dir(_dirs, &dirs, 1, depth);
|
||||||
|
for (i = 0; i < dirs; i++) {
|
||||||
|
if (count >= limit)
|
||||||
|
goto free;
|
||||||
|
|
||||||
|
ret = dir_contains(_dirs[i], target_file, 1, NULL);
|
||||||
|
if (ret == 1) {
|
||||||
|
ret = asprintf(&path[count++], "%s/%s",
|
||||||
|
_dirs[i], target_file);
|
||||||
|
if (ret < 0) {
|
||||||
|
LOGE("compute string failed, out of memory\n");
|
||||||
|
ret = -ENOMEM;
|
||||||
|
goto fail;
|
||||||
|
}
|
||||||
|
} else if (ret > 1) {
|
||||||
|
LOGE("found (%d) (%s) under (%s)??\n",
|
||||||
|
ret, target_file, dir);
|
||||||
|
} else if (ret < 0) {
|
||||||
|
LOGE("dir_contains failed, error (%s)\n",
|
||||||
|
strerror(-ret));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
free:
|
||||||
|
for (i = 0; i < dirs; i++)
|
||||||
|
free(_dirs[i]);
|
||||||
|
|
||||||
|
return count;
|
||||||
|
fail:
|
||||||
|
for (i = 0; i < dirs; i++)
|
||||||
|
free(_dirs[i]);
|
||||||
|
|
||||||
|
for (i = 0; i < count; i++)
|
||||||
|
free(path[i]);
|
||||||
|
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
int read_file(const char *path, unsigned long *size, void **data)
|
||||||
|
{
|
||||||
|
char tmp[1024] = "\0";
|
||||||
|
int len = 0;
|
||||||
|
int fd = 0;
|
||||||
|
int memsize = 1; /* for '\0' */
|
||||||
|
size_t result = 0;
|
||||||
|
char *out = NULL;
|
||||||
|
char *new;
|
||||||
|
|
||||||
|
if (!path || !data || !size) {
|
||||||
|
errno = EINVAL;
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
fd = open(path, O_RDONLY);
|
||||||
|
if (fd < 0)
|
||||||
|
return -1;
|
||||||
|
|
||||||
|
while (1) {
|
||||||
|
result = read(fd, (void *)tmp, 1024);
|
||||||
|
if (result == 0)
|
||||||
|
break;
|
||||||
|
|
||||||
|
memsize += result;
|
||||||
|
new = realloc(out, memsize);
|
||||||
|
if (!new)
|
||||||
|
goto free;
|
||||||
|
|
||||||
|
out = new;
|
||||||
|
memcpy(out + len, tmp, result);
|
||||||
|
out[memsize - 1] = 0;
|
||||||
|
|
||||||
|
len += result;
|
||||||
|
}
|
||||||
|
|
||||||
|
close(fd);
|
||||||
|
|
||||||
|
*data = out;
|
||||||
|
*size = (unsigned int)len;
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
|
||||||
|
free:
|
||||||
|
if (out)
|
||||||
|
free(out);
|
||||||
|
close(fd);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|||||||
@ -88,5 +88,17 @@ void file_reset_init(const char *filename);
|
|||||||
int file_read_int(const char *filename, unsigned int *pcurrent);
|
int file_read_int(const char *filename, unsigned int *pcurrent);
|
||||||
int file_update_int(const char *filename, unsigned int current,
|
int file_update_int(const char *filename, unsigned int current,
|
||||||
unsigned int max);
|
unsigned int max);
|
||||||
|
int do_copy_eof(const char *src, const char *des);
|
||||||
|
int space_available(char *path, int quota);
|
||||||
|
int count_lines_in_file(const char *filename);
|
||||||
|
int read_full_binary_file(const char *path, unsigned long *size,
|
||||||
|
void **data);
|
||||||
|
int file_read_key_value(char *path, char *key, char *value);
|
||||||
|
int file_read_key_value_r(char *path, char *key, char *value);
|
||||||
|
int dir_contains(const char *dir, const char *filename, int exact,
|
||||||
|
char *fullname);
|
||||||
|
int lsdir(const char *dir, char *fullname[], int limit);
|
||||||
|
int find_file(char *dir, char *target_file, int depth, char *path[], int limit);
|
||||||
|
int read_file(const char *path, unsigned long *size, void **data);
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user