1// SPDX-License-Identifier: GPL-2.0-only
2/******************************************************************************
3*******************************************************************************
4**
5**  Copyright (C) Sistina Software, Inc.  1997-2003  All rights reserved.
6**  Copyright (C) 2004-2007 Red Hat, Inc.  All rights reserved.
7**
8**
9*******************************************************************************
10******************************************************************************/
11
12#include "dlm_internal.h"
13#include "midcomms.h"
14#include "lowcomms.h"
15#include "config.h"
16#include "memory.h"
17#include "ast.h"
18
19static struct kmem_cache *writequeue_cache;
20static struct kmem_cache *mhandle_cache;
21static struct kmem_cache *msg_cache;
22static struct kmem_cache *lkb_cache;
23static struct kmem_cache *rsb_cache;
24static struct kmem_cache *cb_cache;
25
26
27int __init dlm_memory_init(void)
28{
29	writequeue_cache = dlm_lowcomms_writequeue_cache_create();
30	if (!writequeue_cache)
31		goto out;
32
33	mhandle_cache = dlm_midcomms_cache_create();
34	if (!mhandle_cache)
35		goto mhandle;
36
37	lkb_cache = kmem_cache_create("dlm_lkb", sizeof(struct dlm_lkb),
38				__alignof__(struct dlm_lkb), 0, NULL);
39	if (!lkb_cache)
40		goto lkb;
41
42	msg_cache = dlm_lowcomms_msg_cache_create();
43	if (!msg_cache)
44		goto msg;
45
46	rsb_cache = kmem_cache_create("dlm_rsb", sizeof(struct dlm_rsb),
47				__alignof__(struct dlm_rsb), 0, NULL);
48	if (!rsb_cache)
49		goto rsb;
50
51	cb_cache = kmem_cache_create("dlm_cb", sizeof(struct dlm_callback),
52				     __alignof__(struct dlm_callback), 0,
53				     NULL);
54	if (!cb_cache)
55		goto cb;
56
57	return 0;
58
59cb:
60	kmem_cache_destroy(rsb_cache);
61rsb:
62	kmem_cache_destroy(msg_cache);
63msg:
64	kmem_cache_destroy(lkb_cache);
65lkb:
66	kmem_cache_destroy(mhandle_cache);
67mhandle:
68	kmem_cache_destroy(writequeue_cache);
69out:
70	return -ENOMEM;
71}
72
73void dlm_memory_exit(void)
74{
75	kmem_cache_destroy(writequeue_cache);
76	kmem_cache_destroy(mhandle_cache);
77	kmem_cache_destroy(msg_cache);
78	kmem_cache_destroy(lkb_cache);
79	kmem_cache_destroy(rsb_cache);
80	kmem_cache_destroy(cb_cache);
81}
82
83char *dlm_allocate_lvb(struct dlm_ls *ls)
84{
85	char *p;
86
87	p = kzalloc(ls->ls_lvblen, GFP_NOFS);
88	return p;
89}
90
91void dlm_free_lvb(char *p)
92{
93	kfree(p);
94}
95
96struct dlm_rsb *dlm_allocate_rsb(struct dlm_ls *ls)
97{
98	struct dlm_rsb *r;
99
100	r = kmem_cache_zalloc(rsb_cache, GFP_NOFS);
101	return r;
102}
103
104void dlm_free_rsb(struct dlm_rsb *r)
105{
106	if (r->res_lvbptr)
107		dlm_free_lvb(r->res_lvbptr);
108	kmem_cache_free(rsb_cache, r);
109}
110
111struct dlm_lkb *dlm_allocate_lkb(struct dlm_ls *ls)
112{
113	struct dlm_lkb *lkb;
114
115	lkb = kmem_cache_zalloc(lkb_cache, GFP_NOFS);
116	return lkb;
117}
118
119void dlm_free_lkb(struct dlm_lkb *lkb)
120{
121	if (test_bit(DLM_DFL_USER_BIT, &lkb->lkb_dflags)) {
122		struct dlm_user_args *ua;
123		ua = lkb->lkb_ua;
124		if (ua) {
125			kfree(ua->lksb.sb_lvbptr);
126			kfree(ua);
127		}
128	}
129
130	/* drop references if they are set */
131	dlm_callback_set_last_ptr(&lkb->lkb_last_cast, NULL);
132	dlm_callback_set_last_ptr(&lkb->lkb_last_cb, NULL);
133
134	kmem_cache_free(lkb_cache, lkb);
135}
136
137struct dlm_mhandle *dlm_allocate_mhandle(gfp_t allocation)
138{
139	return kmem_cache_alloc(mhandle_cache, allocation);
140}
141
142void dlm_free_mhandle(struct dlm_mhandle *mhandle)
143{
144	kmem_cache_free(mhandle_cache, mhandle);
145}
146
147struct writequeue_entry *dlm_allocate_writequeue(void)
148{
149	return kmem_cache_alloc(writequeue_cache, GFP_ATOMIC);
150}
151
152void dlm_free_writequeue(struct writequeue_entry *writequeue)
153{
154	kmem_cache_free(writequeue_cache, writequeue);
155}
156
157struct dlm_msg *dlm_allocate_msg(gfp_t allocation)
158{
159	return kmem_cache_alloc(msg_cache, allocation);
160}
161
162void dlm_free_msg(struct dlm_msg *msg)
163{
164	kmem_cache_free(msg_cache, msg);
165}
166
167struct dlm_callback *dlm_allocate_cb(void)
168{
169	return kmem_cache_alloc(cb_cache, GFP_ATOMIC);
170}
171
172void dlm_free_cb(struct dlm_callback *cb)
173{
174	kmem_cache_free(cb_cache, cb);
175}
176