libio/bug-wmemstream1.c

   1 #include <stdio.h>
   2 #include <string.h>
   3 #include <wchar.h>
   4
   5
   6 static int
   7 do_test (void)
   8 {
   9   size_t size;
  10   wchar_t *buf;
  11   FILE *fp = open_wmemstream (&buf, &size);
  12   if (fp == NULL)
  13     {
  14       puts ("open_wmemstream failed");
  15       return 1;
  16     }
  17
  18   off64_t off = ftello64 (fp);
  19   if (off != 0)
  20     {
  21       puts ("initial position wrong");
  22       return 1;
  23     }
  24
  25   if (fseek (fp, 32768, SEEK_SET) != 0)
  26     {
  27       puts ("fseek failed");
  28       return 1;
  29     }
  30
  31   if (fputws (L"foo", fp) == EOF)
  32     {
  33       puts ("fputws failed");
  34       return 1;
  35     }
  36
  37   if (fclose (fp) == EOF)
  38     {
  39       puts ("fclose failed");
  40       return 1;
  41     }
  42
  43   if (size != 32768 + 3)
  44     {
  45       printf ("expected size %d, got %zu\n", 32768 + 3, size);
  46       return 1;
  47     }
  48
  49   for (int i = 0; i < 32768; ++i)
  50     if (buf[i] != L'\0')
  51       {
  52         printf ("wide character at offset %d is %#x\n",
  53                 i, (unsigned int) buf[i]);
  54         return 1;
  55       }
  56
  57   if (wmemcmp (buf + 32768, L"foo", 3) != 0)
  58     {
  59       puts ("written string incorrect");
  60       return 1;
  61     }
  62
  63   /* Mark the buffer.  */
  64   wmemset (buf, L'A', size);
  65   free (buf);
  66
  67   /* Try again, this time with write mode enabled before the seek.  */
  68   fp = open_wmemstream (&buf, &size);
  69   if (fp == NULL)
  70     {
  71       puts ("2nd open_wmemstream failed");
  72       return 1;
  73     }
  74
  75   off = ftello64 (fp);
  76   if (off != 0)
  77     {
  78       puts ("2nd initial position wrong");
  79       return 1;
  80     }
  81
  82   if (fputws (L"bar", fp) == EOF)
  83     {
  84       puts ("2nd fputws failed");
  85       return 1;
  86     }
  87
  88   if (fseek (fp, 32768, SEEK_SET) != 0)
  89     {
  90       puts ("2nd fseek failed");
  91       return 1;
  92     }
  93
  94   if (fputws (L"foo", fp) == EOF)
  95     {
  96       puts ("3rd fputws failed");
  97       return 1;
  98     }
  99
 100   if (fclose (fp) == EOF)
 101     {
 102       puts ("2nd fclose failed");
 103       return 1;
 104     }
 105
 106   if (size != 32768 + 3)
 107     {
 108       printf ("2nd expected size %d, got %zu\n", 32768 + 3, size);
 109       return 1;
 110     }
 111
 112   if (wmemcmp (buf, L"bar", 3) != 0)
 113     {
 114       puts ("initial string incorrect in 2nd try");
 115       return 1;
 116     }
 117
 118   for (int i = 3; i < 32768; ++i)
 119     if (buf[i] != L'\0')
 120       {
 121         printf ("wide character at offset %d is %#x in 2nd try\n",
 122                 i, (unsigned int) buf[i]);
 123         return 1;
 124       }
 125
 126   if (wmemcmp (buf + 32768, L"foo", 3) != 0)
 127     {
 128       puts ("written string incorrect in 2nd try");
 129       return 1;
 130     }
 131
 132   return 0;
 133 }
 134
 135 #define TEST_FUNCTION do_test ()
 136 #include "../test-skeleton.c"