in ASCII backups, write `|'s within fields as \174 instead of \| so

[moira.git] / backup / dump_db.pc

/*
 *	$Source$
 *	$Author$
 *	$Header$
 *
 *	(c) Copyright 1988 by the Massachusetts Institute of Technology.
 *	For copying and distribution information, please see the file
 *	<mit-copyright.h>.
 *
 * 	This program dumps the SMS database to a series of output files
 * which can be later read back into SMS in the event of a crash.
 * 
 */

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

#include <stdio.h>
#include <sys/file.h>
#include <fcntl.h>
#include <ctype.h>
#include <mit-copyright.h>
EXEC SQL INCLUDE sqlca;
#include "dump_db.h"

/* putc without the line buffer hair */

#define putc1(x, p)	(--(p)->_cnt >= 0 ?\
	(int)(*(unsigned char *)(p)->_ptr++ = (x)) :\
		_flsbuf((unsigned char)(x), p))

FILE *open_file();

EXEC SQL BEGIN DECLARE SECTION;
char *db="moira";
EXEC SQL END DECLARE SECTION;

main(argc, argv)
    int argc;
    char **argv;
{
    char *prefix;
    register int i;
    
    if (argc != 2) {
	fprintf(stderr, "Usage: %s prefix\n", argv[0]);
	exit(1);
    }
    prefix = argv[1];

    EXEC SQL CONNECT :db IDENTIFIED BY :db;

    do_backups(prefix);

    EXEC SQL COMMIT;
    exit(0);
}

dump_int(f, n)
    FILE *f;
    int n;
{
    char buf[1024];
    (void) sprintf(buf, "%d", n);
    dump_str(f, buf);
}

wpunt()
{
    punt("can't write backup file");
}

dump_str(f, str)
    register FILE *f;
    register char *str;
{
    register char *ibp;
    register int c;		/* PCC doesn't put chars in registers.. */
    register int t;

    for (ibp = str; c = (unsigned char) *ibp; ibp++) {
	if(c<32 || c>126 || c==SEP_CHAR || c=='\\') {
	    if (putc1('\\', f) < 0) wpunt();	    
	    t = ((c>>6)&7) + '0';
	    if (putc1(t,f) < 0) wpunt();
	    t = ((c>>3)&7) + '0';
	    if (putc1(t,f) < 0) wpunt();
	    t = (c&7) + '0';
	    if (putc1(t,f) < 0) wpunt();
	} else {
	    if (putc1(c, f) < 0) wpunt();
	}
    }
}

safe_close(stream)
	FILE *stream;
{
	if (fflush(stream) == EOF)
		punt("Unable to fflush");
	if (fsync(fileno(stream)) != 0) 
		punt("Unable to fsync");
	(void) fclose(stream);
}	

FILE *open_file(prefix, suffix)
    char *prefix, *suffix;
{
    char name[BUFSIZ];
    int fd;
    FILE *f;
    
    (void) strcpy(name, prefix);
    (void) strcat(name, suffix);

    fd = open(name, O_CREAT|O_WRONLY|O_EXCL, 0644);
    if (fd < 0) {
	punt(name);
    }
    f = fdopen(fd, "w");
    if (f == NULL) {
	fprintf(stderr, "fdopen of ");
	punt(name);
    }
    fprintf(stderr, "Working on %s\n", name);
    return(f);
}


/*
 * Trim whitespace off both ends of a string.
 */
char *strtrim(save)
    register char *save;
{
    register char *t, *s;

    s = save;
    while (isspace(*s)) s++;
    /* skip to end of string */
    if (*s == '\0') {
	*save = '\0';
	return(save);
    }

    for (t = s; *t; t++) continue; 
    while (t > s) {
	--t;
	if (!isspace(*t)) {
	    t++;
	    break;
	}
    }
    *t = '\0';
    return s;
}