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