dlls/msvcrt/exit.c

   1 /*
   2  * msvcrt.dll exit functions
   3  *
   4  * Copyright 2000 Jon Griffiths
   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 #include <stdio.h>
  21 #include "msvcrt.h"
  22 #include "mtdll.h"
  23 #include "winuser.h"
  24 #include "wine/debug.h"
  25
  26 WINE_DEFAULT_DEBUG_CHANNEL(msvcrt);
  27
  28 /* MT */
  29 #define LOCK_EXIT   _mlock(_EXIT_LOCK1)
  30 #define UNLOCK_EXIT _munlock(_EXIT_LOCK1)
  31
  32 static MSVCRT__onexit_t *MSVCRT_atexit_table = NULL;
  33 static int MSVCRT_atexit_table_size = 0;
  34 static int MSVCRT_atexit_registered = 0; /* Points to free slot */
  35
  36 static const char szMsgBoxTitle[] = "Wine C++ Runtime Library";
  37
  38 extern int MSVCRT_app_type;
  39 extern char *MSVCRT__pgmptr;
  40
  41 void (*CDECL _aexit_rtn)(int) = MSVCRT__exit;
  42
  43 /* INTERNAL: call atexit functions */
  44 static void __MSVCRT__call_atexit(void)
  45 {
  46   /* Note: should only be called with the exit lock held */
  47   TRACE("%d atext functions to call\n", MSVCRT_atexit_registered);
  48   /* Last registered gets executed first */
  49   while (MSVCRT_atexit_registered > 0)
  50   {
  51     MSVCRT_atexit_registered--;
  52     TRACE("next is %p\n",MSVCRT_atexit_table[MSVCRT_atexit_registered]);
  53     if (MSVCRT_atexit_table[MSVCRT_atexit_registered])
  54       (*MSVCRT_atexit_table[MSVCRT_atexit_registered])();
  55     TRACE("returned\n");
  56   }
  57 }
  58
  59 /*********************************************************************
  60  *              __dllonexit (MSVCRT.@)
  61  */
  62 MSVCRT__onexit_t CDECL __dllonexit(MSVCRT__onexit_t func, MSVCRT__onexit_t **start, MSVCRT__onexit_t **end)
  63 {
  64   MSVCRT__onexit_t *tmp;
  65   int len;
  66
  67   TRACE("(%p,%p,%p)\n", func, start, end);
  68
  69   if (!start || !*start || !end || !*end)
  70   {
  71    FIXME("bad table\n");
  72    return NULL;
  73   }
  74
  75   len = (*end - *start);
  76
  77   TRACE("table start %p-%p, %d entries\n", *start, *end, len);
  78
  79   if (++len <= 0)
  80     return NULL;
  81
  82   tmp = MSVCRT_realloc(*start, len * sizeof(tmp));
  83   if (!tmp)
  84     return NULL;
  85   *start = tmp;
  86   *end = tmp + len;
  87   tmp[len - 1] = func;
  88   TRACE("new table start %p-%p, %d entries\n", *start, *end, len);
  89   return func;
  90 }
  91
  92 /*********************************************************************
  93  *              _exit (MSVCRT.@)
  94  */
  95 void CDECL MSVCRT__exit(int exitcode)
  96 {
  97   TRACE("(%d)\n", exitcode);
  98   ExitProcess(exitcode);
  99 }
 100
 101 /* Print out an error message with an option to debug */
 102 static void DoMessageBox(LPCSTR lead, LPCSTR message)
 103 {
 104   MSGBOXPARAMSA msgbox;
 105   char text[2048];
 106   INT ret;
 107
 108   snprintf(text,sizeof(text),"%s\n\nProgram: %s\n%s\n\n"
 109                "Press OK to exit the program, or Cancel to start the Wine debugger.\n ",
 110                lead, MSVCRT__pgmptr, message);
 111
 112   msgbox.cbSize = sizeof(msgbox);
 113   msgbox.hwndOwner = GetActiveWindow();
 114   msgbox.hInstance = 0;
 115   msgbox.lpszText = text;
 116   msgbox.lpszCaption = szMsgBoxTitle;
 117   msgbox.dwStyle = MB_OKCANCEL|MB_ICONERROR;
 118   msgbox.lpszIcon = NULL;
 119   msgbox.dwContextHelpId = 0;
 120   msgbox.lpfnMsgBoxCallback = NULL;
 121   msgbox.dwLanguageId = LANG_NEUTRAL;
 122
 123   ret = MessageBoxIndirectA(&msgbox);
 124   if (ret == IDCANCEL)
 125     DebugBreak();
 126 }
 127
 128 /*********************************************************************
 129  *              _amsg_exit (MSVCRT.@)
 130  */
 131 void CDECL _amsg_exit(int errnum)
 132 {
 133   TRACE("(%d)\n", errnum);
 134   /* FIXME: text for the error number. */
 135   if (MSVCRT_app_type == 2)
 136   {
 137     char text[32];
 138     sprintf(text, "Error: R60%d",errnum);
 139     DoMessageBox("Runtime error!", text);
 140   }
 141   else
 142     _cprintf("\nruntime error R60%d\n",errnum);
 143   _aexit_rtn(255);
 144 }
 145
 146 /*********************************************************************
 147  *              abort (MSVCRT.@)
 148  */
 149 void CDECL MSVCRT_abort(void)
 150 {
 151   TRACE("()\n");
 152   if (MSVCRT_app_type == 2)
 153   {
 154     DoMessageBox("Runtime error!", "abnormal program termination");
 155   }
 156   else
 157     _cputs("\nabnormal program termination\n");
 158   MSVCRT_raise(MSVCRT_SIGABRT);
 159   /* in case raise() returns */
 160   MSVCRT__exit(3);
 161 }
 162
 163 /*********************************************************************
 164  *              _assert (MSVCRT.@)
 165  */
 166 void CDECL MSVCRT__assert(const char* str, const char* file, unsigned int line)
 167 {
 168   TRACE("(%s,%s,%d)\n",str,file,line);
 169   if (MSVCRT_app_type == 2)
 170   {
 171     char text[2048];
 172     snprintf(text, sizeof(text), "File: %s\nLine: %d\n\nExpression: \"%s\"", file, line, str);
 173     DoMessageBox("Assertion failed!", text);
 174   }
 175   else
 176     _cprintf("Assertion failed: %s, file %s, line %d\n\n",str, file, line);
 177   MSVCRT_raise(MSVCRT_SIGABRT);
 178   /* in case raise() returns */
 179   MSVCRT__exit(3);
 180 }
 181
 182 /*********************************************************************
 183  *              _c_exit (MSVCRT.@)
 184  */
 185 void CDECL MSVCRT__c_exit(void)
 186 {
 187   TRACE("(void)\n");
 188   /* All cleanup is done on DLL detach; Return to caller */
 189 }
 190
 191 /*********************************************************************
 192  *              _cexit (MSVCRT.@)
 193  */
 194 void CDECL MSVCRT__cexit(void)
 195 {
 196   TRACE("(void)\n");
 197   LOCK_EXIT;
 198   __MSVCRT__call_atexit();
 199   UNLOCK_EXIT;
 200 }
 201
 202 /*********************************************************************
 203  *              _onexit (MSVCRT.@)
 204  */
 205 MSVCRT__onexit_t CDECL MSVCRT__onexit(MSVCRT__onexit_t func)
 206 {
 207   TRACE("(%p)\n",func);
 208
 209   if (!func)
 210     return NULL;
 211
 212   LOCK_EXIT;
 213   if (MSVCRT_atexit_registered > MSVCRT_atexit_table_size - 1)
 214   {
 215     MSVCRT__onexit_t *newtable;
 216     TRACE("expanding table\n");
 217     newtable = MSVCRT_calloc(sizeof(void *),MSVCRT_atexit_table_size + 32);
 218     if (!newtable)
 219     {
 220       TRACE("failed!\n");
 221       UNLOCK_EXIT;
 222       return NULL;
 223     }
 224     memcpy (newtable, MSVCRT_atexit_table, MSVCRT_atexit_table_size);
 225     MSVCRT_atexit_table_size += 32;
 226     MSVCRT_free (MSVCRT_atexit_table);
 227     MSVCRT_atexit_table = newtable;
 228   }
 229   MSVCRT_atexit_table[MSVCRT_atexit_registered] = func;
 230   MSVCRT_atexit_registered++;
 231   UNLOCK_EXIT;
 232   return func;
 233 }
 234
 235 /*********************************************************************
 236  *              exit (MSVCRT.@)
 237  */
 238 void CDECL MSVCRT_exit(int exitcode)
 239 {
 240   TRACE("(%d)\n",exitcode);
 241   MSVCRT__cexit();
 242   ExitProcess(exitcode);
 243 }
 244
 245 /*********************************************************************
 246  *              atexit (MSVCRT.@)
 247  */
 248 int CDECL MSVCRT_atexit(void (*func)(void))
 249 {
 250   TRACE("(%p)\n", func);
 251   return MSVCRT__onexit((MSVCRT__onexit_t)func) == (MSVCRT__onexit_t)func ? 0 : -1;
 252 }
 253
 254
 255 /*********************************************************************
 256  *              _purecall (MSVCRT.@)
 257  */
 258 void CDECL _purecall(void)
 259 {
 260   TRACE("(void)\n");
 261   _amsg_exit( 25 );
 262 }