version 1.4, 1993/08/03 01:36:10 |
version 1.17, 1994/10/30 21:48:07 |
|
|
|
/* $NetBSD$ */ |
|
|
/* |
/* |
* Copyright (c) 1982, 1986, 1988, 1990 Regents of the University of California. |
* Copyright (c) 1982, 1986, 1988, 1990, 1993 |
* All rights reserved. |
* The Regents of the University of California. All rights reserved. |
* |
* |
* Redistribution and use in source and binary forms, with or without |
* Redistribution and use in source and binary forms, with or without |
* modification, are permitted provided that the following conditions |
* modification, are permitted provided that the following conditions |
|
|
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF |
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF |
* SUCH DAMAGE. |
* SUCH DAMAGE. |
* |
* |
* from: @(#)uipc_socket.c 7.28 (Berkeley) 5/4/91 |
* @(#)uipc_socket.c 8.3 (Berkeley) 4/15/94 |
* $Id$ |
|
*/ |
*/ |
|
|
#include "param.h" |
#include <sys/param.h> |
#include "systm.h" |
#include <sys/systm.h> |
#include "proc.h" |
#include <sys/proc.h> |
#include "file.h" |
#include <sys/file.h> |
#include "malloc.h" |
#include <sys/malloc.h> |
#include "mbuf.h" |
#include <sys/mbuf.h> |
#include "domain.h" |
#include <sys/domain.h> |
#include "kernel.h" |
#include <sys/kernel.h> |
#include "select.h" |
#include <sys/protosw.h> |
#include "protosw.h" |
#include <sys/socket.h> |
#include "socket.h" |
#include <sys/socketvar.h> |
#include "socketvar.h" |
#include <sys/resourcevar.h> |
#include "resourcevar.h" |
|
|
|
/* |
/* |
* Socket operation routines. |
* Socket operation routines. |
|
|
/*ARGSUSED*/ |
/*ARGSUSED*/ |
int |
int |
socreate(dom, aso, type, proto) |
socreate(dom, aso, type, proto) |
|
int dom; |
struct socket **aso; |
struct socket **aso; |
register int type; |
register int type; |
int proto; |
int proto; |
Line 71 socreate(dom, aso, type, proto) |
|
Line 72 socreate(dom, aso, type, proto) |
|
prp = pffindproto(dom, proto, type); |
prp = pffindproto(dom, proto, type); |
else |
else |
prp = pffindtype(dom, type); |
prp = pffindtype(dom, type); |
if (prp == 0 || || !prp->pr_usrreq) |
if (prp == 0 || prp->pr_usrreq == 0) |
return (EPROTONOSUPPORT); |
return (EPROTONOSUPPORT); |
if (prp->pr_type != type) |
if (prp->pr_type != type) |
return (EPROTOTYPE); |
return (EPROTOTYPE); |
Line 82 socreate(dom, aso, type, proto) |
|
Line 83 socreate(dom, aso, type, proto) |
|
so->so_state = SS_PRIV; |
so->so_state = SS_PRIV; |
so->so_proto = prp; |
so->so_proto = prp; |
error = |
error = |
(*prp->pr_usrreq)(so, PRU_ATTACH, |
(*prp->pr_usrreq)(so, PRU_ATTACH, (struct mbuf *)0, |
(struct mbuf *)0, (struct mbuf *)proto, (struct mbuf *)0); |
(struct mbuf *)(long)proto, (struct mbuf *)0); |
if (error) { |
if (error) { |
so->so_state |= SS_NOFDREF; |
so->so_state |= SS_NOFDREF; |
sofree(so); |
sofree(so); |
return (error); |
return (error); |
} |
} |
|
#ifdef COMPAT_SUNOS |
|
if (p->p_emul == EMUL_SUNOS && type == SOCK_DGRAM) |
|
so->so_options |= SO_BROADCAST; |
|
#endif |
*aso = so; |
*aso = so; |
return (0); |
return (0); |
} |
} |
|
|
return (error); |
return (error); |
} |
} |
|
|
|
#define SBLOCKWAIT(f) (((f) & MSG_DONTWAIT) ? M_NOWAIT : M_WAITOK) |
/* |
/* |
* Send on a socket. |
* Send on a socket. |
* If send must go all at once and message is larger than |
* If send must go all at once and message is larger than |
Line 332 sosend(so, addr, uio, top, control, flag |
|
Line 338 sosend(so, addr, uio, top, control, flag |
|
resid = uio->uio_resid; |
resid = uio->uio_resid; |
else |
else |
resid = top->m_pkthdr.len; |
resid = top->m_pkthdr.len; |
|
/* |
|
* In theory resid should be unsigned. |
|
* However, space must be signed, as it might be less than 0 |
|
* if we over-committed, and we must use a signed comparison |
|
* of space and resid. On the other hand, a negative resid |
|
* causes us to loop sending 0-length segments to the protocol. |
|
*/ |
|
if (resid < 0) |
|
return (EINVAL); |
dontroute = |
dontroute = |
(flags & MSG_DONTROUTE) && (so->so_options & SO_DONTROUTE) == 0 && |
(flags & MSG_DONTROUTE) && (so->so_options & SO_DONTROUTE) == 0 && |
(so->so_proto->pr_flags & PR_ATOMIC); |
(so->so_proto->pr_flags & PR_ATOMIC); |
Line 341 sosend(so, addr, uio, top, control, flag |
|
Line 356 sosend(so, addr, uio, top, control, flag |
|
#define snderr(errno) { error = errno; splx(s); goto release; } |
#define snderr(errno) { error = errno; splx(s); goto release; } |
|
|
restart: |
restart: |
if (error = sblock(&so->so_snd)) |
if (error = sblock(&so->so_snd, SBLOCKWAIT(flags))) |
goto out; |
goto out; |
do { |
do { |
s = splnet(); |
s = splnet(); |
|
|
space = sbspace(&so->so_snd); |
space = sbspace(&so->so_snd); |
if (flags & MSG_OOB) |
if (flags & MSG_OOB) |
space += 1024; |
space += 1024; |
if (space < resid + clen && |
if (atomic && resid > so->so_snd.sb_hiwat || |
|
clen > so->so_snd.sb_hiwat) |
|
snderr(EMSGSIZE); |
|
if (space < resid + clen && uio && |
(atomic || space < so->so_snd.sb_lowat || space < clen)) { |
(atomic || space < so->so_snd.sb_lowat || space < clen)) { |
if (atomic && resid > so->so_snd.sb_hiwat || |
|
clen > so->so_snd.sb_hiwat) |
|
snderr(EMSGSIZE); |
|
if (so->so_state & SS_NBIO) |
if (so->so_state & SS_NBIO) |
snderr(EWOULDBLOCK); |
snderr(EWOULDBLOCK); |
sbunlock(&so->so_snd); |
sbunlock(&so->so_snd); |
|
|
#ifdef MAPPED_MBUFS |
#ifdef MAPPED_MBUFS |
len = min(MCLBYTES, resid); |
len = min(MCLBYTES, resid); |
#else |
#else |
if (top == 0) { |
if (atomic && top == 0) { |
len = min(MCLBYTES - max_hdr, resid); |
len = min(MCLBYTES - max_hdr, resid); |
m->m_data += max_hdr; |
m->m_data += max_hdr; |
} else |
} else |
Line 488 soreceive(so, paddr, uio, mp0, controlp, |
|
Line 503 soreceive(so, paddr, uio, mp0, controlp, |
|
struct mbuf **controlp; |
struct mbuf **controlp; |
int *flagsp; |
int *flagsp; |
{ |
{ |
struct proc *p = curproc; /* XXX */ |
|
register struct mbuf *m, **mp; |
register struct mbuf *m, **mp; |
register int flags, len, error, s, offset; |
register int flags, len, error, s, offset; |
struct protosw *pr = so->so_proto; |
struct protosw *pr = so->so_proto; |
Line 507 soreceive(so, paddr, uio, mp0, controlp, |
|
Line 521 soreceive(so, paddr, uio, mp0, controlp, |
|
flags = 0; |
flags = 0; |
if (flags & MSG_OOB) { |
if (flags & MSG_OOB) { |
m = m_get(M_WAIT, MT_DATA); |
m = m_get(M_WAIT, MT_DATA); |
error = (*pr->pr_usrreq)(so, PRU_RCVOOB, |
error = (*pr->pr_usrreq)(so, PRU_RCVOOB, m, |
m, (struct mbuf *)(flags & MSG_PEEK), (struct mbuf *)0); |
(struct mbuf *)(long)(flags & MSG_PEEK), (struct mbuf *)0); |
if (error) |
if (error) |
goto bad; |
goto bad; |
do { |
do { |
|
|
(struct mbuf *)0, (struct mbuf *)0); |
(struct mbuf *)0, (struct mbuf *)0); |
|
|
restart: |
restart: |
if (error = sblock(&so->so_rcv)) |
if (error = sblock(&so->so_rcv, SBLOCKWAIT(flags))) |
return (error); |
return (error); |
s = splnet(); |
s = splnet(); |
|
|
|
|
/* |
/* |
* If we have less data than requested, block awaiting more |
* If we have less data than requested, block awaiting more |
* (subject to any timeout) if: |
* (subject to any timeout) if: |
* 1. the current count is less than the low water mark, or |
* 1. the current count is less than the low water mark, |
* 2. MSG_WAITALL is set, and it is possible to do the entire |
* 2. MSG_WAITALL is set, and it is possible to do the entire |
* receive operation at once if we block (resid <= hiwat). |
* receive operation at once if we block (resid <= hiwat), or |
|
* 3. MSG_DONTWAIT is not set. |
* If MSG_WAITALL is set but resid is larger than the receive buffer, |
* If MSG_WAITALL is set but resid is larger than the receive buffer, |
* we have to do the receive in sections, and thus risk returning |
* we have to do the receive in sections, and thus risk returning |
* a short count if a timeout or signal occurs after we start. |
* a short count if a timeout or signal occurs after we start. |
*/ |
*/ |
while (m == 0 || so->so_rcv.sb_cc < uio->uio_resid && |
if (m == 0 || ((flags & MSG_DONTWAIT) == 0 && |
|
so->so_rcv.sb_cc < uio->uio_resid) && |
(so->so_rcv.sb_cc < so->so_rcv.sb_lowat || |
(so->so_rcv.sb_cc < so->so_rcv.sb_lowat || |
((flags & MSG_WAITALL) && uio->uio_resid <= so->so_rcv.sb_hiwat)) && |
((flags & MSG_WAITALL) && uio->uio_resid <= so->so_rcv.sb_hiwat)) && |
m->m_nextpkt == 0 && (pr->pr_flags & PR_ATOMIC) == 0) { |
m->m_nextpkt == 0 && (pr->pr_flags & PR_ATOMIC) == 0) { |
|
|
#endif |
#endif |
if (so->so_error) { |
if (so->so_error) { |
if (m) |
if (m) |
break; |
goto dontblock; |
error = so->so_error; |
error = so->so_error; |
if ((flags & MSG_PEEK) == 0) |
if ((flags & MSG_PEEK) == 0) |
so->so_error = 0; |
so->so_error = 0; |
|
|
} |
} |
if (so->so_state & SS_CANTRCVMORE) { |
if (so->so_state & SS_CANTRCVMORE) { |
if (m) |
if (m) |
break; |
goto dontblock; |
else |
else |
goto release; |
goto release; |
} |
} |
|
|
} |
} |
if (uio->uio_resid == 0) |
if (uio->uio_resid == 0) |
goto release; |
goto release; |
if (so->so_state & SS_NBIO) { |
if ((so->so_state & SS_NBIO) || (flags & MSG_DONTWAIT)) { |
error = EWOULDBLOCK; |
error = EWOULDBLOCK; |
goto release; |
goto release; |
} |
} |
|
|
goto restart; |
goto restart; |
} |
} |
dontblock: |
dontblock: |
p->p_stats->p_ru.ru_msgrcv++; |
#ifdef notyet /* XXXX */ |
|
if (uio->uio_procp) |
|
uio->uio_procp->p_stats->p_ru.ru_msgrcv++; |
|
#endif |
nextrecord = m->m_nextpkt; |
nextrecord = m->m_nextpkt; |
if (pr->pr_flags & PR_ADDR) { |
if (pr->pr_flags & PR_ADDR) { |
#ifdef DIAGNOSTIC |
#ifdef DIAGNOSTIC |
|
|
so->so_state |= SS_RCVATMARK; |
so->so_state |= SS_RCVATMARK; |
break; |
break; |
} |
} |
} else |
} else { |
offset += len; |
offset += len; |
|
if (offset == so->so_oobmark) |
|
break; |
|
} |
} |
} |
if (flags & MSG_EOR) |
if (flags & MSG_EOR) |
break; |
break; |
|
|
so->so_rcv.sb_mb = nextrecord; |
so->so_rcv.sb_mb = nextrecord; |
if (pr->pr_flags & PR_WANTRCVD && so->so_pcb) |
if (pr->pr_flags & PR_WANTRCVD && so->so_pcb) |
(*pr->pr_usrreq)(so, PRU_RCVD, (struct mbuf *)0, |
(*pr->pr_usrreq)(so, PRU_RCVD, (struct mbuf *)0, |
(struct mbuf *)flags, (struct mbuf *)0, |
(struct mbuf *)(long)flags, (struct mbuf *)0, |
(struct mbuf *)0); |
(struct mbuf *)0); |
} |
} |
if (orig_resid == uio->uio_resid && orig_resid && |
if (orig_resid == uio->uio_resid && orig_resid && |
|
|
return (error); |
return (error); |
} |
} |
|
|
|
int |
soshutdown(so, how) |
soshutdown(so, how) |
register struct socket *so; |
register struct socket *so; |
register int how; |
register int how; |
Line 788 soshutdown(so, how) |
|
Line 811 soshutdown(so, how) |
|
return (0); |
return (0); |
} |
} |
|
|
|
void |
sorflush(so) |
sorflush(so) |
register struct socket *so; |
register struct socket *so; |
{ |
{ |
|
|
struct sockbuf asb; |
struct sockbuf asb; |
|
|
sb->sb_flags |= SB_NOINTR; |
sb->sb_flags |= SB_NOINTR; |
(void) sblock(sb); |
(void) sblock(sb, M_WAITOK); |
s = splimp(); |
s = splimp(); |
socantrcvmore(so); |
socantrcvmore(so); |
sbunlock(sb); |
sbunlock(sb); |
|
|
sbrelease(&asb); |
sbrelease(&asb); |
} |
} |
|
|
|
int |
sosetopt(so, level, optname, m0) |
sosetopt(so, level, optname, m0) |
register struct socket *so; |
register struct socket *so; |
int level, optname; |
int level, optname; |
Line 839 sosetopt(so, level, optname, m0) |
|
Line 864 sosetopt(so, level, optname, m0) |
|
case SO_USELOOPBACK: |
case SO_USELOOPBACK: |
case SO_BROADCAST: |
case SO_BROADCAST: |
case SO_REUSEADDR: |
case SO_REUSEADDR: |
|
case SO_REUSEPORT: |
case SO_OOBINLINE: |
case SO_OOBINLINE: |
if (m == NULL || m->m_len < sizeof (int)) { |
if (m == NULL || m->m_len < sizeof (int)) { |
error = EINVAL; |
error = EINVAL; |
Line 912 sosetopt(so, level, optname, m0) |
|
Line 938 sosetopt(so, level, optname, m0) |
|
error = ENOPROTOOPT; |
error = ENOPROTOOPT; |
break; |
break; |
} |
} |
|
if (error == 0 && so->so_proto && so->so_proto->pr_ctloutput) { |
|
(void) ((*so->so_proto->pr_ctloutput) |
|
(PRCO_SETOPT, so, level, optname, &m0)); |
|
m = NULL; /* freed by protocol */ |
|
} |
} |
} |
bad: |
bad: |
if (m) |
if (m) |
|
|
return (error); |
return (error); |
} |
} |
|
|
|
int |
sogetopt(so, level, optname, mp) |
sogetopt(so, level, optname, mp) |
register struct socket *so; |
register struct socket *so; |
int level, optname; |
int level, optname; |
Line 950 sogetopt(so, level, optname, mp) |
|
Line 982 sogetopt(so, level, optname, mp) |
|
case SO_DEBUG: |
case SO_DEBUG: |
case SO_KEEPALIVE: |
case SO_KEEPALIVE: |
case SO_REUSEADDR: |
case SO_REUSEADDR: |
|
case SO_REUSEPORT: |
case SO_BROADCAST: |
case SO_BROADCAST: |
case SO_OOBINLINE: |
case SO_OOBINLINE: |
*mtod(m, int *) = so->so_options & optname; |
*mtod(m, int *) = so->so_options & optname; |
Line 1002 sogetopt(so, level, optname, mp) |
|
Line 1035 sogetopt(so, level, optname, mp) |
|
} |
} |
} |
} |
|
|
|
void |
sohasoutofband(so) |
sohasoutofband(so) |
register struct socket *so; |
register struct socket *so; |
{ |
{ |