libstdc++-v3/testsuite/22_locale/codecvt/in/wchar_t/5.cc

   1 // 2003-02-06  Petur Runolfsson  <peturr02@ru.is>
   2
   3 // Copyright (C) 2003-2018 Free Software Foundation, Inc.
   4 //
   5 // This file is part of the GNU ISO C++ Library.  This library is free
   6 // software; you can redistribute it and/or modify it under the
   7 // terms of the GNU General Public License as published by the
   8 // Free Software Foundation; either version 3, or (at your option)
   9 // any later version.
  10
  11 // This library is distributed in the hope that it will be useful,
  12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 // GNU General Public License for more details.
  15
  16 // You should have received a copy of the GNU General Public License along
  17 // with this library; see the file COPYING3.  If not see
  18 // <http://www.gnu.org/licenses/>.
  19
  20 // 22.2.1.5 - Template class codecvt [lib.locale.codecvt]
  21
  22 #include <locale>
  23 #include <cstring>
  24 #include <testsuite_hooks.h>
  25
  26 // Need to explicitly set the state(mbstate_t) to zero.
  27 // How to do this is not specified by the ISO C99 standard, so we
  28 // might need to add some operators to make the intuiative case
  29 // work:
  30 //   w_codecvt::state_type state00;
  31 //   state00 = 0;
  32 // or, can use this explicit "C" initialization:
  33 //   w_codecvt::state_type state01 = {0, 0};
  34 // .. except Ulrich says: Use memset. Always use memset. Feel the force...
  35 void
  36 zero_state(std::mbstate_t& state)
  37 { std::memset(&state, 0, sizeof(std::mbstate_t)); }
  38
  39 // Required instantiation
  40 // codecvt<wchar_t, char, mbstate_t>
  41 //
  42 // Test handling of '\0' characters in input
  43 // libstdc++/9246
  44 void test05()
  45 {
  46   using namespace std;
  47   typedef codecvt<wchar_t, char, mbstate_t>     w_codecvt;
  48   typedef codecvt_base::result                  result;
  49   typedef wchar_t                               int_type;
  50   typedef char                                  ext_type;
  51   typedef char_traits<wchar_t>                  int_traits;
  52
  53   const ext_type        e_lit[] = { '\0', '\0', 'f', '\0' };
  54   const ext_type*       efrom_next;
  55   const int_type        i_lit[] = { L'\0', L'\0', L'f', L'\0' };
  56   size_t                size = sizeof(e_lit);
  57   int_type*             i_arr = new int_type[size + 1];
  58   int_type*             i_ref = new int_type[size + 1];
  59   wmemset(i_arr, static_cast<wchar_t>(0xdeadbeef), size + 1);
  60   wmemset(i_ref, static_cast<wchar_t>(0xdeadbeef), size + 1);
  61   int_type*             ito_next;
  62
  63   locale                loc;
  64   const w_codecvt*      cvt = &use_facet<w_codecvt>(loc);
  65
  66   // in
  67   w_codecvt::state_type state01;
  68   zero_state(state01);
  69   result r1 = cvt->in(state01, e_lit, e_lit + size, efrom_next,
  70                       i_arr, i_arr + size, ito_next);
  71   VERIFY( r1 == codecvt_base::ok );
  72   VERIFY( efrom_next == e_lit + size );
  73   VERIFY( ito_next == i_arr + size );
  74   VERIFY( !int_traits::compare(i_arr, i_lit, size) );
  75   VERIFY( !int_traits::compare(ito_next, i_ref, 1) );
  76
  77   delete [] i_arr;
  78   delete [] i_ref;
  79 }
  80
  81 int main ()
  82 {
  83   test05();
  84   return 0;
  85 }