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 "utils.h"
  33 #include "wmc.h"
  34
  35 #define SUPPRESS_YACC_ERROR_MESSAGE
  36
  37 static void generic_msg(const char *s, const char *t, va_list ap)
  38 {
  39         fprintf(stderr, "%s:%d:%d: %s: ", input_name ? input_name : "stdin", line_number, char_number, t);
  40         vfprintf(stderr, s, ap);
  41 }
  42
  43 /*
  44  * The yyerror routine should not exit because we use the error-token
  45  * to determine the syntactic error in the source. However, YACC
  46  * uses the same routine to print an error just before the error
  47  * token is reduced.
  48  * The extra routine 'xyyerror' is used to exit after giving a real
  49  * message.
  50  */
  51 int mcy_error(const char *s, ...)
  52 {
  53 #ifndef SUPPRESS_YACC_ERROR_MESSAGE
  54         va_list ap;
  55         va_start(ap, s);
  56         generic_msg(s, "Yacc error", ap);
  57         va_end(ap);
  58 #endif
  59         return 1;
  60 }
  61
  62 int xyyerror(const char *s, ...)
  63 {
  64         va_list ap;
  65         va_start(ap, s);
  66         generic_msg(s, "Error", ap);
  67         va_end(ap);
  68         exit(1);
  69         return 1;
  70 }
  71
  72 int mcy_warning(const char *s, ...)
  73 {
  74         va_list ap;
  75         va_start(ap, s);
  76         generic_msg(s, "Warning", ap);
  77         va_end(ap);
  78         return 0;
  79 }
  80
  81 void internal_error(const char *file, int line, const char *s, ...)
  82 {
  83         va_list ap;
  84         va_start(ap, s);
  85         fprintf(stderr, "Internal error (please report) %s %d: ", file, line);
  86         vfprintf(stderr, s, ap);
  87         va_end(ap);
  88         exit(3);
  89 }
  90
  91 void fatal_perror( const char *msg, ... )
  92 {
  93         va_list valist;
  94         va_start( valist, msg );
  95         fprintf(stderr, "Error: ");
  96         vfprintf( stderr, msg, valist );
  97         perror( " " );
  98         va_end( valist );
  99         exit(2);
 100 }
 101
 102 void error(const char *s, ...)
 103 {
 104         va_list ap;
 105         va_start(ap, s);
 106         fprintf(stderr, "Error: ");
 107         vfprintf(stderr, s, ap);
 108         va_end(ap);
 109         exit(2);
 110 }
 111
 112 void warning(const char *s, ...)
 113 {
 114         va_list ap;
 115         va_start(ap, s);
 116         fprintf(stderr, "Warning: ");
 117         vfprintf(stderr, s, ap);
 118         va_end(ap);
 119 }
 120
 121 char *dup_basename(const char *name, const char *ext)
 122 {
 123         int namelen;
 124         int extlen = strlen(ext);
 125         char *base;
 126         char *slash;
 127
 128         if(!name)
 129                 name = "wmc.tab";
 130
 131         slash = strrchr(name, '/');
 132         if (slash)
 133                 name = slash + 1;
 134
 135         namelen = strlen(name);
 136
 137         /* +4 for later extension and +1 for '\0' */
 138         base = xmalloc(namelen +4 +1);
 139         strcpy(base, name);
 140         if(!strcasecmp(name + namelen-extlen, ext))
 141         {
 142                 base[namelen - extlen] = '\0';
 143         }
 144         return base;
 145 }
 146
 147 void *xmalloc(size_t size)
 148 {
 149     void *res;
 150
 151     assert(size > 0);
 152     res = malloc(size);
 153     if(res == NULL)
 154     {
 155         error("Virtual memory exhausted.\n");
 156     }
 157     memset(res, 0x55, size);
 158     return res;
 159 }
 160
 161
 162 void *xrealloc(void *p, size_t size)
 163 {
 164     void *res;
 165
 166     assert(size > 0);
 167     res = realloc(p, size);
 168     if(res == NULL)
 169     {
 170         error("Virtual memory exhausted.\n");
 171     }
 172     return res;
 173 }
 174
 175 char *xstrdup(const char *str)
 176 {
 177         char *s;
 178
 179         assert(str != NULL);
 180         s = xmalloc(strlen(str)+1);
 181         return strcpy(s, str);
 182 }
 183
 184 char *strmake( const char* fmt, ... )
 185 {
 186     int n;
 187     size_t size = 100;
 188     va_list ap;
 189
 190     for (;;)
 191     {
 192         char *p = xmalloc( size );
 193         va_start( ap, fmt );
 194         n = vsnprintf( p, size, fmt, ap );
 195         va_end( ap );
 196         if (n == -1) size *= 2;
 197         else if ((size_t)n >= size) size = n + 1;
 198         else return p;
 199         free( p );
 200     }
 201 }
 202
 203 int unistrlen(const WCHAR *s)
 204 {
 205         int n;
 206         for(n = 0; *s; n++, s++)
 207                 ;
 208         return n;
 209 }
 210
 211 WCHAR *unistrcpy(WCHAR *dst, const WCHAR *src)
 212 {
 213         WCHAR *t = dst;
 214         while(*src)
 215                 *t++ = *src++;
 216         *t = 0;
 217         return dst;
 218 }
 219
 220 WCHAR *xunistrdup(const WCHAR * str)
 221 {
 222         WCHAR *s;
 223
 224         assert(str != NULL);
 225         s = xmalloc((unistrlen(str)+1) * sizeof(WCHAR));
 226         return unistrcpy(s, str);
 227 }
 228
 229 int unistricmp(const WCHAR *s1, const WCHAR *s2)
 230 {
 231         int i;
 232         int once = 0;
 233         static const char warn[] = "Don't know the uppercase equivalent of non ascii characters;"
 234                              "comparison might yield wrong results";
 235         while(*s1 && *s2)
 236         {
 237                 if((*s1 & 0xffff) > 0x7f || (*s2 & 0xffff) > 0x7f)
 238                 {
 239                         if(!once)
 240                         {
 241                                 once++;
 242                                 mcy_warning(warn);
 243                         }
 244                         i = *s1++ - *s2++;
 245                 }
 246                 else
 247                         i = toupper(*s1++) - toupper(*s2++);
 248                 if(i)
 249                         return i;
 250         }
 251
 252         if((*s1 & 0xffff) > 0x7f || (*s2 & 0xffff) > 0x7f)
 253         {
 254                 if(!once)
 255                         mcy_warning(warn);
 256                 return *s1 - *s2;
 257         }
 258         else
 259                 return  toupper(*s1) - toupper(*s2);
 260 }
 261
 262 int unistrcmp(const WCHAR *s1, const WCHAR *s2)
 263 {
 264         int i;
 265         while(*s1 && *s2)
 266         {
 267                 i = *s1++ - *s2++;
 268                 if(i)
 269                         return i;
 270         }
 271
 272         return *s1 - *s2;
 273 }
 274
 275 WCHAR *utf8_to_unicode( const char *src, int srclen, int *dstlen )
 276 {
 277     static const char utf8_length[128] =
 278     {
 279         0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0, /* 0x80-0x8f */
 280         0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0, /* 0x90-0x9f */
 281         0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0, /* 0xa0-0xaf */
 282         0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0, /* 0xb0-0xbf */
 283         0,0,1,1,1,1,1,1,1,1,1,1,1,1,1,1, /* 0xc0-0xcf */
 284         1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1, /* 0xd0-0xdf */
 285         2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2, /* 0xe0-0xef */
 286         3,3,3,3,3,0,0,0,0,0,0,0,0,0,0,0  /* 0xf0-0xff */
 287     };
 288     static const unsigned char utf8_mask[4] = { 0x7f, 0x1f, 0x0f, 0x07 };
 289
 290     const char *srcend = src + srclen;
 291     int len, res;
 292     WCHAR *ret, *dst;
 293
 294     dst = ret = xmalloc( (srclen + 1) * sizeof(WCHAR) );
 295     while (src < srcend)
 296     {
 297         unsigned char ch = *src++;
 298         if (ch < 0x80)  /* special fast case for 7-bit ASCII */
 299         {
 300             *dst++ = ch;
 301             continue;
 302         }
 303         len = utf8_length[ch - 0x80];
 304         if (len && src + len <= srcend)
 305         {
 306             res = ch & utf8_mask[len];
 307             switch (len)
 308             {
 309             case 3:
 310                 if ((ch = *src ^ 0x80) >= 0x40) break;
 311                 res = (res << 6) | ch;
 312                 src++;
 313                 if (res < 0x10) break;
 314             case 2:
 315                 if ((ch = *src ^ 0x80) >= 0x40) break;
 316                 res = (res << 6) | ch;
 317                 if (res >= 0x110000 >> 6) break;
 318                 src++;
 319                 if (res < 0x20) break;
 320                 if (res >= 0xd800 >> 6 && res <= 0xdfff >> 6) break;
 321             case 1:
 322                 if ((ch = *src ^ 0x80) >= 0x40) break;
 323                 res = (res << 6) | ch;
 324                 src++;
 325                 if (res < 0x80) break;
 326                 if (res <= 0xffff) *dst++ = res;
 327                 else
 328                 {
 329                     res -= 0x10000;
 330                     *dst++ = 0xd800 | (res >> 10);
 331                     *dst++ = 0xdc00 | (res & 0x3ff);
 332                 }
 333                 continue;
 334             }
 335         }
 336         *dst++ = 0xfffd;
 337     }
 338     *dst = 0;
 339     *dstlen = dst - ret;
 340     return ret;
 341 }
 342
 343 char *unicode_to_utf8( const WCHAR *src, int srclen, int *dstlen )
 344 {
 345     char *ret, *dst;
 346
 347     dst = ret = xmalloc( srclen * 3 + 1 );
 348     for ( ; srclen; srclen--, src++)
 349     {
 350         unsigned int ch = *src;
 351
 352         if (ch < 0x80)  /* 0x00-0x7f: 1 byte */
 353         {
 354             *dst++ = ch;
 355             continue;
 356         }
 357         if (ch < 0x800)  /* 0x80-0x7ff: 2 bytes */
 358         {
 359             dst[1] = 0x80 | (ch & 0x3f);
 360             ch >>= 6;
 361             dst[0] = 0xc0 | ch;
 362             dst += 2;
 363             continue;
 364         }
 365         if (ch >= 0xd800 && ch <= 0xdbff && srclen > 1 && src[1] >= 0xdc00 && src[1] <= 0xdfff)
 366         {
 367             /* 0x10000-0x10ffff: 4 bytes */
 368             ch = 0x10000 + ((ch & 0x3ff) << 10) + (src[1] & 0x3ff);
 369             dst[3] = 0x80 | (ch & 0x3f);
 370             ch >>= 6;
 371             dst[2] = 0x80 | (ch & 0x3f);
 372             ch >>= 6;
 373             dst[1] = 0x80 | (ch & 0x3f);
 374             ch >>= 6;
 375             dst[0] = 0xf0 | ch;
 376             dst += 4;
 377             src++;
 378             srclen--;
 379             continue;
 380         }
 381         if (ch >= 0xd800 && ch <= 0xdfff) ch = 0xfffd;  /* invalid surrogate pair */
 382
 383         /* 0x800-0xffff: 3 bytes */
 384         dst[2] = 0x80 | (ch & 0x3f);
 385         ch >>= 6;
 386         dst[1] = 0x80 | (ch & 0x3f);
 387         ch >>= 6;
 388         dst[0] = 0xe0 | ch;
 389         dst += 3;
 390     }
 391     *dst = 0;
 392     *dstlen = dst - ret;
 393     return ret;
 394 }
 395
 396 /*******************************************************************
 397  *         buffer management
 398  *
 399  * Function for writing to a memory buffer.
 400  */
 401
 402 int byte_swapped = 0;
 403 unsigned char *output_buffer;
 404 size_t output_buffer_pos;
 405 size_t output_buffer_size;
 406
 407 static void check_output_buffer_space( size_t size )
 408 {
 409     if (output_buffer_pos + size >= output_buffer_size)
 410     {
 411         output_buffer_size = max( output_buffer_size * 2, output_buffer_pos + size );
 412         output_buffer = xrealloc( output_buffer, output_buffer_size );
 413     }
 414 }
 415
 416 void init_output_buffer(void)
 417 {
 418     output_buffer_size = 1024;
 419     output_buffer_pos = 0;
 420     output_buffer = xmalloc( output_buffer_size );
 421 }
 422
 423 void flush_output_buffer( const char *name )
 424 {
 425     int fd = open( name, O_WRONLY | O_CREAT | O_TRUNC | O_BINARY, 0666 );
 426     if (fd == -1) error( "Error creating %s\n", name );
 427     if (write( fd, output_buffer, output_buffer_pos ) != output_buffer_pos)
 428         error( "Error writing to %s\n", name );
 429     close( fd );
 430     free( output_buffer );
 431 }
 432
 433 void put_data( const void *data, size_t size )
 434 {
 435     check_output_buffer_space( size );
 436     memcpy( output_buffer + output_buffer_pos, data, size );
 437     output_buffer_pos += size;
 438 }
 439
 440 void put_byte( unsigned char val )
 441 {
 442     check_output_buffer_space( 1 );
 443     output_buffer[output_buffer_pos++] = val;
 444 }
 445
 446 void put_word( unsigned short val )
 447 {
 448     if (byte_swapped) val = (val << 8) | (val >> 8);
 449     put_data( &val, sizeof(val) );
 450 }
 451
 452 void put_dword( unsigned int val )
 453 {
 454     if (byte_swapped)
 455         val = ((val << 24) | ((val << 8) & 0x00ff0000) | ((val >> 8) & 0x0000ff00) | (val >> 24));
 456     put_data( &val, sizeof(val) );
 457 }
 458
 459 void align_output( unsigned int align )
 460 {
 461     size_t size = align - (output_buffer_pos % align);
 462
 463     if (size == align) return;
 464     check_output_buffer_space( size );
 465     memset( output_buffer + output_buffer_pos, 0, size );
 466     output_buffer_pos += size;
 467 }