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 648bb6f7 2005-02-08 devnull 	if((c->rpc=auth_allocrpc()) == nil
 28 648bb6f7 2005-02-08 devnull 	|| auth_rpc(c->rpc, "start", p, strlen(p)) != ARok
 29 648bb6f7 2005-02-08 devnull 	|| auth_rpc(c->rpc, "read", nil, 0) != ARok){
 30 be8b315d 2004-06-17 devnull 	Error:
 31 be8b315d 2004-06-17 devnull 		auth_freechal(c);
 32 be8b315d 2004-06-17 devnull 		free(p);
 33 be8b315d 2004-06-17 devnull 		return nil;
 34 be8b315d 2004-06-17 devnull 	}
 35 be8b315d 2004-06-17 devnull 
 36 be8b315d 2004-06-17 devnull 	if(c->rpc->narg > sizeof(c->chal)-1){
 37 be8b315d 2004-06-17 devnull 		werrstr("buffer too small for challenge");
 38 be8b315d 2004-06-17 devnull 		goto Error;
 39 be8b315d 2004-06-17 devnull 	}
 40 be8b315d 2004-06-17 devnull 	memmove(c->chal, c->rpc->arg, c->rpc->narg);
 41 be8b315d 2004-06-17 devnull 	c->nchal = c->rpc->narg;
 42 be8b315d 2004-06-17 devnull 	free(p);
 43 be8b315d 2004-06-17 devnull 	return c;
 44 be8b315d 2004-06-17 devnull }
 45 be8b315d 2004-06-17 devnull 
 46 be8b315d 2004-06-17 devnull AuthInfo*
 47 be8b315d 2004-06-17 devnull auth_response(Chalstate *c)
 48 be8b315d 2004-06-17 devnull {
 49 be8b315d 2004-06-17 devnull 	int ret;
 50 be8b315d 2004-06-17 devnull 	AuthInfo *ai;
 51 be8b315d 2004-06-17 devnull 
 52 be8b315d 2004-06-17 devnull 	ai = nil;
 53 648bb6f7 2005-02-08 devnull 	if(c->rpc == nil){
 54 be8b315d 2004-06-17 devnull 		werrstr("auth_response: connection not open");
 55 be8b315d 2004-06-17 devnull 		return nil;
 56 be8b315d 2004-06-17 devnull 	}
 57 be8b315d 2004-06-17 devnull 	if(c->resp == nil){
 58 be8b315d 2004-06-17 devnull 		werrstr("auth_response: nil response");
 59 be8b315d 2004-06-17 devnull 		return nil;
 60 be8b315d 2004-06-17 devnull 	}
 61 be8b315d 2004-06-17 devnull 	if(c->nresp == 0){
 62 be8b315d 2004-06-17 devnull 		werrstr("auth_response: unspecified response length");
 63 be8b315d 2004-06-17 devnull 		return nil;
 64 be8b315d 2004-06-17 devnull 	}
 65 be8b315d 2004-06-17 devnull 
 66 be8b315d 2004-06-17 devnull 	if(c->user){
 67 be8b315d 2004-06-17 devnull 		if(auth_rpc(c->rpc, "write", c->user, strlen(c->user)) != ARok){
 68 be8b315d 2004-06-17 devnull 			/*
 69 be8b315d 2004-06-17 devnull 			 * if this fails we're out of phase with factotum.
 70 be8b315d 2004-06-17 devnull 			 * give up.
 71 be8b315d 2004-06-17 devnull 			 */
 72 be8b315d 2004-06-17 devnull 			goto Out;
 73 be8b315d 2004-06-17 devnull 		}
 74 be8b315d 2004-06-17 devnull 	}
 75 be8b315d 2004-06-17 devnull 
 76 be8b315d 2004-06-17 devnull 	if(auth_rpc(c->rpc, "write", c->resp, c->nresp) != ARok){
 77 be8b315d 2004-06-17 devnull 		/*
 78 be8b315d 2004-06-17 devnull 		 * don't close the connection -- maybe we'll try again.
 79 be8b315d 2004-06-17 devnull 		 */
 80 be8b315d 2004-06-17 devnull 		return nil;
 81 be8b315d 2004-06-17 devnull 	}
 82 be8b315d 2004-06-17 devnull 
 83 be8b315d 2004-06-17 devnull 	switch(ret = auth_rpc(c->rpc, "read", nil, 0)){
 84 be8b315d 2004-06-17 devnull 	case ARok:
 85 be8b315d 2004-06-17 devnull 	default:
 86 be8b315d 2004-06-17 devnull 		werrstr("factotum protocol botch %d %s", ret, c->rpc->ibuf);
 87 be8b315d 2004-06-17 devnull 		break;
 88 be8b315d 2004-06-17 devnull 	case ARdone:
 89 be8b315d 2004-06-17 devnull 		ai = auth_getinfo(c->rpc);
 90 be8b315d 2004-06-17 devnull 		break;
 91 be8b315d 2004-06-17 devnull 	}
 92 be8b315d 2004-06-17 devnull 
 93 be8b315d 2004-06-17 devnull Out:
 94 be8b315d 2004-06-17 devnull 	auth_freerpc(c->rpc);
 95 be8b315d 2004-06-17 devnull 	c->rpc = nil;
 96 be8b315d 2004-06-17 devnull 	return ai;
 97 be8b315d 2004-06-17 devnull }
 98 be8b315d 2004-06-17 devnull 
 99 be8b315d 2004-06-17 devnull void
100 be8b315d 2004-06-17 devnull auth_freechal(Chalstate *c)
101 be8b315d 2004-06-17 devnull {
102 be8b315d 2004-06-17 devnull 	if(c == nil)
103 be8b315d 2004-06-17 devnull 		return;
104 be8b315d 2004-06-17 devnull 	if(c->rpc != nil)
105 be8b315d 2004-06-17 devnull 		auth_freerpc(c->rpc);
106 be8b315d 2004-06-17 devnull 	memset(c, 0xBB, sizeof(*c));
107 be8b315d 2004-06-17 devnull 	free(c);
108 be8b315d 2004-06-17 devnull }