Whamcloud - gitweb
d029d99252a5e53d01576f9f6822806dee11b4a3
[fs/lustre-release.git] / lustre / include / obd_support.h
1 /* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*-
2  * vim:expandtab:shiftwidth=8:tabstop=8:
3  *
4  *  Copyright (C) 2001, 2002 Cluster File Systems, Inc.
5  *
6  *   This file is part of Lustre, http://www.lustre.org.
7  *
8  *   Lustre is free software; you can redistribute it and/or
9  *   modify it under the terms of version 2 of the GNU General Public
10  *   License as published by the Free Software Foundation.
11  *
12  *   Lustre is distributed in the hope that it will be useful,
13  *   but WITHOUT ANY WARRANTY; without even the implied warranty of
14  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  *   GNU General Public License for more details.
16  *
17  *   You should have received a copy of the GNU General Public License
18  *   along with Lustre; if not, write to the Free Software
19  *   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
20  *
21  */
22
23 #ifndef _OBD_SUPPORT
24 #define _OBD_SUPPORT
25
26 #include <libcfs/kp30.h>
27 #include <lvfs.h>
28 #include <lprocfs_status.h>
29
30 #if defined(__linux__)
31 #include <linux/obd_support.h>
32 #elif defined(__APPLE__)
33 #include <darwin/obd_support.h>
34 #elif defined(__WINNT__)
35 #include <winnt/obd_support.h>
36 #else
37 #error Unsupported operating system.
38 #endif
39
40 /* global variables */
41 extern struct lprocfs_stats *obd_memory;
42 enum {
43         OBD_MEMORY_STAT = 0,
44         OBD_MEMORY_PAGES_STAT = 1,
45         OBD_STATS_NUM,
46 };
47
48 enum {
49         OBD_FAIL_LOC_NOSET      = 0,
50         OBD_FAIL_LOC_ORSET      = 1,
51         OBD_FAIL_LOC_RESET      = 2
52 };
53
54 extern unsigned long obd_fail_loc;
55 extern unsigned int obd_fail_val;
56 extern unsigned int obd_debug_peer_on_timeout;
57 extern unsigned int obd_dump_on_timeout;
58 extern unsigned int obd_dump_on_eviction;
59 extern unsigned int obd_timeout;          /* seconds */
60 #define PING_INTERVAL max(obd_timeout / 4, 1U)
61 #define RECONNECT_INTERVAL max(obd_timeout / 10, 10U)
62 extern unsigned int ldlm_timeout;
63 extern unsigned int obd_health_check_timeout;
64 extern unsigned int obd_sync_filter;
65 extern unsigned int obd_max_dirty_pages;
66 extern atomic_t obd_dirty_pages;
67 extern cfs_waitq_t obd_race_waitq;
68 extern int obd_race_state;
69 extern unsigned int obd_alloc_fail_rate;
70
71 int __obd_fail_check_set(__u32 id, __u32 value, int set);
72
73 /* Timeout definitions */
74 #define LDLM_TIMEOUT_DEFAULT 20
75 #define OBD_TIMEOUT_DEFAULT 100
76 #define HEALTH_CHECK_COEF 3 / 2
77 #define HEALTH_CHECK_TIMEOUT_DEFAULT (OBD_TIMEOUT_DEFAULT * HEALTH_CHECK_COEF)
78 #define HEALTH_CHECK_TIMEOUT (obd_timeout * HEALTH_CHECK_COEF)
79
80 #define OBD_FAIL_MDS                     0x100
81 #define OBD_FAIL_MDS_HANDLE_UNPACK       0x101
82 #define OBD_FAIL_MDS_GETATTR_NET         0x102
83 #define OBD_FAIL_MDS_GETATTR_PACK        0x103
84 #define OBD_FAIL_MDS_READPAGE_NET        0x104
85 #define OBD_FAIL_MDS_READPAGE_PACK       0x105
86 #define OBD_FAIL_MDS_SENDPAGE            0x106
87 #define OBD_FAIL_MDS_REINT_NET           0x107
88 #define OBD_FAIL_MDS_REINT_UNPACK        0x108
89 #define OBD_FAIL_MDS_REINT_SETATTR       0x109
90 #define OBD_FAIL_MDS_REINT_SETATTR_WRITE 0x10a
91 #define OBD_FAIL_MDS_REINT_CREATE        0x10b
92 #define OBD_FAIL_MDS_REINT_CREATE_WRITE  0x10c
93 #define OBD_FAIL_MDS_REINT_UNLINK        0x10d
94 #define OBD_FAIL_MDS_REINT_UNLINK_WRITE  0x10e
95 #define OBD_FAIL_MDS_REINT_LINK          0x10f
96 #define OBD_FAIL_MDS_REINT_LINK_WRITE    0x110
97 #define OBD_FAIL_MDS_REINT_RENAME        0x111
98 #define OBD_FAIL_MDS_REINT_RENAME_WRITE  0x112
99 #define OBD_FAIL_MDS_OPEN_NET            0x113
100 #define OBD_FAIL_MDS_OPEN_PACK           0x114
101 #define OBD_FAIL_MDS_CLOSE_NET           0x115
102 #define OBD_FAIL_MDS_CLOSE_PACK          0x116
103 #define OBD_FAIL_MDS_CONNECT_NET         0x117
104 #define OBD_FAIL_MDS_CONNECT_PACK        0x118
105 #define OBD_FAIL_MDS_REINT_NET_REP       0x119
106 #define OBD_FAIL_MDS_DISCONNECT_NET      0x11a
107 #define OBD_FAIL_MDS_GETSTATUS_NET       0x11b
108 #define OBD_FAIL_MDS_GETSTATUS_PACK      0x11c
109 #define OBD_FAIL_MDS_STATFS_PACK         0x11d
110 #define OBD_FAIL_MDS_STATFS_NET          0x11e
111 #define OBD_FAIL_MDS_GETATTR_NAME_NET    0x11f
112 #define OBD_FAIL_MDS_PIN_NET             0x120
113 #define OBD_FAIL_MDS_UNPIN_NET           0x121
114 #define OBD_FAIL_MDS_ALL_REPLY_NET       0x122
115 #define OBD_FAIL_MDS_ALL_REQUEST_NET     0x123
116 #define OBD_FAIL_MDS_SYNC_NET            0x124
117 #define OBD_FAIL_MDS_SYNC_PACK           0x125
118 #define OBD_FAIL_MDS_DONE_WRITING_NET    0x126
119 #define OBD_FAIL_MDS_DONE_WRITING_PACK   0x127
120 #define OBD_FAIL_MDS_ALLOC_OBDO          0x128
121 #define OBD_FAIL_MDS_PAUSE_OPEN          0x129
122 #define OBD_FAIL_MDS_STATFS_LCW_SLEEP    0x12a
123 #define OBD_FAIL_MDS_OPEN_CREATE         0x12b
124 #define OBD_FAIL_MDS_OST_SETATTR         0x12c
125 #define OBD_FAIL_MDS_QUOTACHECK_NET      0x12d
126 #define OBD_FAIL_MDS_QUOTACTL_NET        0x12e
127 #define OBD_FAIL_MDS_CLIENT_ADD          0x12f
128 #define OBD_FAIL_MDS_GETXATTR_NET        0x130
129 #define OBD_FAIL_MDS_GETXATTR_PACK       0x131
130 #define OBD_FAIL_MDS_SETXATTR_NET        0x132
131 #define OBD_FAIL_MDS_SETXATTR            0x133
132 #define OBD_FAIL_MDS_SETXATTR_WRITE      0x134
133 #define OBD_FAIL_MDS_FS_SETUP            0x135
134 #define OBD_FAIL_MDS_RESEND              0x136
135 #define OBD_FAIL_MDS_IS_SUBDIR_NET       0x137
136 #define OBD_FAIL_MDS_IS_SUBDIR_PACK      0x138
137 #define OBD_FAIL_MDS_SET_INFO_NET        0x139
138 #define OBD_FAIL_MDS_WRITEPAGE_NET       0x13a
139 #define OBD_FAIL_MDS_WRITEPAGE_PACK      0x13b
140 #define OBD_FAIL_MDS_LLOG_CREATE_FAILED  0x13c
141 #define OBD_FAIL_MDS_OSC_PRECREATE       0x13d
142 #define OBD_FAIL_MDS_LOV_SYNC_RACE       0x13e
143 #define OBD_FAIL_MDS_CLOSE_NET_REP       0x13f
144 #define OBD_FAIL_MDS_LLOG_SYNC_TIMEOUT   0x140
145
146 #define OBD_FAIL_OST                     0x200
147 #define OBD_FAIL_OST_CONNECT_NET         0x201
148 #define OBD_FAIL_OST_DISCONNECT_NET      0x202
149 #define OBD_FAIL_OST_GET_INFO_NET        0x203
150 #define OBD_FAIL_OST_CREATE_NET          0x204
151 #define OBD_FAIL_OST_DESTROY_NET         0x205
152 #define OBD_FAIL_OST_GETATTR_NET         0x206
153 #define OBD_FAIL_OST_SETATTR_NET         0x207
154 #define OBD_FAIL_OST_OPEN_NET            0x208
155 #define OBD_FAIL_OST_CLOSE_NET           0x209
156 #define OBD_FAIL_OST_BRW_NET             0x20a
157 #define OBD_FAIL_OST_PUNCH_NET           0x20b
158 #define OBD_FAIL_OST_STATFS_NET          0x20c
159 #define OBD_FAIL_OST_HANDLE_UNPACK       0x20d
160 #define OBD_FAIL_OST_BRW_WRITE_BULK      0x20e
161 #define OBD_FAIL_OST_BRW_READ_BULK       0x20f
162 #define OBD_FAIL_OST_SYNC_NET            0x210
163 #define OBD_FAIL_OST_ALL_REPLY_NET       0x211
164 #define OBD_FAIL_OST_ALL_REQUESTS_NET    0x212
165 #define OBD_FAIL_OST_LDLM_REPLY_NET      0x213
166 #define OBD_FAIL_OST_BRW_PAUSE_BULK      0x214
167 #define OBD_FAIL_OST_ENOSPC              0x215
168 #define OBD_FAIL_OST_EROFS               0x216
169 #define OBD_FAIL_OST_ENOENT              0x217
170 #define OBD_FAIL_OST_QUOTACHECK_NET      0x218
171 #define OBD_FAIL_OST_QUOTACTL_NET        0x219
172 #define OBD_FAIL_OST_CHECKSUM_RECEIVE    0x21a
173 #define OBD_FAIL_OST_CHECKSUM_SEND       0x21b
174 #define OBD_FAIL_OST_BRW_SIZE            0x21c
175 #define OBD_FAIL_OST_DROP_REQ            0x21d
176 #define OBD_FAIL_OST_SETATTR_CREDITS     0x21e
177 #define OBD_FAIL_OST_HOLD_WRITE_RPC      0x21f
178 #define OBD_FAIL_OST_BRW_WRITE_BULK2     0x220
179 #define OBD_FAIL_OST_LLOG_RECOVERY_TIMEOUT 0x221
180 #define OBD_FAIL_OST_CANCEL_COOKIE_TIMEOUT 0x222
181
182 #define OBD_FAIL_LDLM                    0x300
183 #define OBD_FAIL_LDLM_NAMESPACE_NEW      0x301
184 #define OBD_FAIL_LDLM_ENQUEUE            0x302
185 #define OBD_FAIL_LDLM_CONVERT            0x303
186 #define OBD_FAIL_LDLM_CANCEL             0x304
187 #define OBD_FAIL_LDLM_BL_CALLBACK        0x305
188 #define OBD_FAIL_LDLM_CP_CALLBACK        0x306
189 #define OBD_FAIL_LDLM_GL_CALLBACK        0x307
190 #define OBD_FAIL_LDLM_ENQUEUE_EXTENT_ERR 0x308
191 #define OBD_FAIL_LDLM_ENQUEUE_INTENT_ERR 0x309
192 #define OBD_FAIL_LDLM_CREATE_RESOURCE    0x30a
193 #define OBD_FAIL_LDLM_ENQUEUE_BLOCKED    0x30b
194 #define OBD_FAIL_LDLM_REPLY              0x30c
195 #define OBD_FAIL_LDLM_RECOV_CLIENTS      0x30d
196 #define OBD_FAIL_LDLM_ENQUEUE_OLD_EXPORT 0x30e
197 #define OBD_FAIL_LDLM_GLIMPSE            0x30f
198 #define OBD_FAIL_LDLM_CANCEL_RACE        0x310
199 #define OBD_FAIL_LDLM_CANCEL_EVICT_RACE  0x311
200
201 #define OBD_FAIL_OSC                     0x400
202 #define OBD_FAIL_OSC_BRW_READ_BULK       0x401
203 #define OBD_FAIL_OSC_BRW_WRITE_BULK      0x402
204 #define OBD_FAIL_OSC_LOCK_BL_AST         0x403
205 #define OBD_FAIL_OSC_LOCK_CP_AST         0x404
206 #define OBD_FAIL_OSC_MATCH               0x405
207 #define OBD_FAIL_OSC_BRW_PREP_REQ        0x406
208 #define OBD_FAIL_OSC_SHUTDOWN            0x407
209 #define OBD_FAIL_OSC_CHECKSUM_RECEIVE    0x408
210 #define OBD_FAIL_OSC_CHECKSUM_SEND       0x409
211 #define OBD_FAIL_OSC_BRW_PREP_REQ2       0x40a
212
213 #define OBD_FAIL_PTLRPC                  0x500
214 #define OBD_FAIL_PTLRPC_ACK              0x501
215 #define OBD_FAIL_PTLRPC_RQBD             0x502
216 #define OBD_FAIL_PTLRPC_BULK_GET_NET     0x503
217 #define OBD_FAIL_PTLRPC_BULK_PUT_NET     0x504
218 #define OBD_FAIL_PTLRPC_DROP_RPC         0x505
219 #define OBD_FAIL_PTLRPC_DELAY_SEND       0x506
220 #define OBD_FAIL_PTLRPC_DELAY_RECOV      0x507
221
222 #define OBD_FAIL_OBD_PING_NET            0x600
223 #define OBD_FAIL_OBD_LOG_CANCEL_NET      0x601
224 #define OBD_FAIL_OBD_LOGD_NET            0x602
225 #define OBD_FAIL_OBD_QC_CALLBACK_NET     0x603
226 #define OBD_FAIL_OBD_DQACQ               0x604
227
228 #define OBD_FAIL_TGT_REPLY_NET           0x700
229 #define OBD_FAIL_TGT_CONN_RACE           0x701
230 #define OBD_FAIL_TGT_FORCE_RECONNECT     0x702
231 #define OBD_FAIL_TGT_DELAY_CONNECT       0x703
232 #define OBD_FAIL_TGT_DELAY_RECONNECT     0x704
233 #define OBD_FAIL_TGT_DELAY_PRECREATE     0x705
234
235 #define OBD_FAIL_MDC_REVALIDATE_PAUSE    0x800
236 #define OBD_FAIL_MDC_ENQUEUE_PAUSE       0x801
237
238 #define OBD_FAIL_MGS                     0x900
239 #define OBD_FAIL_MGS_ALL_REQUEST_NET     0x901
240 #define OBD_FAIL_MGS_ALL_REPLY_NET       0x902
241 #define OBD_FAIL_MGC_PROCESS_LOG         0x903
242 #define OBD_FAIL_MGS_SLOW_REQUEST_NET    0x904
243 #define OBD_FAIL_MGS_SLOW_TARGET_REG     0x905
244
245 #define OBD_FAIL_QUOTA_QD_COUNT_32BIT    0xa00
246
247 #define OBD_FAIL_LPROC_REMOVE            0xb00
248
249 #define OBD_FAIL_GENERAL_ALLOC           0xc00
250
251 #define OBD_FAIL_SEQ                     0x1000
252 #define OBD_FAIL_SEQ_QUERY_NET           0x1001
253
254 #define OBD_FAIL_FLD                     0x1100
255 #define OBD_FAIL_FLD_QUERY_NET           0x1101
256
257 #define OBD_FAIL_SEC_CTX                 0x1200
258 #define OBD_FAIL_SEC_CTX_INIT_NET        0x1201
259 #define OBD_FAIL_SEC_CTX_INIT_CONT_NET   0x1202
260 #define OBD_FAIL_SEC_CTX_FINI_NET        0x1203
261
262 /* Failure injection control */
263 #define OBD_FAIL_MASK_SYS    0x0000FF00
264 #define OBD_FAIL_MASK_LOC   (0x000000FF | OBD_FAIL_MASK_SYS)
265
266 #define OBD_FAILED_BIT       30
267 /* OBD_FAILED is 0x40000000 */
268 #define OBD_FAILED          (1 << OBD_FAILED_BIT)
269
270 #define OBD_FAIL_ONCE_BIT    31
271 /* OBD_FAIL_ONCE is 0x80000000 */
272 #define OBD_FAIL_ONCE       (1 << OBD_FAIL_ONCE_BIT)
273
274 /* The following flags aren't made to be combined */
275 #define OBD_FAIL_SKIP        0x20000000 /* skip N times then fail */
276 #define OBD_FAIL_SOME        0x10000000 /* only fail N times */
277 #define OBD_FAIL_RAND        0x08000000 /* fail 1/N of the times */
278 #define OBD_FAIL_USR1        0x04000000 /* user flag */
279
280 #define OBD_FAIL_PRECHECK(id) (obd_fail_loc &&                                \
281                               (obd_fail_loc & OBD_FAIL_MASK_LOC) ==           \
282                               ((id) & OBD_FAIL_MASK_LOC))
283
284 static inline int obd_fail_check_set(__u32 id, __u32 value, int set)
285 {
286         int ret = 0;
287         if (unlikely(OBD_FAIL_PRECHECK(id) &&
288             (ret = __obd_fail_check_set(id, value, set)))) {
289                 CERROR("*** obd_fail_loc=%x ***\n", id);
290         }
291         return ret;
292 }
293
294 /* If id hit obd_fail_loc, return 1, otherwise return 0 */
295 #define OBD_FAIL_CHECK(id) \
296         obd_fail_check_set(id, 0, OBD_FAIL_LOC_NOSET)
297
298 /* If id hit obd_fail_loc, obd_fail_loc |= value and return 1,
299  * otherwise return 0 */
300 #define OBD_FAIL_CHECK_ORSET(id, value) \
301         obd_fail_check_set(id, value, OBD_FAIL_LOC_ORSET)
302
303 /* If id hit obd_fail_loc, obd_fail_loc = value and return 1,
304  * otherwise return 0 */
305 #define OBD_FAIL_CHECK_RESET(id, value) \
306         obd_fail_check_set(id, value, OBD_FAIL_LOC_RESET)
307
308
309 static inline int obd_fail_timeout_set(__u32 id, __u32 value, int secs, int set)
310 {
311         int ret = 0;
312         if (unlikely(OBD_FAIL_PRECHECK(id) &&
313             (ret = __obd_fail_check_set(id, value, set)))) {
314                 CERROR("obd_fail_timeout id %x sleeping for %d secs\n",
315                        id, secs);
316                 set_current_state(TASK_UNINTERRUPTIBLE);
317                 cfs_schedule_timeout(CFS_TASK_UNINT,  cfs_time_seconds(secs));
318                 set_current_state(TASK_RUNNING);
319                 CERROR("obd_fail_timeout id %x awake\n", id);
320         }
321         return ret;
322 }
323
324 /* If id hit obd_fail_loc, sleep secs */
325 #define OBD_FAIL_TIMEOUT(id, secs) \
326         obd_fail_timeout_set(id, 0, secs, OBD_FAIL_LOC_NOSET)
327
328 /* If id hit obd_fail_loc, obd_fail_loc |= value and sleep secs */
329 #define OBD_FAIL_TIMEOUT_ORSET(id, value, secs) \
330         obd_fail_timeout_set(id, value, secs, OBD_FAIL_LOC_ORSET)
331
332 #ifdef __KERNEL__
333 static inline void obd_fail_write(int id, struct super_block *sb)
334 {
335         /* We set FAIL_ONCE because we never "un-fail" a device */
336         if (OBD_FAIL_CHECK_ORSET(id & ~OBD_FAIL_ONCE, OBD_FAIL_ONCE)) {
337 #ifdef LIBCFS_DEBUG
338                 BDEVNAME_DECLARE_STORAGE(tmp);
339                 CERROR("obd_fail_loc=%x, fail write operation on %s\n",
340                        id, ll_bdevname(sb, tmp));
341 #endif
342                 /* TODO-CMD: fix getting jdev */
343                 __lvfs_set_rdonly(lvfs_sbdev(sb), (lvfs_sbdev_type)0);
344         }
345 }
346 #define OBD_FAIL_WRITE(id, sb) obd_fail_write(id, sb)
347
348 /* The idea here is to synchronise two threads to force a race. The
349  * first thread that calls this with a matching fail_loc is put to
350  * sleep. The next thread that calls with the same fail_loc wakes up
351  * the first and continues. */
352 static inline void obd_race(__u32 id)
353 {
354         if (OBD_FAIL_PRECHECK(id)) {
355                 if (unlikely(__obd_fail_check_set(id, 0, OBD_FAIL_LOC_NOSET))) {
356                         obd_race_state = 0;
357                         CERROR("obd_race id %x sleeping\n", id);
358                         OBD_SLEEP_ON(obd_race_waitq, obd_race_state != 0);
359                         CERROR("obd_fail_race id %x awake\n", id);
360                 } else {
361                         CERROR("obd_fail_race id %x waking\n", id);
362                         obd_race_state = 1;
363                         wake_up(&obd_race_waitq);
364                 }
365         }
366 }
367 #define OBD_RACE(id) obd_race(id)
368 #else
369 /* sigh.  an expedient fix until OBD_RACE is fixed up */
370 #define OBD_RACE(foo) do {} while(0)
371 #endif
372
373 #define fixme() CDEBUG(D_OTHER, "FIXME\n");
374
375 extern atomic_t libcfs_kmemory;
376
377 #ifdef LPROCFS
378 #define obd_memory_add(size)                                                  \
379         lprocfs_counter_add(obd_memory, OBD_MEMORY_STAT, (long)(size))
380 #define obd_memory_sub(size)                                                  \
381         lprocfs_counter_sub(obd_memory, OBD_MEMORY_STAT, (long)(size))
382 #define obd_memory_sum()                                                      \
383         lprocfs_stats_collector(obd_memory, OBD_MEMORY_STAT,                  \
384                                 LPROCFS_FIELDS_FLAGS_SUM)
385 #define obd_pages_add(order)                                                  \
386         lprocfs_counter_add(obd_memory, OBD_MEMORY_PAGES_STAT,                \
387                             (long)(1 << (order)))
388 #define obd_pages_sub(order)                                                  \
389         lprocfs_counter_sub(obd_memory, OBD_MEMORY_PAGES_STAT,                \
390                             (long)(1 << (order)))
391 #define obd_pages_sum()                                                       \
392         lprocfs_stats_collector(obd_memory, OBD_MEMORY_PAGES_STAT,            \
393                                 LPROCFS_FIELDS_FLAGS_SUM)
394
395 extern void obd_update_maxusage(void);
396 extern __u64 obd_memory_max(void);
397 extern __u64 obd_pages_max(void);
398
399 #else
400
401 extern __u64 obd_alloc;
402 extern __u64 obd_pages;
403
404 extern __u64 obd_max_alloc;
405 extern __u64 obd_max_pages;
406
407 static inline void obd_memory_add(long size)
408 {
409         obd_alloc += size;
410         if (obd_alloc > obd_max_alloc)
411                 obd_max_alloc = obd_alloc;
412 }
413
414 static inline void obd_memory_sub(long size)
415 {
416         obd_alloc -= size;
417 }
418
419 static inline void obd_pages_add(int order)
420 {
421         obd_pages += 1<< order;
422         if (obd_pages > obd_max_pages)
423                 obd_max_pages = obd_pages;
424 }
425
426 static inline void obd_pages_sub(int order)
427 {
428         obd_pages -= 1<< order;
429 }
430
431 #define obd_memory_sum() (obd_alloc)
432 #define obd_pages_sum()  (obd_pages)
433
434 #define obd_memory_max() (obd_max_alloc)
435 #define obd_pages_max() (obd_max_pages)
436
437 #endif
438
439 #if defined (CONFIG_DEBUG_MEMORY) && defined(__KERNEL__)
440
441 #define OBD_MT_WRONG_SIZE    (1 << 0)
442 #define OBD_MT_ALREADY_FREED (1 << 1)
443 #define OBD_MT_LOC_LEN       128
444
445 struct obd_mem_track {
446         struct hlist_node mt_hash;
447         char              mt_loc[OBD_MT_LOC_LEN];
448         int               mt_flags;
449         void             *mt_ptr;
450         int               mt_size;
451 };
452
453 void lvfs_memdbg_show(void);
454 void lvfs_memdbg_insert(struct obd_mem_track *mt);
455 void lvfs_memdbg_remove(struct obd_mem_track *mt);
456 struct obd_mem_track *lvfs_memdbg_find(void *ptr);
457
458 int lvfs_memdbg_check_insert(struct obd_mem_track *mt);
459 struct obd_mem_track *lvfs_memdbg_check_remove(void *ptr);
460
461 static inline struct obd_mem_track *
462 __new_mem_track(void *ptr, int size,
463                 char *file, int line)
464 {
465         struct obd_mem_track *mt;
466
467         mt = kmalloc(sizeof(*mt), GFP_KERNEL);
468         if (unlikely(!mt))
469                 return NULL;
470
471         snprintf(mt->mt_loc, sizeof(mt->mt_loc) - 1,
472                  "%s:%d", file, line);
473
474         mt->mt_size = size;
475         mt->mt_ptr = ptr;
476         mt->mt_flags = 0;
477         return mt;
478 }
479
480 static inline void
481 __free_mem_track(struct obd_mem_track *mt)
482 {
483         kfree(mt);
484 }
485
486 static inline int
487 __get_mem_track(void *ptr, int size,
488                 char *file, int line)
489 {
490         struct obd_mem_track *mt;
491
492         mt = __new_mem_track(ptr, size, file, line);
493         if (unlikely(!mt)) {
494                 CWARN("Can't allocate new memory track\n");
495                 return 0;
496         }
497
498         if (!lvfs_memdbg_check_insert(mt))
499                 __free_mem_track(mt);
500
501         return 1;
502 }
503
504 static inline int
505 __put_mem_track(void *ptr, int size,
506                 char *file, int line)
507 {
508         struct obd_mem_track *mt;
509
510         if (unlikely(!(mt = lvfs_memdbg_check_remove(ptr)))) {
511                 CWARN("Ptr 0x%p is not allocated. Attempt to free "
512                       "not allocated memory at %s:%d\n", ptr,
513                       file, line);
514                 LBUG();
515                 return 0;
516         } else {
517                 if (unlikely(mt->mt_size != size)) {
518                         if (!(mt->mt_flags & OBD_MT_ALREADY_FREED)) {
519                                 mt->mt_flags |= (OBD_MT_WRONG_SIZE |
520                                                  OBD_MT_ALREADY_FREED);
521
522                                 CWARN("Freeing memory chunk (at 0x%p) of "
523                                       "different size than allocated "
524                                       "(%d != %d) at %s:%d, allocated at %s\n",
525                                       ptr, mt->mt_size, size, file, line,
526                                       mt->mt_loc);
527                         }
528                 } else {
529                         __free_mem_track(mt);
530                 }
531                 return 1;
532         }
533 }
534
535 #define get_mem_track(ptr, size, file, line)                                         \
536         __get_mem_track((ptr), (size), (file), (line))
537
538 #define put_mem_track(ptr, size, file, line)                                         \
539         __put_mem_track((ptr), (size), (file), (line))
540
541 #else /* !CONFIG_DEBUG_MEMORY */
542
543 #define get_mem_track(ptr, size, file, line)                                         \
544         do {} while (0)
545
546 #define put_mem_track(ptr, size, file, line)                                         \
547         do {} while (0)
548 #endif /* !CONFIG_DEBUG_MEMORY */
549
550 #define OBD_DEBUG_MEMUSAGE (1)
551
552 #if OBD_DEBUG_MEMUSAGE
553 #define OBD_ALLOC_POST(ptr, size, name)                                 \
554                 obd_memory_add(size);                                   \
555                 get_mem_track((ptr), (size), __FILE__, __LINE__);       \
556                 CDEBUG(D_MALLOC, name " '" #ptr "': %d at %p.\n",       \
557                        (int)(size), ptr)
558
559 #define OBD_FREE_PRE(ptr, size, name)                                   \
560         LASSERT(ptr);                                                   \
561         put_mem_track((ptr), (size), __FILE__, __LINE__);               \
562         obd_memory_sub(size);                                           \
563         CDEBUG(D_MALLOC, name " '" #ptr "': %d at %p.\n",               \
564                (int)(size), ptr);                                       \
565         POISON(ptr, 0x5a, size)
566
567 #else /* !OBD_DEBUG_MEMUSAGE */
568
569 #define OBD_ALLOC_POST(ptr, size, name) ((void)0)
570 #define OBD_FREE_PRE(ptr, size, name)   ((void)0)
571
572 #endif /* !OBD_DEBUG_MEMUSAGE */
573
574 #ifdef RANDOM_FAIL_ALLOC
575 #define HAS_FAIL_ALLOC_FLAG OBD_FAIL_CHECK(OBD_FAIL_GENERAL_ALLOC)
576 #else
577 #define HAS_FAIL_ALLOC_FLAG 0
578 #endif
579
580 #define OBD_ALLOC_FAIL_BITS 24
581 #define OBD_ALLOC_FAIL_MASK ((1 << OBD_ALLOC_FAIL_BITS) - 1)
582 #define OBD_ALLOC_FAIL_MULT (OBD_ALLOC_FAIL_MASK / 100)
583
584 #if defined(LUSTRE_UTILS) /* this version is for utils only */
585 #define OBD_ALLOC_GFP(ptr, size, gfp_mask)                                    \
586 do {                                                                          \
587         (ptr) = cfs_alloc(size, (gfp_mask));                                  \
588         if (unlikely((ptr) == NULL)) {                                        \
589                 CERROR("kmalloc of '" #ptr "' (%d bytes) failed at %s:%d\n",  \
590                        (int)(size), __FILE__, __LINE__);                      \
591         } else {                                                              \
592                 memset(ptr, 0, size);                                         \
593                 CDEBUG(D_MALLOC, "kmalloced '" #ptr "': %d at %p\n",          \
594                        (int)(size), ptr);                                     \
595         }                                                                     \
596 } while (0)
597 #else /* this version is for the kernel and liblustre */
598 #define OBD_FREE_RTN0(ptr)                                                    \
599 ({                                                                            \
600         cfs_free(ptr);                                                        \
601         (ptr) = NULL;                                                         \
602         0;                                                                    \
603 })
604 #define OBD_ALLOC_GFP(ptr, size, gfp_mask)                                    \
605 do {                                                                          \
606         (ptr) = cfs_alloc(size, (gfp_mask));                                  \
607         if (likely((ptr) != NULL &&                                           \
608                    (!HAS_FAIL_ALLOC_FLAG || obd_alloc_fail_rate == 0 ||       \
609                     !obd_alloc_fail(ptr, #ptr, "km", size,                    \
610                                     __FILE__, __LINE__) ||                    \
611                     OBD_FREE_RTN0(ptr)))){                                    \
612                 memset(ptr, 0, size);                                         \
613                 OBD_ALLOC_POST(ptr, size, "kmalloced");                       \
614         }                                                                     \
615 } while (0)
616 #endif
617
618 #ifndef OBD_ALLOC_MASK
619 # define OBD_ALLOC_MASK CFS_ALLOC_IO
620 #endif
621
622 #define OBD_ALLOC(ptr, size) OBD_ALLOC_GFP(ptr, size, OBD_ALLOC_MASK)
623 #define OBD_ALLOC_WAIT(ptr, size) OBD_ALLOC_GFP(ptr, size, CFS_ALLOC_STD)
624 #define OBD_ALLOC_PTR(ptr) OBD_ALLOC(ptr, sizeof *(ptr))
625 #define OBD_ALLOC_PTR_WAIT(ptr) OBD_ALLOC_WAIT(ptr, sizeof *(ptr))
626
627 #ifdef __arch_um__
628 # define OBD_VMALLOC(ptr, size) OBD_ALLOC(ptr, size)
629 #else
630 # define OBD_VMALLOC(ptr, size)                                               \
631 do {                                                                          \
632         (ptr) = cfs_alloc_large(size);                                        \
633         if (unlikely((ptr) == NULL)) {                                        \
634                 CERROR("vmalloc of '" #ptr "' (%d bytes) failed\n",           \
635                        (int)(size));                                          \
636                 CERROR(LPU64" total bytes allocated by Lustre, %d by LNET\n", \
637                        obd_memory_sum(), atomic_read(&libcfs_kmemory));       \
638         } else {                                                              \
639                 memset(ptr, 0, size);                                         \
640                 OBD_ALLOC_POST(ptr, size, "vmalloced");                       \
641         }                                                                     \
642 } while(0)
643 #endif
644
645 #ifdef CONFIG_DEBUG_SLAB
646 #define POISON(ptr, c, s) do {} while (0)
647 #define POISON_PTR(ptr)  ((void)0)
648 #else
649 #define POISON(ptr, c, s) memset(ptr, c, s)
650 #define POISON_PTR(ptr)  (ptr) = (void *)0xdeadbeef
651 #endif
652
653 #ifdef POISON_BULK
654 #define POISON_PAGE(page, val) do { memset(kmap(page), val, CFS_PAGE_SIZE);   \
655                                     kunmap(page); } while (0)
656 #else
657 #define POISON_PAGE(page, val) do { } while (0)
658 #endif
659
660 #ifdef __KERNEL__
661 #define OBD_FREE(ptr, size)                                                   \
662 do {                                                                          \
663         OBD_FREE_PRE(ptr, size, "kfreed");                                    \
664         cfs_free(ptr);                                                        \
665         POISON_PTR(ptr);                                                      \
666 } while(0)
667
668
669 #ifdef HAVE_RCU
670 # ifdef HAVE_CALL_RCU_PARAM
671 #  define my_call_rcu(rcu, cb)            call_rcu(rcu, cb, rcu)
672 # else
673 #  define my_call_rcu(rcu, cb)            call_rcu(rcu, cb)
674 # endif
675 #else
676 # define my_call_rcu(rcu, cb)             (cb)(rcu)
677 #endif
678
679 #define OBD_FREE_RCU_CB(ptr, size, handle, free_cb)                           \
680 do {                                                                          \
681         struct portals_handle *__h = (handle);                                \
682         LASSERT(handle);                                                      \
683         __h->h_ptr = (ptr);                                                   \
684         __h->h_size = (size);                                                 \
685         __h->h_free_cb = (void (*)(void *, size_t))(free_cb);                 \
686         my_call_rcu(&__h->h_rcu, class_handle_free_cb);                       \
687         POISON_PTR(ptr);                                                      \
688 } while(0)
689 #define OBD_FREE_RCU(ptr, size, handle) OBD_FREE_RCU_CB(ptr, size, handle, NULL)
690
691 #else
692 #define OBD_FREE(ptr, size) ((void)(size), free((ptr)))
693 #define OBD_FREE_RCU(ptr, size, handle) (OBD_FREE(ptr, size))
694 #define OBD_FREE_RCU_CB(ptr, size, handle, cb)     ((*(cb))(ptr, size))
695 #endif /* ifdef __KERNEL__ */
696
697 #ifdef __arch_um__
698 # define OBD_VFREE(ptr, size) OBD_FREE(ptr, size)
699 #else
700 # define OBD_VFREE(ptr, size)                                                 \
701 do {                                                                          \
702         OBD_FREE_PRE(ptr, size, "vfreed");                                    \
703         cfs_free_large(ptr);                                                  \
704         POISON_PTR(ptr);                                                      \
705 } while(0)
706 #endif
707
708 /* we memset() the slab object to 0 when allocation succeeds, so DO NOT
709  * HAVE A CTOR THAT DOES ANYTHING.  its work will be cleared here.  we'd
710  * love to assert on that, but slab.c keeps kmem_cache_s all to itself. */
711 #define OBD_SLAB_FREE_RTN0(ptr, slab)                                         \
712 ({                                                                            \
713         cfs_mem_cache_free((slab), (ptr));                                    \
714         (ptr) = NULL;                                                         \
715         0;                                                                    \
716 })
717 #define OBD_SLAB_ALLOC(ptr, slab, type, size)                                 \
718 do {                                                                          \
719         LASSERT(!in_interrupt());                                             \
720         (ptr) = cfs_mem_cache_alloc(slab, (type));                            \
721         if (likely((ptr) != NULL &&                                           \
722                    (!HAS_FAIL_ALLOC_FLAG || obd_alloc_fail_rate == 0 ||       \
723                     !obd_alloc_fail(ptr, #ptr, "slab-", size,                 \
724                                     __FILE__, __LINE__) ||                    \
725                     OBD_SLAB_FREE_RTN0(ptr, slab)))) {                        \
726                 memset(ptr, 0, size);                                         \
727                 OBD_ALLOC_POST(ptr, size, "slab-alloced");                    \
728         }                                                                     \
729 } while(0)
730
731 #define OBD_FREE_PTR(ptr) OBD_FREE(ptr, sizeof *(ptr))
732
733 #define OBD_SLAB_FREE(ptr, slab, size)                                        \
734 do {                                                                          \
735         OBD_FREE_PRE(ptr, size, "slab-freed");                                \
736         cfs_mem_cache_free(slab, ptr);                                        \
737         POISON_PTR(ptr);                                                      \
738 } while(0)
739
740 #define OBD_SLAB_ALLOC_PTR(ptr, slab)                                         \
741         OBD_SLAB_ALLOC((ptr), (slab), CFS_ALLOC_STD, sizeof *(ptr))
742 #define OBD_SLAB_FREE_PTR(ptr, slab)                                          \
743         OBD_SLAB_FREE((ptr), (slab), sizeof *(ptr))
744
745 #define KEY_IS(str) (keylen >= strlen(str) && strncmp(key, str, keylen) == 0)
746
747 /* Wrapper for contiguous page frame allocation */
748 #define OBD_PAGES_ALLOC(ptr, order, gfp_mask)                                 \
749 do {                                                                          \
750         (ptr) = cfs_alloc_pages(gfp_mask, order);                             \
751         if (unlikely((ptr) == NULL)) {                                        \
752                 CERROR("alloc_pages of '" #ptr "' %d page(s) / "LPU64" bytes "\
753                        "failed\n", (int)(1 << (order)),                       \
754                        (__u64)((1 << (order)) << CFS_PAGE_SHIFT));            \
755                 CERROR(LPU64" total bytes and "LPU64" total pages "           \
756                        "("LPU64" bytes) allocated by Lustre, "                \
757                        "%d total bytes by LNET\n",                            \
758                        obd_memory_sum(),                                      \
759                        obd_pages_sum() << CFS_PAGE_SHIFT,                     \
760                        obd_pages_sum(),                                       \
761                        atomic_read(&libcfs_kmemory));                         \
762         } else {                                                              \
763                 obd_pages_add(order);                                         \
764                 CDEBUG(D_MALLOC, "alloc_pages '" #ptr "': %d page(s) / "      \
765                        LPU64" bytes at %p.\n",                                \
766                        (int)(1 << (order)),                                   \
767                        (__u64)((1 << (order)) << CFS_PAGE_SHIFT), ptr);       \
768         }                                                                     \
769 } while (0)
770
771 #define OBD_PAGE_ALLOC(ptr, gfp_mask)                                         \
772         OBD_PAGES_ALLOC(ptr, 0, gfp_mask)
773
774 #define OBD_PAGES_FREE(ptr, order)                                            \
775 do {                                                                          \
776         LASSERT(ptr);                                                         \
777         obd_pages_sub(order);                                                 \
778         CDEBUG(D_MALLOC, "free_pages '" #ptr "': %d page(s) / "LPU64" bytes " \
779                "at %p.\n",                                                    \
780                (int)(1 << (order)), (__u64)((1 << (order)) << CFS_PAGE_SHIFT),\
781                ptr);                                                          \
782         __cfs_free_pages(ptr, order);                                         \
783         (ptr) = (void *)0xdeadbeef;                                           \
784 } while (0)
785
786 #define OBD_PAGE_FREE(ptr) OBD_PAGES_FREE(ptr, 0)
787
788 #endif