src/support/docstream.cpp

   1 /**
   2  * \file docstream.cpp
   3  * This file is part of LyX, the document processor.
   4  * Licence details can be found in the file COPYING.
   5  *
   6  * \author Georg Baum
   7  *
   8  * Full author contact details are available in file CREDITS.
   9  */
  10
  11 #include <config.h>
  12
  13 #include "support/docstream.h"
  14 #include "support/unicode.h"
  15
  16 #include <cerrno>
  17 #include <cstdio>
  18 #include <cstring>
  19 #include <iconv.h>
  20 #include <locale>
  21
  22 using namespace std;
  23
  24 using lyx::ucs4_codeset;
  25
  26 namespace {
  27
  28 // We use C IO throughout this file, because the facets might be used with
  29 // lyxerr in the future.
  30
  31
  32 /// codecvt facet for conversion of UCS4 (internal representation) to UTF8
  33 /// (external representation) or vice versa
  34 class iconv_codecvt_facet : public codecvt<lyx::char_type, char, mbstate_t>
  35 {
  36         typedef codecvt<lyx::char_type, char, mbstate_t> base;
  37 public:
  38         /// Constructor. You have to specify with \p inout whether you want
  39         /// to use this facet only for input, only for output or for both.
  40         explicit iconv_codecvt_facet(string const & encoding = "UTF-8",
  41                         ios_base::openmode inout = ios_base::in | ios_base::out,
  42                         size_t refs = 0)
  43                 : base(refs), encoding_(encoding)
  44         {
  45                 if (inout & ios_base::in) {
  46                         in_cd_ = iconv_open(ucs4_codeset, encoding.c_str());
  47                         if (in_cd_ == (iconv_t)(-1)) {
  48                                 fprintf(stderr, "Error %d returned from iconv_open(in_cd_): %s\n",
  49                                         errno, strerror(errno));
  50                                 fflush(stderr);
  51                                 throw lyx::iconv_codecvt_facet_exception();
  52                         }
  53                 } else
  54                         in_cd_ = (iconv_t)(-1);
  55                 if (inout & ios_base::out) {
  56                         out_cd_ = iconv_open(encoding.c_str(), ucs4_codeset);
  57                         if (out_cd_ == (iconv_t)(-1)) {
  58                                 fprintf(stderr, "Error %d returned from iconv_open(out_cd_): %s\n",
  59                                         errno, strerror(errno));
  60                                 fflush(stderr);
  61                                 throw lyx::iconv_codecvt_facet_exception();
  62                         }
  63                 } else
  64                         out_cd_ = (iconv_t)(-1);
  65         }
  66 protected:
  67         virtual ~iconv_codecvt_facet()
  68         {
  69                 if (in_cd_ != (iconv_t)(-1))
  70                         if (iconv_close(in_cd_) == -1) {
  71                                 fprintf(stderr, "Error %d returned from iconv_close(in_cd_): %s\n",
  72                                         errno, strerror(errno));
  73                                 fflush(stderr);
  74                         }
  75                 if (out_cd_ != (iconv_t)(-1))
  76                         if (iconv_close(out_cd_) == -1) {
  77                                 fprintf(stderr, "Error %d returned from iconv_close(out_cd_): %s\n",
  78                                         errno, strerror(errno));
  79                                 fflush(stderr);
  80                         }
  81         }
  82         virtual result do_out(state_type &, intern_type const * from,
  83                         intern_type const * from_end, intern_type const *& from_next,
  84                         extern_type * to, extern_type * to_end,
  85                         extern_type *& to_next) const
  86         {
  87 #define WORKAROUND_ICONV_BUG 1
  88 #if WORKAROUND_ICONV_BUG
  89                 // Due to a bug in some iconv versions, when the last char
  90                 // in the buffer is a wide char and the output encoding is
  91                 // ISO-2022-JP and we are going to switch to another encoding,
  92                 // the appropriate escape sequence for changing the character
  93                 // set is not output (see bugs 5216, 5280, and also 5489).
  94                 // As a workaround, we append a nul char in order to force
  95                 // a switch to ASCII, and then remove it from output after
  96                 // the conversion.
  97                 intern_type * from_new = 0;
  98                 intern_type const * from_old = from;
  99                 size_t extra = 0;
 100                 if (*(from_end - 1) >= 0x80 && encoding_ == "ISO-2022-JP") {
 101                         size_t len = from_end - from;
 102                         from_new = new intern_type[len + 1];
 103                         memcpy(from_new, from, len * sizeof(intern_type));
 104                         from_new[len] = 0;
 105                         from_end = from_new + len + 1;
 106                         from = from_new;
 107                         extra = 1;
 108                 }
 109 #endif
 110                 size_t inbytesleft = (from_end - from) * sizeof(intern_type);
 111                 size_t outbytesleft = (to_end - to) * sizeof(extern_type);
 112 #if WORKAROUND_ICONV_BUG
 113                 outbytesleft += extra * sizeof(extern_type);
 114 #endif
 115                 from_next = from;
 116                 to_next = to;
 117                 result const retval = do_iconv(out_cd_,
 118                                 reinterpret_cast<char const **>(&from_next),
 119                                 &inbytesleft, &to_next, &outbytesleft);
 120 #if WORKAROUND_ICONV_BUG
 121                 // Remove from output the nul char that we inserted at the end
 122                 // of the input buffer in order to circumvent an iconv bug.
 123                 if (from_new) {
 124                         --to_next;
 125                         --from_next;
 126                         from_next = from_old + (from_next - from);
 127                         from = from_old;
 128                         delete[] from_new;
 129                 }
 130 #endif
 131                 if (retval == base::error) {
 132                         fprintf(stderr,
 133                                 "Error %d returned from iconv when converting from %s to %s: %s\n",
 134                                 errno, ucs4_codeset, encoding_.c_str(),
 135                                 strerror(errno));
 136                         fputs("Converted input:", stderr);
 137                         for (intern_type const * i = from; i < from_next; ++i) {
 138                                 unsigned int const c = *i;
 139                                 fprintf(stderr, " 0x%04x", c);
 140                         }
 141                         unsigned int const c = *from_next;
 142                         fprintf(stderr, "\nStopped at: 0x%04x\n", c);
 143                         fputs("Unconverted input:", stderr);
 144                         for (intern_type const * i = from_next + 1; i < from_end; ++i) {
 145                                 unsigned int const c = *i;
 146                                 fprintf(stderr, " 0x%04x", c);
 147                         }
 148                         fputs("\nConverted output:", stderr);
 149                         for (extern_type const * i = to; i < to_next; ++i) {
 150                                 // extern_type may be signed, avoid output of
 151                                 // something like 0xffffffc2
 152                                 unsigned int const c =
 153                                         *reinterpret_cast<unsigned char const *>(i);
 154                                 fprintf(stderr, " 0x%02x", c);
 155                         }
 156                         fputc('\n', stderr);
 157                         fflush(stderr);
 158                 }
 159                 return retval;
 160         }
 161         virtual result do_unshift(state_type &, extern_type * to,
 162                         extern_type *, extern_type *& to_next) const
 163         {
 164                 // utf8 does not use shifting
 165                 to_next = to;
 166                 return base::noconv;
 167         }
 168         virtual result do_in(state_type &,
 169                         extern_type const * from, extern_type const * from_end,
 170                         extern_type const *& from_next,
 171                         intern_type * to, intern_type * to_end,
 172                         intern_type *& to_next) const
 173         {
 174                 size_t inbytesleft = (from_end - from) * sizeof(extern_type);
 175                 size_t outbytesleft = (to_end - to) * sizeof(intern_type);
 176                 from_next = from;
 177                 to_next = to;
 178                 result const retval = do_iconv(in_cd_, &from_next, &inbytesleft,
 179                                 reinterpret_cast<char **>(&to_next),
 180                                 &outbytesleft);
 181                 if (retval == base::error) {
 182                         fprintf(stderr,
 183                                 "Error %d returned from iconv when converting from %s to %s: %s\n",
 184                                 errno, encoding_.c_str(), ucs4_codeset,
 185                                 strerror(errno));
 186                         fputs("Converted input:", stderr);
 187                         for (extern_type const * i = from; i < from_next; ++i) {
 188                                 // extern_type may be signed, avoid output of
 189                                 // something like 0xffffffc2
 190                                 unsigned int const c =
 191                                         *reinterpret_cast<unsigned char const *>(i);
 192                                 fprintf(stderr, " 0x%02x", c);
 193                         }
 194                         unsigned int const c =
 195                                 *reinterpret_cast<unsigned char const *>(from_next);
 196                         fprintf(stderr, "\nStopped at: 0x%02x\n", c);
 197                         fputs("Unconverted input:", stderr);
 198                         for (extern_type const * i = from_next + 1; i < from_end; ++i) {
 199                                 unsigned int const c =
 200                                         *reinterpret_cast<unsigned char const *>(i);
 201                                 fprintf(stderr, " 0x%02x", c);
 202                         }
 203                         fputs("\nConverted output:", stderr);
 204                         for (intern_type const * i = to; i < to_next; ++i) {
 205                                 unsigned int const c = *i;
 206                                 fprintf(stderr, " 0x%02x", c);
 207                         }
 208                         fputc('\n', stderr);
 209                         fflush(stderr);
 210                 }
 211                 return retval;
 212         }
 213         virtual int do_encoding() const throw()
 214         {
 215                 return 0;
 216         }
 217         virtual bool do_always_noconv() const throw()
 218         {
 219                 return false;
 220         }
 221         virtual int do_length(state_type & /*state*/, extern_type const * from,
 222                         extern_type const * end, size_t max) const
 223         {
 224                 // The docs are a bit unclear about this method.
 225                 // It seems that we should calculate the actual length of the
 226                 // converted sequence, but that would not make sense, since
 227                 // once could just do the conversion directly.
 228                 // Therefore we just return the number of unconverted
 229                 // characters, since that is the best guess we can do.
 230 #if 0
 231                 intern_type * to = new intern_type[max];
 232                 intern_type * to_end = to + max;
 233                 intern_type * to_next = to;
 234                 extern_type const * from_next = from;
 235                 do_in(state, from, end, from_next, to, to_end, to_next);
 236                 delete[] to;
 237                 return to_next - to;
 238 #else
 239                 size_t const length = end - from;
 240                 return min(length, max);
 241 #endif
 242         }
 243         virtual int do_max_length() const throw()
 244         {
 245                 return lyx::max_encoded_bytes(encoding_);
 246         }
 247 private:
 248         /// Do the actual conversion. The interface is equivalent to that of
 249         /// iconv() (but const correct).
 250         inline base::result do_iconv(iconv_t cd, char const ** from,
 251                         size_t * inbytesleft, char ** to, size_t * outbytesleft) const
 252         {
 253                 char const * const to_start = *to;
 254                 size_t converted = iconv(cd, const_cast<char ICONV_CONST **>(from),
 255                                 inbytesleft, to, outbytesleft);
 256                 if (converted == (size_t)(-1)) {
 257                         switch(errno) {
 258                         case 0:
 259                                 // As strange as it may seem, this
 260                                 // does happen on windows when parsing
 261                                 // comments with accented chars in
 262                                 // tex2lyx. See the following thread
 263                                 // for details
 264                                 // http://thread.gmane.org/gmane.editors.lyx.devel/117636
 265                                 break;
 266                         case EINVAL:
 267                         case E2BIG:
 268                                 return base::partial;
 269                         case EILSEQ:
 270                         default:
 271                                 return base::error;
 272                         }
 273                 }
 274                 if (*to == to_start)
 275                         return base::noconv;
 276                 return base::ok;
 277         }
 278         iconv_t in_cd_;
 279         iconv_t out_cd_;
 280         /// The narrow encoding
 281         string encoding_;
 282 };
 283
 284 } // namespace anon
 285
 286
 287 namespace lyx {
 288
 289 template<class Ios>
 290 void setEncoding(Ios & ios, string const & encoding, ios_base::openmode mode)
 291 {
 292         // We must imbue the stream before openening the file
 293         locale global;
 294         locale locale(global, new iconv_codecvt_facet(encoding, mode));
 295         ios.imbue(locale);
 296 }
 297
 298
 299 const char * iconv_codecvt_facet_exception::what() const throw()
 300 {
 301         return "iconv problem in iconv_codecvt_facet initialization";
 302 }
 303
 304
 305 ifdocstream::ifdocstream() : base()
 306 {
 307         setEncoding(*this, "UTF-8", in);
 308 }
 309
 310
 311 ifdocstream::ifdocstream(const char* s, ios_base::openmode mode,
 312                          string const & encoding)
 313         : base()
 314 {
 315         setEncoding(*this, encoding, in);
 316         open(s, mode);
 317 }
 318
 319
 320 ofdocstream::ofdocstream(): base()
 321 {
 322         setEncoding(*this, "UTF-8", out);
 323 }
 324
 325
 326 ofdocstream::ofdocstream(const char* s, ios_base::openmode mode,
 327                          string const & encoding)
 328         : base()
 329 {
 330         setEncoding(*this, encoding, out);
 331         open(s, mode);
 332 }
 333
 334
 335 void ofdocstream::reset(string const & encoding)
 336 {
 337         setEncoding(*this, encoding, out);
 338 }
 339
 340
 341
 342 SetEnc setEncoding(string const & encoding)
 343 {
 344         return SetEnc(encoding);
 345 }
 346
 347
 348 odocstream & operator<<(odocstream & os, SetEnc e)
 349 {
 350         if (has_facet<iconv_codecvt_facet>(os.rdbuf()->getloc())) {
 351                 // This stream must be a file stream, since we never imbue
 352                 // any other stream with a locale having a iconv_codecvt_facet.
 353                 // Flush the stream so that all pending output is written
 354                 // with the old encoding.
 355                 os.flush();
 356                 locale locale(os.rdbuf()->getloc(),
 357                         new iconv_codecvt_facet(e.encoding, ios_base::out));
 358                 // FIXME Does changing the codecvt facet of an open file
 359                 // stream always work? It does with gcc 4.1, but I have read
 360                 // somewhere that it does not with MSVC.
 361                 // What does the standard say?
 362                 os.imbue(locale);
 363         }
 364         return os;
 365 }
 366
 367
 368 //CHECKME: I just copied the code above, and have no idea whether it
 369 //is correct... (JMarc)
 370 idocstream & operator<<(idocstream & is, SetEnc e)
 371 {
 372         if (has_facet<iconv_codecvt_facet>(is.rdbuf()->getloc())) {
 373                 // This stream must be a file stream, since we never imbue
 374                 // any other stream with a locale having a iconv_codecvt_facet.
 375                 // Flush the stream so that all pending output is written
 376                 // with the old encoding.
 377                 //is.flush();
 378                 locale locale(is.rdbuf()->getloc(),
 379                         new iconv_codecvt_facet(e.encoding, ios_base::in));
 380                 // FIXME Does changing the codecvt facet of an open file
 381                 // stream always work? It does with gcc 4.1, but I have read
 382                 // somewhere that it does not with MSVC.
 383                 // What does the standard say?
 384                 is.imbue(locale);
 385         }
 386         return is;
 387 }
 388
 389
 390 #if ! defined(USE_WCHAR_T)
 391 odocstream & operator<<(odocstream & os, char c)
 392 {
 393         os.put(c);
 394         return os;
 395 }
 396 #endif
 397
 398 }
 399
 400 #if ! defined(USE_WCHAR_T) && defined(__GNUC__)
 401 // We get undefined references to these virtual methods. This looks like
 402 // a bug in gcc. The implementation here does not do anything useful, since
 403 // it is overriden in iconv_codecvt_facet.
 404 namespace std {
 405
 406 template<> codecvt<lyx::char_type, char, mbstate_t>::result
 407 codecvt<lyx::char_type, char, mbstate_t>::do_out(
 408         mbstate_t &, const lyx::char_type *, const lyx::char_type *,
 409         const lyx::char_type *&, char *, char *, char *&) const
 410 {
 411         return error;
 412 }
 413
 414
 415 template<> codecvt<lyx::char_type, char, mbstate_t>::result
 416 codecvt<lyx::char_type, char, mbstate_t>::do_unshift(
 417         mbstate_t &, char *, char *, char *&) const
 418 {
 419         return error;
 420 }
 421
 422
 423 template<> codecvt<lyx::char_type, char, mbstate_t>::result
 424 codecvt<lyx::char_type, char, mbstate_t>::do_in(
 425         mbstate_t &, const char *, const char *, const char *&,
 426         lyx::char_type *, lyx::char_type *, lyx::char_type *&) const
 427 {
 428         return error;
 429 }
 430
 431
 432 template<>
 433 int codecvt<lyx::char_type, char, mbstate_t>::do_encoding() const throw()
 434 {
 435         return 0;
 436 }
 437
 438
 439 template<>
 440 bool codecvt<lyx::char_type, char, mbstate_t>::do_always_noconv() const throw()
 441 {
 442         return true;
 443 }
 444
 445 #if __GNUC__ == 3 && __GNUC_MINOR__ < 4
 446
 447 template<>
 448 int codecvt<lyx::char_type, char, mbstate_t>::do_length(
 449         mbstate_t const &, const char *, const char *, size_t) const
 450 {
 451         return 1;
 452 }
 453
 454 #else
 455
 456 template<>
 457 int codecvt<lyx::char_type, char, mbstate_t>::do_length(
 458         mbstate_t &, const char *, const char *, size_t) const
 459 {
 460         return 1;
 461 }
 462
 463 #endif
 464
 465 template<>
 466 int codecvt<lyx::char_type, char, mbstate_t>::do_max_length() const throw()
 467 {
 468         return 4;
 469 }
 470
 471 } // namespace std
 472 #endif