tools/wmc/utils.c

   1 /*
   2  * Utility routines
   3  *
   4  * Copyright 1998,2000 Bertho A. Stultiens
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) 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 GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #include "config.h"
  22 #include "wine/port.h"
  23
  24 #include <stdio.h>
  25 #include <stdlib.h>
  26 #include <stdarg.h>
  27 #include <string.h>
  28 #include <assert.h>
  29 #include <ctype.h>
  30
  31 #include "wmctypes.h"
  32 #include "winnls.h"
  33 #include "utils.h"
  34 #include "wmc.h"
  35
  36 #define SUPPRESS_YACC_ERROR_MESSAGE
  37
  38 static void generic_msg(const char *s, const char *t, va_list ap)
  39 {
  40         fprintf(stderr, "%s:%d:%d: %s: ", input_name ? input_name : "stdin", line_number, char_number, t);
  41         vfprintf(stderr, s, ap);
  42 }
  43
  44 /*
  45  * The yyerror routine should not exit because we use the error-token
  46  * to determine the syntactic error in the source. However, YACC
  47  * uses the same routine to print an error just before the error
  48  * token is reduced.
  49  * The extra routine 'xyyerror' is used to exit after giving a real
  50  * message.
  51  */
  52 int mcy_error(const char *s, ...)
  53 {
  54 #ifndef SUPPRESS_YACC_ERROR_MESSAGE
  55         va_list ap;
  56         va_start(ap, s);
  57         generic_msg(s, "Yacc error", ap);
  58         va_end(ap);
  59 #endif
  60         return 1;
  61 }
  62
  63 int xyyerror(const char *s, ...)
  64 {
  65         va_list ap;
  66         va_start(ap, s);
  67         generic_msg(s, "Error", ap);
  68         va_end(ap);
  69         exit(1);
  70         return 1;
  71 }
  72
  73 int mcy_warning(const char *s, ...)
  74 {
  75         va_list ap;
  76         va_start(ap, s);
  77         generic_msg(s, "Warning", ap);
  78         va_end(ap);
  79         return 0;
  80 }
  81
  82 void internal_error(const char *file, int line, const char *s, ...)
  83 {
  84         va_list ap;
  85         va_start(ap, s);
  86         fprintf(stderr, "Internal error (please report) %s %d: ", file, line);
  87         vfprintf(stderr, s, ap);
  88         va_end(ap);
  89         exit(3);
  90 }
  91
  92 void fatal_perror( const char *msg, ... )
  93 {
  94         va_list valist;
  95         va_start( valist, msg );
  96         fprintf(stderr, "Error: ");
  97         vfprintf( stderr, msg, valist );
  98         perror( " " );
  99         va_end( valist );
 100         exit(2);
 101 }
 102
 103 void error(const char *s, ...)
 104 {
 105         va_list ap;
 106         va_start(ap, s);
 107         fprintf(stderr, "Error: ");
 108         vfprintf(stderr, s, ap);
 109         va_end(ap);
 110         exit(2);
 111 }
 112
 113 void warning(const char *s, ...)
 114 {
 115         va_list ap;
 116         va_start(ap, s);
 117         fprintf(stderr, "Warning: ");
 118         vfprintf(stderr, s, ap);
 119         va_end(ap);
 120 }
 121
 122 char *dup_basename(const char *name, const char *ext)
 123 {
 124         int namelen;
 125         int extlen = strlen(ext);
 126         char *base;
 127         char *slash;
 128
 129         if(!name)
 130                 name = "wmc.tab";
 131
 132         slash = strrchr(name, '/');
 133         if (slash)
 134                 name = slash + 1;
 135
 136         namelen = strlen(name);
 137
 138         /* +4 for later extension and +1 for '\0' */
 139         base = xmalloc(namelen +4 +1);
 140         strcpy(base, name);
 141         if(!strcasecmp(name + namelen-extlen, ext))
 142         {
 143                 base[namelen - extlen] = '\0';
 144         }
 145         return base;
 146 }
 147
 148 void *xmalloc(size_t size)
 149 {
 150     void *res;
 151
 152     assert(size > 0);
 153     res = malloc(size);
 154     if(res == NULL)
 155     {
 156         error("Virtual memory exhausted.\n");
 157     }
 158     memset(res, 0x55, size);
 159     return res;
 160 }
 161
 162
 163 void *xrealloc(void *p, size_t size)
 164 {
 165     void *res;
 166
 167     assert(size > 0);
 168     res = realloc(p, size);
 169     if(res == NULL)
 170     {
 171         error("Virtual memory exhausted.\n");
 172     }
 173     return res;
 174 }
 175
 176 char *xstrdup(const char *str)
 177 {
 178         char *s;
 179
 180         assert(str != NULL);
 181         s = xmalloc(strlen(str)+1);
 182         return strcpy(s, str);
 183 }
 184
 185 char *strmake( const char* fmt, ... )
 186 {
 187     int n;
 188     size_t size = 100;
 189     va_list ap;
 190
 191     for (;;)
 192     {
 193         char *p = xmalloc( size );
 194         va_start( ap, fmt );
 195         n = vsnprintf( p, size, fmt, ap );
 196         va_end( ap );
 197         if (n == -1) size *= 2;
 198         else if ((size_t)n >= size) size = n + 1;
 199         else return p;
 200         free( p );
 201     }
 202 }
 203
 204 int strendswith( const char *str, const char *end )
 205 {
 206     int l = strlen(str);
 207     int m = strlen(end);
 208     return l >= m && !strcmp( str + l - m, end );
 209 }
 210
 211 int unistrlen(const WCHAR *s)
 212 {
 213         int n;
 214         for(n = 0; *s; n++, s++)
 215                 ;
 216         return n;
 217 }
 218
 219 WCHAR *unistrcpy(WCHAR *dst, const WCHAR *src)
 220 {
 221         WCHAR *t = dst;
 222         while(*src)
 223                 *t++ = *src++;
 224         *t = 0;
 225         return dst;
 226 }
 227
 228 WCHAR *xunistrdup(const WCHAR * str)
 229 {
 230         WCHAR *s;
 231
 232         assert(str != NULL);
 233         s = xmalloc((unistrlen(str)+1) * sizeof(WCHAR));
 234         return unistrcpy(s, str);
 235 }
 236
 237 int unistricmp(const WCHAR *s1, const WCHAR *s2)
 238 {
 239         int i;
 240         int once = 0;
 241         static const char warn[] = "Don't know the uppercase equivalent of non ascii characters;"
 242                              "comparison might yield wrong results";
 243         while(*s1 && *s2)
 244         {
 245                 if((*s1 & 0xffff) > 0x7f || (*s2 & 0xffff) > 0x7f)
 246                 {
 247                         if(!once)
 248                         {
 249                                 once++;
 250                                 mcy_warning(warn);
 251                         }
 252                         i = *s1++ - *s2++;
 253                 }
 254                 else
 255                         i = toupper(*s1++) - toupper(*s2++);
 256                 if(i)
 257                         return i;
 258         }
 259
 260         if((*s1 & 0xffff) > 0x7f || (*s2 & 0xffff) > 0x7f)
 261         {
 262                 if(!once)
 263                         mcy_warning(warn);
 264                 return *s1 - *s2;
 265         }
 266         else
 267                 return  toupper(*s1) - toupper(*s2);
 268 }
 269
 270 int unistrcmp(const WCHAR *s1, const WCHAR *s2)
 271 {
 272         int i;
 273         while(*s1 && *s2)
 274         {
 275                 i = *s1++ - *s2++;
 276                 if(i)
 277                         return i;
 278         }
 279
 280         return *s1 - *s2;
 281 }
 282
 283 WCHAR *utf8_to_unicode( const char *src, int srclen, int *dstlen )
 284 {
 285     static const char utf8_length[128] =
 286     {
 287         0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0, /* 0x80-0x8f */
 288         0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0, /* 0x90-0x9f */
 289         0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0, /* 0xa0-0xaf */
 290         0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0, /* 0xb0-0xbf */
 291         0,0,1,1,1,1,1,1,1,1,1,1,1,1,1,1, /* 0xc0-0xcf */
 292         1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1, /* 0xd0-0xdf */
 293         2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2, /* 0xe0-0xef */
 294         3,3,3,3,3,0,0,0,0,0,0,0,0,0,0,0  /* 0xf0-0xff */
 295     };
 296     static const unsigned char utf8_mask[4] = { 0x7f, 0x1f, 0x0f, 0x07 };
 297
 298     const char *srcend = src + srclen;
 299     int len, res;
 300     WCHAR *ret, *dst;
 301
 302     dst = ret = xmalloc( (srclen + 1) * sizeof(WCHAR) );
 303     while (src < srcend)
 304     {
 305         unsigned char ch = *src++;
 306         if (ch < 0x80)  /* special fast case for 7-bit ASCII */
 307         {
 308             *dst++ = ch;
 309             continue;
 310         }
 311         len = utf8_length[ch - 0x80];
 312         if (len && src + len <= srcend)
 313         {
 314             res = ch & utf8_mask[len];
 315             switch (len)
 316             {
 317             case 3:
 318                 if ((ch = *src ^ 0x80) >= 0x40) break;
 319                 res = (res << 6) | ch;
 320                 src++;
 321                 if (res < 0x10) break;
 322             case 2:
 323                 if ((ch = *src ^ 0x80) >= 0x40) break;
 324                 res = (res << 6) | ch;
 325                 if (res >= 0x110000 >> 6) break;
 326                 src++;
 327                 if (res < 0x20) break;
 328                 if (res >= 0xd800 >> 6 && res <= 0xdfff >> 6) break;
 329             case 1:
 330                 if ((ch = *src ^ 0x80) >= 0x40) break;
 331                 res = (res << 6) | ch;
 332                 src++;
 333                 if (res < 0x80) break;
 334                 if (res <= 0xffff) *dst++ = res;
 335                 else
 336                 {
 337                     res -= 0x10000;
 338                     *dst++ = 0xd800 | (res >> 10);
 339                     *dst++ = 0xdc00 | (res & 0x3ff);
 340                 }
 341                 continue;
 342             }
 343         }
 344         *dst++ = 0xfffd;
 345     }
 346     *dst = 0;
 347     *dstlen = dst - ret;
 348     return ret;
 349 }
 350
 351 char *unicode_to_utf8( const WCHAR *src, int srclen, int *dstlen )
 352 {
 353     char *ret, *dst;
 354
 355     dst = ret = xmalloc( srclen * 3 + 1 );
 356     for ( ; srclen; srclen--, src++)
 357     {
 358         unsigned int ch = *src;
 359
 360         if (ch < 0x80)  /* 0x00-0x7f: 1 byte */
 361         {
 362             *dst++ = ch;
 363             continue;
 364         }
 365         if (ch < 0x800)  /* 0x80-0x7ff: 2 bytes */
 366         {
 367             dst[1] = 0x80 | (ch & 0x3f);
 368             ch >>= 6;
 369             dst[0] = 0xc0 | ch;
 370             dst += 2;
 371             continue;
 372         }
 373         if (ch >= 0xd800 && ch <= 0xdbff && srclen > 1 && src[1] >= 0xdc00 && src[1] <= 0xdfff)
 374         {
 375             /* 0x10000-0x10ffff: 4 bytes */
 376             ch = 0x10000 + ((ch & 0x3ff) << 10) + (src[1] & 0x3ff);
 377             dst[3] = 0x80 | (ch & 0x3f);
 378             ch >>= 6;
 379             dst[2] = 0x80 | (ch & 0x3f);
 380             ch >>= 6;
 381             dst[1] = 0x80 | (ch & 0x3f);
 382             ch >>= 6;
 383             dst[0] = 0xf0 | ch;
 384             dst += 4;
 385             src++;
 386             srclen--;
 387             continue;
 388         }
 389         if (ch >= 0xd800 && ch <= 0xdfff) ch = 0xfffd;  /* invalid surrogate pair */
 390
 391         /* 0x800-0xffff: 3 bytes */
 392         dst[2] = 0x80 | (ch & 0x3f);
 393         ch >>= 6;
 394         dst[1] = 0x80 | (ch & 0x3f);
 395         ch >>= 6;
 396         dst[0] = 0xe0 | ch;
 397         dst += 3;
 398     }
 399     *dst = 0;
 400     *dstlen = dst - ret;
 401     return ret;
 402 }
 403
 404 #ifdef _WIN32
 405
 406 int is_valid_codepage(int id)
 407 {
 408     return IsValidCodePage( id );
 409 }
 410
 411 WCHAR *codepage_to_unicode( int codepage, const char *src, int srclen, int *dstlen )
 412 {
 413     WCHAR *dst = xmalloc( (srclen + 1) * sizeof(WCHAR) );
 414     DWORD ret = MultiByteToWideChar( codepage, MB_ERR_INVALID_CHARS, src, srclen, dst, srclen );
 415     if (!ret) return NULL;
 416     dst[ret] = 0;
 417     *dstlen = ret;
 418     return dst;
 419 }
 420
 421 #else  /* _WIN32 */
 422
 423 struct nls_info
 424 {
 425     unsigned short  codepage;
 426     unsigned short  unidef;
 427     unsigned short  trans_unidef;
 428     unsigned short *cp2uni;
 429     unsigned short *dbcs_offsets;
 430 };
 431
 432 static struct nls_info nlsinfo[128];
 433
 434 static void init_nls_info( struct nls_info *info, unsigned short *ptr )
 435 {
 436     unsigned short hdr_size = ptr[0];
 437
 438     info->codepage      = ptr[1];
 439     info->unidef        = ptr[4];
 440     info->trans_unidef  = ptr[6];
 441     ptr += hdr_size;
 442     info->cp2uni = ++ptr;
 443     ptr += 256;
 444     if (*ptr++) ptr += 256;  /* glyph table */
 445     info->dbcs_offsets  = *ptr ? ptr + 1 : NULL;
 446 }
 447
 448 static const struct nls_info *get_nls_info( unsigned int codepage )
 449 {
 450     struct stat st;
 451     unsigned short *data;
 452     char *path;
 453     unsigned int i;
 454     int fd;
 455
 456     for (i = 0; i < ARRAY_SIZE(nlsinfo) && nlsinfo[i].codepage; i++)
 457         if (nlsinfo[i].codepage == codepage) return &nlsinfo[i];
 458
 459     assert( i < ARRAY_SIZE(nlsinfo) );
 460
 461     for (i = 0; nlsdirs[i]; i++)
 462     {
 463         path = strmake( "%s/c_%03u.nls", nlsdirs[i], codepage );
 464         if ((fd = open( path, O_RDONLY )) != -1) break;
 465         free( path );
 466     }
 467     if (!nlsdirs[i]) return NULL;
 468
 469     fstat( fd, &st );
 470     data = xmalloc( st.st_size );
 471     if (read( fd, data, st.st_size ) != st.st_size) error( "failed to load %s\n", path );
 472     close( fd );
 473     free( path );
 474     init_nls_info( &nlsinfo[i], data );
 475     return &nlsinfo[i];
 476 }
 477
 478 int is_valid_codepage(int cp)
 479 {
 480     return cp == CP_UTF8 || get_nls_info( cp );
 481 }
 482
 483 WCHAR *codepage_to_unicode( int codepage, const char *src, int srclen, int *dstlen )
 484 {
 485     const struct nls_info *info = get_nls_info( codepage );
 486     unsigned int i;
 487     WCHAR dbch, *dst = xmalloc( (srclen + 1) * sizeof(WCHAR) );
 488
 489     if (!info) error( "codepage %u not supported\n", codepage );
 490
 491     if (info->dbcs_offsets)
 492     {
 493         for (i = 0; srclen; i++, srclen--, src++)
 494         {
 495             unsigned short off = info->dbcs_offsets[(unsigned char)*src];
 496             if (off)
 497             {
 498                 if (srclen == 1) return NULL;
 499                 dbch = (src[0] << 8) | (unsigned char)src[1];
 500                 src++;
 501                 srclen--;
 502                 dst[i] = info->dbcs_offsets[off + (unsigned char)*src];
 503                 if (dst[i] == info->unidef && dbch != info->trans_unidef) return NULL;
 504             }
 505             else
 506             {
 507                 dst[i] = info->cp2uni[(unsigned char)*src];
 508                 if (dst[i] == info->unidef && *src != info->trans_unidef) return NULL;
 509             }
 510         }
 511     }
 512     else
 513     {
 514         for (i = 0; i < srclen; i++)
 515         {
 516             dst[i] = info->cp2uni[(unsigned char)src[i]];
 517             if (dst[i] == info->unidef && src[i] != info->trans_unidef) return NULL;
 518         }
 519     }
 520     dst[i] = 0;
 521     *dstlen = i;
 522     return dst;
 523 }
 524
 525 #endif  /* _WIN32 */
 526
 527 /*******************************************************************
 528  *         buffer management
 529  *
 530  * Function for writing to a memory buffer.
 531  */
 532
 533 int byte_swapped = 0;
 534 unsigned char *output_buffer;
 535 size_t output_buffer_pos;
 536 size_t output_buffer_size;
 537
 538 static void check_output_buffer_space( size_t size )
 539 {
 540     if (output_buffer_pos + size >= output_buffer_size)
 541     {
 542         output_buffer_size = max( output_buffer_size * 2, output_buffer_pos + size );
 543         output_buffer = xrealloc( output_buffer, output_buffer_size );
 544     }
 545 }
 546
 547 void init_output_buffer(void)
 548 {
 549     output_buffer_size = 1024;
 550     output_buffer_pos = 0;
 551     output_buffer = xmalloc( output_buffer_size );
 552 }
 553
 554 void flush_output_buffer( const char *name )
 555 {
 556     int fd = open( name, O_WRONLY | O_CREAT | O_TRUNC | O_BINARY, 0666 );
 557     if (fd == -1) error( "Error creating %s\n", name );
 558     if (write( fd, output_buffer, output_buffer_pos ) != output_buffer_pos)
 559         error( "Error writing to %s\n", name );
 560     close( fd );
 561     free( output_buffer );
 562 }
 563
 564 void put_data( const void *data, size_t size )
 565 {
 566     check_output_buffer_space( size );
 567     memcpy( output_buffer + output_buffer_pos, data, size );
 568     output_buffer_pos += size;
 569 }
 570
 571 void put_byte( unsigned char val )
 572 {
 573     check_output_buffer_space( 1 );
 574     output_buffer[output_buffer_pos++] = val;
 575 }
 576
 577 void put_word( unsigned short val )
 578 {
 579     if (byte_swapped) val = (val << 8) | (val >> 8);
 580     put_data( &val, sizeof(val) );
 581 }
 582
 583 void put_dword( unsigned int val )
 584 {
 585     if (byte_swapped)
 586         val = ((val << 24) | ((val << 8) & 0x00ff0000) | ((val >> 8) & 0x0000ff00) | (val >> 24));
 587     put_data( &val, sizeof(val) );
 588 }
 589
 590 void align_output( unsigned int align )
 591 {
 592     size_t size = align - (output_buffer_pos % align);
 593
 594     if (size == align) return;
 595     check_output_buffer_space( size );
 596     memset( output_buffer + output_buffer_pos, 0, size );
 597     output_buffer_pos += size;
 598 }