| /* | |
| * Copyright (c) 1982, 1986, 1988, 1990, 1993 | |
| * The Regents of the University of California. All rights reserved. | |
| * | |
| * Redistribution and use in source and binary forms, with or without | |
| * modification, are permitted provided that the following conditions | |
| * are met: | |
| * 1. Redistributions of source code must retain the above copyright | |
| * notice, this list of conditions and the following disclaimer. | |
| * 2. Redistributions in binary form must reproduce the above copyright | |
| * notice, this list of conditions and the following disclaimer in the | |
| * documentation and/or other materials provided with the distribution. | |
| * 3. Neither the name of the University nor the names of its contributors | |
| * may be used to endorse or promote products derived from this software | |
| * without specific prior written permission. | |
| * | |
| * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND | |
| * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE | |
| * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE | |
| * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE | |
| * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL | |
| * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS | |
| * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) | |
| * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT | |
| * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY | |
| * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF | |
| * SUCH DAMAGE. | |
| * | |
| * @(#)udp_usrreq.c 8.4 (Berkeley) 1/21/94 | |
| * udp_usrreq.c,v 1.4 1994/10/02 17:48:45 phk Exp | |
| */ | |
| /* | |
| * Changes and additions relating to SLiRP | |
| * Copyright (c) 1995 Danny Gasparovski. | |
| * | |
| * Please read the file COPYRIGHT for the | |
| * terms and conditions of the copyright. | |
| */ | |
| #include <slirp.h> | |
| #include "ip_icmp.h" | |
| static uint8_t udp_tos(struct socket *so); | |
| void | |
| udp_init(Slirp *slirp) | |
| { | |
| slirp->udb.so_next = slirp->udb.so_prev = &slirp->udb; | |
| slirp->udp_last_so = &slirp->udb; | |
| } | |
| void udp_cleanup(Slirp *slirp) | |
| { | |
| while (slirp->udb.so_next != &slirp->udb) { | |
| udp_detach(slirp->udb.so_next); | |
| } | |
| } | |
| /* m->m_data points at ip packet header | |
| * m->m_len length ip packet | |
| * ip->ip_len length data (IPDU) | |
| */ | |
| void | |
| udp_input(register struct mbuf *m, int iphlen) | |
| { | |
| Slirp *slirp = m->slirp; | |
| register struct ip *ip; | |
| register struct udphdr *uh; | |
| int len; | |
| struct ip save_ip; | |
| struct socket *so; | |
| DEBUG_CALL("udp_input"); | |
| DEBUG_ARG("m = %lx", (long)m); | |
| DEBUG_ARG("iphlen = %d", iphlen); | |
| /* | |
| * Strip IP options, if any; should skip this, | |
| * make available to user, and use on returned packets, | |
| * but we don't yet have a way to check the checksum | |
| * with options still present. | |
| */ | |
| if(iphlen > sizeof(struct ip)) { | |
| ip_stripoptions(m, (struct mbuf *)0); | |
| iphlen = sizeof(struct ip); | |
| } | |
| /* | |
| * Get IP and UDP header together in first mbuf. | |
| */ | |
| ip = mtod(m, struct ip *); | |
| uh = (struct udphdr *)((caddr_t)ip + iphlen); | |
| /* | |
| * Make mbuf data length reflect UDP length. | |
| * If not enough data to reflect UDP length, drop. | |
| */ | |
| len = ntohs((uint16_t)uh->uh_ulen); | |
| if (ip->ip_len != len) { | |
| if (len > ip->ip_len) { | |
| goto bad; | |
| } | |
| m_adj(m, len - ip->ip_len); | |
| ip->ip_len = len; | |
| } | |
| /* | |
| * Save a copy of the IP header in case we want restore it | |
| * for sending an ICMP error message in response. | |
| */ | |
| save_ip = *ip; | |
| save_ip.ip_len+= iphlen; /* tcp_input subtracts this */ | |
| /* | |
| * Checksum extended UDP header and data. | |
| */ | |
| if (uh->uh_sum) { | |
| memset(&((struct ipovly *)ip)->ih_mbuf, 0, sizeof(struct mbuf_ptr)); | |
| ((struct ipovly *)ip)->ih_x1 = 0; | |
| ((struct ipovly *)ip)->ih_len = uh->uh_ulen; | |
| if(cksum(m, len + sizeof(struct ip))) { | |
| goto bad; | |
| } | |
| } | |
| /* | |
| * handle DHCP/BOOTP | |
| */ | |
| if (ntohs(uh->uh_dport) == BOOTP_SERVER && | |
| (ip->ip_dst.s_addr == slirp->vhost_addr.s_addr || | |
| ip->ip_dst.s_addr == 0xffffffff)) { | |
| bootp_input(m); | |
| goto bad; | |
| } | |
| /* | |
| * handle TFTP | |
| */ | |
| if (ntohs(uh->uh_dport) == TFTP_SERVER && | |
| ip->ip_dst.s_addr == slirp->vhost_addr.s_addr) { | |
| tftp_input(m); | |
| goto bad; | |
| } | |
| if (slirp->restricted) { | |
| goto bad; | |
| } | |
| /* | |
| * Locate pcb for datagram. | |
| */ | |
| so = slirp->udp_last_so; | |
| if (so == &slirp->udb || so->so_lport != uh->uh_sport || | |
| so->so_laddr.s_addr != ip->ip_src.s_addr) { | |
| struct socket *tmp; | |
| for (tmp = slirp->udb.so_next; tmp != &slirp->udb; | |
| tmp = tmp->so_next) { | |
| if (tmp->so_lport == uh->uh_sport && | |
| tmp->so_laddr.s_addr == ip->ip_src.s_addr) { | |
| so = tmp; | |
| break; | |
| } | |
| } | |
| if (tmp == &slirp->udb) { | |
| so = NULL; | |
| } else { | |
| slirp->udp_last_so = so; | |
| } | |
| } | |
| if (so == NULL) { | |
| /* | |
| * If there's no socket for this packet, | |
| * create one | |
| */ | |
| so = socreate(slirp); | |
| if (!so) { | |
| goto bad; | |
| } | |
| if(udp_attach(so) == -1) { | |
| DEBUG_MISC(" udp_attach errno = %d-%s\n", | |
| errno,strerror(errno)); | |
| sofree(so); | |
| goto bad; | |
| } | |
| /* | |
| * Setup fields | |
| */ | |
| so->so_laddr = ip->ip_src; | |
| so->so_lport = uh->uh_sport; | |
| if ((so->so_iptos = udp_tos(so)) == 0) | |
| so->so_iptos = ip->ip_tos; | |
| /* | |
| * XXXXX Here, check if it's in udpexec_list, | |
| * and if it is, do the fork_exec() etc. | |
| */ | |
| } | |
| so->so_faddr = ip->ip_dst; /* XXX */ | |
| so->so_fport = uh->uh_dport; /* XXX */ | |
| iphlen += sizeof(struct udphdr); | |
| m->m_len -= iphlen; | |
| m->m_data += iphlen; | |
| /* | |
| * Now we sendto() the packet. | |
| */ | |
| if(sosendto(so,m) == -1) { | |
| m->m_len += iphlen; | |
| m->m_data -= iphlen; | |
| *ip=save_ip; | |
| DEBUG_MISC("udp tx errno = %d-%s\n",errno,strerror(errno)); | |
| icmp_error(m, ICMP_UNREACH,ICMP_UNREACH_NET, 0,strerror(errno)); | |
| } | |
| m_free(so->so_m); /* used for ICMP if error on sorecvfrom */ | |
| /* restore the orig mbuf packet */ | |
| m->m_len += iphlen; | |
| m->m_data -= iphlen; | |
| *ip=save_ip; | |
| so->so_m=m; /* ICMP backup */ | |
| return; | |
| bad: | |
| m_free(m); | |
| } | |
| int udp_output2(struct socket *so, struct mbuf *m, | |
| struct sockaddr_in *saddr, struct sockaddr_in *daddr, | |
| int iptos) | |
| { | |
| register struct udpiphdr *ui; | |
| int error = 0; | |
| DEBUG_CALL("udp_output"); | |
| DEBUG_ARG("so = %lx", (long)so); | |
| DEBUG_ARG("m = %lx", (long)m); | |
| DEBUG_ARG("saddr = %lx", (long)saddr->sin_addr.s_addr); | |
| DEBUG_ARG("daddr = %lx", (long)daddr->sin_addr.s_addr); | |
| /* | |
| * Adjust for header | |
| */ | |
| m->m_data -= sizeof(struct udpiphdr); | |
| m->m_len += sizeof(struct udpiphdr); | |
| /* | |
| * Fill in mbuf with extended UDP header | |
| * and addresses and length put into network format. | |
| */ | |
| ui = mtod(m, struct udpiphdr *); | |
| memset(&ui->ui_i.ih_mbuf, 0 , sizeof(struct mbuf_ptr)); | |
| ui->ui_x1 = 0; | |
| ui->ui_pr = IPPROTO_UDP; | |
| ui->ui_len = htons(m->m_len - sizeof(struct ip)); | |
| /* XXXXX Check for from-one-location sockets, or from-any-location sockets */ | |
| ui->ui_src = saddr->sin_addr; | |
| ui->ui_dst = daddr->sin_addr; | |
| ui->ui_sport = saddr->sin_port; | |
| ui->ui_dport = daddr->sin_port; | |
| ui->ui_ulen = ui->ui_len; | |
| /* | |
| * Stuff checksum and output datagram. | |
| */ | |
| ui->ui_sum = 0; | |
| if ((ui->ui_sum = cksum(m, m->m_len)) == 0) | |
| ui->ui_sum = 0xffff; | |
| ((struct ip *)ui)->ip_len = m->m_len; | |
| ((struct ip *)ui)->ip_ttl = IPDEFTTL; | |
| ((struct ip *)ui)->ip_tos = iptos; | |
| error = ip_output(so, m); | |
| return (error); | |
| } | |
| int udp_output(struct socket *so, struct mbuf *m, | |
| struct sockaddr_in *addr) | |
| { | |
| Slirp *slirp = so->slirp; | |
| struct sockaddr_in saddr, daddr; | |
| saddr = *addr; | |
| if ((so->so_faddr.s_addr & slirp->vnetwork_mask.s_addr) == | |
| slirp->vnetwork_addr.s_addr) { | |
| uint32_t inv_mask = ~slirp->vnetwork_mask.s_addr; | |
| if ((so->so_faddr.s_addr & inv_mask) == inv_mask) { | |
| saddr.sin_addr = slirp->vhost_addr; | |
| } else if (addr->sin_addr.s_addr == loopback_addr.s_addr || | |
| so->so_faddr.s_addr != slirp->vhost_addr.s_addr) { | |
| saddr.sin_addr = so->so_faddr; | |
| } | |
| } | |
| daddr.sin_addr = so->so_laddr; | |
| daddr.sin_port = so->so_lport; | |
| return udp_output2(so, m, &saddr, &daddr, so->so_iptos); | |
| } | |
| int | |
| udp_attach(struct socket *so) | |
| { | |
| if((so->s = qemu_socket(AF_INET,SOCK_DGRAM,0)) != -1) { | |
| so->so_expire = curtime + SO_EXPIRE; | |
| insque(so, &so->slirp->udb); | |
| } | |
| return(so->s); | |
| } | |
| void | |
| udp_detach(struct socket *so) | |
| { | |
| closesocket(so->s); | |
| sofree(so); | |
| } | |
| static const struct tos_t udptos[] = { | |
| {0, 53, IPTOS_LOWDELAY, 0}, /* DNS */ | |
| {0, 0, 0, 0} | |
| }; | |
| static uint8_t | |
| udp_tos(struct socket *so) | |
| { | |
| int i = 0; | |
| while(udptos[i].tos) { | |
| if ((udptos[i].fport && ntohs(so->so_fport) == udptos[i].fport) || | |
| (udptos[i].lport && ntohs(so->so_lport) == udptos[i].lport)) { | |
| so->so_emu = udptos[i].emu; | |
| return udptos[i].tos; | |
| } | |
| i++; | |
| } | |
| return 0; | |
| } | |
| struct socket * | |
| udp_listen(Slirp *slirp, uint32_t haddr, u_int hport, uint32_t laddr, | |
| u_int lport, int flags) | |
| { | |
| struct sockaddr_in addr; | |
| struct socket *so; | |
| socklen_t addrlen = sizeof(struct sockaddr_in); | |
| so = socreate(slirp); | |
| if (!so) { | |
| return NULL; | |
| } | |
| so->s = qemu_socket(AF_INET,SOCK_DGRAM,0); | |
| so->so_expire = curtime + SO_EXPIRE; | |
| insque(so, &slirp->udb); | |
| addr.sin_family = AF_INET; | |
| addr.sin_addr.s_addr = haddr; | |
| addr.sin_port = hport; | |
| if (bind(so->s,(struct sockaddr *)&addr, addrlen) < 0) { | |
| udp_detach(so); | |
| return NULL; | |
| } | |
| socket_set_fast_reuse(so->s); | |
| getsockname(so->s,(struct sockaddr *)&addr,&addrlen); | |
| so->so_fport = addr.sin_port; | |
| if (addr.sin_addr.s_addr == 0 || | |
| addr.sin_addr.s_addr == loopback_addr.s_addr) { | |
| so->so_faddr = slirp->vhost_addr; | |
| } else { | |
| so->so_faddr = addr.sin_addr; | |
| } | |
| so->so_lport = lport; | |
| so->so_laddr.s_addr = laddr; | |
| if (flags != SS_FACCEPTONCE) | |
| so->so_expire = 0; | |
| so->so_state &= SS_PERSISTENT_MASK; | |
| so->so_state |= SS_ISFCONNECTED | flags; | |
| return so; | |
| } |