1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117 |
- // ProcessCopy.cpp: implementation of the CProcessCopy class.
- //
- //////////////////////////////////////////////////////////////////////
- #include "stdafx.h"
- #include "CP_Main.h"
- #include "ProcessCopy.h"
- #include "DatabaseUtilities.h"
- #include ".\processcopy.h"
- #ifdef _DEBUG
- #undef THIS_FILE
- static char THIS_FILE[]=__FILE__;
- #define new DEBUG_NEW
- #endif
- /*----------------------------------------------------------------------------*\
- COleDataObjectEx
- \*----------------------------------------------------------------------------*/
- HGLOBAL COleDataObjectEx::GetGlobalData(CLIPFORMAT cfFormat, LPFORMATETC lpFormatEtc)
- {
- HGLOBAL hGlobal = COleDataObject::GetGlobalData(cfFormat, lpFormatEtc);
- if( hGlobal )
- return hGlobal;
- // The data isn't in global memory, so try getting an IStream interface to it.
- STGMEDIUM stg;
-
- if( !GetData(cfFormat, &stg) )
- {
- ASSERT(0); // does this ever happen?
- return 0;
- }
- switch(stg.tymed)
- {
- case TYMED_HGLOBAL:
- hGlobal = stg.hGlobal;
- break;
-
- case TYMED_ISTREAM:
- {
- UINT uDataSize;
- LARGE_INTEGER li;
- ULARGE_INTEGER uli;
-
- li.HighPart = li.LowPart = 0;
-
- if ( SUCCEEDED( stg.pstm->Seek ( li, STREAM_SEEK_END, &uli )))
- {
- hGlobal = GlobalAlloc(GMEM_MOVEABLE | GMEM_SHARE, uli.LowPart );
- void* pv = GlobalLock(hGlobal);
- stg.pstm->Seek(li, STREAM_SEEK_SET, NULL);
- HRESULT result = stg.pstm->Read(pv, uli.LowPart, (PULONG)&uDataSize);
- GlobalUnlock(hGlobal);
- if( FAILED(result) )
- hGlobal = GlobalFree(hGlobal);
- }
- break; // case TYMED_ISTREAM
- }
- } // end switch
- ReleaseStgMedium(&stg);
- return hGlobal;
- }
- /*----------------------------------------------------------------------------*\
- CClipFormat - holds the data of one clip format.
- \*----------------------------------------------------------------------------*/
- /*----------------------------------------------------------------------------*\
- CClipFormats - holds an array of CClipFormat
- \*----------------------------------------------------------------------------*/
- // returns a pointer to the CClipFormat in this array which matches the given type
- // or NULL if that type doesn't exist in this array.
- CClipFormat* CClipFormats::FindFormat( UINT cfType )
- {
- CClipFormat* pCF;
- int count = GetCount();
- for( int i=0; i < count; i++ )
- {
- pCF = &GetAt(i);
- if( pCF->m_cfType == cfType )
- return pCF;
- }
- return NULL;
- }
- /*----------------------------------------------------------------------------*\
- CClip - holds multiple CClipFormats and CopyClipboard() statistics
- \*----------------------------------------------------------------------------*/
- CClip::CClip() : m_ID(0), m_lTotalCopySize(0)
- {}
- CClip::~CClip()
- {
- int count = m_Formats.GetCount();
- // in proper handling, m_Formats should be empty
- ASSERT( count == 0 );
- EmptyFormats();
- }
- void CClip::Clear()
- {
- m_ID = 0;
- m_Time = 0;
- m_Desc = "";
- m_lTotalCopySize = 0;
- EmptyFormats();
- }
- void CClip::EmptyFormats()
- {
- // free global memory in m_Formats
- for( int i = m_Formats.GetCount()-1; i >= 0; i-- )
- {
- m_Formats[i].Free();
- m_Formats.RemoveAt( i );
- }
- }
- // Adds a new Format to this Clip by copying the given data.
- bool CClip::AddFormat( CLIPFORMAT cfType, void* pData, UINT nLen )
- {
- ASSERT( pData && nLen );
- HGLOBAL hGlobal = ::NewGlobalP( pData, nLen );
- ASSERT( hGlobal );
- // update the Clip statistics
- m_Time = m_Time.GetCurrentTime();
- m_lTotalCopySize += nLen;
- if( !SetDescFromText( hGlobal ) )
- SetDescFromType();
- CClipFormat format(cfType,hGlobal);
- CClipFormat *pFormat;
- pFormat = m_Formats.FindFormat(cfType);
- // if the format type already exists as part of this clip, replace the data
- if( pFormat )
- {
- pFormat->Free();
- pFormat->m_hgData = format.m_hgData;
- }
- else
- m_Formats.Add(format);
- format.m_hgData = 0; // now owned by m_Formats
- return true;
- }
- #define EXIT_LoadFromClipboard(ret) { oleData.Release(); g_bCopyingClipboard = false; return(ret); }
- bool g_bCopyingClipboard = false; // for debugging reentrance
- // Fills this CClip with the contents of the clipboard.
- bool CClip::LoadFromClipboard( CClipTypes* pClipTypes )
- {
- COleDataObjectEx oleData;
- CClipTypes defaultTypes;
- CClipTypes* pTypes = pClipTypes;
- //ASSERT( !g_bCopyingClipboard );
- // For some reason, this can actually happen with *very* fast copies.
- // This is probably due to the OLE functions processing messages.
- // This *might* be able to be avoided by directly using the win32 Clipboard API
- // If this does happen, we just ignore the request by returning failure.
- if( g_bCopyingClipboard )
- return false;
- g_bCopyingClipboard = true;
- // m_Formats should be empty when this is called.
- ASSERT( m_Formats.GetCount() == 0 );
- //Attach to the clipboard
- if( !oleData.AttachClipboard() )
- {
- ASSERT(0); // does this ever happen?
- EXIT_LoadFromClipboard(false);
- }
- oleData.EnsureClipboardObject();
- // If the data is supposed to be private, then return
- if( oleData.IsDataAvailable( theApp.m_cfIgnoreClipboard ) )
- EXIT_LoadFromClipboard( false );
- // if no types were given, get only the first (most important) type.
- // (subsequent types could be synthetic due to automatic type conversions)
- if( pTypes == NULL || pTypes->GetCount() == 0 )
- {
- ASSERT(0); // this feature is not currently used... it is an error if it is.
- FORMATETC formatEtc;
- oleData.BeginEnumFormats();
- oleData.GetNextFormat(&formatEtc);
- defaultTypes.Add( formatEtc.cfFormat );
- pTypes = &defaultTypes;
- }
- // reset copy stats
- m_lTotalCopySize = 0;
- m_Desc = "[Ditto Error] !!BAD DESCRIPTION!!";
- // Get Description String
- // NOTE: We make sure that the description always corresponds to the
- // data saved by using the exact same globalmem instance as the source
- // for both... i.e. we only fetch the description format type once.
- CClipFormat cfDesc;
- bool bIsDescSet = false;
- cfDesc.m_cfType = CF_TEXT;
- if( oleData.IsDataAvailable( cfDesc.m_cfType ) )
- {
- cfDesc.m_hgData = oleData.GetGlobalData( cfDesc.m_cfType );
- bIsDescSet = SetDescFromText( cfDesc.m_hgData );
- }
- // Get global data for each supported type on the clipboard
- UINT nSize;
- CClipFormat cf;
- int numTypes = pTypes->GetCount();
- for(int i = 0; i < numTypes; i++)
- {
- cf.m_cfType = pTypes->GetAt(i);
- // is this the description we already fetched?
- if( cf.m_cfType == cfDesc.m_cfType )
- {
- cf = cfDesc;
- cfDesc.m_hgData = 0; // cf owns it now (to go into m_Formats)
- }
- else if( !oleData.IsDataAvailable(cf.m_cfType) )
- continue;
- else
- cf.m_hgData = oleData.GetGlobalData( cf.m_cfType );
- if( cf.m_hgData )
- {
- nSize = GlobalSize( cf.m_hgData );
- if( nSize > 0 )
- {
- m_Formats.Add( cf );
- m_lTotalCopySize += nSize;
- }
- else
- {
- ASSERT(FALSE); // a valid GlobalMem with 0 size is strange
- cf.Free();
- }
- cf.m_hgData = 0; // m_Formats owns it now
- }
- }
- m_Time = CTime::GetCurrentTime();
- if( !bIsDescSet )
- SetDescFromType();
- // if the description was in a type that is not supported,
- // we have to free it since it wasn't added to m_Formats
- if( cfDesc.m_hgData )
- cfDesc.Free();
- if( m_Formats.GetSize() == 0 )
- EXIT_LoadFromClipboard( false );
- EXIT_LoadFromClipboard( true );
- }
- bool CClip::SetDescFromText( HGLOBAL hgData )
- {
- if( hgData == 0 )
- return false;
- bool bRet = false;
- char* text = (char *) GlobalLock(hgData);
- ULONG ulBufLen = GlobalSize(hgData);
- ASSERT( text != NULL );
- if( ulBufLen > LENGTH_OF_TEXT_SNIPET )
- ulBufLen = LENGTH_OF_TEXT_SNIPET;
- if( ulBufLen > 0 )
- {
- char* buf = m_Desc.GetBuffer(ulBufLen);
- memcpy(buf, text, ulBufLen); // in most cases, last char == null
- buf[ulBufLen-1] = '\0'; // just in case not null terminated
- m_Desc.ReleaseBuffer(); // scans for the null
- bRet = m_Desc.GetLength() > 0;
- }
-
- //Unlock the data
- GlobalUnlock(hgData);
- return bRet;
- }
- bool CClip::SetDescFromType()
- {
- if( m_Formats.GetCount() <= 0 )
- return false;
- m_Desc = GetFormatName( m_Formats[0].m_cfType );
- return m_Desc.GetLength() > 0;
- }
- bool CClip::AddToDB()
- {
- bool bResult;
- try
- {
- CMainTable recset;
- if( FindDuplicate( recset, g_Opt.m_bAllowDuplicates ) )
- {
- m_ID = recset.m_lID;
- recset.Edit();
- recset.m_lDate = (long) m_Time.GetTime(); // update the copy Time
- recset.Update();
- recset.Close();
- EmptyFormats(); // delete this clip's data from memory.
- return true;
- }
- if( recset.IsOpen() )
- recset.Close();
- }
- CATCHDAO
- bResult = AddToMainTable() && AddToDataTable();
- // should be emptied by AddToDataTable
- ASSERT( m_Formats.GetCount() == 0 );
- return bResult;
- }
- // if a duplicate exists, set recset to the duplicate and return true
- bool CClip::FindDuplicate( CMainTable& recset, BOOL bCheckLastOnly )
- {
- ASSERT( m_lTotalCopySize > 0 );
- try
- {
- recset.m_strSort = "lDate DESC";
- if( bCheckLastOnly )
- {
- recset.Open("SELECT * FROM Main");
- recset.MoveFirst();
- // if an entry exists and they are the same size and the format data matches
- if( !recset.IsBOF() && !recset.IsEOF() &&
- m_lTotalCopySize == recset.m_lTotalCopySize &&
- (CompareFormatDataTo(recset.m_lID) == 0) )
- { return true; }
- return false;
- }
- // Look for any other entries that have the same size
- recset.Open("SELECT * FROM Main WHERE lTotalCopySize = %d", m_lTotalCopySize);
- while( !recset.IsEOF() )
- {
- //if there is any then look if it is an exact match
- if( CompareFormatDataTo(recset.m_lID) == 0 )
- return true;
- recset.MoveNext();
- }
- }
- CATCHDAO
- return false;
- }
- int CClip::CompareFormatDataTo( long lID )
- {
- int nRet = 0;
- int nRecs, nFormats;
- CClipFormat* pFormat = NULL;
- try
- {
- CDataTable recset;
- recset.Open("SELECT * FROM Data WHERE lParentID = %d", lID);
- // Verify the same number of saved types
- recset.MoveLast();
- nRecs = recset.GetRecordCount();
- nFormats = m_Formats.GetCount();
- nRet = nFormats - nRecs;
- if( nRet != 0 )
- { recset.Close(); return nRet; }
- // For each format type in the db
-
- for( recset.MoveFirst(); !recset.IsEOF(); recset.MoveNext() )
- {
- pFormat = m_Formats.FindFormat( GetFormatID(recset.m_strClipBoardFormat) );
- // Verify the format exists
- if( !pFormat )
- { recset.Close(); return -1; }
- // Compare the size
- nRet = ::GlobalSize(pFormat->m_hgData) - recset.m_ooData.m_dwDataLength;
- if( nRet != 0 )
- { recset.Close(); return nRet; }
- // Binary compare
- nRet = CompareGlobalHH( recset.m_ooData.m_hData,
- pFormat->m_hgData,
- recset.m_ooData.m_dwDataLength );
- if( nRet != 0 )
- { recset.Close(); return nRet; }
- }
- recset.Close();
- }
- CATCHDAO
- return 0;
- }
- // assigns m_ID
- bool CClip::AddToMainTable()
- {
- long lDate;
- try
- {
- CMainTable recset;
- // recset.m_strSort = "lDate DESC";
- recset.Open("SELECT * FROM Main");
- lDate = (long) m_Time.GetTime();
- recset.AddNew();
- recset.m_lDate = lDate;
- recset.m_strType = GetFormatName( m_Formats[0].m_cfType );
- recset.m_strText = m_Desc;
- recset.m_lTotalCopySize = m_lTotalCopySize;
- recset.Update();
- recset.MoveLast();
-
- m_ID = recset.m_lID;
- recset.Close();
- }
- catch(CDaoException* e)
- {
- ASSERT(FALSE);
- e->Delete();
- return false;
- }
- return true;
- }
- // Empties m_Formats as it saves them to the Data Table.
- bool CClip::AddToDataTable()
- {
- ASSERT( m_ID != 0 );
- try
- {
- CDataTable recset;
- CClipFormat* pCF;
- recset.Open(AFX_DAO_USE_DEFAULT_TYPE, "SELECT * FROM Data" ,NULL);
- for( int i = m_Formats.GetCount()-1; i >= 0 ; i-- )
- {
- pCF = & m_Formats.GetAt(i);
- recset.AddNew();
- recset.m_lParentID = m_ID;
- recset.m_strClipBoardFormat = GetFormatName( pCF->m_cfType );
- // the recset takes ownership of the HGLOBAL
- recset.ReplaceData( pCF->m_hgData, GlobalSize(pCF->m_hgData) );
- recset.Update();
- m_Formats.RemoveAt( i ); // the recset now owns the global
- }
- return true;
- }
- CATCHDAO
- return false;
- }
- // changes m_Time to be later than the latest clip entry in the db
- // ensures that pClip's time is not older than the last clip added
- // old times can happen on fast copies (<1 sec).
- void CClip::MakeLatestTime()
- {
- long lDate;
- try
- {
- CMainTable recset;
- recset.m_strSort = "lDate DESC";
- recset.Open("SELECT * FROM Main");
- recset.MoveFirst();
- lDate = (long) m_Time.GetTime();
- if( lDate <= recset.m_lDate )
- {
- lDate = recset.m_lDate + 1;
- m_Time = lDate;
- }
- recset.Close();
- }
- CATCHDAO
- }
- // STATICS
- // deletes from both Main and Data Tables
- BOOL CClip::Delete( int id )
- {
- CString csMainSQL;
- CString csDataSQL;
- BOOL bRet = FALSE;
- csMainSQL.Format( "DELETE FROM Main WHERE lID = %d", id );
- csDataSQL.Format( "DELETE FROM Data WHERE lParentID = %d", id );
- try
- {
- theApp.EnsureOpenDB();
- theApp.m_pDatabase->Execute(csMainSQL, dbFailOnError);
- theApp.m_pDatabase->Execute(csDataSQL, dbFailOnError);
- bRet = TRUE;
- }
- catch(CDaoException* e)
- {
- AfxMessageBox(e->m_pErrorInfo->m_strDescription);
- e->Delete();
- }
- return bRet;
- }
- BOOL CClip::Delete( ARRAY& IDs )
- {
- int count = IDs.GetSize();
- if(count <= 0)
- return FALSE;
- BOOL bRet = TRUE;
- // delete one at a time rather than all in one query
- // in order to avoid the "query too large" error for large deletes
- for( int i=0; i < count && bRet; i++ )
- bRet = bRet && Delete( IDs[i] );
- return bRet;
- }
- BOOL CClip::DeleteAll()
- {
- BOOL bRet = FALSE;
- try
- {
- theApp.EnsureOpenDB();
- theApp.m_pDatabase->Execute("DELETE * FROM Main", dbFailOnError);
- theApp.m_pDatabase->Execute("DELETE * FROM Data", dbFailOnError);
- bRet = TRUE;
- }
- catch(CDaoException* e)
- {
- AfxMessageBox(e->m_pErrorInfo->m_strDescription);
- e->Delete();
- }
- return bRet;
- }
- // Allocates a Global containing the requested Clip Format Data
- HGLOBAL CClip::LoadFormat( long lID, UINT cfType )
- {
- HGLOBAL hGlobal = 0;
- try
- {
- CDataTable recset;
- //Open the data table for all that have the parent id and format
- CString csSQL;
- csSQL.Format("SELECT * FROM Data WHERE lParentID = %d AND strClipBoardFormat = \'%s\'", lID, GetFormatName(cfType));
- recset.Open(AFX_DAO_USE_DEFAULT_TYPE, csSQL);
- if( !recset.IsBOF() && !recset.IsEOF() )
- {
- // create a new HGLOBAL duplicate
- hGlobal = NewGlobalH( recset.m_ooData.m_hData, recset.m_ooData.m_dwDataLength );
- // XOR take the recset's HGLOBAL... is this SAFE???
- // hGlobal = recset.TakeData();
- if( !hGlobal || ::GlobalSize(hGlobal) == 0 )
- {
- TRACE0( GetErrorString(::GetLastError()) );
- //::_RPT0( _CRT_WARN, GetErrorString(::GetLastError()) );
- ASSERT(FALSE);
- }
- }
- recset.Close();
- }
- CATCHDAO
- return hGlobal;
- }
- bool CClip::LoadFormats( long lID, CClipFormats& formats )
- {
- CClipFormat cf;
- HGLOBAL hGlobal = 0;
- formats.RemoveAll();
- try
- {
- CDataTable recset;
- //Open the data table for all that have the parent id
- CString csSQL;
- csSQL.Format("SELECT * FROM Data WHERE lParentID = %d", lID);
- recset.Open(AFX_DAO_USE_DEFAULT_TYPE, csSQL);
- while( !recset.IsEOF() )
- {
- // create a new HGLOBAL duplicate
- hGlobal = NewGlobalH( recset.m_ooData.m_hData, recset.m_ooData.m_dwDataLength );
- // XOR take the recset's HGLOBAL... is this SAFE???
- // hGlobal = recset.TakeData();
- if( !hGlobal || ::GlobalSize(hGlobal) == 0 )
- {
- TRACE0( GetErrorString(::GetLastError()) );
- //::_RPT0( _CRT_WARN, GetErrorString(::GetLastError()) );
- ASSERT(FALSE);
- }
- cf.m_cfType = GetFormatID( recset.m_strClipBoardFormat );
- cf.m_hgData = hGlobal;
- formats.Add( cf );
- recset.MoveNext();
- }
- cf.m_hgData = 0; // formats owns all the data
- recset.Close();
- }
- CATCHDAO
- return formats.GetCount() > 0;
- }
- void CClip::LoadTypes( long lID, CClipTypes& types )
- {
- types.RemoveAll();
- try
- {
- CDataTable recset;
- //Open the data table for all that have the parent id
- CString csSQL;
- csSQL.Format("SELECT * FROM Data WHERE lParentID = %d", lID );
- recset.Open(AFX_DAO_USE_DEFAULT_TYPE, csSQL);
- while( !recset.IsEOF() )
- {
- types.Add( GetFormatID( recset.m_strClipBoardFormat ) );
- recset.MoveNext();
- }
- recset.Close();
- }
- CATCHDAO
- }
- /*----------------------------------------------------------------------------*\
- CClipList
- \*----------------------------------------------------------------------------*/
- CClipList::~CClipList()
- {
- CClip* pClip;
- while( GetCount() )
- {
- pClip = RemoveHead();
- DELETE_PTR( pClip );
- }
- }
- // returns the number of clips actually saved
- // while this does empty the Format Data, it does not delete the Clips.
- int CClipList::AddToDB( bool bLatestTime, bool bShowStatus )
- {
- int savedCount = 0;
- int nRemaining = 0;
- CClip* pClip;
- POSITION pos;
- bool bResult;
- nRemaining = GetCount();
- pos = GetHeadPosition();
- while( pos )
- {
- if( bShowStatus )
- {
- theApp.SetStatus( StrF("%d",nRemaining) );
- nRemaining--;
- }
- pClip = GetNext( pos );
- ASSERT( pClip );
- if( bLatestTime )
- pClip->MakeLatestTime();
- bResult = pClip->AddToDB();
- if( bResult )
- savedCount++;
- }
- if( bShowStatus )
- theApp.SetStatus();
- return savedCount;
- }
- /*----------------------------------------------------------------------------*\
- CClipboardViewer
- \*----------------------------------------------------------------------------*/
- IMPLEMENT_DYNAMIC(CClipboardViewer, CWnd)
- BEGIN_MESSAGE_MAP(CClipboardViewer, CWnd)
- //{{AFX_MSG_MAP(CClipboardViewer)
- ON_WM_CREATE()
- ON_WM_CHANGECBCHAIN()
- ON_WM_DRAWCLIPBOARD()
- ON_WM_TIMER()
- //}}AFX_MSG_MAP
- ON_MESSAGE(WM_RECONNECT_TO_COPY_CHAIN, OnReconnectToCopyChain)
- ON_MESSAGE(WM_IS_TOP_VIEWER, OnGetIsTopViewer)
- ON_WM_DESTROY()
- END_MESSAGE_MAP()
- /////////////////////////////////////////////////////////////////////////////
- // CClipboardViewer construction/destruction
- CClipboardViewer::CClipboardViewer( CCopyThread* pHandler )
- {
- m_hNextClipboardViewer = 0;
- m_bCalling_SetClipboardViewer = false;
- m_lReconectCount = 0;
- m_bIsConnected = false;
- m_pHandler = pHandler;
- ASSERT(m_pHandler);
- }
- CClipboardViewer::~CClipboardViewer()
- {
- }
- void CClipboardViewer::Create()
- {
- CString strParentClass = AfxRegisterWndClass(0);
- CWnd::CreateEx(0, strParentClass, "Ditto Clipboard Viewer", 0, -1, -1, 0, 0, 0, 0);
- }
- // connects as a clipboard viewer
- void CClipboardViewer::Connect()
- {
- ASSERT( ::IsWindow(m_hWnd) );
- if( m_bIsConnected )
- return;
- //Set up the clip board viewer
- m_bCalling_SetClipboardViewer = true;
- m_hNextClipboardViewer = CWnd::SetClipboardViewer();
- m_bCalling_SetClipboardViewer = false;
- m_bIsConnected = true;
- }
- // disconnects as a clipboard viewer
- void CClipboardViewer::Disconnect()
- {
- if( !m_bIsConnected )
- return;
- ASSERT( ::IsWindow(m_hWnd) );
- CWnd::ChangeClipboardChain( m_hNextClipboardViewer );
- m_hNextClipboardViewer = 0;
- m_bIsConnected = false;
- }
- /////////////////////////////////////////////////////////////////////////////
- // CClipboardViewer message handlers
- int CClipboardViewer::OnCreate(LPCREATESTRUCT lpCreateStruct)
- {
- if (CWnd::OnCreate(lpCreateStruct) == -1)
- return -1;
- // is it important for us to be the top viewer?
- // SetTimer(TIMER_CHECK_TOP_LEVEL_VIEWER, ONE_MINUTE, 0);
- //Set up the clip board viewer
- Connect();
- return 0;
- }
- void CClipboardViewer::OnDestroy()
- {
- Disconnect();
- CWnd::OnDestroy();
- }
- void CClipboardViewer::OnChangeCbChain(HWND hWndRemove, HWND hWndAfter)
- {
- // If the next window is closing, repair the chain.
- if(m_hNextClipboardViewer == hWndRemove)
- {
- m_hNextClipboardViewer = hWndAfter;
- }
- // Otherwise, pass the message to the next link.
- else if (m_hNextClipboardViewer != NULL)
- {
- ::SendMessage ( m_hNextClipboardViewer, WM_CHANGECBCHAIN,
- (WPARAM) hWndRemove, (LPARAM) hWndAfter );
- }
- }
- //Message that the clipboard data has changed
- void CClipboardViewer::OnDrawClipboard()
- {
- // don't process the event when we first attach
- if( m_pHandler && !m_bCalling_SetClipboardViewer )
- m_pHandler->OnClipboardChange();
- // pass the event to the next Clipboard viewer in the chain
- if( m_hNextClipboardViewer != NULL )
- ::SendMessage(m_hNextClipboardViewer, WM_DRAWCLIPBOARD, 0, 0);
- }
- void CClipboardViewer::OnTimer(UINT nIDEvent)
- {
- switch(nIDEvent)
- {
- case TIMER_CHECK_TOP_LEVEL_VIEWER:
- {
- if( GetClipboardViewer() != this )
- {
- OnReconnectToCopyChain(0, 0);
- m_lReconectCount++;
- if(m_lReconectCount > 10)
- KillTimer(TIMER_CHECK_TOP_LEVEL_VIEWER);
- }
- break;
- }
- }
- CWnd::OnTimer(nIDEvent);
- }
- LRESULT CClipboardViewer::OnReconnectToCopyChain(WPARAM wParam, LPARAM lParam)
- {
- if( GetClipboardViewer() != this )
- {
- Disconnect();
- Connect();
- return TRUE;
- }
- return FALSE;
- }
- LRESULT CClipboardViewer::OnGetIsTopViewer(WPARAM wParam, LPARAM lParam)
- {
- return (GetClipboardViewer() == this);
- }
- /*----------------------------------------------------------------------------*\
- CCopyConfig
- \*----------------------------------------------------------------------------*/
- /*----------------------------------------------------------------------------*\
- CCopyThread
- \*----------------------------------------------------------------------------*/
- IMPLEMENT_DYNCREATE(CCopyThread, CWinThread)
- CCopyThread::CCopyThread()
- {
- m_bQuit = false;
- m_bAutoDelete = false;
- m_bConfigChanged = false;
- m_pClips = new CClipList;
- m_pClipboardViewer = new CClipboardViewer(this);
- ::InitializeCriticalSection(&m_CS);
- }
- CCopyThread::~CCopyThread()
- {
- m_LocalConfig.DeleteTypes();
- m_SharedConfig.DeleteTypes();
- DELETE_PTR( m_pClipboardViewer );
- if( m_pClips )
- ASSERT( m_pClips->GetCount() == 0 );
- DELETE_PTR( m_pClips );
- ::DeleteCriticalSection(&m_CS);
- }
- // perform and per-thread initialization here
- BOOL CCopyThread::InitInstance()
- {
- ASSERT( ::GetCurrentThreadId() == m_nThreadID );
- SetThreadName(m_nThreadID, "COPY");
- // the window is created within this thread and therefore uses its message queue
- m_pClipboardViewer->Create();
- return TRUE;
- }
- // perform any per-thread cleanup here
- int CCopyThread::ExitInstance()
- {
- ASSERT( m_bQuit ); // make sure we intended to quit
- m_pClipboardViewer->Disconnect();
- return CWinThread::ExitInstance();
- }
- // Called within Copy Thread:
- void CCopyThread::OnClipboardChange()
- {
- SyncConfig(); // synchronize with the main thread's copy configuration
- // if we are told not to copy on change, then we have nothing to do.
- if( !m_LocalConfig.m_bCopyOnChange )
- return;
- CClip* pClip = new CClip;
- bool bResult = pClip->LoadFromClipboard( m_LocalConfig.m_pSupportedTypes );
- if( !bResult )
- {
- delete pClip;
- return; // error
- }
- AddToClips( pClip );
- if( m_LocalConfig.m_bAsyncCopy )
- ::PostMessage(m_LocalConfig.m_hClipHandler, WM_CLIPBOARD_COPIED, (WPARAM)pClip, 0);
- else
- ::SendMessage(m_LocalConfig.m_hClipHandler, WM_CLIPBOARD_COPIED, (WPARAM)pClip, 0);
- }
- void CCopyThread::SyncConfig()
- {
- // atomic read
- if( m_bConfigChanged )
- {
- CClipTypes* pTypes = NULL;
- Hold();
- pTypes = m_LocalConfig.m_pSupportedTypes;
- m_LocalConfig = m_SharedConfig;
- // NULL means that it shouldn't have been sync'ed
- if( m_SharedConfig.m_pSupportedTypes == NULL )
- { // let m_LocalConfig keep its types
- m_LocalConfig.m_pSupportedTypes = pTypes; // undo sync
- pTypes = NULL; // nothing to delete
- }
- else
- m_SharedConfig.m_pSupportedTypes = NULL; // now owned by LocalConfig
- Release();
- // delete old types
- if( pTypes )
- delete pTypes;
- }
- }
- void CCopyThread::AddToClips( CClip* pClip )
- {
- Hold();
- if( !m_pClips )
- m_pClips = new CClipList;
- m_pClips->AddTail( pClip ); // m_pClips now owns pClip
- Release();
- }
- // Shared (use thread-safe access functions below)
- // Called within Main thread:
- CClipList* CCopyThread::GetClips()
- {
- CClipList* pRet;
- CClipList* pClips = new CClipList;
- Hold();
- pRet = m_pClips;
- m_pClips = pClips;
- Release();
- return pRet;
- }
- void CCopyThread::SetSupportedTypes( CClipTypes* pTypes )
- {
- CClipTypes* pTemp;
- Hold();
- pTemp = m_SharedConfig.m_pSupportedTypes;
- m_SharedConfig.m_pSupportedTypes = pTypes;
- m_bConfigChanged = true;
- Release();
- if( pTemp )
- delete pTemp;
- }
- HWND CCopyThread::SetClipHandler( HWND hWnd )
- {
- HWND hRet;
- Hold();
- hRet = m_SharedConfig.m_hClipHandler;
- m_SharedConfig.m_hClipHandler = hWnd;
- m_bConfigChanged = (hRet != hWnd);
- Release();
- return hRet;
- }
- HWND CCopyThread::GetClipHandler()
- {
- HWND hRet;
- Hold();
- hRet = m_SharedConfig.m_hClipHandler;
- Release();
- return hRet;
- }
- bool CCopyThread::SetCopyOnChange( bool bVal )
- {
- bool bRet;
- Hold();
- bRet = m_SharedConfig.m_bCopyOnChange;
- m_SharedConfig.m_bCopyOnChange = bVal;
- m_bConfigChanged = (bRet != bVal);
- Release();
- return bRet;
- }
- bool CCopyThread::GetCopyOnChange()
- {
- bool bRet;
- Hold();
- bRet = m_SharedConfig.m_bCopyOnChange;
- Release();
- return bRet;
- }
- bool CCopyThread::SetAsyncCopy( bool bVal )
- {
- bool bRet;
- Hold();
- bRet = m_SharedConfig.m_bAsyncCopy;
- m_SharedConfig.m_bAsyncCopy = bVal;
- m_bConfigChanged = (bRet != bVal);
- Release();
- return bRet;
- }
- bool CCopyThread::GetAsyncCopy()
- {
- bool bRet;
- Hold();
- bRet = m_SharedConfig.m_bAsyncCopy;
- Release();
- return bRet;
- }
- void CCopyThread::Init( CCopyConfig cfg )
- {
- ASSERT( m_LocalConfig.m_pSupportedTypes == NULL );
- m_LocalConfig = m_SharedConfig = cfg;
- // let m_LocalConfig own the m_pSupportedTypes
- m_SharedConfig.m_pSupportedTypes = NULL;
- }
- bool CCopyThread::Quit()
- {
- m_bQuit = true;
- m_pClipboardViewer->PostMessage( WM_QUIT );
- return CWinThread::PostThreadMessage( WM_QUIT, NULL, NULL ) != FALSE;
- }
- BEGIN_MESSAGE_MAP(CCopyThread, CWinThread)
- END_MESSAGE_MAP()
- // CCopyThread message handlers
- /*----------------------------------------------------------------------------*\
- CProcessCopy
- \*----------------------------------------------------------------------------*/
- CProcessCopy::CProcessCopy()
- {
- }
- CProcessCopy::~CProcessCopy()
- {
- }
|