#include <windows.h>
#include "leashdll.h"
#include <leashwin.h>
#include "leash-int.h"
HINSTANCE hLeashInst;
HINSTANCE hKrb5 = 0;
HINSTANCE hKrb524 = 0;
HINSTANCE hSecur32 = 0;
HINSTANCE hComErr = 0;
HINSTANCE hService = 0;
HINSTANCE hProfile = 0;
HINSTANCE hPsapi = 0;
HINSTANCE hToolHelp32 = 0;
HINSTANCE hCcapi = 0;
// krb5 functions
DECL_FUNC_PTR(krb5_change_password);
DECL_FUNC_PTR(krb5_get_init_creds_opt_alloc);
DECL_FUNC_PTR(krb5_get_init_creds_opt_free);
DECL_FUNC_PTR(krb5_get_init_creds_opt_init);
DECL_FUNC_PTR(krb5_get_init_creds_opt_set_tkt_life);
DECL_FUNC_PTR(krb5_get_init_creds_opt_set_renew_life);
DECL_FUNC_PTR(krb5_get_init_creds_opt_set_forwardable);
DECL_FUNC_PTR(krb5_get_init_creds_opt_set_proxiable);
DECL_FUNC_PTR(krb5_get_init_creds_opt_set_address_list);
DECL_FUNC_PTR(krb5_get_init_creds_opt_set_out_ccache);
DECL_FUNC_PTR(krb5_get_init_creds_password);
DECL_FUNC_PTR(krb5_build_principal_ext);
DECL_FUNC_PTR(krb5_cc_get_name);
DECL_FUNC_PTR(krb5_cc_resolve);
DECL_FUNC_PTR(krb5_cc_default);
DECL_FUNC_PTR(krb5_cc_default_name);
DECL_FUNC_PTR(krb5_cc_set_default_name);
DECL_FUNC_PTR(krb5_cc_initialize);
DECL_FUNC_PTR(krb5_cc_destroy);
DECL_FUNC_PTR(krb5_cc_close);
DECL_FUNC_PTR(krb5_cc_store_cred);
DECL_FUNC_PTR(krb5_cc_copy_creds);
// DECL_FUNC_PTR(krb5_cc_retrieve_cred);
DECL_FUNC_PTR(krb5_cc_get_principal);
DECL_FUNC_PTR(krb5_cc_start_seq_get);
DECL_FUNC_PTR(krb5_cc_next_cred);
DECL_FUNC_PTR(krb5_cc_end_seq_get);
// DECL_FUNC_PTR(krb5_cc_remove_cred);
DECL_FUNC_PTR(krb5_cc_set_flags);
// DECL_FUNC_PTR(krb5_cc_get_type);
DECL_FUNC_PTR(krb5_free_context);
DECL_FUNC_PTR(krb5_free_cred_contents);
DECL_FUNC_PTR(krb5_free_principal);
DECL_FUNC_PTR(krb5_get_in_tkt_with_password);
DECL_FUNC_PTR(krb5_init_context);
DECL_FUNC_PTR(krb5_parse_name);
DECL_FUNC_PTR(krb5_timeofday);
DECL_FUNC_PTR(krb5_timestamp_to_sfstring);
DECL_FUNC_PTR(krb5_unparse_name);
DECL_FUNC_PTR(krb5_get_credentials);
DECL_FUNC_PTR(krb5_mk_req);
DECL_FUNC_PTR(krb5_sname_to_principal);
DECL_FUNC_PTR(krb5_get_credentials_renew);
DECL_FUNC_PTR(krb5_free_data);
DECL_FUNC_PTR(krb5_free_data_contents);
// DECL_FUNC_PTR(krb5_get_realm_domain);
DECL_FUNC_PTR(krb5_free_unparsed_name);
DECL_FUNC_PTR(krb5_os_localaddr);
DECL_FUNC_PTR(krb5_copy_keyblock_contents);
DECL_FUNC_PTR(krb5_copy_data);
DECL_FUNC_PTR(krb5_free_creds);
DECL_FUNC_PTR(krb5_build_principal);
DECL_FUNC_PTR(krb5_get_renewed_creds);
DECL_FUNC_PTR(krb5_get_default_config_files);
DECL_FUNC_PTR(krb5_free_config_files);
DECL_FUNC_PTR(krb5_get_default_realm);
DECL_FUNC_PTR(krb5_free_ticket);
DECL_FUNC_PTR(krb5_decode_ticket);
DECL_FUNC_PTR(krb5_get_host_realm);
DECL_FUNC_PTR(krb5_free_host_realm);
DECL_FUNC_PTR(krb5_c_random_make_octets);
DECL_FUNC_PTR(krb5_free_addresses);
DECL_FUNC_PTR(krb5_free_default_realm);
DECL_FUNC_PTR(krb5_principal_compare);
DECL_FUNC_PTR(krb5_string_to_deltat);
DECL_FUNC_PTR(krb5_is_config_principal);
DECL_FUNC_PTR(krb5_cccol_cursor_new);
DECL_FUNC_PTR(krb5_cccol_cursor_free);
DECL_FUNC_PTR(krb5_cccol_cursor_next);
DECL_FUNC_PTR(krb5_cc_cache_match);
DECL_FUNC_PTR(krb5_cc_get_type);
DECL_FUNC_PTR(krb5_cc_new_unique);
DECL_FUNC_PTR(krb5_cc_support_switch);
DECL_FUNC_PTR(krb5_cc_switch);
DECL_FUNC_PTR(krb5_cc_get_full_name);
DECL_FUNC_PTR(krb5_free_string);
DECL_FUNC_PTR(krb5int_cc_user_set_default_name);
// ComErr functions
DECL_FUNC_PTR(com_err);
DECL_FUNC_PTR(error_message);
// Profile functions
DECL_FUNC_PTR(profile_init);
DECL_FUNC_PTR(profile_release);
DECL_FUNC_PTR(profile_get_subsection_names);
DECL_FUNC_PTR(profile_free_list);
DECL_FUNC_PTR(profile_get_string);
DECL_FUNC_PTR(profile_release_string);
DECL_FUNC_PTR(profile_get_integer);
// Service functions
DECL_FUNC_PTR(OpenSCManagerA);
DECL_FUNC_PTR(OpenServiceA);
DECL_FUNC_PTR(QueryServiceStatus);
DECL_FUNC_PTR(CloseServiceHandle);
DECL_FUNC_PTR(LsaNtStatusToWinError);
// LSA Functions
DECL_FUNC_PTR(LsaConnectUntrusted);
DECL_FUNC_PTR(LsaLookupAuthenticationPackage);
DECL_FUNC_PTR(LsaCallAuthenticationPackage);
DECL_FUNC_PTR(LsaFreeReturnBuffer);
DECL_FUNC_PTR(LsaGetLogonSessionData);
// CCAPI Functions
DECL_FUNC_PTR(cc_initialize);
DECL_FUNC_PTR(cc_shutdown);
DECL_FUNC_PTR(cc_get_NC_info);
DECL_FUNC_PTR(cc_free_NC_info);
FUNC_INFO k5_fi[] = {
MAKE_FUNC_INFO(krb5_change_password),
MAKE_FUNC_INFO(krb5_get_init_creds_opt_alloc),
MAKE_FUNC_INFO(krb5_get_init_creds_opt_free),
MAKE_FUNC_INFO(krb5_get_init_creds_opt_init),
MAKE_FUNC_INFO(krb5_get_init_creds_opt_set_tkt_life),
MAKE_FUNC_INFO(krb5_get_init_creds_opt_set_renew_life),
MAKE_FUNC_INFO(krb5_get_init_creds_opt_set_forwardable),
MAKE_FUNC_INFO(krb5_get_init_creds_opt_set_proxiable),
MAKE_FUNC_INFO(krb5_get_init_creds_opt_set_address_list),
MAKE_FUNC_INFO(krb5_get_init_creds_opt_set_out_ccache),
MAKE_FUNC_INFO(krb5_get_init_creds_password),
MAKE_FUNC_INFO(krb5_build_principal_ext),
MAKE_FUNC_INFO(krb5_cc_get_name),
MAKE_FUNC_INFO(krb5_cc_resolve),
MAKE_FUNC_INFO(krb5_cc_default),
MAKE_FUNC_INFO(krb5_cc_default_name),
MAKE_FUNC_INFO(krb5_cc_set_default_name),
MAKE_FUNC_INFO(krb5_cc_initialize),
MAKE_FUNC_INFO(krb5_cc_destroy),
MAKE_FUNC_INFO(krb5_cc_close),
MAKE_FUNC_INFO(krb5_cc_copy_creds),
MAKE_FUNC_INFO(krb5_cc_store_cred),
// MAKE_FUNC_INFO(krb5_cc_retrieve_cred),
MAKE_FUNC_INFO(krb5_cc_get_principal),
MAKE_FUNC_INFO(krb5_cc_start_seq_get),
MAKE_FUNC_INFO(krb5_cc_next_cred),
MAKE_FUNC_INFO(krb5_cc_end_seq_get),
// MAKE_FUNC_INFO(krb5_cc_remove_cred),
MAKE_FUNC_INFO(krb5_cc_set_flags),
// MAKE_FUNC_INFO(krb5_cc_get_type),
MAKE_FUNC_INFO(krb5_free_context),
MAKE_FUNC_INFO(krb5_free_cred_contents),
MAKE_FUNC_INFO(krb5_free_principal),
MAKE_FUNC_INFO(krb5_get_in_tkt_with_password),
MAKE_FUNC_INFO(krb5_init_context),
MAKE_FUNC_INFO(krb5_parse_name),
MAKE_FUNC_INFO(krb5_timeofday),
MAKE_FUNC_INFO(krb5_timestamp_to_sfstring),
MAKE_FUNC_INFO(krb5_unparse_name),
MAKE_FUNC_INFO(krb5_get_credentials),
MAKE_FUNC_INFO(krb5_mk_req),
MAKE_FUNC_INFO(krb5_sname_to_principal),
MAKE_FUNC_INFO(krb5_get_credentials_renew),
MAKE_FUNC_INFO(krb5_free_data),
MAKE_FUNC_INFO(krb5_free_data_contents),
// MAKE_FUNC_INFO(krb5_get_realm_domain),
MAKE_FUNC_INFO(krb5_free_unparsed_name),
MAKE_FUNC_INFO(krb5_os_localaddr),
MAKE_FUNC_INFO(krb5_copy_keyblock_contents),
MAKE_FUNC_INFO(krb5_copy_data),
MAKE_FUNC_INFO(krb5_free_creds),
MAKE_FUNC_INFO(krb5_build_principal),
MAKE_FUNC_INFO(krb5_get_renewed_creds),
MAKE_FUNC_INFO(krb5_free_addresses),
MAKE_FUNC_INFO(krb5_get_default_config_files),
MAKE_FUNC_INFO(krb5_free_config_files),
MAKE_FUNC_INFO(krb5_get_default_realm),
MAKE_FUNC_INFO(krb5_free_ticket),
MAKE_FUNC_INFO(krb5_decode_ticket),
MAKE_FUNC_INFO(krb5_get_host_realm),
MAKE_FUNC_INFO(krb5_free_host_realm),
MAKE_FUNC_INFO(krb5_c_random_make_octets),
MAKE_FUNC_INFO(krb5_free_default_realm),
MAKE_FUNC_INFO(krb5_principal_compare),
MAKE_FUNC_INFO(krb5_string_to_deltat),
MAKE_FUNC_INFO(krb5_is_config_principal),
MAKE_FUNC_INFO(krb5_cccol_cursor_new),
MAKE_FUNC_INFO(krb5_cccol_cursor_next),
MAKE_FUNC_INFO(krb5_cccol_cursor_free),
MAKE_FUNC_INFO(krb5_cc_cache_match),
MAKE_FUNC_INFO(krb5_cc_get_type),
MAKE_FUNC_INFO(krb5_cc_new_unique),
MAKE_FUNC_INFO(krb5_cc_support_switch),
MAKE_FUNC_INFO(krb5_cc_switch),
MAKE_FUNC_INFO(krb5_cc_get_full_name),
MAKE_FUNC_INFO(krb5_free_string),
MAKE_FUNC_INFO(krb5int_cc_user_set_default_name),
END_FUNC_INFO
};
FUNC_INFO profile_fi[] = {
MAKE_FUNC_INFO(profile_init),
MAKE_FUNC_INFO(profile_release),
MAKE_FUNC_INFO(profile_get_subsection_names),
MAKE_FUNC_INFO(profile_free_list),
MAKE_FUNC_INFO(profile_get_string),
MAKE_FUNC_INFO(profile_release_string),
MAKE_FUNC_INFO(profile_get_integer),
END_FUNC_INFO
};
FUNC_INFO ce_fi[] = {
MAKE_FUNC_INFO(com_err),
MAKE_FUNC_INFO(error_message),
END_FUNC_INFO
};
FUNC_INFO service_fi[] = {
MAKE_FUNC_INFO(OpenSCManagerA),
MAKE_FUNC_INFO(OpenServiceA),
MAKE_FUNC_INFO(QueryServiceStatus),
MAKE_FUNC_INFO(CloseServiceHandle),
MAKE_FUNC_INFO(LsaNtStatusToWinError),
END_FUNC_INFO
};
FUNC_INFO lsa_fi[] = {
MAKE_FUNC_INFO(LsaConnectUntrusted),
MAKE_FUNC_INFO(LsaLookupAuthenticationPackage),
MAKE_FUNC_INFO(LsaCallAuthenticationPackage),
MAKE_FUNC_INFO(LsaFreeReturnBuffer),
MAKE_FUNC_INFO(LsaGetLogonSessionData),
END_FUNC_INFO
};
// CCAPI v2
FUNC_INFO ccapi_fi[] = {
MAKE_FUNC_INFO(cc_initialize),
MAKE_FUNC_INFO(cc_shutdown),
MAKE_FUNC_INFO(cc_get_NC_info),
MAKE_FUNC_INFO(cc_free_NC_info),
END_FUNC_INFO
};
// psapi functions
DECL_FUNC_PTR(GetModuleFileNameExA);
DECL_FUNC_PTR(EnumProcessModules);
FUNC_INFO psapi_fi[] = {
MAKE_FUNC_INFO(GetModuleFileNameExA),
MAKE_FUNC_INFO(EnumProcessModules),
END_FUNC_INFO
};
// toolhelp functions
DECL_FUNC_PTR(CreateToolhelp32Snapshot);
DECL_FUNC_PTR(Module32First);
DECL_FUNC_PTR(Module32Next);
FUNC_INFO toolhelp_fi[] = {
MAKE_FUNC_INFO(CreateToolhelp32Snapshot),
MAKE_FUNC_INFO(Module32First),
MAKE_FUNC_INFO(Module32Next),
END_FUNC_INFO
};
BOOL WINAPI
DllMain(
HANDLE hinstDLL,
DWORD fdwReason,
LPVOID lpReserved
)
{
hLeashInst = hinstDLL;
switch (fdwReason)
{
case DLL_PROCESS_ATTACH:
{
OSVERSIONINFO osvi;
LoadFuncs(KRB5_DLL, k5_fi, &hKrb5, 0, 1, 0, 0);
LoadFuncs(COMERR_DLL, ce_fi, &hComErr, 0, 0, 1, 0);
LoadFuncs(SERVICE_DLL, service_fi, &hService, 0, 1, 0, 0);
LoadFuncs(SECUR32_DLL, lsa_fi, &hSecur32, 0, 1, 1, 1);
LoadFuncs(PROFILE_DLL, profile_fi, &hProfile, 0, 1, 0, 0);
LoadFuncs(CCAPI_DLL, ccapi_fi, &hCcapi, 0, 1, 0, 0);
memset(&osvi, 0, sizeof(OSVERSIONINFO));
osvi.dwOSVersionInfoSize = sizeof(OSVERSIONINFO);
GetVersionEx(&osvi);
// XXX: We should really use feature testing, first
// checking for CreateToolhelp32Snapshot. If that's
// not around, we try the psapi stuff.
//
// Only load LSA functions if on NT/2000/XP
if(osvi.dwPlatformId == VER_PLATFORM_WIN32_WINDOWS)
{
// Windows 9x
LoadFuncs(TOOLHELPDLL, toolhelp_fi, &hToolHelp32, 0, 1, 0, 0);
hPsapi = 0;
}
else if(osvi.dwPlatformId == VER_PLATFORM_WIN32_NT)
{
// Windows NT
LoadFuncs(PSAPIDLL, psapi_fi, &hPsapi, 0, 1, 0, 0);
hToolHelp32 = 0;
}
/*
* Register window class for the MITPasswordControl that
* replaces normal edit controls for password input.
* zero any fields we don't explicitly set
*/
hLeashInst = hinstDLL;
Register_MITPasswordEditControl(hLeashInst);
return TRUE;
}
case DLL_PROCESS_DETACH:
if (hKrb5)
FreeLibrary(hKrb5);
if (hCcapi)
FreeLibrary(hCcapi);
if (hProfile)
FreeLibrary(hProfile);
if (hComErr)
FreeLibrary(hComErr);
if (hService)
FreeLibrary(hService);
if (hSecur32)
FreeLibrary(hSecur32);
if (hPsapi)
FreeLibrary(hPsapi);
if (hToolHelp32)
FreeLibrary(hToolHelp32);
return TRUE;
default:
return TRUE;
}
}