1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26#include <linux/types.h>
27#include <linux/slab.h>
28#include <linux/in.h>
29#include <linux/random.h>
30#include <net/sock.h>
31#include <net/ipv6.h>
32#include <net/sctp/sctp.h>
33#include <net/sctp/sm.h>
34
35
36static void sctp_endpoint_bh_rcv(struct work_struct *work);
37
38
39
40
41static struct sctp_endpoint *sctp_endpoint_init(struct sctp_endpoint *ep,
42 struct sock *sk,
43 gfp_t gfp)
44{
45 struct net *net = sock_net(sk);
46 struct sctp_hmac_algo_param *auth_hmacs = NULL;
47 struct sctp_chunks_param *auth_chunks = NULL;
48 struct sctp_shared_key *null_key;
49 int err;
50
51 ep->digest = kzalloc(SCTP_SIGNATURE_SIZE, gfp);
52 if (!ep->digest)
53 return NULL;
54
55 ep->auth_enable = net->sctp.auth_enable;
56 if (ep->auth_enable) {
57
58
59
60
61 auth_hmacs = kzalloc(struct_size(auth_hmacs, hmac_ids,
62 SCTP_AUTH_NUM_HMACS), gfp);
63 if (!auth_hmacs)
64 goto nomem;
65
66 auth_chunks = kzalloc(sizeof(*auth_chunks) +
67 SCTP_NUM_CHUNK_TYPES, gfp);
68 if (!auth_chunks)
69 goto nomem;
70
71
72
73
74
75
76 auth_hmacs->param_hdr.type = SCTP_PARAM_HMAC_ALGO;
77 auth_hmacs->param_hdr.length =
78 htons(sizeof(struct sctp_paramhdr) + 2);
79 auth_hmacs->hmac_ids[0] = htons(SCTP_AUTH_HMAC_ID_SHA1);
80
81
82 auth_chunks->param_hdr.type = SCTP_PARAM_CHUNKS;
83 auth_chunks->param_hdr.length =
84 htons(sizeof(struct sctp_paramhdr));
85
86
87
88
89 if (net->sctp.addip_enable) {
90 auth_chunks->chunks[0] = SCTP_CID_ASCONF;
91 auth_chunks->chunks[1] = SCTP_CID_ASCONF_ACK;
92 auth_chunks->param_hdr.length =
93 htons(sizeof(struct sctp_paramhdr) + 2);
94 }
95
96
97
98
99 err = sctp_auth_init_hmacs(ep, gfp);
100 if (err)
101 goto nomem;
102 }
103
104
105
106 ep->base.type = SCTP_EP_TYPE_SOCKET;
107
108
109 refcount_set(&ep->base.refcnt, 1);
110 ep->base.dead = false;
111
112
113 sctp_inq_init(&ep->base.inqueue);
114
115
116 sctp_inq_set_th_handler(&ep->base.inqueue, sctp_endpoint_bh_rcv);
117
118
119 sctp_bind_addr_init(&ep->base.bind_addr, 0);
120
121
122 INIT_LIST_HEAD(&ep->asocs);
123
124
125 ep->sndbuf_policy = net->sctp.sndbuf_policy;
126
127 sk->sk_data_ready = sctp_data_ready;
128 sk->sk_write_space = sctp_write_space;
129 sock_set_flag(sk, SOCK_USE_WRITE_QUEUE);
130
131
132 ep->rcvbuf_policy = net->sctp.rcvbuf_policy;
133
134
135 get_random_bytes(ep->secret_key, sizeof(ep->secret_key));
136
137
138 INIT_LIST_HEAD(&ep->endpoint_shared_keys);
139 null_key = sctp_auth_shkey_create(0, gfp);
140 if (!null_key)
141 goto nomem_shkey;
142
143 list_add(&null_key->key_list, &ep->endpoint_shared_keys);
144
145
146
147
148 ep->auth_hmacs_list = auth_hmacs;
149 ep->auth_chunk_list = auth_chunks;
150 ep->prsctp_enable = net->sctp.prsctp_enable;
151 ep->reconf_enable = net->sctp.reconf_enable;
152
153
154 ep->base.sk = sk;
155 sock_hold(ep->base.sk);
156
157 return ep;
158
159nomem_shkey:
160 sctp_auth_destroy_hmacs(ep->auth_hmacs);
161nomem:
162
163 kfree(auth_hmacs);
164 kfree(auth_chunks);
165 kfree(ep->digest);
166 return NULL;
167
168}
169
170
171
172
173struct sctp_endpoint *sctp_endpoint_new(struct sock *sk, gfp_t gfp)
174{
175 struct sctp_endpoint *ep;
176
177
178 ep = kzalloc(sizeof(*ep), gfp);
179 if (!ep)
180 goto fail;
181
182 if (!sctp_endpoint_init(ep, sk, gfp))
183 goto fail_init;
184
185 SCTP_DBG_OBJCNT_INC(ep);
186 return ep;
187
188fail_init:
189 kfree(ep);
190fail:
191 return NULL;
192}
193
194
195void sctp_endpoint_add_asoc(struct sctp_endpoint *ep,
196 struct sctp_association *asoc)
197{
198 struct sock *sk = ep->base.sk;
199
200
201
202
203
204 if (asoc->temp)
205 return;
206
207
208 list_add_tail(&asoc->asocs, &ep->asocs);
209
210
211 if (sctp_style(sk, TCP) && sctp_sstate(sk, LISTENING))
212 sk->sk_ack_backlog++;
213}
214
215
216
217
218void sctp_endpoint_free(struct sctp_endpoint *ep)
219{
220 ep->base.dead = true;
221
222 inet_sk_set_state(ep->base.sk, SCTP_SS_CLOSED);
223
224
225 sctp_unhash_endpoint(ep);
226
227 sctp_endpoint_put(ep);
228}
229
230
231static void sctp_endpoint_destroy(struct sctp_endpoint *ep)
232{
233 struct sock *sk;
234
235 if (unlikely(!ep->base.dead)) {
236 WARN(1, "Attempt to destroy undead endpoint %p!\n", ep);
237 return;
238 }
239
240
241 kfree(ep->digest);
242
243
244
245
246 sctp_auth_destroy_keys(&ep->endpoint_shared_keys);
247 kfree(ep->auth_hmacs_list);
248 kfree(ep->auth_chunk_list);
249
250
251 sctp_auth_destroy_hmacs(ep->auth_hmacs);
252
253
254 sctp_inq_free(&ep->base.inqueue);
255 sctp_bind_addr_free(&ep->base.bind_addr);
256
257 memset(ep->secret_key, 0, sizeof(ep->secret_key));
258
259 sk = ep->base.sk;
260
261 if (sctp_sk(sk)->bind_hash)
262 sctp_put_port(sk);
263
264 sctp_sk(sk)->ep = NULL;
265
266 sock_put(sk);
267
268 kfree(ep);
269 SCTP_DBG_OBJCNT_DEC(ep);
270}
271
272
273void sctp_endpoint_hold(struct sctp_endpoint *ep)
274{
275 refcount_inc(&ep->base.refcnt);
276}
277
278
279
280
281void sctp_endpoint_put(struct sctp_endpoint *ep)
282{
283 if (refcount_dec_and_test(&ep->base.refcnt))
284 sctp_endpoint_destroy(ep);
285}
286
287
288struct sctp_endpoint *sctp_endpoint_is_match(struct sctp_endpoint *ep,
289 struct net *net,
290 const union sctp_addr *laddr)
291{
292 struct sctp_endpoint *retval = NULL;
293
294 if ((htons(ep->base.bind_addr.port) == laddr->v4.sin_port) &&
295 net_eq(sock_net(ep->base.sk), net)) {
296 if (sctp_bind_addr_match(&ep->base.bind_addr, laddr,
297 sctp_sk(ep->base.sk)))
298 retval = ep;
299 }
300
301 return retval;
302}
303
304
305
306
307
308struct sctp_association *sctp_endpoint_lookup_assoc(
309 const struct sctp_endpoint *ep,
310 const union sctp_addr *paddr,
311 struct sctp_transport **transport)
312{
313 struct sctp_association *asoc = NULL;
314 struct sctp_transport *t;
315
316 *transport = NULL;
317
318
319
320
321 if (!ep->base.bind_addr.port)
322 return NULL;
323
324 rcu_read_lock();
325 t = sctp_epaddr_lookup_transport(ep, paddr);
326 if (!t)
327 goto out;
328
329 *transport = t;
330 asoc = t->asoc;
331out:
332 rcu_read_unlock();
333 return asoc;
334}
335
336
337
338
339bool sctp_endpoint_is_peeled_off(struct sctp_endpoint *ep,
340 const union sctp_addr *paddr)
341{
342 struct sctp_sockaddr_entry *addr;
343 struct sctp_bind_addr *bp;
344 struct net *net = sock_net(ep->base.sk);
345
346 bp = &ep->base.bind_addr;
347
348
349
350 list_for_each_entry(addr, &bp->address_list, list) {
351 if (sctp_has_association(net, &addr->a, paddr))
352 return true;
353 }
354
355 return false;
356}
357
358
359
360
361static void sctp_endpoint_bh_rcv(struct work_struct *work)
362{
363 struct sctp_endpoint *ep =
364 container_of(work, struct sctp_endpoint,
365 base.inqueue.immediate);
366 struct sctp_association *asoc;
367 struct sock *sk;
368 struct net *net;
369 struct sctp_transport *transport;
370 struct sctp_chunk *chunk;
371 struct sctp_inq *inqueue;
372 union sctp_subtype subtype;
373 enum sctp_state state;
374 int error = 0;
375 int first_time = 1;
376
377 if (ep->base.dead)
378 return;
379
380 asoc = NULL;
381 inqueue = &ep->base.inqueue;
382 sk = ep->base.sk;
383 net = sock_net(sk);
384
385 while (NULL != (chunk = sctp_inq_pop(inqueue))) {
386 subtype = SCTP_ST_CHUNK(chunk->chunk_hdr->type);
387
388
389
390
391 if (first_time && (subtype.chunk == SCTP_CID_AUTH)) {
392 struct sctp_chunkhdr *next_hdr;
393
394 next_hdr = sctp_inq_peek(inqueue);
395 if (!next_hdr)
396 goto normal;
397
398
399
400
401
402
403 if (next_hdr->type == SCTP_CID_COOKIE_ECHO) {
404 chunk->auth_chunk = skb_clone(chunk->skb,
405 GFP_ATOMIC);
406 chunk->auth = 1;
407 continue;
408 }
409 }
410normal:
411
412
413
414
415
416
417 if (NULL == chunk->asoc) {
418 asoc = sctp_endpoint_lookup_assoc(ep,
419 sctp_source(chunk),
420 &transport);
421 chunk->asoc = asoc;
422 chunk->transport = transport;
423 }
424
425 state = asoc ? asoc->state : SCTP_STATE_CLOSED;
426 if (sctp_auth_recv_cid(subtype.chunk, asoc) && !chunk->auth)
427 continue;
428
429
430
431
432 if (asoc && sctp_chunk_is_data(chunk))
433 asoc->peer.last_data_from = chunk->transport;
434 else {
435 SCTP_INC_STATS(sock_net(ep->base.sk), SCTP_MIB_INCTRLCHUNKS);
436 if (asoc)
437 asoc->stats.ictrlchunks++;
438 }
439
440 if (chunk->transport)
441 chunk->transport->last_time_heard = ktime_get();
442
443 error = sctp_do_sm(net, SCTP_EVENT_T_CHUNK, subtype, state,
444 ep, asoc, chunk, GFP_ATOMIC);
445
446 if (error && chunk)
447 chunk->pdiscard = 1;
448
449
450
451
452 if (!sctp_sk(sk)->ep)
453 break;
454
455 if (first_time)
456 first_time = 0;
457 }
458}
459