1 files changed, 136 insertions, 8 deletions

diff --git a/src/framework/mem/dmm_huge.c b/src/framework/mem/dmm_huge.c
index 735c507..6e3fc42 100644
--- a/src/framework/mem/dmm_huge.c
+++ b/src/framework/mem/dmm_huge.c
@@ -15,39 +15,167 @@
 */
 #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_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 pid index */
+#define DMM_HUGE_FMT "%s/dmm-%d-%d"     /* HUGE_DIR/dmm-pid-index */
 
 inline static void
-huge_set_path (struct dmm_share *share, int index)
+huge_set_path (char path[DMM_SHARE_PATH_MAX], pid_t pid, int index)
 {
-  (void) snprintf (share->path, sizeof (share->path), DMM_HUGE_FMT,
-                   DMM_HUGE_DIR, share->pid, index);
+  (void) snprintf (path, DMM_SHARE_PATH_MAX, DMM_HUGE_FMT,
+                   DMM_HUGE_DIR, pid, index);
+  path[DMM_SHARE_PATH_MAX - 1] = 0;
 }
 
 int
 dmm_huge_create (struct dmm_share *share)
 {
-  return -1;
+  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, 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)
 {
-  return -1;
+  (void) munmap (share->base, share->size);
+  (void) unlink (share->path);
+
+  return 0;
 }
 
 int
 dmm_huge_attach (struct dmm_share *share)
 {
-  return -1;
+  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;
+    }
+
+  fd = open (share->path, O_RDWR);
+  if (fd < 0)
+    {
+      NSFW_LOGERR ("Open file failed, path:'%s', errno=%d",
+                   share->path, errno);
+      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);
+          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);
+      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);
+      return -1;
+    }
+
+  NSFW_LOGINF ("Memory attached, base=%p", base);
+
+  (void) nsfw_base_close (fd);
+  return 0;
 }
 
 int
 dmm_huge_detach (struct dmm_share *share)
 {
-  return -1;
+  (void) munmap (share->base, share->size);
+
+  return 0;
 }