#include <sys/types.h>
#include <sys/stat.h>
#include <pwd.h>
+#include <time.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
*q = '\0';
}
+int parse(char * const buf, const int argc, const char *argv[const])
+{
+ char *p, *q;
+ int i;
+
+ for (i = 0, p = buf; *p; p = q+1, i++) {
+ for (q = p; *q && *q != ':' && *q != '\r' && *q != '\n'; q++) ;
+ *q = '\0';
+ if (*p && i < argc) argv[i] = p;
+ }
+ return i != argc;
+}
+
struct _auth_obj authfile(const char *tokenid,
const char *userid, const char *password,
void (*update_nonce)(char *nonce, const int nonsize),
struct _auth_obj ret = {0};
mode_t oldmask;
FILE *fp = NULL;
- char *fn;
+ char *fn, *nfn;
int fnl;
char *buf = NULL;
- const char *wtokenid = "", *wuserid = NULL, *wnonce = NULL;
- const char *hablob = NULL;
+ struct {
+ const char *tokenid;
+ const char *userid;
+ const char *nonce;
+ const char *hablob;
+ } w = {"", NULL, NULL, NULL};
unsigned char *ablob = NULL;
int blobsize = 0;
char *newnonce;
}
fn = alloca(fnl);
make_path(fn, tokenid, userid);
+ nfn = alloca(fnl+32);
+ snprintf(nfn, fnl+32, "%s.%d.%ld", fn, (int)getpid(), (long)time(NULL));
fp = fopen(fn, "r");
if (fp) {
struct stat st;
if (fstat(fd, &st)) st.st_size = 2047;
if (st.st_size > 2047) st.st_size = 2047;
buf = alloca(st.st_size + 1);
- if (fgets(buf, st.st_size + 1, fp)) {
- char *p;
-
- p = &buf[strlen(buf) - 1];
- while (*p == '\n' || *p == '\r') *p-- = '\0';
- wtokenid = strtok(buf, ":");
- wuserid = strtok(NULL, ":");
- wnonce = strtok(NULL, ":");
- hablob = strtok(NULL, ":");
- } else {
+ if (!fgets(buf, st.st_size + 1, fp)) {
ret.err = strerror(errno);
+ } else if (parse(buf, sizeof(w)/sizeof(char*),
+ (const char ** const)&w)){
+ ret.err = "error: unparseable auth file";
}
fclose(fp);
}
if (ret.err) return ret;
- if (hablob) {
- int hlen = strlen(hablob);
+ if (w.hablob) {
+ int hlen = strlen(w.hablob);
if (hlen % 32 != 0) {
ret.err = "error: auth string has wrong length";
} else if (hlen !=
- strspn(hablob, "0123456789abcdefABCDEF")) {
+ strspn(w.hablob, "0123456789abcdefABCDEF")) {
ret.err = "error: auth string not hexadecimal";
} else {
int i;
blobsize = hlen/2;
ablob = alloca(blobsize);
for (i = 0; i < blobsize; i++)
- sscanf(&hablob[i*2], "%2hhx", &ablob[i]);
+ sscanf(&w.hablob[i*2], "%2hhx", &ablob[i]);
}
}
if (ret.err) return ret;
- nonsize = wnonce ? strlen(wnonce)*2 : 32;
+ nonsize = w.nonce ? strlen(w.nonce)*2 : 32;
if (nonsize < 32) nonsize = 32;
newnonce = alloca(nonsize);
- if (wnonce) strcpy(newnonce, wnonce);
+ if (w.nonce) strcpy(newnonce, w.nonce);
else memset(newnonce, 0, nonsize);
update_nonce(newnonce, nonsize);
- ao = authobj(userid?userid:wuserid, password,
- wnonce, newnonce, secret, secsize,
+ ao = authobj(userid?userid:w.userid, password,
+ w.nonce, newnonce, secret, secsize,
payload, paylsize, ablob, blobsize,
fetch_key);
}
oldmask = umask(077);
- if ((fp = fopen(fn, "w"))) {
+ if ((fp = fopen(nfn, "w"))) {
int i;
- if (fprintf(fp, "%s:%s:%s:", tokenid?tokenid:wtokenid,
- userid?userid:wuserid, newnonce) < 0) {
+ if (fprintf(fp, "%s:%s:%s:", tokenid?tokenid:w.tokenid,
+ userid?userid:w.userid, newnonce) < 0) {
ret.err = strerror(errno);
} else for (i = 0; i < ao.datasize; i++)
if (fprintf(fp, "%02x", ao.data[i]) < 0) {
ret.err = strerror(errno);
}
(void)umask(oldmask);
+ if (ret.err) {
+ unlink(nfn); /* may not exist but no matter */
+ } else if (rename(nfn, fn)) {
+ ret.err = strerror(errno);
+ }
if (!ret.err) {
- int bufsize = (wuserid?strlen(wuserid)+1:0) + ao.paylsize;
+ int bufsize = (w.userid?strlen(w.userid)+1:0) + ao.paylsize + 1;
if (bufsize) {
if ((ret.buffer = malloc(bufsize)) == NULL) {
ret.err = "authfile malloc failed";
} else {
unsigned char *p = ret.buffer;
- if (wuserid) {
- strcpy((char*)p, wuserid);
+ if (w.userid) {
+ strcpy((char*)p, w.userid);
ret.data = p;
- ret.datasize = strlen(wuserid)+1;
- p += strlen(wuserid)+1;
+ ret.datasize = strlen(w.userid)+1;
+ p += strlen(w.userid)+1;
}
if (ao.payload) {
memcpy(p, ao.payload, ao.paylsize);
+ p[ao.paylsize] = '\0';
ret.payload = p;
- ret.paylsize = ao.paylsize;
+ ret.paylsize = ao.paylsize+1;
}
}
}