#if !defined(lint) && defined(SCCSIDS)
static char sccsid[] = "@(#)svc.c 1.41 87/10/13 Copyr 1984 Sun Micro";
#endif
#include "autoconf.h"
#if HAVE_SYS_PARAM_H
#include <sys/param.h>
#endif
#include <gssrpc/rpc.h>
#include <gssrpc/pmap_clnt.h>
#include <stdio.h>
#include <string.h>
#include <errno.h>
#ifdef FD_SETSIZE
static SVCXPRT **xports;
extern int gssrpc_svc_fdset_init;
#else
#ifdef NBBY
#define NOFILE (sizeof(int) * NBBY)
#else
#define NOFILE (sizeof(int) * 8)
#endif
static SVCXPRT *xports[NOFILE];
#endif
#define NULL_SVC ((struct svc_callout *)0)
#define RQCRED_SIZE 1024
static struct svc_callout {
struct svc_callout *sc_next;
rpcprog_t sc_prog;
rpcprog_t sc_vers;
void (*sc_dispatch)();
} *svc_head;
static struct svc_callout *svc_find(rpcprog_t, rpcvers_t,
struct svc_callout **);
static void svc_do_xprt(SVCXPRT *xprt);
void
xprt_register(SVCXPRT *xprt)
{
register int sock = xprt->xp_sock;
#ifdef FD_SETSIZE
if (gssrpc_svc_fdset_init == 0) {
FD_ZERO(&svc_fdset);
gssrpc_svc_fdset_init++;
}
if (xports == NULL) {
xports = (SVCXPRT **)
mem_alloc(FD_SETSIZE * sizeof(SVCXPRT *));
memset(xports, 0, FD_SETSIZE * sizeof(SVCXPRT *));
}
if (sock < FD_SETSIZE) {
xports[sock] = xprt;
FD_SET(sock, &svc_fdset);
}
#else
if (sock < NOFILE) {
xports[sock] = xprt;
svc_fds |= (1 << sock);
}
#endif
if (sock > svc_maxfd)
svc_maxfd = sock;
}
void
xprt_unregister(SVCXPRT *xprt)
{
register int sock = xprt->xp_sock;
#ifdef FD_SETSIZE
if ((sock < FD_SETSIZE) && (xports[sock] == xprt)) {
xports[sock] = (SVCXPRT *)0;
FD_CLR(sock, &svc_fdset);
}
#else
if ((sock < NOFILE) && (xports[sock] == xprt)) {
xports[sock] = (SVCXPRT *)0;
svc_fds &= ~(1 << sock);
}
#endif
if (svc_maxfd <= sock) {
while ((svc_maxfd > 0) && xports[svc_maxfd] == 0)
svc_maxfd--;
}
}
bool_t
svc_register(
SVCXPRT *xprt,
rpcprog_t prog,
rpcvers_t vers,
void (*dispatch)(),
int protocol)
{
struct svc_callout *prev;
register struct svc_callout *s;
if ((s = svc_find(prog, vers, &prev)) != NULL_SVC) {
if (s->sc_dispatch == dispatch)
goto pmap_it;
return (FALSE);
}
s = (struct svc_callout *)mem_alloc(sizeof(struct svc_callout));
if (s == (struct svc_callout *)0) {
return (FALSE);
}
s->sc_prog = prog;
s->sc_vers = vers;
s->sc_dispatch = dispatch;
s->sc_next = svc_head;
svc_head = s;
pmap_it:
if (protocol) {
return (pmap_set(prog, vers, protocol, xprt->xp_port));
}
return (TRUE);
}
void
svc_unregister(
rpcprog_t prog,
rpcvers_t vers)
{
struct svc_callout *prev;
register struct svc_callout *s;
if ((s = svc_find(prog, vers, &prev)) == NULL_SVC)
return;
if (prev == NULL_SVC) {
svc_head = s->sc_next;
} else {
prev->sc_next = s->sc_next;
}
s->sc_next = NULL_SVC;
mem_free((char *) s, (u_int) sizeof(struct svc_callout));
(void)pmap_unset(prog, vers);
}
static struct svc_callout *
svc_find(
rpcprog_t prog,
rpcvers_t vers,
struct svc_callout **prev)
{
register struct svc_callout *s, *p;
p = NULL_SVC;
for (s = svc_head; s != NULL_SVC; s = s->sc_next) {
if ((s->sc_prog == prog) && (s->sc_vers == vers))
goto done;
p = s;
}
done:
*prev = p;
return (s);
}
bool_t
svc_sendreply(
register SVCXPRT *xprt,
xdrproc_t xdr_results,
caddr_t xdr_location)
{
struct rpc_msg rply;
rply.rm_direction = REPLY;
rply.rm_reply.rp_stat = MSG_ACCEPTED;
rply.acpted_rply.ar_verf = xprt->xp_verf;
rply.acpted_rply.ar_stat = SUCCESS;
rply.acpted_rply.ar_results.where = xdr_location;
rply.acpted_rply.ar_results.proc = xdr_results;
return (SVC_REPLY(xprt, &rply));
}
void
svcerr_noproc(register SVCXPRT *xprt)
{
struct rpc_msg rply;
rply.rm_direction = REPLY;
rply.rm_reply.rp_stat = MSG_ACCEPTED;
rply.acpted_rply.ar_verf = xprt->xp_verf;
rply.acpted_rply.ar_stat = PROC_UNAVAIL;
SVC_REPLY(xprt, &rply);
}
void
svcerr_decode(register SVCXPRT *xprt)
{
struct rpc_msg rply;
rply.rm_direction = REPLY;
rply.rm_reply.rp_stat = MSG_ACCEPTED;
rply.acpted_rply.ar_verf = xprt->xp_verf;
rply.acpted_rply.ar_stat = GARBAGE_ARGS;
SVC_REPLY(xprt, &rply);
}
void
svcerr_systemerr(register SVCXPRT *xprt)
{
struct rpc_msg rply;
rply.rm_direction = REPLY;
rply.rm_reply.rp_stat = MSG_ACCEPTED;
rply.acpted_rply.ar_verf = xprt->xp_verf;
rply.acpted_rply.ar_stat = SYSTEM_ERR;
SVC_REPLY(xprt, &rply);
}
void
svcerr_auth(
SVCXPRT *xprt,
enum auth_stat why)
{
struct rpc_msg rply;
rply.rm_direction = REPLY;
rply.rm_reply.rp_stat = MSG_DENIED;
rply.rjcted_rply.rj_stat = AUTH_ERROR;
rply.rjcted_rply.rj_why = why;
SVC_REPLY(xprt, &rply);
}
void
svcerr_weakauth(SVCXPRT *xprt)
{
svcerr_auth(xprt, AUTH_TOOWEAK);
}
void
svcerr_noprog(register SVCXPRT *xprt)
{
struct rpc_msg rply;
rply.rm_direction = REPLY;
rply.rm_reply.rp_stat = MSG_ACCEPTED;
rply.acpted_rply.ar_verf = xprt->xp_verf;
rply.acpted_rply.ar_stat = PROG_UNAVAIL;
SVC_REPLY(xprt, &rply);
}
void
svcerr_progvers(
register SVCXPRT *xprt,
rpcvers_t low_vers,
rpcvers_t high_vers)
{
struct rpc_msg rply;
rply.rm_direction = REPLY;
rply.rm_reply.rp_stat = MSG_ACCEPTED;
rply.acpted_rply.ar_verf = xprt->xp_verf;
rply.acpted_rply.ar_stat = PROG_MISMATCH;
rply.acpted_rply.ar_vers.low = low_vers;
rply.acpted_rply.ar_vers.high = high_vers;
SVC_REPLY(xprt, &rply);
}
void
svc_getreq(int rdfds)
{
#ifdef FD_SETSIZE
fd_set readfds;
int i, mask;
FD_ZERO(&readfds);
for (i=0, mask=1; rdfds; i++, mask <<=1) {
if (rdfds & mask)
FD_SET(i, &readfds);
rdfds &= ~mask;
}
svc_getreqset(&readfds);
#else
int readfds = rdfds & svc_fds;
svc_getreqset(&readfds);
#endif
}
#ifdef FD_SETSIZE
#define FDSET_TYPE fd_set
#else
#define FDSET_TYPE int
#endif
void
svc_getreqset(FDSET_TYPE *readfds)
{
#ifndef FD_SETSIZE
int readfds_local = *readfds;
#endif
register SVCXPRT *xprt;
register int sock;
#ifdef FD_SETSIZE
for (sock = 0; sock <= svc_maxfd; sock++) {
if (!FD_ISSET(sock, readfds))
continue;
xprt = xports[sock];
svc_do_xprt(xprt);
}
#else
for (sock = 0; readfds_local != 0; sock++, readfds_local >>= 1) {
if ((readfds_local & 1) == 0)
continue;
xprt = xports[sock];
svc_do_xprt(xprt);
}
#endif
}
extern struct svc_auth_ops svc_auth_gss_ops;
static void
svc_do_xprt(SVCXPRT *xprt)
{
caddr_t rawcred, rawverf, cookedcred;
struct rpc_msg msg;
struct svc_req r;
bool_t no_dispatch;
int prog_found;
rpcvers_t low_vers;
rpcvers_t high_vers;
enum xprt_stat stat;
rawcred = mem_alloc(MAX_AUTH_BYTES);
rawverf = mem_alloc(MAX_AUTH_BYTES);
cookedcred = mem_alloc(RQCRED_SIZE);
if (rawcred == NULL || rawverf == NULL || cookedcred == NULL)
return;
msg.rm_call.cb_cred.oa_base = rawcred;
msg.rm_call.cb_verf.oa_base = rawverf;
r.rq_clntcred = cookedcred;
do {
register struct svc_callout *s;
enum auth_stat why;
if (!SVC_RECV(xprt, &msg))
goto call_done;
r.rq_xprt = xprt;
r.rq_prog = msg.rm_call.cb_prog;
r.rq_vers = msg.rm_call.cb_vers;
r.rq_proc = msg.rm_call.cb_proc;
r.rq_cred = msg.rm_call.cb_cred;
no_dispatch = FALSE;
why = gssrpc__authenticate(&r, &msg, &no_dispatch);
if (why != AUTH_OK) {
svcerr_auth(xprt, why);
goto call_done;
} else if (no_dispatch) {
goto call_done;
}
prog_found = FALSE;
low_vers = (rpcvers_t) -1L;
high_vers = 0;
for (s = svc_head; s != NULL_SVC; s = s->sc_next) {
if (s->sc_prog == r.rq_prog) {
if (s->sc_vers == r.rq_vers) {
(*s->sc_dispatch)(&r, xprt);
goto call_done;
}
prog_found = TRUE;
if (s->sc_vers < low_vers)
low_vers = s->sc_vers;
if (s->sc_vers > high_vers)
high_vers = s->sc_vers;
}
}
if (prog_found)
svcerr_progvers(xprt,
low_vers, high_vers);
else
svcerr_noprog(xprt);
call_done:
if ((stat = SVC_STAT(xprt)) == XPRT_DIED){
SVC_DESTROY(xprt);
break;
} else if ((xprt->xp_auth != NULL) &&
(xprt->xp_auth->svc_ah_ops != &svc_auth_gss_ops)) {
xprt->xp_auth = NULL;
}
} while (stat == XPRT_MOREREQS);
mem_free(rawcred, MAX_AUTH_BYTES);
mem_free(rawverf, MAX_AUTH_BYTES);
mem_free(cookedcred, RQCRED_SIZE);
}