#include <sys/param.h>
#include <sys/systm.h>
#include <sys/mbuf.h>
#include <sys/proc.h>
#include <sys/domain.h>
#include <sys/protosw.h>
#include <sys/socket.h>
#include <sys/socketvar.h>
#include <kern/locks.h>
#include <net/raw_cb.h>
decl_lck_mtx_data(,raw_mtx_data);
lck_mtx_t *raw_mtx = &raw_mtx_data;
lck_attr_t *raw_mtx_attr;
lck_grp_t *raw_mtx_grp;
lck_grp_attr_t *raw_mtx_grp_attr;
void
raw_init(struct protosw *pp, struct domain *dp)
{
#pragma unused(pp, dp)
static int raw_initialized = 0;
if (!raw_initialized) {
raw_initialized = 1;
raw_mtx_grp_attr = lck_grp_attr_alloc_init();
raw_mtx_grp = lck_grp_alloc_init("rawcb", raw_mtx_grp_attr);
raw_mtx_attr = lck_attr_alloc_init();
lck_mtx_init(raw_mtx, raw_mtx_grp, raw_mtx_attr);
LIST_INIT(&rawcb_list);
}
}
void
raw_input(struct mbuf *m0, struct sockproto *proto, struct sockaddr *src,
struct sockaddr *dst)
{
struct rawcb *rp;
struct mbuf *m = m0;
int sockets = 0;
struct socket *last;
int error;
last = NULL;
lck_mtx_lock(raw_mtx);
LIST_FOREACH(rp, &rawcb_list, list) {
if (rp->rcb_proto.sp_family != proto->sp_family)
continue;
if (rp->rcb_proto.sp_protocol &&
rp->rcb_proto.sp_protocol != proto->sp_protocol)
continue;
#define equal(a1, a2) \
(bcmp((caddr_t)(a1), (caddr_t)(a2), a1->sa_len) == 0)
if (rp->rcb_laddr && !equal(rp->rcb_laddr, dst))
continue;
if (rp->rcb_faddr && !equal(rp->rcb_faddr, src))
continue;
if (last) {
struct mbuf *n;
n = m_copy(m, 0, (int)M_COPYALL);
if (n) {
socket_lock(last, 1);
if (sbappendaddr(&last->so_rcv, src,
n, (struct mbuf *)0, &error) != 0) {
sorwakeup(last);
sockets++;
}
socket_unlock(last, 1);
}
}
last = rp->rcb_socket;
}
if (last) {
socket_lock(last, 1);
if (sbappendaddr(&last->so_rcv, src,
m, (struct mbuf *)0, &error) != 0) {
sorwakeup(last);
sockets++;
}
socket_unlock(last, 1);
} else
m_freem(m);
lck_mtx_unlock(raw_mtx);
}
void
raw_ctlinput(int cmd, __unused struct sockaddr *arg, __unused void *dummy)
{
if (cmd < 0 || cmd > PRC_NCMDS)
return;
}
static int
raw_uabort(struct socket *so)
{
struct rawcb *rp = sotorawcb(so);
lck_mtx_t * mutex_held;
if (so->so_proto->pr_getlock != NULL)
mutex_held = (*so->so_proto->pr_getlock)(so, 0);
else
mutex_held = so->so_proto->pr_domain->dom_mtx;
lck_mtx_assert(mutex_held, LCK_MTX_ASSERT_OWNED);
if (rp == 0)
return EINVAL;
raw_disconnect(rp);
sofree(so);
soisdisconnected(so);
return 0;
}
static int
raw_uattach(struct socket *so, int proto, __unused struct proc *p)
{
struct rawcb *rp = sotorawcb(so);
if (rp == 0)
return EINVAL;
if ((so->so_state & SS_PRIV) == 0)
return (EPERM);
return raw_attach(so, proto);
}
static int
raw_ubind(__unused struct socket *so, __unused struct sockaddr *nam, __unused struct proc *p)
{
return EINVAL;
}
static int
raw_uconnect(__unused struct socket *so, __unused struct sockaddr *nam, __unused struct proc *p)
{
return EINVAL;
}
static int
raw_udetach(struct socket *so)
{
struct rawcb *rp = sotorawcb(so);
lck_mtx_t * mutex_held;
if (so->so_proto->pr_getlock != NULL)
mutex_held = (*so->so_proto->pr_getlock)(so, 0);
else
mutex_held = so->so_proto->pr_domain->dom_mtx;
lck_mtx_assert(mutex_held, LCK_MTX_ASSERT_OWNED);
if (rp == 0)
return EINVAL;
raw_detach(rp);
return 0;
}
static int
raw_udisconnect(struct socket *so)
{
struct rawcb *rp = sotorawcb(so);
if (rp == 0)
return EINVAL;
if (rp->rcb_faddr == 0) {
return ENOTCONN;
}
raw_disconnect(rp);
soisdisconnected(so);
return 0;
}
static int
raw_upeeraddr(struct socket *so, struct sockaddr **nam)
{
struct rawcb *rp = sotorawcb(so);
if (rp == 0)
return EINVAL;
if (rp->rcb_faddr == 0) {
return ENOTCONN;
}
*nam = dup_sockaddr(rp->rcb_faddr, 1);
return 0;
}
static int
raw_usend(struct socket *so, int flags, struct mbuf *m,
struct sockaddr *nam, struct mbuf *control, __unused struct proc *p)
{
int error;
struct rawcb *rp = sotorawcb(so);
lck_mtx_t * mutex_held;
if (so->so_proto->pr_getlock != NULL)
mutex_held = (*so->so_proto->pr_getlock)(so, 0);
else
mutex_held = so->so_proto->pr_domain->dom_mtx;
lck_mtx_assert(mutex_held, LCK_MTX_ASSERT_OWNED);
if (rp == 0) {
error = EINVAL;
goto release;
}
if (flags & PRUS_OOB) {
error = EOPNOTSUPP;
goto release;
}
if (so->so_proto->pr_output == NULL) {
error = EOPNOTSUPP;
goto release;
}
if (control && control->m_len) {
error = EOPNOTSUPP;
goto release;
}
if (nam) {
if (rp->rcb_faddr) {
error = EISCONN;
goto release;
}
rp->rcb_faddr = nam;
} else if (rp->rcb_faddr == 0) {
error = ENOTCONN;
goto release;
}
error = (*so->so_proto->pr_output)(m, so);
m = NULL;
if (nam)
rp->rcb_faddr = NULL;
release:
if (m != NULL)
m_freem(m);
return (error);
}
static int
raw_ushutdown(struct socket *so)
{
struct rawcb *rp = sotorawcb(so);
lck_mtx_t * mutex_held;
if (so->so_proto->pr_getlock != NULL)
mutex_held = (*so->so_proto->pr_getlock)(so, 0);
else
mutex_held = so->so_proto->pr_domain->dom_mtx;
lck_mtx_assert(mutex_held, LCK_MTX_ASSERT_OWNED);
if (rp == 0)
return EINVAL;
socantsendmore(so);
return 0;
}
static int
raw_usockaddr(struct socket *so, struct sockaddr **nam)
{
struct rawcb *rp = sotorawcb(so);
if (rp == 0)
return EINVAL;
if (rp->rcb_laddr == 0)
return EINVAL;
*nam = dup_sockaddr(rp->rcb_laddr, 1);
return 0;
}
struct pr_usrreqs raw_usrreqs = {
.pru_abort = raw_uabort,
.pru_attach = raw_uattach,
.pru_bind = raw_ubind,
.pru_connect = raw_uconnect,
.pru_detach = raw_udetach,
.pru_disconnect = raw_udisconnect,
.pru_peeraddr = raw_upeeraddr,
.pru_send = raw_usend,
.pru_shutdown = raw_ushutdown,
.pru_sockaddr = raw_usockaddr,
.pru_sosend = sosend,
.pru_soreceive = soreceive,
};