156 lines
5.1 KiB
C
156 lines
5.1 KiB
C
/* socketvar.h */
|
|
|
|
|
|
#ifndef _SYS_SOCKETVAR_H
|
|
#define _SYS_SOCKETVAR_H
|
|
|
|
#include <OS.h>
|
|
#include <sys/uio.h>
|
|
#include <sys/socket.h>
|
|
|
|
struct mbuf;
|
|
|
|
struct sockbuf {
|
|
uint32 sb_cc; /* actual chars in buffer */
|
|
uint32 sb_hiwat; /* max actual char count (high water mark) */
|
|
uint32 sb_mbcnt; /* chars of mbufs used */
|
|
uint32 sb_mbmax; /* max chars of mbufs to use */
|
|
int32 sb_lowat; /* low water mark */
|
|
struct mbuf *sb_mb; /* the mbuf chain */
|
|
int16 sb_flags; /* flags, see below */
|
|
int32 sb_timeo; /* timeout for read/write */
|
|
sem_id sb_sleep; /* our sleep sem */
|
|
sem_id sb_pop; /* sem to wait on... */
|
|
};
|
|
|
|
#define SB_MAX (256*1024) /* default for max chars in sockbuf */
|
|
#define SB_LOCK 0x01 /* lock on data queue */
|
|
#define SB_WANT 0x02 /* someone is waiting to lock */
|
|
#define SB_WAIT 0x04 /* someone is waiting for data/space */
|
|
#define SB_SEL 0x08 /* someone is selecting */
|
|
#define SB_ASYNC 0x10 /* ASYNC I/O, need signals */
|
|
#define SB_NOINTR 0x40 /* operations not interruptible */
|
|
#define SB_KNOTE 0x80 /* kernel note attached */
|
|
#define SB_NOTIFY (SB_WAIT|SB_SEL|SB_ASYNC)
|
|
|
|
typedef void (*socket_event_callback)(void * socket, uint32 event, void * cookie);
|
|
|
|
struct socket {
|
|
uint16 so_type; /* type of socket */
|
|
uint16 so_options; /* socket options */
|
|
int16 so_linger; /* dreaded linger value */
|
|
int16 so_state; /* socket state */
|
|
char *so_pcb; /* pointer to the control block */
|
|
sem_id so_lock; /* socket lock */
|
|
sem_id so_timeo; /* our wait channel */
|
|
|
|
struct protosw *so_proto; /* pointer to protocol module */
|
|
|
|
struct socket *so_head;
|
|
struct socket *so_q0;
|
|
struct socket *so_q;
|
|
|
|
int16 so_q0len;
|
|
int16 so_qlen;
|
|
int16 so_qlimit;
|
|
int32 so_error;
|
|
// pid_t so_pgid;
|
|
uint32 so_oobmark;
|
|
|
|
/* our send/recv buffers */
|
|
struct sockbuf so_snd;
|
|
struct sockbuf so_rcv;
|
|
|
|
// event callback
|
|
socket_event_callback event_callback;
|
|
void * event_callback_cookie;
|
|
int sel_ev;
|
|
};
|
|
|
|
/* Select event bit mask */
|
|
#define SEL_READ 0x01
|
|
#define SEL_WRITE 0x02
|
|
#define SEL_EX 0x04
|
|
|
|
/*
|
|
* Socket state bits.
|
|
*/
|
|
#define SS_NOFDREF 0x001 /* no file table ref any more */
|
|
#define SS_ISCONNECTED 0x002 /* socket connected to a peer */
|
|
#define SS_ISCONNECTING 0x004 /* in process of connecting to peer */
|
|
#define SS_ISDISCONNECTING 0x008 /* in process of disconnecting */
|
|
#define SS_CANTSENDMORE 0x010 /* can't send more data to peer */
|
|
#define SS_CANTRCVMORE 0x020 /* can't receive more data from peer */
|
|
#define SS_RCVATMARK 0x040 /* at mark on input */
|
|
#define SS_ISDISCONNECTED 0x800 /* socket disconnected from peer */
|
|
|
|
#define SS_PRIV 0x080 /* privileged for broadcast, raw... */
|
|
#define SS_NBIO 0x100 /* non-blocking ops */
|
|
#define SS_ASYNC 0x200 /* async i/o notify */
|
|
#define SS_ISCONFIRMING 0x400 /* deciding to accept connection req */
|
|
#define SS_CONNECTOUT 0x1000 /* connect, not accept, at this end */
|
|
|
|
/* helpful defines... */
|
|
|
|
/* adjust counters in sb reflecting freeing of m */
|
|
#define sbfree(sb, m) { \
|
|
(sb)->sb_cc -= (m)->m_len; \
|
|
(sb)->sb_mbcnt -= MSIZE; \
|
|
if ((m)->m_flags & M_EXT) \
|
|
(sb)->sb_mbcnt -= (m)->m_ext.ext_size; \
|
|
}
|
|
|
|
#define sbspace(sb) \
|
|
((uint32) min((int)((sb)->sb_hiwat - (sb)->sb_cc), \
|
|
(int)((sb)->sb_mbmax - (sb)->sb_mbcnt)))
|
|
|
|
/* do we have to send all at once on a socket? */
|
|
#define sosendallatonce(so) \
|
|
((so)->so_proto->pr_flags & PR_ATOMIC)
|
|
|
|
/* adjust counters in sb reflecting allocation of m */
|
|
#define sballoc(sb, m) { \
|
|
(sb)->sb_cc += (m)->m_len; \
|
|
(sb)->sb_mbcnt += MSIZE; \
|
|
if ((m)->m_flags & M_EXT) \
|
|
(sb)->sb_mbcnt += (m)->m_ext.ext_size; \
|
|
}
|
|
|
|
#define soreadable(so) \
|
|
((so)->so_rcv.sb_cc >= (so)->so_rcv.sb_lowat || \
|
|
((so)->so_state & SS_CANTRCVMORE) || \
|
|
(so)->so_qlen || (so)->so_error)
|
|
|
|
#define sowriteable(so) \
|
|
((sbspace(&(so)->so_snd) >= (so)->so_snd.sb_lowat) && \
|
|
(((so)->so_state & SS_ISCONNECTED) || \
|
|
(((so)->so_proto->pr_flags & PR_CONNREQUIRED) == 0) || \
|
|
((so)->so_state & SS_CANTSENDMORE) || (so)->so_error))
|
|
|
|
#define M_WAITOK 0x0000
|
|
#define M_NOWAIT 0x0001
|
|
|
|
/*
|
|
* Set lock on sockbuf sb; sleep if lock is already held.
|
|
* Unless SB_NOINTR is set on sockbuf, sleep is interruptible.
|
|
* Returns error without lock if sleep is interrupted.
|
|
*/
|
|
#define sblock(sb, wf) ((sb)->sb_flags & SB_LOCK ? \
|
|
(((wf) == M_WAITOK) ? sockbuf_lock(sb) : EWOULDBLOCK) : \
|
|
((sb)->sb_flags |= SB_LOCK), 0)
|
|
|
|
/* release lock on sockbuf sb */
|
|
#define sbunlock(sb) { \
|
|
(sb)->sb_flags &= ~SB_LOCK; \
|
|
if ((sb)->sb_flags & SB_WANT) { \
|
|
(sb)->sb_flags &= ~SB_WANT; \
|
|
wakeup((sb)->sb_sleep); \
|
|
} \
|
|
}
|
|
|
|
#define sorwakeup(so) sowakeup((so), &(so)->so_rcv)
|
|
/* we don't handle upcall for sockets */
|
|
#define sowwakeup(so) sowakeup((so), &(so)->so_snd)
|
|
|
|
#endif /* _SYS_SOCKETVAR_H */
|