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 /*******************************************************************
 276  *         buffer management
 277  *
 278  * Function for writing to a memory buffer.
 279  */
 280
 281 int byte_swapped = 0;
 282 unsigned char *output_buffer;
 283 size_t output_buffer_pos;
 284 size_t output_buffer_size;
 285
 286 static void check_output_buffer_space( size_t size )
 287 {
 288     if (output_buffer_pos + size >= output_buffer_size)
 289     {
 290         output_buffer_size = max( output_buffer_size * 2, output_buffer_pos + size );
 291         output_buffer = xrealloc( output_buffer, output_buffer_size );
 292     }
 293 }
 294
 295 void init_output_buffer(void)
 296 {
 297     output_buffer_size = 1024;
 298     output_buffer_pos = 0;
 299     output_buffer = xmalloc( output_buffer_size );
 300 }
 301
 302 void flush_output_buffer( const char *name )
 303 {
 304     int fd = open( name, O_WRONLY | O_CREAT | O_TRUNC | O_BINARY, 0666 );
 305     if (fd == -1) error( "Error creating %s\n", name );
 306     if (write( fd, output_buffer, output_buffer_pos ) != output_buffer_pos)
 307         error( "Error writing to %s\n", name );
 308     close( fd );
 309     free( output_buffer );
 310 }
 311
 312 void put_data( const void *data, size_t size )
 313 {
 314     check_output_buffer_space( size );
 315     memcpy( output_buffer + output_buffer_pos, data, size );
 316     output_buffer_pos += size;
 317 }
 318
 319 void put_byte( unsigned char val )
 320 {
 321     check_output_buffer_space( 1 );
 322     output_buffer[output_buffer_pos++] = val;
 323 }
 324
 325 void put_word( unsigned short val )
 326 {
 327     if (byte_swapped) val = (val << 8) | (val >> 8);
 328     put_data( &val, sizeof(val) );
 329 }
 330
 331 void put_dword( unsigned int val )
 332 {
 333     if (byte_swapped)
 334         val = ((val << 24) | ((val << 8) & 0x00ff0000) | ((val >> 8) & 0x0000ff00) | (val >> 24));
 335     put_data( &val, sizeof(val) );
 336 }
 337
 338 void align_output( unsigned int align )
 339 {
 340     size_t size = align - (output_buffer_pos % align);
 341
 342     if (size == align) return;
 343     check_output_buffer_space( size );
 344     memset( output_buffer + output_buffer_pos, 0, size );
 345     output_buffer_pos += size;
 346 }