2 * Copyright (c) 2003, 2004 Jeffrey M. Hsu. All rights reserved.
3 * Copyright (c) 2003, 2004 The DragonFly Project. All rights reserved.
5 * This code is derived from software contributed to The DragonFly Project
8 * Redistribution and use in source and binary forms, with or without
9 * modification, are permitted provided that the following conditions
11 * 1. Redistributions of source code must retain the above copyright
12 * notice, this list of conditions and the following disclaimer.
13 * 2. Redistributions in binary form must reproduce the above copyright
14 * notice, this list of conditions and the following disclaimer in the
15 * documentation and/or other materials provided with the distribution.
16 * 3. Neither the name of The DragonFly Project nor the names of its
17 * contributors may be used to endorse or promote products derived
18 * from this software without specific, prior written permission.
20 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
21 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
22 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
23 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
24 * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
25 * INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING,
26 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
27 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
28 * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
29 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
30 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
33 * $DragonFly: src/sys/netinet/ip_demux.c,v 1.45 2008/11/11 10:46:58 sephe Exp $
38 #include <sys/param.h>
39 #include <sys/systm.h>
40 #include <sys/kernel.h>
41 #include <sys/socket.h>
42 #include <sys/socketvar.h>
43 #include <sys/thread.h>
44 #include <sys/sysctl.h>
45 #include <sys/globaldata.h>
48 #include <net/netisr.h>
50 #include <netinet/in_systm.h>
51 #include <netinet/in.h>
52 #include <netinet/in_var.h>
53 #include <netinet/in_pcb.h>
54 #include <netinet/ip.h>
55 #include <netinet/ip_var.h>
56 #include <netinet/tcp.h>
57 #include <netinet/tcpip.h>
58 #include <netinet/tcp_var.h>
59 #include <netinet/udp.h>
60 #include <netinet/udp_var.h>
62 extern struct thread netisr_cpu
[];
63 extern int udp_mpsafe_thread
;
65 static struct thread tcp_thread
[MAXCPU
];
66 static struct thread udp_thread
[MAXCPU
];
69 INP_MPORT_HASH(in_addr_t faddr
, in_addr_t laddr
,
70 in_port_t fport
, in_port_t lport
)
73 * Use low order bytes.
76 #if (BYTE_ORDER == LITTLE_ENDIAN)
77 KASSERT(ncpus2
< 256, ("need different hash function")); /* XXX JH */
78 return (((faddr
>> 24) ^ (fport
>> 8) ^ (laddr
>> 24) ^ (lport
>> 8)) &
81 return ((faddr
^ fport
^ laddr
^ lport
) & ncpus2_mask
);
86 * If the packet is a valid IP datagram, upon returning of this function
87 * following things are promised:
89 * o IP header (including any possible IP options) is in one mbuf (m_len).
90 * o IP header length is not less than the minimum (sizeof(struct ip)).
91 * o IP total length is not less than IP header length.
92 * o IP datagram resides completely in the mbuf chain,
93 * i.e. pkthdr.len >= IP total length.
95 * If the packet is a UDP datagram,
96 * o IP header (including any possible IP options) and UDP header are in
98 * o IP total length is not less than (IP header length + UDP header length).
100 * If the packet is a TCP segment,
101 * o IP header (including any possible IP options) and TCP header (including
102 * any possible TCP options) are in one mbuf (m_len).
103 * o TCP header length is not less than the minimum (sizeof(struct tcphdr)).
104 * o IP total length is not less than (IP header length + TCP header length).
107 ip_lengthcheck(struct mbuf
**mp
)
109 struct mbuf
*m
= *mp
;
113 int thoff
; /* TCP data offset */
115 /* The packet must be at least the size of an IP header. */
116 if (m
->m_pkthdr
.len
< sizeof(struct ip
)) {
117 ipstat
.ips_tooshort
++;
121 /* The fixed IP header must reside completely in the first mbuf. */
122 if (m
->m_len
< sizeof(struct ip
)) {
123 m
= m_pullup(m
, sizeof(struct ip
));
125 ipstat
.ips_toosmall
++;
130 ip
= mtod(m
, struct ip
*);
132 /* Bound check the packet's stated IP header length. */
133 iphlen
= ip
->ip_hl
<< 2;
134 if (iphlen
< sizeof(struct ip
)) { /* minimum header length */
135 ipstat
.ips_badhlen
++;
139 /* The full IP header must reside completely in the one mbuf. */
140 if (m
->m_len
< iphlen
) {
141 m
= m_pullup(m
, iphlen
);
143 ipstat
.ips_badhlen
++;
146 ip
= mtod(m
, struct ip
*);
149 iplen
= ntohs(ip
->ip_len
);
152 * Check that the amount of data in the buffers is as
153 * at least much as the IP header would have us expect.
155 if (m
->m_pkthdr
.len
< iplen
) {
156 ipstat
.ips_tooshort
++;
161 * Fragments other than the first fragment don't have much
162 * length information.
164 if (ntohs(ip
->ip_off
) & IP_OFFMASK
)
168 * The TCP/IP or UDP/IP header must be entirely contained within
169 * the first fragment of a packet. Packet filters will break if they
172 * Since the packet will be trimmed to ip_len we must also make sure
173 * the potentially trimmed down length is still sufficient to hold
178 if (iplen
< iphlen
+ sizeof(struct tcphdr
)) {
179 ++tcpstat
.tcps_rcvshort
;
182 if (m
->m_len
< iphlen
+ sizeof(struct tcphdr
)) {
183 m
= m_pullup(m
, iphlen
+ sizeof(struct tcphdr
));
185 tcpstat
.tcps_rcvshort
++;
188 ip
= mtod(m
, struct ip
*);
190 th
= (struct tcphdr
*)((caddr_t
)ip
+ iphlen
);
191 thoff
= th
->th_off
<< 2;
192 if (thoff
< sizeof(struct tcphdr
) ||
193 thoff
+ iphlen
> ntohs(ip
->ip_len
)) {
194 tcpstat
.tcps_rcvbadoff
++;
197 if (m
->m_len
< iphlen
+ thoff
) {
198 m
= m_pullup(m
, iphlen
+ thoff
);
200 tcpstat
.tcps_rcvshort
++;
206 if (iplen
< iphlen
+ sizeof(struct udphdr
)) {
207 ++udpstat
.udps_hdrops
;
210 if (m
->m_len
< iphlen
+ sizeof(struct udphdr
)) {
211 m
= m_pullup(m
, iphlen
+ sizeof(struct udphdr
));
213 udpstat
.udps_hdrops
++;
220 if (iplen
< iphlen
) {
238 * Map a packet to a protocol processing thread and return the thread's port.
239 * If an error occurs, the passed mbuf will be freed, *mptr will be set
240 * to NULL, and NULL will be returned. If no error occurs, the passed mbuf
241 * may be modified and a port pointer will be returned.
244 ip_mport(struct mbuf
**mptr
, int dir
)
251 int thoff
; /* TCP data offset */
255 if (!ip_lengthcheck(mptr
))
259 ip
= mtod(m
, struct ip
*);
260 iphlen
= ip
->ip_hl
<< 2;
263 * XXX generic packet handling defrag on CPU 0 for now.
265 if (ntohs(ip
->ip_off
) & (IP_MF
| IP_OFFMASK
))
266 return (&netisr_cpu
[0].td_msgport
);
270 th
= (struct tcphdr
*)((caddr_t
)ip
+ iphlen
);
271 thoff
= th
->th_off
<< 2;
272 cpu
= INP_MPORT_HASH(ip
->ip_src
.s_addr
, ip
->ip_dst
.s_addr
,
273 th
->th_sport
, th
->th_dport
);
274 port
= &tcp_thread
[cpu
].td_msgport
;
277 uh
= (struct udphdr
*)((caddr_t
)ip
+ iphlen
);
279 if (IN_MULTICAST(ntohl(ip
->ip_dst
.s_addr
)) ||
280 (dir
== IP_MPORT_IN
&&
281 in_broadcast(ip
->ip_dst
, m
->m_pkthdr
.rcvif
))) {
284 cpu
= INP_MPORT_HASH(ip
->ip_src
.s_addr
,
285 ip
->ip_dst
.s_addr
, uh
->uh_sport
, uh
->uh_dport
);
287 port
= &udp_thread
[cpu
].td_msgport
;
290 port
= &netisr_cpu
[0].td_msgport
;
298 ip_mport_in(struct mbuf
**mptr
)
300 return ip_mport(mptr
, IP_MPORT_IN
);
304 * Map a TCP socket to a protocol processing thread.
307 tcp_soport(struct socket
*so
, struct sockaddr
*nam __unused
,
308 struct mbuf
**dummy __unused
, int req
)
312 /* The following processing all take place on Protocol Thread 0. */
313 if (req
== PRU_BIND
|| req
== PRU_CONNECT
|| req
== PRU_ATTACH
||
315 return (&tcp_thread
[0].td_msgport
);
318 if (!inp
) /* connection reset by peer */
319 return (&tcp_thread
[0].td_msgport
);
322 * Already bound and connected or listening. For TCP connections,
323 * the (faddr, fport, laddr, lport) association cannot change now.
325 * Note: T/TCP code needs some reorganization to fit into
328 * Rely on type-stable memory and check in protocol handler
329 * to fix race condition here w/ deallocation of inp. XXX JH
331 return (&tcp_thread
[INP_MPORT_HASH(inp
->inp_faddr
.s_addr
,
332 inp
->inp_laddr
.s_addr
, inp
->inp_fport
, inp
->inp_lport
)].td_msgport
);
336 * Used to route icmp messages to the proper protocol thread for ctlinput
340 tcp_ctlport(int cmd
, struct sockaddr
*sa
, void *vip
)
344 struct in_addr faddr
;
347 faddr
= ((struct sockaddr_in
*)sa
)->sin_addr
;
348 if (sa
->sa_family
!= AF_INET
|| faddr
.s_addr
== INADDR_ANY
)
350 if (ip
== NULL
|| PRC_IS_REDIRECT(cmd
) || cmd
== PRC_HOSTDEAD
) {
352 * Message will be forwarded to all TCP protocol threads
355 * netisr0 (the msgport we return here)
358 * | domsg <----------------------------+
362 * V forwardmsg forwardmsg |
363 * tcp0 ------------> tcp1 ------------> tcpN
365 return cpu0_ctlport(cmd
, sa
, vip
);
367 th
= (struct tcphdr
*)((caddr_t
)ip
+ (ip
->ip_hl
<< 2));
368 cpu
= tcp_addrcpu(faddr
.s_addr
, th
->th_dport
,
369 ip
->ip_src
.s_addr
, th
->th_sport
);
371 return(&tcp_thread
[cpu
].td_msgport
);
375 tcp_addrport(in_addr_t faddr
, in_port_t fport
, in_addr_t laddr
, in_port_t lport
)
377 return (&tcp_thread
[tcp_addrcpu(faddr
, fport
,
378 laddr
, lport
)].td_msgport
);
382 * Map a UDP socket to a protocol processing thread.
385 udp_soport(struct socket
*so
, struct sockaddr
*nam __unused
,
386 struct mbuf
**dummy __unused
, int req
)
391 * The following processing all take place on Protocol Thread 0:
393 * attach() has a null socket parameter
394 * Fast and slow timeouts pass in null socket parameter
396 if (req
== PRU_BIND
|| so
== NULL
)
397 return (&udp_thread
[0].td_msgport
);
401 if (IN_MULTICAST(ntohl(inp
->inp_laddr
.s_addr
)))
402 return (&udp_thread
[0].td_msgport
);
405 * Rely on type-stable memory and check in protocol handler
406 * to fix race condition here w/ deallocation of inp. XXX JH
409 return (&udp_thread
[INP_MPORT_HASH(inp
->inp_faddr
.s_addr
,
410 inp
->inp_laddr
.s_addr
, inp
->inp_fport
, inp
->inp_lport
)].td_msgport
);
414 * Used to route icmp messages to the proper protocol thread for ctlinput
418 udp_ctlport(int cmd
, struct sockaddr
*sa
, void *vip
)
422 struct in_addr faddr
;
425 faddr
= ((struct sockaddr_in
*)sa
)->sin_addr
;
426 if (sa
->sa_family
!= AF_INET
|| faddr
.s_addr
== INADDR_ANY
)
428 if (PRC_IS_REDIRECT(cmd
)) {
430 * See the comment in tcp_ctlport; the only difference
431 * is that message is forwarded to UDP protocol theads.
433 return cpu0_ctlport(cmd
, sa
, vip
);
434 } else if (ip
== NULL
|| cmd
== PRC_HOSTDEAD
) {
437 * Once UDP inpcbs are CPU localized, we should do
438 * the same forwarding as PRC_IS_REDIRECT(cmd)
442 uh
= (struct udphdr
*)((caddr_t
)ip
+ (ip
->ip_hl
<< 2));
444 cpu
= INP_MPORT_HASH(faddr
.s_addr
, ip
->ip_src
.s_addr
,
445 uh
->uh_dport
, uh
->uh_sport
);
447 return (&udp_thread
[cpu
].td_msgport
);
451 * Map a network address to a processor.
454 tcp_addrcpu(in_addr_t faddr
, in_port_t fport
, in_addr_t laddr
, in_port_t lport
)
456 return (INP_MPORT_HASH(faddr
, laddr
, fport
, lport
));
460 udp_addrcpu(in_addr_t faddr
, in_port_t fport
, in_addr_t laddr
, in_port_t lport
)
462 if (IN_MULTICAST(ntohl(laddr
)))
465 return (INP_MPORT_HASH(faddr
, laddr
, fport
, lport
));
469 * Return LWKT port for cpu.
474 return (&tcp_thread
[cpu
].td_msgport
);
480 return (&udp_thread
[cpu
].td_msgport
);
484 tcp_thread_init(void)
488 for (cpu
= 0; cpu
< ncpus2
; cpu
++) {
489 lwkt_create(tcpmsg_service_loop
, NULL
, NULL
,
490 &tcp_thread
[cpu
], TDF_NETWORK
| TDF_MPSAFE
, cpu
,
491 "tcp_thread %d", cpu
);
492 netmsg_service_port_init(&tcp_thread
[cpu
].td_msgport
);
497 udp_thread_init(void)
501 for (cpu
= 0; cpu
< ncpus2
; cpu
++) {
502 lwkt_create(netmsg_service_loop
, &udp_mpsafe_thread
, NULL
,
503 &udp_thread
[cpu
], TDF_NETWORK
| TDF_MPSAFE
, cpu
,
504 "udp_thread %d", cpu
);
505 netmsg_service_port_init(&udp_thread
[cpu
].td_msgport
);