// 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.

#pragma warning(disable: 244 267 311 312)

#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;
}