1 be8b315d 2004-06-17 devnull #include <u.h>
  2 be8b315d 2004-06-17 devnull #include <libc.h>
  3 be8b315d 2004-06-17 devnull #include <auth.h>
  4 be8b315d 2004-06-17 devnull #include <authsrv.h>
  5 be8b315d 2004-06-17 devnull #include "authlocal.h"
  6 be8b315d 2004-06-17 devnull 
  7 be8b315d 2004-06-17 devnull Chalstate*
  8 be8b315d 2004-06-17 devnull auth_challenge(char *fmt, ...)
  9 be8b315d 2004-06-17 devnull {
 10 be8b315d 2004-06-17 devnull 	char *p;
 11 be8b315d 2004-06-17 devnull 	va_list arg;
 12 be8b315d 2004-06-17 devnull 	Chalstate *c;
 13 be8b315d 2004-06-17 devnull 
 14 be8b315d 2004-06-17 devnull 	quotefmtinstall();	/* just in case */
 15 be8b315d 2004-06-17 devnull 	va_start(arg, fmt);
 16 be8b315d 2004-06-17 devnull 	p = vsmprint(fmt, arg);
 17 be8b315d 2004-06-17 devnull 	va_end(arg);
 18 be8b315d 2004-06-17 devnull 	if(p == nil)
 19 be8b315d 2004-06-17 devnull 		return nil;
 20 be8b315d 2004-06-17 devnull 
 21 be8b315d 2004-06-17 devnull 	c = mallocz(sizeof(*c), 1);
 22 be8b315d 2004-06-17 devnull 	if(c == nil){
 23 be8b315d 2004-06-17 devnull 		free(p);
 24 be8b315d 2004-06-17 devnull 		return nil;
 25 be8b315d 2004-06-17 devnull 	}
 26 be8b315d 2004-06-17 devnull 
 27 be8b315d 2004-06-17 devnull 	if((c->afd = open("/mnt/factotum/rpc", ORDWR)) < 0){
 28 be8b315d 2004-06-17 devnull 	Error:
 29 be8b315d 2004-06-17 devnull 		auth_freechal(c);
 30 be8b315d 2004-06-17 devnull 		free(p);
 31 be8b315d 2004-06-17 devnull 		return nil;
 32 be8b315d 2004-06-17 devnull 	}
 33 be8b315d 2004-06-17 devnull 
 34 be8b315d 2004-06-17 devnull 	if((c->rpc=auth_allocrpc(c->afd)) == nil
 35 be8b315d 2004-06-17 devnull 	|| auth_rpc(c->rpc, "start", p, strlen(p)) != ARok
 36 be8b315d 2004-06-17 devnull 	|| auth_rpc(c->rpc, "read", nil, 0) != ARok)
 37 be8b315d 2004-06-17 devnull 		goto Error;
 38 be8b315d 2004-06-17 devnull 
 39 be8b315d 2004-06-17 devnull 	if(c->rpc->narg > sizeof(c->chal)-1){
 40 be8b315d 2004-06-17 devnull 		werrstr("buffer too small for challenge");
 41 be8b315d 2004-06-17 devnull 		goto Error;
 42 be8b315d 2004-06-17 devnull 	}
 43 be8b315d 2004-06-17 devnull 	memmove(c->chal, c->rpc->arg, c->rpc->narg);
 44 be8b315d 2004-06-17 devnull 	c->nchal = c->rpc->narg;
 45 be8b315d 2004-06-17 devnull 	free(p);
 46 be8b315d 2004-06-17 devnull 	return c;
 47 be8b315d 2004-06-17 devnull }
 48 be8b315d 2004-06-17 devnull 
 49 be8b315d 2004-06-17 devnull AuthInfo*
 50 be8b315d 2004-06-17 devnull auth_response(Chalstate *c)
 51 be8b315d 2004-06-17 devnull {
 52 be8b315d 2004-06-17 devnull 	int ret;
 53 be8b315d 2004-06-17 devnull 	AuthInfo *ai;
 54 be8b315d 2004-06-17 devnull 
 55 be8b315d 2004-06-17 devnull 	ai = nil;
 56 be8b315d 2004-06-17 devnull 	if(c->afd < 0){
 57 be8b315d 2004-06-17 devnull 		werrstr("auth_response: connection not open");
 58 be8b315d 2004-06-17 devnull 		return nil;
 59 be8b315d 2004-06-17 devnull 	}
 60 be8b315d 2004-06-17 devnull 	if(c->resp == nil){
 61 be8b315d 2004-06-17 devnull 		werrstr("auth_response: nil response");
 62 be8b315d 2004-06-17 devnull 		return nil;
 63 be8b315d 2004-06-17 devnull 	}
 64 be8b315d 2004-06-17 devnull 	if(c->nresp == 0){
 65 be8b315d 2004-06-17 devnull 		werrstr("auth_response: unspecified response length");
 66 be8b315d 2004-06-17 devnull 		return nil;
 67 be8b315d 2004-06-17 devnull 	}
 68 be8b315d 2004-06-17 devnull 
 69 be8b315d 2004-06-17 devnull 	if(c->user){
 70 be8b315d 2004-06-17 devnull 		if(auth_rpc(c->rpc, "write", c->user, strlen(c->user)) != ARok){
 71 be8b315d 2004-06-17 devnull 			/*
 72 be8b315d 2004-06-17 devnull 			 * if this fails we're out of phase with factotum.
 73 be8b315d 2004-06-17 devnull 			 * give up.
 74 be8b315d 2004-06-17 devnull 			 */
 75 be8b315d 2004-06-17 devnull 			goto Out;
 76 be8b315d 2004-06-17 devnull 		}
 77 be8b315d 2004-06-17 devnull 	}
 78 be8b315d 2004-06-17 devnull 
 79 be8b315d 2004-06-17 devnull 	if(auth_rpc(c->rpc, "write", c->resp, c->nresp) != ARok){
 80 be8b315d 2004-06-17 devnull 		/*
 81 be8b315d 2004-06-17 devnull 		 * don't close the connection -- maybe we'll try again.
 82 be8b315d 2004-06-17 devnull 		 */
 83 be8b315d 2004-06-17 devnull 		return nil;
 84 be8b315d 2004-06-17 devnull 	}
 85 be8b315d 2004-06-17 devnull 
 86 be8b315d 2004-06-17 devnull 	switch(ret = auth_rpc(c->rpc, "read", nil, 0)){
 87 be8b315d 2004-06-17 devnull 	case ARok:
 88 be8b315d 2004-06-17 devnull 	default:
 89 be8b315d 2004-06-17 devnull 		werrstr("factotum protocol botch %d %s", ret, c->rpc->ibuf);
 90 be8b315d 2004-06-17 devnull 		break;
 91 be8b315d 2004-06-17 devnull 	case ARdone:
 92 be8b315d 2004-06-17 devnull 		ai = auth_getinfo(c->rpc);
 93 be8b315d 2004-06-17 devnull 		break;
 94 be8b315d 2004-06-17 devnull 	}
 95 be8b315d 2004-06-17 devnull 
 96 be8b315d 2004-06-17 devnull Out:
 97 be8b315d 2004-06-17 devnull 	close(c->afd);
 98 be8b315d 2004-06-17 devnull 	auth_freerpc(c->rpc);
 99 be8b315d 2004-06-17 devnull 	c->afd = -1;
100 be8b315d 2004-06-17 devnull 	c->rpc = nil;
101 be8b315d 2004-06-17 devnull 	return ai;
102 be8b315d 2004-06-17 devnull }
103 be8b315d 2004-06-17 devnull 
104 be8b315d 2004-06-17 devnull void
105 be8b315d 2004-06-17 devnull auth_freechal(Chalstate *c)
106 be8b315d 2004-06-17 devnull {
107 be8b315d 2004-06-17 devnull 	if(c == nil)
108 be8b315d 2004-06-17 devnull 		return;
109 be8b315d 2004-06-17 devnull 
110 be8b315d 2004-06-17 devnull 	if(c->afd >= 0)
111 be8b315d 2004-06-17 devnull 		close(c->afd);
112 be8b315d 2004-06-17 devnull 	if(c->rpc != nil)
113 be8b315d 2004-06-17 devnull 		auth_freerpc(c->rpc);
114 be8b315d 2004-06-17 devnull 
115 be8b315d 2004-06-17 devnull 	memset(c, 0xBB, sizeof(*c));
116 be8b315d 2004-06-17 devnull 	free(c);
117 be8b315d 2004-06-17 devnull }