programs/winetest/util.c

   1 /*
   2  * Utility functions.
   3  *
   4  * Copyright 2003 Dimitrie O. Paun
   5  * Copyright 2003 Ferenc Wagner
   6  *
   7  * This library is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU Lesser General Public
   9  * License as published by the Free Software Foundation; either
  10  * version 2.1 of the License, or (at your option) any later version.
  11  *
  12  * This library is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * Lesser General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Lesser General Public
  18  * License along with this library; if not, write to the Free Software
  19  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  20  */
  21
  22 #include <stdarg.h>
  23 #include <windef.h>
  24 #include <winbase.h>
  25
  26 #include "winetest.h"
  27
  28 HANDLE logfile = 0;
  29
  30 void *xalloc (size_t len)
  31 {
  32     void *p = malloc( len );
  33
  34     if (!p) report (R_FATAL, "Out of memory.");
  35     return p;
  36 }
  37
  38 void *xrealloc (void *op, size_t len)
  39 {
  40     void *p = realloc(op, len);
  41
  42     if (len && !p) report (R_FATAL, "Out of memory.");
  43     return p;
  44 }
  45
  46 char *xstrdup( const char *str )
  47 {
  48     char* res = strdup( str );
  49     if (!res) report (R_FATAL, "Out of memory.");
  50     return res;
  51 }
  52
  53 static char *vstrfmtmake (size_t *lenp, const char *fmt, va_list ap)
  54 {
  55     size_t size = 1000;
  56     char *p;
  57     int n;
  58
  59     p = xalloc(size);
  60     while (1) {
  61         n = vsnprintf (p, size, fmt, ap);
  62         if (n < 0) size *= 2;   /* Windows */
  63         else if ((unsigned)n >= size) size = n+1; /* glibc */
  64         else break;
  65         p = xrealloc(p, size);
  66     }
  67     if (lenp) *lenp = n;
  68     return p;
  69 }
  70
  71 char *vstrmake (size_t *lenp, va_list ap)
  72 {
  73     const char *fmt;
  74
  75     fmt = va_arg (ap, const char*);
  76     return vstrfmtmake (lenp, fmt, ap);
  77 }
  78
  79 char * WINAPIV strmake (size_t *lenp, ...)
  80 {
  81     va_list ap;
  82     char *p;
  83
  84     va_start (ap, lenp);
  85     p = vstrmake (lenp, ap);
  86     va_end (ap);
  87     return p;
  88 }
  89
  90 void WINAPIV xprintf (const char *fmt, ...)
  91 {
  92     va_list ap;
  93     size_t size;
  94     DWORD written;
  95     char *buffer, *head;
  96
  97     va_start (ap, fmt);
  98     buffer = vstrfmtmake (&size, fmt, ap);
  99     head = buffer;
 100     va_end (ap);
 101     while (size) {
 102         if (!WriteFile( logfile, head, size, &written, NULL ))
 103             report (R_FATAL, "Can't write logs: %u", GetLastError());
 104         head += written;
 105         size -= written;
 106     }
 107     free(buffer);
 108 }
 109
 110 int
 111 goodtagchar (char c)
 112 {
 113     return (('a'<=c && c<='z') ||
 114             ('A'<=c && c<='Z') ||
 115             ('0'<=c && c<='9') ||
 116             c=='-' || c=='.');
 117 }
 118
 119 const char *
 120 findbadtagchar (const char *tag)
 121 {
 122     while (*tag)
 123         if (goodtagchar (*tag)) tag++;
 124         else return tag;
 125     return NULL;
 126 }