btoa.c 5.79 KB
Newer Older
1 2 3
/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
4 5 6 7

#include "plgetopt.h"
#include "secutil.h"
#include "nssb64.h"
8
#include <errno.h>
9 10 11

#if defined(XP_WIN) || (defined(__sun) && !defined(SVR4))
#if !defined(WIN32)
12 13
extern int fread(char *, size_t, size_t, FILE *);
extern int fwrite(char *, size_t, size_t, FILE *);
14 15 16 17 18 19 20 21 22
extern int fprintf(FILE *, char *, ...);
#endif
#endif

#if defined(WIN32)
#include "fcntl.h"
#include "io.h"
#endif

23 24
static PRInt32
output_ascii(void *arg, const char *obuf, PRInt32 size)
25 26 27 28 29 30
{
    FILE *outFile = arg;
    int nb;

    nb = fwrite(obuf, 1, size, outFile);
    if (nb != size) {
31 32
        PORT_SetError(SEC_ERROR_IO);
        return -1;
33 34 35 36 37 38 39 40 41 42 43 44 45 46 47
    }

    return nb;
}

static SECStatus
encode_file(FILE *outFile, FILE *inFile)
{
    NSSBase64Encoder *cx;
    int nb;
    SECStatus status = SECFailure;
    unsigned char ibuf[4096];

    cx = NSSBase64Encoder_Create(output_ascii, outFile);
    if (!cx) {
48
        return -1;
49 50 51
    }

    for (;;) {
52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68
        if (feof(inFile))
            break;
        nb = fread(ibuf, 1, sizeof(ibuf), inFile);
        if (nb != sizeof(ibuf)) {
            if (nb == 0) {
                if (ferror(inFile)) {
                    PORT_SetError(SEC_ERROR_IO);
                    goto loser;
                }
                /* eof */
                break;
            }
        }

        status = NSSBase64Encoder_Update(cx, ibuf, nb);
        if (status != SECSuccess)
            goto loser;
69 70
    }

71 72
    status = NSSBase64Encoder_Destroy(cx, PR_FALSE);
    if (status != SECSuccess)
73
        return status;
74 75 76 77 78 79 80 81

    /*
     * Add a trailing CRLF.  Note this must be done *after* the call
     * to Destroy above (because only then are we sure all data has
     * been written out).
     */
    fwrite("\r\n", 1, 2, outFile);
    return SECSuccess;
82

83 84
loser:
    (void)NSSBase64Encoder_Destroy(cx, PR_TRUE);
85 86 87
    return status;
}

88 89
static void
Usage(char *progName)
90 91
{
    fprintf(stderr,
92 93
            "Usage: %s [-i input] [-o output]\n",
            progName);
94
    fprintf(stderr, "%-20s Define an input file to use (default is stdin)\n",
95
            "-i input");
96
    fprintf(stderr, "%-20s Define an output file to use (default is stdout)\n",
97
            "-o output");
98
    fprintf(stderr, "%-20s Wrap output in BEGIN/END lines and the given suffix\n",
99
            "-w suffix");
100
    fprintf(stderr, "%-20s (use \"c\" as a shortcut for suffix CERTIFICATE)\n",
101
            "");
102 103
}

104 105
int
main(int argc, char **argv)
106 107 108
{
    char *progName;
    SECStatus rv;
109 110 111
    FILE *inFile = NULL, *outFile = NULL;
    PRBool closeIn = PR_TRUE, closeOut = PR_TRUE;
    PLOptState *optstate = NULL;
112
    PLOptStatus status;
113
    char *suffix = NULL;
114
    int exitCode = -1;
115 116

    progName = strrchr(argv[0], '/');
117
    if (!progName)
118 119
        progName = strrchr(argv[0], '\\');
    progName = progName ? progName + 1 : argv[0];
120 121

    /* Parse command line arguments */
122
    optstate = PL_CreateOptState(argc, argv, "i:o:w:");
123
    PORT_Assert(optstate);
124
    while ((status = PL_GetNextOpt(optstate)) == PL_OPT_OK) {
125 126 127
        switch (optstate->option) {
            default:
                Usage(progName);
128
                goto loser;
129 130 131 132 133 134 135
                break;

            case 'i':
                inFile = fopen(optstate->value, "rb");
                if (!inFile) {
                    fprintf(stderr, "%s: unable to open \"%s\" for reading\n",
                            progName, optstate->value);
136
                    goto loser;
137 138 139 140 141 142 143 144
                }
                break;

            case 'o':
                outFile = fopen(optstate->value, "wb");
                if (!outFile) {
                    fprintf(stderr, "%s: unable to open \"%s\" for writing\n",
                            progName, optstate->value);
145
                    goto loser;
146 147 148 149 150 151 152 153 154 155
                }
                break;

            case 'w':
                if (!strcmp(optstate->value, "c"))
                    suffix = strdup("CERTIFICATE");
                else
                    suffix = strdup(optstate->value);
                break;
        }
156
    }
157
    if (status == PL_OPT_BAD)
158
        Usage(progName);
159 160
    if (!inFile) {
#if defined(WIN32)
161 162 163 164 165 166 167 168 169
        /* If we're going to read binary data from stdin, we must put stdin
        ** into O_BINARY mode or else incoming \r\n's will become \n's.
        */

        int smrv = _setmode(_fileno(stdin), _O_BINARY);
        if (smrv == -1) {
            fprintf(stderr,
                    "%s: Cannot change stdin to binary mode. Use -i option instead.\n",
                    progName);
170
            goto loser;
171
        }
172
#endif
173
        inFile = stdin;
174
        closeIn = PR_FALSE;
175
    }
176 177
    if (!outFile) {
#if defined(WIN32)
178 179 180 181 182 183 184 185 186
        /* We're going to write binary data to stdout. We must put stdout
        ** into O_BINARY mode or else outgoing \r\n's will become \r\r\n's.
        */

        int smrv = _setmode(_fileno(stdout), _O_BINARY);
        if (smrv == -1) {
            fprintf(stderr,
                    "%s: Cannot change stdout to binary mode. Use -o option instead.\n",
                    progName);
187
            goto loser;
188
        }
189
#endif
190
        outFile = stdout;
191
        closeOut = PR_FALSE;
192
    }
193
    if (suffix) {
194
        fprintf(outFile, "-----BEGIN %s-----\n", suffix);
195
    }
196 197
    rv = encode_file(outFile, inFile);
    if (rv != SECSuccess) {
198 199
        fprintf(stderr, "%s: lossage: error=%d errno=%d\n",
                progName, PORT_GetError(), errno);
200
        goto loser;
201
    }
202
    if (suffix) {
203
        fprintf(outFile, "-----END %s-----\n", suffix);
204
    }
205 206
    exitCode = 0;
loser:
207
    PL_DestroyOptState(optstate);
208 209 210 211 212 213
    if (inFile && closeIn) {
        fclose(inFile);
    }
    if (outFile && closeOut) {
        fclose(outFile);
    }
214 215 216
    if (suffix) {
        PORT_Free(suffix);
    }
217
    return exitCode;
218
}