开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 行业应用 > 金融证券系统 > 查看源码
QSProfile.cpp
资源名称：clking_src.rar　[点击查看]
上传用户：zhanglf88
上传日期：2013-11-19
资源大小：6036k
文件大小：18k
源码类别：

金融证券系统

开发平台：

Visual C++

QSProfile.cpp：源码内容
  1. /*
  2. All rights reserved.
  3. */
  5. #include "stdafx.h"
  6. #include "QSProfile.h"
  7. #include "ProxySocket.h"
  9. #ifdef _DEBUG
  10. #undef THIS_FILE
  11. static char THIS_FILE[]=__FILE__;
  12. #define new DEBUG_NEW
  13. #endif
  16. //////////////////////////////////////////////////////////////////////
  17. // CQSProfile
  19. CQSProfile & AfxGetQSProfile()
  20. {
  21. static CQSProfile s_qsprofile;
  22. return s_qsprofile;
  23. }
  25. //////////////////////////////////////////////////////////////////////
  26. // Construction/Destruction
  27. //////////////////////////////////////////////////////////////////////
  29. CQSProfile::CQSProfile()
  30. {
  31. m_nAccessType = 0; // INTERNET_OPEN_TYPE_PRECONFIG
  32. m_nProxyType = CProxySocket::TypeHTTP;
  33. m_nProxyPort = 80; // INTERNET_DEFAULT_HTTP_PORT
  35. m_nCurrentServer = -1;
  36. m_bCycleConnect = TRUE;
  37. }
  39. CQSProfile::~CQSProfile()
  40. {
  41. }
  43. char szQuoteServerProfileName[] = "QuoteServer.ini";
  45. char szProfile_Proxy[] = "Proxy";
  46. char szProfile_AccessType[] = "AccessType";
  47. char szProfile_ProxyType[] = "ProxyType";
  48. char szProfile_ProxyAddress[] = "ProxyAddress";
  49. char szProfile_ProxyPort[] = "ProxyPort";
  50. char szProfile_ProxyUser[] = "ProxyUser";
  51. char szProfile_ProxyPasswd[] = "ProxyPasswd";
  53. char szProfile_Server[] = "Server";
  54. char szProfile_ServerNumber[] = "ServerNumber";
  55. char szProfile_CurrentServer[] = "CurrentServer";
  56. char szProfile_CycleConnect[] = "CycleConnect";
  58. char szProfile_ServerFmt[] = "-Server%d";
  59. char szProfile_Name[] = "Name";
  60. char szProfile_Address[] = "Address";
  61. char szProfile_Port[] = "Port";
  62. char szProfile_User[] = "User";
  63. char szProfile_Passwd[] = "Passwd";
  64. char szProfile_SavePasswd[] = "SavePasswd";
  66. void CQSProfile::LoadProfile()
  67. {
  68. CString sFileName = GetWorkDirectory() + szQuoteServerProfileName;
  69. m_nCurrentServer = ::GetPrivateProfileInt( szProfile_Server, szProfile_CurrentServer, 0, sFileName );
  70. m_bCycleConnect = ::GetPrivateProfileInt( szProfile_Server, szProfile_CycleConnect, TRUE, sFileName );
  71. int nServerNumber = ::GetPrivateProfileInt( szProfile_Server, szProfile_ServerNumber, 0, sFileName );
  73. m_aQuoteServers.SetSize( 0, nServerNumber + 3 );
  74. char buffer[1024];
  75. for( int k=0; k<nServerNumber; k++ )
  76. {
  77. CString sSession;
  78. sSession.Format( szProfile_ServerFmt, k );
  80. CQuoteServer qs;
  82. memset( buffer, 0, sizeof(buffer) );
  83. ::GetPrivateProfileString( sSession, szProfile_Name, "", buffer, sizeof(buffer)-2, sFileName );
  84. qs.m_strName = buffer;
  86. memset( buffer, 0, sizeof(buffer) );
  87. ::GetPrivateProfileString( sSession, szProfile_Address, "", buffer, sizeof(buffer)-2, sFileName );
  88. qs.m_strAddress = buffer;
  90. qs.m_nPort = ::GetPrivateProfileInt( sSession, szProfile_Port, 8001, sFileName );
  92. memset( buffer, 0, sizeof(buffer) );
  93. ::GetPrivateProfileString( sSession, szProfile_User, "", buffer, sizeof(buffer)-2, sFileName );
  94. qs.m_strUser = buffer;
  96. memset( buffer, 0, sizeof(buffer) );
  97. ::GetPrivateProfileString( sSession, szProfile_Passwd, "", buffer, sizeof(buffer)-2, sFileName );
  98. qs.m_strPasswd = buffer;
  100. qs.m_bSavePasswd = ::GetPrivateProfileInt( sSession, szProfile_SavePasswd, TRUE, sFileName );
  102. if( qs.m_strName.IsEmpty() )
  103. qs.m_strName = qs.m_strAddress;
  104. if( !qs.m_strName.IsEmpty() )
  105. m_aQuoteServers.Add( qs );
  106. }
  108. LoadAliases();
  109. }
  111. void CQSProfile::StoreProfile()
  112. {
  113. CString sFileName = GetWorkDirectory() + szQuoteServerProfileName;
  114. CString sTemp;
  115. sTemp.Format( "%d", m_aQuoteServers.GetSize() );
  116. ::WritePrivateProfileString( szProfile_Server, szProfile_ServerNumber, sTemp, sFileName );
  117. sTemp.Format( "%d", m_nCurrentServer );
  118. ::WritePrivateProfileString( szProfile_Server, szProfile_CurrentServer, sTemp, sFileName );
  119. sTemp.Format( "%d", m_bCycleConnect );
  120. ::WritePrivateProfileString( szProfile_Server, szProfile_CycleConnect, sTemp, sFileName );
  121. ::GetPrivateProfileInt( szProfile_Server, szProfile_ServerNumber, m_aQuoteServers.GetSize(), sFileName );
  123. for( int k=0; k<m_aQuoteServers.GetSize(); k++ )
  124. {
  125. CQuoteServer & qs = m_aQuoteServers.ElementAt(k);
  127. CString sSession;
  128. sSession.Format( szProfile_ServerFmt, k );
  130. ::WritePrivateProfileString( sSession, szProfile_Name, qs.m_strName, sFileName );
  131. ::WritePrivateProfileString( sSession, szProfile_Address, qs.m_strAddress, sFileName );
  132. sTemp.Format( "%d", qs.m_nPort );
  133. ::WritePrivateProfileString( sSession, szProfile_Port, sTemp, sFileName );
  134. ::WritePrivateProfileString( sSession, szProfile_User, qs.m_strUser, sFileName );
  135. ::WritePrivateProfileString( sSession, szProfile_Passwd, qs.m_strPasswd, sFileName );
  136. sTemp.Format( "%d", qs.m_bSavePasswd );
  137. ::WritePrivateProfileString( sSession, szProfile_SavePasswd, sTemp, sFileName );
  138. }
  139. }
  141. static HKEY QSP_GetSectionKey( HKEY hKey, LPCTSTR szRegKeyGroup, LPCTSTR szRegKeyCompany, LPCTSTR szRegKeyApp, LPCTSTR lpszVersion, LPCTSTR lpszSection )
  142. {
  143. HKEY hGroupKey = NULL;
  144. HKEY hCompanyKey = NULL;
  145. HKEY hAppKey = NULL;
  146. HKEY hVersionKey = NULL;
  147. HKEY hSectionKey = NULL;
  148. if (RegOpenKeyEx( hKey, szRegKeyGroup, 0, KEY_WRITE|KEY_READ,
  149. &hGroupKey) == ERROR_SUCCESS)
  150. {
  151. DWORD dw;
  152. if (RegCreateKeyEx(hGroupKey, szRegKeyCompany, 0, REG_NONE,
  153. REG_OPTION_NON_VOLATILE, KEY_WRITE|KEY_READ, NULL,
  154. &hCompanyKey, &dw) == ERROR_SUCCESS )
  155. {
  156. RegCreateKeyEx(hCompanyKey, szRegKeyApp, 0, REG_NONE,
  157. REG_OPTION_NON_VOLATILE, KEY_WRITE|KEY_READ, NULL,
  158. &hAppKey, &dw);
  159. }
  160. }
  161. if (hGroupKey != NULL)
  162. RegCloseKey(hGroupKey);
  163. if (hCompanyKey != NULL)
  164. RegCloseKey(hCompanyKey);
  166. DWORD dw;
  167. if( NULL == hAppKey )
  168. return FALSE;
  169. RegCreateKeyEx(hAppKey, lpszVersion, 0, REG_NONE,
  170. REG_OPTION_NON_VOLATILE, KEY_WRITE|KEY_READ, NULL,
  171. &hVersionKey, &dw);
  172. RegCloseKey(hAppKey);
  174. if( NULL == hVersionKey )
  175. return FALSE;
  176. RegCreateKeyEx(hVersionKey, lpszSection, 0, REG_NONE,
  177. REG_OPTION_NON_VOLATILE, KEY_WRITE|KEY_READ, NULL,
  178. &hSectionKey, &dw);
  179. RegCloseKey(hVersionKey);
  181. return hSectionKey;
  182. }
  184. static DWORD QSP_RegQueryDWORD(HKEY hKey, LPCTSTR szRegKeyGroup, LPCTSTR szRegKeyCompany, LPCTSTR szRegKeyApp, LPCTSTR lpszVersion, LPCTSTR lpszSection, LPCTSTR lpszEntry, DWORD dwDefault )
  185. // hKey = HKEY_LOCAL_MACHINE, HKEY_CLASSES_ROOT, or so
  186. {
  187. HKEY hSecKey = QSP_GetSectionKey(hKey, szRegKeyGroup, szRegKeyCompany, szRegKeyApp, lpszVersion, lpszSection);
  188. if (hSecKey == NULL)
  189. return dwDefault;
  190. DWORD dwValue;
  191. DWORD dwType, dwCount = 4;
  192. LONG lResult = RegQueryValueEx(hSecKey, (LPTSTR)lpszEntry, NULL, &dwType,
  193. (LPBYTE)&dwValue, &dwCount);
  194. RegCloseKey(hSecKey);
  195. if (lResult == ERROR_SUCCESS)
  196. {
  197. ASSERT(dwType == REG_DWORD);
  198. return dwValue;
  199. }
  200. return dwDefault;
  201. }
  203. static CString QSP_RegQueryString(HKEY hKey, LPCTSTR szRegKeyGroup, LPCTSTR szRegKeyCompany, LPCTSTR szRegKeyApp, LPCTSTR lpszVersion, LPCTSTR lpszSection, LPCTSTR lpszEntry, LPCTSTR lpszDefault )
  204. // hKey = HKEY_LOCAL_MACHINE, HKEY_CLASSES_ROOT, or so
  205. {
  206. HKEY hSecKey = QSP_GetSectionKey(hKey, szRegKeyGroup, szRegKeyCompany, szRegKeyApp, lpszVersion, lpszSection);
  207. if (hSecKey == NULL)
  208. return lpszDefault;
  209. CString strValue;
  210. DWORD dwType, dwCount;
  211. LONG lResult = RegQueryValueEx(hSecKey, (LPTSTR)lpszEntry, NULL, &dwType,
  212. NULL, &dwCount);
  213. if (lResult == ERROR_SUCCESS)
  214. {
  215. ASSERT(dwType == REG_SZ);
  216. lResult = RegQueryValueEx(hSecKey, (LPTSTR)lpszEntry, NULL, &dwType,
  217. (LPBYTE)strValue.GetBuffer(dwCount/sizeof(TCHAR)), &dwCount);
  218. strValue.ReleaseBuffer();
  219. }
  220. RegCloseKey(hSecKey);
  221. if (lResult == ERROR_SUCCESS)
  222. {
  223. ASSERT(dwType == REG_SZ);
  224. return strValue;
  225. }
  226. return lpszDefault;
  227. }
  229. static BOOL QSP_ParseProxyString( LPCTSTR lpszProxy, int & nProxyType, CString & rProxyAddress, UINT & nProxyPort )
  230. {
  231. CString strProxy = lpszProxy;
  232. int nIndex1, nIndex2, nIndex3;
  234. char szSocks[] = "socks=";
  235. nIndex1 = strProxy.Find( szSocks );
  236. if( -1 != nIndex1 )
  237. {
  238. nIndex1 = nIndex1 + strlen(szSocks);
  239. nIndex2 = strProxy.Find( ":", nIndex1 );
  240. if( -1 != nIndex2 )
  241. {
  242. nIndex3 = strProxy.Find( ";", nIndex2 );
  243. if( -1 == nIndex3 ) nIndex3 = strProxy.GetLength();
  244. nProxyType = CProxySocket::TypeSocks5;
  245. rProxyAddress = strProxy.Mid( nIndex1, nIndex2-nIndex1 );
  246. nProxyPort = atol(strProxy.Mid( nIndex2+1, nIndex3-nIndex2-1) );
  247. return TRUE;
  248. }
  249. }
  251. char szHttp[] = "http=";
  252. nIndex1 = strProxy.Find( szHttp );
  253. if( -1 != nIndex1 )
  254. {
  255. nIndex1 = nIndex1 + strlen(szHttp);
  256. nIndex2 = strProxy.Find( ":", nIndex1 );
  257. if( -1 != nIndex2 )
  258. {
  259. nIndex3 = strProxy.Find( ";", nIndex2 );
  260. if( -1 == nIndex3 ) nIndex3 = strProxy.GetLength();
  261. nProxyType = CProxySocket::TypeHTTP;
  262. rProxyAddress = strProxy.Mid( nIndex1, nIndex2-nIndex1 );
  263. nProxyPort = atol(strProxy.Mid( nIndex2+1, nIndex3-nIndex2-1) );
  264. return TRUE;
  265. }
  266. }
  268. // All is the same
  269. nIndex1 = 0;
  270. nIndex2 = strProxy.Find( ":", nIndex1 );
  271. if( -1 != nIndex2 )
  272. {
  273. nIndex3 = strProxy.Find( ";", nIndex2 );
  274. if( -1 == nIndex3 ) nIndex3 = strProxy.GetLength();
  275. nProxyType = CProxySocket::TypeSocks5;
  276. rProxyAddress = strProxy.Mid( nIndex1, nIndex2-nIndex1 );
  277. nProxyPort = atol(strProxy.Mid( nIndex2+1, nIndex3-nIndex2-1) );
  278. return TRUE;
  279. }
  281. return FALSE;
  282. }
  284. BOOL CQSProfile::GetCompoundConfig( int & nProxyType, CString & strProxyAddress, UINT & nProxyPort )
  285. {
  286. int nAccessType = GetAccessType();
  287. nProxyType = GetProxyType();
  288. strProxyAddress = GetProxyAddress();
  289. nProxyPort = GetProxyPort();
  290. if( INTERNET_OPEN_TYPE_PRECONFIG == nAccessType )
  291. {
  292. DWORD dwProxyEnable = QSP_RegQueryDWORD( HKEY_CURRENT_USER, _T("Software"), _T("Microsoft"), _T("Windows"),
  293. _T("CurrentVersion"), _T("Internet Settings"), _T("ProxyEnable"), 0 );
  294. if( dwProxyEnable )
  295. {
  296. CString strProxy = QSP_RegQueryString( HKEY_CURRENT_USER, _T("Software"), _T("Microsoft"), _T("Windows"),
  297. _T("CurrentVersion"), _T("Internet Settings"), _T("ProxyServer"), "" );
  298. QSP_ParseProxyString( strProxy, nProxyType, strProxyAddress, nProxyPort );
  299. }
  300. else
  301. nProxyType = CProxySocket::TypeNone;
  302. }
  303. else if( INTERNET_OPEN_TYPE_DIRECT == nAccessType )
  304. nProxyType = CProxySocket::TypeNone;
  305. return TRUE;
  306. }
  308. int CQSProfile::GetAccessType( )
  309. {
  310. /* if( INTERNET_OPEN_TYPE_DIRECT != m_nAccessType
  311. && INTERNET_OPEN_TYPE_PRECONFIG != m_nAccessType
  312. && INTERNET_OPEN_TYPE_PROXY != m_nAccessType )
  313. return INTERNET_OPEN_TYPE_DIRECT;
  314. */
  315. CString sFileName = GetWorkDirectory() + szQuoteServerProfileName;
  316. m_nAccessType = ::GetPrivateProfileInt( szProfile_Proxy, szProfile_AccessType, 0, sFileName );
  317. return m_nAccessType;
  318. }
  320. void CQSProfile::SetAccessType( int nAccessType )
  321. {
  322. m_nAccessType = nAccessType;
  324. CString sFileName = GetWorkDirectory() + szQuoteServerProfileName;
  325. CString sTemp;
  326. sTemp.Format( "%d", m_nAccessType );
  327. ::WritePrivateProfileString( szProfile_Proxy, szProfile_AccessType, sTemp, sFileName );
  328. }
  330. int CQSProfile::GetProxyType( )
  331. {
  332. CString sFileName = GetWorkDirectory() + szQuoteServerProfileName;
  333. m_nProxyType = ::GetPrivateProfileInt( szProfile_Proxy, szProfile_ProxyType, 3, sFileName );
  334. return m_nProxyType;
  335. }
  337. void CQSProfile::SetProxyType( int nProxyType )
  338. {
  339. m_nProxyType = nProxyType;
  341. CString sFileName = GetWorkDirectory() + szQuoteServerProfileName;
  342. CString sTemp;
  343. sTemp.Format( "%d", m_nProxyType );
  344. ::WritePrivateProfileString( szProfile_Proxy, szProfile_ProxyType, sTemp, sFileName );
  345. }
  347. CString CQSProfile::GetProxyAddress( )
  348. {
  349. CString sFileName = GetWorkDirectory() + szQuoteServerProfileName;
  350. char szAddress[CProxySocket::MaxLenAddress+2];
  351. memset( szAddress, 0, sizeof(szAddress) );
  352. ::GetPrivateProfileString( szProfile_Proxy, szProfile_ProxyAddress, "", szAddress, sizeof(szAddress)-2, sFileName );
  353. m_strProxyAddress = szAddress;
  354. return m_strProxyAddress;
  355. }
  357. void CQSProfile::SetProxyAddress( LPCTSTR lpszProxyAddress )
  358. {
  359. m_strProxyAddress = lpszProxyAddress;
  361. CString sFileName = GetWorkDirectory() + szQuoteServerProfileName;
  362. ::WritePrivateProfileString( szProfile_Proxy, szProfile_ProxyAddress, m_strProxyAddress, sFileName );
  363. }
  365. WORD CQSProfile::GetProxyPort( )
  366. {
  367. CString sFileName = GetWorkDirectory() + szQuoteServerProfileName;
  368. m_nProxyPort = ::GetPrivateProfileInt( szProfile_Proxy, szProfile_ProxyPort, 80, sFileName );
  369. return m_nProxyPort;
  370. }
  372. void CQSProfile::SetProxyPort( WORD nPort )
  373. {
  374. m_nProxyPort = nPort;
  376. CString sFileName = GetWorkDirectory() + szQuoteServerProfileName;
  377. CString sTemp;
  378. sTemp.Format( "%d", m_nProxyPort );
  379. ::WritePrivateProfileString( szProfile_Proxy, szProfile_ProxyPort, sTemp, sFileName );
  380. }
  382. CString CQSProfile::GetProxyUser( )
  383. {
  384. CString sFileName = GetWorkDirectory() + szQuoteServerProfileName;
  385. char szUser[CProxySocket::MaxLenUser+2];
  386. memset( szUser, 0, sizeof(szUser) );
  387. ::GetPrivateProfileString( szProfile_Proxy, szProfile_ProxyUser, "", szUser, sizeof(szUser)-2, sFileName );
  388. m_strProxyUser = szUser;
  389. return m_strProxyUser;
  390. }
  392. void CQSProfile::SetProxyUser( LPCTSTR lpszProxyUser )
  393. {
  394. m_strProxyUser = lpszProxyUser;
  396. CString sFileName = GetWorkDirectory() + szQuoteServerProfileName;
  397. ::WritePrivateProfileString( szProfile_Proxy, szProfile_ProxyUser, m_strProxyUser, sFileName );
  398. }
  400. CString CQSProfile::GetProxyPasswd( )
  401. {
  402. CString sFileName = GetWorkDirectory() + szQuoteServerProfileName;
  403. char szPasswd[CProxySocket::MaxLenPasswd+2];
  404. memset( szPasswd, 0, sizeof(szPasswd) );
  405. ::GetPrivateProfileString( szProfile_Proxy, szProfile_ProxyPasswd, "", szPasswd, sizeof(szPasswd)-2, sFileName );
  406. m_strProxyPasswd = szPasswd;
  407. return m_strProxyPasswd;
  408. }
  410. void CQSProfile::SetProxyPasswd( LPCTSTR lpszProxyPasswd )
  411. {
  412. m_strProxyPasswd = lpszProxyPasswd;
  414. CString sFileName = GetWorkDirectory() + szQuoteServerProfileName;
  415. ::WritePrivateProfileString( szProfile_Proxy, szProfile_ProxyPasswd, m_strProxyPasswd, sFileName );
  416. }
  418. static CString QSP_AfxGetFileDirectory( CString strPathName, CString strDefault )
  419. {
  420. if( strPathName.IsEmpty() )
  421. return strDefault;
  423. CString strResult = strPathName;
  424. if( strResult.GetLength() > 0 )
  425. {
  426. int nIndex = strResult.ReverseFind( '\' );
  427. if( -1 == nIndex )
  428. nIndex = strResult.ReverseFind( '/' );
  429. if( -1 == nIndex )
  430. strResult = strDefault;
  431. else
  432. strResult = strResult.Left( nIndex );
  433. }
  435. if( strResult.GetLength() <= 0 )
  436. return strDefault;
  438. int nLen = strResult.GetLength();
  439. if( nLen > 0 && strResult[nLen-1] != '\' && strResult[nLen-1] != '/' )
  440. strResult += "\";
  442. return strResult;
  443. }
  445. CString CQSProfile::GetWorkDirectory( )
  446. {
  447. if( m_strWorkDirectory.IsEmpty() )
  448. {
  449. TCHAR szLongPathName[_MAX_PATH];
  450. ::GetModuleFileName( AfxGetModuleHandle(), szLongPathName, _MAX_PATH);
  451. m_strWorkDirectory = QSP_AfxGetFileDirectory( szLongPathName, "" );
  452. }
  453. return m_strWorkDirectory;
  454. }
  456. int CQSProfile::GetCurrentServerIndex()
  457. {
  458. if( m_nCurrentServer < 0 || m_nCurrentServer >= m_aQuoteServers.GetSize() )
  459. m_nCurrentServer = 0;
  460. if( 0 == m_aQuoteServers.GetSize() )
  461. m_nCurrentServer = -1;
  462. return m_nCurrentServer;
  463. }
  465. CQuoteServer CQSProfile::GetCurrentServer()
  466. {
  467. CQuoteServer qs;
  468. int nIndex = GetCurrentServerIndex();
  469. if( nIndex >= 0 && nIndex < m_aQuoteServers.GetSize() )
  470. qs = m_aQuoteServers.ElementAt(nIndex);
  471. return qs;
  472. }
  474. void CQSProfile::SetCurrentServer( int nServer )
  475. {
  476. m_nCurrentServer = nServer;
  477. }
  479. BOOL CQSProfile::GetCycleConnect()
  480. {
  481. return m_bCycleConnect;
  482. }
  484. void CQSProfile::SetCycleConnect( BOOL bCycleConnect )
  485. {
  486. m_bCycleConnect = bCycleConnect;
  487. }
  489. CQuoteServerArray & CQSProfile::GetQuoteServers( )
  490. {
  491. return m_aQuoteServers;
  492. }
  494. int CQSProfile::AddQuoteServers( CQuoteServerArray & aNew )
  495. {
  496. int nCount = 0;
  497. for( int i=0; i<aNew.GetSize(); i++ )
  498. {
  499. CQuoteServer & qs = aNew.ElementAt(i);
  500. for( int j=0; j<m_aQuoteServers.GetSize(); j++ )
  501. {
  502. if( 0 == qs.m_strName.CompareNoCase(m_aQuoteServers.ElementAt(j).m_strName)
  503. || 0 == qs.m_strAddress.CompareNoCase(m_aQuoteServers.ElementAt(j).m_strAddress) )
  504. {
  505. m_aQuoteServers.SetAt( j, qs );
  506. nCount ++;
  507. break;
  508. }
  509. }
  510. if( j == m_aQuoteServers.GetSize() )
  511. {
  512. m_aQuoteServers.Add( qs );
  513. nCount ++;
  514. }
  515. }
  516. return nCount;
  517. }
  519. void SerializeAlias( CArchive & ar, CAliasArray & aAliases )
  520. {
  521. char szFileMagic[] = "NETTSALIAS";
  522. DWORD dwFileVersion = 0x20040825;
  524. if( ar.IsStoring() )
  525. {
  526. ar.WriteString( szFileMagic );
  527. ar << dwFileVersion;
  529. ar << (int)aAliases.GetSize();
  530. for( int i=0; i<aAliases.GetSize(); i++ )
  531. {
  532. CAlias & a = aAliases.ElementAt(i);
  533. ar << a.m_strName;
  534. ar << a.m_strValue;
  535. }
  536. }
  537. else
  538. {
  539. TCHAR buffer[64];
  540. memset( buffer, 0, sizeof(buffer) );
  541. ar.ReadString( buffer, strlen(szFileMagic) );
  542. if( 0 != strncmp( szFileMagic, buffer, strlen(szFileMagic) ) )
  543. return;
  545. DWORD dwVer;
  546. ar >> dwVer;
  547. if( dwVer > dwFileVersion )
  548. return;
  550. int size;
  551. ar >> size;
  552. if( size < 0 || size > 10000 ) // too big
  553. return;
  555. aAliases.SetSize( 0, size+2 );
  556. for( int i=0; i<size; i++ )
  557. {
  558. CAlias a;
  559. ar >> a.m_strName;
  560. ar >> a.m_strValue;
  561. aAliases.Add( a );
  562. }
  563. }
  564. }
  566. CString GetNetTSAliasFileName( )
  567. {
  568. TCHAR buffer[MAX_PATH];
  569. memset( buffer, 0, sizeof(buffer) );
  570. GetWindowsDirectory(buffer, sizeof(buffer));
  571. buffer[sizeof(buffer)-1] = 0;
  572. CString string = buffer;
  573. int nLen = string.GetLength();
  574. if( nLen > 0 && '\' != string[nLen-1] )
  575. string += '\';
  576. return string + "NetTSAlias";
  577. }
  579. BOOL CQSProfile::StoreAliases( CAliasArray & aNew )
  580. {
  581. m_aAliases.SetSize( 0, aNew.GetSize() + 2 );
  582. for( int i=0; i<aNew.GetSize(); i++ )
  583. {
  584. CAlias & alias = aNew.ElementAt(i);
  585. m_aAliases.Add( alias );
  586. }
  588. CFile file;
  589. if( file.Open( GetNetTSAliasFileName(), CFile::modeWrite | CFile::modeCreate ) )
  590. {
  591. CArchive ar( &file, CArchive::store );
  592. SerializeAlias( ar, m_aAliases );
  593. ar.Close();
  594. file.Close();
  595. return TRUE;
  596. }
  597. return FALSE;
  598. }
  600. BOOL CQSProfile::LoadAliases( )
  601. {
  602. CFile file;
  603. if( file.Open( GetNetTSAliasFileName(), CFile::modeRead ) )
  604. {
  605. DWORD dwLen = (DWORD)file.GetLength();
  606. if( 0 == dwLen || dwLen > 102400 ) // too big
  607. {
  608. file.Close();
  609. return FALSE;
  610. }
  612. CArchive ar( &file, CArchive::load );
  613. SerializeAlias( ar, m_aAliases );
  614. ar.Close();
  615. file.Close();
  616. return TRUE;
  617. }
  618. return FALSE;
  619. }
  621. CString CQSProfile::GetAlias( LPCTSTR lpszName )
  622. {
  623. for( int i=0; i<m_aAliases.GetSize(); i++ )
  624. {
  625. CAlias & alias = m_aAliases.ElementAt(i);
  626. if( 0 == alias.m_strName.CompareNoCase(lpszName) )
  627. return alias.m_strValue;
  628. }
  629. return lpszName;
  630. }