include/linux/vserver/limit.h

   1 /* _VX_LIMIT_H defined below */
   2
   3 #if     defined(__KERNEL__) && defined(_VX_INFO_DEF_)
   4
   5 #include <asm/atomic.h>
   6 #include <asm/resource.h>
   7
   8 /* context sub struct */
   9
  10 #define NUM_LIMITS      20
  11
  12 #define VLIMIT_NSOCK    16
  13
  14
  15 extern const char *vlimit_name[NUM_LIMITS];
  16
  17 struct _vx_limit {
  18         atomic_t ticks;
  19
  20         unsigned long rlim[NUM_LIMITS];         /* Context limit */
  21         unsigned long rmax[NUM_LIMITS];         /* Context maximum */
  22         atomic_t rcur[NUM_LIMITS];              /* Current value */
  23         atomic_t lhit[NUM_LIMITS];              /* Limit hits */
  24 };
  25
  26 static inline void vx_info_init_limit(struct _vx_limit *limit)
  27 {
  28         int lim;
  29
  30         for (lim=0; lim<NUM_LIMITS; lim++) {
  31                 limit->rlim[lim] = RLIM_INFINITY;
  32                 limit->rmax[lim] = 0;
  33                 atomic_set(&limit->rcur[lim], 0);
  34                 atomic_set(&limit->lhit[lim], 0);
  35         }
  36 }
  37
  38 static inline void vx_info_exit_limit(struct _vx_limit *limit)
  39 {
  40 #ifdef  CONFIG_VSERVER_DEBUG
  41         unsigned long value;
  42         unsigned int lim;
  43
  44         for (lim=0; lim<NUM_LIMITS; lim++) {
  45                 value = atomic_read(&limit->rcur[lim]);
  46                 if (value)
  47                         printk("!!! limit: %p[%s,%d] = %ld on exit.\n",
  48                                 limit, vlimit_name[lim], lim, value);
  49         }
  50 #endif
  51 }
  52
  53 static inline void vx_limit_fixup(struct _vx_limit *limit)
  54 {
  55         unsigned long value;
  56         unsigned int lim;
  57
  58         for (lim=0; lim<NUM_LIMITS; lim++) {
  59                 value = atomic_read(&limit->rcur[lim]);
  60                 if (value > limit->rmax[lim])
  61                         limit->rmax[lim] = value;
  62                 if (limit->rmax[lim] > limit->rlim[lim])
  63                         limit->rmax[lim] = limit->rlim[lim];
  64         }
  65 }
  66
  67 #define VX_LIMIT_FMT    ":\t%10d\t%10ld\t%10ld\t%6d\n"
  68
  69 #define VX_LIMIT_ARG(r)                         \
  70                 ,atomic_read(&limit->rcur[r])   \
  71                 ,limit->rmax[r]                 \
  72                 ,limit->rlim[r]                 \
  73                 ,atomic_read(&limit->lhit[r])
  74
  75 static inline int vx_info_proc_limit(struct _vx_limit *limit, char *buffer)
  76 {
  77         vx_limit_fixup(limit);
  78         return sprintf(buffer,
  79                 "PROC"  VX_LIMIT_FMT
  80                 "VM"    VX_LIMIT_FMT
  81                 "VML"   VX_LIMIT_FMT
  82                 "RSS"   VX_LIMIT_FMT
  83                 "FILES" VX_LIMIT_FMT
  84                 "SOCK"  VX_LIMIT_FMT
  85                 VX_LIMIT_ARG(RLIMIT_NPROC)
  86                 VX_LIMIT_ARG(RLIMIT_AS)
  87                 VX_LIMIT_ARG(RLIMIT_MEMLOCK)
  88                 VX_LIMIT_ARG(RLIMIT_RSS)
  89                 VX_LIMIT_ARG(RLIMIT_NOFILE)
  90                 VX_LIMIT_ARG(VLIMIT_NSOCK)
  91                 );
  92 }
  93
  94 #else   /* _VX_INFO_DEF_ */
  95 #ifndef _VX_LIMIT_H
  96 #define _VX_LIMIT_H
  97
  98 #include "switch.h"
  99
 100 #define VXD_RLIMIT(r,l)         (VXD_CBIT(limit, (l)) && ((r) == (l)))
 101
 102 /*  rlimit vserver commands */
 103
 104 #define VCMD_get_rlimit         VC_CMD(RLIMIT, 1, 0)
 105 #define VCMD_set_rlimit         VC_CMD(RLIMIT, 2, 0)
 106 #define VCMD_get_rlimit_mask    VC_CMD(RLIMIT, 3, 0)
 107
 108 struct  vcmd_ctx_rlimit_v0 {
 109         uint32_t id;
 110         uint64_t minimum;
 111         uint64_t softlimit;
 112         uint64_t maximum;
 113 };
 114
 115 struct  vcmd_ctx_rlimit_mask_v0 {
 116         uint32_t minimum;
 117         uint32_t softlimit;
 118         uint32_t maximum;
 119 };
 120
 121 #define CRLIM_UNSET             (0ULL)
 122 #define CRLIM_INFINITY          (~0ULL)
 123 #define CRLIM_KEEP              (~1ULL)
 124
 125 #ifdef  __KERNEL__
 126
 127 #include <linux/compiler.h>
 128
 129 extern int vc_get_rlimit(uint32_t, void __user *);
 130 extern int vc_set_rlimit(uint32_t, void __user *);
 131 extern int vc_get_rlimit_mask(uint32_t, void __user *);
 132
 133 struct sysinfo;
 134
 135 void vx_vsi_meminfo(struct sysinfo *);
 136 void vx_vsi_swapinfo(struct sysinfo *);
 137
 138
 139 #endif  /* __KERNEL__ */
 140
 141 #endif  /* _VX_LIMIT_H */
 142 #endif
 143
 144