#include <stdio.h>
#include "libsockets.h"
#include "libkeef.h"        // for strmatch(), etc.

#if defined (UNIX)
  #include <signal.h>       // for signal()
  #include <sys/signal.h>   // for SIGBUS, etc.
  #include <stdarg.h>
#else if defined(WIN32) // Windows
  #include <winsock.h>      // for closesocket()
  #include <direct.h>       // for _getcwd()
  #include <process.h>      // for getpid()

  #define  sleep(SECS)           Sleep(SECS*1000)
  #define  close(SOCKET)         closesocket(SOCKET)
  #define  read(SOCKET,BUF,LEN)  recv(SOCKET,BUF,LEN,/*flags=*/0)
  #define  write(SOCKET,BUF,LEN) send(SOCKET,BUF,LEN,/*flags=*/0)
#endif

#include <time.h>            // for time(), strftime(), etc.

/*global*/int g_fDebug     = 0;

#define	ErrorLog	printf
#define	DebugLog	if (g_fDebug) printf

#define MAXLEN_HOSTNAME   200

#ifndef PATH_SLASH
 #ifdef UNIX
  #define PATH_SLASH '/'
 #else
  #define PATH_SLASH '\\'
 #endif
#endif

#ifndef TXTNL
 #define TXTNL  "\r\n"
#endif

#define MAXLEN_OBSERVATIONS 200
#define MAXLEN_IP           (3+1+3+1+3+1+3+1)
#define MAX_IPS             20
#define MAXLEN_WEBSERVER    100
#define MAXLEN_PROTOCOL     50
#define MAXLEN_PROTOCODE    100

typedef struct
{
	char* pszSourceURL;
	BOOL  fVignette;
	BOOL  fWebObjects;
	BOOL  fOpenMarket;
	BOOL  fFutureTense;
	BOOL  fColdFusion;
	// - - - - - - - - - - - - - - -
	BOOL  fHostnameResolves;
	char* pszCanonicalName;
	BOOL  fDNSRoundRobin;
	char  szIP[(MAXLEN_IP+3) * MAX_IPS];
	// - - - - - - - - - - - - - - -
	BOOL  fHostConnects;
	BOOL  fHostResponds;
	char  szProtocol[MAXLEN_PROTOCOL+1];
	char  szProtocode[MAXLEN_PROTOCODE+1];
	char  szWebServer[MAXLEN_WEBSERVER+1];
	BOOL  fCookies;
	BOOL  fFrames;
	BOOL  fClientSideScripting;
}
ANALYSIS;

//-----------------------------------------------------------------------------
static char *g_Synopsis[] = {
//-----------------------------------------------------------------------------
	" ",
	"analyze - hit site to determine certain things by sockets",
	" ",
	"usage: analyze [-v] [-html] <FullURL>",
	" ",
	"where:",
	"  -v        : verbosity for sockets debugging",
	"  -html     : put output in HTML wrapper",
	"  <FullURL> : just copy and paste from a browser",
	" ",
	"History:",
	"  05 Aug 99 : keef wrote",
	" ",
	"(c) Copyright 1999 Ridgeware, Inc.",
	" ",
	NULL
};

//-----------------------------------------------------------------------------
static void Exit (int iExitCode)
//-----------------------------------------------------------------------------
{
	#ifdef WIN32
	WSACleanup();
	#endif

	exit (iExitCode);

} // Exit

//-----------------------------------------------------------------------------
static void SynopsisAndExit ()
//-----------------------------------------------------------------------------
{
	for (int ii=0; g_Synopsis[ii]; ++ii)
	{
		printf ("%s\n", g_Synopsis[ii]);
	}

	Exit(0);

} // SynopsisAndExit

#define TF(X) ((X)?"TRUE":"false")

//-----------------------------------------------------------------------------
void HtmlReport (WEBURL* pWebURL, ANALYSIS* pAnalysis)
//-----------------------------------------------------------------------------
{
	printf ("<HR>\n");
	printf ("<B>%s</B>\n", pAnalysis->pszSourceURL);
	printf ("<HR>\n");
	printf ("<BR>\n");
	printf ("<I>Straight off the URL...</I>\n");
	printf ("<PRE>\n");
	printf ("  %-20s ==> %s\n", "Hostname",     pWebURL->szWebHostname);
	printf ("  %-20s ==> %s\n", "Vignette?",    TF(pAnalysis->fVignette));
	printf ("  %-20s ==> %s\n", "WebObjects?",  TF(pAnalysis->fWebObjects));
	printf ("  %-20s ==> %s\n", "OpenMarket?",  TF(pAnalysis->fOpenMarket));
	printf ("  %-20s ==> %s\n", "FutureTense?", TF(pAnalysis->fFutureTense));
	printf ("  %-20s ==> %s\n", "ColdFusion?",  TF(pAnalysis->fColdFusion));
	printf ("\n</PRE>");
	printf ("<I>Using DNS (domain name service)...</I>\n");
	printf ("<PRE>\n");
	printf ("  %-20s ==> %s\n", "Hostname Resolves?", TF(pAnalysis->fHostnameResolves));
	//printf ("  %-20s ==> %s\n", "Canonical Hostname", pAnalysis->pszCanonicalName);
	printf ("  %-20s ==> %s\n", "DNS Round Robin?", TF(pAnalysis->fDNSRoundRobin));
	if (pAnalysis->fDNSRoundRobin)
		printf ("  %-20s\n  %s\n", "IP Address Pool:", pAnalysis->szIP);
	else
		printf ("  %-20s ==> %s\n", "Current IP Address", pAnalysis->szIP);
	printf ("\n</PRE>");
	printf ("<I>By connecting to the host...</I>\n");
	printf ("<PRE>\n");
	printf ("  %-20s ==> %s\n", "Host Connects?",   TF(pAnalysis->fHostConnects));
	printf ("  %-20s ==> %s\n", "Host Responds?",   TF(pAnalysis->fHostResponds));
	printf ("  %-20s ==> %s\n", "Protocol Version", pAnalysis->szProtocol);
	printf ("  %-20s ==> %s\n", "Protocol Response", pAnalysis->szProtocode);
	printf ("  %-20s ==> %s\n", "HTTP Software",    pAnalysis->szWebServer);
	printf ("  %-20s ==> %s\n", "Cookies?",         TF(pAnalysis->fCookies));
	printf ("  %-20s ==> %s\n", "Frames?",          TF(pAnalysis->fFrames));
	printf ("  %-20s ==> %s\n", "Client-side Scripts?", TF(pAnalysis->fClientSideScripting));
	printf ("</PRE>\n");

} // HtmlReport


//-----------------------------------------------------------------------------
void ConsoleReport (WEBURL* pWebURL, ANALYSIS* pAnalysis)
//-----------------------------------------------------------------------------
{
	printf ("----------------------------------------------------------------------\n");
	printf ("%s\n", pAnalysis->pszSourceURL);
	printf ("----------------------------------------------------------------------\n");
	printf ("\n");
	printf ("  Straight off the URL...\n");
	printf ("  %-20s ==> %s\n", "Hostname",     pWebURL->szWebHostname);
	printf ("  %-20s ==> %s\n", "Vignette?",    TF(pAnalysis->fVignette));
	printf ("  %-20s ==> %s\n", "WebObjects?",  TF(pAnalysis->fWebObjects));
	printf ("  %-20s ==> %s\n", "OpenMarket?",  TF(pAnalysis->fOpenMarket));
	printf ("  %-20s ==> %s\n", "FutureTense?", TF(pAnalysis->fFutureTense));
	printf ("  %-20s ==> %s\n", "ColdFusion?",  TF(pAnalysis->fColdFusion));
	printf ("\n");
	printf ("  Using DNS (domain name service)...\n");
	printf ("  %-20s ==> %s\n", "Hostname Resolves?", TF(pAnalysis->fHostnameResolves));
	//printf ("  %-20s ==> %s\n", "Canonical Hostname", pAnalysis->pszCanonicalName);
	printf ("  %-20s ==> %s\n", "DNS Round Robin?", TF(pAnalysis->fDNSRoundRobin));
	if (pAnalysis->fDNSRoundRobin)
		printf ("  %-20s\n  %s\n", "IP Address Pool:", pAnalysis->szIP);
	else
		printf ("  %-20s ==> %s\n", "Current IP Address", pAnalysis->szIP);
	printf ("\n");
	printf ("  By connecting to the host...\n");
	printf ("  %-20s ==> %s\n", "Host Connects?",   TF(pAnalysis->fHostConnects));
	printf ("  %-20s ==> %s\n", "Host Responds?",   TF(pAnalysis->fHostResponds));
	printf ("  %-20s ==> %s\n", "Protocol Version", pAnalysis->szProtocol);
	printf ("  %-20s ==> %s\n", "Protocol Response", pAnalysis->szProtocode);
	printf ("  %-20s ==> %s\n", "HTTP Software",    pAnalysis->szWebServer);
	printf ("  %-20s ==> %s\n", "Cookies?",         TF(pAnalysis->fCookies));
	printf ("  %-20s ==> %s\n", "Frames?",          TF(pAnalysis->fFrames));
	printf ("  %-20s ==> %s\n", "Client-side Scripts?", TF(pAnalysis->fClientSideScripting));

} // ConsoleReport

//------------------------------------------------------------------------------
main (int argc, char* argv[])
//------------------------------------------------------------------------------
{
	char*  pszFullURL  = NULL;
	char** pArgArray[] = {&pszFullURL, NULL};
	int    iThisArg    = 0;
	int    iMaxArgs    = 1;
	BOOL   fHTML       = FALSE;
	ANALYSIS analysis;

	for (int ii=1; ii<argc; ++ii)
	{
		if (strmatch (argv[ii], "-v"))
			g_fDebug = 1;
		else if (strmatch (argv[ii], "-html"))
			fHTML = TRUE;
		else
		{
			if (iThisArg >= iMaxArgs)
			{
				printf ("analyze: argument '%s' not understood.\n", argv[ii]);
				SynopsisAndExit();
			}
			
			*(pArgArray[iThisArg]) = argv[ii];
			++iThisArg;
		}
	}

	// check required args:
	if (!pszFullURL || !(*pszFullURL))
		SynopsisAndExit ();

	WEBURL WebURL;

	DebugLog ("Parse the URL...\n");
	if (!ParseWebURL (pszFullURL, &WebURL))
	{
		ErrorLog ("analyze: invalid URL '%s'.\n", pszFullURL);
		Exit (-1);
	}

	memset (&analysis, sizeof(analysis), 0);
	analysis.pszSourceURL = pszFullURL;
	analysis.fVignette    = (strchr (WebURL.szWebRestURL, ',') != NULL);
	analysis.fWebObjects  = FALSE; // TODO
	analysis.fOpenMarket  = FALSE; // TODO
	analysis.fFutureTense = FALSE; // TODO
	analysis.fColdFusion  = (strstr (WebURL.szWebRestURL, ".cfm") != NULL);

	#ifdef WIN32
	WORD wVersionRequested = MAKEWORD(1,1); 
	WSADATA wsaData; 
	if (WSAStartup (wVersionRequested, &wsaData))
	{
		ErrorLog ("analyze: could not find a compatible WINSOCK DLL\n");
		Exit (-1);
 	}
	#endif

	struct sockaddr_in AddrServer;
	//char   szCanonical[MAXLEN_HOSTNAME+1];
	int    iLineNum = 0;
	int    fdSocket;

	DebugLog ("Resolve hostname into an IP address...\n");
	if (!ValidateHost (&AddrServer, WebURL.szWebHostname)) /*szCanonical, MAXLEN_HOSTNAME*/
	{
		analysis.fHostnameResolves = FALSE;
		goto Report;
	}

	analysis.fHostnameResolves = TRUE;
	analysis.fDNSRoundRobin = FALSE;
	strcpy (analysis.szIP, addr2ascii (AF_INET, &(AddrServer.sin_addr.s_addr), sizeof(AddrServer.sin_addr.s_addr), NULL));
	analysis.pszCanonicalName = ""; /*szCanonical*/

	DebugLog ("Look for DNS round-robin...\n");
	for (int ii=0; ii<MAX_IPS; ++ii)
	{
		struct sockaddr_in AddrServerAgain;
		ValidateHost (&AddrServerAgain, WebURL.szWebHostname);
		if (AddrServer.sin_addr.s_addr == AddrServer.sin_addr.s_addr)
			break; //for
		analysis.fDNSRoundRobin = TRUE;
		strcat (analysis.szIP, ", ");
		strcat (analysis.szIP, addr2ascii (AF_INET, &(AddrServerAgain.sin_addr.s_addr), sizeof(AddrServer.sin_addr.s_addr), NULL));
	}

	if (WebURL.iWebProtocol != WP_HTTP)
		goto Report;

	AddrServer.sin_port = htons(WebURL.iPortNum);
	char szCommand[500+1];

	DebugLog ("Now actually talk to site to get more info...\n");
	sprintf (szCommand, "GET %s HTTP/1.0\n\n", WebURL.szWebRestURL);

	fdSocket = SendCommand (&AddrServer, WebURL.szWebHostname, szCommand);
	if (fdSocket <= 0)
	{
		analysis.fHostConnects = FALSE;
		goto Report;
	}
	analysis.fHostConnects = TRUE;

	#define MAXLEN_BUFFER 2000
	static char szLine[MAXLEN_BUFFER+1];

	analysis.fCookies = FALSE;
	analysis.fFrames  = FALSE;
	analysis.fClientSideScripting = FALSE;

	while (SocketReadLine (fdSocket, szLine, MAXLEN_BUFFER))
	{
		// HTTP/1.1 200 OK
		// Server: Netscape-Enterprise/3.6 SP1
		// Date: Mon, 12 Jul 1999 18:27:13 GMT
		// Last-modified: 01-01-1970 01:01:01 GMT
		// Set-cookie: LLBEAN=SELF:1:092834l...; path=/; domain=.llbean.com; expires=Fri, 02 Aug 2002 20:28:31 GMT
		// Content-type: text/html
		// Last-modified: Wed, 21 Apr 1999 16:37:37 GMT
		// Content-length: 890
		// Accept-ranges: bytes
		// Connection: close
		//
		// <html>
		// ...

		DebugLog ("-| %s\n", szLine);
		if (++iLineNum == 1)
		{
			char* pSpace = strchr (szLine, ' ');
			if (pSpace)
				*pSpace = 0;

			strncpy (analysis.szProtocol, szLine, MAXLEN_PROTOCOL);
			analysis.szProtocol[MAXLEN_PROTOCOL] = 0;

			if (pSpace)
			{
				*pSpace = ' ';
				strncpy (analysis.szProtocode, pSpace+1, MAXLEN_PROTOCODE);
				analysis.szProtocode[MAXLEN_PROTOCODE] = 0;
			}
		}
		else if (strmatchN (szLine, "Server:"))
		{
			strncpy (analysis.szWebServer, szLine+sizeof("Server:"), MAXLEN_WEBSERVER);
			analysis.szWebServer[MAXLEN_WEBSERVER] = 0;
		}

		if (strmatchN (szLine, "Set-cookie:"))
			analysis.fCookies = TRUE;

		if (strstr (szLine, "<frameset"))
			analysis.fFrames = TRUE;
		else if (strstr (szLine, "<FRAMESET"))
			analysis.fFrames = TRUE;

		if (strstr (szLine, "<script"))
			analysis.fClientSideScripting = TRUE;
		else if (strstr (szLine, "<SCRIPT"))
			analysis.fClientSideScripting = TRUE;
	}

	DebugLog ("CloseSocket(): closing fdSocket = %d...\n", fdSocket);
	close (fdSocket);

	Report:
	if (fHTML)
	{
		HtmlReport (&WebURL, &analysis);
	}
	else
	{
		ConsoleReport (&WebURL, &analysis);
	}

	return (0);

} // main - analyze