| /* 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/. */ |
| |
| #include "plgetopt.h" |
| #include "secutil.h" |
| #include "nssb64.h" |
| #include <errno.h> |
| |
| #if defined(XP_WIN) || (defined(__sun) && !defined(SVR4)) |
| #if !defined(WIN32) |
| extern int fread(char *, size_t, size_t, FILE *); |
| extern int fwrite(char *, size_t, size_t, FILE *); |
| extern int fprintf(FILE *, char *, ...); |
| #endif |
| #endif |
| |
| #if defined(WIN32) |
| #include "fcntl.h" |
| #include "io.h" |
| #endif |
| |
| static PRInt32 |
| output_binary(void *arg, const unsigned char *obuf, PRInt32 size) |
| { |
| FILE *outFile = arg; |
| int nb; |
| |
| nb = fwrite(obuf, 1, size, outFile); |
| if (nb != size) { |
| PORT_SetError(SEC_ERROR_IO); |
| return -1; |
| } |
| |
| return nb; |
| } |
| |
| static PRBool |
| isBase64Char(char c) |
| { |
| return ((c >= 'A' && c <= 'Z') || |
| (c >= 'a' && c <= 'z') || |
| (c >= '0' && c <= '9') || |
| c == '+' || c == '/' || |
| c == '='); |
| } |
| |
| static SECStatus |
| decode_file(FILE *outFile, FILE *inFile) |
| { |
| NSSBase64Decoder *cx; |
| SECStatus status = SECFailure; |
| char ibuf[4096]; |
| const char *ptr; |
| |
| cx = NSSBase64Decoder_Create(output_binary, outFile); |
| if (!cx) { |
| return -1; |
| } |
| |
| for (;;) { |
| if (feof(inFile)) |
| break; |
| if (!fgets(ibuf, sizeof(ibuf), inFile)) { |
| if (ferror(inFile)) { |
| PORT_SetError(SEC_ERROR_IO); |
| goto loser; |
| } |
| /* eof */ |
| break; |
| } |
| for (ptr = ibuf; *ptr; ++ptr) { |
| char c = *ptr; |
| if (c == '\n' || c == '\r') { |
| break; /* found end of line */ |
| } |
| if (!isBase64Char(c)) { |
| ptr = ibuf; /* ignore line */ |
| break; |
| } |
| } |
| if (ibuf == ptr) { |
| continue; /* skip empty or non-base64 line */ |
| } |
| |
| status = NSSBase64Decoder_Update(cx, ibuf, ptr - ibuf); |
| if (status != SECSuccess) |
| goto loser; |
| } |
| |
| return NSSBase64Decoder_Destroy(cx, PR_FALSE); |
| |
| loser: |
| (void)NSSBase64Decoder_Destroy(cx, PR_TRUE); |
| return status; |
| } |
| |
| static void |
| Usage(char *progName) |
| { |
| fprintf(stderr, |
| "Usage: %s [-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"); |
| } |
| |
| int |
| main(int argc, char **argv) |
| { |
| char *progName; |
| SECStatus rv; |
| FILE *inFile = NULL, *outFile = NULL; |
| PRBool closeIn = PR_TRUE, closeOut = PR_TRUE; |
| PLOptState *optstate = NULL; |
| PLOptStatus status; |
| int exitCode = -1; |
| |
| progName = strrchr(argv[0], '/'); |
| progName = progName ? progName + 1 : argv[0]; |
| |
| /* Parse command line arguments */ |
| optstate = PL_CreateOptState(argc, argv, "?hi:o:"); |
| while ((status = PL_GetNextOpt(optstate)) == PL_OPT_OK) { |
| switch (optstate->option) { |
| case '?': |
| case 'h': |
| Usage(progName); |
| goto loser; |
| break; |
| |
| case 'i': |
| inFile = fopen(optstate->value, "r"); |
| if (!inFile) { |
| fprintf(stderr, "%s: unable to open \"%s\" for reading\n", |
| progName, optstate->value); |
| goto loser; |
| } |
| break; |
| |
| case 'o': |
| outFile = fopen(optstate->value, "wb"); |
| if (!outFile) { |
| fprintf(stderr, "%s: unable to open \"%s\" for writing\n", |
| progName, optstate->value); |
| goto loser; |
| } |
| break; |
| } |
| } |
| if (!inFile) { |
| inFile = stdin; |
| closeIn = PR_FALSE; |
| } |
| if (!outFile) { |
| #if defined(WIN32) |
| 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); |
| goto loser; |
| } |
| #endif |
| outFile = stdout; |
| closeOut = PR_FALSE; |
| } |
| rv = decode_file(outFile, inFile); |
| if (rv != SECSuccess) { |
| fprintf(stderr, "%s: lossage: error=%d errno=%d\n", |
| progName, PORT_GetError(), errno); |
| goto loser; |
| } |
| exitCode = 0; |
| loser: |
| if (optstate) { |
| PL_DestroyOptState(optstate); |
| } |
| if (inFile && closeIn) { |
| fclose(inFile); |
| } |
| if (outFile && closeOut) { |
| fclose(outFile); |
| } |
| return exitCode; |
| } |