static INT_PTR GetEvent(WPARAM wParam,LPARAM lParam)
{
struct DBEvent *dbe;
DBEVENTINFO *dbei=(DBEVENTINFO*)lParam;
int bytesToCopy,i;
if(dbei==NULL||dbei->cbSize!=sizeof(DBEVENTINFO)) return 1;
if(dbei->cbBlob > 0 && dbei->pBlob == NULL) {
dbei->cbBlob = 0;
return 1;
}
EnterCriticalSection(&csDbAccess);
dbe=(struct DBEvent*)DBRead(wParam,sizeof(struct DBEvent),NULL);
if(dbe->signature!=DBEVENT_SIGNATURE) {
LeaveCriticalSection(&csDbAccess);
return 1;
}
dbei->szModule=GetModuleNameByOfs(dbe->ofsModuleName);
dbei->timestamp=dbe->timestamp;
dbei->flags=dbe->flags;
dbei->eventType=dbe->eventType;
if(dbei->cbBlob<dbe->cbBlob) bytesToCopy=dbei->cbBlob;
else bytesToCopy=dbe->cbBlob;
dbei->cbBlob=dbe->cbBlob;
for(i=0;;i+=MAXCACHEDREADSIZE) {
if(bytesToCopy-i<=MAXCACHEDREADSIZE) {
CopyMemory(dbei->pBlob+i,DBRead(wParam+offsetof(struct DBEvent,blob)+i,bytesToCopy-i,NULL),bytesToCopy-i);
break;
}
CopyMemory(dbei->pBlob+i,DBRead(wParam+offsetof(struct DBEvent,blob)+i,MAXCACHEDREADSIZE,NULL),MAXCACHEDREADSIZE);
}
STDMETHODIMP_(BOOL) CDbxKyoto::GetEvent(MEVENT hDbEvent, DBEVENTINFO *dbei)
{
if (hDbEvent == 0 || dbei == NULL || dbei->cbSize != sizeof(DBEVENTINFO)) return 1;
if (dbei->cbBlob > 0 && dbei->pBlob == NULL) {
dbei->cbBlob = 0;
return 1;
}
mir_cslock lck(m_csDbAccess);
VisitorCopyRec visitor;
if (!m_dbEvents.accept((LPCSTR)&hDbEvent, sizeof(MEVENT), &visitor, false))
return 1;
DBEvent *dbe = (DBEvent*)visitor.vbuf_;
if (dbe->dwSignature != DBEVENT_SIGNATURE)
return 1;
dbei->szModule = GetModuleNameByOfs(dbe->ofsModuleName);
dbei->timestamp = dbe->timestamp;
dbei->flags = dbe->flags;
dbei->eventType = dbe->wEventType;
int bytesToCopy = (dbei->cbBlob < dbe->cbBlob) ? dbei->cbBlob : dbe->cbBlob;
dbei->cbBlob = dbe->cbBlob;
if (bytesToCopy && dbei->pBlob) {
BYTE *pSrc = (BYTE*)visitor.vbuf_ + sizeof(DBEvent);
if (dbe->flags & DBEF_ENCRYPTED) {
dbei->flags &= ~DBEF_ENCRYPTED;
size_t len;
BYTE* pBlob = (BYTE*)m_crypto->decodeBuffer(pSrc, dbe->cbBlob, &len);
if (pBlob == NULL)
return 1;
memcpy(dbei->pBlob, pBlob, bytesToCopy);
if (bytesToCopy > (int)len)
memset(dbei->pBlob + len, 0, bytesToCopy - len);
mir_free(pBlob);
}
else memcpy(dbei->pBlob, pSrc, bytesToCopy);
}
return 0;
}
STDMETHODIMP_(BOOL) CDb3Mmap::GetEvent(MEVENT hDbEvent, DBEVENTINFO *dbei)
{
if (dbei == nullptr) return 1;
if (dbei->cbBlob > 0 && dbei->pBlob == nullptr) {
dbei->cbBlob = 0;
return 1;
}
mir_cslock lck(m_csDbAccess);
DBEvent *dbe = AdaptEvent((DWORD)hDbEvent, 0);
if (dbe->signature != DBEVENT_SIGNATURE)
return 1;
dbei->szModule = GetModuleNameByOfs(dbe->ofsModuleName);
dbei->timestamp = dbe->timestamp;
dbei->flags = dbe->flags;
dbei->eventType = dbe->wEventType;
int bytesToCopy = (dbei->cbBlob < dbe->cbBlob) ? dbei->cbBlob : dbe->cbBlob;
dbei->cbBlob = dbe->cbBlob;
if (bytesToCopy && dbei->pBlob) {
BYTE *pSrc;
if (m_dbHeader.version >= DB_095_1_VERSION)
pSrc = DBRead(DWORD(hDbEvent) + offsetof(DBEvent, blob), nullptr);
else
pSrc = DBRead(DWORD(hDbEvent) + offsetof(DBEvent_094, blob), nullptr);
if (dbe->flags & DBEF_ENCRYPTED) {
dbei->flags &= ~DBEF_ENCRYPTED;
size_t len;
BYTE* pBlob = (BYTE*)m_crypto->decodeBuffer(pSrc, dbe->cbBlob, &len);
if (pBlob == nullptr)
return 1;
memcpy(dbei->pBlob, pBlob, bytesToCopy);
if (bytesToCopy > (int)len)
memset(dbei->pBlob + len, 0, bytesToCopy - len);
mir_free(pBlob);
}
else memcpy(dbei->pBlob, pSrc, bytesToCopy);
}
return 0;
}
STDMETHODIMP_(BOOL) CDb3Mmap::GetEvent(HANDLE hDbEvent, DBEVENTINFO *dbei)
{
if (dbei == NULL || dbei->cbSize != sizeof(DBEVENTINFO)) return 1;
if (dbei->cbBlob > 0 && dbei->pBlob == NULL) {
dbei->cbBlob = 0;
return 1;
}
mir_cslock lck(m_csDbAccess);
DBEvent *dbe = (DBEvent*)DBRead((DWORD)hDbEvent, sizeof(DBEvent), NULL);
if (dbe->signature != DBEVENT_SIGNATURE)
return 1;
dbei->szModule = GetModuleNameByOfs(dbe->ofsModuleName);
dbei->timestamp = dbe->timestamp;
dbei->flags = dbe->flags;
dbei->eventType = dbe->wEventType;
int bytesToCopy = (dbei->cbBlob < dbe->cbBlob) ? dbei->cbBlob : dbe->cbBlob;
dbei->cbBlob = dbe->cbBlob;
if (bytesToCopy && dbei->pBlob) {
BYTE *pSrc = DBRead(DWORD(hDbEvent) + offsetof(DBEvent, blob), bytesToCopy, NULL);
if (dbe->flags & DBEF_ENCRYPTED) {
dbei->flags &= ~DBEF_ENCRYPTED;
size_t len;
BYTE* pBlob = (BYTE*)m_crypto->decodeBuffer(pSrc, dbe->cbBlob, &len);
if (pBlob == NULL)
return 1;
memcpy(dbei->pBlob, pBlob, bytesToCopy);
if (bytesToCopy > (int)len)
memset(dbei->pBlob + len, 0, bytesToCopy - len);
mir_free(pBlob);
}
else MoveMemory(dbei->pBlob, pSrc, bytesToCopy);
}
return 0;
}
void CDb3Mmap::ToggleSettingsEncryption(MCONTACT contactID)
{
DWORD ofsContact = GetContactOffset(contactID);
if (ofsContact == 0)
return;
DBContact *contact = (DBContact*)DBRead(ofsContact, NULL);
if (contact->ofsFirstSettings == 0)
return;
// fast cycle through all settings
DBContactSettings *setting = (DBContactSettings*)DBRead(contact->ofsFirstSettings, NULL);
DWORD offset = contact->ofsFirstSettings;
char *szModule = GetModuleNameByOfs(setting->ofsModuleName);
if (szModule == NULL)
return;
while (true) {
OBJLIST<VarDescr> arSettings(10);
char szSetting[256];
int bytesRemaining, len;
DWORD ofsBlobPtr = offset + offsetof(DBContactSettings, blob), ofsNext = setting->ofsNext;
PBYTE pBlob = (PBYTE)DBRead(ofsBlobPtr, &bytesRemaining);
while (pBlob[0]) {
NeedBytes(1);
len = pBlob[0];
memcpy(szSetting, pBlob + 1, len); szSetting[len] = 0;
NeedBytes(1 + pBlob[0]);
MoveAlong(1 + pBlob[0]);
NeedBytes(5);
switch (pBlob[0]) {
case DBVT_ASCIIZ:
len = *(PWORD)(pBlob + 1);
// we need to convert a string into utf8 and encrypt it
if (!m_bEncrypted) {
BYTE bSave = pBlob[len + 3]; pBlob[len + 3] = 0;
arSettings.insert(new VarDescr(szSetting, mir_utf8encode((LPCSTR)pBlob + 3)));
pBlob[len + 3] = bSave;
}
NeedBytes(3 + len);
break;
case DBVT_UTF8:
len = *(PWORD)(pBlob + 1);
// we need to encrypt these strings
if (!m_bEncrypted) {
BYTE bSave = pBlob[len + 3]; pBlob[len + 3] = 0;
arSettings.insert(new VarDescr(szSetting, (LPCSTR)pBlob + 3));
pBlob[len + 3] = bSave;
}
NeedBytes(3 + len);
break;
case DBVT_ENCRYPTED:
len = *(PWORD)(pBlob + 1);
// we need to decrypt these strings
if (m_bEncrypted && !IsSettingEncrypted(szModule, szSetting))
arSettings.insert(new VarDescr(szSetting, pBlob + 3, len));
NeedBytes(3 + len);
break;
case DBVT_BLOB:
NeedBytes(3 + *(PWORD)(pBlob + 1));
break;
}
NeedBytes(3);
MoveAlong(1 + GetSettingValueLength(pBlob));
NeedBytes(1);
}
for (int i = 0; i < arSettings.getCount(); i++) {
VarDescr &p = arSettings[i];
if (!m_bEncrypted) {
size_t encodedLen;
BYTE *pResult = m_crypto->encodeString(p.szValue, &encodedLen);
if (pResult != NULL) {
DBCONTACTWRITESETTING dbcws = { szModule, p.szVar };
dbcws.value.type = DBVT_ENCRYPTED;
dbcws.value.pbVal = pResult;
dbcws.value.cpbVal = (WORD)encodedLen;
WriteContactSetting(contactID, &dbcws);
mir_free(pResult);
}
}
else {
size_t realLen;
ptrA decoded(m_crypto->decodeString((PBYTE)(char*)p.szValue, p.iLen, &realLen));
if (decoded != NULL) {
DBCONTACTWRITESETTING dbcws = { szModule, p.szVar };
dbcws.value.type = DBVT_UNENCRYPTED;
dbcws.value.pszVal = decoded;
dbcws.value.cchVal = (WORD)realLen;
WriteContactSetting(contactID, &dbcws);
}
}
}
if (!ofsNext)
break;
setting = (DBContactSettings*)DBRead(offset = ofsNext, NULL);
if ((szModule = GetModuleNameByOfs(setting->ofsModuleName)) == NULL)
break;
}
}
