CDATA.CPP

//**************************************************************************** 
//  Module:     NMCHAT.EXE 
//  File:       CDATA.CPP 
//  Content:     
//               
// 
//  Copyright (c) Microsoft Corporation 1997 
// 
// THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF  
// ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED TO  
// THE IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A  
// PARTICULAR PURPOSE. 
//**************************************************************************** 
 
#include "precomp.h" 
 
INmChannelData * g_pChannelData; 
CDataNotify * g_pNotifyData; 
 
//**************************************************************************** 
// 
// VOID DisplayMsg(LPTSTR pszMsg, INmMember * pMember) 
// 
//**************************************************************************** 
 
VOID DisplayMsg(LPTSTR pszMsg, INmMember * pMember) 
{ 
#ifdef NM2_CHAT_PROTOCOL 
#ifndef UNICODE 
// Convert UNICODE to ANSI 
char sz[MAX_PATH]; 
WideCharToMultiByte(CP_ACP, 0, (LPWSTR) pszMsg, -1, sz, MAX_PATH, NULL, NULL); 
pszMsg = sz; 
#endif /* UNICODE */ 
#endif /* NM2_CHAT_PROTOCOL */ 
 
DisplaySysMsg(pszMsg, pMember); 
} 
 
 
//**************************************************************************** 
// 
// VOID DisplaySysMsg(LPTSTR pszMsg, INmMember * pMember) 
// 
//**************************************************************************** 
 
VOID DisplaySysMsg(LPTSTR pszMsg, INmMember * pMember) 
{ 
LPTSTR pszUser = NULL; 
 
if (NULL != pMember) 
{ 
PUSER pUser = PUserFromINmMember(pMember); 
if (NULL != pUser) 
pszUser = pUser->pszName; 
} 
 
if (NULL == pszUser) 
{ 
pszUser = TEXT(""); 
} 
 
DisplayMessage(pszMsg, pszUser); 
} 
 
 
//**************************************************************************** 
// 
// VOID DisplayMessage(LPTSTR pszMsg, LPTSTR pszUser) 
// 
//**************************************************************************** 
 
VOID DisplayMessage(LPTSTR pszMsg, LPTSTR pszUser) 
{ 
int iItem; 
LV_ITEM  lvItem; 
 
ClearStruct(&lvItem); 
lvItem.iItem = 0x0FFFFFFF; 
lvItem.mask = LVIF_PARAM | LVIF_TEXT; 
lvItem.pszText = pszUser; 
lvItem.cchTextMax = lstrlen(lvItem.pszText); 
iItem = ListView_InsertItem(ghwndMsg, &lvItem); 
 
{ 
TCHAR sz[MAX_PATH]; 
SYSTEMTIME sysTime; 
GetLocalTime(&sysTime); 
wsprintf(sz, "%02d:%02d:%02d", sysTime.wHour, sysTime.wMinute, sysTime.wSecond); 
ListView_SetItemText(ghwndMsg, iItem, ILV_TIME, sz); 
} 
 
ListView_SetItemText(ghwndMsg, iItem, ILV_TEXT, pszMsg); 
} 
 
 
//**************************************************************************** 
// 
// HRESULT HookData(INmChannelData * pChannelData) 
// 
//**************************************************************************** 
 
HRESULT HookData(INmChannelData * pChannelData) 
{ 
HRESULT hr; 
CDataNotify * pNotify; 
 
ASSERT(NULL != pChannelData); 
ASSERT(NULL == g_pChannelData); 
 
g_pChannelData = pChannelData; 
 
    // Connect to the channel object 
    pNotify = new CDataNotify(); 
    if (pNotify == NULL) 
    { 
        hr = E_OUTOFMEMORY; 
    } 
    else 
    { 
hr = pNotify->Connect(pChannelData); 
if (SUCCEEDED(hr)) 
{ 
g_pNotifyData = pNotify; 
} 
} 
 
if (FAILED(hr)) 
{ 
FreeDataChannel(); 
} 
 
return hr; 
} 
 
 
//**************************************************************************** 
// 
// VOID FreeDataChannel(void) 
// 
//**************************************************************************** 
 
VOID FreeDataChannel(void) 
{ 
// Release all notify 
if (NULL != g_pNotifyData) 
{ 
g_pNotifyData->Disconnect(); 
g_pNotifyData->Release(); 
g_pNotifyData = NULL; 
} 
 
if (NULL != g_pChannelData) 
{ 
g_pChannelData->Release(); 
g_pChannelData = NULL; 
} 
} 
 
 
//**************************************************************************** 
// 
// CLASS CDataNotify 
// 
//**************************************************************************** 
 
 
//**************************************************************************** 
// 
// Constructor 
// 
//**************************************************************************** 
 
CDataNotify::CDataNotify() : RefCount(), CNotify() 
{ 
} 
 
 
//**************************************************************************** 
// 
// Destructor 
// 
//**************************************************************************** 
 
CDataNotify::~CDataNotify() 
{ 
} 
 
 
//**************************************************************************** 
// 
// Methods from IUnknown 
// 
//**************************************************************************** 
 
ULONG STDMETHODCALLTYPE CDataNotify::AddRef(void) 
{ 
return RefCount::AddRef(); 
} 
 
 
ULONG STDMETHODCALLTYPE CDataNotify::Release(void) 
{ 
return RefCount::Release(); 
} 
 
HRESULT STDMETHODCALLTYPE CDataNotify::QueryInterface(REFIID riid, PVOID *ppvObject) 
{ 
HRESULT hr = S_OK; 
 
if (riid == IID_IUnknown) 
{ 
*ppvObject = (IUnknown *)this; 
} 
else if (riid == IID_INmChannelDataNotify) 
{ 
*ppvObject = (INmChannelDataNotify *)this; 
} 
else 
{ 
hr = E_NOINTERFACE; 
*ppvObject = NULL; 
} 
 
if (S_OK == hr) 
{ 
AddRef(); 
} 
 
return hr; 
} 
 
 
//**************************************************************************** 
// 
// Methods from ICNotify 
// 
//**************************************************************************** 
 
HRESULT STDMETHODCALLTYPE CDataNotify::Connect(IUnknown *pUnk) 
{ 
return CNotify::Connect(pUnk, IID_INmChannelDataNotify, (IUnknown *)this); 
} 
 
HRESULT STDMETHODCALLTYPE CDataNotify::Disconnect(void) 
{ 
return CNotify::Disconnect(); 
} 
 
 
//**************************************************************************** 
// 
// Methods from INmChannelNotify 
// 
//**************************************************************************** 
 
HRESULT STDMETHODCALLTYPE CDataNotify::NmUI(CONFN uNotify) 
{ 
return S_OK; 
} 
 
HRESULT STDMETHODCALLTYPE CDataNotify::MemberChanged(NM_MEMBER_NOTIFY uNotify, INmMember *pMember) 
{ 
if (NM_MEMBER_ADDED == uNotify) 
{ 
AddMember(pMember); 
} 
else if (NM_MEMBER_REMOVED == uNotify) 
{ 
RemoveMember(pMember); 
} 
 
return S_OK; 
} 
 
HRESULT STDMETHODCALLTYPE CDataNotify::DataSent(INmMember *pMember, ULONG uSize, LPBYTE pb) 
{ 
LPTSTR psz; 
 
#ifdef NM2_CHAT_PROTOCOL 
ASSERT(uSize > CB_NM2_HDR); 
psz = (LPTSTR) (pb + *(LPDWORD) pb);  // skip past header 
#else 
psz = (LPTSTR) pb; 
#endif /* NM2_CHAT_PROTOCOL */ 
 
DisplayMsg(psz, g_pMemberLocal); 
 
return S_OK; 
} 
 
HRESULT STDMETHODCALLTYPE CDataNotify::DataReceived(INmMember *pMember, ULONG uSize, LPBYTE pb, ULONG dwFlags) 
{ 
LPTSTR psz; 
 
// Warning : Doesn't handle broken packets 
// TODO: Check dwFlags & NM_DF_SEGMENT_END 
 
#ifdef NM2_CHAT_PROTOCOL 
ASSERT(uSize > CB_NM2_HDR); 
psz = (LPTSTR) (pb + *(LPDWORD) pb);  // skip past header 
#else 
psz = (LPTSTR) pb; 
#endif /* NM2_CHAT_PROTOCOL */ 
 
DisplayMsg(psz, pMember); 
 
return S_OK; 
}