]> git.proxmox.com Git - mirror_ubuntu-jammy-kernel.git/blob - net/packet/af_packet.c
projects / mirror_ubuntu-jammy-kernel.git / blob
? search:
summary | shortlog | log | commit | commitdiff | tree
blame | history | raw | HEAD
Merge branch 'fix/hda' of git://git.kernel.org/pub/scm/linux/kernel/git/tiwai/sound...
[mirror_ubuntu-jammy-kernel.git] / net / packet / af_packet.c
1 /*
2 * INET An implementation of the TCP/IP protocol suite for the LINUX
3 * operating system. INET is implemented using the BSD Socket
4 * interface as the means of communication with the user level.
5 *
6 * PACKET - implements raw packet sockets.
7 *
8 * Authors: Ross Biro
9 * Fred N. van Kempen, <waltje@uWalt.NL.Mugnet.ORG>
10 * Alan Cox, <gw4pts@gw4pts.ampr.org>
11 *
12 * Fixes:
13 * Alan Cox : verify_area() now used correctly
14 * Alan Cox : new skbuff lists, look ma no backlogs!
15 * Alan Cox : tidied skbuff lists.
16 * Alan Cox : Now uses generic datagram routines I
17 * added. Also fixed the peek/read crash
18 * from all old Linux datagram code.
19 * Alan Cox : Uses the improved datagram code.
20 * Alan Cox : Added NULL's for socket options.
21 * Alan Cox : Re-commented the code.
22 * Alan Cox : Use new kernel side addressing
23 * Rob Janssen : Correct MTU usage.
24 * Dave Platt : Counter leaks caused by incorrect
25 * interrupt locking and some slightly
26 * dubious gcc output. Can you read
27 * compiler: it said _VOLATILE_
28 * Richard Kooijman : Timestamp fixes.
29 * Alan Cox : New buffers. Use sk->mac.raw.
30 * Alan Cox : sendmsg/recvmsg support.
31 * Alan Cox : Protocol setting support
32 * Alexey Kuznetsov : Untied from IPv4 stack.
33 * Cyrus Durgin : Fixed kerneld for kmod.
34 * Michal Ostrowski : Module initialization cleanup.
35 * Ulises Alonso : Frame number limit removal and
36 * packet_set_ring memory leak.
37 * Eric Biederman : Allow for > 8 byte hardware addresses.
38 * The convention is that longer addresses
39 * will simply extend the hardware address
40 * byte arrays at the end of sockaddr_ll
41 * and packet_mreq.
42 * Johann Baudy : Added TX RING.
43 *
44 * This program is free software; you can redistribute it and/or
45 * modify it under the terms of the GNU General Public License
46 * as published by the Free Software Foundation; either version
47 * 2 of the License, or (at your option) any later version.
48 *
49 */
50
51 #include <linux/types.h>
52 #include <linux/mm.h>
53 #include <linux/capability.h>
54 #include <linux/fcntl.h>
55 #include <linux/socket.h>
56 #include <linux/in.h>
57 #include <linux/inet.h>
58 #include <linux/netdevice.h>
59 #include <linux/if_packet.h>
60 #include <linux/wireless.h>
61 #include <linux/kernel.h>
62 #include <linux/kmod.h>
63 #include <net/net_namespace.h>
64 #include <net/ip.h>
65 #include <net/protocol.h>
66 #include <linux/skbuff.h>
67 #include <net/sock.h>
68 #include <linux/errno.h>
69 #include <linux/timer.h>
70 #include <asm/system.h>
71 #include <asm/uaccess.h>
72 #include <asm/ioctls.h>
73 #include <asm/page.h>
74 #include <asm/cacheflush.h>
75 #include <asm/io.h>
76 #include <linux/proc_fs.h>
77 #include <linux/seq_file.h>
78 #include <linux/poll.h>
79 #include <linux/module.h>
80 #include <linux/init.h>
81 #include <linux/mutex.h>
82 #include <linux/if_vlan.h>
83
84 #ifdef CONFIG_INET
85 #include <net/inet_common.h>
86 #endif
87
88 /*
89 Assumptions:
90 - if device has no dev->hard_header routine, it adds and removes ll header
91 inside itself. In this case ll header is invisible outside of device,
92 but higher levels still should reserve dev->hard_header_len.
93 Some devices are enough clever to reallocate skb, when header
94 will not fit to reserved space (tunnel), another ones are silly
95 (PPP).
96 - packet socket receives packets with pulled ll header,
97 so that SOCK_RAW should push it back.
98
99 On receive:
100 -----------
101
102 Incoming, dev->hard_header!=NULL
103 mac_header -> ll header
104 data -> data
105
106 Outgoing, dev->hard_header!=NULL
107 mac_header -> ll header
108 data -> ll header
109
110 Incoming, dev->hard_header==NULL
111 mac_header -> UNKNOWN position. It is very likely, that it points to ll
112 header. PPP makes it, that is wrong, because introduce
113 assymetry between rx and tx paths.
114 data -> data
115
116 Outgoing, dev->hard_header==NULL
117 mac_header -> data. ll header is still not built!
118 data -> data
119
120 Resume
121 If dev->hard_header==NULL we are unlikely to restore sensible ll header.
122
123
124 On transmit:
125 ------------
126
127 dev->hard_header != NULL
128 mac_header -> ll header
129 data -> ll header
130
131 dev->hard_header == NULL (ll header is added by device, we cannot control it)
132 mac_header -> data
133 data -> data
134
135 We should set nh.raw on output to correct posistion,
136 packet classifier depends on it.
137 */
138
139 /* Private packet socket structures. */
140
141 struct packet_mclist {
142 struct packet_mclist *next;
143 int ifindex;
144 int count;
145 unsigned short type;
146 unsigned short alen;
147 unsigned char addr[MAX_ADDR_LEN];
148 };
149 /* identical to struct packet_mreq except it has
150 * a longer address field.
151 */
152 struct packet_mreq_max {
153 int mr_ifindex;
154 unsigned short mr_type;
155 unsigned short mr_alen;
156 unsigned char mr_address[MAX_ADDR_LEN];
157 };
158
159 #ifdef CONFIG_PACKET_MMAP
160 static int packet_set_ring(struct sock *sk, struct tpacket_req *req,
161 int closing, int tx_ring);
162
163 struct packet_ring_buffer {
164 char **pg_vec;
165 unsigned int head;
166 unsigned int frames_per_block;
167 unsigned int frame_size;
168 unsigned int frame_max;
169
170 unsigned int pg_vec_order;
171 unsigned int pg_vec_pages;
172 unsigned int pg_vec_len;
173
174 atomic_t pending;
175 };
176
177 struct packet_sock;
178 static int tpacket_snd(struct packet_sock *po, struct msghdr *msg);
179 #endif
180
181 static void packet_flush_mclist(struct sock *sk);
182
183 struct packet_sock {
184 /* struct sock has to be the first member of packet_sock */
185 struct sock sk;
186 struct tpacket_stats stats;
187 #ifdef CONFIG_PACKET_MMAP
188 struct packet_ring_buffer rx_ring;
189 struct packet_ring_buffer tx_ring;
190 int copy_thresh;
191 #endif
192 spinlock_t bind_lock;
193 struct mutex pg_vec_lock;
194 unsigned int running:1, /* prot_hook is attached*/
195 auxdata:1,
196 origdev:1;
197 int ifindex; /* bound device */
198 __be16 num;
199 struct packet_mclist *mclist;
200 #ifdef CONFIG_PACKET_MMAP
201 atomic_t mapped;
202 enum tpacket_versions tp_version;
203 unsigned int tp_hdrlen;
204 unsigned int tp_reserve;
205 unsigned int tp_loss:1;
206 #endif
207 struct packet_type prot_hook ____cacheline_aligned_in_smp;
208 };
209
210 struct packet_skb_cb {
211 unsigned int origlen;
212 union {
213 struct sockaddr_pkt pkt;
214 struct sockaddr_ll ll;
215 } sa;
216 };
217
218 #define PACKET_SKB_CB(__skb) ((struct packet_skb_cb *)((__skb)->cb))
219
220 #ifdef CONFIG_PACKET_MMAP
221
222 static void __packet_set_status(struct packet_sock *po, void *frame, int status)
223 {
224 union {
225 struct tpacket_hdr *h1;
226 struct tpacket2_hdr *h2;
227 void *raw;
228 } h;
229
230 h.raw = frame;
231 switch (po->tp_version) {
232 case TPACKET_V1:
233 h.h1->tp_status = status;
234 flush_dcache_page(virt_to_page(&h.h1->tp_status));
235 break;
236 case TPACKET_V2:
237 h.h2->tp_status = status;
238 flush_dcache_page(virt_to_page(&h.h2->tp_status));
239 break;
240 default:
241 pr_err("TPACKET version not supported\n");
242 BUG();
243 }
244
245 smp_wmb();
246 }
247
248 static int __packet_get_status(struct packet_sock *po, void *frame)
249 {
250 union {
251 struct tpacket_hdr *h1;
252 struct tpacket2_hdr *h2;
253 void *raw;
254 } h;
255
256 smp_rmb();
257
258 h.raw = frame;
259 switch (po->tp_version) {
260 case TPACKET_V1:
261 flush_dcache_page(virt_to_page(&h.h1->tp_status));
262 return h.h1->tp_status;
263 case TPACKET_V2:
264 flush_dcache_page(virt_to_page(&h.h2->tp_status));
265 return h.h2->tp_status;
266 default:
267 pr_err("TPACKET version not supported\n");
268 BUG();
269 return 0;
270 }
271 }
272
273 static void *packet_lookup_frame(struct packet_sock *po,
274 struct packet_ring_buffer *rb,
275 unsigned int position,
276 int status)
277 {
278 unsigned int pg_vec_pos, frame_offset;
279 union {
280 struct tpacket_hdr *h1;
281 struct tpacket2_hdr *h2;
282 void *raw;
283 } h;
284
285 pg_vec_pos = position / rb->frames_per_block;
286 frame_offset = position % rb->frames_per_block;
287
288 h.raw = rb->pg_vec[pg_vec_pos] + (frame_offset * rb->frame_size);
289
290 if (status != __packet_get_status(po, h.raw))
291 return NULL;
292
293 return h.raw;
294 }
295
296 static inline void *packet_current_frame(struct packet_sock *po,
297 struct packet_ring_buffer *rb,
298 int status)
299 {
300 return packet_lookup_frame(po, rb, rb->head, status);
301 }
302
303 static inline void *packet_previous_frame(struct packet_sock *po,
304 struct packet_ring_buffer *rb,
305 int status)
306 {
307 unsigned int previous = rb->head ? rb->head - 1 : rb->frame_max;
308 return packet_lookup_frame(po, rb, previous, status);
309 }
310
311 static inline void packet_increment_head(struct packet_ring_buffer *buff)
312 {
313 buff->head = buff->head != buff->frame_max ? buff->head+1 : 0;
314 }
315
316 #endif
317
318 static inline struct packet_sock *pkt_sk(struct sock *sk)
319 {
320 return (struct packet_sock *)sk;
321 }
322
323 static void packet_sock_destruct(struct sock *sk)
324 {
325 WARN_ON(atomic_read(&sk->sk_rmem_alloc));
326 WARN_ON(atomic_read(&sk->sk_wmem_alloc));
327
328 if (!sock_flag(sk, SOCK_DEAD)) {
329 pr_err("Attempt to release alive packet socket: %p\n", sk);
330 return;
331 }
332
333 sk_refcnt_debug_dec(sk);
334 }
335
336
337 static const struct proto_ops packet_ops;
338
339 static const struct proto_ops packet_ops_spkt;
340
341 static int packet_rcv_spkt(struct sk_buff *skb, struct net_device *dev,
342 struct packet_type *pt, struct net_device *orig_dev)
343 {
344 struct sock *sk;
345 struct sockaddr_pkt *spkt;
346
347 /*
348 * When we registered the protocol we saved the socket in the data
349 * field for just this event.
350 */
351
352 sk = pt->af_packet_priv;
353
354 /*
355 * Yank back the headers [hope the device set this
356 * right or kerboom...]
357 *
358 * Incoming packets have ll header pulled,
359 * push it back.
360 *
361 * For outgoing ones skb->data == skb_mac_header(skb)
362 * so that this procedure is noop.
363 */
364
365 if (skb->pkt_type == PACKET_LOOPBACK)
366 goto out;
367
368 if (!net_eq(dev_net(dev), sock_net(sk)))
369 goto out;
370
371 skb = skb_share_check(skb, GFP_ATOMIC);
372 if (skb == NULL)
373 goto oom;
374
375 /* drop any routing info */
376 skb_dst_drop(skb);
377
378 /* drop conntrack reference */
379 nf_reset(skb);
380
381 spkt = &PACKET_SKB_CB(skb)->sa.pkt;
382
383 skb_push(skb, skb->data - skb_mac_header(skb));
384
385 /*
386 * The SOCK_PACKET socket receives _all_ frames.
387 */
388
389 spkt->spkt_family = dev->type;
390 strlcpy(spkt->spkt_device, dev->name, sizeof(spkt->spkt_device));
391 spkt->spkt_protocol = skb->protocol;
392
393 /*
394 * Charge the memory to the socket. This is done specifically
395 * to prevent sockets using all the memory up.
396 */
397
398 if (sock_queue_rcv_skb(sk, skb) == 0)
399 return 0;
400
401 out:
402 kfree_skb(skb);
403 oom:
404 return 0;
405 }
406
407
408 /*
409 * Output a raw packet to a device layer. This bypasses all the other
410 * protocol layers and you must therefore supply it with a complete frame
411 */
412
413 static int packet_sendmsg_spkt(struct kiocb *iocb, struct socket *sock,
414 struct msghdr *msg, size_t len)
415 {
416 struct sock *sk = sock->sk;
417 struct sockaddr_pkt *saddr = (struct sockaddr_pkt *)msg->msg_name;
418 struct sk_buff *skb;
419 struct net_device *dev;
420 __be16 proto = 0;
421 int err;
422
423 /*
424 * Get and verify the address.
425 */
426
427 if (saddr) {
428 if (msg->msg_namelen < sizeof(struct sockaddr))
429 return -EINVAL;
430 if (msg->msg_namelen == sizeof(struct sockaddr_pkt))
431 proto = saddr->spkt_protocol;
432 } else
433 return -ENOTCONN; /* SOCK_PACKET must be sent giving an address */
434
435 /*
436 * Find the device first to size check it
437 */
438
439 saddr->spkt_device[13] = 0;
440 rcu_read_lock();
441 dev = dev_get_by_name_rcu(sock_net(sk), saddr->spkt_device);
442 err = -ENODEV;
443 if (dev == NULL)
444 goto out_unlock;
445
446 err = -ENETDOWN;
447 if (!(dev->flags & IFF_UP))
448 goto out_unlock;
449
450 /*
451 * You may not queue a frame bigger than the mtu. This is the lowest level
452 * raw protocol and you must do your own fragmentation at this level.
453 */
454
455 err = -EMSGSIZE;
456 if (len > dev->mtu + dev->hard_header_len)
457 goto out_unlock;
458
459 err = -ENOBUFS;
460 skb = sock_wmalloc(sk, len + LL_RESERVED_SPACE(dev), 0, GFP_KERNEL);
461
462 /*
463 * If the write buffer is full, then tough. At this level the user
464 * gets to deal with the problem - do your own algorithmic backoffs.
465 * That's far more flexible.
466 */
467
468 if (skb == NULL)
469 goto out_unlock;
470
471 /*
472 * Fill it in
473 */
474
475 /* FIXME: Save some space for broken drivers that write a
476 * hard header at transmission time by themselves. PPP is the
477 * notable one here. This should really be fixed at the driver level.
478 */
479 skb_reserve(skb, LL_RESERVED_SPACE(dev));
480 skb_reset_network_header(skb);
481
482 /* Try to align data part correctly */
483 if (dev->header_ops) {
484 skb->data -= dev->hard_header_len;
485 skb->tail -= dev->hard_header_len;
486 if (len < dev->hard_header_len)
487 skb_reset_network_header(skb);
488 }
489
490 /* Returns -EFAULT on error */
491 err = memcpy_fromiovec(skb_put(skb, len), msg->msg_iov, len);
492 skb->protocol = proto;
493 skb->dev = dev;
494 skb->priority = sk->sk_priority;
495 skb->mark = sk->sk_mark;
496 if (err)
497 goto out_free;
498
499 /*
500 * Now send it
501 */
502
503 dev_queue_xmit(skb);
504 rcu_read_unlock();
505 return len;
506
507 out_free:
508 kfree_skb(skb);
509 out_unlock:
510 rcu_read_unlock();
511 return err;
512 }
513
514 static inline unsigned int run_filter(struct sk_buff *skb, struct sock *sk,
515 unsigned int res)
516 {
517 struct sk_filter *filter;
518
519 rcu_read_lock_bh();
520 filter = rcu_dereference(sk->sk_filter);
521 if (filter != NULL)
522 res = sk_run_filter(skb, filter->insns, filter->len);
523 rcu_read_unlock_bh();
524
525 return res;
526 }
527
528 /*
529 This function makes lazy skb cloning in hope that most of packets
530 are discarded by BPF.
531
532 Note tricky part: we DO mangle shared skb! skb->data, skb->len
533 and skb->cb are mangled. It works because (and until) packets
534 falling here are owned by current CPU. Output packets are cloned
535 by dev_queue_xmit_nit(), input packets are processed by net_bh
536 sequencially, so that if we return skb to original state on exit,
537 we will not harm anyone.
538 */
539
540 static int packet_rcv(struct sk_buff *skb, struct net_device *dev,
541 struct packet_type *pt, struct net_device *orig_dev)
542 {
543 struct sock *sk;
544 struct sockaddr_ll *sll;
545 struct packet_sock *po;
546 u8 *skb_head = skb->data;
547 int skb_len = skb->len;
548 unsigned int snaplen, res;
549
550 if (skb->pkt_type == PACKET_LOOPBACK)
551 goto drop;
552
553 sk = pt->af_packet_priv;
554 po = pkt_sk(sk);
555
556 if (!net_eq(dev_net(dev), sock_net(sk)))
557 goto drop;
558
559 skb->dev = dev;
560
561 if (dev->header_ops) {
562 /* The device has an explicit notion of ll header,
563 exported to higher levels.
564
565 Otherwise, the device hides datails of it frame
566 structure, so that corresponding packet head
567 never delivered to user.
568 */
569 if (sk->sk_type != SOCK_DGRAM)
570 skb_push(skb, skb->data - skb_mac_header(skb));
571 else if (skb->pkt_type == PACKET_OUTGOING) {
572 /* Special case: outgoing packets have ll header at head */
573 skb_pull(skb, skb_network_offset(skb));
574 }
575 }
576
577 snaplen = skb->len;
578
579 res = run_filter(skb, sk, snaplen);
580 if (!res)
581 goto drop_n_restore;
582 if (snaplen > res)
583 snaplen = res;
584
585 if (atomic_read(&sk->sk_rmem_alloc) + skb->truesize >=
586 (unsigned)sk->sk_rcvbuf)
587 goto drop_n_acct;
588
589 if (skb_shared(skb)) {
590 struct sk_buff *nskb = skb_clone(skb, GFP_ATOMIC);
591 if (nskb == NULL)
592 goto drop_n_acct;
593
594 if (skb_head != skb->data) {
595 skb->data = skb_head;
596 skb->len = skb_len;
597 }
598 kfree_skb(skb);
599 skb = nskb;
600 }
601
602 BUILD_BUG_ON(sizeof(*PACKET_SKB_CB(skb)) + MAX_ADDR_LEN - 8 >
603 sizeof(skb->cb));
604
605 sll = &PACKET_SKB_CB(skb)->sa.ll;
606 sll->sll_family = AF_PACKET;
607 sll->sll_hatype = dev->type;
608 sll->sll_protocol = skb->protocol;
609 sll->sll_pkttype = skb->pkt_type;
610 if (unlikely(po->origdev))
611 sll->sll_ifindex = orig_dev->ifindex;
612 else
613 sll->sll_ifindex = dev->ifindex;
614
615 sll->sll_halen = dev_parse_header(skb, sll->sll_addr);
616
617 PACKET_SKB_CB(skb)->origlen = skb->len;
618
619 if (pskb_trim(skb, snaplen))
620 goto drop_n_acct;
621
622 skb_set_owner_r(skb, sk);
623 skb->dev = NULL;
624 skb_dst_drop(skb);
625
626 /* drop conntrack reference */
627 nf_reset(skb);
628
629 spin_lock(&sk->sk_receive_queue.lock);
630 po->stats.tp_packets++;
631 skb->dropcount = atomic_read(&sk->sk_drops);
632 __skb_queue_tail(&sk->sk_receive_queue, skb);
633 spin_unlock(&sk->sk_receive_queue.lock);
634 sk->sk_data_ready(sk, skb->len);
635 return 0;
636
637 drop_n_acct:
638 po->stats.tp_drops = atomic_inc_return(&sk->sk_drops);
639
640 drop_n_restore:
641 if (skb_head != skb->data && skb_shared(skb)) {
642 skb->data = skb_head;
643 skb->len = skb_len;
644 }
645 drop:
646 consume_skb(skb);
647 return 0;
648 }
649
650 #ifdef CONFIG_PACKET_MMAP
651 static int tpacket_rcv(struct sk_buff *skb, struct net_device *dev,
652 struct packet_type *pt, struct net_device *orig_dev)
653 {
654 struct sock *sk;
655 struct packet_sock *po;
656 struct sockaddr_ll *sll;
657 union {
658 struct tpacket_hdr *h1;
659 struct tpacket2_hdr *h2;
660 void *raw;
661 } h;
662 u8 *skb_head = skb->data;
663 int skb_len = skb->len;
664 unsigned int snaplen, res;
665 unsigned long status = TP_STATUS_LOSING|TP_STATUS_USER;
666 unsigned short macoff, netoff, hdrlen;
667 struct sk_buff *copy_skb = NULL;
668 struct timeval tv;
669 struct timespec ts;
670
671 if (skb->pkt_type == PACKET_LOOPBACK)
672 goto drop;
673
674 sk = pt->af_packet_priv;
675 po = pkt_sk(sk);
676
677 if (!net_eq(dev_net(dev), sock_net(sk)))
678 goto drop;
679
680 if (dev->header_ops) {
681 if (sk->sk_type != SOCK_DGRAM)
682 skb_push(skb, skb->data - skb_mac_header(skb));
683 else if (skb->pkt_type == PACKET_OUTGOING) {
684 /* Special case: outgoing packets have ll header at head */
685 skb_pull(skb, skb_network_offset(skb));
686 }
687 }
688
689 if (skb->ip_summed == CHECKSUM_PARTIAL)
690 status |= TP_STATUS_CSUMNOTREADY;
691
692 snaplen = skb->len;
693
694 res = run_filter(skb, sk, snaplen);
695 if (!res)
696 goto drop_n_restore;
697 if (snaplen > res)
698 snaplen = res;
699
700 if (sk->sk_type == SOCK_DGRAM) {
701 macoff = netoff = TPACKET_ALIGN(po->tp_hdrlen) + 16 +
702 po->tp_reserve;
703 } else {
704 unsigned maclen = skb_network_offset(skb);
705 netoff = TPACKET_ALIGN(po->tp_hdrlen +
706 (maclen < 16 ? 16 : maclen)) +
707 po->tp_reserve;
708 macoff = netoff - maclen;
709 }
710
711 if (macoff + snaplen > po->rx_ring.frame_size) {
712 if (po->copy_thresh &&
713 atomic_read(&sk->sk_rmem_alloc) + skb->truesize <
714 (unsigned)sk->sk_rcvbuf) {
715 if (skb_shared(skb)) {
716 copy_skb = skb_clone(skb, GFP_ATOMIC);
717 } else {
718 copy_skb = skb_get(skb);
719 skb_head = skb->data;
720 }
721 if (copy_skb)
722 skb_set_owner_r(copy_skb, sk);
723 }
724 snaplen = po->rx_ring.frame_size - macoff;
725 if ((int)snaplen < 0)
726 snaplen = 0;
727 }
728
729 spin_lock(&sk->sk_receive_queue.lock);
730 h.raw = packet_current_frame(po, &po->rx_ring, TP_STATUS_KERNEL);
731 if (!h.raw)
732 goto ring_is_full;
733 packet_increment_head(&po->rx_ring);
734 po->stats.tp_packets++;
735 if (copy_skb) {
736 status |= TP_STATUS_COPY;
737 __skb_queue_tail(&sk->sk_receive_queue, copy_skb);
738 }
739 if (!po->stats.tp_drops)
740 status &= ~TP_STATUS_LOSING;
741 spin_unlock(&sk->sk_receive_queue.lock);
742
743 skb_copy_bits(skb, 0, h.raw + macoff, snaplen);
744
745 switch (po->tp_version) {
746 case TPACKET_V1:
747 h.h1->tp_len = skb->len;
748 h.h1->tp_snaplen = snaplen;
749 h.h1->tp_mac = macoff;
750 h.h1->tp_net = netoff;
751 if (skb->tstamp.tv64)
752 tv = ktime_to_timeval(skb->tstamp);
753 else
754 do_gettimeofday(&tv);
755 h.h1->tp_sec = tv.tv_sec;
756 h.h1->tp_usec = tv.tv_usec;
757 hdrlen = sizeof(*h.h1);
758 break;
759 case TPACKET_V2:
760 h.h2->tp_len = skb->len;
761 h.h2->tp_snaplen = snaplen;
762 h.h2->tp_mac = macoff;
763 h.h2->tp_net = netoff;
764 if (skb->tstamp.tv64)
765 ts = ktime_to_timespec(skb->tstamp);
766 else
767 getnstimeofday(&ts);
768 h.h2->tp_sec = ts.tv_sec;
769 h.h2->tp_nsec = ts.tv_nsec;
770 h.h2->tp_vlan_tci = vlan_tx_tag_get(skb);
771 hdrlen = sizeof(*h.h2);
772 break;
773 default:
774 BUG();
775 }
776
777 sll = h.raw + TPACKET_ALIGN(hdrlen);
778 sll->sll_halen = dev_parse_header(skb, sll->sll_addr);
779 sll->sll_family = AF_PACKET;
780 sll->sll_hatype = dev->type;
781 sll->sll_protocol = skb->protocol;
782 sll->sll_pkttype = skb->pkt_type;
783 if (unlikely(po->origdev))
784 sll->sll_ifindex = orig_dev->ifindex;
785 else
786 sll->sll_ifindex = dev->ifindex;
787
788 __packet_set_status(po, h.raw, status);
789 smp_mb();
790 {
791 struct page *p_start, *p_end;
792 u8 *h_end = h.raw + macoff + snaplen - 1;
793
794 p_start = virt_to_page(h.raw);
795 p_end = virt_to_page(h_end);
796 while (p_start <= p_end) {
797 flush_dcache_page(p_start);
798 p_start++;
799 }
800 }
801
802 sk->sk_data_ready(sk, 0);
803
804 drop_n_restore:
805 if (skb_head != skb->data && skb_shared(skb)) {
806 skb->data = skb_head;
807 skb->len = skb_len;
808 }
809 drop:
810 kfree_skb(skb);
811 return 0;
812
813 ring_is_full:
814 po->stats.tp_drops++;
815 spin_unlock(&sk->sk_receive_queue.lock);
816
817 sk->sk_data_ready(sk, 0);
818 kfree_skb(copy_skb);
819 goto drop_n_restore;
820 }
821
822 static void tpacket_destruct_skb(struct sk_buff *skb)
823 {
824 struct packet_sock *po = pkt_sk(skb->sk);
825 void *ph;
826
827 BUG_ON(skb == NULL);
828
829 if (likely(po->tx_ring.pg_vec)) {
830 ph = skb_shinfo(skb)->destructor_arg;
831 BUG_ON(__packet_get_status(po, ph) != TP_STATUS_SENDING);
832 BUG_ON(atomic_read(&po->tx_ring.pending) == 0);
833 atomic_dec(&po->tx_ring.pending);
834 __packet_set_status(po, ph, TP_STATUS_AVAILABLE);
835 }
836
837 sock_wfree(skb);
838 }
839
840 static int tpacket_fill_skb(struct packet_sock *po, struct sk_buff *skb,
841 void *frame, struct net_device *dev, int size_max,
842 __be16 proto, unsigned char *addr)
843 {
844 union {
845 struct tpacket_hdr *h1;
846 struct tpacket2_hdr *h2;
847 void *raw;
848 } ph;
849 int to_write, offset, len, tp_len, nr_frags, len_max;
850 struct socket *sock = po->sk.sk_socket;
851 struct page *page;
852 void *data;
853 int err;
854
855 ph.raw = frame;
856
857 skb->protocol = proto;
858 skb->dev = dev;
859 skb->priority = po->sk.sk_priority;
860 skb->mark = po->sk.sk_mark;
861 skb_shinfo(skb)->destructor_arg = ph.raw;
862
863 switch (po->tp_version) {
864 case TPACKET_V2:
865 tp_len = ph.h2->tp_len;
866 break;
867 default:
868 tp_len = ph.h1->tp_len;
869 break;
870 }
871 if (unlikely(tp_len > size_max)) {
872 pr_err("packet size is too long (%d > %d)\n", tp_len, size_max);
873 return -EMSGSIZE;
874 }
875
876 skb_reserve(skb, LL_RESERVED_SPACE(dev));
877 skb_reset_network_header(skb);
878
879 data = ph.raw + po->tp_hdrlen - sizeof(struct sockaddr_ll);
880 to_write = tp_len;
881
882 if (sock->type == SOCK_DGRAM) {
883 err = dev_hard_header(skb, dev, ntohs(proto), addr,
884 NULL, tp_len);
885 if (unlikely(err < 0))
886 return -EINVAL;
887 } else if (dev->hard_header_len) {
888 /* net device doesn't like empty head */
889 if (unlikely(tp_len <= dev->hard_header_len)) {
890 pr_err("packet size is too short (%d < %d)\n",
891 tp_len, dev->hard_header_len);
892 return -EINVAL;
893 }
894
895 skb_push(skb, dev->hard_header_len);
896 err = skb_store_bits(skb, 0, data,
897 dev->hard_header_len);
898 if (unlikely(err))
899 return err;
900
901 data += dev->hard_header_len;
902 to_write -= dev->hard_header_len;
903 }
904
905 err = -EFAULT;
906 page = virt_to_page(data);
907 offset = offset_in_page(data);
908 len_max = PAGE_SIZE - offset;
909 len = ((to_write > len_max) ? len_max : to_write);
910
911 skb->data_len = to_write;
912 skb->len += to_write;
913 skb->truesize += to_write;
914 atomic_add(to_write, &po->sk.sk_wmem_alloc);
915
916 while (likely(to_write)) {
917 nr_frags = skb_shinfo(skb)->nr_frags;
918
919 if (unlikely(nr_frags >= MAX_SKB_FRAGS)) {
920 pr_err("Packet exceed the number of skb frags(%lu)\n",
921 MAX_SKB_FRAGS);
922 return -EFAULT;
923 }
924
925 flush_dcache_page(page);
926 get_page(page);
927 skb_fill_page_desc(skb,
928 nr_frags,
929 page++, offset, len);
930 to_write -= len;
931 offset = 0;
932 len_max = PAGE_SIZE;
933 len = ((to_write > len_max) ? len_max : to_write);
934 }
935
936 return tp_len;
937 }
938
939 static int tpacket_snd(struct packet_sock *po, struct msghdr *msg)
940 {
941 struct socket *sock;
942 struct sk_buff *skb;
943 struct net_device *dev;
944 __be16 proto;
945 int ifindex, err, reserve = 0;
946 void *ph;
947 struct sockaddr_ll *saddr = (struct sockaddr_ll *)msg->msg_name;
948 int tp_len, size_max;
949 unsigned char *addr;
950 int len_sum = 0;
951 int status = 0;
952
953 sock = po->sk.sk_socket;
954
955 mutex_lock(&po->pg_vec_lock);
956
957 err = -EBUSY;
958 if (saddr == NULL) {
959 ifindex = po->ifindex;
960 proto = po->num;
961 addr = NULL;
962 } else {
963 err = -EINVAL;
964 if (msg->msg_namelen < sizeof(struct sockaddr_ll))
965 goto out;
966 if (msg->msg_namelen < (saddr->sll_halen
967 + offsetof(struct sockaddr_ll,
968 sll_addr)))
969 goto out;
970 ifindex = saddr->sll_ifindex;
971 proto = saddr->sll_protocol;
972 addr = saddr->sll_addr;
973 }
974
975 dev = dev_get_by_index(sock_net(&po->sk), ifindex);
976 err = -ENXIO;
977 if (unlikely(dev == NULL))
978 goto out;
979
980 reserve = dev->hard_header_len;
981
982 err = -ENETDOWN;
983 if (unlikely(!(dev->flags & IFF_UP)))
984 goto out_put;
985
986 size_max = po->tx_ring.frame_size
987 - (po->tp_hdrlen - sizeof(struct sockaddr_ll));
988
989 if (size_max > dev->mtu + reserve)
990 size_max = dev->mtu + reserve;
991
992 do {
993 ph = packet_current_frame(po, &po->tx_ring,
994 TP_STATUS_SEND_REQUEST);
995
996 if (unlikely(ph == NULL)) {
997 schedule();
998 continue;
999 }
1000
1001 status = TP_STATUS_SEND_REQUEST;
1002 skb = sock_alloc_send_skb(&po->sk,
1003 LL_ALLOCATED_SPACE(dev)
1004 + sizeof(struct sockaddr_ll),
1005 0, &err);
1006
1007 if (unlikely(skb == NULL))
1008 goto out_status;
1009
1010 tp_len = tpacket_fill_skb(po, skb, ph, dev, size_max, proto,
1011 addr);
1012
1013 if (unlikely(tp_len < 0)) {
1014 if (po->tp_loss) {
1015 __packet_set_status(po, ph,
1016 TP_STATUS_AVAILABLE);
1017 packet_increment_head(&po->tx_ring);
1018 kfree_skb(skb);
1019 continue;
1020 } else {
1021 status = TP_STATUS_WRONG_FORMAT;
1022 err = tp_len;
1023 goto out_status;
1024 }
1025 }
1026
1027 skb->destructor = tpacket_destruct_skb;
1028 __packet_set_status(po, ph, TP_STATUS_SENDING);
1029 atomic_inc(&po->tx_ring.pending);
1030
1031 status = TP_STATUS_SEND_REQUEST;
1032 err = dev_queue_xmit(skb);
1033 if (unlikely(err > 0 && (err = net_xmit_errno(err)) != 0))
1034 goto out_xmit;
1035 packet_increment_head(&po->tx_ring);
1036 len_sum += tp_len;
1037 } while (likely((ph != NULL) ||
1038 ((!(msg->msg_flags & MSG_DONTWAIT)) &&
1039 (atomic_read(&po->tx_ring.pending))))
1040 );
1041
1042 err = len_sum;
1043 goto out_put;
1044
1045 out_xmit:
1046 skb->destructor = sock_wfree;
1047 atomic_dec(&po->tx_ring.pending);
1048 out_status:
1049 __packet_set_status(po, ph, status);
1050 kfree_skb(skb);
1051 out_put:
1052 dev_put(dev);
1053 out:
1054 mutex_unlock(&po->pg_vec_lock);
1055 return err;
1056 }
1057 #endif
1058
1059 static int packet_snd(struct socket *sock,
1060 struct msghdr *msg, size_t len)
1061 {
1062 struct sock *sk = sock->sk;
1063 struct sockaddr_ll *saddr = (struct sockaddr_ll *)msg->msg_name;
1064 struct sk_buff *skb;
1065 struct net_device *dev;
1066 __be16 proto;
1067 unsigned char *addr;
1068 int ifindex, err, reserve = 0;
1069
1070 /*
1071 * Get and verify the address.
1072 */
1073
1074 if (saddr == NULL) {
1075 struct packet_sock *po = pkt_sk(sk);
1076
1077 ifindex = po->ifindex;
1078 proto = po->num;
1079 addr = NULL;
1080 } else {
1081 err = -EINVAL;
1082 if (msg->msg_namelen < sizeof(struct sockaddr_ll))
1083 goto out;
1084 if (msg->msg_namelen < (saddr->sll_halen + offsetof(struct sockaddr_ll, sll_addr)))
1085 goto out;
1086 ifindex = saddr->sll_ifindex;
1087 proto = saddr->sll_protocol;
1088 addr = saddr->sll_addr;
1089 }
1090
1091
1092 dev = dev_get_by_index(sock_net(sk), ifindex);
1093 err = -ENXIO;
1094 if (dev == NULL)
1095 goto out_unlock;
1096 if (sock->type == SOCK_RAW)
1097 reserve = dev->hard_header_len;
1098
1099 err = -ENETDOWN;
1100 if (!(dev->flags & IFF_UP))
1101 goto out_unlock;
1102
1103 err = -EMSGSIZE;
1104 if (len > dev->mtu+reserve)
1105 goto out_unlock;
1106
1107 skb = sock_alloc_send_skb(sk, len + LL_ALLOCATED_SPACE(dev),
1108 msg->msg_flags & MSG_DONTWAIT, &err);
1109 if (skb == NULL)
1110 goto out_unlock;
1111
1112 skb_reserve(skb, LL_RESERVED_SPACE(dev));
1113 skb_reset_network_header(skb);
1114
1115 err = -EINVAL;
1116 if (sock->type == SOCK_DGRAM &&
1117 dev_hard_header(skb, dev, ntohs(proto), addr, NULL, len) < 0)
1118 goto out_free;
1119
1120 /* Returns -EFAULT on error */
1121 err = memcpy_fromiovec(skb_put(skb, len), msg->msg_iov, len);
1122 if (err)
1123 goto out_free;
1124
1125 skb->protocol = proto;
1126 skb->dev = dev;
1127 skb->priority = sk->sk_priority;
1128 skb->mark = sk->sk_mark;
1129
1130 /*
1131 * Now send it
1132 */
1133
1134 err = dev_queue_xmit(skb);
1135 if (err > 0 && (err = net_xmit_errno(err)) != 0)
1136 goto out_unlock;
1137
1138 dev_put(dev);
1139
1140 return len;
1141
1142 out_free:
1143 kfree_skb(skb);
1144 out_unlock:
1145 if (dev)
1146 dev_put(dev);
1147 out:
1148 return err;
1149 }
1150
1151 static int packet_sendmsg(struct kiocb *iocb, struct socket *sock,
1152 struct msghdr *msg, size_t len)
1153 {
1154 #ifdef CONFIG_PACKET_MMAP
1155 struct sock *sk = sock->sk;
1156 struct packet_sock *po = pkt_sk(sk);
1157 if (po->tx_ring.pg_vec)
1158 return tpacket_snd(po, msg);
1159 else
1160 #endif
1161 return packet_snd(sock, msg, len);
1162 }
1163
1164 /*
1165 * Close a PACKET socket. This is fairly simple. We immediately go
1166 * to 'closed' state and remove our protocol entry in the device list.
1167 */
1168
1169 static int packet_release(struct socket *sock)
1170 {
1171 struct sock *sk = sock->sk;
1172 struct packet_sock *po;
1173 struct net *net;
1174 #ifdef CONFIG_PACKET_MMAP
1175 struct tpacket_req req;
1176 #endif
1177
1178 if (!sk)
1179 return 0;
1180
1181 net = sock_net(sk);
1182 po = pkt_sk(sk);
1183
1184 write_lock_bh(&net->packet.sklist_lock);
1185 sk_del_node_init(sk);
1186 sock_prot_inuse_add(net, sk->sk_prot, -1);
1187 write_unlock_bh(&net->packet.sklist_lock);
1188
1189 /*
1190 * Unhook packet receive handler.
1191 */
1192
1193 if (po->running) {
1194 /*
1195 * Remove the protocol hook
1196 */
1197 dev_remove_pack(&po->prot_hook);
1198 po->running = 0;
1199 po->num = 0;
1200 __sock_put(sk);
1201 }
1202
1203 packet_flush_mclist(sk);
1204
1205 #ifdef CONFIG_PACKET_MMAP
1206 memset(&req, 0, sizeof(req));
1207
1208 if (po->rx_ring.pg_vec)
1209 packet_set_ring(sk, &req, 1, 0);
1210
1211 if (po->tx_ring.pg_vec)
1212 packet_set_ring(sk, &req, 1, 1);
1213 #endif
1214
1215 /*
1216 * Now the socket is dead. No more input will appear.
1217 */
1218
1219 sock_orphan(sk);
1220 sock->sk = NULL;
1221
1222 /* Purge queues */
1223
1224 skb_queue_purge(&sk->sk_receive_queue);
1225 sk_refcnt_debug_release(sk);
1226
1227 sock_put(sk);
1228 return 0;
1229 }
1230
1231 /*
1232 * Attach a packet hook.
1233 */
1234
1235 static int packet_do_bind(struct sock *sk, struct net_device *dev, __be16 protocol)
1236 {
1237 struct packet_sock *po = pkt_sk(sk);
1238 /*
1239 * Detach an existing hook if present.
1240 */
1241
1242 lock_sock(sk);
1243
1244 spin_lock(&po->bind_lock);
1245 if (po->running) {
1246 __sock_put(sk);
1247 po->running = 0;
1248 po->num = 0;
1249 spin_unlock(&po->bind_lock);
1250 dev_remove_pack(&po->prot_hook);
1251 spin_lock(&po->bind_lock);
1252 }
1253
1254 po->num = protocol;
1255 po->prot_hook.type = protocol;
1256 po->prot_hook.dev = dev;
1257
1258 po->ifindex = dev ? dev->ifindex : 0;
1259
1260 if (protocol == 0)
1261 goto out_unlock;
1262
1263 if (!dev || (dev->flags & IFF_UP)) {
1264 dev_add_pack(&po->prot_hook);
1265 sock_hold(sk);
1266 po->running = 1;
1267 } else {
1268 sk->sk_err = ENETDOWN;
1269 if (!sock_flag(sk, SOCK_DEAD))
1270 sk->sk_error_report(sk);
1271 }
1272
1273 out_unlock:
1274 spin_unlock(&po->bind_lock);
1275 release_sock(sk);
1276 return 0;
1277 }
1278
1279 /*
1280 * Bind a packet socket to a device
1281 */
1282
1283 static int packet_bind_spkt(struct socket *sock, struct sockaddr *uaddr,
1284 int addr_len)
1285 {
1286 struct sock *sk = sock->sk;
1287 char name[15];
1288 struct net_device *dev;
1289 int err = -ENODEV;
1290
1291 /*
1292 * Check legality
1293 */
1294
1295 if (addr_len != sizeof(struct sockaddr))
1296 return -EINVAL;
1297 strlcpy(name, uaddr->sa_data, sizeof(name));
1298
1299 dev = dev_get_by_name(sock_net(sk), name);
1300 if (dev) {
1301 err = packet_do_bind(sk, dev, pkt_sk(sk)->num);
1302 dev_put(dev);
1303 }
1304 return err;
1305 }
1306
1307 static int packet_bind(struct socket *sock, struct sockaddr *uaddr, int addr_len)
1308 {
1309 struct sockaddr_ll *sll = (struct sockaddr_ll *)uaddr;
1310 struct sock *sk = sock->sk;
1311 struct net_device *dev = NULL;
1312 int err;
1313
1314
1315 /*
1316 * Check legality
1317 */
1318
1319 if (addr_len < sizeof(struct sockaddr_ll))
1320 return -EINVAL;
1321 if (sll->sll_family != AF_PACKET)
1322 return -EINVAL;
1323
1324 if (sll->sll_ifindex) {
1325 err = -ENODEV;
1326 dev = dev_get_by_index(sock_net(sk), sll->sll_ifindex);
1327 if (dev == NULL)
1328 goto out;
1329 }
1330 err = packet_do_bind(sk, dev, sll->sll_protocol ? : pkt_sk(sk)->num);
1331 if (dev)
1332 dev_put(dev);
1333
1334 out:
1335 return err;
1336 }
1337
1338 static struct proto packet_proto = {
1339 .name = "PACKET",
1340 .owner = THIS_MODULE,
1341 .obj_size = sizeof(struct packet_sock),
1342 };
1343
1344 /*
1345 * Create a packet of type SOCK_PACKET.
1346 */
1347
1348 static int packet_create(struct net *net, struct socket *sock, int protocol,
1349 int kern)
1350 {
1351 struct sock *sk;
1352 struct packet_sock *po;
1353 __be16 proto = (__force __be16)protocol; /* weird, but documented */
1354 int err;
1355
1356 if (!capable(CAP_NET_RAW))
1357 return -EPERM;
1358 if (sock->type != SOCK_DGRAM && sock->type != SOCK_RAW &&
1359 sock->type != SOCK_PACKET)
1360 return -ESOCKTNOSUPPORT;
1361
1362 sock->state = SS_UNCONNECTED;
1363
1364 err = -ENOBUFS;
1365 sk = sk_alloc(net, PF_PACKET, GFP_KERNEL, &packet_proto);
1366 if (sk == NULL)
1367 goto out;
1368
1369 sock->ops = &packet_ops;
1370 if (sock->type == SOCK_PACKET)
1371 sock->ops = &packet_ops_spkt;
1372
1373 sock_init_data(sock, sk);
1374
1375 po = pkt_sk(sk);
1376 sk->sk_family = PF_PACKET;
1377 po->num = proto;
1378
1379 sk->sk_destruct = packet_sock_destruct;
1380 sk_refcnt_debug_inc(sk);
1381
1382 /*
1383 * Attach a protocol block
1384 */
1385
1386 spin_lock_init(&po->bind_lock);
1387 mutex_init(&po->pg_vec_lock);
1388 po->prot_hook.func = packet_rcv;
1389
1390 if (sock->type == SOCK_PACKET)
1391 po->prot_hook.func = packet_rcv_spkt;
1392
1393 po->prot_hook.af_packet_priv = sk;
1394
1395 if (proto) {
1396 po->prot_hook.type = proto;
1397 dev_add_pack(&po->prot_hook);
1398 sock_hold(sk);
1399 po->running = 1;
1400 }
1401
1402 write_lock_bh(&net->packet.sklist_lock);
1403 sk_add_node(sk, &net->packet.sklist);
1404 sock_prot_inuse_add(net, &packet_proto, 1);
1405 write_unlock_bh(&net->packet.sklist_lock);
1406 return 0;
1407 out:
1408 return err;
1409 }
1410
1411 /*
1412 * Pull a packet from our receive queue and hand it to the user.
1413 * If necessary we block.
1414 */
1415
1416 static int packet_recvmsg(struct kiocb *iocb, struct socket *sock,
1417 struct msghdr *msg, size_t len, int flags)
1418 {
1419 struct sock *sk = sock->sk;
1420 struct sk_buff *skb;
1421 int copied, err;
1422 struct sockaddr_ll *sll;
1423
1424 err = -EINVAL;
1425 if (flags & ~(MSG_PEEK|MSG_DONTWAIT|MSG_TRUNC|MSG_CMSG_COMPAT))
1426 goto out;
1427
1428 #if 0
1429 /* What error should we return now? EUNATTACH? */
1430 if (pkt_sk(sk)->ifindex < 0)
1431 return -ENODEV;
1432 #endif
1433
1434 /*
1435 * Call the generic datagram receiver. This handles all sorts
1436 * of horrible races and re-entrancy so we can forget about it
1437 * in the protocol layers.
1438 *
1439 * Now it will return ENETDOWN, if device have just gone down,
1440 * but then it will block.
1441 */
1442
1443 skb = skb_recv_datagram(sk, flags, flags & MSG_DONTWAIT, &err);
1444
1445 /*
1446 * An error occurred so return it. Because skb_recv_datagram()
1447 * handles the blocking we don't see and worry about blocking
1448 * retries.
1449 */
1450
1451 if (skb == NULL)
1452 goto out;
1453
1454 /*
1455 * If the address length field is there to be filled in, we fill
1456 * it in now.
1457 */
1458
1459 sll = &PACKET_SKB_CB(skb)->sa.ll;
1460 if (sock->type == SOCK_PACKET)
1461 msg->msg_namelen = sizeof(struct sockaddr_pkt);
1462 else
1463 msg->msg_namelen = sll->sll_halen + offsetof(struct sockaddr_ll, sll_addr);
1464
1465 /*
1466 * You lose any data beyond the buffer you gave. If it worries a
1467 * user program they can ask the device for its MTU anyway.
1468 */
1469
1470 copied = skb->len;
1471 if (copied > len) {
1472 copied = len;
1473 msg->msg_flags |= MSG_TRUNC;
1474 }
1475
1476 err = skb_copy_datagram_iovec(skb, 0, msg->msg_iov, copied);
1477 if (err)
1478 goto out_free;
1479
1480 sock_recv_ts_and_drops(msg, sk, skb);
1481
1482 if (msg->msg_name)
1483 memcpy(msg->msg_name, &PACKET_SKB_CB(skb)->sa,
1484 msg->msg_namelen);
1485
1486 if (pkt_sk(sk)->auxdata) {
1487 struct tpacket_auxdata aux;
1488
1489 aux.tp_status = TP_STATUS_USER;
1490 if (skb->ip_summed == CHECKSUM_PARTIAL)
1491 aux.tp_status |= TP_STATUS_CSUMNOTREADY;
1492 aux.tp_len = PACKET_SKB_CB(skb)->origlen;
1493 aux.tp_snaplen = skb->len;
1494 aux.tp_mac = 0;
1495 aux.tp_net = skb_network_offset(skb);
1496 aux.tp_vlan_tci = vlan_tx_tag_get(skb);
1497
1498 put_cmsg(msg, SOL_PACKET, PACKET_AUXDATA, sizeof(aux), &aux);
1499 }
1500
1501 /*
1502 * Free or return the buffer as appropriate. Again this
1503 * hides all the races and re-entrancy issues from us.
1504 */
1505 err = (flags&MSG_TRUNC) ? skb->len : copied;
1506
1507 out_free:
1508 skb_free_datagram(sk, skb);
1509 out:
1510 return err;
1511 }
1512
1513 static int packet_getname_spkt(struct socket *sock, struct sockaddr *uaddr,
1514 int *uaddr_len, int peer)
1515 {
1516 struct net_device *dev;
1517 struct sock *sk = sock->sk;
1518
1519 if (peer)
1520 return -EOPNOTSUPP;
1521
1522 uaddr->sa_family = AF_PACKET;
1523 rcu_read_lock();
1524 dev = dev_get_by_index_rcu(sock_net(sk), pkt_sk(sk)->ifindex);
1525 if (dev)
1526 strlcpy(uaddr->sa_data, dev->name, 15);
1527 else
1528 memset(uaddr->sa_data, 0, 14);
1529 rcu_read_unlock();
1530 *uaddr_len = sizeof(*uaddr);
1531
1532 return 0;
1533 }
1534
1535 static int packet_getname(struct socket *sock, struct sockaddr *uaddr,
1536 int *uaddr_len, int peer)
1537 {
1538 struct net_device *dev;
1539 struct sock *sk = sock->sk;
1540 struct packet_sock *po = pkt_sk(sk);
1541 DECLARE_SOCKADDR(struct sockaddr_ll *, sll, uaddr);
1542
1543 if (peer)
1544 return -EOPNOTSUPP;
1545
1546 sll->sll_family = AF_PACKET;
1547 sll->sll_ifindex = po->ifindex;
1548 sll->sll_protocol = po->num;
1549 rcu_read_lock();
1550 dev = dev_get_by_index_rcu(sock_net(sk), po->ifindex);
1551 if (dev) {
1552 sll->sll_hatype = dev->type;
1553 sll->sll_halen = dev->addr_len;
1554 memcpy(sll->sll_addr, dev->dev_addr, dev->addr_len);
1555 } else {
1556 sll->sll_hatype = 0; /* Bad: we have no ARPHRD_UNSPEC */
1557 sll->sll_halen = 0;
1558 }
1559 rcu_read_unlock();
1560 *uaddr_len = offsetof(struct sockaddr_ll, sll_addr) + sll->sll_halen;
1561
1562 return 0;
1563 }
1564
1565 static int packet_dev_mc(struct net_device *dev, struct packet_mclist *i,
1566 int what)
1567 {
1568 switch (i->type) {
1569 case PACKET_MR_MULTICAST:
1570 if (what > 0)
1571 return dev_mc_add(dev, i->addr, i->alen, 0);
1572 else
1573 return dev_mc_delete(dev, i->addr, i->alen, 0);
1574 break;
1575 case PACKET_MR_PROMISC:
1576 return dev_set_promiscuity(dev, what);
1577 break;
1578 case PACKET_MR_ALLMULTI:
1579 return dev_set_allmulti(dev, what);
1580 break;
1581 case PACKET_MR_UNICAST:
1582 if (what > 0)
1583 return dev_unicast_add(dev, i->addr);
1584 else
1585 return dev_unicast_delete(dev, i->addr);
1586 break;
1587 default:
1588 break;
1589 }
1590 return 0;
1591 }
1592
1593 static void packet_dev_mclist(struct net_device *dev, struct packet_mclist *i, int what)
1594 {
1595 for ( ; i; i = i->next) {
1596 if (i->ifindex == dev->ifindex)
1597 packet_dev_mc(dev, i, what);
1598 }
1599 }
1600
1601 static int packet_mc_add(struct sock *sk, struct packet_mreq_max *mreq)
1602 {
1603 struct packet_sock *po = pkt_sk(sk);
1604 struct packet_mclist *ml, *i;
1605 struct net_device *dev;
1606 int err;
1607
1608 rtnl_lock();
1609
1610 err = -ENODEV;
1611 dev = __dev_get_by_index(sock_net(sk), mreq->mr_ifindex);
1612 if (!dev)
1613 goto done;
1614
1615 err = -EINVAL;
1616 if (mreq->mr_alen > dev->addr_len)
1617 goto done;
1618
1619 err = -ENOBUFS;
1620 i = kmalloc(sizeof(*i), GFP_KERNEL);
1621 if (i == NULL)
1622 goto done;
1623
1624 err = 0;
1625 for (ml = po->mclist; ml; ml = ml->next) {
1626 if (ml->ifindex == mreq->mr_ifindex &&
1627 ml->type == mreq->mr_type &&
1628 ml->alen == mreq->mr_alen &&
1629 memcmp(ml->addr, mreq->mr_address, ml->alen) == 0) {
1630 ml->count++;
1631 /* Free the new element ... */
1632 kfree(i);
1633 goto done;
1634 }
1635 }
1636
1637 i->type = mreq->mr_type;
1638 i->ifindex = mreq->mr_ifindex;
1639 i->alen = mreq->mr_alen;
1640 memcpy(i->addr, mreq->mr_address, i->alen);
1641 i->count = 1;
1642 i->next = po->mclist;
1643 po->mclist = i;
1644 err = packet_dev_mc(dev, i, 1);
1645 if (err) {
1646 po->mclist = i->next;
1647 kfree(i);
1648 }
1649
1650 done:
1651 rtnl_unlock();
1652 return err;
1653 }
1654
1655 static int packet_mc_drop(struct sock *sk, struct packet_mreq_max *mreq)
1656 {
1657 struct packet_mclist *ml, **mlp;
1658
1659 rtnl_lock();
1660
1661 for (mlp = &pkt_sk(sk)->mclist; (ml = *mlp) != NULL; mlp = &ml->next) {
1662 if (ml->ifindex == mreq->mr_ifindex &&
1663 ml->type == mreq->mr_type &&
1664 ml->alen == mreq->mr_alen &&
1665 memcmp(ml->addr, mreq->mr_address, ml->alen) == 0) {
1666 if (--ml->count == 0) {
1667 struct net_device *dev;
1668 *mlp = ml->next;
1669 dev = __dev_get_by_index(sock_net(sk), ml->ifindex);
1670 if (dev)
1671 packet_dev_mc(dev, ml, -1);
1672 kfree(ml);
1673 }
1674 rtnl_unlock();
1675 return 0;
1676 }
1677 }
1678 rtnl_unlock();
1679 return -EADDRNOTAVAIL;
1680 }
1681
1682 static void packet_flush_mclist(struct sock *sk)
1683 {
1684 struct packet_sock *po = pkt_sk(sk);
1685 struct packet_mclist *ml;
1686
1687 if (!po->mclist)
1688 return;
1689
1690 rtnl_lock();
1691 while ((ml = po->mclist) != NULL) {
1692 struct net_device *dev;
1693
1694 po->mclist = ml->next;
1695 dev = __dev_get_by_index(sock_net(sk), ml->ifindex);
1696 if (dev != NULL)
1697 packet_dev_mc(dev, ml, -1);
1698 kfree(ml);
1699 }
1700 rtnl_unlock();
1701 }
1702
1703 static int
1704 packet_setsockopt(struct socket *sock, int level, int optname, char __user *optval, unsigned int optlen)
1705 {
1706 struct sock *sk = sock->sk;
1707 struct packet_sock *po = pkt_sk(sk);
1708 int ret;
1709
1710 if (level != SOL_PACKET)
1711 return -ENOPROTOOPT;
1712
1713 switch (optname) {
1714 case PACKET_ADD_MEMBERSHIP:
1715 case PACKET_DROP_MEMBERSHIP:
1716 {
1717 struct packet_mreq_max mreq;
1718 int len = optlen;
1719 memset(&mreq, 0, sizeof(mreq));
1720 if (len < sizeof(struct packet_mreq))
1721 return -EINVAL;
1722 if (len > sizeof(mreq))
1723 len = sizeof(mreq);
1724 if (copy_from_user(&mreq, optval, len))
1725 return -EFAULT;
1726 if (len < (mreq.mr_alen + offsetof(struct packet_mreq, mr_address)))
1727 return -EINVAL;
1728 if (optname == PACKET_ADD_MEMBERSHIP)
1729 ret = packet_mc_add(sk, &mreq);
1730 else
1731 ret = packet_mc_drop(sk, &mreq);
1732 return ret;
1733 }
1734
1735 #ifdef CONFIG_PACKET_MMAP
1736 case PACKET_RX_RING:
1737 case PACKET_TX_RING:
1738 {
1739 struct tpacket_req req;
1740
1741 if (optlen < sizeof(req))
1742 return -EINVAL;
1743 if (copy_from_user(&req, optval, sizeof(req)))
1744 return -EFAULT;
1745 return packet_set_ring(sk, &req, 0, optname == PACKET_TX_RING);
1746 }
1747 case PACKET_COPY_THRESH:
1748 {
1749 int val;
1750
1751 if (optlen != sizeof(val))
1752 return -EINVAL;
1753 if (copy_from_user(&val, optval, sizeof(val)))
1754 return -EFAULT;
1755
1756 pkt_sk(sk)->copy_thresh = val;
1757 return 0;
1758 }
1759 case PACKET_VERSION:
1760 {
1761 int val;
1762
1763 if (optlen != sizeof(val))
1764 return -EINVAL;
1765 if (po->rx_ring.pg_vec || po->tx_ring.pg_vec)
1766 return -EBUSY;
1767 if (copy_from_user(&val, optval, sizeof(val)))
1768 return -EFAULT;
1769 switch (val) {
1770 case TPACKET_V1:
1771 case TPACKET_V2:
1772 po->tp_version = val;
1773 return 0;
1774 default:
1775 return -EINVAL;
1776 }
1777 }
1778 case PACKET_RESERVE:
1779 {
1780 unsigned int val;
1781
1782 if (optlen != sizeof(val))
1783 return -EINVAL;
1784 if (po->rx_ring.pg_vec || po->tx_ring.pg_vec)
1785 return -EBUSY;
1786 if (copy_from_user(&val, optval, sizeof(val)))
1787 return -EFAULT;
1788 po->tp_reserve = val;
1789 return 0;
1790 }
1791 case PACKET_LOSS:
1792 {
1793 unsigned int val;
1794
1795 if (optlen != sizeof(val))
1796 return -EINVAL;
1797 if (po->rx_ring.pg_vec || po->tx_ring.pg_vec)
1798 return -EBUSY;
1799 if (copy_from_user(&val, optval, sizeof(val)))
1800 return -EFAULT;
1801 po->tp_loss = !!val;
1802 return 0;
1803 }
1804 #endif
1805 case PACKET_AUXDATA:
1806 {
1807 int val;
1808
1809 if (optlen < sizeof(val))
1810 return -EINVAL;
1811 if (copy_from_user(&val, optval, sizeof(val)))
1812 return -EFAULT;
1813
1814 po->auxdata = !!val;
1815 return 0;
1816 }
1817 case PACKET_ORIGDEV:
1818 {
1819 int val;
1820
1821 if (optlen < sizeof(val))
1822 return -EINVAL;
1823 if (copy_from_user(&val, optval, sizeof(val)))
1824 return -EFAULT;
1825
1826 po->origdev = !!val;
1827 return 0;
1828 }
1829 default:
1830 return -ENOPROTOOPT;
1831 }
1832 }
1833
1834 static int packet_getsockopt(struct socket *sock, int level, int optname,
1835 char __user *optval, int __user *optlen)
1836 {
1837 int len;
1838 int val;
1839 struct sock *sk = sock->sk;
1840 struct packet_sock *po = pkt_sk(sk);
1841 void *data;
1842 struct tpacket_stats st;
1843
1844 if (level != SOL_PACKET)
1845 return -ENOPROTOOPT;
1846
1847 if (get_user(len, optlen))
1848 return -EFAULT;
1849
1850 if (len < 0)
1851 return -EINVAL;
1852
1853 switch (optname) {
1854 case PACKET_STATISTICS:
1855 if (len > sizeof(struct tpacket_stats))
1856 len = sizeof(struct tpacket_stats);
1857 spin_lock_bh(&sk->sk_receive_queue.lock);
1858 st = po->stats;
1859 memset(&po->stats, 0, sizeof(st));
1860 spin_unlock_bh(&sk->sk_receive_queue.lock);
1861 st.tp_packets += st.tp_drops;
1862
1863 data = &st;
1864 break;
1865 case PACKET_AUXDATA:
1866 if (len > sizeof(int))
1867 len = sizeof(int);
1868 val = po->auxdata;
1869
1870 data = &val;
1871 break;
1872 case PACKET_ORIGDEV:
1873 if (len > sizeof(int))
1874 len = sizeof(int);
1875 val = po->origdev;
1876
1877 data = &val;
1878 break;
1879 #ifdef CONFIG_PACKET_MMAP
1880 case PACKET_VERSION:
1881 if (len > sizeof(int))
1882 len = sizeof(int);
1883 val = po->tp_version;
1884 data = &val;
1885 break;
1886 case PACKET_HDRLEN:
1887 if (len > sizeof(int))
1888 len = sizeof(int);
1889 if (copy_from_user(&val, optval, len))
1890 return -EFAULT;
1891 switch (val) {
1892 case TPACKET_V1:
1893 val = sizeof(struct tpacket_hdr);
1894 break;
1895 case TPACKET_V2:
1896 val = sizeof(struct tpacket2_hdr);
1897 break;
1898 default:
1899 return -EINVAL;
1900 }
1901 data = &val;
1902 break;
1903 case PACKET_RESERVE:
1904 if (len > sizeof(unsigned int))
1905 len = sizeof(unsigned int);
1906 val = po->tp_reserve;
1907 data = &val;
1908 break;
1909 case PACKET_LOSS:
1910 if (len > sizeof(unsigned int))
1911 len = sizeof(unsigned int);
1912 val = po->tp_loss;
1913 data = &val;
1914 break;
1915 #endif
1916 default:
1917 return -ENOPROTOOPT;
1918 }
1919
1920 if (put_user(len, optlen))
1921 return -EFAULT;
1922 if (copy_to_user(optval, data, len))
1923 return -EFAULT;
1924 return 0;
1925 }
1926
1927
1928 static int packet_notifier(struct notifier_block *this, unsigned long msg, void *data)
1929 {
1930 struct sock *sk;
1931 struct hlist_node *node;
1932 struct net_device *dev = data;
1933 struct net *net = dev_net(dev);
1934
1935 read_lock(&net->packet.sklist_lock);
1936 sk_for_each(sk, node, &net->packet.sklist) {
1937 struct packet_sock *po = pkt_sk(sk);
1938
1939 switch (msg) {
1940 case NETDEV_UNREGISTER:
1941 if (po->mclist)
1942 packet_dev_mclist(dev, po->mclist, -1);
1943 /* fallthrough */
1944
1945 case NETDEV_DOWN:
1946 if (dev->ifindex == po->ifindex) {
1947 spin_lock(&po->bind_lock);
1948 if (po->running) {
1949 __dev_remove_pack(&po->prot_hook);
1950 __sock_put(sk);
1951 po->running = 0;
1952 sk->sk_err = ENETDOWN;
1953 if (!sock_flag(sk, SOCK_DEAD))
1954 sk->sk_error_report(sk);
1955 }
1956 if (msg == NETDEV_UNREGISTER) {
1957 po->ifindex = -1;
1958 po->prot_hook.dev = NULL;
1959 }
1960 spin_unlock(&po->bind_lock);
1961 }
1962 break;
1963 case NETDEV_UP:
1964 spin_lock(&po->bind_lock);
1965 if (dev->ifindex == po->ifindex && po->num &&
1966 !po->running) {
1967 dev_add_pack(&po->prot_hook);
1968 sock_hold(sk);
1969 po->running = 1;
1970 }
1971 spin_unlock(&po->bind_lock);
1972 break;
1973 }
1974 }
1975 read_unlock(&net->packet.sklist_lock);
1976 return NOTIFY_DONE;
1977 }
1978
1979
1980 static int packet_ioctl(struct socket *sock, unsigned int cmd,
1981 unsigned long arg)
1982 {
1983 struct sock *sk = sock->sk;
1984
1985 switch (cmd) {
1986 case SIOCOUTQ:
1987 {
1988 int amount = sk_wmem_alloc_get(sk);
1989
1990 return put_user(amount, (int __user *)arg);
1991 }
1992 case SIOCINQ:
1993 {
1994 struct sk_buff *skb;
1995 int amount = 0;
1996
1997 spin_lock_bh(&sk->sk_receive_queue.lock);
1998 skb = skb_peek(&sk->sk_receive_queue);
1999 if (skb)
2000 amount = skb->len;
2001 spin_unlock_bh(&sk->sk_receive_queue.lock);
2002 return put_user(amount, (int __user *)arg);
2003 }
2004 case SIOCGSTAMP:
2005 return sock_get_timestamp(sk, (struct timeval __user *)arg);
2006 case SIOCGSTAMPNS:
2007 return sock_get_timestampns(sk, (struct timespec __user *)arg);
2008
2009 #ifdef CONFIG_INET
2010 case SIOCADDRT:
2011 case SIOCDELRT:
2012 case SIOCDARP:
2013 case SIOCGARP:
2014 case SIOCSARP:
2015 case SIOCGIFADDR:
2016 case SIOCSIFADDR:
2017 case SIOCGIFBRDADDR:
2018 case SIOCSIFBRDADDR:
2019 case SIOCGIFNETMASK:
2020 case SIOCSIFNETMASK:
2021 case SIOCGIFDSTADDR:
2022 case SIOCSIFDSTADDR:
2023 case SIOCSIFFLAGS:
2024 if (!net_eq(sock_net(sk), &init_net))
2025 return -ENOIOCTLCMD;
2026 return inet_dgram_ops.ioctl(sock, cmd, arg);
2027 #endif
2028
2029 default:
2030 return -ENOIOCTLCMD;
2031 }
2032 return 0;
2033 }
2034
2035 #ifndef CONFIG_PACKET_MMAP
2036 #define packet_mmap sock_no_mmap
2037 #define packet_poll datagram_poll
2038 #else
2039
2040 static unsigned int packet_poll(struct file *file, struct socket *sock,
2041 poll_table *wait)
2042 {
2043 struct sock *sk = sock->sk;
2044 struct packet_sock *po = pkt_sk(sk);
2045 unsigned int mask = datagram_poll(file, sock, wait);
2046
2047 spin_lock_bh(&sk->sk_receive_queue.lock);
2048 if (po->rx_ring.pg_vec) {
2049 if (!packet_previous_frame(po, &po->rx_ring, TP_STATUS_KERNEL))
2050 mask |= POLLIN | POLLRDNORM;
2051 }
2052 spin_unlock_bh(&sk->sk_receive_queue.lock);
2053 spin_lock_bh(&sk->sk_write_queue.lock);
2054 if (po->tx_ring.pg_vec) {
2055 if (packet_current_frame(po, &po->tx_ring, TP_STATUS_AVAILABLE))
2056 mask |= POLLOUT | POLLWRNORM;
2057 }
2058 spin_unlock_bh(&sk->sk_write_queue.lock);
2059 return mask;
2060 }
2061
2062
2063 /* Dirty? Well, I still did not learn better way to account
2064 * for user mmaps.
2065 */
2066
2067 static void packet_mm_open(struct vm_area_struct *vma)
2068 {
2069 struct file *file = vma->vm_file;
2070 struct socket *sock = file->private_data;
2071 struct sock *sk = sock->sk;
2072
2073 if (sk)
2074 atomic_inc(&pkt_sk(sk)->mapped);
2075 }
2076
2077 static void packet_mm_close(struct vm_area_struct *vma)
2078 {
2079 struct file *file = vma->vm_file;
2080 struct socket *sock = file->private_data;
2081 struct sock *sk = sock->sk;
2082
2083 if (sk)
2084 atomic_dec(&pkt_sk(sk)->mapped);
2085 }
2086
2087 static const struct vm_operations_struct packet_mmap_ops = {
2088 .open = packet_mm_open,
2089 .close = packet_mm_close,
2090 };
2091
2092 static void free_pg_vec(char **pg_vec, unsigned int order, unsigned int len)
2093 {
2094 int i;
2095
2096 for (i = 0; i < len; i++) {
2097 if (likely(pg_vec[i]))
2098 free_pages((unsigned long) pg_vec[i], order);
2099 }
2100 kfree(pg_vec);
2101 }
2102
2103 static inline char *alloc_one_pg_vec_page(unsigned long order)
2104 {
2105 gfp_t gfp_flags = GFP_KERNEL | __GFP_COMP | __GFP_ZERO | __GFP_NOWARN;
2106
2107 return (char *) __get_free_pages(gfp_flags, order);
2108 }
2109
2110 static char **alloc_pg_vec(struct tpacket_req *req, int order)
2111 {
2112 unsigned int block_nr = req->tp_block_nr;
2113 char **pg_vec;
2114 int i;
2115
2116 pg_vec = kzalloc(block_nr * sizeof(char *), GFP_KERNEL);
2117 if (unlikely(!pg_vec))
2118 goto out;
2119
2120 for (i = 0; i < block_nr; i++) {
2121 pg_vec[i] = alloc_one_pg_vec_page(order);
2122 if (unlikely(!pg_vec[i]))
2123 goto out_free_pgvec;
2124 }
2125
2126 out:
2127 return pg_vec;
2128
2129 out_free_pgvec:
2130 free_pg_vec(pg_vec, order, block_nr);
2131 pg_vec = NULL;
2132 goto out;
2133 }
2134
2135 static int packet_set_ring(struct sock *sk, struct tpacket_req *req,
2136 int closing, int tx_ring)
2137 {
2138 char **pg_vec = NULL;
2139 struct packet_sock *po = pkt_sk(sk);
2140 int was_running, order = 0;
2141 struct packet_ring_buffer *rb;
2142 struct sk_buff_head *rb_queue;
2143 __be16 num;
2144 int err;
2145
2146 rb = tx_ring ? &po->tx_ring : &po->rx_ring;
2147 rb_queue = tx_ring ? &sk->sk_write_queue : &sk->sk_receive_queue;
2148
2149 err = -EBUSY;
2150 if (!closing) {
2151 if (atomic_read(&po->mapped))
2152 goto out;
2153 if (atomic_read(&rb->pending))
2154 goto out;
2155 }
2156
2157 if (req->tp_block_nr) {
2158 /* Sanity tests and some calculations */
2159 err = -EBUSY;
2160 if (unlikely(rb->pg_vec))
2161 goto out;
2162
2163 switch (po->tp_version) {
2164 case TPACKET_V1:
2165 po->tp_hdrlen = TPACKET_HDRLEN;
2166 break;
2167 case TPACKET_V2:
2168 po->tp_hdrlen = TPACKET2_HDRLEN;
2169 break;
2170 }
2171
2172 err = -EINVAL;
2173 if (unlikely((int)req->tp_block_size <= 0))
2174 goto out;
2175 if (unlikely(req->tp_block_size & (PAGE_SIZE - 1)))
2176 goto out;
2177 if (unlikely(req->tp_frame_size < po->tp_hdrlen +
2178 po->tp_reserve))
2179 goto out;
2180 if (unlikely(req->tp_frame_size & (TPACKET_ALIGNMENT - 1)))
2181 goto out;
2182
2183 rb->frames_per_block = req->tp_block_size/req->tp_frame_size;
2184 if (unlikely(rb->frames_per_block <= 0))
2185 goto out;
2186 if (unlikely((rb->frames_per_block * req->tp_block_nr) !=
2187 req->tp_frame_nr))
2188 goto out;
2189
2190 err = -ENOMEM;
2191 order = get_order(req->tp_block_size);
2192 pg_vec = alloc_pg_vec(req, order);
2193 if (unlikely(!pg_vec))
2194 goto out;
2195 }
2196 /* Done */
2197 else {
2198 err = -EINVAL;
2199 if (unlikely(req->tp_frame_nr))
2200 goto out;
2201 }
2202
2203 lock_sock(sk);
2204
2205 /* Detach socket from network */
2206 spin_lock(&po->bind_lock);
2207 was_running = po->running;
2208 num = po->num;
2209 if (was_running) {
2210 __dev_remove_pack(&po->prot_hook);
2211 po->num = 0;
2212 po->running = 0;
2213 __sock_put(sk);
2214 }
2215 spin_unlock(&po->bind_lock);
2216
2217 synchronize_net();
2218
2219 err = -EBUSY;
2220 mutex_lock(&po->pg_vec_lock);
2221 if (closing || atomic_read(&po->mapped) == 0) {
2222 err = 0;
2223 #define XC(a, b) ({ __typeof__ ((a)) __t; __t = (a); (a) = (b); __t; })
2224 spin_lock_bh(&rb_queue->lock);
2225 pg_vec = XC(rb->pg_vec, pg_vec);
2226 rb->frame_max = (req->tp_frame_nr - 1);
2227 rb->head = 0;
2228 rb->frame_size = req->tp_frame_size;
2229 spin_unlock_bh(&rb_queue->lock);
2230
2231 order = XC(rb->pg_vec_order, order);
2232 req->tp_block_nr = XC(rb->pg_vec_len, req->tp_block_nr);
2233
2234 rb->pg_vec_pages = req->tp_block_size/PAGE_SIZE;
2235 po->prot_hook.func = (po->rx_ring.pg_vec) ?
2236 tpacket_rcv : packet_rcv;
2237 skb_queue_purge(rb_queue);
2238 #undef XC
2239 if (atomic_read(&po->mapped))
2240 pr_err("packet_mmap: vma is busy: %d\n",
2241 atomic_read(&po->mapped));
2242 }
2243 mutex_unlock(&po->pg_vec_lock);
2244
2245 spin_lock(&po->bind_lock);
2246 if (was_running && !po->running) {
2247 sock_hold(sk);
2248 po->running = 1;
2249 po->num = num;
2250 dev_add_pack(&po->prot_hook);
2251 }
2252 spin_unlock(&po->bind_lock);
2253
2254 release_sock(sk);
2255
2256 if (pg_vec)
2257 free_pg_vec(pg_vec, order, req->tp_block_nr);
2258 out:
2259 return err;
2260 }
2261
2262 static int packet_mmap(struct file *file, struct socket *sock,
2263 struct vm_area_struct *vma)
2264 {
2265 struct sock *sk = sock->sk;
2266 struct packet_sock *po = pkt_sk(sk);
2267 unsigned long size, expected_size;
2268 struct packet_ring_buffer *rb;
2269 unsigned long start;
2270 int err = -EINVAL;
2271 int i;
2272
2273 if (vma->vm_pgoff)
2274 return -EINVAL;
2275
2276 mutex_lock(&po->pg_vec_lock);
2277
2278 expected_size = 0;
2279 for (rb = &po->rx_ring; rb <= &po->tx_ring; rb++) {
2280 if (rb->pg_vec) {
2281 expected_size += rb->pg_vec_len
2282 * rb->pg_vec_pages
2283 * PAGE_SIZE;
2284 }
2285 }
2286
2287 if (expected_size == 0)
2288 goto out;
2289
2290 size = vma->vm_end - vma->vm_start;
2291 if (size != expected_size)
2292 goto out;
2293
2294 start = vma->vm_start;
2295 for (rb = &po->rx_ring; rb <= &po->tx_ring; rb++) {
2296 if (rb->pg_vec == NULL)
2297 continue;
2298
2299 for (i = 0; i < rb->pg_vec_len; i++) {
2300 struct page *page = virt_to_page(rb->pg_vec[i]);
2301 int pg_num;
2302
2303 for (pg_num = 0; pg_num < rb->pg_vec_pages;
2304 pg_num++, page++) {
2305 err = vm_insert_page(vma, start, page);
2306 if (unlikely(err))
2307 goto out;
2308 start += PAGE_SIZE;
2309 }
2310 }
2311 }
2312
2313 atomic_inc(&po->mapped);
2314 vma->vm_ops = &packet_mmap_ops;
2315 err = 0;
2316
2317 out:
2318 mutex_unlock(&po->pg_vec_lock);
2319 return err;
2320 }
2321 #endif
2322
2323
2324 static const struct proto_ops packet_ops_spkt = {
2325 .family = PF_PACKET,
2326 .owner = THIS_MODULE,
2327 .release = packet_release,
2328 .bind = packet_bind_spkt,
2329 .connect = sock_no_connect,
2330 .socketpair = sock_no_socketpair,
2331 .accept = sock_no_accept,
2332 .getname = packet_getname_spkt,
2333 .poll = datagram_poll,
2334 .ioctl = packet_ioctl,
2335 .listen = sock_no_listen,
2336 .shutdown = sock_no_shutdown,
2337 .setsockopt = sock_no_setsockopt,
2338 .getsockopt = sock_no_getsockopt,
2339 .sendmsg = packet_sendmsg_spkt,
2340 .recvmsg = packet_recvmsg,
2341 .mmap = sock_no_mmap,
2342 .sendpage = sock_no_sendpage,
2343 };
2344
2345 static const struct proto_ops packet_ops = {
2346 .family = PF_PACKET,
2347 .owner = THIS_MODULE,
2348 .release = packet_release,
2349 .bind = packet_bind,
2350 .connect = sock_no_connect,
2351 .socketpair = sock_no_socketpair,
2352 .accept = sock_no_accept,
2353 .getname = packet_getname,
2354 .poll = packet_poll,
2355 .ioctl = packet_ioctl,
2356 .listen = sock_no_listen,
2357 .shutdown = sock_no_shutdown,
2358 .setsockopt = packet_setsockopt,
2359 .getsockopt = packet_getsockopt,
2360 .sendmsg = packet_sendmsg,
2361 .recvmsg = packet_recvmsg,
2362 .mmap = packet_mmap,
2363 .sendpage = sock_no_sendpage,
2364 };
2365
2366 static const struct net_proto_family packet_family_ops = {
2367 .family = PF_PACKET,
2368 .create = packet_create,
2369 .owner = THIS_MODULE,
2370 };
2371
2372 static struct notifier_block packet_netdev_notifier = {
2373 .notifier_call = packet_notifier,
2374 };
2375
2376 #ifdef CONFIG_PROC_FS
2377 static inline struct sock *packet_seq_idx(struct net *net, loff_t off)
2378 {
2379 struct sock *s;
2380 struct hlist_node *node;
2381
2382 sk_for_each(s, node, &net->packet.sklist) {
2383 if (!off--)
2384 return s;
2385 }
2386 return NULL;
2387 }
2388
2389 static void *packet_seq_start(struct seq_file *seq, loff_t *pos)
2390 __acquires(seq_file_net(seq)->packet.sklist_lock)
2391 {
2392 struct net *net = seq_file_net(seq);
2393 read_lock(&net->packet.sklist_lock);
2394 return *pos ? packet_seq_idx(net, *pos - 1) : SEQ_START_TOKEN;
2395 }
2396
2397 static void *packet_seq_next(struct seq_file *seq, void *v, loff_t *pos)
2398 {
2399 struct net *net = seq_file_net(seq);
2400 ++*pos;
2401 return (v == SEQ_START_TOKEN)
2402 ? sk_head(&net->packet.sklist)
2403 : sk_next((struct sock *)v) ;
2404 }
2405
2406 static void packet_seq_stop(struct seq_file *seq, void *v)
2407 __releases(seq_file_net(seq)->packet.sklist_lock)
2408 {
2409 struct net *net = seq_file_net(seq);
2410 read_unlock(&net->packet.sklist_lock);
2411 }
2412
2413 static int packet_seq_show(struct seq_file *seq, void *v)
2414 {
2415 if (v == SEQ_START_TOKEN)
2416 seq_puts(seq, "sk RefCnt Type Proto Iface R Rmem User Inode\n");
2417 else {
2418 struct sock *s = v;
2419 const struct packet_sock *po = pkt_sk(s);
2420
2421 seq_printf(seq,
2422 "%p %-6d %-4d %04x %-5d %1d %-6u %-6u %-6lu\n",
2423 s,
2424 atomic_read(&s->sk_refcnt),
2425 s->sk_type,
2426 ntohs(po->num),
2427 po->ifindex,
2428 po->running,
2429 atomic_read(&s->sk_rmem_alloc),
2430 sock_i_uid(s),
2431 sock_i_ino(s));
2432 }
2433
2434 return 0;
2435 }
2436
2437 static const struct seq_operations packet_seq_ops = {
2438 .start = packet_seq_start,
2439 .next = packet_seq_next,
2440 .stop = packet_seq_stop,
2441 .show = packet_seq_show,
2442 };
2443
2444 static int packet_seq_open(struct inode *inode, struct file *file)
2445 {
2446 return seq_open_net(inode, file, &packet_seq_ops,
2447 sizeof(struct seq_net_private));
2448 }
2449
2450 static const struct file_operations packet_seq_fops = {
2451 .owner = THIS_MODULE,
2452 .open = packet_seq_open,
2453 .read = seq_read,
2454 .llseek = seq_lseek,
2455 .release = seq_release_net,
2456 };
2457
2458 #endif
2459
2460 static int packet_net_init(struct net *net)
2461 {
2462 rwlock_init(&net->packet.sklist_lock);
2463 INIT_HLIST_HEAD(&net->packet.sklist);
2464
2465 if (!proc_net_fops_create(net, "packet", 0, &packet_seq_fops))
2466 return -ENOMEM;
2467
2468 return 0;
2469 }
2470
2471 static void packet_net_exit(struct net *net)
2472 {
2473 proc_net_remove(net, "packet");
2474 }
2475
2476 static struct pernet_operations packet_net_ops = {
2477 .init = packet_net_init,
2478 .exit = packet_net_exit,
2479 };
2480
2481
2482 static void __exit packet_exit(void)
2483 {
2484 unregister_netdevice_notifier(&packet_netdev_notifier);
2485 unregister_pernet_subsys(&packet_net_ops);
2486 sock_unregister(PF_PACKET);
2487 proto_unregister(&packet_proto);
2488 }
2489
2490 static int __init packet_init(void)
2491 {
2492 int rc = proto_register(&packet_proto, 0);
2493
2494 if (rc != 0)
2495 goto out;
2496
2497 sock_register(&packet_family_ops);
2498 register_pernet_subsys(&packet_net_ops);
2499 register_netdevice_notifier(&packet_netdev_notifier);
2500 out:
2501 return rc;
2502 }
2503
2504 module_init(packet_init);
2505 module_exit(packet_exit);
2506 MODULE_LICENSE("GPL");
2507 MODULE_ALIAS_NETPROTO(PF_PACKET);
Ubuntu Jammy Linux kernel mirror