include/linux/vs_dlimit.h

   1 #ifndef _VX_VS_DLIMIT_H
   2 #define _VX_VS_DLIMIT_H
   3
   4
   5 #include "vserver/dlimit.h"
   6 #include "vserver/debug.h"
   7
   8
   9 #define get_dl_info(i)  __get_dl_info(i,__FILE__,__LINE__)
  10
  11 static inline struct dl_info *__get_dl_info(struct dl_info *dli,
  12         const char *_file, int _line)
  13 {
  14         if (!dli)
  15                 return NULL;
  16         vxlprintk(VXD_CBIT(dlim, 4), "get_dl_info(%p[#%d.%d])",
  17                 dli, dli?dli->dl_xid:0, dli?atomic_read(&dli->dl_usecnt):0,
  18                 _file, _line);
  19         atomic_inc(&dli->dl_usecnt);
  20         return dli;
  21 }
  22
  23
  24 #define free_dl_info(i) \
  25         call_rcu(&i->dl_rcu, rcu_free_dl_info);
  26
  27 #define put_dl_info(i)  __put_dl_info(i,__FILE__,__LINE__)
  28
  29 static inline void __put_dl_info(struct dl_info *dli,
  30         const char *_file, int _line)
  31 {
  32         if (!dli)
  33                 return;
  34         vxlprintk(VXD_CBIT(dlim, 4), "put_dl_info(%p[#%d.%d])",
  35                 dli, dli?dli->dl_xid:0, dli?atomic_read(&dli->dl_usecnt):0,
  36                 _file, _line);
  37         if (atomic_dec_and_test(&dli->dl_usecnt))
  38                 free_dl_info(dli);
  39 }
  40
  41
  42 #define __dlimit_char(d)        ((d)?'*':' ')
  43
  44 static inline int __dl_alloc_space(struct super_block *sb,
  45         xid_t xid, dlsize_t nr, const char *file, int line)
  46 {
  47         struct dl_info *dli = NULL;
  48         int ret = 0;
  49
  50         if (nr == 0)
  51                 goto out;
  52         dli = locate_dl_info(sb, xid);
  53         if (!dli)
  54                 goto out;
  55
  56         spin_lock(&dli->dl_lock);
  57         ret = (dli->dl_space_used + nr > dli->dl_space_total);
  58         if (!ret)
  59                 dli->dl_space_used += nr;
  60         spin_unlock(&dli->dl_lock);
  61         put_dl_info(dli);
  62 out:
  63         vxlprintk(VXD_CBIT(dlim, 1),
  64                 "ALLOC (%p,#%d)%c %lld bytes (%d)",
  65                 sb, xid, __dlimit_char(dli), (long long)nr,
  66                 ret, file, line);
  67         return ret;
  68 }
  69
  70 static inline void __dl_free_space(struct super_block *sb,
  71         xid_t xid, dlsize_t nr, const char *_file, int _line)
  72 {
  73         struct dl_info *dli = NULL;
  74
  75         if (nr == 0)
  76                 goto out;
  77         dli = locate_dl_info(sb, xid);
  78         if (!dli)
  79                 goto out;
  80
  81         spin_lock(&dli->dl_lock);
  82         if (dli->dl_space_used > nr)
  83                 dli->dl_space_used -= nr;
  84         else
  85                 dli->dl_space_used = 0;
  86         spin_unlock(&dli->dl_lock);
  87         put_dl_info(dli);
  88 out:
  89         vxlprintk(VXD_CBIT(dlim, 1),
  90                 "FREE  (%p,#%d)%c %lld bytes",
  91                 sb, xid, __dlimit_char(dli), (long long)nr,
  92                 _file, _line);
  93 }
  94
  95 static inline int __dl_alloc_inode(struct super_block *sb,
  96         xid_t xid, const char *_file, int _line)
  97 {
  98         struct dl_info *dli;
  99         int ret = 0;
 100
 101         dli = locate_dl_info(sb, xid);
 102         if (!dli)
 103                 goto out;
 104
 105         spin_lock(&dli->dl_lock);
 106         ret = (dli->dl_inodes_used >= dli->dl_inodes_total);
 107         if (!ret)
 108                 dli->dl_inodes_used++;
 109 #if 0
 110         else
 111                 vxwprintk("DLIMIT hit (%p,#%d), inode %d>=%d @ %s:%d",
 112                         sb, xid,
 113                         dli->dl_inodes_used, dli->dl_inodes_total,
 114                         file, line);
 115 #endif
 116         spin_unlock(&dli->dl_lock);
 117         put_dl_info(dli);
 118 out:
 119         vxlprintk(VXD_CBIT(dlim, 0),
 120                 "ALLOC (%p,#%d)%c inode (%d)",
 121                 sb, xid, __dlimit_char(dli), ret, _file, _line);
 122         return ret;
 123 }
 124
 125 static inline void __dl_free_inode(struct super_block *sb,
 126         xid_t xid, const char *_file, int _line)
 127 {
 128         struct dl_info *dli;
 129
 130         dli = locate_dl_info(sb, xid);
 131         if (!dli)
 132                 goto out;
 133
 134         spin_lock(&dli->dl_lock);
 135         if (dli->dl_inodes_used > 1)
 136                 dli->dl_inodes_used--;
 137         else
 138                 dli->dl_inodes_used = 0;
 139         spin_unlock(&dli->dl_lock);
 140         put_dl_info(dli);
 141 out:
 142         vxlprintk(VXD_CBIT(dlim, 0),
 143                 "FREE  (%p,#%d)%c inode",
 144                 sb, xid, __dlimit_char(dli), _file, _line);
 145 }
 146
 147 static inline void __dl_adjust_block(struct super_block *sb, xid_t xid,
 148         unsigned int *free_blocks, unsigned int *root_blocks,
 149         const char *_file, int _line)
 150 {
 151         struct dl_info *dli;
 152         uint64_t broot, bfree;
 153
 154         dli = locate_dl_info(sb, xid);
 155         if (!dli)
 156                 return;
 157
 158         spin_lock(&dli->dl_lock);
 159         broot = (dli->dl_space_total -
 160                 (dli->dl_space_total >> 10) * dli->dl_nrlmult)
 161                 >> sb->s_blocksize_bits;
 162         bfree = (dli->dl_space_total - dli->dl_space_used)
 163                         >> sb->s_blocksize_bits;
 164         spin_unlock(&dli->dl_lock);
 165
 166         vxlprintk(VXD_CBIT(dlim, 2),
 167                 "ADJUST: %lld,%lld on %d,%d [mult=%d]",
 168                 (long long)bfree, (long long)broot,
 169                 *free_blocks, *root_blocks, dli->dl_nrlmult,
 170                 _file, _line);
 171         if (free_blocks) {
 172                 if (*free_blocks > bfree)
 173                         *free_blocks = bfree;
 174         }
 175         if (root_blocks) {
 176                 if (*root_blocks > broot)
 177                         *root_blocks = broot;
 178         }
 179         put_dl_info(dli);
 180 }
 181
 182
 183 #define DLIMIT_ALLOC_BLOCK(sb, xid, nr) \
 184         __dl_alloc_space(sb, xid, \
 185                 ((dlsize_t)(nr)) << (sb)->s_blocksize_bits, \
 186                 __FILE__, __LINE__ )
 187
 188 #define DLIMIT_FREE_BLOCK(sb, xid, nr) \
 189         __dl_free_space(sb, xid, \
 190                 ((dlsize_t)(nr)) << (sb)->s_blocksize_bits, \
 191                 __FILE__, __LINE__ )
 192
 193 #define DLIMIT_ALLOC_INODE(sb, xid) \
 194         __dl_alloc_inode(sb, xid, __FILE__, __LINE__ )
 195
 196 #define DLIMIT_FREE_INODE(sb, xid) \
 197         __dl_free_inode(sb, xid, __FILE__, __LINE__ )
 198
 199
 200 #define DLIMIT_ADJUST_BLOCK(sb, xid, fb, rb) \
 201         __dl_adjust_block(sb, xid, fb, rb, __FILE__, __LINE__ )
 202
 203
 204 #else
 205 #warning duplicate inclusion
 206 #endif