#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 <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 lck_mtx_t *nd6_mutex;
extern struct protosw inetsw[];
static int in6_mcmatch(struct inpcb *, struct in6_addr *, struct ifnet *);
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 int
in6_mcmatch(
struct inpcb *in6p,
register struct in6_addr *ia6,
struct ifnet *ifp)
{
struct ip6_moptions *im6o = in6p->in6p_moptions;
struct in6_multi_mship *imm;
if (im6o == NULL)
return 0;
lck_mtx_lock(nd6_mutex);
for (imm = im6o->im6o_memberships.lh_first; imm != NULL;
imm = imm->i6mm_chain.le_next) {
if ((ifp == NULL ||
imm->i6mm_maddr->in6m_ifp == ifp) &&
IN6_ARE_ADDR_EQUAL(&imm->i6mm_maddr->in6m_addr,
ia6)) {
lck_mtx_unlock(nd6_mutex);
return 1;
}
}
lck_mtx_unlock(nd6_mutex);
return 0;
}
static void
udp6_append(struct inpcb *last, struct ip6_hdr *ip6,
struct sockaddr_in6 *udp_in6, struct mbuf *n, int off)
{
struct mbuf *opts = NULL;
#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 ||
last->in6p_socket->so_options & SO_TIMESTAMP)
ip6_savecontrol(last, &opts, ip6, n);
m_adj(n, off);
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)
{
struct mbuf *m = *mp;
register struct ip6_hdr *ip6;
register struct udphdr *uh;
register struct inpcb *in6p;
struct mbuf *opts = NULL;
int off = *offp;
int plen, ulen;
struct sockaddr_in6 udp_in6;
struct inpcbinfo *pcbinfo = &udbinfo;
IP6_EXTHDR_CHECK(m, off, sizeof(struct udphdr), return IPPROTO_DONE);
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;
}
#ifndef __APPLE__
if (uh->uh_sum == 0)
udpstat.udps_nosum++;
#endif
else if (in6_cksum(m, IPPROTO_UDP, off, ulen) != 0) {
udpstat.udps_badsum++;
goto bad;
}
if (IN6_IS_ADDR_MULTICAST(&ip6->ip6_dst)) {
int reuse_sock = 0, mcast_delivered = 0;
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;
}
if (!IN6_IS_ADDR_UNSPECIFIED(&in6p->in6p_laddr)) {
if (!IN6_ARE_ADDR_EQUAL(&in6p->in6p_laddr,
&ip6->ip6_dst) &&
!in6_mcmatch(in6p, &ip6->ip6_dst,
m->m_pkthdr.rcvif)) {
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;
}
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
|| in6p->in6p_socket->so_options & SO_TIMESTAMP)
ip6_savecontrol(in6p, &opts, ip6, m);
m_adj(m, off + sizeof(struct udphdr));
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, notify);
} else
(void) in6_pcbnotify(&udbinfo, sa, 0, (struct sockaddr *)&sa6_src,
0, cmd, 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;
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);
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;
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
};