mirror of
https://github.com/projectacrn/acrn-hypervisor.git
synced 2025-05-03 05:56:57 +00:00
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>
1158 lines
22 KiB
C
1158 lines
22 KiB
C
/*
|
|
* Copyright (C) 2018 Intel Corporation
|
|
* SPDX-License-Identifier: BSD-3-Clause
|
|
*/
|
|
|
|
/*
|
|
* Copyright (C) 2018 Intel Corporation
|
|
*
|
|
* 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.
|
|
*/
|
|
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <sys/mman.h>
|
|
#include <sys/types.h>
|
|
#include <sys/stat.h>
|
|
#include <fcntl.h>
|
|
#include <unistd.h>
|
|
#include <errno.h>
|
|
#include <malloc.h>
|
|
#include <sys/sendfile.h>
|
|
#include <sys/vfs.h>
|
|
#include <sys/wait.h>
|
|
#include <stdlib.h>
|
|
#include "fsutils.h"
|
|
#include "cmdutils.h"
|
|
#include "strutils.h"
|
|
#include "log_sys.h"
|
|
|
|
#define MAX_SEARCH_DIRS 4096
|
|
|
|
/**
|
|
* Help function to do fclose. In some cases (full partition),
|
|
* file closure could fail, so get error message here.
|
|
*
|
|
* @param filename Path of file.
|
|
* @param fp FILE pointer open by fopen.
|
|
*
|
|
* @return Upon successful completion 0 is returned.
|
|
* Otherwise, EOF is returned and errno is set to indicate the error.
|
|
*/
|
|
static int close_file(const char *filename, FILE *fp)
|
|
{
|
|
int res;
|
|
|
|
errno = 0;
|
|
res = fclose(fp);
|
|
if (res != 0) {
|
|
/* File closure could fail in some cases (full partition) */
|
|
LOGE("fclose on %s failed - error is %s\n",
|
|
filename, strerror(errno));
|
|
}
|
|
|
|
return res;
|
|
}
|
|
|
|
/**
|
|
* Create the dir, make parent directories as needed.
|
|
*
|
|
* @param path Target dir to make.
|
|
* @param mode Dir mode.
|
|
*
|
|
* @return 0 if successful, or a negative errno-style value if not.
|
|
*/
|
|
int mkdir_p(char *path)
|
|
{
|
|
if (!path)
|
|
return -EINVAL;
|
|
|
|
/* return 0 if path exists */
|
|
return exec_out2file(NULL, "mkdir -p %s", path);
|
|
}
|
|
|
|
/**
|
|
* Count file lines.
|
|
* This function defaults to all text files ending with \n.
|
|
*
|
|
* @param mfile File opened by mmap_file.
|
|
*
|
|
* @return file lines if successful, or a negative errno-style value if not.
|
|
*/
|
|
int mm_count_lines(struct mm_file_t *mfile)
|
|
{
|
|
if (!mfile || mfile->size < 0)
|
|
return -EINVAL;
|
|
|
|
return strcnt(mfile->begin, '\n');
|
|
}
|
|
|
|
/**
|
|
* Get the head of specified line from area mapped by mmap_file.
|
|
*
|
|
* @param mfile File opened by mmap_file.
|
|
* @param line Target line.
|
|
*
|
|
* @return a pointer to the head of line if successful, or NULL if not.
|
|
*/
|
|
char *mm_get_line(struct mm_file_t *mfile, int line)
|
|
{
|
|
char *begin = mfile->begin;
|
|
char *next;
|
|
int i;
|
|
char *ret;
|
|
|
|
if (line <= 0 || line > mm_count_lines(mfile))
|
|
return NULL;
|
|
else if (line == 1)
|
|
ret = begin;
|
|
else {
|
|
next = begin;
|
|
for (i = 2; i <= line; i++)
|
|
next = strchr(next, '\n') + 1;
|
|
ret = next;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
/**
|
|
* Replace the content in specified line.
|
|
*
|
|
* @param mfile File opened by mmap_file.
|
|
* @param replace New content.
|
|
* @param line Target line.
|
|
*
|
|
* @return 0 lines if successful, or a negative errno-style value if not.
|
|
*/
|
|
int mm_replace_str_line(struct mm_file_t *mfile, char *replace, int line)
|
|
{
|
|
int len, rmlen, offset;
|
|
int res;
|
|
int move_size = 0;
|
|
char *add_buf;
|
|
char *to_replace;
|
|
int oldsize = mfile->size;
|
|
int fd = mfile->fd;
|
|
|
|
len = strlen(replace);
|
|
add_buf = malloc(len + 1);
|
|
if (add_buf == NULL)
|
|
return -ENOMEM;
|
|
|
|
if (replace[len - 1] != '\n')
|
|
sprintf(add_buf, "%s\n", replace);
|
|
else
|
|
sprintf(add_buf, "%s", replace);
|
|
len = strlen(add_buf);
|
|
|
|
to_replace = mm_get_line(mfile, line);
|
|
if (to_replace == NULL) {
|
|
LOGE("no line %d in %s\n", line, mfile->path);
|
|
free(add_buf);
|
|
return -EINVAL;
|
|
}
|
|
rmlen = strlinelen(to_replace);
|
|
offset = len - rmlen;
|
|
|
|
/* resize the file and add/del the space to add_buf */
|
|
if (offset > 0) {
|
|
int newsize = oldsize + offset;
|
|
|
|
/* shift right */
|
|
res = ftruncate(fd, newsize);
|
|
if (res < 0) {
|
|
free(add_buf);
|
|
return -errno;
|
|
}
|
|
|
|
if ((newsize / PAGE_SIZE) > (oldsize / PAGE_SIZE)) {
|
|
/* the size crosses PAGESIZE, we need remap */
|
|
char *old_addr;
|
|
|
|
old_addr = mfile->begin;
|
|
mfile->begin = mremap(old_addr, oldsize, newsize,
|
|
MREMAP_MAYMOVE);
|
|
if (mfile->begin == MAP_FAILED) {
|
|
free(add_buf);
|
|
return -errno;
|
|
}
|
|
/* refresh the target place after remap */
|
|
to_replace = mm_get_line(mfile, line);
|
|
}
|
|
|
|
mfile->size = newsize;
|
|
move_size = mfile->begin + oldsize - to_replace;
|
|
memmove(to_replace + offset, to_replace, move_size);
|
|
} else if (offset < 0) {
|
|
/* shift left the next line */
|
|
move_size = mfile->begin + oldsize - to_replace - rmlen;
|
|
memmove(to_replace + len, to_replace + rmlen, move_size);
|
|
mfile->size += offset;
|
|
|
|
res = ftruncate(fd, oldsize + offset);
|
|
if (res < 0) {
|
|
free(add_buf);
|
|
return -errno;
|
|
}
|
|
} /* else we don't need shift */
|
|
|
|
memcpy(to_replace, add_buf, len);
|
|
free(add_buf);
|
|
|
|
return 0;
|
|
}
|
|
|
|
/**
|
|
* Open a file and map it. The structure mm_file_t maintains the major
|
|
* infomations of this file.
|
|
*
|
|
* @param path File path need mmap.
|
|
*
|
|
* @return a pointer to the structure mm_file_t if succuessful, or NULL if not.
|
|
*/
|
|
struct mm_file_t *mmap_file(const char *path)
|
|
{
|
|
struct mm_file_t *mfile;
|
|
int size;
|
|
|
|
mfile = malloc(sizeof(struct mm_file_t));
|
|
if (!mfile) {
|
|
LOGE("malloc failed, error (%s)\n", strerror(errno));
|
|
return NULL;
|
|
}
|
|
|
|
mfile->path = strdup(path);
|
|
if (!mfile->path) {
|
|
LOGE("strdup failed, error (%s)\n", strerror(errno));
|
|
goto free_mfile;
|
|
}
|
|
|
|
mfile->fd = open(mfile->path, O_RDWR);
|
|
if (mfile->fd < 0) {
|
|
LOGE("open (%s) failed, error (%s)\n", path, strerror(errno));
|
|
goto free_path;
|
|
}
|
|
|
|
mfile->size = get_file_size(path);
|
|
if (mfile->size < 0) {
|
|
LOGE("get filesize of (%s) failed, error (%s)\n",
|
|
path, strerror(errno));
|
|
goto close_fd;
|
|
}
|
|
size = mfile->size > 0 ? mfile->size : PAGE_SIZE;
|
|
mfile->begin = mmap(NULL, size, PROT_READ|PROT_WRITE, MAP_SHARED,
|
|
mfile->fd, 0);
|
|
if (mfile->begin == MAP_FAILED) {
|
|
LOGE("mmap (%s) failed, error (%s)\n", path, strerror(errno));
|
|
goto close_fd;
|
|
}
|
|
|
|
return mfile;
|
|
|
|
close_fd:
|
|
close(mfile->fd);
|
|
free_path:
|
|
free(mfile->path);
|
|
free_mfile:
|
|
free(mfile);
|
|
return NULL;
|
|
|
|
}
|
|
|
|
/**
|
|
* Close and unmap the file opened by mmap_file.
|
|
*
|
|
* @param mfile File opened by mmap_file.
|
|
*/
|
|
void unmap_file(struct mm_file_t *mfile)
|
|
{
|
|
close(mfile->fd);
|
|
munmap(mfile->begin, mfile->size);
|
|
free(mfile->path);
|
|
free(mfile);
|
|
}
|
|
|
|
/**
|
|
* Copy the tail data from a file which supports mmap(2)-like operations
|
|
* to new file.
|
|
*
|
|
* @param src File path to copy, this file supports mmap(2)-like operations
|
|
* (i.e., it cannot be a socket).
|
|
* @param dest New file path to generate.
|
|
* @param limit Size of data, if limit equals 0, this function
|
|
* will copy entire file.
|
|
*
|
|
* @return The number of bytes written to new file if successful,
|
|
* or a negative errno-style value if not.
|
|
*/
|
|
int do_copy_tail(char *src, char *dest, int limit)
|
|
{
|
|
int rc = 0;
|
|
int fsrc = -1, fdest = -1;
|
|
struct stat info;
|
|
off_t offset = 0;
|
|
|
|
if (src == NULL || dest == NULL)
|
|
return -EINVAL;
|
|
|
|
if (stat(src, &info) < 0)
|
|
return -errno;
|
|
|
|
fsrc = open(src, O_RDONLY);
|
|
if (fsrc < 0)
|
|
return -errno;
|
|
|
|
fdest = open(dest, O_WRONLY | O_CREAT | O_TRUNC, 0660);
|
|
if (fdest < 0) {
|
|
close(fsrc);
|
|
return -errno;
|
|
}
|
|
|
|
if ((limit == 0) || (info.st_size < limit))
|
|
limit = info.st_size;
|
|
|
|
if (info.st_size > limit)
|
|
offset = info.st_size - limit;
|
|
|
|
rc = sendfile(fdest, fsrc, &offset, limit);
|
|
|
|
close(fsrc);
|
|
close(fdest);
|
|
|
|
return rc == -1 ? -errno : rc;
|
|
}
|
|
|
|
/**
|
|
* Move a file.
|
|
*
|
|
* @param src Old file path.
|
|
* @param dest New file path.
|
|
*
|
|
* @return 0 if successful, or a negative errno-style value if not.
|
|
*/
|
|
int do_mv(char *src, char *dest)
|
|
{
|
|
struct stat info;
|
|
|
|
if (src == NULL || dest == NULL)
|
|
return -EINVAL;
|
|
|
|
if (stat(src, &info) < 0)
|
|
return -errno;
|
|
|
|
/* check if destination exists */
|
|
if (stat(dest, &info)) {
|
|
/* an error, unless the destination was missing */
|
|
if (errno != ENOENT) {
|
|
LOGE("failed on '%s', err:%s", dest, strerror(errno));
|
|
return -errno;
|
|
}
|
|
}
|
|
/* attempt to move it */
|
|
if (rename(src, dest)) {
|
|
LOGE("failed on '%s', err:%s\n", src, strerror(errno));
|
|
return -errno;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
/**
|
|
* Add a string to the end of file.
|
|
*
|
|
* @param filename Path of file.
|
|
* @param text String need to be appended.
|
|
*
|
|
* @return 0 if successful, or a negative errno-style value if not.
|
|
*/
|
|
int append_file(char *filename, char *text)
|
|
{
|
|
int fd, res, len;
|
|
|
|
if (!filename || !text)
|
|
return -EINVAL;
|
|
|
|
len = strlen(text);
|
|
if (!len)
|
|
return -EINVAL;
|
|
|
|
fd = open(filename, O_RDWR | O_APPEND);
|
|
if (fd < 0)
|
|
return -errno;
|
|
|
|
res = write(fd, text, len);
|
|
close(fd);
|
|
|
|
return (res == -1 ? -errno : res);
|
|
}
|
|
|
|
/**
|
|
* Replace a string from the beginning of file.
|
|
*
|
|
* @param filename Path of file.
|
|
* @param text String need to replace.
|
|
*
|
|
* @return 0 if successful, or a negative errno-style value if not.
|
|
*/
|
|
int replace_file_head(char *filename, char *text)
|
|
{
|
|
FILE *fp;
|
|
|
|
fp = fopen(filename, "r+");
|
|
if (fp == NULL)
|
|
return -errno;
|
|
|
|
errno = 0;
|
|
fputs(text, fp);
|
|
close_file(filename, fp);
|
|
if (errno != 0)
|
|
return -errno;
|
|
|
|
return 0;
|
|
}
|
|
|
|
/**
|
|
* Change the entire file content to specified string.
|
|
* The file is created if it does not exist, otherwise it is truncated.
|
|
*
|
|
* @param filename Path of file.
|
|
* @param text String need to be appended.
|
|
*
|
|
* @return 0 if successful, or a negative errno-style value if not.
|
|
*/
|
|
int overwrite_file(char *filename, char *value)
|
|
{
|
|
FILE *fp;
|
|
int ret = 0;
|
|
|
|
if (!filename || !value)
|
|
return -EINVAL;
|
|
|
|
fp = fopen(filename, "w+");
|
|
if (fp == NULL)
|
|
return -errno;
|
|
|
|
if (fprintf(fp, "%s", value) <= 0)
|
|
ret = -errno;
|
|
|
|
close_file(filename, fp);
|
|
|
|
return ret;
|
|
}
|
|
|
|
/**
|
|
* Read line from file descriptor.
|
|
*
|
|
* @param fd File descriptor.
|
|
* @param[out] buffer Content of line.
|
|
*
|
|
* @return length of line if successful, or a negative errno-style value if not.
|
|
*/
|
|
int readline(int fd, char buffer[MAXLINESIZE])
|
|
{
|
|
int size = 0, res;
|
|
char *pbuffer = &buffer[0];
|
|
|
|
/* Read the file until end of line or file */
|
|
while ((res = read(fd, pbuffer, 1)) == 1 && size < MAXLINESIZE-1) {
|
|
if (pbuffer[0] == '\n') {
|
|
buffer[++size] = 0;
|
|
return size;
|
|
}
|
|
pbuffer++;
|
|
size++;
|
|
}
|
|
|
|
/* Check the last read result */
|
|
if (res < 0) {
|
|
/* ernno is checked in the upper layer as we could
|
|
* print the filename here
|
|
*/
|
|
return res;
|
|
}
|
|
/* last line */
|
|
buffer[size] = 0;
|
|
|
|
return size;
|
|
}
|
|
|
|
/**
|
|
* Read the first line from file.
|
|
*
|
|
* @param file Path of file.
|
|
* @param[out] string String read out.
|
|
* @param[out] size Size of string.
|
|
*
|
|
* @return length of string if successful, or a negative errno-style value
|
|
* if not.
|
|
*/
|
|
int file_read_string(const char *file, char *string, int size)
|
|
{
|
|
FILE *fp;
|
|
char *res;
|
|
char *end;
|
|
|
|
if (!file || !string)
|
|
return -EINVAL;
|
|
|
|
if (!file_exists(file))
|
|
return -ENOENT;
|
|
|
|
fp = fopen(file, "r");
|
|
if (!fp)
|
|
return -errno;
|
|
|
|
res = fgets(string, size, fp);
|
|
|
|
close_file(file, fp);
|
|
|
|
if (!res)
|
|
return -errno;
|
|
|
|
end = strchr(string, '\n');
|
|
if (end)
|
|
*end = 0;
|
|
return strlen(string);
|
|
}
|
|
|
|
/**
|
|
* Reset the string in file to zero.
|
|
*
|
|
* @param filename Path of file.
|
|
*/
|
|
void file_reset_init(const char *filename)
|
|
{
|
|
FILE *fp;
|
|
|
|
fp = fopen(filename, "w");
|
|
if (fp == NULL) {
|
|
LOGE("Cannot reset %s - %s\n", filename,
|
|
strerror(errno));
|
|
return;
|
|
}
|
|
fprintf(fp, "%4u", 0);
|
|
close_file(filename, fp);
|
|
}
|
|
|
|
/**
|
|
* Read the string in file and convert it to type int.
|
|
*
|
|
* @param filename Path of file.
|
|
* @param[out] pcurrent The result read out.
|
|
*
|
|
* @return 0 if successful, or a negative errno-style value if not.
|
|
*/
|
|
int file_read_int(const char *filename, unsigned int *pcurrent)
|
|
{
|
|
FILE *fp;
|
|
int res;
|
|
|
|
/* Open file in reading */
|
|
fp = fopen(filename, "r");
|
|
if (fp != NULL) {
|
|
/* Read current value from file */
|
|
res = fscanf(fp, "%4u", pcurrent);
|
|
if (res != 1) {
|
|
/* Set it to 0 by default */
|
|
*pcurrent = 0;
|
|
LOGE("read KO res=%d, current=%d - error is %s\n",
|
|
res, *pcurrent, strerror(errno));
|
|
res = 0;
|
|
}
|
|
/* Close file */
|
|
res = close_file(filename, fp);
|
|
} else if (errno == ENOENT) {
|
|
LOGI("File %s does not exist, fall back to folder 0.\n",
|
|
filename);
|
|
/* Initialize file */
|
|
file_reset_init(filename);
|
|
*pcurrent = 0;
|
|
res = 0;
|
|
} else {
|
|
LOGE("Cannot open file %s - error is %s.\n", filename,
|
|
strerror(errno));
|
|
res = -errno;
|
|
}
|
|
|
|
return res;
|
|
}
|
|
|
|
/**
|
|
* Update the string in file to specified value.
|
|
*
|
|
* @param filename Path of file.
|
|
* @param current The number to record.
|
|
* @param max The max value file can record, this function would record
|
|
* remainder((current+1) % max)) if the specified value exceeds max.
|
|
*
|
|
* @return 0 if successful, or a negative value if not.
|
|
*/
|
|
int file_update_int(const char *filename, unsigned int current,
|
|
unsigned int max)
|
|
{
|
|
FILE *fp;
|
|
int res;
|
|
|
|
/* Open file in reading and writing */
|
|
fp = fopen(filename, "r+");
|
|
if (fp == NULL) {
|
|
LOGE("Cannot open the file %s in update mode\n", filename);
|
|
return -1;
|
|
}
|
|
|
|
/* Write new current value in file */
|
|
res = fprintf(fp, "%4u", ((current + 1) % max));
|
|
if (res <= 0) {
|
|
LOGE("Cannot update file %s - error is %s.\n",
|
|
filename, strerror(errno));
|
|
/* Close file */
|
|
close_file(filename, fp);
|
|
return -1;
|
|
}
|
|
|
|
/* Close file */
|
|
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;
|
|
}
|