1 files changed, 192 insertions, 0 deletions
diff --git a/src/framework/mem/basic_mem/dmm_huge.c b/src/framework/mem/basic_mem/dmm_huge.c
new file mode 100644
index 0000000..6267b6c
--- /dev/null
+++ b/src/framework/mem/basic_mem/dmm_huge.c
@@ -0,0 +1,192 @@
+/*
+*
+* Copyright (c) 2018 Huawei Technologies Co.,Ltd.
+* 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 <sys/types.h>
+#include <sys/stat.h>
+#include <unistd.h>
+#include <fcntl.h>
+#include <sys/mman.h>
+#include <errno.h>
+
+#include "nstack_securec.h"
+#include "nstack_log.h"
+#include "nsfw_base_linux_api.h"
+
+#include "dmm_config.h"
+#include "dmm_share.h"
+#include "dmm_fs.h"
+
+#define DMM_HUGE_FMT "%s/dmm-%d-%d"     /* HUGE_DIR/dmm-pid-index */
+
+inline static void
+huge_set_path(char path[DMM_SHARE_PATH_MAX], pid_t pid, int index)
+{
+    (void) snprintf_s(path, DMM_SHARE_PATH_MAX, DMM_SHARE_PATH_MAX - 1,
+                      DMM_HUGE_FMT, DMM_HUGE_DIR, pid, index);
+    path[DMM_SHARE_PATH_MAX - 1] = 0;
+}
+
+int dmm_huge_create(struct dmm_share *share)
+{
+    int fd, ret;
+    void *base, *hint = (void *) DMM_MAIN_SHARE_BASE;
+
+    if (share->type != DMM_SHARE_HUGE)
+    {
+        NSFW_LOGERR("Type error, type:%d", share->type);
+        return -1;
+    }
+
+    huge_set_path(share->path, share->pid, 0);
+
+    NSFW_LOGINF("Start create share memory, path:'%s' size:%lu",
+                share->path, share->size);
+
+    fd = open(share->path, O_RDWR | O_CREAT, 0666);
+    if (fd < 0)
+    {
+        NSFW_LOGERR("Open file failed, path:'%s', errno=%d",
+                    share->path, errno);
+        return -1;
+    }
+
+    ret = ftruncate(fd, (off_t) share->size);
+    if (ret < 0)
+    {
+        NSFW_LOGERR("Set file size failed, path:'%s', errno=%d",
+                    share->path, errno);
+        (void) nsfw_base_close(fd);
+        return -1;
+    }
+
+    base = mmap(hint, share->size, PROT_READ | PROT_WRITE,
+                MAP_HUGETLB | MAP_SHARED | MAP_POPULATE, fd, 0);
+    if (base == MAP_FAILED)
+    {
+        NSFW_LOGERR("Map failed, path:'%s' size:%lu, errno=%d",
+                    share->path, share->size, errno);
+        (void) nsfw_base_close(fd);
+        return -1;
+    }
+    else if (hint && hint != MAP_FAILED && hint != base)
+    {
+        NSFW_LOGERR
+            ("Map address failed, path:'%s' hint:%p size:%lu, base:%p",
+             share->path, hint, share->size, base);
+        (void) munmap(base, share->size);
+        (void) nsfw_base_close(fd);
+        return -1;
+    }
+
+    share->base = base;
+
+    NSFW_LOGINF("Share memory created, size:%lu, base=%p", share->size, base);
+
+    (void) nsfw_base_close(fd);
+    return 0;
+}
+
+int dmm_huge_delete(struct dmm_share *share)
+{
+    (void) munmap(share->base, share->size);
+    (void) unlink(share->path);
+
+    return 0;
+}
+
+int dmm_huge_attach(struct dmm_share *share)
+{
+    int fd;
+    void *base;
+
+    NSFW_LOGINF("Start attach, share:%p"
+                " {type:%d pid:%d base:%p size:%lu path:'%s'}",
+                share, share->type, share->pid,
+                share->base, share->size, share->path);
+
+    if (share->type != DMM_SHARE_HUGE)
+    {
+        NSFW_LOGERR("Type error, type:%d", share->type);
+        return -1;
+    }
+
+    char *real_path = realpath(share->path, NULL);
+    if (NULL == real_path)
+    {
+        NSFW_LOGERR("Open file failed, path:'%s', errno=%d",
+                    share->path, errno);
+        return -1;
+    }
+    fd = open(real_path, O_RDWR);
+    if (fd < 0)
+    {
+        NSFW_LOGERR("Open file failed, path:'%s', errno=%d",
+                    share->path, errno);
+        (void) free(real_path);
+        return -1;
+    }
+
+    if (share->size <= 0)
+    {
+        share->size = dmm_file_size(fd);
+        if (share->size == 0)
+        {
+            NSFW_LOGERR("No file size '%s'", share->path);
+            (void) nsfw_base_close(fd);
+            (void) free(real_path);
+            return -1;
+        }
+    }
+
+    base = mmap(share->base, share->size, PROT_READ | PROT_WRITE,
+                MAP_SHARED, fd, 0);
+    if (base == MAP_FAILED)
+    {
+        NSFW_LOGERR("mmap failed, path:'%s' base:%p size:%lu, errno=%d",
+                    share->path, share->base, share->size, errno);
+        (void) nsfw_base_close(fd);
+        (void) free(real_path);
+        return -1;
+    }
+
+    if (NULL == share->base)
+    {
+        share->base = base;
+    }
+    else if (base != share->base)
+    {
+        NSFW_LOGERR("mmap address error, path:'%s' share->base:%p, base:%p",
+                    share->path, share->base, base);
+        (void) munmap(base, share->size);
+        (void) nsfw_base_close(fd);
+        (void) free(real_path);
+        return -1;
+    }
+
+    NSFW_LOGINF("Memory attached, base=%p", base);
+
+    (void) nsfw_base_close(fd);
+    (void) free(real_path);
+    return 0;
+}
+
+int dmm_huge_detach(struct dmm_share *share)
+{
+    (void) munmap(share->base, share->size);
+
+    return 0;
+}