NS/releases/3.05f/source/util/Zassert.cpp
2005-07-14 17:41:41 +00:00

319 lines
No EOL
9.3 KiB
C++

// ZASSERT AND STACKTRACE are an improved assertion system which augments
// the WIN32 assert.h standard _assert.
// (c) 1999 Zachary B. Simpson
// Code may be used and distributed freely as long as all
// changes are noted appropriately.
#ifdef WIN32
#include "windows.h"
#else
#include "netdb.h"
#include "unistd.h"
#include "sys/types.h"
#include "sys/socket.h"
#include "netinet/in.h"
#include "arpa/inet.h"
#endif
#include "string.h"
#include "stdio.h"
#include "stdlib.h"
#include "stacktrace.h"
#include "zassert.h"
#include "malloc.h"
#ifdef WIN32
#define CLOSESOCKET closesocket
#else
#define CLOSESOCKET ::close
#endif
#ifdef WIN32
// Assert Box Global Variables.
HWND hAssertBox = NULL;
HWND hTextArea = NULL;
HWND hEditArea = NULL;
HWND hOKButton = NULL;
HWND hEmailButton = NULL;
HWND hCopyButton = NULL;
void (*emailButtonFuncPtr)(char *);
// WndProc that runs the assert dialog box.
long CALLBACK assertBoxWndProc( HWND hWnd, UINT message, UINT wParam, long lParam ) {
switch( message ) {
case WM_SIZE: {
HDC hDC = GetDC( hOKButton );
SIZE s;
GetTextExtentPoint32( hDC, "W", 1, &s );
int margin = 10;
int w = LOWORD(lParam);
int h = HIWORD(lParam);
int bw = (w - 4*margin) / 3;
int bh = s.cy + 6;
int upperH = h - 3*margin - bh;
int textH = (upperH * 6) / 10 - margin;
int editH = (upperH * 4) / 10 ;
int editY = 2*margin + textH;
int x = margin;
int y = h - margin - bh;
MoveWindow( hOKButton, x, y, bw, bh, TRUE ); x += bw+margin;
MoveWindow( hEmailButton, x, y, bw, bh, TRUE ); x += bw+margin;
MoveWindow( hCopyButton, x, y, bw, bh, TRUE ); x += bw+margin;
MoveWindow( hTextArea, margin, margin, w-2*margin, textH, TRUE );
MoveWindow( hEditArea, margin, editY, w-2*margin, editH, TRUE );
break;
}
case WM_PAINT: {
// Paint the loading screen
PAINTSTRUCT ps;
HDC hdc = BeginPaint( hWnd, &ps );
EndPaint( hWnd, &ps );
break;
}
case WM_COMMAND: {
HWND who = (HWND)lParam;
if( who == hOKButton ) {
PostMessage( hWnd, WM_CLOSE, 0, 0 );
}
else if( who == hEmailButton ) {
int textSize =
GetWindowTextLength( hTextArea )
+ GetWindowTextLength( hEditArea )
;
char *msg = (char *)alloca( textSize );
GetWindowText( hTextArea, msg, textSize+1 );
GetWindowText( hEditArea, msg+strlen(msg), textSize-strlen(msg)+1 );
if( emailButtonFuncPtr ) {
(*emailButtonFuncPtr)(msg);
}
EnableWindow( hEmailButton, FALSE );
}
else if( who == hCopyButton ) {
int textSize =
GetWindowTextLength( hTextArea )
+ GetWindowTextLength( hEditArea )
;
if( !OpenClipboard(NULL) ) {
break;
}
EmptyClipboard();
HGLOBAL hglbCopy = GlobalAlloc( GMEM_DDESHARE, (textSize+1) * sizeof(TCHAR) );
if( hglbCopy == NULL ) {
CloseClipboard();
break;
}
char *globalStr = (char *)GlobalLock( hglbCopy );
GetWindowText( hTextArea, globalStr, textSize+1 );
GetWindowText( hEditArea, globalStr+strlen(globalStr), textSize-strlen(globalStr)+1 );
GlobalUnlock (hglbCopy);
SetClipboardData( CF_TEXT, hglbCopy );
CloseClipboard();
EnableWindow( hCopyButton, FALSE );
}
}
}
return DefWindowProc(hWnd, message, wParam, lParam );
}
void *createAssertBox(
char *msg,
void (*emailButtonFuncPtr)(char *),
char *titleText,
char *okButtonText,
char *emailButtonText,
char *copyButtonText,
char *editAreaText,
void *parentBox
) {
if( !titleText ) {
titleText = "Assertion Failed";
}
if( !okButtonText ) {
okButtonText = "OK";
}
if( !emailButtonText ) {
emailButtonText = "E-Mail";
}
if( !copyButtonText ) {
copyButtonText = "Copy";
}
if( !editAreaText ) {
editAreaText =
"Note: If you don't see your mouse cursor, try alt-tabbing away, then back again.\r\n\r\n Your name and e-mail address: \r\nSteps to reproduce bug (what were you doing?): \r\n\r\nMake sure to press the e-mail button to submit this bug. Thanks for making Natural Selection better!";
}
::emailButtonFuncPtr = emailButtonFuncPtr;
WNDCLASSEX wc;
memset(&wc, 0, sizeof(WNDCLASSEX) );
wc.cbSize = sizeof(wc);
wc.style = CS_HREDRAW | CS_VREDRAW;
wc.hCursor = LoadCursor( NULL, IDC_ARROW );
wc.lpfnWndProc = (WNDPROC)assertBoxWndProc;
wc.lpszClassName = "AssertBox";
wc.hbrBackground = (HBRUSH)(COLOR_BTNFACE+1);
RegisterClassEx(&wc);
hAssertBox = CreateWindowEx(
WS_EX_DLGMODALFRAME, "AssertBox", titleText, WS_SIZEBOX,
CW_USEDEFAULT, CW_USEDEFAULT, 100, 100, NULL, NULL, NULL, NULL
);
hTextArea = CreateWindowEx(
WS_EX_STATICEDGE, "STATIC", msg, WS_CHILD,
0, 0, 0, 0, hAssertBox, NULL, NULL, NULL
);
hEditArea = CreateWindowEx(
WS_EX_STATICEDGE, "EDIT", editAreaText, WS_CHILD|ES_MULTILINE|ES_WANTRETURN|ES_AUTOVSCROLL,
0, 0, 0, 0, hAssertBox, NULL, NULL, NULL
);
hOKButton = CreateWindowEx(
0, "BUTTON", okButtonText, BS_CENTER | BS_VCENTER | BS_DEFPUSHBUTTON | WS_CHILD,
0, 0, 0, 0, hAssertBox, NULL, NULL, NULL
);
hEmailButton = CreateWindowEx(
0, "BUTTON", emailButtonText, BS_CENTER | BS_VCENTER | BS_PUSHBUTTON | WS_CHILD,
0, 0, 0, 0, hAssertBox, NULL, NULL, NULL
);
hCopyButton = CreateWindowEx(
0, "BUTTON", copyButtonText, BS_CENTER | BS_VCENTER | BS_PUSHBUTTON | WS_CHILD,
0, 0, 0, 0, hAssertBox, NULL, NULL, NULL
);
if( !emailButtonFuncPtr ) {
EnableWindow( hEmailButton, FALSE );
}
int cx = GetSystemMetrics(SM_CXSCREEN);
int cy = GetSystemMetrics(SM_CYSCREEN);
MoveWindow ( hAssertBox, (cx-640)/2, (cy-300)/2, 640, 300, TRUE );
ShowWindow( hTextArea, SW_SHOW );
ShowWindow( hEditArea, SW_SHOW );
ShowWindow( hOKButton, SW_SHOW );
ShowWindow( hEmailButton, SW_SHOW );
ShowWindow( hCopyButton, SW_SHOW );
ShowWindow( hAssertBox, SW_SHOW );
HFONT hGuiFont = (HFONT)GetStockObject(DEFAULT_GUI_FONT);
HFONT hFixFont = (HFONT)GetStockObject(ANSI_FIXED_FONT);
SendMessage( hTextArea, WM_SETFONT, (int)hFixFont, MAKELONG( TRUE, 0 ) );
SendMessage( hEditArea, WM_SETFONT, (int)hGuiFont, MAKELONG( TRUE, 0 ) );
SendMessage( hOKButton, WM_SETFONT, (int)hGuiFont, MAKELONG( TRUE, 0 ) );
SendMessage( hEmailButton, WM_SETFONT, (int)hGuiFont, MAKELONG( TRUE, 0 ) );
SendMessage( hCopyButton, WM_SETFONT, (int)hGuiFont, MAKELONG( TRUE, 0 ) );
//SetFocus( hEditArea );
SendMessage( hEditArea, EM_SETSEL, 0, MAKELONG(1000, -1) );
//if( emailButtonFuncPtr ) {
// (*emailButtonFuncPtr)(msg);
//}
return hAssertBox;
}
extern HWND hRunAssertBox;
void theExitRunAssertBox(void)
{
runAssertBox(hRunAssertBox);
}
//void runAssertBox( HWND hWnd ) {
void runAssertBox( void* inWnd ) {
HWND hWnd = (HWND)(inWnd);
MSG m;
while( GetMessage( &m, hWnd, 0, 0) ) {
if( m.message == WM_CLOSE ) {
return;
}
TranslateMessage( &m );
DispatchMessage( &m );
}
}
#endif
int emailMsgTo( char *msg, char *emailAddress ) {
struct hostent *hostEnt;
int ip = 0;
sockaddr_in sockAddr;
int sockFD;
int _send, _recv, _connect;
char *computerName = "";
char *emailAt;
char *s = (char *)alloca( strlen(msg) + 256 ); // send buffer
char r[2048]; // recv buffer
#ifdef WIN32
WSAData wsaData;
int ret = WSAStartup(0x0101, &wsaData);
if( ret != 0 ) goto error;
#endif
sockFD = socket( PF_INET, SOCK_STREAM, IPPROTO_TCP );
if( sockFD < 0 ) goto error;
// Parse the emailAddress to get the copmuter name after the '@'
emailAt = strchr( emailAddress, '@' );
if( emailAt ) {
computerName = emailAt+1;
}
ip = inet_addr( computerName );
if( ip == INADDR_NONE ) {
hostEnt = gethostbyname( computerName );
if( hostEnt ) {
ip = *(int *)hostEnt->h_addr;
}
}
memset( &sockAddr, 0, sizeof(sockAddr) );
sockAddr.sin_family = AF_INET;
sockAddr.sin_port = htons(25); // 25 = SMTP port
sockAddr.sin_addr.s_addr = ip;
_connect = connect( sockFD, (struct sockaddr *)&sockAddr, sizeof(sockAddr) );
if( _connect < 0 ) goto error;
memset( r, 0, 256 );
_recv = recv( sockFD, r, 256, 0 );
if( r[0] != '2' ) goto error;
memset( r, 0, 256 );
sprintf( s, "HELO overmind.org\r\n" );
_send = send( sockFD, s, strlen(s), 0 );
if( _send < 0 ) goto error;
_recv = recv( sockFD, r, 256, 0 );
if( r[0] != '2' ) goto error;
memset( r, 0, 256 );
sprintf( s, "MAIL FROM: ns-assert@overmind.org\r\n" );
_send = send( sockFD, s, strlen(s), 0 );
if( _send < 0 ) goto error;
_recv = recv( sockFD, r, 256, 0 );
if( r[0] != '2' ) goto error;
memset( r, 0, 256 );
sprintf( s, "RCPT TO: %s\r\n", emailAddress );
_send = send( sockFD, s, strlen(s), 0 );
if( _send < 0 ) goto error;
_recv = recv( sockFD, r, 256, 0 );
if( r[0] != '2' ) goto error;
memset( r, 0, 256 );
sprintf( s, "DATA\r\n" );
_send = send( sockFD, s, strlen(s), 0 );
if( _send < 0 ) goto error;
_recv = recv( sockFD, r, 256, 0 );
if( r[0] != '3' ) goto error;
memset( r, 0, 256 );
sprintf( s, "From ns-assert@overmind.org\r\nSubject: NS Assert\r\n%s\r\n.\r\n", msg );
_send = send( sockFD, s, strlen(s), 0 );
if( _send < 0 ) goto error;
_recv = recv( sockFD, r, 256, 0 );
if( r[0] != '2' ) goto error;
memset( r, 0, 256 );
sprintf( s, "QUIT\r\n" );
_send = send( sockFD, s, strlen(s), 0 );
if( _send < 0 ) goto error;
_recv = recv( sockFD, r, 256, 0 );
CLOSESOCKET( sockFD );
return 1;
error:
CLOSESOCKET( sockFD );
return 0;
}