From c2df6e7bbf6efdb36f751379c154623191864a1f Mon Sep 17 00:00:00 2001 From: George Hazan Date: Sat, 21 Jul 2012 18:25:03 +0000 Subject: Dbx_mmap_SA ready git-svn-id: http://svn.miranda-ng.org/main/trunk@1094 1316c22d-e87f-b044-9b9b-93d7a3e3ba9c --- plugins/Dbx_mmap_SA/init.cpp | 164 +++++++++++++++++++------------------------ 1 file changed, 71 insertions(+), 93 deletions(-) (limited to 'plugins/Dbx_mmap_SA/init.cpp') diff --git a/plugins/Dbx_mmap_SA/init.cpp b/plugins/Dbx_mmap_SA/init.cpp index cf06474dde..2566e14471 100644 --- a/plugins/Dbx_mmap_SA/init.cpp +++ b/plugins/Dbx_mmap_SA/init.cpp @@ -23,21 +23,20 @@ Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. #include "commonheaders.h" -int hLangpack; +void InitSecurity(void); +void UnloadSecurity(void); -extern TCHAR szDbPath[MAX_PATH]; +int hLangpack; HINSTANCE g_hInst = NULL; -CDdxMmap* g_Db = NULL; - PLUGININFOEX pluginInfo = { sizeof(PLUGININFOEX), __PLUGIN_NAME, __VERSION_DWORD, "Provides Miranda database support: global settings, contacts, history, settings per contact. Enhanced modification with Encryption support.", "Miranda-IM project, modification by FYR and chaos.persei, nullbie, Billy_Bons", - "chaos.persei@gmail.com; ashpynov@gmail.com; bio@msx.ru; ghazan@miranda-im.org", + "chaos.persei@gmail.com; ashpynov@gmail.com; bio@msx.ru; ghazan@miranda.im", "Copyright 2000-2011 Miranda IM project, FYR, chaos.persei, induction, nullbie", "http://dbmmapmod.googlecode.com/", UNICODE_AWARE, @@ -45,132 +44,98 @@ PLUGININFOEX pluginInfo = { { 0x28ff9b91, 0x3e4d, 0x4f1c, { 0xb4, 0x7c, 0xc6, 0x41, 0xb0, 0x37, 0xff, 0x40 } } }; +LIST g_Dbs(1, (LIST::FTSortFunc)HandleKeySort); + ///////////////////////////////////////////////////////////////////////////////////////// // returns 0 if the profile is created, EMKPRF* static int makeDatabase(const TCHAR *profile, int *error) { - HANDLE hFile = CreateFile(profile, GENERIC_READ|GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, 0, NULL); - if ( hFile != INVALID_HANDLE_VALUE ) { - CreateDbHeaders(hFile); - CloseHandle(hFile); + CDdxMmapSA *tmp = new CDdxMmapSA(profile); + if (tmp->Create() == ERROR_SUCCESS) { + tmp->CreateDbHeaders(); + delete tmp; return 0; } - if ( error != NULL ) *error = EMKPRF_CREATEFAILED; + delete tmp; + if (error != NULL) *error = EMKPRF_CREATEFAILED; return 1; } // returns 0 if the given profile has a valid header -static int grokHeader(const TCHAR *profile, int *error ) +static int grokHeader(const TCHAR *profile, int *error) { - int rc = 1; - int chk = 0; - struct DBHeader hdr; - HANDLE hFile = INVALID_HANDLE_VALUE; - DWORD dummy = 0; - - hFile = CreateFile(profile, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, NULL); - if ( hFile == INVALID_HANDLE_VALUE ) { - if ( error != NULL ) *error = EGROKPRF_CANTREAD; + CDdxMmapSA *tmp = new CDdxMmapSA(profile); + if (tmp->Load(true) != ERROR_SUCCESS) { + delete tmp; + if (error != NULL) *error = EGROKPRF_CANTREAD; return 1; } - // read the header, which can fail (for various reasons) - if ( !ReadFile(hFile, &hdr, sizeof(struct DBHeader), &dummy, NULL)) { - if ( error != NULL) *error = EGROKPRF_CANTREAD; - CloseHandle(hFile); - return 1; - } - chk = CheckDbHeaders(&hdr); + + int chk = tmp->CheckDbHeaders(); + delete tmp; if ( chk == 0 ) { // all the internal tests passed, hurrah - rc = 0; - if ( error != NULL ) *error = 0; - } else { - // didn't pass at all, or some did. - switch ( chk ) { - case 1: - { - // "Miranda ICQ DB" wasn't present - if ( error != NULL ) *error = EGROKPRF_UNKHEADER; - break; - } - case 2: - { - // header was present, but version information newer - if ( error != NULL ) *error = EGROKPRF_VERNEWER; - break; - } - case 3: - { - // header/version OK, internal data missing - if ( error != NULL ) *error = EGROKPRF_DAMAGED; - break; - } - } // switch - } //if - CloseHandle(hFile); - return rc; + if (error != NULL) *error = 0; + return 0; + } + + // didn't pass at all, or some did. + switch ( chk ) { + case 1: + // "Miranda ICQ DB" wasn't present + if (error != NULL) *error = EGROKPRF_UNKHEADER; + break; + + case 2: + // header was present, but version information newer + if (error != NULL) *error = EGROKPRF_VERNEWER; + break; + + case 3: + // header/version OK, internal data missing + if (error != NULL) *error = EGROKPRF_DAMAGED; + break; + } + + return 1; } // returns 0 if all the APIs are injected otherwise, 1 static MIDatabase* LoadDatabase(const TCHAR *profile) { - if (g_Db) delete g_Db; - g_Db = new CDdxMmap(profile); - - // don't need thread notifications - _tcsncpy(szDbPath, profile, SIZEOF(szDbPath)); - // set the memory, lists & UTF8 manager mir_getLP( &pluginInfo ); - { // Are we running under unicode Miranda core ? - char szVer[MAX_PATH]; - CallService(MS_SYSTEM_GETVERSIONTEXT, MAX_PATH, (LPARAM)szVer); - _strlwr(szVer); // make sure it is lowercase - - /* WARNING!!! You CANNOT remove this code */ - if (strstr(szVer, "coffee") != NULL) - { - // We are running under damn violators - void (*f)(); - - MessageBox(0, TranslateT("Running mmap_sa is forbidden under license violating products, sorry"), TranslateT("Warning!"), MB_OK); - - f = NULL; - f(); - } - /* end of protected code */ + CDdxMmapSA* db = new CDdxMmapSA(profile); + if (db->Load(false) != ERROR_SUCCESS) { + delete db; + return NULL; } - // inject all APIs and hooks into the core - LoadDatabaseModule(); - - return g_Db; + g_Dbs.insert(db); + return db; } static int UnloadDatabase(MIDatabase* db) { - UnloadDatabaseModule(); - delete g_Db; g_Db = 0; + g_Dbs.remove((CDdxMmapSA*)db); + delete (CDdxMmapSA*)db; return 0; } -static DATABASELINK dblink = { +static DATABASELINK dblink = +{ sizeof(DATABASELINK), - "db3x secured_mmap driver", - _T("db3x mmap database support"), + "db3x secure mmap driver", + _T("db3x secure mmap database support"), makeDatabase, grokHeader, LoadDatabase, - UnloadDatabase, + UnloadDatabase }; -BOOL WINAPI DllMain(HINSTANCE hInstDLL, DWORD dwReason, LPVOID reserved) -{ - g_hInst = hInstDLL; - return TRUE; -} +///////////////////////////////////////////////////////////////////////////////////////// extern "C" __declspec(dllexport) PLUGININFOEX * MirandaPluginInfoEx(DWORD mirandaVersion) { @@ -181,10 +146,23 @@ extern "C" __declspec(dllexport) const MUUID MirandaInterfaces[] = {MIID_DATABAS extern "C" __declspec(dllexport) int Load(void) { + InitSecurity(); + InitPreset(); + + RegisterDatabasePlugin(&dblink); return 0; } extern "C" __declspec(dllexport) int Unload(void) { + g_Dbs.destroy(); + UnloadSecurity(); + UninitPreset(); return 0; -} \ No newline at end of file +} + +BOOL WINAPI DllMain(HINSTANCE hInstDLL, DWORD dwReason, LPVOID reserved) +{ + g_hInst = hInstDLL; + return TRUE; +} -- cgit v1.2.3