#include <sys/param.h>
#include <sys/kernel.h>
#include <sys/malloc.h>
#include <sys/mbuf.h>
#include <sys/protosw.h>
#include <sys/socket.h>
#include <sys/socketvar.h>
#include <sys/sysctl.h>
#include <sys/errno.h>
#include <sys/stat.h>
#include <sys/systm.h>
#include <sys/syslog.h>
#include <sys/proc.h>
#include <sys/kauth.h>
#include <net/if.h>
#include <net/route.h>
#include <net/if_types.h>
#include <net/ntstat.h>
#include <netinet/in.h>
#include <netinet/in_systm.h>
#include <netinet/ip.h>
#include <netinet/in_pcb.h>
#include <netinet/in_var.h>
#include <netinet/ip_var.h>
#include <netinet/udp.h>
#include <netinet/udp_var.h>
#include <netinet/ip6.h>
#include <netinet6/ip6_var.h>
#include <netinet6/in6_pcb.h>
#include <netinet/icmp6.h>
#include <netinet6/udp6_var.h>
#include <netinet6/ip6protosw.h>
#if IPSEC
#include <netinet6/ipsec.h>
#include <netinet6/ipsec6.h>
extern int ipsec_bypass;
#endif
extern struct protosw inetsw[];
static int udp6_detach(struct socket *so);
static void udp6_append(struct inpcb *, struct ip6_hdr *,
struct sockaddr_in6 *, struct mbuf *, int);
extern void ipfwsyslog( int level, const char *format,...);
extern int fw_verbose;
#if IPFIREWALL
#define log_in_vain_log( a ) { \
if ( (log_in_vain == 3 ) && (fw_verbose == 2)) { \
ipfwsyslog a ; \
} \
else log a ; \
}
#else
#define log_in_vain_log( a ) { log a; }
#endif
static void
udp6_append(struct inpcb *last, __unused struct ip6_hdr *ip6,
struct sockaddr_in6 *udp_in6, struct mbuf *n, int off)
{
struct mbuf *opts = NULL;
int ret = 0;
#if CONFIG_MACF_NET
if (mac_inpcb_check_deliver(last, n, AF_INET6, SOCK_DGRAM) != 0) {
m_freem(n);
return;
}
#endif
if ((last->in6p_flags & IN6P_CONTROLOPTS) != 0 ||
(last->in6p_socket->so_options & SO_TIMESTAMP) != 0 ||
(last->in6p_socket->so_options & SO_TIMESTAMP_MONOTONIC) != 0) {
ret = ip6_savecontrol(last, n, &opts);
if (ret != 0) {
m_freem(n);
m_freem(opts);
return;
}
}
m_adj(n, off);
if (nstat_collect) {
locked_add_64(&last->inp_stat->rxpackets, 1);
locked_add_64(&last->inp_stat->rxbytes, n->m_pkthdr.len);
}
so_recv_data_stat(last->in6p_socket, n, 0);
if (sbappendaddr(&last->in6p_socket->so_rcv,
(struct sockaddr *)udp_in6, n, opts, NULL) == 0)
udpstat.udps_fullsock++;
else
sorwakeup(last->in6p_socket);
}
int
udp6_input(
struct mbuf **mp,
int *offp,
int proto)
{
#pragma unused(proto)
struct mbuf *m = *mp;
struct ifnet *ifp;
register struct ip6_hdr *ip6;
register struct udphdr *uh;
register struct inpcb *in6p;
struct mbuf *opts = NULL;
int off = *offp;
int plen, ulen, ret = 0;
struct sockaddr_in6 udp_in6;
struct inpcbinfo *pcbinfo = &udbinfo;
struct sockaddr_in6 fromsa;
IP6_EXTHDR_CHECK(m, off, sizeof(struct udphdr), return IPPROTO_DONE);
ifp = m->m_pkthdr.rcvif;
ip6 = mtod(m, struct ip6_hdr *);
#if defined(NFAITH) && 0 < NFAITH
if (faithprefix(&ip6->ip6_dst)) {
m_freem(m);
return IPPROTO_DONE;
}
#endif
udpstat.udps_ipackets++;
plen = ntohs(ip6->ip6_plen) - off + sizeof(*ip6);
uh = (struct udphdr *)((caddr_t)ip6 + off);
ulen = ntohs((u_short)uh->uh_ulen);
if (plen != ulen) {
udpstat.udps_badlen++;
goto bad;
}
if (uh->uh_dport == 0)
goto bad;
if (uh->uh_sum) {
if ((apple_hwcksum_rx != 0) && (m->m_pkthdr.csum_flags & CSUM_DATA_VALID)) {
uh->uh_sum = m->m_pkthdr.csum_data;
uh->uh_sum ^= 0xffff;
}
else {
if (in6_cksum(m, IPPROTO_UDP, off, ulen) != 0) {
udpstat.udps_badsum++;
goto bad;
}
}
}
#ifndef __APPLE__
else
udpstat.udps_nosum++;
#endif
init_sin6(&fromsa, m);
fromsa.sin6_port = uh->uh_sport;
if (IN6_IS_ADDR_MULTICAST(&ip6->ip6_dst)) {
int reuse_sock = 0, mcast_delivered = 0;
struct ip6_moptions *imo;
struct mbuf *n = NULL;
init_sin6(&udp_in6, m);
udp_in6.sin6_port = uh->uh_sport;
lck_rw_lock_shared(pcbinfo->mtx);
LIST_FOREACH(in6p, &udb, inp_list) {
if ((in6p->inp_vflag & INP_IPV6) == 0)
continue;
if (in_pcb_checkstate(in6p, WNT_ACQUIRE, 0) == WNT_STOPUSING)
continue;
udp_lock(in6p->in6p_socket, 1, 0);
if (in_pcb_checkstate(in6p, WNT_RELEASE, 1) == WNT_STOPUSING) {
udp_unlock(in6p->in6p_socket, 1, 0);
continue;
}
if (in6p->in6p_lport != uh->uh_dport) {
udp_unlock(in6p->in6p_socket, 1, 0);
continue;
}
imo = in6p->in6p_moptions;
if (imo && IN6_IS_ADDR_MULTICAST(&ip6->ip6_dst)) {
struct sockaddr_in6 mcaddr;
int blocked;
IM6O_LOCK(imo);
bzero(&mcaddr, sizeof(struct sockaddr_in6));
mcaddr.sin6_len = sizeof(struct sockaddr_in6);
mcaddr.sin6_family = AF_INET6;
mcaddr.sin6_addr = ip6->ip6_dst;
blocked = im6o_mc_filter(imo, ifp,
(struct sockaddr *)&mcaddr,
(struct sockaddr *)&fromsa);
IM6O_UNLOCK(imo);
if (blocked != MCAST_PASS) {
udp_unlock(in6p->in6p_socket, 1, 0);
continue;
}
}
if (!IN6_IS_ADDR_UNSPECIFIED(&in6p->in6p_faddr)) {
if (!IN6_ARE_ADDR_EQUAL(&in6p->in6p_faddr,
&ip6->ip6_src) ||
in6p->in6p_fport != uh->uh_sport) {
udp_unlock(in6p->in6p_socket, 1, 0);
continue;
}
}
reuse_sock = in6p->inp_socket->so_options &
(SO_REUSEPORT | SO_REUSEADDR);
{
#if IPSEC
int skipit = 0;
if (ipsec_bypass == 0) {
if (ipsec6_in_reject_so(m, in6p->inp_socket)) {
IPSEC_STAT_INCREMENT(ipsec6stat.in_polvio);
skipit = 1;
}
}
if (skipit == 0)
#endif
{
if (reuse_sock)
n = m_copy(m, 0, M_COPYALL);
udp6_append(in6p, ip6, &udp_in6, m,
off + sizeof (struct udphdr));
mcast_delivered++;
}
udp_unlock(in6p->in6p_socket, 1, 0);
}
if (reuse_sock == 0 || ((m = n) == NULL))
break;
ip6 = mtod(m, struct ip6_hdr *);
uh = (struct udphdr *)((caddr_t)ip6 + off);
}
lck_rw_done(pcbinfo->mtx);
if (mcast_delivered == 0) {
udpstat.udps_noport++;
#ifndef __APPLE__
udpstat.udps_noportmcast++;
#endif
goto bad;
}
if (reuse_sock != 0)
m_freem(m);
return IPPROTO_DONE;
}
in6p = in6_pcblookup_hash(&udbinfo, &ip6->ip6_src, uh->uh_sport,
&ip6->ip6_dst, uh->uh_dport, 1,
m->m_pkthdr.rcvif);
if (in6p == 0) {
if (log_in_vain) {
char buf[INET6_ADDRSTRLEN];
strlcpy(buf, ip6_sprintf(&ip6->ip6_dst), sizeof(buf));
if (log_in_vain != 3)
log(LOG_INFO,
"Connection attempt to UDP %s:%d from %s:%d\n",
buf, ntohs(uh->uh_dport),
ip6_sprintf(&ip6->ip6_src), ntohs(uh->uh_sport));
else if (!(m->m_flags & (M_BCAST | M_MCAST)) &&
!IN6_ARE_ADDR_EQUAL(&ip6->ip6_dst, &ip6->ip6_src))
log_in_vain_log((LOG_INFO,
"Connection attempt to UDP %s:%d from %s:%d\n",
buf, ntohs(uh->uh_dport),
ip6_sprintf(&ip6->ip6_src), ntohs(uh->uh_sport)));
}
udpstat.udps_noport++;
if (m->m_flags & M_MCAST) {
printf("UDP6: M_MCAST is set in a unicast packet.\n");
#ifndef __APPLE__
udpstat.udps_noportmcast++;
#endif
goto bad;
}
icmp6_error(m, ICMP6_DST_UNREACH, ICMP6_DST_UNREACH_NOPORT, 0);
return IPPROTO_DONE;
}
#if IPSEC
if (ipsec_bypass == 0) {
if (ipsec6_in_reject_so(m, in6p->in6p_socket)) {
IPSEC_STAT_INCREMENT(ipsec6stat.in_polvio);
in_pcb_checkstate(in6p, WNT_RELEASE, 0);
goto bad;
}
}
#endif
udp_lock(in6p->in6p_socket, 1, 0);
if (in_pcb_checkstate(in6p, WNT_RELEASE, 1) == WNT_STOPUSING) {
udp_unlock(in6p->in6p_socket, 1, 0);
goto bad;
}
init_sin6(&udp_in6, m);
udp_in6.sin6_port = uh->uh_sport;
if ((in6p->in6p_flags & IN6P_CONTROLOPTS) != 0 ||
(in6p->in6p_socket->so_options & SO_TIMESTAMP) != 0 ||
(in6p->in6p_socket->so_options & SO_TIMESTAMP_MONOTONIC) != 0) {
ret = ip6_savecontrol(in6p, m, &opts);
if (ret != 0) {
udp_unlock(in6p->in6p_socket, 1, 0);
goto bad;
}
}
m_adj(m, off + sizeof(struct udphdr));
if (nstat_collect) {
locked_add_64(&in6p->inp_stat->rxpackets, 1);
locked_add_64(&in6p->inp_stat->rxbytes, m->m_pkthdr.len);
}
so_recv_data_stat(in6p->in6p_socket, m, 0);
if (sbappendaddr(&in6p->in6p_socket->so_rcv,
(struct sockaddr *)&udp_in6,
m, opts, NULL) == 0) {
m = NULL;
opts = NULL;
udpstat.udps_fullsock++;
udp_unlock(in6p->in6p_socket, 1, 0);
goto bad;
}
sorwakeup(in6p->in6p_socket);
udp_unlock(in6p->in6p_socket, 1, 0);
return IPPROTO_DONE;
bad:
if (m)
m_freem(m);
if (opts)
m_freem(opts);
return IPPROTO_DONE;
}
void
udp6_ctlinput(
int cmd,
struct sockaddr *sa,
void *d)
{
struct udphdr uh;
struct ip6_hdr *ip6;
struct mbuf *m;
int off = 0;
struct ip6ctlparam *ip6cp = NULL;
const struct sockaddr_in6 *sa6_src = NULL;
void (*notify)(struct inpcb *, int) = udp_notify;
struct udp_portonly {
u_int16_t uh_sport;
u_int16_t uh_dport;
} *uhp;
if (sa->sa_family != AF_INET6 ||
sa->sa_len != sizeof(struct sockaddr_in6))
return;
if ((unsigned)cmd >= PRC_NCMDS)
return;
if (PRC_IS_REDIRECT(cmd))
notify = in6_rtchange, d = NULL;
else if (cmd == PRC_HOSTDEAD)
d = NULL;
else if (inet6ctlerrmap[cmd] == 0)
return;
if (d != NULL) {
ip6cp = (struct ip6ctlparam *)d;
m = ip6cp->ip6c_m;
ip6 = ip6cp->ip6c_ip6;
off = ip6cp->ip6c_off;
sa6_src = ip6cp->ip6c_src;
} else {
m = NULL;
ip6 = NULL;
sa6_src = &sa6_any;
}
if (ip6) {
if (m->m_pkthdr.len < off + sizeof(*uhp))
return;
bzero(&uh, sizeof(uh));
m_copydata(m, off, sizeof(*uhp), (caddr_t)&uh);
(void) in6_pcbnotify(&udbinfo, sa, uh.uh_dport,
(struct sockaddr*)ip6cp->ip6c_src,
uh.uh_sport, cmd, NULL, notify);
} else
(void) in6_pcbnotify(&udbinfo, sa, 0, (struct sockaddr *)&sa6_src,
0, cmd, NULL, notify);
}
#ifndef __APPLE__
static int
udp6_getcred SYSCTL_HANDLER_ARGS
{
struct sockaddr_in6 addrs[2];
struct inpcb *inp;
int error, s;
error = proc_suser(req->p);
if (error)
return (error);
if (req->newlen != sizeof(addrs))
return (EINVAL);
if (req->oldlen != sizeof(*(kauth_cred_t)0))
return (EINVAL);
error = SYSCTL_IN(req, addrs, sizeof(addrs));
if (error)
return (error);
s = splnet();
inp = in6_pcblookup_hash(&udbinfo, &addrs[1].sin6_addr,
addrs[1].sin6_port,
&addrs[0].sin6_addr, addrs[0].sin6_port,
1, NULL);
if (!inp || !inp->inp_socket || !inp->inp_socket->so_cred) {
error = ENOENT;
goto out;
}
error = SYSCTL_OUT(req, inp->inp_socket->so_cred->pc_ucred,
sizeof(*(kauth_cred_t)0));
out:
splx(s);
return (error);
}
SYSCTL_PROC(_net_inet6_udp6, OID_AUTO, getcred, CTLTYPE_OPAQUE|CTLFLAG_RW,
0, 0,
udp6_getcred, "S,ucred", "Get the ucred of a UDP6 connection");
#endif
static int
udp6_abort(struct socket *so)
{
struct inpcb *inp;
inp = sotoinpcb(so);
if (inp == 0)
return EINVAL;
soisdisconnected(so);
in6_pcbdetach(inp);
return 0;
}
static int
udp6_attach(struct socket *so, __unused int proto, struct proc *p)
{
struct inpcb *inp;
int error;
inp = sotoinpcb(so);
if (inp != 0)
return EINVAL;
error = in_pcballoc(so, &udbinfo, p);
if (error)
return error;
if (so->so_snd.sb_hiwat == 0 || so->so_rcv.sb_hiwat == 0) {
error = soreserve(so, udp_sendspace, udp_recvspace);
if (error)
return error;
}
inp = (struct inpcb *)so->so_pcb;
inp->inp_vflag |= INP_IPV6;
if (ip6_mapped_addr_on)
inp->inp_vflag |= INP_IPV4;
inp->in6p_hops = -1;
inp->in6p_cksum = -1;
inp->inp_ip_ttl = ip_defttl;
nstat_udp_new_pcb(inp);
return 0;
}
static int
udp6_bind(struct socket *so, struct sockaddr *nam, struct proc *p)
{
struct inpcb *inp;
int error;
inp = sotoinpcb(so);
if (inp == 0)
return EINVAL;
inp->inp_vflag &= ~INP_IPV4;
inp->inp_vflag |= INP_IPV6;
if ((inp->inp_flags & IN6P_IPV6_V6ONLY) == 0) {
struct sockaddr_in6 *sin6_p;
sin6_p = (struct sockaddr_in6 *)nam;
if (IN6_IS_ADDR_UNSPECIFIED(&sin6_p->sin6_addr))
inp->inp_vflag |= INP_IPV4;
else if (IN6_IS_ADDR_V4MAPPED(&sin6_p->sin6_addr)) {
struct sockaddr_in sin;
in6_sin6_2_sin(&sin, sin6_p);
inp->inp_vflag |= INP_IPV4;
inp->inp_vflag &= ~INP_IPV6;
error = in_pcbbind(inp, (struct sockaddr *)&sin, p);
return error;
}
}
error = in6_pcbbind(inp, nam, p);
return error;
}
static int
udp6_connect(struct socket *so, struct sockaddr *nam, struct proc *p)
{
struct inpcb *inp;
int error;
inp = sotoinpcb(so);
if (inp == 0)
return EINVAL;
if ((inp->inp_flags & IN6P_IPV6_V6ONLY) == 0) {
struct sockaddr_in6 *sin6_p;
sin6_p = (struct sockaddr_in6 *)nam;
if (IN6_IS_ADDR_V4MAPPED(&sin6_p->sin6_addr)) {
struct sockaddr_in sin;
if (inp->inp_faddr.s_addr != INADDR_ANY)
return EISCONN;
in6_sin6_2_sin(&sin, sin6_p);
error = in_pcbconnect(inp, (struct sockaddr *)&sin, p, NULL);
if (error == 0) {
inp->inp_vflag |= INP_IPV4;
inp->inp_vflag &= ~INP_IPV6;
soisconnected(so);
}
return error;
}
}
if (!IN6_IS_ADDR_UNSPECIFIED(&inp->in6p_faddr))
return EISCONN;
error = in6_pcbconnect(inp, nam, p);
if (error == 0) {
if (ip6_mapped_addr_on || (inp->inp_flags & IN6P_IPV6_V6ONLY) == 0) {
inp->inp_vflag &= ~INP_IPV4;
inp->inp_vflag |= INP_IPV6;
}
soisconnected(so);
}
return error;
}
static int
udp6_detach(struct socket *so)
{
struct inpcb *inp;
inp = sotoinpcb(so);
if (inp == 0)
return EINVAL;
in6_pcbdetach(inp);
return 0;
}
static int
udp6_disconnect(struct socket *so)
{
struct inpcb *inp;
inp = sotoinpcb(so);
if (inp == 0)
return EINVAL;
if (inp->inp_vflag & INP_IPV4) {
struct pr_usrreqs *pru;
pru = ip_protox[IPPROTO_UDP]->pr_usrreqs;
return ((*pru->pru_disconnect)(so));
}
if (IN6_IS_ADDR_UNSPECIFIED(&inp->in6p_faddr))
return ENOTCONN;
in6_pcbdisconnect(inp);
inp->in6p_laddr = in6addr_any;
inp->in6p_last_outif = 0;
so->so_state &= ~SS_ISCONNECTED;
return 0;
}
static int
udp6_send(struct socket *so, int flags, struct mbuf *m, struct sockaddr *addr,
struct mbuf *control, struct proc *p)
{
struct inpcb *inp;
int error = 0;
inp = sotoinpcb(so);
if (inp == 0) {
error = EINVAL;
goto bad;
}
if (addr) {
if (addr->sa_len != sizeof(struct sockaddr_in6)) {
error = EINVAL;
goto bad;
}
if (addr->sa_family != AF_INET6) {
error = EAFNOSUPPORT;
goto bad;
}
}
if (ip6_mapped_addr_on || (inp->inp_flags & IN6P_IPV6_V6ONLY) == 0) {
int hasv4addr;
struct sockaddr_in6 *sin6 = 0;
if (addr == 0)
hasv4addr = (inp->inp_vflag & INP_IPV4);
else {
sin6 = (struct sockaddr_in6 *)addr;
hasv4addr = IN6_IS_ADDR_V4MAPPED(&sin6->sin6_addr)
? 1 : 0;
}
if (hasv4addr) {
struct pr_usrreqs *pru;
if (sin6)
in6_sin6_2_sin_in_sock(addr);
pru = ip_protox[IPPROTO_UDP]->pr_usrreqs;
error = ((*pru->pru_send)(so, flags, m, addr, control,
p));
return error;
}
}
return udp6_output(inp, m, addr, control, p);
bad:
m_freem(m);
return(error);
}
struct pr_usrreqs udp6_usrreqs = {
udp6_abort, pru_accept_notsupp, udp6_attach, udp6_bind, udp6_connect,
pru_connect2_notsupp, in6_control, udp6_detach, udp6_disconnect,
pru_listen_notsupp, in6_mapped_peeraddr, pru_rcvd_notsupp,
pru_rcvoob_notsupp, udp6_send, pru_sense_null, udp_shutdown,
in6_mapped_sockaddr, sosend, soreceive, pru_sopoll_notsupp
};