netscape-revival

security/cmd/wrap.c

#include "sec.h"

#if defined(__sun) && !defined(SVR4)
extern int getopt(int, char**, char*);
extern int fprintf(FILE *, char *, ...);
extern int fwrite(char *, size_t, size_t, FILE*);
extern void fflush(FILE *);
#endif
extern char *optarg;

static void Usage(char *progName)
{
    fprintf(stderr,
	    "Usage: %s -t type [-i input] [-o output]\n",
	    progName);
    fprintf(stderr, "%-20s Define an input file to use (default is stdin)\n",
	    "-i input");
    fprintf(stderr, "%-20s Define an output file to use (default is stdout)\n",
	    "-o output");
    fprintf(stderr, "%-20s Set the type tag to use for output\n",
	    "-t type");
    fprintf(stderr, "Legal type tags are:\n");
    fprintf(stderr, "    %s\n", SEC_CT_CERTIFICATE);
    fprintf(stderr, "    %s\n", SEC_CT_CERTIFICATE_REQUEST);
    exit(-1);
}

int main(int argc, char **argv)
{
    char *progName, *type;
    int o, rv, nb;
    FILE *inFile, *outFile;
    SECTypedData td;
    SECItem wrap;
    PRArenaPool *arena;

    progName = strrchr(argv[0], '/');
    progName = progName ? progName+1 : argv[0];
    
    arena = PORT_NewArena(DER_DEFAULT_CHUNKSIZE);
    if (!arena) {
	fprintf(stderr, "%s: unable to allocate memory\n",
		progName);
	return -1;
    }
    
    /* Parse command line arguments */
    inFile = 0;
    outFile = 0;
    type = 0;
    while ((o = getopt(argc, argv, "t:i:o:")) != -1) {
	switch (o) {
	  case '?':
	    Usage(progName);
	    break;

	  case 't':
	    type = PORT_Strdup(optarg);
	    if (PORT_Strcmp(type, SEC_CT_CERTIFICATE) &&
		PORT_Strcmp(type, SEC_CT_CERTIFICATE_REQUEST)) {
		/* Loser */
		Usage(progName);
	    }
	    break;

	  case 'i':
	    inFile = fopen(optarg, "r");
	    if (!inFile) {
		fprintf(stderr, "%s: unable to open \"%s\" for reading\n",
			progName, optarg);
		return -1;
	    }
	    break;

	  case 'o':
	    outFile = fopen(optarg, "w");
	    if (!outFile) {
		fprintf(stderr, "%s: unable to open \"%s\" for writing\n",
			progName, optarg);
		return -1;
	    }
	    break;
	}
    }
    if (!type) Usage(progName);
    if (!inFile) inFile = stdin;
    if (!outFile) outFile = stdout;

    /* First read in the naked der */
    rv = DER_Read(arena, &td.data, inFile);
    if (rv) {
	fprintf(stderr, "%s: input file is not good der (error=%d)\n",
		progName, PORT_GetError());
	return -1;
    }

    /* Now wrap it with a type tag */ 
    td.type.data = (unsigned char*) type;
    td.type.len = PORT_Strlen(type);
    rv = DER_Encode(arena, &wrap, SECTypedDataTemplate, &td);
    if (rv) {
	fprintf(stderr, "%s: unable to der encode result (error=%d)\n",
		progName, PORT_GetError());
	return -1;
    }
    nb = fwrite(wrap.data, 1, wrap.len, outFile);
    if (nb != wrap.len) {
	fprintf(stderr, "%s: write error: errno=%d\n",
		progName, errno);
	return -1;
    }
    fflush(outFile);
    return 0;
}