Commit | Line | Data |
---|---|---|
7c657876 ACM |
1 | /* |
2 | * net/dccp/ccid.c | |
3 | * | |
4 | * An implementation of the DCCP protocol | |
5 | * Arnaldo Carvalho de Melo <acme@conectiva.com.br> | |
6 | * | |
7 | * CCID infrastructure | |
8 | * | |
9 | * This program is free software; you can redistribute it and/or modify it | |
10 | * under the terms of the GNU General Public License version 2 as | |
11 | * published by the Free Software Foundation. | |
12 | */ | |
13 | ||
14 | #include "ccid.h" | |
15 | ||
ddebc973 GR |
16 | static struct ccid_operations *ccids[] = { |
17 | &ccid2_ops, | |
18 | #ifdef CONFIG_IP_DCCP_CCID3 | |
19 | &ccid3_ops, | |
20 | #endif | |
21 | }; | |
22 | ||
23 | static struct ccid_operations *ccid_by_number(const u8 id) | |
24 | { | |
25 | int i; | |
26 | ||
27 | for (i = 0; i < ARRAY_SIZE(ccids); i++) | |
28 | if (ccids[i]->ccid_id == id) | |
29 | return ccids[i]; | |
30 | return NULL; | |
31 | } | |
32 | ||
33 | /* check that up to @array_len members in @ccid_array are supported */ | |
34 | bool ccid_support_check(u8 const *ccid_array, u8 array_len) | |
35 | { | |
36 | while (array_len > 0) | |
37 | if (ccid_by_number(ccid_array[--array_len]) == NULL) | |
38 | return false; | |
39 | return true; | |
40 | } | |
41 | ||
42 | /** | |
43 | * ccid_get_builtin_ccids - Populate a list of built-in CCIDs | |
44 | * @ccid_array: pointer to copy into | |
45 | * @array_len: value to return length into | |
46 | * This function allocates memory - caller must see that it is freed after use. | |
47 | */ | |
48 | int ccid_get_builtin_ccids(u8 **ccid_array, u8 *array_len) | |
49 | { | |
50 | *ccid_array = kmalloc(ARRAY_SIZE(ccids), gfp_any()); | |
51 | if (*ccid_array == NULL) | |
52 | return -ENOBUFS; | |
53 | ||
54 | for (*array_len = 0; *array_len < ARRAY_SIZE(ccids); *array_len += 1) | |
55 | (*ccid_array)[*array_len] = ccids[*array_len]->ccid_id; | |
56 | return 0; | |
57 | } | |
58 | ||
59 | int ccid_getsockopt_builtin_ccids(struct sock *sk, int len, | |
60 | char __user *optval, int __user *optlen) | |
61 | { | |
62 | u8 *ccid_array, array_len; | |
63 | int err = 0; | |
64 | ||
65 | if (len < ARRAY_SIZE(ccids)) | |
66 | return -EINVAL; | |
67 | ||
68 | if (ccid_get_builtin_ccids(&ccid_array, &array_len)) | |
69 | return -ENOBUFS; | |
70 | ||
71 | if (put_user(array_len, optlen) || | |
72 | copy_to_user(optval, ccid_array, array_len)) | |
73 | err = -EFAULT; | |
74 | ||
75 | kfree(ccid_array); | |
76 | return err; | |
77 | } | |
78 | ||
e18b890b | 79 | static struct kmem_cache *ccid_kmem_cache_create(int obj_size, const char *fmt,...) |
7c657876 | 80 | { |
e18b890b | 81 | struct kmem_cache *slab; |
91f0ebf7 ACM |
82 | char slab_name_fmt[32], *slab_name; |
83 | va_list args; | |
84 | ||
85 | va_start(args, fmt); | |
86 | vsnprintf(slab_name_fmt, sizeof(slab_name_fmt), fmt, args); | |
87 | va_end(args); | |
88 | ||
89 | slab_name = kstrdup(slab_name_fmt, GFP_KERNEL); | |
90 | if (slab_name == NULL) | |
91 | return NULL; | |
92 | slab = kmem_cache_create(slab_name, sizeof(struct ccid) + obj_size, 0, | |
20c2df83 | 93 | SLAB_HWCACHE_ALIGN, NULL); |
91f0ebf7 ACM |
94 | if (slab == NULL) |
95 | kfree(slab_name); | |
96 | return slab; | |
97 | } | |
98 | ||
e18b890b | 99 | static void ccid_kmem_cache_destroy(struct kmem_cache *slab) |
91f0ebf7 ACM |
100 | { |
101 | if (slab != NULL) { | |
102 | const char *name = kmem_cache_name(slab); | |
103 | ||
104 | kmem_cache_destroy(slab); | |
105 | kfree(name); | |
106 | } | |
107 | } | |
108 | ||
ddebc973 | 109 | static int ccid_activate(struct ccid_operations *ccid_ops) |
91f0ebf7 ACM |
110 | { |
111 | int err = -ENOBUFS; | |
112 | ||
113 | ccid_ops->ccid_hc_rx_slab = | |
114 | ccid_kmem_cache_create(ccid_ops->ccid_hc_rx_obj_size, | |
84a97b0a GR |
115 | "ccid%u_hc_rx_sock", |
116 | ccid_ops->ccid_id); | |
91f0ebf7 ACM |
117 | if (ccid_ops->ccid_hc_rx_slab == NULL) |
118 | goto out; | |
119 | ||
120 | ccid_ops->ccid_hc_tx_slab = | |
121 | ccid_kmem_cache_create(ccid_ops->ccid_hc_tx_obj_size, | |
84a97b0a GR |
122 | "ccid%u_hc_tx_sock", |
123 | ccid_ops->ccid_id); | |
91f0ebf7 ACM |
124 | if (ccid_ops->ccid_hc_tx_slab == NULL) |
125 | goto out_free_rx_slab; | |
7c657876 | 126 | |
ddebc973 | 127 | pr_info("CCID: Activated CCID %d (%s)\n", |
91f0ebf7 | 128 | ccid_ops->ccid_id, ccid_ops->ccid_name); |
ddebc973 | 129 | err = 0; |
91f0ebf7 | 130 | out: |
7c657876 | 131 | return err; |
91f0ebf7 ACM |
132 | out_free_rx_slab: |
133 | ccid_kmem_cache_destroy(ccid_ops->ccid_hc_rx_slab); | |
134 | ccid_ops->ccid_hc_rx_slab = NULL; | |
135 | goto out; | |
7c657876 ACM |
136 | } |
137 | ||
ddebc973 | 138 | static void ccid_deactivate(struct ccid_operations *ccid_ops) |
7c657876 | 139 | { |
91f0ebf7 ACM |
140 | ccid_kmem_cache_destroy(ccid_ops->ccid_hc_tx_slab); |
141 | ccid_ops->ccid_hc_tx_slab = NULL; | |
142 | ccid_kmem_cache_destroy(ccid_ops->ccid_hc_rx_slab); | |
143 | ccid_ops->ccid_hc_rx_slab = NULL; | |
144 | ||
ddebc973 | 145 | pr_info("CCID: Deactivated CCID %d (%s)\n", |
91f0ebf7 | 146 | ccid_ops->ccid_id, ccid_ops->ccid_name); |
7c657876 ACM |
147 | } |
148 | ||
e5fd56ca | 149 | struct ccid *ccid_new(const u8 id, struct sock *sk, bool rx) |
7c657876 | 150 | { |
ddebc973 | 151 | struct ccid_operations *ccid_ops = ccid_by_number(id); |
91f0ebf7 | 152 | struct ccid *ccid = NULL; |
7c657876 | 153 | |
91f0ebf7 | 154 | if (ccid_ops == NULL) |
ddebc973 | 155 | goto out; |
7c657876 | 156 | |
91f0ebf7 | 157 | ccid = kmem_cache_alloc(rx ? ccid_ops->ccid_hc_rx_slab : |
e5fd56ca | 158 | ccid_ops->ccid_hc_tx_slab, gfp_any()); |
91f0ebf7 | 159 | if (ccid == NULL) |
ddebc973 | 160 | goto out; |
91f0ebf7 ACM |
161 | ccid->ccid_ops = ccid_ops; |
162 | if (rx) { | |
163 | memset(ccid + 1, 0, ccid_ops->ccid_hc_rx_obj_size); | |
164 | if (ccid->ccid_ops->ccid_hc_rx_init != NULL && | |
165 | ccid->ccid_ops->ccid_hc_rx_init(ccid, sk) != 0) | |
166 | goto out_free_ccid; | |
167 | } else { | |
168 | memset(ccid + 1, 0, ccid_ops->ccid_hc_tx_obj_size); | |
169 | if (ccid->ccid_ops->ccid_hc_tx_init != NULL && | |
170 | ccid->ccid_ops->ccid_hc_tx_init(ccid, sk) != 0) | |
171 | goto out_free_ccid; | |
172 | } | |
7c657876 | 173 | out: |
7c657876 | 174 | return ccid; |
91f0ebf7 ACM |
175 | out_free_ccid: |
176 | kmem_cache_free(rx ? ccid_ops->ccid_hc_rx_slab : | |
177 | ccid_ops->ccid_hc_tx_slab, ccid); | |
7c657876 ACM |
178 | ccid = NULL; |
179 | goto out; | |
180 | } | |
181 | ||
91f0ebf7 ACM |
182 | void ccid_hc_rx_delete(struct ccid *ccid, struct sock *sk) |
183 | { | |
e5fd56ca GR |
184 | if (ccid != NULL) { |
185 | if (ccid->ccid_ops->ccid_hc_rx_exit != NULL) | |
186 | ccid->ccid_ops->ccid_hc_rx_exit(sk); | |
187 | kmem_cache_free(ccid->ccid_ops->ccid_hc_rx_slab, ccid); | |
188 | } | |
91f0ebf7 | 189 | } |
7c657876 | 190 | |
91f0ebf7 ACM |
191 | void ccid_hc_tx_delete(struct ccid *ccid, struct sock *sk) |
192 | { | |
e5fd56ca GR |
193 | if (ccid != NULL) { |
194 | if (ccid->ccid_ops->ccid_hc_tx_exit != NULL) | |
195 | ccid->ccid_ops->ccid_hc_tx_exit(sk); | |
196 | kmem_cache_free(ccid->ccid_ops->ccid_hc_tx_slab, ccid); | |
197 | } | |
7c657876 ACM |
198 | } |
199 | ||
ddebc973 GR |
200 | int __init ccid_initialize_builtins(void) |
201 | { | |
202 | int i, err; | |
203 | ||
204 | for (i = 0; i < ARRAY_SIZE(ccids); i++) { | |
205 | err = ccid_activate(ccids[i]); | |
206 | if (err) | |
207 | goto unwind_registrations; | |
208 | } | |
209 | return 0; | |
210 | ||
211 | unwind_registrations: | |
212 | while(--i >= 0) | |
213 | ccid_deactivate(ccids[i]); | |
214 | return err; | |
215 | } | |
216 | ||
217 | void ccid_cleanup_builtins(void) | |
218 | { | |
219 | int i; | |
220 | ||
221 | for (i = 0; i < ARRAY_SIZE(ccids); i++) | |
222 | ccid_deactivate(ccids[i]); | |
223 | } |