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.10 (Berkeley) 05/05/89 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 "in.h" 29 #include "in_systm.h" 30 #include "../net/if.h" 31 #include "../net/route.h" 32 #include "in_pcb.h" 33 #include "in_var.h" 34 #include "protosw.h" 35 36 struct in_addr zeroin_addr; 37 38 in_pcballoc(so, head) 39 struct socket *so; 40 struct inpcb *head; 41 { 42 struct mbuf *m; 43 register struct inpcb *inp; 44 45 m = m_getclr(M_DONTWAIT, MT_PCB); 46 if (m == NULL) 47 return (ENOBUFS); 48 inp = mtod(m, struct inpcb *); 49 inp->inp_head = head; 50 inp->inp_socket = so; 51 insque(inp, head); 52 so->so_pcb = (caddr_t)inp; 53 return (0); 54 } 55 56 in_pcbbind(inp, nam) 57 register struct inpcb *inp; 58 struct mbuf *nam; 59 { 60 register struct socket *so = inp->inp_socket; 61 register struct inpcb *head = inp->inp_head; 62 register struct sockaddr_in *sin; 63 u_short lport = 0; 64 65 if (in_ifaddr == 0) 66 return (EADDRNOTAVAIL); 67 if (inp->inp_lport || inp->inp_laddr.s_addr != INADDR_ANY) 68 return (EINVAL); 69 if (nam == 0) 70 goto noname; 71 sin = mtod(nam, struct sockaddr_in *); 72 if (nam->m_len != sizeof (*sin)) 73 return (EINVAL); 74 if (sin->sin_addr.s_addr != INADDR_ANY) { 75 int tport = sin->sin_port; 76 77 sin->sin_port = 0; /* yech... */ 78 if (ifa_ifwithaddr((struct sockaddr *)sin) == 0) 79 return (EADDRNOTAVAIL); 80 sin->sin_port = tport; 81 } 82 lport = sin->sin_port; 83 if (lport) { 84 u_short aport = ntohs(lport); 85 int wild = 0; 86 87 /* GROSS */ 88 if (aport < IPPORT_RESERVED && u.u_uid != 0) 89 return (EACCES); 90 /* even GROSSER, but this is the Internet */ 91 if ((so->so_options & SO_REUSEADDR) == 0 && 92 ((so->so_proto->pr_flags & PR_CONNREQUIRED) == 0 || 93 (so->so_options & SO_ACCEPTCONN) == 0)) 94 wild = INPLOOKUP_WILDCARD; 95 if (in_pcblookup(head, 96 zeroin_addr, 0, sin->sin_addr, lport, wild)) 97 return (EADDRINUSE); 98 } 99 inp->inp_laddr = sin->sin_addr; 100 noname: 101 if (lport == 0) 102 do { 103 if (head->inp_lport++ < IPPORT_RESERVED || 104 head->inp_lport > IPPORT_USERRESERVED) 105 head->inp_lport = IPPORT_RESERVED; 106 lport = htons(head->inp_lport); 107 } while (in_pcblookup(head, 108 zeroin_addr, 0, inp->inp_laddr, lport, 0)); 109 inp->inp_lport = lport; 110 return (0); 111 } 112 113 /* 114 * Connect from a socket to a specified address. 115 * Both address and port must be specified in argument sin. 116 * If don't have a local address for this socket yet, 117 * then pick one. 118 */ 119 in_pcbconnect(inp, nam) 120 register struct inpcb *inp; 121 struct mbuf *nam; 122 { 123 struct in_ifaddr *ia; 124 struct sockaddr_in *ifaddr; 125 register struct sockaddr_in *sin = mtod(nam, struct sockaddr_in *); 126 127 if (nam->m_len != sizeof (*sin)) 128 return (EINVAL); 129 if (sin->sin_family != AF_INET) 130 return (EAFNOSUPPORT); 131 if (sin->sin_port == 0) 132 return (EADDRNOTAVAIL); 133 if (in_ifaddr) { 134 /* 135 * If the destination address is INADDR_ANY, 136 * use the primary local address. 137 * If the supplied address is INADDR_BROADCAST, 138 * and the primary interface supports broadcast, 139 * choose the broadcast address for that interface. 140 */ 141 #define satosin(sa) ((struct sockaddr_in *)(sa)) 142 if (sin->sin_addr.s_addr == INADDR_ANY) 143 sin->sin_addr = IA_SIN(in_ifaddr)->sin_addr; 144 else if (sin->sin_addr.s_addr == (u_long)INADDR_BROADCAST && 145 (in_ifaddr->ia_ifp->if_flags & IFF_BROADCAST)) 146 sin->sin_addr = satosin(&in_ifaddr->ia_broadaddr)->sin_addr; 147 } 148 if (inp->inp_laddr.s_addr == INADDR_ANY) { 149 register struct route *ro; 150 struct ifnet *ifp; 151 152 ia = (struct in_ifaddr *)0; 153 /* 154 * If route is known or can be allocated now, 155 * our src addr is taken from the i/f, else punt. 156 */ 157 ro = &inp->inp_route; 158 if (ro->ro_rt && 159 (satosin(&ro->ro_dst)->sin_addr.s_addr != 160 sin->sin_addr.s_addr || 161 inp->inp_socket->so_options & SO_DONTROUTE)) { 162 RTFREE(ro->ro_rt); 163 ro->ro_rt = (struct rtentry *)0; 164 } 165 if ((inp->inp_socket->so_options & SO_DONTROUTE) == 0 && /*XXX*/ 166 (ro->ro_rt == (struct rtentry *)0 || 167 ro->ro_rt->rt_ifp == (struct ifnet *)0)) { 168 /* No route yet, so try to acquire one */ 169 ro->ro_dst.sa_family = AF_INET; 170 ro->ro_dst.sa_len = sizeof(struct sockaddr_in); 171 ((struct sockaddr_in *) &ro->ro_dst)->sin_addr = 172 sin->sin_addr; 173 rtalloc(ro); 174 } 175 /* 176 * If we found a route, use the address 177 * corresponding to the outgoing interface 178 * unless it is the loopback (in case a route 179 * to our address on another net goes to loopback). 180 */ 181 if (ro->ro_rt && (ifp = ro->ro_rt->rt_ifp) && 182 (ifp->if_flags & IFF_LOOPBACK) == 0) 183 for (ia = in_ifaddr; ia; ia = ia->ia_next) 184 if (ia->ia_ifp == ifp) 185 break; 186 if (ia == 0) { 187 int fport = sin->sin_port; 188 189 sin->sin_port = 0; 190 ia = (struct in_ifaddr *) 191 ifa_ifwithdstaddr((struct sockaddr *)sin); 192 sin->sin_port = fport; 193 if (ia == 0) 194 ia = in_iaonnetof(in_netof(sin->sin_addr)); 195 if (ia == 0) 196 ia = in_ifaddr; 197 if (ia == 0) 198 return (EADDRNOTAVAIL); 199 } 200 ifaddr = (struct sockaddr_in *)&ia->ia_addr; 201 } 202 if (in_pcblookup(inp->inp_head, 203 sin->sin_addr, 204 sin->sin_port, 205 inp->inp_laddr.s_addr ? inp->inp_laddr : ifaddr->sin_addr, 206 inp->inp_lport, 207 0)) 208 return (EADDRINUSE); 209 if (inp->inp_laddr.s_addr == INADDR_ANY) { 210 if (inp->inp_lport == 0) 211 (void)in_pcbbind(inp, (struct mbuf *)0); 212 inp->inp_laddr = ifaddr->sin_addr; 213 } 214 inp->inp_faddr = sin->sin_addr; 215 inp->inp_fport = sin->sin_port; 216 return (0); 217 } 218 219 in_pcbdisconnect(inp) 220 struct inpcb *inp; 221 { 222 223 inp->inp_faddr.s_addr = INADDR_ANY; 224 inp->inp_fport = 0; 225 if (inp->inp_socket->so_state & SS_NOFDREF) 226 in_pcbdetach(inp); 227 } 228 229 in_pcbdetach(inp) 230 struct inpcb *inp; 231 { 232 struct socket *so = inp->inp_socket; 233 234 so->so_pcb = 0; 235 sofree(so); 236 if (inp->inp_options) 237 (void)m_free(inp->inp_options); 238 if (inp->inp_route.ro_rt) 239 rtfree(inp->inp_route.ro_rt); 240 remque(inp); 241 (void) m_free(dtom(inp)); 242 } 243 244 in_setsockaddr(inp, nam) 245 register struct inpcb *inp; 246 struct mbuf *nam; 247 { 248 register struct sockaddr_in *sin; 249 250 nam->m_len = sizeof (*sin); 251 sin = mtod(nam, struct sockaddr_in *); 252 bzero((caddr_t)sin, sizeof (*sin)); 253 sin->sin_family = AF_INET; 254 sin->sin_len = sizeof(*sin); 255 sin->sin_port = inp->inp_lport; 256 sin->sin_addr = inp->inp_laddr; 257 } 258 259 in_setpeeraddr(inp, nam) 260 struct inpcb *inp; 261 struct mbuf *nam; 262 { 263 register struct sockaddr_in *sin; 264 265 nam->m_len = sizeof (*sin); 266 sin = mtod(nam, struct sockaddr_in *); 267 bzero((caddr_t)sin, sizeof (*sin)); 268 sin->sin_family = AF_INET; 269 sin->sin_len = sizeof(*sin); 270 sin->sin_port = inp->inp_fport; 271 sin->sin_addr = inp->inp_faddr; 272 } 273 274 /* 275 * Pass some notification to all connections of a protocol 276 * associated with address dst. Call the protocol specific 277 * routine (if any) to handle each connection. 278 */ 279 in_pcbnotify(head, dst, errno, notify) 280 struct inpcb *head; 281 register struct in_addr *dst; 282 int errno, (*notify)(); 283 { 284 register struct inpcb *inp, *oinp; 285 int s = splimp(); 286 287 for (inp = head->inp_next; inp != head;) { 288 if (inp->inp_faddr.s_addr != dst->s_addr || 289 inp->inp_socket == 0) { 290 inp = inp->inp_next; 291 continue; 292 } 293 if (errno) 294 inp->inp_socket->so_error = errno; 295 oinp = inp; 296 inp = inp->inp_next; 297 if (notify) 298 (*notify)(oinp); 299 } 300 splx(s); 301 } 302 303 /* 304 * Check for alternatives when higher level complains 305 * about service problems. For now, invalidate cached 306 * routing information. If the route was created dynamically 307 * (by a redirect), time to try a default gateway again. 308 */ 309 in_losing(inp) 310 struct inpcb *inp; 311 { 312 register struct rtentry *rt; 313 314 if ((rt = inp->inp_route.ro_rt)) { 315 rt_missmsg(RTM_LOSING, &inp->inp_route.ro_dst, 316 rt->rt_gateway, (struct sockaddr *)rt_mask(rt), 317 (struct sockaddr *)0, rt->rt_flags, 0); 318 if (rt->rt_flags & RTF_DYNAMIC) 319 (void) rtrequest(RTM_DELETE, rt_key(rt), 320 rt->rt_gateway, rt_mask(rt), rt->rt_flags, 321 (struct rtentry **)0); 322 inp->inp_route.ro_rt = 0; 323 rtfree(rt); 324 /* 325 * A new route can be allocated 326 * the next time output is attempted. 327 */ 328 } 329 } 330 331 /* 332 * After a routing change, flush old routing 333 * and allocate a (hopefully) better one. 334 */ 335 in_rtchange(inp) 336 register struct inpcb *inp; 337 { 338 if (inp->inp_route.ro_rt) { 339 rtfree(inp->inp_route.ro_rt); 340 inp->inp_route.ro_rt = 0; 341 /* 342 * A new route can be allocated the next time 343 * output is attempted. 344 */ 345 } 346 } 347 348 struct inpcb * 349 in_pcblookup(head, faddr, fport, laddr, lport, flags) 350 struct inpcb *head; 351 struct in_addr faddr, laddr; 352 u_short fport, lport; 353 int flags; 354 { 355 register struct inpcb *inp, *match = 0; 356 int matchwild = 3, wildcard; 357 358 for (inp = head->inp_next; inp != head; inp = inp->inp_next) { 359 if (inp->inp_lport != lport) 360 continue; 361 wildcard = 0; 362 if (inp->inp_laddr.s_addr != INADDR_ANY) { 363 if (laddr.s_addr == INADDR_ANY) 364 wildcard++; 365 else if (inp->inp_laddr.s_addr != laddr.s_addr) 366 continue; 367 } else { 368 if (laddr.s_addr != INADDR_ANY) 369 wildcard++; 370 } 371 if (inp->inp_faddr.s_addr != INADDR_ANY) { 372 if (faddr.s_addr == INADDR_ANY) 373 wildcard++; 374 else if (inp->inp_faddr.s_addr != faddr.s_addr || 375 inp->inp_fport != fport) 376 continue; 377 } else { 378 if (faddr.s_addr != INADDR_ANY) 379 wildcard++; 380 } 381 if (wildcard && (flags & INPLOOKUP_WILDCARD) == 0) 382 continue; 383 if (wildcard < matchwild) { 384 match = inp; 385 matchwild = wildcard; 386 if (matchwild == 0) 387 break; 388 } 389 } 390 return (match); 391 } 392