mirror of git://git.gnupg.org/gnupg.git
182 lines
3.6 KiB
C
182 lines
3.6 KiB
C
/* t-b64.c - Module tests for b64enc.c and b64dec.c
|
|
* Copyright (C) 2008 Free Software Foundation, Inc.
|
|
*
|
|
* This file is part of GnuPG.
|
|
*
|
|
* GnuPG is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* GnuPG is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
/*
|
|
|
|
As of now this is only a test program for manual tests.
|
|
|
|
*/
|
|
|
|
|
|
|
|
#include <config.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
|
|
#include "util.h"
|
|
|
|
#define pass() do { ; } while(0)
|
|
#define fail(a) do { fprintf (stderr, "%s:%d: test %d failed\n",\
|
|
__FILE__,__LINE__, (a)); \
|
|
errcount++; \
|
|
} while(0)
|
|
|
|
static int verbose;
|
|
static int errcount;
|
|
|
|
static void
|
|
test_b64enc_pgp (const char *string)
|
|
{
|
|
gpg_error_t err;
|
|
struct b64state state;
|
|
|
|
if (!string)
|
|
string = "a";
|
|
|
|
err = b64enc_start (&state, stdout, "PGP MESSAGE");
|
|
if (err)
|
|
fail (1);
|
|
|
|
err = b64enc_write (&state, string, strlen (string));
|
|
if (err)
|
|
fail (2);
|
|
|
|
err = b64enc_finish (&state);
|
|
if (err)
|
|
fail (3);
|
|
|
|
pass ();
|
|
}
|
|
|
|
|
|
static void
|
|
test_b64enc_file (const char *fname)
|
|
{
|
|
gpg_error_t err;
|
|
struct b64state state;
|
|
FILE *fp;
|
|
char buffer[50];
|
|
size_t nread;
|
|
|
|
fp = fname ? fopen (fname, "r") : stdin;
|
|
if (!fp)
|
|
{
|
|
fprintf (stderr, "%s:%d: can't open '%s': %s\n",
|
|
__FILE__, __LINE__, fname? fname:"[stdin]", strerror (errno));
|
|
fail (0);
|
|
}
|
|
|
|
err = b64enc_start (&state, stdout, "DATA");
|
|
if (err)
|
|
fail (1);
|
|
|
|
while ( (nread = fread (buffer, 1, sizeof buffer, fp)) )
|
|
{
|
|
err = b64enc_write (&state, buffer, nread);
|
|
if (err)
|
|
fail (2);
|
|
}
|
|
|
|
err = b64enc_finish (&state);
|
|
if (err)
|
|
fail (3);
|
|
|
|
fclose (fp);
|
|
pass ();
|
|
}
|
|
|
|
static void
|
|
test_b64dec_file (const char *fname)
|
|
{
|
|
gpg_error_t err;
|
|
struct b64state state;
|
|
FILE *fp;
|
|
char buffer[50];
|
|
size_t nread, nbytes;
|
|
|
|
fp = fname ? fopen (fname, "r") : stdin;
|
|
if (!fp)
|
|
{
|
|
fprintf (stderr, "%s:%d: can't open '%s': %s\n",
|
|
__FILE__, __LINE__, fname? fname:"[stdin]", strerror (errno));
|
|
fail (0);
|
|
}
|
|
|
|
err = b64dec_start (&state, "");
|
|
if (err)
|
|
fail (1);
|
|
|
|
while ( (nread = fread (buffer, 1, sizeof buffer, fp)) )
|
|
{
|
|
err = b64dec_proc (&state, buffer, nread, &nbytes);
|
|
if (err)
|
|
{
|
|
if (gpg_err_code (err) == GPG_ERR_EOF)
|
|
break;
|
|
fail (2);
|
|
}
|
|
else if (nbytes)
|
|
fwrite (buffer, 1, nbytes, stdout);
|
|
}
|
|
|
|
err = b64dec_finish (&state);
|
|
if (err)
|
|
fail (3);
|
|
|
|
fclose (fp);
|
|
pass ();
|
|
}
|
|
|
|
|
|
|
|
int
|
|
main (int argc, char **argv)
|
|
{
|
|
int do_encode = 0;
|
|
int do_decode = 0;
|
|
|
|
if (argc)
|
|
{ argc--; argv++; }
|
|
if (argc && !strcmp (argv[0], "--verbose"))
|
|
{
|
|
verbose = 1;
|
|
argc--; argv++;
|
|
}
|
|
|
|
if (argc && !strcmp (argv[0], "--encode"))
|
|
{
|
|
do_encode = 1;
|
|
argc--; argv++;
|
|
}
|
|
else if (argc && !strcmp (argv[0], "--decode"))
|
|
{
|
|
do_decode = 1;
|
|
argc--; argv++;
|
|
}
|
|
|
|
if (do_encode)
|
|
test_b64enc_file (argc? *argv: NULL);
|
|
else if (do_decode)
|
|
test_b64dec_file (argc? *argv: NULL);
|
|
else
|
|
test_b64enc_pgp (argc? *argv: NULL);
|
|
|
|
return !!errcount;
|
|
}
|