Merge pull request torvalds#258 from liuyuan10/memfix

Fix boot mem size and lkl_set_fd_limit
laijs · Nov 3, 2016 · 487de9a · 487de9a
2 parents 468cb6e + 20679a9
commit 487de9a
Show file tree

Hide file tree

Showing 5 changed files with 24 additions and 4 deletions.
diff --git a/arch/lkl/include/asm/page.h b/arch/lkl/include/asm/page.h
@@ -5,7 +5,7 @@
 
 #ifndef __ASSEMBLY__
 void free_mem(void);
-void bootmem_init(int mem_size);
+void bootmem_init(unsigned long mem_size);
 #endif
 
 #include <asm-generic/page.h>

diff --git a/arch/lkl/mm/bootmem.c b/arch/lkl/mm/bootmem.c
@@ -7,9 +7,9 @@ static unsigned long _memory_start, mem_size;
 
 void *empty_zero_page;
 
-void __init bootmem_init(int mem_size)
+void __init bootmem_init(unsigned long mem_size)
 {
- int bootmap_size;
+ unsigned long bootmap_size;
 
  _memory_start = (unsigned long)lkl_ops->mem_alloc(mem_size);
  memory_start = _memory_start;

diff --git a/tools/lkl/include/lkl.h b/tools/lkl/include/lkl.h
@@ -437,6 +437,12 @@ int lkl_if_del_ip(int ifindex, int af, void *addr, unsigned int netprefix_len);
  */
 int lkl_if_wait_ipv6_dad(int ifindex, void *addr);
 
+/**
+ * lkl_set_fd_limit - set the maximum number of file descriptors allowed
+ * @fd_limit - fd max limit
+ */
+int lkl_set_fd_limit(unsigned int fd_limit);
+
 #ifdef __cplusplus
 }
 #endif

diff --git a/tools/lkl/lib/fs.c b/tools/lkl/lib/fs.c
@@ -363,3 +363,12 @@ int lkl_dirfd(struct lkl_dir *dir)
 {
  return dir->fd;
 }
+
+int lkl_set_fd_limit(unsigned int fd_limit)
+{
+ struct lkl_rlimit rlim = {
+ .rlim_cur = fd_limit,
+ .rlim_max = fd_limit,
+ };
+ return lkl_sys_setrlimit(LKL_RLIMIT_NOFILE, &rlim);
+}
diff --git a/tools/lkl/lib/hijack/init.c b/tools/lkl/lib/hijack/init.c
@@ -323,7 +323,7 @@ hijack_init(void)
  if (single_cpu_mode == 1)
  PinToFirstCpu(&ori_cpu);
 
- ret = lkl_start_kernel(&lkl_host_ops, 64 * 1024 * 1024, "");
+ ret = lkl_start_kernel(&lkl_host_ops, 64 * 1024 * 1024UL, "");
  if (ret) {
  fprintf(stderr, "can't start kernel: %s\n", lkl_strerror(ret));
  return;
@@ -335,6 +335,11 @@ hijack_init(void)
  if (single_cpu_mode)
  PinToCpus(&ori_cpu);
 
+ ret = lkl_set_fd_limit(65535);
+ if (ret)
+ fprintf(stderr, "lkl_set_fd_limit failed: %s\n",
+ lkl_strerror(ret));
+
  /* fillup FDs up to LKL_FD_OFFSET */
  ret = lkl_sys_mknod("/dev_null", LKL_S_IFCHR | 0600, LKL_MKDEV(1, 3));
  dev_null = lkl_sys_open("/dev_null", LKL_O_RDONLY, 0);