1 /* 2 * Copyright (c) 1982, 1986 Regents of the University of California. 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms are permitted 6 * provided that the above copyright notice and this paragraph are 7 * duplicated in all such forms and that any documentation, 8 * advertising materials, and other materials related to such 9 * distribution and use acknowledge that the software was developed 10 * by the University of California, Berkeley. The name of the 11 * University may not be used to endorse or promote products derived 12 * from this software without specific prior written permission. 13 * THIS SOFTWARE IS PROVIDED ``AS IS'' AND WITHOUT ANY EXPRESS OR 14 * IMPLIED WARRANTIES, INCLUDING, WITHOUT LIMITATION, THE IMPLIED 15 * WARRANTIES OF MERCHANTIBILITY AND FITNESS FOR A PARTICULAR PURPOSE. 16 * 17 * @(#)in_pcb.c 7.11 (Berkeley) 04/03/90 18 */ 19 20 #include "param.h" 21 #include "systm.h" 22 #include "user.h" 23 #include "malloc.h" 24 #include "mbuf.h" 25 #include "socket.h" 26 #include "socketvar.h" 27 #include "ioctl.h" 28 #include "../net/if.h" 29 #include "../net/route.h" 30 #include "in.h" 31 #include "in_systm.h" 32 #include "ip.h" 33 #include "in_pcb.h" 34 #include "in_var.h" 35 #include "protosw.h" 36 37 struct in_addr zeroin_addr; 38 39 in_pcballoc(so, head) 40 struct socket *so; 41 struct inpcb *head; 42 { 43 struct mbuf *m; 44 register struct inpcb *inp; 45 46 m = m_getclr(M_DONTWAIT, MT_PCB); 47 if (m == NULL) 48 return (ENOBUFS); 49 inp = mtod(m, struct inpcb *); 50 inp->inp_head = head; 51 inp->inp_socket = so; 52 insque(inp, head); 53 so->so_pcb = (caddr_t)inp; 54 return (0); 55 } 56 57 in_pcbbind(inp, nam) 58 register struct inpcb *inp; 59 struct mbuf *nam; 60 { 61 register struct socket *so = inp->inp_socket; 62 register struct inpcb *head = inp->inp_head; 63 register struct sockaddr_in *sin; 64 u_short lport = 0; 65 66 if (in_ifaddr == 0) 67 return (EADDRNOTAVAIL); 68 if (inp->inp_lport || inp->inp_laddr.s_addr != INADDR_ANY) 69 return (EINVAL); 70 if (nam == 0) 71 goto noname; 72 sin = mtod(nam, struct sockaddr_in *); 73 if (nam->m_len != sizeof (*sin)) 74 return (EINVAL); 75 if (sin->sin_addr.s_addr != INADDR_ANY) { 76 int tport = sin->sin_port; 77 78 sin->sin_port = 0; /* yech... */ 79 if (ifa_ifwithaddr((struct sockaddr *)sin) == 0) 80 return (EADDRNOTAVAIL); 81 sin->sin_port = tport; 82 } 83 lport = sin->sin_port; 84 if (lport) { 85 u_short aport = ntohs(lport); 86 int wild = 0; 87 88 /* GROSS */ 89 if (aport < IPPORT_RESERVED && u.u_uid != 0) 90 return (EACCES); 91 /* even GROSSER, but this is the Internet */ 92 if ((so->so_options & SO_REUSEADDR) == 0 && 93 ((so->so_proto->pr_flags & PR_CONNREQUIRED) == 0 || 94 (so->so_options & SO_ACCEPTCONN) == 0)) 95 wild = INPLOOKUP_WILDCARD; 96 if (in_pcblookup(head, 97 zeroin_addr, 0, sin->sin_addr, lport, wild)) 98 return (EADDRINUSE); 99 } 100 inp->inp_laddr = sin->sin_addr; 101 noname: 102 if (lport == 0) 103 do { 104 if (head->inp_lport++ < IPPORT_RESERVED || 105 head->inp_lport > IPPORT_USERRESERVED) 106 head->inp_lport = IPPORT_RESERVED; 107 lport = htons(head->inp_lport); 108 } while (in_pcblookup(head, 109 zeroin_addr, 0, inp->inp_laddr, lport, 0)); 110 inp->inp_lport = lport; 111 return (0); 112 } 113 114 /* 115 * Connect from a socket to a specified address. 116 * Both address and port must be specified in argument sin. 117 * If don't have a local address for this socket yet, 118 * then pick one. 119 */ 120 in_pcbconnect(inp, nam) 121 register struct inpcb *inp; 122 struct mbuf *nam; 123 { 124 struct in_ifaddr *ia; 125 struct sockaddr_in *ifaddr; 126 register struct sockaddr_in *sin = mtod(nam, struct sockaddr_in *); 127 128 if (nam->m_len != sizeof (*sin)) 129 return (EINVAL); 130 if (sin->sin_family != AF_INET) 131 return (EAFNOSUPPORT); 132 if (sin->sin_port == 0) 133 return (EADDRNOTAVAIL); 134 if (in_ifaddr) { 135 /* 136 * If the destination address is INADDR_ANY, 137 * use the primary local address. 138 * If the supplied address is INADDR_BROADCAST, 139 * and the primary interface supports broadcast, 140 * choose the broadcast address for that interface. 141 */ 142 #define satosin(sa) ((struct sockaddr_in *)(sa)) 143 if (sin->sin_addr.s_addr == INADDR_ANY) 144 sin->sin_addr = IA_SIN(in_ifaddr)->sin_addr; 145 else if (sin->sin_addr.s_addr == (u_long)INADDR_BROADCAST && 146 (in_ifaddr->ia_ifp->if_flags & IFF_BROADCAST)) 147 sin->sin_addr = satosin(&in_ifaddr->ia_broadaddr)->sin_addr; 148 } 149 if (inp->inp_laddr.s_addr == INADDR_ANY) { 150 register struct route *ro; 151 struct ifnet *ifp; 152 153 ia = (struct in_ifaddr *)0; 154 /* 155 * If route is known or can be allocated now, 156 * our src addr is taken from the i/f, else punt. 157 */ 158 ro = &inp->inp_route; 159 if (ro->ro_rt && 160 (satosin(&ro->ro_dst)->sin_addr.s_addr != 161 sin->sin_addr.s_addr || 162 inp->inp_socket->so_options & SO_DONTROUTE)) { 163 RTFREE(ro->ro_rt); 164 ro->ro_rt = (struct rtentry *)0; 165 } 166 if ((inp->inp_socket->so_options & SO_DONTROUTE) == 0 && /*XXX*/ 167 (ro->ro_rt == (struct rtentry *)0 || 168 ro->ro_rt->rt_ifp == (struct ifnet *)0)) { 169 /* No route yet, so try to acquire one */ 170 ro->ro_dst.sa_family = AF_INET; 171 ro->ro_dst.sa_len = sizeof(struct sockaddr_in); 172 ((struct sockaddr_in *) &ro->ro_dst)->sin_addr = 173 sin->sin_addr; 174 rtalloc(ro); 175 } 176 /* 177 * If we found a route, use the address 178 * corresponding to the outgoing interface 179 * unless it is the loopback (in case a route 180 * to our address on another net goes to loopback). 181 */ 182 if (ro->ro_rt && (ifp = ro->ro_rt->rt_ifp) && 183 (ifp->if_flags & IFF_LOOPBACK) == 0) 184 for (ia = in_ifaddr; ia; ia = ia->ia_next) 185 if (ia->ia_ifp == ifp) 186 break; 187 if (ia == 0) { 188 int fport = sin->sin_port; 189 190 sin->sin_port = 0; 191 ia = (struct in_ifaddr *) 192 ifa_ifwithdstaddr((struct sockaddr *)sin); 193 sin->sin_port = fport; 194 if (ia == 0) 195 ia = in_iaonnetof(in_netof(sin->sin_addr)); 196 if (ia == 0) 197 ia = in_ifaddr; 198 if (ia == 0) 199 return (EADDRNOTAVAIL); 200 } 201 ifaddr = (struct sockaddr_in *)&ia->ia_addr; 202 } 203 if (in_pcblookup(inp->inp_head, 204 sin->sin_addr, 205 sin->sin_port, 206 inp->inp_laddr.s_addr ? inp->inp_laddr : ifaddr->sin_addr, 207 inp->inp_lport, 208 0)) 209 return (EADDRINUSE); 210 if (inp->inp_laddr.s_addr == INADDR_ANY) { 211 if (inp->inp_lport == 0) 212 (void)in_pcbbind(inp, (struct mbuf *)0); 213 inp->inp_laddr = ifaddr->sin_addr; 214 } 215 inp->inp_faddr = sin->sin_addr; 216 inp->inp_fport = sin->sin_port; 217 return (0); 218 } 219 220 in_pcbdisconnect(inp) 221 struct inpcb *inp; 222 { 223 224 inp->inp_faddr.s_addr = INADDR_ANY; 225 inp->inp_fport = 0; 226 if (inp->inp_socket->so_state & SS_NOFDREF) 227 in_pcbdetach(inp); 228 } 229 230 in_pcbdetach(inp) 231 struct inpcb *inp; 232 { 233 struct socket *so = inp->inp_socket; 234 235 so->so_pcb = 0; 236 sofree(so); 237 if (inp->inp_options) 238 (void)m_free(inp->inp_options); 239 if (inp->inp_route.ro_rt) 240 rtfree(inp->inp_route.ro_rt); 241 remque(inp); 242 (void) m_free(dtom(inp)); 243 } 244 245 in_setsockaddr(inp, nam) 246 register struct inpcb *inp; 247 struct mbuf *nam; 248 { 249 register struct sockaddr_in *sin; 250 251 nam->m_len = sizeof (*sin); 252 sin = mtod(nam, struct sockaddr_in *); 253 bzero((caddr_t)sin, sizeof (*sin)); 254 sin->sin_family = AF_INET; 255 sin->sin_len = sizeof(*sin); 256 sin->sin_port = inp->inp_lport; 257 sin->sin_addr = inp->inp_laddr; 258 } 259 260 in_setpeeraddr(inp, nam) 261 struct inpcb *inp; 262 struct mbuf *nam; 263 { 264 register struct sockaddr_in *sin; 265 266 nam->m_len = sizeof (*sin); 267 sin = mtod(nam, struct sockaddr_in *); 268 bzero((caddr_t)sin, sizeof (*sin)); 269 sin->sin_family = AF_INET; 270 sin->sin_len = sizeof(*sin); 271 sin->sin_port = inp->inp_fport; 272 sin->sin_addr = inp->inp_faddr; 273 } 274 275 /* 276 * Pass some notification to all connections of a protocol 277 * associated with address dst. The local address and/or port numbers 278 * may be specified to limit the search. The "usual action" will be 279 * taken, depending on the ctlinput cmd. The caller must filter any 280 * cmds that are uninteresting (e.g., no error in the map). 281 * Call the protocol specific routine (if any) to report 282 * any errors for each matching socket. 283 * 284 * Must be called at splnet. 285 */ 286 in_pcbnotify(head, dst, fport, laddr, lport, cmd, notify) 287 struct inpcb *head; 288 struct sockaddr *dst; 289 u_short fport, lport; 290 struct in_addr laddr; 291 int cmd, (*notify)(); 292 { 293 register struct inpcb *inp, *oinp; 294 struct in_addr faddr; 295 int errno; 296 int in_rtchange(); 297 extern u_char inetctlerrmap[]; 298 299 if ((unsigned)cmd > PRC_NCMDS || dst->sa_family != AF_INET) 300 return; 301 faddr = ((struct sockaddr_in *)dst)->sin_addr; 302 if (faddr.s_addr == INADDR_ANY) 303 return; 304 305 /* 306 * Redirects go to all references to the destination, 307 * and use in_rtchange to invalidate the route cache. 308 * Dead host indications: notify all references to the destination. 309 * Otherwise, if we have knowledge of the local port and address, 310 * deliver only to that socket. 311 */ 312 if (PRC_IS_REDIRECT(cmd) || cmd == PRC_HOSTDEAD) { 313 fport = 0; 314 lport = 0; 315 laddr.s_addr = 0; 316 if (cmd != PRC_HOSTDEAD) 317 notify = in_rtchange; 318 } 319 errno = inetctlerrmap[cmd]; 320 for (inp = head->inp_next; inp != head;) { 321 if (inp->inp_faddr.s_addr != faddr.s_addr || 322 inp->inp_socket == 0 || 323 (lport && inp->inp_lport != lport) || 324 (laddr.s_addr && inp->inp_laddr.s_addr != laddr.s_addr) || 325 (fport && inp->inp_fport != fport)) { 326 inp = inp->inp_next; 327 continue; 328 } 329 if (errno) 330 inp->inp_socket->so_error = errno; 331 oinp = inp; 332 inp = inp->inp_next; 333 if (notify) 334 (*notify)(oinp); 335 } 336 } 337 338 /* 339 * Check for alternatives when higher level complains 340 * about service problems. For now, invalidate cached 341 * routing information. If the route was created dynamically 342 * (by a redirect), time to try a default gateway again. 343 */ 344 in_losing(inp) 345 struct inpcb *inp; 346 { 347 register struct rtentry *rt; 348 349 if ((rt = inp->inp_route.ro_rt)) { 350 rt_missmsg(RTM_LOSING, &inp->inp_route.ro_dst, 351 rt->rt_gateway, (struct sockaddr *)rt_mask(rt), 352 (struct sockaddr *)0, rt->rt_flags, 0); 353 if (rt->rt_flags & RTF_DYNAMIC) 354 (void) rtrequest(RTM_DELETE, rt_key(rt), 355 rt->rt_gateway, rt_mask(rt), rt->rt_flags, 356 (struct rtentry **)0); 357 inp->inp_route.ro_rt = 0; 358 rtfree(rt); 359 /* 360 * A new route can be allocated 361 * the next time output is attempted. 362 */ 363 } 364 } 365 366 /* 367 * After a routing change, flush old routing 368 * and allocate a (hopefully) better one. 369 */ 370 in_rtchange(inp) 371 register struct inpcb *inp; 372 { 373 if (inp->inp_route.ro_rt) { 374 rtfree(inp->inp_route.ro_rt); 375 inp->inp_route.ro_rt = 0; 376 /* 377 * A new route can be allocated the next time 378 * output is attempted. 379 */ 380 } 381 } 382 383 struct inpcb * 384 in_pcblookup(head, faddr, fport, laddr, lport, flags) 385 struct inpcb *head; 386 struct in_addr faddr, laddr; 387 u_short fport, lport; 388 int flags; 389 { 390 register struct inpcb *inp, *match = 0; 391 int matchwild = 3, wildcard; 392 393 for (inp = head->inp_next; inp != head; inp = inp->inp_next) { 394 if (inp->inp_lport != lport) 395 continue; 396 wildcard = 0; 397 if (inp->inp_laddr.s_addr != INADDR_ANY) { 398 if (laddr.s_addr == INADDR_ANY) 399 wildcard++; 400 else if (inp->inp_laddr.s_addr != laddr.s_addr) 401 continue; 402 } else { 403 if (laddr.s_addr != INADDR_ANY) 404 wildcard++; 405 } 406 if (inp->inp_faddr.s_addr != INADDR_ANY) { 407 if (faddr.s_addr == INADDR_ANY) 408 wildcard++; 409 else if (inp->inp_faddr.s_addr != faddr.s_addr || 410 inp->inp_fport != fport) 411 continue; 412 } else { 413 if (faddr.s_addr != INADDR_ANY) 414 wildcard++; 415 } 416 if (wildcard && (flags & INPLOOKUP_WILDCARD) == 0) 417 continue; 418 if (wildcard < matchwild) { 419 match = inp; 420 matchwild = wildcard; 421 if (matchwild == 0) 422 break; 423 } 424 } 425 return (match); 426 } 427