DBMS=ORACLE

[moira.git] / server / qrtn.dc

/*
 *	$Source$
 *	$Author$
 *	$Header$
 *
 *	Copyright (C) 1987, 1988 by the Massachusetts Institute of Technology
 *	For copying and distribution information, please see the file
 *	<mit-copyright.h>.
 * 
 */

#ifndef lint
static char *rcsid_qrtn_dc = "$Header$";
#endif lint

#include <mit-copyright.h>
#include <string.h>
#include "mr_server.h"
#include "query.h"
EXEC SQL INCLUDE sqlca;  /* SQL Communications Area */
EXEC SQL INCLUDE sqlda;  /* SQL Descriptor Area */
#include "qrtn.h"

SQLDA *mr_sqlda;
EXEC SQL BEGIN DECLARE SECTION; 
int mr_sig_length;
int idummy;                            
char cdummy[MR_CDUMMY_LEN];            
char stmt_buf[MR_STMTBUF_LEN];
EXEC SQL END DECLARE SECTION; 

char *Argv[QMAXARGS];
extern char *table_name[];
extern char *sqlbuffer[QMAXARGS];

int dbms_errno = 0;
int mr_errcode = 0;
EXEC SQL BEGIN DECLARE SECTION; 
int query_timeout = 30;
char *database = "moira";
EXEC SQL END DECLARE SECTION;
extern char *whoami;
extern FILE *journal;

int mr_verify_query(client *cl, struct query *q, int argc, char *argv_ro[]);
int do_retrieve(struct query *q, char *pqual, char *psort,
		int (*action)(), char *actarg);
int do_update(struct query *q, char *argv[], char *qual,
	      int (*action)(), char *actarg);
int do_append(struct query *q, char *argv[], char *pqual,
	      int (*action)(), char *actarg);
int do_delete(struct query *q, char *qual,
	      int (*action)(), char *actarg);
void build_sql_stmt(char *result_buf, char *cmd, char *targetlist,
		    char *argv[], char *qual);
char *build_sort(struct validate *v, char *sort);

/* from qvalidate.dc */
int validate_fields(struct query *q, char *argv[], struct valobj *vo, int n);
int validate_row(struct query *q, char *argv[], struct validate *v);


/*
 * dbmserr: Called when the DBMS indicates an error.
 */

void dbmserr(void)
{
    EXEC SQL BEGIN DECLARE SECTION; 
    char err_msg[256];
    EXEC SQL END DECLARE SECTION;
    int bufsize=256, msglength=0;

    dbms_errno = -sqlca.sqlcode;
    mr_errcode = MR_DBMS_ERR;
    com_err(whoami, MR_DBMS_ERR, " code %d\n", dbms_errno);
    sqlglm(err_msg, &bufsize, &msglength);
    err_msg[msglength]=0;
    com_err(whoami, 0, "SQL error text = %s", err_msg);
    critical_alert("MOIRA", "Moira server encountered DBMS ERROR %d\n%s",
		       dbms_errno, err_msg);
}

/* This is declarative, not executed.  Applies from here on, in this file. */
EXEC SQL WHENEVER SQLERROR DO dbmserr();

int mr_open_database(void)
{
    register int i;
    SQLDA *mr_alloc_sqlda();
    static first_open = 1;

    if (first_open) {
	first_open = 0;

	/* initialize local argv */
	for (i = 0; i < 16; i++)
	  Argv[i] = malloc(ARGLEN);

	mr_sqlda = mr_alloc_sqlda();       

	incremental_init();
	flush_cache();
    }
	
    dbms_errno = 0;
    mr_errcode = 0;

    /* open the database */
    EXEC SQL CONNECT :database IDENTIFIED BY :database;

    if(dbms_errno) 
        return(mr_errcode);

    EXEC SQL SELECT data_length INTO :mr_sig_length FROM user_tab_columns WHERE table_name='USERS' and column_name='SIGNATURE';
    EXEC SQL COMMIT WORK;
    if(dbms_errno)
        return(mr_errcode);

    return(MR_SUCCESS);
}

void mr_close_database(void)
{
    flush_cache();
    EXEC SQL COMMIT RELEASE;
}

int mr_check_access(cl, name, argc, argv_ro)
     client *cl;
     char *name, *argv_ro[];
     int argc;
{
    struct query *q;

    dbms_errno = 0;
    mr_errcode = 0;

    q = get_query_by_name(name, cl->args->mr_version_no);
    if (q == (struct query *)0)
	return(MR_NO_HANDLE);

    return(mr_verify_query(cl, q, argc, argv_ro));    
}

int mr_process_query(cl, name, argc, argv_ro, action, actarg)
     client *cl;
     char *name, *argv_ro[], *actarg;
     int argc, (*action)();
{
    register struct query *q;
    register int status;
    register struct validate *v;
    char qual[256];
    char sort[32];
    char *pqual;
    char *psort;
    EXEC SQL BEGIN DECLARE SECTION; 
    char *table;
    EXEC SQL END DECLARE SECTION; 
    struct save_queue *sq;

    dbms_errno = 0;
    mr_errcode = 0;

    /* list queries command */
    if (!strcmp(name, "_list_queries")) {
	list_queries(cl->args->mr_version_no, action, actarg);
	return(MR_SUCCESS);
    }

    /* help query command */
    if (!strcmp(name, "_help")) {
	if (argc < 1)
	    return(MR_ARGS);
	q = get_query_by_name(argv_ro[0], cl->args->mr_version_no);
	if (q == (struct query *)0) return(MR_NO_HANDLE);
	help_query(q, action, actarg);
	return(MR_SUCCESS);
    }

    /* get query structure, return error if named query does not exist */
    q = get_query_by_name(name, cl->args->mr_version_no);
    if (q == (struct query *)0) return(MR_NO_HANDLE);
    v = q->validate;

    /* setup argument vector, verify access and arguments */
    if ((status = mr_verify_query(cl, q, argc, argv_ro)) != MR_SUCCESS)
	goto out;

    /* perform any special query pre-processing */
    if (v && v->pre_rtn) {
	status = (*v->pre_rtn)(q, Argv, cl, 0);
	if (status != MR_SUCCESS)
	    goto out;
    }

    switch (q->type) {
    case RETRIEVE:
	/* for queries that do not permit wildcarding, check if row
	   uniquely exists */
	if (v && v->field) {
	    status = validate_row(q, Argv, v); 
	    if (status != MR_EXISTS) break;
	}

	/* build "where" clause if needed */
	if (q->qual) {
	    build_qual(q->qual, q->argc, Argv, qual); 
	    pqual = qual;
	} else {
	    pqual = 0;
	}

	/* build "sort" clause if needed */
	if (v && v->valobj) {
	    psort = build_sort(v, sort);
	} else {
	    psort = 0;
	}

	/* if there is a followup routine, then we must save the results */
	/* of the first query for use by the followup routine */
	/* if q->rvar = NULL, perform post_rtn only */
	if (q->rvar) {
	    if (v && v->post_rtn) {
		sq = sq_create();
		status = do_retrieve(q, pqual, psort, sq_save_args, sq);
		if (status != MR_SUCCESS) {
		    sq_destroy(sq);
		    break;
		}
		status = (*v->post_rtn)(q, sq, v, action, actarg, cl);
	    } else {
		/* normal retrieve */
		status = do_retrieve(q, pqual, psort, action, actarg);
	    }
	    if (status != MR_SUCCESS) break;
	} else {
	    status = (*v->post_rtn)(q, Argv, cl, action, actarg);
	}

	break;

    case UPDATE:
	/* see if row already exists */
	if (v->field) {
	    status = validate_row(q, Argv, v);
	    if (status != MR_EXISTS) break;
	}

	/* build "where" clause and perform update */
	/* if q->rvar = NULL, perform post_rtn only */
	if (q->rvar) {
	    build_qual(q->qual, q->argc, Argv, qual);
	    incremental_before(q->rtable, qual, argv_ro);
	    status = do_update(q, &Argv[q->argc], qual, action, actarg);
	    incremental_after(q->rtable, qual, argv_ro);
	    if (status != MR_SUCCESS) break;
	    flush_name(argv_ro[0], q->rtable);
	    table = table_name[q->rtable];
	    if (strcmp(q->shortname, "sshi") && strcmp(q->shortname, "ssif")) {
		EXEC SQL UPDATE tblstats
                  SET updates = updates + 1, modtime = SYSDATE
		  WHERE table_name = :table;
	    }
	}

	/* execute followup routine (if any) */
	if (v->post_rtn) status = (*v->post_rtn)(q, Argv, cl);

	break;

    case APPEND:
	/* see if row already exists */
	if (v->field) {
	    status = validate_row(q, Argv, v);
	    if (status != MR_NO_MATCH) break;
	}

	/* build "where" clause if needed */
	if (q->qual) {
	    build_qual(q->qual, q->argc, Argv, qual);
	    pqual = qual;
	} else {
	    pqual = 0;
	}

	/* perform the append */
	/* if q->rvar = NULL, perform post_rtn only */
	if (q->rvar) {
	    incremental_clear_before();
	    status = do_append(q, &Argv[q->argc], pqual, action, actarg);
	    if (status != MR_SUCCESS) break;
	    if (v && v->object_id) {
		sprintf(qual, "%s.%s = %s",q->rvar, v->object_id, 
			Argv[q->argc+q->vcnt]);
		incremental_after(q->rtable, qual, argv_ro);
	    } else
	      incremental_after(q->rtable, pqual, argv_ro);

	    table = table_name[q->rtable];
	    EXEC SQL UPDATE tblstats
	      SET appends = appends + 1, modtime = SYSDATE
	      WHERE table_name = :table;
	}
	
	/* execute followup routine */
	if (v->post_rtn) status = (*v->post_rtn)(q, Argv, cl);
	break;

    case DELETE:
	/* see if row already exists */
	if (v->field) {
	    status = validate_row(q, Argv, v);
	    if (status != MR_EXISTS) break;
	}

	/* build "where" clause and perform delete */
	/* if q->rvar = NULL, perform post_rtn only */
	if (q->rvar) {
	    build_qual(q->qual, q->argc, Argv, qual);
	    table = table_name[q->rtable];
     	    incremental_before(q->rtable, qual, argv_ro);
	    status = do_delete(q, qual, action, actarg);
	    incremental_clear_after();
	    if (status != MR_SUCCESS) break;
	    flush_name(argv_ro[0], q->rtable);
	    EXEC SQL UPDATE tblstats
	      SET deletes = deletes + 1, modtime = SYSDATE
	      WHERE table_name = :table;
	}

	/* execute followup routine */
	if (v->post_rtn) status = (*v->post_rtn)(q, Argv, cl);
	break;

    }

out:
    if (status == MR_SUCCESS && dbms_errno != 0) {
	com_err(whoami, MR_INTERNAL, "Server didn't notice DBMS ERROR %d",
		       dbms_errno);
	status = mr_errcode;
    }

    if (q->type == RETRIEVE) {
        EXEC SQL COMMIT WORK;
    } else {
        if (status == MR_SUCCESS) {
        EXEC SQL COMMIT WORK;
	    if (journal) {
		char buf[1024];
		int i;
		extern time_t now;

		fprintf(journal, "%% %s %s %s",
			cl->clname, cl->entity, ctime(&now));
		fprintf(journal, "%s[%d] ", q->name, cl->args->mr_version_no);
		for (i = 0; i < argc; i++) {
		    if (i != 0) {
			putc(' ', journal);
		    }
		    requote(buf, argv_ro[i], sizeof(buf));
		    fputs(buf, journal);
		}
		putc('\n', journal);
		fflush(journal);
	    }
	    incremental_update();
	} else {
	    cache_abort();
            EXEC SQL ROLLBACK WORK;
	    incremental_flush();
	}
    }
    cache_commit(); /* commit following abort is safe */

    if (status != MR_SUCCESS && log_flags & LOG_RES)
	com_err(whoami, status, " (Query failed)");
    return(status);
}

int build_qual(fmt_buf, argc, argv, qual)
     char *fmt_buf, *argv[], *qual;
     int argc;
{
    char *res, *fmt;

    for(res=qual, fmt=fmt_buf; *fmt; fmt++) {
	if(*fmt=='%') {
	    if(*++fmt) {
		switch(*fmt) {                       
		  case '%':                              /* %% -> % */
		    *res++ = *fmt;           
		    break;
		  case 's':
		    if(*argv[0]) {
		        char *p=*argv;
			while(*p) {
			    if(*p=='\'') *res++='\'';	/* double the ' */
			    *res++=*p++;
			}
		    }
		    argv++;
		    break;
		  case 'd':
		    res+=sprintf(res,"%d",*(int *)*argv++);
		    break;
		  default:                               /* Swallow other %? pairs */
		    break;
		}
	    } else break;
	} else *res++ = *fmt;                            /* text -> result buffer */
    }
    *res='\0';
}

char *build_sort(v, sort)
     struct validate *v;
     char *sort;
{
    register struct valobj *vo;
    register int n;
    char elem[16];

    n = v->objcnt;
    vo = v->valobj;
    *sort = 0;

    while (--n >= 0) {
	if (vo->type == V_SORT) {
	    sprintf(elem, "%d", vo->index + 1);    /* Result column number */
	    if (*sort) strcat(sort, ", ");
	    strcat(sort, elem);
	}
	vo++;
    }

    return ((*sort) ? sort : 0);
}


/* Build arguement vector, verify query and arguments */

int privileged;

int mr_verify_query(cl, q, argc, argv_ro)
     client *cl;
     struct query *q;
     int argc;
     char *argv_ro[];
{
    register int argreq;
    register int status;
    register struct validate *v = q->validate;
    register int i;
    register char *to,*fr,*stop;

    privileged = 0;

    /* copy the arguments into a local argv that we can modify */
    if (argc >= QMAXARGS)
      return(MR_ARGS);
    for (i = 0; i < argc; i++) {
	for (to=Argv[i], fr=argv_ro[i], stop=to+ARGLEN; (*fr) && (to<stop);)
	  *to++ = *fr++;

	if (*fr) 
	  return(MR_ARG_TOO_LONG);
	*to='\0';

	if (*--to == '\\')
	  return(MR_BAD_CHAR);
    }

    /* check initial query access */
    status = check_query_access(q, Argv, cl);
    if (status != MR_SUCCESS && status != MR_PERM)
	return(status);
    if (status == MR_SUCCESS)
	privileged++;

    /* check argument count */
    argreq = q->argc;
    if (q->type == UPDATE || q->type == APPEND) argreq += q->vcnt;
    if (argc != argreq) return(MR_ARGS);

    /* validate arguments */
    if (v && v->valobj) {
	status = validate_fields(q, Argv, v->valobj, v->objcnt);
	if (status != MR_SUCCESS) return(status);
    }

    /* perform special query access check */
    if (!privileged && v && v->acs_rtn) {
	status = (*v->acs_rtn)(q, Argv, cl);
	if (status != MR_SUCCESS && status != MR_PERM)
	    return(status);
	if (status == MR_SUCCESS)
	    return(MR_SUCCESS);
    }

    return(privileged ? MR_SUCCESS : MR_PERM);
}


/* This routine caches info from the database.  Each query acl is stored
 * in the query structure, and whether that acl contains everybody.
 */

int check_query_access(q, argv, cl)
     struct query *q;
     char *argv[];
     client *cl;
{
    EXEC SQL BEGIN DECLARE SECTION; 
    char *name;
    int acl_id;
    static int def_uid;
    EXEC SQL END DECLARE SECTION; 
    int client_id;
    char *client_type;

    /* initialize default uid */
    if (def_uid == 0) {
        EXEC SQL SELECT users_id INTO :def_uid FROM users WHERE login='default';
    }

    /* get query access control list */
    if (q->acl != 0)
      acl_id = q->acl;
    else {
	name = q->shortname;
	EXEC SQL SELECT list_id INTO :acl_id FROM capacls WHERE tag = :name;
        if (sqlca.sqlcode < 0) return(MR_DBMS_ERR);
        if (sqlca.sqlcode == SQL_NO_MATCH) return(MR_PERM);
	q->acl = acl_id;

	/* check for default access */
        EXEC SQL SELECT member_id INTO :acl_id FROM imembers
          WHERE list_id = :acl_id AND member_type = 'USER' 
	    AND member_id = :def_uid;
        if (sqlca.sqlerrd[2] == 0)
          q->everybody = 0;
        else
          q->everybody = 1;
    }

    if (q->everybody)
      return(MR_SUCCESS);

    if (get_client(cl, &client_type, &client_id) != MR_SUCCESS)
      return(MR_PERM);
    if (find_member("LIST", acl_id, client_type, client_id))
      return(MR_SUCCESS);
    else
      return(MR_PERM);
}


int get_client(cl, client_type, client_id)
     client *cl;
     char **client_type;
     int *client_id;
{
    if (cl->users_id > 0) {
	*client_id = cl->users_id;
	*client_type = "USER";
	return(MR_SUCCESS);
    }

    if (cl->client_id < 0) {
	*client_id = -cl->users_id;
	*client_type = "KERBEROS";
	return(MR_SUCCESS);
    }

    return(MR_PERM);
}

int find_member(list_type, list_id, member_type, member_id)
     char *list_type, *member_type;
     int list_id, member_id;
{
    EXEC SQL BEGIN DECLARE SECTION; 
    int flag;
    EXEC SQL END DECLARE SECTION; 

    if (!strcmp(strtrim(list_type), strtrim(member_type)) &&
	list_id == member_id)
	return(1);

    /* see if client is a direct member of list */
    flag = 0;
    EXEC SQL SELECT member_id INTO :flag FROM imembers
      WHERE list_id = :list_id AND member_type = :member_type 
	AND member_id = :member_id;
    if(flag!=0) flag=1;             /** Not strictly necessary */
    if (sqlca.sqlcode == 0)
      return(flag);            
    return(0);
}


int do_retrieve(q, pqual, psort, action, actarg)
     struct query *q;
     char *pqual, *psort, *actarg;
     int (*action)();
{
    build_sql_stmt(stmt_buf,"SELECT",q->tlist,NULL,pqual);
    if(psort) {
      strcat(stmt_buf," ORDER BY ");
      strcat(stmt_buf,psort);
    }

    return do_for_all_rows(stmt_buf, q->vcnt, action, actarg);
}

char *sqlstrstr(str, pat)
     char *str, *pat;
{
    register char *p=pat;
    
    do {
	if(*str=='\'') {    /* Skip over single-quote delimited substrings */ 
	    while(*++str && (*str!='\'')) 
		; 
	    continue;
	}
	if(*str==*p) {
	    register char *s;
	    s=str; 
	    while(*++p && (*++s==*p))
		; 
	    if(*p) p=pat;  /* failed */
	}
    } while(*p && *++str);

    if(!*str) str=NULL;
    return(str);
}

void optimize_sql_stmt(buf)
     char *buf;
{
    char *point=buf, *pat, *eopat, *esc1, *esc2, *csr;

    for(point=buf; point=sqlstrstr(point,"LIKE"); point++)  {
	/* Now pointing to string "LIKE" */

	/* Look at next word */
	for(pat=point+4; *pat==' '; pat++) ;

	/* Is it a single-quote delimited string? */
	if(*pat!='\'') continue;

	/* look for "escape" clause - save escape character */
	/* 1. Find end of pattern */
	for(eopat=pat+1; 1; eopat++) {
	    if(*eopat=='\'') {
		if(eopat[1]=='\'')  /* single-quote is self-escaping */
		    eopat++;
		else
		    break;
	    }
	}

	/* 2. Look at next word */
	for(esc1=eopat; *++esc1==' ';) ;

	/* 3. esc1=0 if not "ESCAPE '?'", where the ? may be any character. */
	if(strncmp(esc1,"ESCAPE",6)) esc1=NULL;

	if(esc1) {
	    for(esc2=esc1+6; *esc2==' '; esc2++) ;
	    
	    if(*esc2++!='\'') continue; /* Bad SQL syntax. Skip. */
	    /* esc2 now points at the escape character itself */
	    if(esc2[1]!='\'') continue; /* Weird escape string. Skip. */
	} else {
	    esc2="\\";
	}

	/* Is pattern free from special characters? */
	for(csr=pat; csr<eopat; csr++) 
	    if((*csr=='%') || (*csr=='_') || (*csr==*esc2)) break;
	if(csr!=eopat) continue; /* Uses pattern matching. Skip. */

	/* Optimize the query statement */
	/* 1. Change "LIKE" to " =  " */
	memcpy(point," =  ",4);
	
	/* 2. Change "ESCAPE" to "      " */
	if(esc1) {
	    memset(esc1,' ',6);
	    /* 3. Change  "'*'" to "   " */
	    /*    (Changes '''' to "    ") */
	    if(esc2) memset(esc2-1,' ',(*esc2=='\'')?4:3); 
	}
    }
}

void build_sql_stmt(result_buf, cmd, targetlist, argv, qual)
     char *result_buf, *cmd, *targetlist, *argv[], *qual;
{
    char fmt_buf[MR_STMTBUF_LEN];
    register char *res, *fmt;

    if(qual)
      sprintf(fmt_buf,"%s %s WHERE %s",cmd,targetlist,qual);
    else
      sprintf(fmt_buf,"%s %s",cmd,targetlist);

    for(res=result_buf, fmt=fmt_buf; *fmt; fmt++) {
	if(*fmt=='%') {
	    if(*++fmt) {
		switch(*fmt) {                       
		  case '%':                              /* %% -> % */
		    *res++ = *fmt;           
		    break;
		  case 's':
		    if(*argv[0]) {
		        char *p=*argv;
			while(*p) {
			    if(*p=='\'') *res++='\'';	/* double the ' */
			    *res++=*p++;
			}
		    }
		    argv++;
		    break;
		  case 'd':
		    res+=sprintf(res,"%d",*(int *)*argv++);
		    break;
		  default:                               /* Swallow other %? pairs */
		    break;
		}
	    } else break;
	} else *res++ = *fmt;                            /* text -> result buffer */
    }
    *res='\0';

    optimize_sql_stmt(result_buf);
}

int do_update(q, argv, qual, action, actarg)
     struct query *q;
     char *argv[], *qual, *actarg;
     int (*action)();
{
    build_sql_stmt(stmt_buf,"UPDATE",q->tlist,argv,qual);
    EXEC SQL EXECUTE IMMEDIATE :stmt_buf;
    if (mr_errcode) return(mr_errcode);
    return(MR_SUCCESS);
}

int do_append(q, argv, pqual, action, actarg)
     struct query *q;
     char *argv[], *pqual, *actarg;
     int (*action)();
{
    build_sql_stmt(stmt_buf,"INSERT",q->tlist,argv,pqual);
    EXEC SQL EXECUTE IMMEDIATE :stmt_buf;
    if (mr_errcode) return(mr_errcode);
    return(MR_SUCCESS);
}

int do_delete(q, qual, action, actarg)
     struct query *q;
     char *qual, *actarg;
     int (*action)();
{
    sprintf(stmt_buf,"DELETE FROM %s WHERE %s",table_name[q->rtable],qual);
    EXEC SQL EXECUTE IMMEDIATE :stmt_buf;
    if (mr_errcode) return(mr_errcode);
    return(MR_SUCCESS);
}


/**
 ** set_next_object_id - set next object id in values table
 **
 ** Inputs: object - object name in values table and in objects
 **	    table - name of table objects are found in
 **	    limit - should the ID be range limited
 **
 ** - called before an APPEND operation to set the next object id to
 **   be used for the new record to the next free value
 **
 **/

int set_next_object_id(object, table, limit)
EXEC SQL BEGIN DECLARE SECTION;
     char *object;
EXEC SQL END DECLARE SECTION;
     enum tables table;
     int limit;
{
    EXEC SQL BEGIN DECLARE SECTION; 
    int value;
    EXEC SQL END DECLARE SECTION;
    int starting_value;
    
    EXEC SQL SELECT value INTO :value FROM numvalues WHERE name = :object;
    if (sqlca.sqlerrd[2] != 1)
	return(MR_NO_ID);

    starting_value=value;
    while (1) { 
	if (limit && value > MAX_ID_VALUE) 
	    value = MIN_ID_VALUE;

	sprintf(stmt_buf,"SELECT %s FROM %s WHERE %s=%d",
		object,table_name[table],object,value);
	dosql(sqlbuffer);
	if (sqlca.sqlcode < 0) return(mr_errcode);
	if (sqlca.sqlcode == SQL_NO_MATCH) break;

	value++;
	if (limit && value == starting_value) {   
	    com_err(whoami,0,"All id values have been used");
	    return(MR_NO_ID);
	}
    }

    if (LOG_RES)
        com_err(whoami, 0, "setting ID %s to %d", object, value);
    EXEC SQL UPDATE numvalues SET value = :value WHERE name = :object;
    return(MR_SUCCESS);
}


/* Turn a kerberos name into the user's ID of the account that principal
 * owns.  Sets the kerberos ID and user ID.
 */

int set_krb_mapping(name, login, ok, kid, uid)
     char *name, *login;
     int ok, *kid, *uid;
{
    EXEC SQL BEGIN DECLARE SECTION; 
    int u_id, k_id;
    char *krbname;
    EXEC SQL END DECLARE SECTION; 

    krbname = name;
    *kid = 0;
    *uid = 0;

    EXEC SQL SELECT km.users_id, km.string_id INTO :u_id, :k_id
      FROM krbmap km, strings str
      WHERE km.string_id = str.string_id AND str.string = :krbname;
    EXEC SQL COMMIT WORK;

    if (dbms_errno) return(mr_errcode);
    
    if (sqlca.sqlerrd[2] == 1) {    /* rowcount */
	*kid = -k_id;
	*uid = u_id;
	return(MR_SUCCESS);
    }

    if (name_to_id(name, STRINGS_TABLE, &k_id) == MR_SUCCESS)
      *kid = -k_id;

    if (!ok) {             
	*uid = *kid;
	return(MR_SUCCESS);
    }

    if (name_to_id(login, USERS_TABLE, uid) != MR_SUCCESS)
      *uid = 0;

    if (*kid == 0)
      *kid = *uid;
    if (dbms_errno) return(mr_errcode);
    return(MR_SUCCESS);
}


/* For now this just checks the argc's.  It should also see that there
 * are no duplicate names.
 */

void sanity_check_queries(void)
{
    register int i;
    int maxv = 0, maxa = 0;
    extern int QueryCount2;
    extern struct query Queries2[];

#define MAX(x,y) ((x) > (y) ? (x) : (y))

    for (i = 0; i < QueryCount2; i++) {
	maxv = MAX(maxv, Queries2[i].vcnt);
	maxa = MAX(maxa, Queries2[i].argc);
    }
    if (MAX(maxv, maxa) > QMAXARGS) {
	com_err(whoami, 0, "A query has more args than QMAXARGS");
	exit(1);
    }
}


/* Generically do a SELECT, storing the results in the provided buffers */

void dosql(buffers)
     char *buffers[];
{
  int i, errcode=0, errlen;

  EXEC SQL PREPARE inc_stmt FROM :stmt_buf;
  if(sqlca.sqlcode) return;
  EXEC SQL DECLARE inc_crs CURSOR FOR inc_stmt;
  EXEC SQL OPEN inc_crs;
  mr_sqlda->N = QMAXARGS;
  EXEC SQL DESCRIBE SELECT LIST FOR inc_stmt INTO mr_sqlda;
  mr_sqlda->N = mr_sqlda->F;
  for(i=0; i<mr_sqlda->N; i++) {
    mr_sqlda->V[i]=buffers[i];
    mr_sqlda->T[i]=97;
    mr_sqlda->L[i]=ARGLEN;
  }
  EXEC SQL FETCH inc_crs USING DESCRIPTOR mr_sqlda;

  /* if we got an error from the FETCH, we have to preserve it or the
     close will reset it and the caller will think nothing happened */
  if(sqlca.sqlcode) {
    errcode=sqlca.sqlcode;
    errlen=sqlca.sqlerrm.sqlerrml;
  }

  EXEC SQL CLOSE inc_crs;
  if(errcode) {
    sqlca.sqlcode=errcode;
    sqlca.sqlerrm.sqlerrml=errlen;
  }
}

int do_for_all_rows(query, count, action, actarg)
EXEC SQL BEGIN DECLARE SECTION;
     char *query;
EXEC SQL END DECLARE SECTION;
     int count, (*action)(), actarg;
{
  int i, rowcount=0;

  EXEC SQL PREPARE stmt FROM :query;
  if(sqlca.sqlcode) return;
  EXEC SQL DECLARE curs CURSOR FOR stmt;
  EXEC SQL OPEN curs;
  mr_sqlda->N = count;
  EXEC SQL DESCRIBE SELECT LIST FOR stmt INTO mr_sqlda;
  mr_sqlda->N = mr_sqlda->F;
  for(i=0; i<mr_sqlda->N; i++) {
    mr_sqlda->V[i]=sqlbuffer[i];
    mr_sqlda->T[i]=97;
    mr_sqlda->L[i]=ARGLEN;
  }

  while(1) {
    EXEC SQL FETCH curs USING DESCRIPTOR mr_sqlda;
    if(sqlca.sqlcode != 0) break;
    (*action)(count, sqlbuffer, actarg);
    rowcount++;
  }
  EXEC SQL CLOSE curs;
  
  if (mr_errcode) return(mr_errcode);
  return ((rowcount == 0) ? MR_NO_MATCH : MR_SUCCESS);
}


/* eof:qrtn.dc */