libcxx/src/support/ibm/xlocale_zos.cpp

   1 //===----------------------------------------------------------------------===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8
   9 #include <__assert>
  10 #include <__support/ibm/xlocale.h>
  11 #include <sstream>
  12 #include <vector>
  13
  14 #ifdef __cplusplus
  15 extern "C" {
  16 #endif // __cplusplus
  17
  18 locale_t newlocale(int category_mask, const char* locale, locale_t base) {
  19   // Maintain current locale name(s) to restore later.
  20   std::string current_loc_name(setlocale(LC_ALL, 0));
  21
  22   // Check for errors.
  23   if (category_mask == LC_ALL_MASK && setlocale(LC_ALL, locale) == NULL) {
  24     errno = EINVAL;
  25     return (locale_t)0;
  26   } else {
  27     for (int _Cat = 0; _Cat <= _LC_MAX; ++_Cat) {
  28       if ((_CATMASK(_Cat) & category_mask) != 0 && setlocale(_Cat, locale) == NULL) {
  29         setlocale(LC_ALL, current_loc_name.c_str());
  30         errno = EINVAL;
  31         return (locale_t)0;
  32       }
  33     }
  34   }
  35
  36   // Create new locale.
  37   locale_t newloc = new locale_struct();
  38
  39   if (base) {
  40     if (category_mask != LC_ALL_MASK) {
  41       // Copy base when it will not be overwritten.
  42       memcpy(newloc, base, sizeof (locale_struct));
  43       newloc->category_mask = category_mask | base->category_mask;
  44     }
  45     delete base;
  46   } else {
  47     newloc->category_mask = category_mask;
  48   }
  49
  50   if (category_mask & LC_COLLATE_MASK)
  51     newloc->lc_collate = locale;
  52   if (category_mask & LC_CTYPE_MASK)
  53     newloc->lc_ctype = locale;
  54   if (category_mask & LC_MONETARY_MASK)
  55     newloc->lc_monetary = locale;
  56   if (category_mask & LC_NUMERIC_MASK)
  57     newloc->lc_numeric = locale;
  58   if (category_mask & LC_TIME_MASK)
  59     newloc->lc_time = locale;
  60   if (category_mask & LC_MESSAGES_MASK)
  61     newloc->lc_messages = locale;
  62
  63   // Restore current locale.
  64   setlocale(LC_ALL, current_loc_name.c_str());
  65   return (locale_t)newloc;
  66 }
  67
  68 void freelocale(locale_t locobj) {
  69   delete locobj;
  70 }
  71
  72 locale_t uselocale(locale_t newloc) {
  73   // Maintain current locale name(s).
  74   std::string current_loc_name(setlocale(LC_ALL, 0));
  75
  76   if (newloc) {
  77     // Set locales and check for errors.
  78     bool is_error =
  79       (newloc->category_mask & LC_COLLATE_MASK &&
  80         setlocale(LC_COLLATE, newloc->lc_collate.c_str()) == NULL) ||
  81       (newloc->category_mask & LC_CTYPE_MASK &&
  82         setlocale(LC_CTYPE, newloc->lc_ctype.c_str()) == NULL) ||
  83       (newloc->category_mask & LC_MONETARY_MASK &&
  84         setlocale(LC_MONETARY, newloc->lc_monetary.c_str()) == NULL) ||
  85       (newloc->category_mask & LC_NUMERIC_MASK &&
  86         setlocale(LC_NUMERIC, newloc->lc_numeric.c_str()) == NULL) ||
  87       (newloc->category_mask & LC_TIME_MASK &&
  88         setlocale(LC_TIME, newloc->lc_time.c_str()) == NULL) ||
  89       (newloc->category_mask & LC_MESSAGES_MASK &&
  90         setlocale(LC_MESSAGES, newloc->lc_messages.c_str()) == NULL);
  91
  92     if (is_error) {
  93       setlocale(LC_ALL, current_loc_name.c_str());
  94       errno = EINVAL;
  95       return (locale_t)0;
  96     }
  97   }
  98
  99   // Construct and return previous locale.
 100   locale_t previous_loc = new locale_struct();
 101
 102   // current_loc_name might be a comma-separated locale name list.
 103   if (current_loc_name.find(',') != std::string::npos) {
 104     // Tokenize locale name list.
 105     const char delimiter = ',';
 106     std::vector<std::string> tokenized;
 107     std::stringstream ss(current_loc_name);
 108     std::string s;
 109
 110     while (std::getline(ss, s, delimiter)) {
 111         tokenized.push_back(s);
 112     }
 113
 114     _LIBCPP_ASSERT_UNCATEGORIZED(tokenized.size() >= _NCAT, "locale-name list is too short");
 115
 116     previous_loc->lc_collate = tokenized[LC_COLLATE];
 117     previous_loc->lc_ctype = tokenized[LC_CTYPE];
 118     previous_loc->lc_monetary = tokenized[LC_MONETARY];
 119     previous_loc->lc_numeric = tokenized[LC_NUMERIC];
 120     previous_loc->lc_time = tokenized[LC_TIME];
 121     // Skip LC_TOD.
 122     previous_loc->lc_messages = tokenized[LC_MESSAGES];
 123   } else {
 124     previous_loc->lc_collate = current_loc_name;
 125     previous_loc->lc_ctype = current_loc_name;
 126     previous_loc->lc_monetary = current_loc_name;
 127     previous_loc->lc_numeric = current_loc_name;
 128     previous_loc->lc_time = current_loc_name;
 129     previous_loc->lc_messages = current_loc_name;
 130   }
 131
 132   previous_loc->category_mask = LC_ALL_MASK;
 133   return previous_loc;
 134 }
 135
 136 #ifdef __cplusplus
 137 }
 138 #endif // __cplusplus