DatabaseUtilities.cpp 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806
  1. // DatabaseUtilites.cpp: implementation of the CDatabaseUtilites class.
  2. //
  3. //////////////////////////////////////////////////////////////////////
  4. #include "stdafx.h"
  5. #include "CP_Main.h"
  6. #include "DatabaseUtilities.h"
  7. #include "ProcessPaste.h"
  8. #include <io.h>
  9. #include "AccessToSqlite.h"
  10. #include "Path.h"
  11. #include "InternetUpdate.h"
  12. //////////////////////////////////////////////////////////////////////
  13. // Construction/Destruction
  14. //////////////////////////////////////////////////////////////////////
  15. BOOL CreateBackup(CString csPath)
  16. {
  17. CString csOriginal;
  18. int count = 0;
  19. // create a backup of the existing database
  20. do
  21. {
  22. count++;
  23. csOriginal = csPath + StrF(_T(".%03d"), count);
  24. // in case of some weird infinite loop
  25. if( count > 50 )
  26. {
  27. ASSERT(0);
  28. return FALSE;
  29. }
  30. } while( !::CopyFile(csPath, csOriginal, TRUE));
  31. return TRUE;
  32. }
  33. CString GetDBName()
  34. {
  35. return CGetSetOptions::GetDBPath();
  36. }
  37. CString GetOLDDefaultDBName()
  38. {
  39. CString csDefaultPath;
  40. LPMALLOC pMalloc;
  41. if(SUCCEEDED(::SHGetMalloc(&pMalloc)))
  42. {
  43. LPITEMIDLIST pidlPrograms;
  44. SHGetSpecialFolderLocation(NULL, CSIDL_APPDATA, &pidlPrograms);
  45. TCHAR string[MAX_PATH];
  46. SHGetPathFromIDList(pidlPrograms, string);
  47. pMalloc->Free(pidlPrograms);
  48. pMalloc->Release();
  49. csDefaultPath = string;
  50. csDefaultPath += "\\Ditto\\";
  51. csDefaultPath += "DittoDB.mdb";
  52. }
  53. return csDefaultPath;
  54. }
  55. CString GetDefaultDBName()
  56. {
  57. CString csDefaultPath = _T("c:\\program files\\Ditto\\");
  58. if(g_Opt.m_bU3)
  59. {
  60. csDefaultPath = CGetSetOptions::GetPath(PATH_DATABASE);
  61. }
  62. else
  63. {
  64. //If portable then default to the running path
  65. if(CGetSetOptions::GetIsPortableDitto())
  66. {
  67. csDefaultPath.Empty();
  68. }
  69. else
  70. {
  71. LPMALLOC pMalloc;
  72. if(SUCCEEDED(::SHGetMalloc(&pMalloc)))
  73. {
  74. LPITEMIDLIST pidlPrograms;
  75. SHGetSpecialFolderLocation(NULL, CSIDL_APPDATA, &pidlPrograms);
  76. TCHAR string[MAX_PATH];
  77. SHGetPathFromIDList(pidlPrograms, string);
  78. pMalloc->Free(pidlPrograms);
  79. pMalloc->Release();
  80. csDefaultPath = string;
  81. }
  82. FIX_CSTRING_PATH(csDefaultPath);
  83. csDefaultPath += "Ditto\\";
  84. }
  85. }
  86. CString csTempName = csDefaultPath + "Ditto.db";
  87. int i = 1;
  88. while(FileExists(csTempName))
  89. {
  90. csTempName.Format(_T("%sDitto_%d.db"), csDefaultPath, i);
  91. i++;
  92. }
  93. csDefaultPath = csTempName;
  94. return csDefaultPath;
  95. }
  96. BOOL CheckDBExists(CString csDBPath)
  97. {
  98. //If this is the first time running this version then convert the old database to the new db
  99. if(csDBPath.IsEmpty() && g_Opt.m_bU3 == false)
  100. {
  101. csDBPath = GetDefaultDBName();
  102. if(FileExists(csDBPath) == FALSE && CGetSetOptions::GetIsPortableDitto() == FALSE)
  103. {
  104. CString csOldDB = CGetSetOptions::GetDBPathOld();
  105. if(csOldDB.IsEmpty())
  106. {
  107. csOldDB = GetOLDDefaultDBName();
  108. }
  109. if(FileExists(csOldDB))
  110. {
  111. //create the new sqlite db
  112. CreateDB(csDBPath);
  113. CAccessToSqlite Convert;
  114. Convert.ConvertDatabase(csDBPath, csOldDB);
  115. }
  116. }
  117. }
  118. BOOL bRet = FALSE;
  119. if(FileExists(csDBPath) == FALSE)
  120. {
  121. csDBPath = GetDefaultDBName();
  122. nsPath::CPath FullPath(csDBPath);
  123. CString csPath = FullPath.GetPath().GetStr();
  124. if(csPath.IsEmpty() == false && FileExists(csDBPath) == FALSE)
  125. {
  126. CreateDirectory(csPath, NULL);
  127. }
  128. // -- create a new one
  129. bRet = CreateDB(csDBPath);
  130. }
  131. else
  132. {
  133. if(ValidDB(csDBPath) == FALSE)
  134. {
  135. //Db existed but was bad
  136. CString csMarkAsBad;
  137. csMarkAsBad = csDBPath;
  138. csMarkAsBad.Replace(_T("."), _T("_BAD."));
  139. CString csPath = GetDefaultDBName();
  140. CString cs;
  141. cs.Format(_T("%s \"%s\",\n")
  142. _T("%s \"%s\",\n")
  143. _T("%s,\n")
  144. _T("\"%s\""),
  145. theApp.m_Language.GetString("Database_Format", "Unrecognized Database Format"),
  146. csDBPath,
  147. theApp.m_Language.GetString("File_Renamed", "the file will be renamed"),
  148. csMarkAsBad,
  149. theApp.m_Language.GetString("New_Database", "and a new database will be created"),
  150. csPath);
  151. AfxMessageBox(cs);
  152. CFile::Rename(csDBPath, csMarkAsBad);
  153. csDBPath = csPath;
  154. bRet = CreateDB(csDBPath);
  155. }
  156. else
  157. {
  158. bRet = TRUE;
  159. }
  160. }
  161. if(bRet)
  162. {
  163. bRet = OpenDatabase(csDBPath);
  164. }
  165. return bRet;
  166. }
  167. BOOL OpenDatabase(CString csDB)
  168. {
  169. try
  170. {
  171. theApp.m_db.close();
  172. theApp.m_db.open(csDB);
  173. CGetSetOptions::SetDBPath(csDB);
  174. return TRUE;
  175. }
  176. CATCH_SQLITE_EXCEPTION
  177. return FALSE;
  178. }
  179. BOOL ValidDB(CString csPath, BOOL bUpgrade)
  180. {
  181. try
  182. {
  183. BOOL didBackup = FALSE;
  184. CString backupFilePrefix = _T("Before_Update_To");
  185. CppSQLite3DB db;
  186. db.open(csPath);
  187. db.execQuery(_T("SELECT lID, lDate, mText, lShortCut, lDontAutoDelete, ")
  188. _T("CRC, bIsGroup, lParentID, QuickPasteText ")
  189. _T("FROM Main"));
  190. db.execQuery(_T("SELECT lID, lParentID, strClipBoardFormat, ooData FROM Data"));
  191. db.execQuery(_T("SELECT lID, TypeText FROM Types"));
  192. try
  193. {
  194. db.execDML(_T("DROP TRIGGER delete_data_trigger"));
  195. }
  196. catch(CppSQLite3Exception& e)
  197. {
  198. e.errorCode();
  199. }
  200. try
  201. {
  202. db.execDML(_T("DROP TRIGGER delete_copy_buffer_trigger"));
  203. }
  204. catch(CppSQLite3Exception& e)
  205. {
  206. e.errorCode();
  207. }
  208. //This was added later so try to add each time and catch the exception here
  209. try
  210. {
  211. db.execDML(_T("CREATE TRIGGER delete_data_trigger BEFORE DELETE ON Main FOR EACH ROW\n")
  212. _T("BEGIN\n")
  213. _T("INSERT INTO MainDeletes VALUES(old.lID, datetime('now'));\n")
  214. _T("END\n"));
  215. }
  216. catch(CppSQLite3Exception& e)
  217. {
  218. if(didBackup == FALSE)
  219. didBackup = BackupDB(csPath, backupFilePrefix);
  220. e.errorCode();
  221. }
  222. //This was added later so try to add each time and catch the exception here
  223. try
  224. {
  225. db.execQuery(_T("SELECT lID, lClipID, lCopyBuffer FROM CopyBuffers"));
  226. }
  227. catch(CppSQLite3Exception& e)
  228. {
  229. if(didBackup == FALSE)
  230. didBackup = BackupDB(csPath, backupFilePrefix);
  231. e.errorCode();
  232. db.execDML(_T("CREATE TABLE CopyBuffers(")
  233. _T("lID INTEGER PRIMARY KEY AUTOINCREMENT, ")
  234. _T("lClipID INTEGER,")
  235. _T("lCopyBuffer INTEGER)"));
  236. }
  237. //This was added later so try to add each time and catch the exception here
  238. try
  239. {
  240. db.execQuery(_T("SELECT clipId FROM MainDeletes"));
  241. }
  242. catch(CppSQLite3Exception& e)
  243. {
  244. if(didBackup == FALSE)
  245. didBackup = BackupDB(csPath, backupFilePrefix);
  246. e.errorCode();
  247. db.execDML(_T("CREATE TABLE MainDeletes(")
  248. _T("clipID INTEGER,")
  249. _T("modifiedDate)"));
  250. db.execDML(_T("CREATE TRIGGER MainDeletes_delete_data_trigger BEFORE DELETE ON MainDeletes FOR EACH ROW\n")
  251. _T("BEGIN\n")
  252. _T("DELETE FROM CopyBuffers WHERE lClipID = old.clipID;\n")
  253. _T("DELETE FROM Data WHERE lParentID = old.clipID;\n")
  254. _T("END\n"));
  255. }
  256. try
  257. {
  258. db.execDML(_T("CREATE INDEX Main_ParentId on Main(lParentID DESC)"));
  259. db.execDML(_T("CREATE INDEX Main_IsGroup on Main(bIsGroup DESC)"));
  260. db.execDML(_T("CREATE INDEX Main_ShortCut on Main(lShortCut DESC)"));
  261. }
  262. catch(CppSQLite3Exception& e)
  263. {
  264. e.errorCode();
  265. }
  266. try
  267. {
  268. db.execQuery(_T("SELECT clipOrder, clipGroupOrder FROM Main"));
  269. }
  270. catch(CppSQLite3Exception& e)
  271. {
  272. if(didBackup == FALSE)
  273. didBackup = BackupDB(csPath, backupFilePrefix);
  274. db.execDML(_T("ALTER TABLE Main ADD clipOrder REAL"));
  275. db.execDML(_T("ALTER TABLE Main ADD clipGroupOrder REAL"));
  276. db.execDML(_T("Update Main set clipOrder = lDate, clipGroupOrder = lDate"));
  277. db.execDML(_T("CREATE INDEX Main_ClipOrder on Main(clipOrder DESC)"));
  278. db.execDML(_T("CREATE INDEX Main_ClipGroupOrder on Main(clipGroupOrder DESC)"));
  279. db.execDML(_T("DROP INDEX Main_Date"));
  280. e.errorCode();
  281. }
  282. try
  283. {
  284. db.execQuery(_T("SELECT globalShortCut FROM Main"));
  285. }
  286. catch(CppSQLite3Exception& e)
  287. {
  288. if(didBackup == FALSE)
  289. didBackup = BackupDB(csPath, backupFilePrefix);
  290. db.execDML(_T("ALTER TABLE Main ADD globalShortCut INTEGER"));
  291. e.errorCode();
  292. }
  293. try
  294. {
  295. db.execQuery(_T("SELECT lastPasteDate FROM Main"));
  296. }
  297. catch(CppSQLite3Exception& e)
  298. {
  299. if(didBackup == FALSE)
  300. didBackup = BackupDB(csPath, backupFilePrefix);
  301. db.execDML(_T("ALTER TABLE Main ADD lastPasteDate INTEGER"));
  302. db.execDML(_T("Update Main set lastPasteDate = lDate"));
  303. db.execDMLEx(_T("Update Main set lastPasteDate = %d where lastPasteDate <= 0"), CTime::GetCurrentTime().GetTime());
  304. e.errorCode();
  305. }
  306. }
  307. CATCH_SQLITE_EXCEPTION_AND_RETURN(FALSE)
  308. return TRUE;
  309. }
  310. BOOL BackupDB(CString dbPath, CString prefix)
  311. {
  312. CString backup = GetFilePath(dbPath);
  313. CInternetUpdate update;
  314. long runningVersion = update.GetRunningVersion();
  315. CString versionString = update.GetVersionString(runningVersion);
  316. backup += GetFileName(dbPath) += _T("_") + prefix + _T("_") + versionString;
  317. backup.Replace(_T(".db"), _T(""));
  318. backup.Replace(_T("."), _T("_"));
  319. CString temp = backup;
  320. temp += _T(".db");
  321. int i = 1;
  322. while(FileExists(temp))
  323. {
  324. temp.Format(_T("%s_%d.db"), backup, i);
  325. i++;
  326. }
  327. backup = temp;
  328. BOOL ret = CopyFile(dbPath, backup, TRUE);
  329. return ret;
  330. }
  331. BOOL CreateDB(CString csFile)
  332. {
  333. try
  334. {
  335. CppSQLite3DB db;
  336. db.open(csFile);
  337. db.execDML(_T("PRAGMA auto_vacuum = 1"));
  338. db.execDML(_T("CREATE TABLE Main(")
  339. _T("lID INTEGER PRIMARY KEY AUTOINCREMENT, ")
  340. _T("lDate INTEGER, ")
  341. _T("mText TEXT, ")
  342. _T("lShortCut INTEGER, ")
  343. _T("lDontAutoDelete INTEGER, ")
  344. _T("CRC INTEGER, ")
  345. _T("bIsGroup INTEGER, ")
  346. _T("lParentID INTEGER, ")
  347. _T("QuickPasteText TEXT, ")
  348. _T("clipOrder REAL, ")
  349. _T("clipGroupOrder REAL, ")
  350. _T("globalShortCut INTEGER, ")
  351. _T("lastPasteDate INTEGER);"));
  352. db.execDML(_T("CREATE TABLE Data(")
  353. _T("lID INTEGER PRIMARY KEY AUTOINCREMENT, ")
  354. _T("lParentID INTEGER, ")
  355. _T("strClipBoardFormat TEXT, ")
  356. _T("ooData BLOB);"));
  357. db.execDML(_T("CREATE TABLE Types(")
  358. _T("lID INTEGER PRIMARY KEY AUTOINCREMENT, ")
  359. _T("TypeText TEXT);"));
  360. db.execDML(_T("CREATE UNIQUE INDEX Main_ID on Main(lID ASC)"));
  361. db.execDML(_T("CREATE UNIQUE INDEX Data_ID on Data(lID ASC)"));
  362. db.execDML(_T("CREATE INDEX Main_ClipOrder on Main(clipOrder DESC)"));
  363. db.execDML(_T("CREATE INDEX Main_ClipGroupOrder on Main(clipGroupOrder DESC)"));
  364. db.execDML(_T("CREATE INDEX Main_ParentId on Main(lParentID DESC)"));
  365. db.execDML(_T("CREATE INDEX Main_IsGroup on Main(bIsGroup DESC)"));
  366. db.execDML(_T("CREATE INDEX Main_ShortCut on Main(lShortCut DESC)"));
  367. db.execDML(_T("CREATE TRIGGER delete_data_trigger BEFORE DELETE ON Main FOR EACH ROW\n")
  368. _T("BEGIN\n")
  369. _T("INSERT INTO MainDeletes VALUES(old.lID, datetime('now'));\n")
  370. _T("END\n"));
  371. db.execDML(_T("CREATE TABLE CopyBuffers(")
  372. _T("lID INTEGER PRIMARY KEY AUTOINCREMENT, ")
  373. _T("lClipID INTEGER, ")
  374. _T("lCopyBuffer INTEGER)"));
  375. db.execDML(_T("CREATE TABLE MainDeletes(")
  376. _T("clipID INTEGER,")
  377. _T("modifiedDate)"));
  378. db.execDML(_T("CREATE TRIGGER MainDeletes_delete_data_trigger BEFORE DELETE ON MainDeletes FOR EACH ROW\n")
  379. _T("BEGIN\n")
  380. _T("DELETE FROM CopyBuffers WHERE lClipID = old.clipID;\n")
  381. _T("DELETE FROM Data WHERE lParentID = old.clipID;\n")
  382. _T("END\n"));
  383. db.close();
  384. }
  385. CATCH_SQLITE_EXCEPTION_AND_RETURN(FALSE)
  386. return TRUE;
  387. }
  388. BOOL CompactDatabase()
  389. {
  390. // if(!theApp.CloseDB())
  391. // return FALSE;
  392. //
  393. // CString csDBName = GetDBName();
  394. // CString csTempDBName = csDBName;
  395. // csTempDBName.Replace(".mdb", "TempDBName.mdb");
  396. //
  397. // //Compact the database
  398. // try
  399. // {
  400. // CDaoWorkspace::CompactDatabase(csDBName, csTempDBName);//, dbLangGeneral, 0, "andrew");//DATABASE_PASSWORD);
  401. // }
  402. // catch(CDaoException* e)
  403. // {
  404. // AfxMessageBox(e->m_pErrorInfo->m_strDescription);
  405. // DeleteFile(csTempDBName);
  406. // e->Delete();
  407. // return FALSE;
  408. // }
  409. // catch(CMemoryException* e)
  410. // {
  411. // AfxMessageBox("Memory Exception");
  412. // DeleteFile(csTempDBName);
  413. // e->Delete();
  414. // return FALSE;
  415. // }
  416. //
  417. // //Since compacting the database creates a new db delete the old one and replace it
  418. // //with the compacted db
  419. // if(DeleteFile(csDBName))
  420. // {
  421. // try
  422. // {
  423. // CFile::Rename(csTempDBName, csDBName);
  424. // }
  425. // catch(CFileException *e)
  426. // {
  427. // e->ReportError();
  428. // e->Delete();
  429. // return FALSE;
  430. // }
  431. // }
  432. // else
  433. // AfxMessageBox("Error Compacting Database");
  434. return TRUE;
  435. }
  436. BOOL RepairDatabase()
  437. {
  438. // if(!theApp.CloseDB())
  439. // return FALSE;
  440. // try
  441. // {
  442. // CDaoWorkspace::RepairDatabase(GetDBName());
  443. // }
  444. // catch(CDaoException *e)
  445. // {
  446. // AfxMessageBox(e->m_pErrorInfo->m_strDescription);
  447. // e->Delete();
  448. // return FALSE;
  449. // }
  450. return TRUE;
  451. }
  452. BOOL RemoveOldEntries()
  453. {
  454. Log(StrF(_T("Beginning of RemoveOldEntries MaxEntries: %d - Keep days: %d"), CGetSetOptions::GetMaxEntries(), CGetSetOptions::GetExpiredEntries()));
  455. try
  456. {
  457. CppSQLite3DB db;
  458. CString csDbPath = CGetSetOptions::GetDBPath();
  459. db.open(csDbPath);
  460. if(CGetSetOptions::GetCheckForMaxEntries())
  461. {
  462. long lMax = CGetSetOptions::GetMaxEntries();
  463. if(lMax >= 0)
  464. {
  465. CClipIDs IDs;
  466. int clipId;
  467. CppSQLite3Query q = db.execQueryEx(_T("SELECT lID, lShortCut, lParentID, lDontAutoDelete FROM Main WHERE bIsGroup = 0 ORDER BY clipOrder DESC LIMIT -1 OFFSET %d"), lMax);
  468. while(q.eof() == false)
  469. {
  470. int shortcut = q.getIntField(_T("lShortCut"));
  471. int dontDelete = q.getIntField(_T("lDontAutoDelete"));
  472. int parentId = q.getIntField(_T("lParentID"));
  473. //Only delete entries that have no shortcut and don't have the flag set
  474. if(shortcut == 0 &&
  475. dontDelete == 0 &&
  476. parentId <= 0)
  477. {
  478. clipId = q.getIntField(_T("lID"));
  479. IDs.Add(clipId);
  480. Log(StrF(_T("From MaxEntries - Deleting Id: %d"), clipId));
  481. }
  482. q.nextRow();
  483. }
  484. if(IDs.GetCount() > 0)
  485. {
  486. IDs.DeleteIDs(false, db);
  487. }
  488. }
  489. }
  490. if(CGetSetOptions::GetCheckForExpiredEntries())
  491. {
  492. long lExpire = CGetSetOptions::GetExpiredEntries();
  493. if(lExpire)
  494. {
  495. CTime now = CTime::GetCurrentTime();
  496. now -= CTimeSpan(lExpire, 0, 0, 0);
  497. CClipIDs IDs;
  498. CppSQLite3Query q = db.execQueryEx(_T("SELECT lID FROM Main ")
  499. _T("WHERE lastPasteDate < %d AND ")
  500. _T("bIsGroup = 0 AND lShortCut = 0 AND lParentID <= 0 AND lDontAutoDelete = 0"), now.GetTime());
  501. while(q.eof() == false)
  502. {
  503. IDs.Add(q.getIntField(_T("lID")));
  504. Log(StrF(_T("From Clips Expire - Deleting Id: %d"), q.getIntField(_T("lID"))));
  505. q.nextRow();
  506. }
  507. if(IDs.GetCount() > 0)
  508. {
  509. IDs.DeleteIDs(false, db);
  510. }
  511. }
  512. }
  513. int toDeleteCount = db.execScalar(_T("SELECT COUNT(clipID) FROM MainDeletes"));
  514. Log(StrF(_T("Before Deleting emptied out data, count: %d"), toDeleteCount));
  515. //Only delete 1 at a time, was finding that it was taking a long time to delete clips, locking the db and causing other queries
  516. //to lock up
  517. CppSQLite3Query q = db.execQueryEx(_T("SELECT * FROM MainDeletes LIMIT %d"), CGetSetOptions::GetMainDeletesDeleteCount());
  518. int deleteCount = 0;
  519. while(q.eof() == false)
  520. {
  521. //delete any data items sitting out there that the main table data was deleted
  522. //this was done to speed up deleted from the main table
  523. deleteCount = db.execDMLEx(_T("DELETE FROM MainDeletes WHERE clipID=%d"), q.getIntField(_T("clipID")));
  524. q.nextRow();
  525. }
  526. toDeleteCount = db.execScalar(_T("SELECT COUNT(clipID) FROM MainDeletes"));
  527. Log(StrF(_T("After Deleting emptied out data rows, Count: %d, toDelete: %d"), deleteCount, toDeleteCount));
  528. }
  529. CATCH_SQLITE_EXCEPTION
  530. Log(_T("End of RemoveOldEntries"));
  531. return TRUE;
  532. }
  533. BOOL EnsureDirectory(CString csPath)
  534. {
  535. TCHAR drive[_MAX_DRIVE];
  536. TCHAR dir[_MAX_DIR];
  537. TCHAR fname[_MAX_FNAME];
  538. TCHAR ext[_MAX_EXT];
  539. SPLITPATH(csPath, drive, dir, fname, ext);
  540. CString csDir(drive);
  541. csDir += dir;
  542. if(FileExists(csDir) == FALSE)
  543. {
  544. if(CreateDirectory(csDir, NULL))
  545. return TRUE;
  546. }
  547. else
  548. return TRUE;
  549. return FALSE;
  550. }
  551. // BOOL RunZippApp(CString csCommandLine)
  552. // {
  553. // CString csLocalPath = GETENV(_T("U3_HOST_EXEC_PATH"));
  554. // FIX_CSTRING_PATH(csLocalPath);
  555. //
  556. // CString csZippApp = GETENV(_T("U3_DEVICE_EXEC_PATH"));
  557. // FIX_CSTRING_PATH(csZippApp);
  558. // csZippApp += "7za.exe";
  559. //
  560. // csZippApp += " ";
  561. // csZippApp += csCommandLine;
  562. //
  563. // Log(csZippApp);
  564. //
  565. // STARTUPINFO StartupInfo;
  566. // PROCESS_INFORMATION ProcessInformation;
  567. //
  568. // ZeroMemory(&StartupInfo, sizeof(StartupInfo));
  569. // StartupInfo.cb = sizeof(StartupInfo);
  570. // ZeroMemory(&ProcessInformation, sizeof(ProcessInformation));
  571. //
  572. // StartupInfo.dwFlags = STARTF_USESHOWWINDOW;
  573. // StartupInfo.wShowWindow = SW_HIDE;
  574. //
  575. // BOOL bRet = CreateProcess(NULL, csZippApp.GetBuffer(csZippApp.GetLength()), NULL, NULL, FALSE,
  576. // CREATE_DEFAULT_ERROR_MODE | NORMAL_PRIORITY_CLASS, NULL, csLocalPath,
  577. // &StartupInfo, &ProcessInformation);
  578. //
  579. // if(bRet)
  580. // {
  581. // WaitForSingleObject(ProcessInformation.hProcess, INFINITE);
  582. //
  583. // DWORD dwExitCode;
  584. // GetExitCodeProcess(ProcessInformation.hProcess, &dwExitCode);
  585. //
  586. // CString cs;
  587. // cs.Format(_T("Exit code from unzip = %d"), dwExitCode);
  588. // Log(cs);
  589. //
  590. // if(dwExitCode != 0)
  591. // {
  592. // bRet = FALSE;
  593. // }
  594. // }
  595. // else
  596. // {
  597. // bRet = FALSE;
  598. // Log(_T("Create Process Failed"));
  599. // }
  600. //
  601. // csZippApp.ReleaseBuffer();
  602. //
  603. // return bRet;
  604. // }
  605. // BOOL CopyDownDatabase()
  606. // {
  607. // BOOL bRet = FALSE;
  608. //
  609. // CString csZippedPath = GETENV(_T("U3_APP_DATA_PATH"));
  610. // FIX_CSTRING_PATH(csZippedPath);
  611. //
  612. // CString csUnZippedPath = csZippedPath;
  613. // csUnZippedPath += "Ditto.db";
  614. //
  615. // csZippedPath += "Ditto.7z";
  616. //
  617. // CString csLocalPath = GETENV(_T("U3_HOST_EXEC_PATH"));
  618. // FIX_CSTRING_PATH(csLocalPath);
  619. //
  620. // if(FileExists(csZippedPath))
  621. // {
  622. // CString csCommandLine;
  623. //
  624. // //e = extract
  625. // //surround command line arguments with quotes
  626. // //-aoa = overight files with extracted files
  627. //
  628. // csCommandLine += "e ";
  629. // csCommandLine += "\"";
  630. // csCommandLine += csZippedPath;
  631. // csCommandLine += "\"";
  632. // csCommandLine += " -o";
  633. // csCommandLine += "\"";
  634. // csCommandLine += csLocalPath;
  635. // csCommandLine += "\"";
  636. // csCommandLine += " -aoa";
  637. //
  638. // bRet = RunZippApp(csCommandLine);
  639. //
  640. // csLocalPath += "Ditto.db";
  641. // }
  642. // else if(FileExists(csUnZippedPath))
  643. // {
  644. // csLocalPath += "Ditto.db";
  645. // bRet = CopyFile(csUnZippedPath, csLocalPath, FALSE);
  646. // }
  647. //
  648. // if(FileExists(csLocalPath) == FALSE)
  649. // {
  650. // Log(_T("Failed to copy files from device zip file"));
  651. // }
  652. //
  653. // g_Opt.nLastDbWriteTime = GetLastWriteTime(csLocalPath);
  654. //
  655. // return bRet;
  656. // }
  657. //BOOL CopyUpDatabase()
  658. //{
  659. // CStringA csZippedPath = "C:\\";//getenv("U3_APP_DATA_PATH");
  660. // FIX_CSTRING_PATH(csZippedPath);
  661. // csZippedPath += "Ditto.zip";
  662. // CStringA csLocalPath = GetDBName();//getenv("U3_HOST_EXEC_PATH");
  663. // //FIX_CSTRING_PATH(csLocalPath);
  664. // //csLocalPath += "Ditto.db";
  665. //
  666. // CZipper Zip;
  667. //
  668. // if(Zip.OpenZip(csZippedPath))
  669. // {
  670. // Zip.AddFileToZip(csLocalPath);
  671. // }
  672. //
  673. // return TRUE;
  674. //}