Match-id-d23b01861e8b27062a1d27792272a26cc174c10a

This commit is contained in:
BianTanggui
2020-08-24 22:36:23 +08:00
parent bedc26e00b
commit 8bfa3f9400
12 changed files with 126 additions and 165 deletions

View File

@@ -70,13 +70,13 @@ int ParseFileByLine(char* buffer, int bufferSize, ParseFileLine fn, const char*
char resolvedPath[PATH_MAX] = {0x0};
if (realpath(filepath, resolvedPath) == NULL && errno != ENOENT) {
LogError("error: cannot canonicalize path %s.", filepath);
LOG_ERROR("error: cannot canonicalize path %s.", filepath);
return -1;
}
fp = fopen(resolvedPath, "r");
if (fp == NULL) {
LogError("cannot open file.");
LOG_ERROR("cannot open file.");
return -1;
}
@@ -154,20 +154,20 @@ int SetupDeviceCgroup(FILE *cgroupAllow, const char *devName)
ret = sprintf_s(devPath, BUF_SIZE, "/dev/%s", devName);
if (ret < 0) {
LogError("error: failed to assemble dev path for %s.", devName);
LOG_ERROR("error: failed to assemble dev path for %s.", devName);
return -1;
}
ret = stat((const char *)devPath, &devStat);
if (ret < 0) {
LogError("error: failed to get stat of %s.", devPath);
LOG_ERROR("error: failed to get stat of %s.", devPath);
return -1;
}
bool isFailed = fprintf(cgroupAllow, "c %u:%u rw", major(devStat.st_rdev), minor(devStat.st_rdev)) < 0 ||
fflush(cgroupAllow) == EOF || ferror(cgroupAllow) < 0;
if (isFailed) {
LogError("error: write devices failed.");
LOG_ERROR("error: write devices failed.");
return -1;
}
@@ -180,19 +180,19 @@ int SetupDriverCgroup(FILE *cgroupAllow)
ret = SetupDeviceCgroup(cgroupAllow, DAVINCI_MANAGER);
if (ret < 0) {
LogError("error: failed to setup cgroup for %s.", DAVINCI_MANAGER);
LOG_ERROR("error: failed to setup cgroup for %s.", DAVINCI_MANAGER);
return -1;
}
ret = SetupDeviceCgroup(cgroupAllow, DEVMM_SVM);
if (ret < 0) {
LogError("error: failed to setup cgroup for %s.", DEVMM_SVM);
LOG_ERROR("error: failed to setup cgroup for %s.", DEVMM_SVM);
return -1;
}
ret = SetupDeviceCgroup(cgroupAllow, HISI_HDC);
if (ret < 0) {
LogError("error: failed to setup cgroup for %s.", HISI_HDC);
LOG_ERROR("error: failed to setup cgroup for %s.", HISI_HDC);
return -1;
}
@@ -207,13 +207,13 @@ int GetCgroupPath(int pid, char *effPath, size_t maxSize)
ret = sprintf_s(mountPath, BUF_SIZE, "/proc/%d/mountinfo", (int)getppid());
if (ret < 0) {
LogError("error: assemble mount info path failed: ppid(%d).", getppid());
LOG_ERROR("error: assemble mount info path failed: ppid(%d).", getppid());
return -1;
}
ret = ParseFileByLine(mount, BUF_SIZE, GetCgroupMount, mountPath);
if (ret < 0) {
LogError("error: cat file content failed.");
LOG_ERROR("error: cat file content failed.");
return -1;
}
@@ -221,13 +221,13 @@ int GetCgroupPath(int pid, char *effPath, size_t maxSize)
char cgroupPath[BUF_SIZE] = {0x0};
ret = sprintf_s(cgroupPath, BUF_SIZE, "/proc/%d/cgroup", pid);
if (ret < 0) {
LogError("error: assemble cgroup path failed: pid(%d).", pid);
LOG_ERROR("error: assemble cgroup path failed: pid(%d).", pid);
return -1;
}
ret = ParseFileByLine(cgroup, BUF_SIZE, GetCgroupRoot, cgroupPath);
if (ret < 0) {
LogError("error: cat file content failed.");
LOG_ERROR("error: cat file content failed.");
return -1;
}
@@ -236,7 +236,7 @@ int GetCgroupPath(int pid, char *effPath, size_t maxSize)
ret = sprintf_s(effPath, maxSize, "%s%s%s", mount, cgroup, ALLOW_PATH);
if (ret < 0) {
LogError("error: assemble cgroup device path failed.");
LOG_ERROR("error: assemble cgroup device path failed.");
return -1;
}
@@ -251,20 +251,20 @@ int SetupCgroup(const struct ParsedConfig *config)
FILE *cgroupAllow = NULL;
if (realpath(config->cgroupPath, resolvedCgroupPath) == NULL && errno != ENOENT) {
LogError("error: cannot canonicalize cgroup path: %s.", config->cgroupPath);
LOG_ERROR("error: cannot canonicalize cgroup path: %s.", config->cgroupPath);
return -1;
}
cgroupAllow = fopen((const char *)resolvedCgroupPath, "a");
if (cgroupAllow == NULL) {
LogError("error: failed to open cgroup file: %s.", resolvedCgroupPath);
LOG_ERROR("error: failed to open cgroup file: %s.", resolvedCgroupPath);
return -1;
}
ret = SetupDriverCgroup(cgroupAllow);
if (ret < 0) {
fclose(cgroupAllow);
LogError("error: failed to setup driver cgroup.");
LOG_ERROR("error: failed to setup driver cgroup.");
return -1;
}
@@ -272,14 +272,14 @@ int SetupCgroup(const struct ParsedConfig *config)
ret = sprintf_s(deviceName, BUF_SIZE, "%s%u", DEVICE_NAME, config->devices[idx]);
if (ret < 0) {
fclose(cgroupAllow);
LogError("error: failed to assemble device path for no.%u.", config->devices[idx]);
LOG_ERROR("error: failed to assemble device path for no.%u.", config->devices[idx]);
return -1;
}
ret = SetupDeviceCgroup(cgroupAllow, (const char *)deviceName);
if (ret < 0) {
fclose(cgroupAllow);
LogError("error: failed to setup cgroup for %s.", deviceName);
LOG_ERROR("error: failed to setup cgroup for %s.", deviceName);
return -1;
}
}

View File

@@ -6,7 +6,6 @@
#include <stdio.h>
#include <stdlib.h>
#include <stdarg.h>
#include <limits.h>
#include <errno.h>
#include <sys/time.h>
@@ -32,13 +31,13 @@ int OpenLog(const char *logFile)
}
if (realpath(logFile, realPath) == NULL && errno != ENOENT) {
LogError("error: cannot canonicalize log file path %s.", logFile);
LOG_ERROR("error: cannot canonicalize log file path %s.", logFile);
return -1;
}
g_logFile = fopen((const char *)realPath, "ae");
if (g_logFile == NULL) {
LogError("error: failed to open log file %s.", realPath);
LOG_ERROR("error: failed to open log file %s.", realPath);
return -1;
}
@@ -53,7 +52,7 @@ void CloseLog()
}
}
static void WriteLog(char level, const char *fmt, va_list args)
void WriteLog(char level, const char *content)
{
struct timeval tv = {0};
struct tm *tm = NULL;
@@ -70,51 +69,6 @@ static void WriteLog(char level, const char *fmt, va_list args)
}
fprintf(g_logFile, "[%c %s.%06ld %d] ", level, buf, tv.tv_usec, g_pid);
vfprintf(g_logFile, fmt, args);
fprintf(g_logFile, "%s", content);
fputc('\n', g_logFile);
}
void LogError(const char *fmt, ...)
{
va_list args;
va_start(args, fmt);
va_list argsCopy;
va_copy(argsCopy, args);
WriteLog('E', fmt, argsCopy);
va_end(argsCopy);
vfprintf(stderr, fmt, args);
va_end(args);
}
void LogInfo(const char *fmt, ...)
{
va_list args;
va_start(args, fmt);
va_list argsCopy;
va_copy(argsCopy, args);
WriteLog('I', fmt, argsCopy);
va_end(argsCopy);
vfprintf(stdout, fmt, args);
va_end(args);
}
void LogWarning(const char *fmt, ...)
{
va_list args;
va_start(args, fmt);
va_list argsCopy;
va_copy(argsCopy, args);
WriteLog('W', fmt, argsCopy);
va_end(argsCopy);
vfprintf(stderr, fmt, args);
va_end(args);
}

View File

@@ -5,12 +5,35 @@
#ifndef _LOGGING_H
#define _LOGGING_H
#include <stdio.h>
#include <stdarg.h>
#include "securec.h"
#include "basic.h"
void SetPidForLog(int pid);
int OpenLog(const char *logFile);
void CloseLog();
void WriteLog(char level, const char *content);
void LogError(const char *fmt, ...);
void LogInfo(const char *fmt, ...);
void LogWarning(const char *fmt, ...);
#define LOG(level, fmt, ...) \
do { \
char content[BUF_SIZE] = {0}; \
int ret = sprintf_s(content, BUF_SIZE, fmt, ##__VA_ARGS__); \
if (ret < 0) { \
break; \
} \
WriteLog(level, (const char *)content); \
fprintf(stderr, "%s", (const char *)content); \
} while (0)
#define LOG_ERROR(fmt, ...) \
do { \
LOG('E', fmt, ##__VA_ARGS__); \
} while (0)
#define LOG_WARNING(fmt, ...) \
do { \
LOG('W', fmt, ##__VA_ARGS__); \
} while (0)
#endif

View File

@@ -44,7 +44,7 @@ static bool DevicesCmdArgParser(struct CmdArgs *args, const char *arg)
{
errno_t err = strcpy_s(args->devices, BUF_SIZE, arg);
if (err != EOK) {
LogError("error: failed to get devices from cmd args.");
LOG_ERROR("error: failed to get devices from cmd args.");
return false;
}
@@ -56,12 +56,12 @@ static bool PidCmdArgParser(struct CmdArgs *args, const char *arg)
errno = 0;
args->pid = strtol(optarg, NULL, DECIMAL);
if (errno != 0) {
LogError("error: failed to convert pid string from cmd args, pid string: %s.", arg);
LOG_ERROR("error: failed to convert pid string from cmd args, pid string: %s.", arg);
return false;
}
if (args->pid <= 0) {
LogError("error: invalid pid %d.", args->pid);
LOG_ERROR("error: invalid pid %d.", args->pid);
return false;
}
@@ -72,7 +72,7 @@ static bool RootfsCmdArgParser(struct CmdArgs *args, const char *arg)
{
errno_t err = strcpy_s(args->rootfs, BUF_SIZE, arg);
if (err != EOK) {
LogError("error: failed to get rootfs path from cmd args");
LOG_ERROR("error: failed to get rootfs path from cmd args");
return false;
}
@@ -83,7 +83,7 @@ static bool OptionsCmdArgParser(struct CmdArgs *args, const char *arg)
{
errno_t err = strcpy_s(args->options, BUF_SIZE, arg);
if (err != EOK) {
LogError("error: failed to get options string from cmd args");
LOG_ERROR("error: failed to get options string from cmd args");
return false;
}
@@ -93,7 +93,7 @@ static bool OptionsCmdArgParser(struct CmdArgs *args, const char *arg)
#define NUM_OF_CMD_ARGS 4
static struct {
const int c;
const char c;
CmdArgParser parser;
} g_cmdArgParsers[NUM_OF_CMD_ARGS] = {
{'d', DevicesCmdArgParser},
@@ -102,7 +102,7 @@ static struct {
{'o', OptionsCmdArgParser}
};
static int ParseOneCmdArg(struct CmdArgs *args, int indicator, const char *value)
static int ParseOneCmdArg(struct CmdArgs *args, char indicator, const char *value)
{
int i;
for (i = 0; i < NUM_OF_CMD_ARGS; i++) {
@@ -112,13 +112,13 @@ static int ParseOneCmdArg(struct CmdArgs *args, int indicator, const char *value
}
if (i == NUM_OF_CMD_ARGS) {
LogError("error: unrecognized cmd arg: indicate char: %c, value: %s.", indicator, value);
LOG_ERROR("error: unrecognized cmd arg: indicate char: %c, value: %s.", indicator, value);
return -1;
}
bool isOK = g_cmdArgParsers[i].parser(args, value);
if (!isOK) {
LogError("error: failed while parsing cmd arg, indicate char: %c, value: %s.", indicator, value);
LOG_ERROR("error: failed while parsing cmd arg, indicate char: %c, value: %s.", indicator, value);
return -1;
}
@@ -140,14 +140,14 @@ static int ParseDeviceIDs(unsigned int *idList, size_t *idListSize, char *device
token = strtok_s(devices, sep, &context);
while (token != NULL) {
if (idx >= *idListSize) {
LogError("error: too many devices(%u), support %u devices maximally", idx, *idListSize);
LOG_ERROR("error: too many devices(%u), support %u devices maximally", idx, *idListSize);
return -1;
}
errno = 0;
idList[idx] = strtoul((const char *)token, NULL, DECIMAL);
if (errno != 0) {
LogError("error: failed to convert device id (%s) from cmd args, caused by: %s.", token, strerror(errno));
LOG_ERROR("error: failed to convert device id (%s) from cmd args, caused by: %s.", token, strerror(errno));
return -1;
}
@@ -166,38 +166,38 @@ int DoPrepare(const struct CmdArgs *args, struct ParsedConfig *config)
err = strcpy_s(config->rootfs, BUF_SIZE, args->rootfs);
if (err != EOK) {
LogError("error: failed to copy rootfs path to parsed config.");
LOG_ERROR("error: failed to copy rootfs path to parsed config.");
return -1;
}
ret = ParseDeviceIDs(config->devices, &config->devicesNr, (char *)args->devices);
if (ret < 0) {
LogError("error: failed to parse device ids from cmdline argument");
LOG_ERROR("error: failed to parse device ids from cmdline argument");
return -1;
}
ret = GetNsPath(args->pid, "mnt", config->containerNsPath, BUF_SIZE);
if (ret < 0) {
LogError("error: failed to get container mnt ns path: pid(%d).", args->pid);
LOG_ERROR("error: failed to get container mnt ns path: pid(%d).", args->pid);
return -1;
}
ret = GetCgroupPath(args->pid, config->cgroupPath, BUF_SIZE);
if (ret < 0) {
LogError("error: failed to get cgroup path.");
LOG_ERROR("error: failed to get cgroup path.");
return -1;
}
char originNsPath[BUF_SIZE] = {0};
ret = GetSelfNsPath("mnt", originNsPath, BUF_SIZE);
if (ret < 0) {
LogError("error: failed to get self ns path.");
LOG_ERROR("error: failed to get self ns path.");
return -1;
}
config->originNsFd = open((const char *)originNsPath, O_RDONLY); // proc接口非外部输入
if (config->originNsFd < 0) {
LogError("error: failed to get self ns fd: %s.", originNsPath);
LOG_ERROR("error: failed to get self ns fd: %s.", originNsPath);
return -1;
}
@@ -213,28 +213,28 @@ int SetupContainer(struct CmdArgs *args)
ret = DoPrepare(args, &config);
if (ret < 0) {
LogError("error: failed to prepare nesessary config.");
LOG_ERROR("error: failed to prepare nesessary config.");
return -1;
}
// enter container's mount namespace
ret = EnterNsByPath((const char *)config.containerNsPath, CLONE_NEWNS);
if (ret < 0) {
LogError("error: failed to set to container ns: %s.", config.containerNsPath);
LOG_ERROR("error: failed to set to container ns: %s.", config.containerNsPath);
close(config.originNsFd);
return -1;
}
ret = DoMounting(&config);
if (ret < 0) {
LogError("error: failed to do mounting.");
LOG_ERROR("error: failed to do mounting.");
close(config.originNsFd);
return -1;
}
ret = SetupCgroup(&config);
if (ret < 0) {
LogError("error: failed to set up cgroup.");
LOG_ERROR("error: failed to set up cgroup.");
close(config.originNsFd);
return -1;
}
@@ -242,7 +242,7 @@ int SetupContainer(struct CmdArgs *args)
// back to original namespace
ret = EnterNsByFd(config.originNsFd, CLONE_NEWNS);
if (ret < 0) {
LogError("error: failed to set ns back.");
LOG_ERROR("error: failed to set ns back.");
close(config.originNsFd);
return -1;
}
@@ -259,29 +259,24 @@ int Process(int argc, char **argv)
struct CmdArgs args = {0};
while ((c = getopt_long(argc, argv, "d:p:r:o", g_cmdOpts, &optionIndex)) != -1) {
ret = ParseOneCmdArg(&args, c, optarg);
ret = ParseOneCmdArg(&args, (char)c, optarg);
if (ret < 0) {
LogError("error: failed to parse cmd args.");
LOG_ERROR("error: failed to parse cmd args.");
return -1;
}
}
if (!IsCmdArgsValid(&args)) {
LogError("error: information not completed or valid.");
return -1;
}
ret = ParseRuntimeOptions(args.options);
if (ret < 0) {
LogError("error: failed to parse runtime options.");
LOG_ERROR("error: information not completed or valid.");
return -1;
}
ParseRuntimeOptions(args.options);
SetPidForLog(args.pid);
ret = OpenLog(DEFAULT_LOG_FILE);
if (ret < 0) {
LogError("error: failed to open log file %s.", DEFAULT_LOG_FILE);
LOG_ERROR("error: failed to open log file %s.", DEFAULT_LOG_FILE);
return -1;
}

View File

@@ -22,13 +22,13 @@ int Mount(const char *src, const char *dst)
ret = mount(src, dst, NULL, mountFlags, NULL);
if (ret < 0) {
LogError("error: failed to mount.");
LOG_ERROR("error: failed to mount.");
return -1;
}
ret = mount(NULL, dst, NULL, remountFlags, NULL);
if (ret < 0) {
LogError("error: failed to re-mount.");
LOG_ERROR("error: failed to re-mount.");
return -1;
}
@@ -64,13 +64,13 @@ static int GetDeviceMntSrcDst(const char *rootfs, const char *deviceName,
}
if (realpath(unresolvedDst, resolvedDst) == NULL && errno != ENOENT) {
LogError("error: cannot canonicalize device dst: %s.", dst);
LOG_ERROR("error: cannot canonicalize device dst: %s.", dst);
return -1;
}
err = strcpy_s(dst, dstBufSize, (const char *)resolvedDst);
if (err != EOK) {
LogError("error: failed to copy resolved device mnt path to dst: %s.", resolvedDst);
LOG_ERROR("error: failed to copy resolved device mnt path to dst: %s.", resolvedDst);
return -1;
}
@@ -86,26 +86,26 @@ int MountDevice(const char *rootfs, const char *deviceName)
ret = GetDeviceMntSrcDst(rootfs, deviceName, &pathInfo);
if (ret < 0) {
LogError("error: failed to get device mount src and(or) dst path, device name: %s.", deviceName);
LOG_ERROR("error: failed to get device mount src and(or) dst path, device name: %s.", deviceName);
return -1;
}
struct stat srcStat;
ret = stat((const char *)src, &srcStat);
if (ret < 0) {
LogError("error: failed to stat src: %s.", src);
LOG_ERROR("error: failed to stat src: %s.", src);
return -1;
}
ret = CreateFile(dst, srcStat.st_mode);
if (ret < 0) {
LogError("error: failed to create mount dst file: %s.", dst);
LOG_ERROR("error: failed to create mount dst file: %s.", dst);
return -1;
}
ret = Mount(src, dst);
if (ret < 0) {
LogError("error: failed to mount dev.");
LOG_ERROR("error: failed to mount dev.");
return -1;
}
@@ -119,13 +119,13 @@ int DoDeviceMounting(const char *rootfs, const unsigned int ids[], size_t idsNr)
for (size_t idx = 0; idx < idsNr; idx++) {
int ret = sprintf_s(deviceName, BUF_SIZE, "%s%u", DEVICE_NAME, ids[idx]);
if (ret < 0) {
LogError("error: assemble device name failed, id: %u.", ids[idx]);
LOG_ERROR("error: assemble device name failed, id: %u.", ids[idx]);
return -1;
}
ret = MountDevice(rootfs, deviceName);
if (ret < 0) {
LogError("error: failed to mount device %s.", deviceName);
LOG_ERROR("error: failed to mount device %s.", deviceName);
return -1;
}
}
@@ -140,26 +140,26 @@ int MountFile(const char *rootfs, const char *filepath)
ret = sprintf_s(dst, BUF_SIZE, "%s%s", rootfs, filepath);
if (ret < 0) {
LogError("error: failed to assemble file mounting path, file: %s.", filepath);
LOG_ERROR("error: failed to assemble file mounting path, file: %s.", filepath);
return -1;
}
struct stat srcStat;
ret = stat(filepath, &srcStat);
if (ret < 0) {
LogWarning("warning: failed to find file %s on host, skipping", filepath);
LOG_WARNING("warning: failed to find file %s on host, skipping", filepath);
return 0;
}
ret = CreateFile(dst, srcStat.st_mode);
if (ret < 0) {
LogError("error: failed to create mount dst file: %s.", dst);
LOG_ERROR("error: failed to create mount dst file: %s.", dst);
return -1;
}
ret = Mount(filepath, dst);
if (ret < 0) {
LogError("error: failed to mount dev.");
LOG_ERROR("error: failed to mount dev.");
return -1;
}
@@ -179,19 +179,19 @@ int MountDir(const char *rootfs, const char *src)
struct stat srcStat;
ret = stat(src, &srcStat);
if (ret < 0) {
LogWarning("warning: failed to find dir %s on host, skipping", src);
LOG_WARNING("warning: failed to find dir %s on host, skipping", src);
return 0;
}
ret = MakeDirWithParent(dst, DEFAULT_DIR_MODE);
if (ret < 0) {
LogError("error: failed to make dir: %s.", dst);
LOG_ERROR("error: failed to make dir: %s.", dst);
return -1;
}
ret = Mount(src, dst);
if (ret < 0) {
LogError("error: failed to mount dir: %s to %s.", src, dst);
LOG_ERROR("error: failed to mount dir: %s to %s.", src, dst);
return -1;
}
@@ -203,19 +203,19 @@ int DoCtrlDeviceMounting(const char *rootfs)
/* device */
int ret = MountDevice(rootfs, DAVINCI_MANAGER);
if (ret < 0) {
LogError("error: failed to mount device %s.", DAVINCI_MANAGER);
LOG_ERROR("error: failed to mount device %s.", DAVINCI_MANAGER);
return -1;
}
ret = MountDevice(rootfs, DEVMM_SVM);
if (ret < 0) {
LogError("error: failed to mount device %s.", DEVMM_SVM);
LOG_ERROR("error: failed to mount device %s.", DEVMM_SVM);
return -1;
}
ret = MountDevice(rootfs, HISI_HDC);
if (ret < 0) {
LogError("error: failed to mount device %s.", HISI_HDC);
LOG_ERROR("error: failed to mount device %s.", HISI_HDC);
return -1;
}
@@ -227,31 +227,31 @@ int DoDirectoryMounting(const char *rootfs)
/* directory */
int ret = MountDir(rootfs, ASCEND_DRIVER_LIB64_PATH);
if (ret < 0) {
LogError("error: failed to do mount %s.", ASCEND_DRIVER_LIB64_PATH);
LOG_ERROR("error: failed to do mount %s.", ASCEND_DRIVER_LIB64_PATH);
return -1;
}
ret = MountDir(rootfs, ASCEND_DRIVER_TOOLS_PATH);
if (ret < 0) {
LogError("error: failed to do mount %s.", ASCEND_DRIVER_TOOLS_PATH);
LOG_ERROR("error: failed to do mount %s.", ASCEND_DRIVER_TOOLS_PATH);
return -1;
}
ret = MountDir(rootfs, ASCEND_DRIVER_INC_PATH);
if (ret < 0) {
LogError("error: failed to do mount %s.", ASCEND_DRIVER_INC_PATH);
LOG_ERROR("error: failed to do mount %s.", ASCEND_DRIVER_INC_PATH);
return -1;
}
ret = MountDir(rootfs, ASCEND_ADDONS_PATH);
if (ret < 0) {
LogError("error: failed to do mount %s.", ASCEND_ADDONS_PATH);
LOG_ERROR("error: failed to do mount %s.", ASCEND_ADDONS_PATH);
return -1;
}
ret = MountDir(rootfs, ASCEND_DCMI_PATH);
if (ret < 0) {
LogError("error: failed to do mount %s.", ASCEND_DCMI_PATH);
LOG_ERROR("error: failed to do mount %s.", ASCEND_DCMI_PATH);
return -1;
}
@@ -262,13 +262,13 @@ int DoFileMounting(const char *rootfs)
{
int ret = MountFile(rootfs, ASCEND_NPU_SMI_PATH);
if (ret < 0) {
LogError("error: failed to do mount %s.", ASCEND_NPU_SMI_PATH);
LOG_ERROR("error: failed to do mount %s.", ASCEND_NPU_SMI_PATH);
return -1;
}
MountFile(rootfs, ASCEND_SLOG_CONF_PATH);
if (ret < 0) {
LogError("error: failed to do mount %s.", ASCEND_SLOG_CONF_PATH);
LOG_ERROR("error: failed to do mount %s.", ASCEND_SLOG_CONF_PATH);
return -1;
}
@@ -281,13 +281,13 @@ int DoMounting(const struct ParsedConfig *config)
ret = DoDeviceMounting(config->rootfs, config->devices, config->devicesNr);
if (ret < 0) {
LogError("error: failed to mount devices.");
LOG_ERROR("error: failed to mount devices.");
return -1;
}
ret = DoCtrlDeviceMounting(config->rootfs);
if (ret < 0) {
LogError("error: failed to mount ctrl devices.");
LOG_ERROR("error: failed to mount ctrl devices.");
return -1;
}
@@ -297,13 +297,13 @@ int DoMounting(const struct ParsedConfig *config)
ret = DoFileMounting(config->rootfs);
if (ret < 0) {
LogError("error: failed to mount files.");
LOG_ERROR("error: failed to mount files.");
return -1;
}
ret = DoDirectoryMounting(config->rootfs);
if (ret < 0) {
LogError("error: failed to do mount directories.");
LOG_ERROR("error: failed to do mount directories.");
return -1;
}

View File

@@ -28,7 +28,7 @@ int EnterNsByFd(int fd, int nsType)
{
int ret = setns(fd, nsType);
if (ret < 0) {
LogError("error: failed to set ns: fd(%d).", fd);
LOG_ERROR("error: failed to set ns: fd(%d).", fd);
return -1;
}
@@ -42,13 +42,13 @@ int EnterNsByPath(const char *path, int nsType)
fd = open(path, O_RDONLY); // proc文件接口非外部输入
if (fd < 0) {
LogError("error: failed to open ns path: %s.", path);
LOG_ERROR("error: failed to open ns path: %s.", path);
return -1;
}
ret = EnterNsByFd(fd, nsType);
if (ret < 0) {
LogError("error: failed to set ns: %s.", path);
LOG_ERROR("error: failed to set ns: %s.", path);
close(fd);
return -1;
}

View File

@@ -22,7 +22,7 @@ static struct {
{NULL, NULL}
};
int ParseRuntimeOptions(const char *options)
void ParseRuntimeOptions(const char *options)
{
// set defaults value
g_runtimeOptions.noDrv = false;
@@ -47,7 +47,6 @@ int ParseRuntimeOptions(const char *options)
}
free(runtimeOptions);
return 0;
}
bool IsOptionNoDrvSet()

View File

@@ -7,7 +7,7 @@
#include <stdbool.h>
int ParseRuntimeOptions(const char *options);
void ParseRuntimeOptions(const char *options);
bool IsOptionNoDrvSet();
bool IsOptionVerboseSet();

View File

@@ -100,19 +100,19 @@ int CreateFile(const char *path, mode_t mode)
int ret = MakeDirWithParent(parentDir, DEFAULT_DIR_MODE);
if (ret < 0) {
LogError("error: failed to make parent dir for file: %s", path);
LOG_ERROR("error: failed to make parent dir for file: %s", path);
return -1;
}
char resolvedPath[PATH_MAX] = {0};
if (realpath(path, resolvedPath) == NULL && errno != ENOENT) {
LogError("error: failed to resolve path %s.", path);
LOG_ERROR("error: failed to resolve path %s.", path);
return -1;
}
int fd = open(resolvedPath, O_NOFOLLOW | O_CREAT, mode);
if (fd < 0) {
LogError("error: cannot create file: %s.", resolvedPath);
LOG_ERROR("error: cannot create file: %s.", resolvedPath);
return -1;
}
close(fd);

View File

@@ -1,3 +1,7 @@
/*
* Copyright (c) Huawei Technologies Co., Ltd. 2020-2020. All rights reserved.
* Description: 测试框架
*/
#ifndef __PUBLIC_STUB_H__
#define __PUBLIC_STUB_H__

View File

@@ -1,3 +1,7 @@
/*
* Copyright (c) Huawei Technologies Co., Ltd. 2020-2020. All rights reserved.
* Description: 测试框架
*/
#ifndef __UT_COMM_H__
#define __UT_COMM_H__

View File

@@ -1,33 +1,15 @@
/*
* Copyright (c) Huawei Technologies Co., Ltd. 2020-2020. All rights reserved.
* Description: 测试框架
*/
#include "gtest/gtest.h"
#include <stdio.h>
#include <signal.h>
#include <execinfo.h>
using namespace testing;
void handler(int sig)
{
void *array[20];
size_t size;
// get void*'s for all entries on the stack
size = backtrace(array, 20);
// print out all the frames to stderr
fprintf(stderr, "Error: signal %d:\n", sig);
backtrace_symbols_fd(array, size, STDERR_FILENO);
exit(sig);
}
int main(int argc, char* argv[])
{
signal(SIGABRT, handler);
signal(SIGFPE, handler);
signal(SIGILL, handler);
signal(SIGINT, handler);
signal(SIGSEGV, handler);
signal(SIGTERM, handler);
//testing::InitGoogleTest(&argc, argv);
int result = testing::Init_UT(argc, argv,true);