1 /* 2 * Copyright (c) 1984, 1985 Regents of the University of California. 3 * All rights reserved. The Berkeley software License Agreement 4 * specifies the terms and conditions for redistribution. 5 * 6 * @(#)ns_pcb.c 6.8 (Berkeley) 06/02/86 7 */ 8 9 #include "param.h" 10 #include "systm.h" 11 #include "dir.h" 12 #include "user.h" 13 #include "mbuf.h" 14 #include "socket.h" 15 #include "socketvar.h" 16 #include "../net/if.h" 17 #include "../net/route.h" 18 #include "protosw.h" 19 20 #include "ns.h" 21 #include "ns_if.h" 22 #include "ns_pcb.h" 23 24 struct ns_addr zerons_addr; 25 26 ns_pcballoc(so, head) 27 struct socket *so; 28 struct nspcb *head; 29 { 30 struct mbuf *m; 31 register struct nspcb *nsp; 32 33 m = m_getclr(M_DONTWAIT, MT_PCB); 34 if (m == NULL) 35 return (ENOBUFS); 36 nsp = mtod(m, struct nspcb *); 37 nsp->nsp_socket = so; 38 insque(nsp, head); 39 so->so_pcb = (caddr_t)nsp; 40 return (0); 41 } 42 43 ns_pcbbind(nsp, nam) 44 register struct nspcb *nsp; 45 struct mbuf *nam; 46 { 47 register struct sockaddr_ns *sns; 48 u_short lport = 0; 49 50 if(nsp->nsp_lport || !ns_nullhost(nsp->nsp_laddr)) 51 return (EINVAL); 52 if (nam == 0) 53 goto noname; 54 sns = mtod(nam, struct sockaddr_ns *); 55 if (nam->m_len != sizeof (*sns)) 56 return (EINVAL); 57 if (!ns_nullhost(sns->sns_addr)) { 58 int tport = sns->sns_port; 59 60 sns->sns_port = 0; /* yech... */ 61 if (ifa_ifwithaddr((struct sockaddr *)sns) == 0) 62 return (EADDRNOTAVAIL); 63 sns->sns_port = tport; 64 } 65 lport = sns->sns_port; 66 if (lport) { 67 u_short aport = ntohs(lport); 68 69 if (aport < NSPORT_RESERVED && u.u_uid != 0) 70 return (EACCES); 71 if (ns_pcblookup(&zerons_addr, lport, 0)) 72 return (EADDRINUSE); 73 } 74 nsp->nsp_laddr = sns->sns_addr; 75 noname: 76 if (lport == 0) 77 do { 78 if (nspcb.nsp_lport++ < NSPORT_RESERVED) 79 nspcb.nsp_lport = NSPORT_RESERVED; 80 lport = htons(nspcb.nsp_lport); 81 } while (ns_pcblookup(&zerons_addr, lport, 0)); 82 nsp->nsp_lport = lport; 83 return (0); 84 } 85 86 /* 87 * Connect from a socket to a specified address. 88 * Both address and port must be specified in argument sns. 89 * If don't have a local address for this socket yet, 90 * then pick one. 91 */ 92 ns_pcbconnect(nsp, nam) 93 struct nspcb *nsp; 94 struct mbuf *nam; 95 { 96 struct ns_ifaddr *ia; 97 register struct sockaddr_ns *sns = mtod(nam, struct sockaddr_ns *); 98 register struct ns_addr *dst; 99 100 if (nam->m_len != sizeof (*sns)) 101 return (EINVAL); 102 if (sns->sns_family != AF_NS) 103 return (EAFNOSUPPORT); 104 if (sns->sns_port==0 || ns_nullhost(sns->sns_addr)) 105 return (EADDRNOTAVAIL); 106 if (ns_nullhost(nsp->nsp_laddr) && 107 (!ns_neteq(nsp->nsp_lastdst, sns->sns_addr))) { 108 register struct route *ro; 109 struct ifnet *ifp; 110 ro = &nsp->nsp_route; 111 dst = &satons_addr(ro->ro_dst); 112 113 ia = ns_iaonnetof(&sns->sns_addr); 114 if (ia == 0 || 115 (ia->ia_ifp->if_flags & IFF_UP) == 0) { 116 /* 117 * If route is known or can be allocated now, 118 * our src addr is taken from the i/f, else punt. 119 */ 120 if (ro->ro_rt && 121 !ns_hosteq(*dst, sns->sns_addr)) { 122 RTFREE(ro->ro_rt); 123 ro->ro_rt = (struct rtentry *)0; 124 } 125 if ((ro->ro_rt == (struct rtentry *)0) || 126 (ifp = ro->ro_rt->rt_ifp) == (struct ifnet *)0) { 127 /* No route yet, so try to acquire one */ 128 ro->ro_dst.sa_family = AF_NS; 129 *dst = sns->sns_addr; 130 dst->x_port = 0; 131 rtalloc(ro); 132 if (ro->ro_rt == 0) 133 ifp = (struct ifnet *)0; 134 else 135 ifp = ro->ro_rt->rt_ifp; 136 } 137 if (ifp) { 138 for (ia = ns_ifaddr; ia; ia = ia->ia_next) 139 if (ia->ia_ifp == ifp) 140 break; 141 } 142 if (ia == 0) 143 ia = ns_ifaddr; 144 if (ia == 0) 145 return (EADDRNOTAVAIL); 146 } else if (ro->ro_rt) { 147 if (ns_neteq(*dst, sns->sns_addr)) { 148 /* 149 * This assume that we have no GH 150 * type routes. 151 */ 152 if (ro->ro_rt->rt_flags & RTF_HOST) { 153 if (!ns_hosteq(*dst, sns->sns_addr)) 154 goto re_route; 155 156 } 157 if ((ro->ro_rt->rt_flags & RTF_GATEWAY) == 0) { 158 dst->x_host = sns->sns_addr.x_host; 159 } 160 /* 161 * Otherwise, we go through the same gateway 162 * and dst is already set up. 163 */ 164 } else { 165 re_route: 166 RTFREE(ro->ro_rt); 167 ro->ro_rt = (struct rtentry *)0; 168 } 169 } 170 nsp->nsp_laddr.x_net = satons_addr(ia->ia_addr).x_net; 171 nsp->nsp_lastdst = sns->sns_addr; 172 } 173 if (ns_pcblookup(&sns->sns_addr, nsp->nsp_lport, 0)) 174 return (EADDRINUSE); 175 if (ns_nullhost(nsp->nsp_laddr)) { 176 if (nsp->nsp_lport == 0) 177 (void) ns_pcbbind(nsp, (struct mbuf *)0); 178 nsp->nsp_laddr.x_host = ns_thishost; 179 } 180 nsp->nsp_faddr = sns->sns_addr; 181 /* Includes nsp->nsp_fport = sns->sns_port; */ 182 return (0); 183 } 184 185 ns_pcbdisconnect(nsp) 186 struct nspcb *nsp; 187 { 188 189 nsp->nsp_faddr = zerons_addr; 190 if (nsp->nsp_socket->so_state & SS_NOFDREF) 191 ns_pcbdetach(nsp); 192 } 193 194 ns_pcbdetach(nsp) 195 struct nspcb *nsp; 196 { 197 struct socket *so = nsp->nsp_socket; 198 199 so->so_pcb = 0; 200 sofree(so); 201 if (nsp->nsp_route.ro_rt) 202 rtfree(nsp->nsp_route.ro_rt); 203 remque(nsp); 204 (void) m_free(dtom(nsp)); 205 } 206 207 ns_setsockaddr(nsp, nam) 208 register struct nspcb *nsp; 209 struct mbuf *nam; 210 { 211 register struct sockaddr_ns *sns = mtod(nam, struct sockaddr_ns *); 212 213 nam->m_len = sizeof (*sns); 214 sns = mtod(nam, struct sockaddr_ns *); 215 bzero((caddr_t)sns, sizeof (*sns)); 216 sns->sns_family = AF_NS; 217 sns->sns_addr = nsp->nsp_laddr; 218 } 219 220 ns_setpeeraddr(nsp, nam) 221 register struct nspcb *nsp; 222 struct mbuf *nam; 223 { 224 register struct sockaddr_ns *sns = mtod(nam, struct sockaddr_ns *); 225 226 nam->m_len = sizeof (*sns); 227 sns = mtod(nam, struct sockaddr_ns *); 228 bzero((caddr_t)sns, sizeof (*sns)); 229 sns->sns_family = AF_NS; 230 sns->sns_addr = nsp->nsp_faddr; 231 } 232 233 /* 234 * Pass some notification to all connections of a protocol 235 * associated with address dst. Call the 236 * protocol specific routine to handle each connection. 237 * Also pass an extra paramter via the nspcb. (which may in fact 238 * be a parameter list!) 239 */ 240 ns_pcbnotify(dst, errno, notify, param) 241 register struct ns_addr *dst; 242 long param; 243 int errno, (*notify)(); 244 { 245 register struct nspcb *nsp, *oinp; 246 int s = splimp(); 247 248 for (nsp = (&nspcb)->nsp_next; nsp != (&nspcb);) { 249 if (!ns_hosteq(*dst,nsp->nsp_faddr)) { 250 next: 251 nsp = nsp->nsp_next; 252 continue; 253 } 254 if (nsp->nsp_socket == 0) 255 goto next; 256 if (errno) 257 nsp->nsp_socket->so_error = errno; 258 oinp = nsp; 259 nsp = nsp->nsp_next; 260 oinp->nsp_notify_param = param; 261 (*notify)(oinp); 262 } 263 splx(s); 264 } 265 266 #ifdef notdef 267 /* 268 * After a routing change, flush old routing 269 * and allocate a (hopefully) better one. 270 */ 271 ns_rtchange(nsp) 272 struct nspcb *nsp; 273 { 274 if (nsp->nsp_route.ro_rt) { 275 rtfree(nsp->nsp_route.ro_rt); 276 nsp->nsp_route.ro_rt = 0; 277 /* 278 * A new route can be allocated the next time 279 * output is attempted. 280 */ 281 } 282 /* SHOULD NOTIFY HIGHER-LEVEL PROTOCOLS */ 283 } 284 #endif 285 286 struct nspcb * 287 ns_pcblookup(faddr, lport, wildp) 288 struct ns_addr *faddr; 289 u_short lport; 290 { 291 register struct nspcb *nsp, *match = 0; 292 int matchwild = 3, wildcard; 293 u_short fport; 294 295 fport = faddr->x_port; 296 for (nsp = (&nspcb)->nsp_next; nsp != (&nspcb); nsp = nsp->nsp_next) { 297 if (nsp->nsp_lport != lport) 298 continue; 299 wildcard = 0; 300 if (ns_nullhost(nsp->nsp_faddr)) { 301 if (!ns_nullhost(*faddr)) 302 wildcard++; 303 } else { 304 if (ns_nullhost(*faddr)) 305 wildcard++; 306 else { 307 if (!ns_hosteq(nsp->nsp_faddr, *faddr)) 308 continue; 309 if( nsp->nsp_fport != fport) { 310 if(nsp->nsp_fport != 0) 311 continue; 312 else 313 wildcard++; 314 } 315 } 316 } 317 if (wildcard && wildp==0) 318 continue; 319 if (wildcard < matchwild) { 320 match = nsp; 321 matchwild = wildcard; 322 if (wildcard == 0) 323 break; 324 } 325 } 326 return (match); 327 } 328