Quelle  t-str.c

/*

Copyright 2012-2014, 2016 Free Software Foundation, Inc.

This file is part of the GNU MP Library test suite.

The GNU MP Library test suite 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.

The GNU MP Library test suite 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
the GNU MP Library test suite.  If not, see https://www.gnu.org/licenses/.  */

#include <assert.h>
#include <limits.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>

#include "testutils.h"

#define MAXBITS 400
#define COUNT 2000

#define GMP_LIMB_BITS (sizeof(mp_limb_t) * CHAR_BIT)
#define MAXLIMBS ((MAXBITS + GMP_LIMB_BITS - 1) / GMP_LIMB_BITS)

static void
test_small (void)
{
  struct {
    const char *input;
    const char *decimal;
  } data[] = {
    { "183407", "183407" },
    { " 763959", "763959 " },
    { "9 81999", "981999" },
    { "10\t7398 ", "107398" },
    { "-9585 44", "-00958544" },
    { "-0", "0000" },
    { " -000  ", "0" },
    { "0704436", "231710" },
    /* Check the case of large number of leading zeros. */
    { "0000000000000000000000000", "0000000000000000000000000" },
    { "000000000000000000000000704436", "000000000000000000000000231710" },
    { " 02503517", "689999" },
    { "0 1312143", "365667" },
    { "-03 274062", "-882738" },
    { "012\t242", "005282" },
    { "0b11010111110010001111", "883855" },
    { " 0b11001010010100001", "103585" },
    { "-0b101010110011101111", "-175343" },
    { "0b 1111111011011100110", "521958" },
    { "0b1 1111110111001000011", "1044035" },
    { " 0x53dfc", "343548" },
    { "0xfA019", "1024025" },
    { "0x 642d1", "410321" },
    { "0x5 8067", "360551" },
    { "-0xd6Be6", "-879590" },
    { "\t0B1110000100000000011", "460803" },
    { "0B\t1111110010010100101", "517285" },
    { "0B1\t010111101101110100", "359284" },
    { "-0B101\t1001101111111001", "-367609" },
    { "0B10001001010111110000", "562672" },
    { "0Xe4B7e", "936830" },
    { "0X1E4bf", "124095" },
    { "-0Xfdb90", "-1039248" },
    { "0X7fc47", "523335" },
    { "0X8167c", "530044" },
    /* Some invalid inputs */
    { "", NULL },
    { "0x", NULL },
    { "0b", NULL },
    { "0z", NULL },
    { "-", NULL },
    { "-0x ", NULL },
    { "0|1", NULL },
    { "4+4", NULL },
    { "0ab", NULL },
    { "10x0", NULL },
    { "0xxab", NULL },
    { "ab", NULL },
    { "0%#", NULL },
    { "$foo", NULL },
    { NULL, NULL }
  };
  unsigned i;
  mpz_t a, b;
  mpz_init (b);

  for (i = 0; data[i].input; i++)
    {
      int res = mpz_init_set_str (a, data[i].input, 0);
      if (data[i].decimal)
 {
   if (res != 0)
     {
       fprintf (stderr, "mpz_set_str returned -1, input: %s\n",
         data[i].input);
       abort ();
     }
   if (mpz_set_str (b, data[i].decimal, 10) != 0)
     {
       fprintf (stderr, "mpz_set_str returned -1, decimal input: %s\n",
         data[i].input);
       abort ();
     }
   if (mpz_cmp (a, b) != 0)
     {
       fprintf (stderr, "mpz_set_str failed for input: %s\n",
         data[i].input);

       dump ("got", a);
       dump ("ref", b);
       abort ();
     }
 }
      else if (res != -1)
 {
   fprintf (stderr, "mpz_set_str returned %d, invalid input: %s\n",
     res, data[i].input);
   abort ();
 }
      mpz_clear (a);
    }

  mpz_clear (b);
}

void
testmain (int argc, char **argv)
{
  unsigned i;
  char *ap;
  char *bp;
  char *rp;
  size_t bn, rn, arn, bps;

  mpz_t a, b;

  FILE *tmp;

  test_small ();

  mpz_init (a);
  mpz_init (b);

  tmp = tmpfile ();
  if (!tmp)
    fprintf (stderr,
      "Failed to create temporary file. Skipping mpz_out_str tests.\n");

  if (mpz_out_str (tmp, 63, a) != 0)
    {
      printf ("mpz_out_str did not return 0 (error) with base > 62\n");
      abort ();
    }

  if (mpz_out_str (tmp, -37, a) != 0)
    {
      printf ("mpz_out_str did not return 0 (error) with base < -37\n");
      abort ();
    }

  for (i = 0; i < COUNT; i++)
    {
      int base;
      for (base = 0; base <= 62; base += 1 + (base == 0))
 {
   hex_random_str_op (MAXBITS, (i&1 || base > 36) ? base: -base, &ap, ;&rp);
   if (mpz_set_str (a, ap, 16) != 0)
     {
       fprintf (stderr, "mpz_set_str failed on input %s\n", ap);
       abort ();
     }

   rn = strlen (rp);
   arn = rn - (rp[0] == '-');

   bn = mpz_sizeinbase (a, base ? base : 10);
   if (bn < arn || bn > (arn + 1))
     {
       fprintf (stderr, "mpz_sizeinbase failed:\n");
       dump ("a", a);
       fprintf (stderr, "r = %s\n", rp);
       fprintf (stderr, "  base %d, correct size %u, got %u\n",
         base, (unsigned) arn, (unsigned)bn);
       abort ();
     }
   bp = mpz_get_str (NULL, (i&1 || base > 36) ? base: -base, a);
   bps = strlen(bp) + 1;
   if (strcmp (bp, rp))
     {
       fprintf (stderr, "mpz_get_str failed:\n");
       dump ("a", a);
       fprintf (stderr, "b = %s\n", bp);
       fprintf (stderr, "  base = %d\n", base);
       fprintf (stderr, "r = %s\n", rp);
       abort ();
     }

   /* Just a few tests with file i/o. */
   if (tmp && i < 20)
     {
       size_t tn;
       rewind (tmp);
       tn = mpz_out_str (tmp, (i&1 || base > 36) ? base: -base, a);
       if (tn != rn)
  {
    fprintf (stderr, "mpz_out_str, bad return value:\n");
    dump ("a", a);
    fprintf (stderr, "r = %s\n", rp);
    fprintf (stderr, "  base %d, correct size %u, got %u\n",
      base, (unsigned) rn, (unsigned)tn);
    abort ();
  }
       rewind (tmp);
       memset (bp, 0, rn);
       tn = fread (bp, 1, rn, tmp);
       if (tn != rn)
  {
    fprintf (stderr,
      "fread failed, expected %lu bytes, got only %lu.\n",
      (unsigned long) rn, (unsigned long) tn);
    abort ();
  }

       if (memcmp (bp, rp, rn) != 0)
  {
    fprintf (stderr, "mpz_out_str failed:\n");
    dump ("a", a);
    fprintf (stderr, "b = %s\n", bp);
    fprintf (stderr, "  base = %d\n", base);
    fprintf (stderr, "r = %s\n", rp);
    abort ();
  }
     }

   mpz_set_str (b, rp, base);

   if (mpz_cmp (a, b))
     {
       fprintf (stderr, "mpz_set_str failed:\n");
       fprintf (stderr, "r = %s\n", rp);
       fprintf (stderr, "  base = %d\n", base);
       fprintf (stderr, "r = %s\n", ap);
       fprintf (stderr, "  base = 16\n");
       dump ("b", b);
       dump ("r", a);
       abort ();
     }

   /* Test mpn interface */
   if (base && mpz_sgn (a))
     {
       size_t i;
       const char *absr;
       mp_limb_t t[MAXLIMBS];
       size_t tn = mpz_size (a);

       assert (tn <= MAXLIMBS);
       mpn_copyi (t, a->_mp_d, tn);

       bn = mpn_get_str ((unsigned char *) bp, base, t, tn);
       if (bn != arn)
  {
    fprintf (stderr, "mpn_get_str failed:\n");
    fprintf (stderr, "returned length: %lu (bad)\n", (unsigned long) bn);
    fprintf (stderr, "expected: %lu\n", (unsigned long) arn);
    fprintf (stderr, "  base = %d\n", base);
    fprintf (stderr, "r = %s\n", ap);
    fprintf (stderr, "  base = 16\n");
    dump ("b", b);
    dump ("r", a);
    abort ();
  }
       absr = rp + (rp[0] == '-');

       for (i = 0; i < bn; i++)
  {
    unsigned char digit = absr[i];
    char value;
    if (digit >= '0' && digit <= '9')
      value = digit - '0';
    else if (digit >= 'a' && digit <= 'z')
      value = digit - 'a' + ((base > 36) ? 36 : 10);
    else if (digit >= 'A' && digit <= 'Z')
      value = digit - 'A' + 10;
    else
      {
        fprintf (stderr, "Internal error in test.\n");
        abort();
      }
    if (bp[i] != value)
      {
        fprintf (stderr, "mpn_get_str failed:\n");
        fprintf (stderr, "digit %lu: %d (bad)\n", (unsigned long) i, bp[i]);
        fprintf (stderr, "expected: %d\n", value);
        fprintf (stderr, "  base = %d\n", base);
        fprintf (stderr, "r = %s\n", ap);
        fprintf (stderr, "  base = 16\n");
        dump ("b", b);
        dump ("r", a);
        abort ();
      }
  }
       tn = mpn_set_str (t, (unsigned char *) bp, bn, base);
       if (tn != mpz_size (a) || mpn_cmp (t, a->_mp_d, tn))
  {
    fprintf (stderr, "mpn_set_str failed:\n");
    fprintf (stderr, "r = %s\n", rp);
    fprintf (stderr, "  base = %d\n", base);
    fprintf (stderr, "r = %s\n", ap);
    fprintf (stderr, "  base = 16\n");
    dump ("r", a);
    abort ();
  }
     }
   free (ap);
   free (rp);
   testfree (bp, bps);
 }
    }
  mpz_clear (a);
  mpz_clear (b);
}