zishuo/ascend-docker-runtime
1
0
Fork 0
mirror of https://github.com/Ascend/ascend-docker-runtime.git synced 2025-10-11 18:00:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

Match-id-2dde9f027721ce1b2cc97d2c9c433ea457db8afd

Browse Source
This commit is contained in:
BianTanggui
2020-07-06 11:13:00 +08:00
parent b0c6f0d387
commit a9ec644488
7 changed files with 62 additions and 62 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
cli/src/cgrp.c
View File

@@ -73,20 +73,20 @@ int SetupDeviceCgroup(FILE *cgroupAllow, const char *devName)
ret = snprintf_s(devPath, BUF_SIZE, BUF_SIZE, "/dev/%s", devName);
if (ret < 0) {
logError("error: failed to assemble dev path for %s\n", devName);
LogError("error: failed to assemble dev path for %s\n", devName);
return -1;
}
ret = stat((const char *)devPath, &devStat);
if (ret < 0) {
logError("error: failed to get stat of %s\n", devPath);
LogError("error: failed to get stat of %s\n", 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\n");
LogError("error: write devices failed\n");
return -1;
}
@@ -99,19 +99,19 @@ int SetupDriverCgroup(FILE *cgroupAllow)
ret = SetupDeviceCgroup(cgroupAllow, DAVINCI_MANAGER);
if (ret < 0) {
logError("error: failed to setup cgroup for %s\n", DAVINCI_MANAGER);
LogError("error: failed to setup cgroup for %s\n", DAVINCI_MANAGER);
return -1;
}
ret = SetupDeviceCgroup(cgroupAllow, DEVMM_SVM);
if (ret < 0) {
logError("error: failed to setup cgroup for %s\n", DEVMM_SVM);
LogError("error: failed to setup cgroup for %s\n", DEVMM_SVM);
return -1;
}
ret = SetupDeviceCgroup(cgroupAllow, HISI_HDC);
if (ret < 0) {
logError("error: failed to setup cgroup for %s\n", HISI_HDC);
LogError("error: failed to setup cgroup for %s\n", HISI_HDC);
return -1;
}
@@ -126,13 +126,13 @@ int GetCgroupPath(const struct CmdArgs *args, char *effPath, const size_t maxSiz
ret = snprintf_s(mountPath, BUF_SIZE, BUF_SIZE, "/proc/%d/mountinfo", (int)getppid());
if (ret < 0) {
logError("error: assemble mount info path failed: ppid(%d)\n", getppid());
LogError("error: assemble mount info path failed: ppid(%d)\n", getppid());
return -1;
}
ret = CatFileContent(mount, BUF_SIZE, GetCgroupMount, mountPath);
if (ret < 0) {
logError("error: cat file content failed\n");
LogError("error: cat file content failed\n");
return -1;
}
@@ -140,13 +140,13 @@ int GetCgroupPath(const struct CmdArgs *args, char *effPath, const size_t maxSiz
char cgroupPath[BUF_SIZE] = {0x0};
ret = snprintf_s(cgroupPath, BUF_SIZE, BUF_SIZE, "/proc/%d/cgroup", args->pid);
if (ret < 0) {
logError("error: assemble cgroup path failed: pid(%d)\n", args->pid);
LogError("error: assemble cgroup path failed: pid(%d)\n", args->pid);
return -1;
}
ret = CatFileContent(cgroup, BUF_SIZE, GetCgroupRoot, cgroupPath);
if (ret < 0) {
logError("error: cat file content failed\n");
LogError("error: cat file content failed\n");
return -1;
}
@@ -155,7 +155,7 @@ int GetCgroupPath(const struct CmdArgs *args, char *effPath, const size_t maxSiz
ret = snprintf_s(effPath, BUF_SIZE, maxSize, "%s%s%s", mount, cgroup, ALLOW_PATH);
if (ret < 0) {
logError("error: assemble cgroup device path failed: \n");
LogError("error: assemble cgroup device path failed: \n");
return -1;
}
@@ -170,7 +170,7 @@ int SetupCgroup(struct CmdArgs *args, const char *cgroupPath)
FILE *cgroupAllow = NULL;
if (realpath(cgroupPath, resolvedCgroupPath) == NULL && errno != ENOENT) {
logError("error: cannot canonicalize cgroup path: %s\n", cgroupPath);
LogError("error: cannot canonicalize cgroup path: %s\n", cgroupPath);
return -1;
}
@@ -184,14 +184,14 @@ int SetupCgroup(struct CmdArgs *args, const char *cgroupPath)
cgroupAllow = fopen((const char *)resolvedCgroupPath, "a");
if (cgroupAllow == NULL) {
logError("error: failed to open cgroup file: %s\n", resolvedCgroupPath);
LogError("error: failed to open cgroup file: %s\n", resolvedCgroupPath);
return -1;
}
ret = SetupDriverCgroup(cgroupAllow);
if (ret < 0) {
fclose(cgroupAllow);
logError("error: failed to setup driver cgroup\n");
LogError("error: failed to setup driver cgroup\n");
return -1;
}
@@ -200,14 +200,14 @@ int SetupCgroup(struct CmdArgs *args, const char *cgroupPath)
ret = snprintf_s(deviceName, BUF_SIZE, BUF_SIZE, "%s%s", DEVICE_NAME, token);
if (ret < 0) {
fclose(cgroupAllow);
logError("error: failed to assemble device path for no.%s\n", token);
LogError("error: failed to assemble device path for no.%s\n", token);
return -1;
}
ret = SetupDeviceCgroup(cgroupAllow, (const char *)deviceName);
if (ret < 0) {
fclose(cgroupAllow);
logError("error: failed to setup cgroup %s\n", token);
LogError("error: failed to setup cgroup %s\n", token);
return -1;
}

4
cli/src/logging.c
View File

@@ -7,7 +7,7 @@
#include <stdio.h>
#include <stdarg.h>
void logError(const char *fmt, ...)
void LogError(const char *fmt, ...)
{
va_list args;
@@ -16,7 +16,7 @@ void logError(const char *fmt, ...)
va_end(args);
}
void logInfo(const char *fmt, ...)
void LogInfo(const char *fmt, ...)
{
va_list args;

4
cli/src/logging.h
View File

@@ -5,7 +5,7 @@
#ifndef _LOGGING_H
#define _LOGGING_H
void logError(const char *fmt, ...);
void logInfo(const char *fmt, ...);
void LogError(const char *fmt, ...);
void LogInfo(const char *fmt, ...);
#endif

22
cli/src/main.c
View File

@@ -44,26 +44,26 @@ int DoPrepare(const struct CmdArgs *args, struct ParsedConfig *config)
ret = GetNsPath(args->pid, "mnt", config->containerNsPath, BUF_SIZE);
if (ret < 0) {
logError("error: failed to get container mnt ns path: pid(%d)\n", args->pid);
LogError("error: failed to get container mnt ns path: pid(%d)\n", args->pid);
return -1;
}
ret = GetCgroupPath(args, config->cgroupPath, BUF_SIZE);
if (ret < 0) {
logError("error: failed to get cgroup path\n");
LogError("error: failed to get cgroup path\n");
return -1;
}
char originNsPath[BUF_SIZE] = {0};
ret = GetSelfNsPath("mnt", originNsPath, BUF_SIZE);
if (ret < 0) {
logError("error: failed to get self ns path\n");
LogError("error: failed to get self ns path\n");
return -1;
}
config->originNsFd = open((const char *)originNsPath, O_RDONLY); // proc接口，非外部输入
if (config->originNsFd < 0) {
logError("error: failed to get self ns fd: %s\n", originNsPath);
LogError("error: failed to get self ns fd: %s\n", originNsPath);
return -1;
}
@@ -77,28 +77,28 @@ int SetupContainer(struct CmdArgs *args)
ret = DoPrepare(args, &config);
if (ret < 0) {
logError("error: failed to prepare nesessary config\n");
LogError("error: failed to prepare nesessary config\n");
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\n", config.containerNsPath);
LogError("error: failed to set to container ns: %s\n", config.containerNsPath);
close(config.originNsFd);
return -1;
}
ret = DoMounting(args);
if (ret < 0) {
logError("error: failed to do mounting\n");
LogError("error: failed to do mounting\n");
close(config.originNsFd);
return -1;
}
ret = SetupCgroup(args, (const char *)config.cgroupPath);
if (ret < 0) {
logError("error: failed to set up cgroup\n");
LogError("error: failed to set up cgroup\n");
close(config.originNsFd);
return -1;
}
@@ -106,7 +106,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\n");
LogError("error: failed to set ns back\n");
close(config.originNsFd);
return -1;
}
@@ -151,14 +151,14 @@ int Process(int argc, char **argv)
}
break;
default:
logError("unrecongnized option\n");
LogError("unrecongnized option\n");
isSucceed = false; // unrecognized option
break;
}
}
if (!isSucceed || !IsCmdArgsValid(&args)) {
logError("error: information not completed or valid.\n");
LogError("error: information not completed or valid.\n");
return -1;
}

38
cli/src/mount.c
View File

@@ -41,13 +41,13 @@ static int GetDeviceMntSrcDst(const char *rootfs, const char *deviceName,
}
if (realpath(unresolvedDst, resolvedDst) == NULL && errno != ENOENT) {
logError("error: cannot canonicalize device dst: %s\n", dst);
LogError("error: cannot canonicalize device dst: %s\n", 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\n", resolvedDst);
LogError("error: failed to copy resolved device mnt path to dst: %s\n", resolvedDst);
return -1;
}
@@ -63,26 +63,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\n", deviceName);
LogError("error: failed to get device mount src and(or) dst path, device name: %s\n", deviceName);
return -1;
}
struct stat srcStat;
ret = stat((const char *)src, &srcStat);
if (ret < 0) {
logError("error: failed to stat src: %s\n", src);
LogError("error: failed to stat src: %s\n", src);
return -1;
}
ret = CreateFile(dst, srcStat.st_mode);
if (ret < 0) {
logError("error: failed to create mount dst file: %s\n", dst);
LogError("error: failed to create mount dst file: %s\n", dst);
return -1;
}
ret = Mount(src, dst);
if (ret < 0) {
logError("error: failed to mount dev\n");
LogError("error: failed to mount dev\n");
return -1;
}
@@ -105,13 +105,13 @@ int DoDeviceMounting(const char *rootfs, const char *devicesList)
while (token != NULL) {
int ret = snprintf_s(deviceName, BUF_SIZE, BUF_SIZE, "%s%s", DEVICE_NAME, token);
if (ret < 0) {
logError("error: assemble device name failed, id: %s\n", token);
LogError("error: assemble device name failed, id: %s\n", token);
return -1;
}
ret = MountDevice(rootfs, deviceName);
if (ret < 0) {
logError("error: failed to mount device no. %s\n", token);
LogError("error: failed to mount device no. %s\n", token);
return -1;
}
@@ -144,7 +144,7 @@ int MountDir(const char *rootfs, const char *src)
mode_t parentMode = DEFAULT_DIR_MODE;
ret = MakeParentDir(parentDir, parentMode);
if (ret < 0) {
logError("error: failed to make dir: %s\n", parentDir);
LogError("error: failed to make dir: %s\n", parentDir);
return -1;
}
}
@@ -153,14 +153,14 @@ int MountDir(const char *rootfs, const char *src)
const mode_t curMode = srcStat.st_mode;
ret = MkDir(dst, curMode);
if (ret < 0) {
logError("error: failed to make dir: %s\n", dst);
LogError("error: failed to make dir: %s\n", dst);
return -1;
}
}
ret = Mount(src, dst);
if (ret < 0) {
logError("error: failed to mount dir: %s to %s\n", src, dst);
LogError("error: failed to mount dir: %s to %s\n", src, dst);
return -1;
}
@@ -172,19 +172,19 @@ int DoCtrlDeviceMounting(const char *rootfs)
/* device */
int ret = MountDevice(rootfs, DAVINCI_MANAGER);
if (ret < 0) {
logError("error: failed to mount device %s\n", DAVINCI_MANAGER);
LogError("error: failed to mount device %s\n", DAVINCI_MANAGER);
return -1;
}
ret = MountDevice(rootfs, DEVMM_SVM);
if (ret < 0) {
logError("error: failed to mount device %s\n", DEVMM_SVM);
LogError("error: failed to mount device %s\n", DEVMM_SVM);
return -1;
}
ret = MountDevice(rootfs, HISI_HDC);
if (ret < 0) {
logError("error: failed to mount device %s\n", HISI_HDC);
LogError("error: failed to mount device %s\n", HISI_HDC);
return -1;
}
@@ -196,13 +196,13 @@ int DoDirectoryMounting(const char *rootfs)
/* directory */
int ret = MountDir(rootfs, ASCEND_DRIVER_PATH);
if (ret < 0) {
logError("error: failed to do mount %s\n", ASCEND_DRIVER_PATH);
LogError("error: failed to do mount %s\n", ASCEND_DRIVER_PATH);
return -1;
}
ret = MountDir(rootfs, ASCEND_ADDONS_PATH);
if (ret < 0) {
logError("error: failed to do mount %s\n", ASCEND_ADDONS_PATH);
LogError("error: failed to do mount %s\n", ASCEND_ADDONS_PATH);
return -1;
}
@@ -215,19 +215,19 @@ int DoMounting(const struct CmdArgs *args)
ret = DoDeviceMounting(args->rootfs, args->devices);
if (ret < 0) {
logError("error: failed to do mounts\n");
LogError("error: failed to do mounts\n");
return -1;
}
ret = DoCtrlDeviceMounting(args->rootfs);
if (ret < 0) {
logError("error: failed to do mount files\n");
LogError("error: failed to do mount files\n");
return -1;
}
ret = DoDirectoryMounting(args->rootfs);
if (ret < 0) {
logError("error: failed to do mount directory\n");
LogError("error: failed to do mount directory\n");
return -1;
}

6
cli/src/ns.c
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)\n", fd);
LogError("error: failed to set ns: fd(%d)\n", 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\n", path);
LogError("error: failed to open ns path: %s\n", path);
return -1;
}
ret = EnterNsByFd(fd, nsType);
if (ret < 0) {
logError("error: failed to set ns: %s\n", path);
LogError("error: failed to set ns: %s\n", path);
close(fd);
return -1;
}

18
cli/src/utils.c
View File

@@ -75,13 +75,13 @@ int CatFileContent(char* buffer, int bufferSize, ParseFileLine fn, const char* f
char resolvedPath[PATH_MAX] = {0x0};
if (realpath(filepath, resolvedPath) == NULL && errno != ENOENT) {
logError("error: cannot canonicalize path %s\n", filepath);
LogError("error: cannot canonicalize path %s\n", filepath);
return -1;
}
fp = fopen(resolvedPath, "r");
if (fp == NULL) {
logError("cannot open file.\n");
LogError("cannot open file.\n");
return -1;
}
@@ -122,11 +122,11 @@ int CheckDirExists(const char *dir)
{
DIR *ptr = opendir(dir);
if (NULL == ptr) {
logError("path %s not exist\n", dir);
LogError("path %s not exist\n", dir);
return -1;
}
logInfo("path %s exist\n", dir);
LogInfo("path %s exist\n", dir);
closedir(ptr);
return 0;
}
@@ -169,7 +169,7 @@ int MakeParentDir(const char *path, mode_t mode)
struct stat s;
int ret = stat(path, &s);
if (ret < 0) {
logError("error: failed to stat path: %s\n", path);
LogError("error: failed to stat path: %s\n", path);
return (MkDir(path, mode));
}
@@ -180,13 +180,13 @@ int CreateFile(const char *path, mode_t mode)
{
char resolvedPath[PATH_MAX] = {0};
if (realpath(path, resolvedPath) == NULL && errno != ENOENT) {
logError("error: failed to resolve path %s\n", path);
LogError("error: failed to resolve path %s\n", path);
return -1;
}
int fd = open(resolvedPath, O_NOFOLLOW | O_CREAT, mode);
if (fd < 0) {
logError("error: cannot create file: %s\n", resolvedPath);
LogError("error: cannot create file: %s\n", resolvedPath);
return -1;
}
close(fd);
@@ -200,13 +200,13 @@ int Mount(const char *src, const char *dst)
ret = mount(src, dst, NULL, MS_BIND, NULL);
if (ret < 0) {
logError("error: failed to mount\n");
LogError("error: failed to mount\n");
return -1;
}
ret = mount(NULL, dst, NULL, remountFlags, NULL);
if (ret < 0) {
logError("error: failed to re-mount\n");
LogError("error: failed to re-mount\n");
return -1;
}