#include "k5-int.h"

#ifdef KRB5

#include "krb5_err.h"

#include "kv5m_err.h"

#include "asn1_err.h"

#include "kdb5_err.h"

#include "profile.h"

extern void krb5_stdcc_shutdown();

#endif

#ifdef GSSAPI

#include "gssapi/generic/gssapi_err_generic.h"

#include "gssapi/krb5/gssapi_err_krb5.h"

#endif

/*

 * #defines for MIT-specific time-based timebombs and/or version

 * server for the Kerberos DLL.

 */

#ifdef SAP_TIMEBOMB

#define TIMEBOMB 865141200	/* 1-Jun-97 */

#define TIMEBOMB_PRODUCT "SAPGUI"

#define TIMEBOMB_WARN  15

#define TIMEBOMB_INFO "  Please see the web page at:\nhttp://web.mit.edu/reeng/www/saphelp for more information"

#define TIMEBOMB_ERROR KRB5_APPL_EXPIRED

#endif

#ifdef KRB_TIMEBOMB

#define TIMEBOMB 865141200	/* 1-Jun-97 */

#define TIMEBOMB_PRODUCT "Kerberos V5"

#define TIMEBOMB_WARN 15

#define TIMEBOMB_INFO "  Please see the web page at:\nhttp://web.mit.edu/reeng/www/saphelp for more information"

#define TIMEBOMB_ERROR KRB5_LIB_EXPIRED

#endif

/*

 * #defines for using MIT's version server DLL

 */

#ifdef SAP_VERSERV

#define APP_TITLE "KRB5-SAP"

#define APP_VER "3.0f"

#define APP_INI "krb5sap.ini"

#define VERSERV_ERROR 	KRB5_APPL_EXPIRED

#endif

#ifdef VERSERV

#define WINDOWS

#include <ver.h>

#include <vs.h>

#include <v.h>

/*

 * This function will get the version resource information from the

 * application using the DLL.  This allows us to Version Serve

 * arbitrary third party applications.  If there is an error, or we

 * decide that we should not version check the calling application

 * then VSflag will be FALSE when the function returns.

 *

 * The buffers passed into this function must be at least

 * APPVERINFO_SIZE bytes long.

 */

#define APPVERINFO_SIZE 256

void GetCallingAppVerInfo( char *AppTitle, char *AppVer, char *AppIni,

			  BOOL *VSflag)

{

	char CallerFilename[_MAX_PATH];

	LONG *lpLangInfo;

	DWORD hVersionInfoID, size;

	GLOBALHANDLE hVersionInfo;

	LPSTR lpVersionInfo;

	int dumint, retval;

	char *cp;

	char *revAppTitle;

	char szVerQ[90];

	LPBYTE locAppTitle;

	LPBYTE locAppVer;

	char locAppIni[_MAX_PATH];

#ifndef _WIN32

	WORD wStackSeg;

#endif /* !_WIN32 */

	/* first we need to get the calling module's filename */

#ifndef _WIN32

	_asm {

		mov wStackSeg, ss

	};

	retval = GetModuleFileName((HMODULE)wStackSeg, CallerFilename,

		_MAX_PATH);

#else

	/*

	 * Note: this may only work for single threaded applications,

	 * we'll live and learn ...

	 */

        retval = GetModuleFileName( NULL, CallerFilename, _MAX_PATH);

#endif

	if ( retval == 0 ) {

		VSflag = FALSE;

		return;

	}

	size = GetFileVersionInfoSize( CallerFilename, &hVersionInfoID);

	if( size == 0 ) {

		/*

		 * hey , I bet we don't have a version resource, let's

		 * punt

		 */

		*VSflag = FALSE;

		return;

	}

	hVersionInfo = GlobalAlloc(GHND, size);

	lpVersionInfo = GlobalLock(hVersionInfo);

	retval = GetFileVersionInfo( CallerFilename, hVersionInfoID, size,

				    lpVersionInfo);

	retval = VerQueryValue(lpVersionInfo, "\\VarFileInfo\\Translation",

			       (LPSTR *)&lpLangInfo, &dumint);

	wsprintf(szVerQ,

		 "\\StringFileInfo\\%04x%04x\\",

		 LOWORD(*lpLangInfo), HIWORD(*lpLangInfo));

	cp = szVerQ + lstrlen(szVerQ);

	lstrcpy(cp, "ProductName");

	/* try a localAppTitle and then a strcpy 4/2/97 */

	locAppTitle = 0;

	locAppVer = 0;

	retval = VerQueryValue(lpVersionInfo, szVerQ, &locAppTitle,

			       &dumint);

	lstrcpy(cp, "ProductVersion");

	retval = VerQueryValue(lpVersionInfo, szVerQ, &locAppVer,

			       &dumint);

	if (!locAppTitle || !locAppVer) {

	  	/* Punt, we don't have the right version resource records */

		*VSflag = FALSE;

		return;

	}

	/*

	 * We don't have a way to determine that INI file of the

	 * application at the moment so let's just use krb5.ini

	 */

	strncpy( locAppIni, KERBEROS_INI, sizeof(locAppIni) - 1 );

	locAppIni[ sizeof(locAppIni) - 1 ] = '\0';

	strncpy( AppTitle, locAppTitle, APPVERINFO_SIZE);

	AppTitle[APPVERINFO_SIZE - 1] = '\0';

	strncpy( AppVer, locAppVer, APPVERINFO_SIZE);

	AppVer[APPVERINFO_SIZE - 1] = '\0';

	strncpy( AppIni, locAppIni, APPVERINFO_SIZE);

	AppIni[APPVERINFO_SIZE - 1] = '\0';

	/*

	 * We also need to determine if we want to suppress version

	 * checking of this application.  Does the tail of the

	 * AppTitle end in a "-v" ?

	 */

	revAppTitle = _strrev( _strdup(AppTitle));

	if( revAppTitle[0] == 'v' || revAppTitle[0] == 'V'  &&

	   revAppTitle[1] == '-' ) {

		VSflag = FALSE;

	}

	return;

}

/*

 * Use the version server to give us some control on distribution and usage

 * We're going to test track as well

 */

static int CallVersionServer(app_title, app_version, app_ini, code_cover)

	char *app_title;

	char *app_version;

	char *app_ini;

	char *code_cover;

{

	VS_Request vrequest;

	VS_Status  vstatus;

	SetCursor(LoadCursor(NULL, IDC_WAIT));

	/*

	 * We should be able to pass in code_cover below, but things

	 * are breaking under Windows 16 for no good reason.

	 */

	vrequest = VSFormRequest((LPSTR) app_title, (LPSTR) app_version,

				 (LPSTR) app_ini,

				 NULL /* code_cover */, NULL,

				 V_CHECK_AND_LOG);

	SetCursor(LoadCursor(NULL, IDC_ARROW));

	/*

	 * If the user presses cancel when registering the test

	 * tracker, we'll let them continue.

	 */

	if (ReqStatus(vrequest) == V_E_CANCEL) {

		VSDestroyRequest(vrequest);

		return 0;

	}

	vstatus = VSProcessRequest(vrequest);

	/*

	 * Only complain periodically, if the test tracker isn't

	 * working...

	 */

	if (v_complain(vstatus, app_ini)) {

		WinVSReportRequest(vrequest, NULL,

				   "Version Server Status Report");

	}

	if (vstatus == V_REQUIRED) {

		SetCursor(LoadCursor(NULL, IDC_WAIT));

		VSDestroyRequest(vrequest);

		return( -1 );

	}

	VSDestroyRequest(vrequest);

	return (0);

}

#endif

#ifdef TIMEBOMB

static krb5_error_code do_timebomb()

{

	char buf[1024];

	long timeleft;

	static first_time = 1;

	timeleft = TIMEBOMB - time(0);

	if (timeleft <= 0) {

		if (first_time) {

			sprintf(buf, "Your version of %s has expired.\n",

				TIMEBOMB_PRODUCT);

			buf[sizeof(buf) - 1] = '\0';

			strncat(buf, "Please upgrade it.", sizeof(buf) - 1 - strlen(buf));

#ifdef TIMEBOMB_INFO

			strncat(buf, TIMEBOMB_INFO, sizeof(buf) - 1 - strlen(buf));

#endif

			MessageBox(NULL, buf, "", MB_OK);

			first_time = 0;

		}

		return TIMEBOMB_ERROR;

	}

	timeleft = timeleft / ((long) 60*60*24);

	if (timeleft < TIMEBOMB_WARN) {

		if (first_time) {

			sprintf(buf, "Your version of %s will expire in %ld days.\n",

				TIMEBOMB_PRODUCT, timeleft);

			strncat(buf, "Please upgrade it soon.", sizeof(buf) - 1 - strlen(buf));

#ifdef TIMEBOMB_INFO

			strncat(buf, TIMEBOMB_INFO, sizeof(buf) - 1 - strlen(buf));

#endif

			MessageBox(NULL, buf, "", MB_OK);

			first_time = 0;

		}

	}

	return 0;

}

#endif

/*

 * This was originally called from LibMain; unfortunately, Windows 3.1

 * doesn't allow you to make messaging calls from LibMain.  So, we now

 * do the timebomb/version server stuff from krb5_init_context().

 */

krb5_error_code krb5_vercheck()

{

	static int verchecked = 0;

	if (verchecked)

		return 0;

#ifdef TIMEBOMB

	krb5_error_code retval = do_timebomb();

	if (retval)

		return retval;

#endif

#ifdef VERSERV

	{

#ifdef APP_TITLE

		if (CallVersionServer(APP_TITLE, APP_VER, APP_INI, NULL))

			return VERSERV_ERROR;

#else

		char AppTitle[APPVERINFO_SIZE];

		char AppVer[APPVERINFO_SIZE];

		char AppIni[APPVERINFO_SIZE];

		BOOL VSflag=TRUE;

		GetCallingAppVerInfo( AppTitle, AppVer, AppIni, &VSflag);

		if (VSflag) {

			if (CallVersionServer(AppTitle, AppVer, AppIni, NULL))

				return KRB5_APPL_EXPIRED;

		}

#endif

	}

#endif

        verchecked = 1;

	return 0;

}

static HINSTANCE hlibinstance;

HINSTANCE get_lib_instance()

{

    return hlibinstance;

}

#define DLL_STARTUP 0

#define DLL_SHUTDOWN 1

static int

control(int mode)

{

    switch(mode) {

    case DLL_STARTUP:

	break;

    case DLL_SHUTDOWN:

#ifdef KRB5

	krb5_stdcc_shutdown();

#endif

	break;

#if defined(ENABLE_THREADS) && defined(SUPPORTLIB)

    case DLL_THREAD_DETACH:

	krb5int_thread_detach_hook();

	return 0;

#endif

    default:

	return -1;

    }

#if defined KRB5

    switch (mode) {

    case DLL_STARTUP:

	profile_library_initializer__auxinit();

	cryptoint_initialize_library__auxinit();

	krb5int_lib_init__auxinit();

	break;

    case DLL_SHUTDOWN:

	krb5int_lib_fini();

	cryptoint_cleanup_library();

	profile_library_finalizer();

	break;

    }

#elif defined GSSAPI

    switch (mode) {

    case DLL_STARTUP:

	gssint_mechglue_init__auxinit();

	break;

    case DLL_SHUTDOWN:

	gssint_mechglue_fini();

	break;

    }

#elif defined COMERR

    switch (mode) {

    case DLL_STARTUP:

	com_err_initialize__auxinit();

	break;

    case DLL_SHUTDOWN:

	com_err_terminate();

	break;

    }

#elif defined PROFILELIB

    switch (mode) {

    case DLL_STARTUP:

	profile_library_initializer__auxinit();

	break;

    case DLL_SHUTDOWN:

	profile_library_finalizer();

	break;

    }

#elif defined SUPPORTLIB

    switch (mode) {

    case DLL_STARTUP:

      krb5int_thread_support_init__auxinit();

      break;

    case DLL_SHUTDOWN:

      krb5int_thread_support_fini();

      break;

    }

#else

# error "Don't know the init/fini functions for this library."

#endif

    return 0;

}

#ifdef _WIN32

BOOL WINAPI DllMain (HANDLE hModule, DWORD fdwReason, LPVOID lpReserved)

{

    switch (fdwReason)

    {

        case DLL_PROCESS_ATTACH:

	    hlibinstance = (HINSTANCE) hModule;

	    if (control(DLL_STARTUP))

		return FALSE;

	    break;

        case DLL_THREAD_ATTACH:

	    break;

        case DLL_THREAD_DETACH:

	    if (control(DLL_THREAD_DETACH))

		return FALSE;

	    break;

        case DLL_PROCESS_DETACH:

	    if (control(DLL_SHUTDOWN))

		return FALSE;

	    break;

        default:

	    return FALSE;

    }

    return TRUE;   // successful DLL_PROCESS_ATTACH

}

#else

BOOL CALLBACK

LibMain (hInst, wDataSeg, cbHeap, CmdLine)

HINSTANCE hInst;

WORD wDataSeg;

WORD cbHeap;

LPSTR CmdLine;

{

    hlibinstance = hInst;

    if (control(DLL_STARTUP))

	return 0;

    else

	return 1;

}

int CALLBACK __export

WEP(nParam)

	int nParam;

{

    if (control(DLL_SHUTDOWN))

	return 0;

    else

	return 1;

}

#endif