reactos/lib/ntdll/dbg/winedbg.c

   1 /*
   2  * Debugging functions for WINE
   3  */
   4 #define NTOS_USER_MODE
   5 #include <ntos.h>
   6 #include <wine/debugtools.h>
   7
   8 DECLARE_DEBUG_CHANNEL(tid);
   9
  10 /* ---------------------------------------------------------------------- */
  11
  12 struct debug_info
  13 {
  14     char *str_pos;       /* current position in strings buffer */
  15     char *out_pos;       /* current position in output buffer */
  16     char  strings[1024]; /* buffer for temporary strings */
  17     char  output[1024];  /* current output line */
  18 };
  19
  20 static struct debug_info tmp;
  21
  22 /* get the debug info pointer for the current thread */
  23 static inline struct debug_info *get_info(void)
  24 {
  25     struct debug_info *info = NtCurrentTeb()->WineDebugInfo;
  26     if (!info)
  27     {
  28         if (!tmp.str_pos)
  29         {
  30             tmp.str_pos = tmp.strings;
  31             tmp.out_pos = tmp.output;
  32         }
  33         if (!RtlGetProcessHeap()) return &tmp;
  34         /* setup the temp structure in case HeapAlloc wants to print something */
  35         NtCurrentTeb()->WineDebugInfo = &tmp;
  36         info = RtlAllocateHeap( RtlGetProcessHeap(), 0, sizeof(*info) );
  37         info->str_pos = info->strings;
  38         info->out_pos = info->output;
  39         NtCurrentTeb()->WineDebugInfo = info;
  40     }
  41     return info;
  42 }
  43
  44 /* allocate some tmp space for a string */
  45 static void *gimme1(int n)
  46 {
  47     struct debug_info *info = get_info();
  48     char *res = info->str_pos;
  49
  50     if (res + n >= &info->strings[sizeof(info->strings)]) res = info->strings;
  51     info->str_pos = res + n;
  52     return res;
  53 }
  54
  55 /* release extra space that we requested in gimme1() */
  56 static inline void release( void *ptr )
  57 {
  58     struct debug_info *info = NtCurrentTeb()->WineDebugInfo;
  59     info->str_pos = ptr;
  60 }
  61
  62 /***********************************************************************
  63  *              wine_dbgstr_an (NTDLL.@)
  64  */
  65 const char *wine_dbgstr_an( const char *src, int n )
  66 {
  67     char *dst, *res;
  68
  69     if (!((WORD)(DWORD)(src) >> 16))
  70     {
  71         if (!src) return "(null)";
  72         res = gimme1(6);
  73         sprintf(res, "#%04x", (WORD)(DWORD)(src) );
  74         return res;
  75     }
  76     if (n < 0) n = 0;
  77     else if (n > 200) n = 200;
  78     dst = res = gimme1 (n * 4 + 6);
  79     *dst++ = '"';
  80     while (n-- > 0 && *src)
  81     {
  82         unsigned char c = *src++;
  83         switch (c)
  84         {
  85         case '\n': *dst++ = '\\'; *dst++ = 'n'; break;
  86         case '\r': *dst++ = '\\'; *dst++ = 'r'; break;
  87         case '\t': *dst++ = '\\'; *dst++ = 't'; break;
  88         case '"': *dst++ = '\\'; *dst++ = '"'; break;
  89         case '\\': *dst++ = '\\'; *dst++ = '\\'; break;
  90         default:
  91             if (c >= ' ' && c <= 126)
  92                 *dst++ = c;
  93             else
  94             {
  95                 *dst++ = '\\';
  96                 *dst++ = '0' + ((c >> 6) & 7);
  97                 *dst++ = '0' + ((c >> 3) & 7);
  98                 *dst++ = '0' + ((c >> 0) & 7);
  99             }
 100         }
 101     }
 102     *dst++ = '"';
 103     if (*src)
 104     {
 105         *dst++ = '.';
 106         *dst++ = '.';
 107         *dst++ = '.';
 108     }
 109     *dst++ = '\0';
 110     release( dst );
 111     return res;
 112 }
 113
 114 /***********************************************************************
 115  *              wine_dbgstr_wn (NTDLL.@)
 116  */
 117 const char *wine_dbgstr_wn( const WCHAR *src, int n )
 118 {
 119     char *dst, *res;
 120
 121     if (!((WORD)(DWORD)(src) >> 16))
 122     {
 123         if (!src) return "(null)";
 124         res = gimme1(6);
 125         sprintf(res, "#%04x", (WORD)(DWORD)(src) );
 126         return res;
 127     }
 128     if (n < 0) n = 0;
 129     else if (n > 200) n = 200;
 130     dst = res = gimme1 (n * 5 + 7);
 131     *dst++ = 'L';
 132     *dst++ = '"';
 133     while (n-- > 0 && *src)
 134     {
 135         WCHAR c = *src++;
 136         switch (c)
 137         {
 138         case '\n': *dst++ = '\\'; *dst++ = 'n'; break;
 139         case '\r': *dst++ = '\\'; *dst++ = 'r'; break;
 140         case '\t': *dst++ = '\\'; *dst++ = 't'; break;
 141         case '"': *dst++ = '\\'; *dst++ = '"'; break;
 142         case '\\': *dst++ = '\\'; *dst++ = '\\'; break;
 143         default:
 144             if (c >= ' ' && c <= 126)
 145                 *dst++ = c;
 146             else
 147             {
 148                 *dst++ = '\\';
 149                 sprintf(dst,"%04x",c);
 150                 dst+=4;
 151             }
 152         }
 153     }
 154     *dst++ = '"';
 155     if (*src)
 156     {
 157         *dst++ = '.';
 158         *dst++ = '.';
 159         *dst++ = '.';
 160     }
 161     *dst++ = '\0';
 162     release( dst );
 163     return res;
 164 }
 165
 166 /***********************************************************************
 167  *              wine_dbgstr_guid (NTDLL.@)
 168  */
 169 const char *wine_dbgstr_guid( const GUID *id )
 170 {
 171     char *str;
 172
 173     if (!id) return "(null)";
 174     if (!((WORD)(DWORD)(id) >> 16))
 175     {
 176         str = gimme1(12);
 177         sprintf( str, "<guid-0x%04x>", (WORD)(DWORD)(id) );
 178     }
 179     else
 180     {
 181         str = gimme1(40);
 182         sprintf( str, "{%08lx-%04x-%04x-%02x%02x-%02x%02x%02x%02x%02x%02x}",
 183                  id->Data1, id->Data2, id->Data3,
 184                  id->Data4[0], id->Data4[1], id->Data4[2], id->Data4[3],
 185                  id->Data4[4], id->Data4[5], id->Data4[6], id->Data4[7] );
 186     }
 187     return str;
 188 }
 189
 190 /***********************************************************************
 191  *              wine_dbg_vprintf (NTDLL.@)
 192  */
 193 int wine_dbg_vprintf( const char *format, va_list args )
 194 {
 195     struct debug_info *info = get_info();
 196     char *p;
 197
 198     int ret = _vsnprintf( info->out_pos, sizeof(info->output) - (info->out_pos - info->output),
 199                          format, args );
 200
 201     p = strrchr( info->out_pos, '\n' );
 202     if (!p) info->out_pos += ret;
 203     else
 204     {
 205         char *pos = info->output;
 206         char saved_ch;
 207         p++;
 208         saved_ch = *p;
 209         *p = 0;
 210         DbgPrint(pos);
 211         *p = saved_ch;
 212         /* move beginning of next line to start of buffer */
 213         while ((*pos = *p++)) pos++;
 214         info->out_pos = pos;
 215     }
 216     return ret;
 217 }
 218
 219 /***********************************************************************
 220  *              wine_dbg_printf (NTDLL.@)
 221  */
 222 int wine_dbg_printf(const char *format, ...)
 223 {
 224     int ret;
 225     va_list valist;
 226
 227     va_start(valist, format);
 228     ret = wine_dbg_vprintf( format, valist );
 229     va_end(valist);
 230     return ret;
 231 }
 232
 233 /***********************************************************************
 234  *              wine_dbg_log (NTDLL.@)
 235  */
 236 int wine_dbg_log(enum __DEBUG_CLASS cls, const char *channel,
 237                  const char *function, const char *format, ... )
 238 {
 239     static const char *classes[__DBCL_COUNT] = { "fixme", "err", "warn", "trace" };
 240     va_list valist;
 241     int ret = 0;
 242
 243     va_start(valist, format);
 244     if (TRACE_ON(tid))
 245         ret = wine_dbg_printf( "%08lx:", NtCurrentTeb()->Cid.UniqueThread);
 246     if (cls < __DBCL_COUNT)
 247         ret += wine_dbg_printf( "%s:%s:%s ", classes[cls], channel + 1, function );
 248     if (format)
 249         ret += wine_dbg_vprintf( format, valist );
 250     va_end(valist);
 251     return ret;
 252 }