store/workben/t_store.cxx

   1 /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
   2 /*
   3  * This file is part of the LibreOffice project.
   4  *
   5  * This Source Code Form is subject to the terms of the Mozilla Public
   6  * License, v. 2.0. If a copy of the MPL was not distributed with this
   7  * file, You can obtain one at http://mozilla.org/MPL/2.0/.
   8  *
   9  * This file incorporates work covered by the following license notice:
  10  *
  11  *   Licensed to the Apache Software Foundation (ASF) under one or more
  12  *   contributor license agreements. See the NOTICE file distributed
  13  *   with this work for additional information regarding copyright
  14  *   ownership. The ASF licenses this file to you under the Apache
  15  *   License, Version 2.0 (the "License"); you may not use this file
  16  *   except in compliance with the License. You may obtain a copy of
  17  *   the License at http://www.apache.org/licenses/LICENSE-2.0 .
  18  */
  19
  20
  21 #define _T_STORE_CXX
  22 #include <sal/types.h>
  23 #include <osl/diagnose.h>
  24 #include <osl/thread.h>
  25 #include <osl/time.h>
  26 #include <rtl/ustring.hxx>
  27 #include <store/store.hxx>
  28
  29 #include <stdio.h>
  30
  31 #if (defined(WNT) && defined(PROFILE))
  32 extern "C"
  33 {
  34     void StartCAP (void);
  35     void StopCAP  (void);
  36     void DumpCAP  (void);
  37 }
  38 #endif /* PROFILE */
  39
  40
  41 /*========================================================================
  42  *
  43  * Internals.
  44  *
  45  *======================================================================*/
  46 #define _DEMOSTOR_BUFSIZ          512  /* 4096, 1024, 512 */
  47 #define _DEMOSTOR_LOOPS           1000 /* 1000, 2000 */
  48
  49 #define _DEMOSTOR_REMOVE          0
  50 #define _DEMOSTOR_REBUILD         0
  51
  52 enum Options
  53 {
  54     OPTION_HELP    = 0x0001,
  55     OPTION_FILE    = 0x0002,
  56     OPTION_PAUSE   = 0x0004,
  57     OPTION_REBUILD = 0x0008,
  58
  59     OPTION_DUMP    = 0x0010,
  60     OPTION_ITER    = 0x0020,
  61     OPTION_LINK    = 0x0040,
  62
  63     OPTION_READ    = 0x0100,
  64     OPTION_WRITE   = 0x0200,
  65     OPTION_CREAT   = 0x0400,
  66     OPTION_TRUNC   = 0x0800
  67 };
  68
  69 inline sal_Char ascii_toLowerCase (sal_Char ch)
  70 {
  71     if ((ch >= 0x41) && (ch <= 0x5A))
  72         return (ch + 0x20);
  73     else
  74         return (ch);
  75 }
  76
  77 /*========================================================================
  78  *
  79  * Timing.
  80  *
  81  *======================================================================*/
  82 struct OTime : public TimeValue
  83 {
  84     OTime (void)
  85     {
  86         Seconds = 0;
  87         Nanosec = 0;
  88     }
  89
  90     static OTime getSystemTime (void)
  91     {
  92         OTime tv;
  93         osl_getSystemTime (&tv);
  94         return tv;
  95     }
  96
  97     OTime& operator-= (const OTime& rPast)
  98     {
  99         Seconds -= rPast.Seconds;
 100         if (Nanosec < rPast.Nanosec)
 101         {
 102             Seconds -= 1;
 103             Nanosec += 1000000000;
 104         }
 105         Nanosec -= rPast.Nanosec;
 106         return *this;
 107     }
 108
 109     friend OTime operator- (const OTime& rTimeA, const OTime& rTimeB)
 110     {
 111         OTime aTimeC (rTimeA);
 112         aTimeC -= rTimeB;
 113         return aTimeC;
 114     }
 115 };
 116
 117 /*========================================================================
 118  *
 119  * DirectoryTraveller.
 120  *
 121  *======================================================================*/
 122 typedef store::OStoreDirectory Directory;
 123
 124 class DirectoryTraveller : public Directory::traveller
 125 {
 126     typedef store::OStoreFile   file;
 127     typedef Directory::iterator iter;
 128
 129     store::OStoreFile m_aFile;
 130     OUString          m_aPath;
 131
 132     sal_uInt32  m_nOptions;
 133     unsigned int  m_nLevel;
 134     unsigned int  m_nCount;
 135
 136 public:
 137     DirectoryTraveller (
 138         const file&     rFile,
 139         const OUString &rPath,
 140         const OUString &rName,
 141         sal_uInt32      nOptions,
 142         unsigned int nLevel = 0);
 143
 144     virtual ~DirectoryTraveller (void);
 145
 146     virtual sal_Bool visit (const iter& it);
 147 };
 148
 149 /*
 150  * DirectoryTraveller.
 151  */
 152 DirectoryTraveller::DirectoryTraveller (
 153     const file&     rFile,
 154     const OUString &rPath,
 155     const OUString &rName,
 156     sal_uInt32      nOptions,
 157     unsigned int nLevel)
 158     : m_aFile    (rFile),
 159       m_aPath    (rPath),
 160       m_nOptions (nOptions),
 161       m_nLevel   (nLevel),
 162       m_nCount   (0)
 163 {
 164     m_aPath += rName + "/";
 165 }
 166
 167 /*
 168  * ~DirectoryTraveller.
 169  */
 170 DirectoryTraveller::~DirectoryTraveller (void)
 171 {
 172 }
 173
 174 /*
 175  * visit.
 176  */
 177 sal_Bool DirectoryTraveller::visit (const iter& it)
 178 {
 179     m_nCount++;
 180     if (m_nOptions & OPTION_DUMP)
 181     {
 182         OString aName (it.m_pszName, it.m_nLength, RTL_TEXTENCODING_UTF8);
 183         printf ("Visit(%u,%u): %s [0x%08x] %d [Bytes]\n",
 184                 m_nLevel, m_nCount,
 185                 aName.pData->buffer, (unsigned int)(it.m_nAttrib), (unsigned int)(it.m_nSize));
 186     }
 187     if (it.m_nAttrib & STORE_ATTRIB_ISDIR)
 188     {
 189         OUString  aName (it.m_pszName, it.m_nLength);
 190         if (aName.compareToAscii ("XTextViewCursorSupplier") == 0)
 191         {
 192             m_nCount += 1 - 1;
 193         }
 194         Directory aSubDir;
 195
 196         storeError eErrCode = aSubDir.create (
 197             m_aFile, m_aPath, aName, store_AccessReadOnly);
 198         if (eErrCode == store_E_None)
 199         {
 200             sal_uInt32 nRefCount = 0;
 201             m_aFile.getRefererCount (nRefCount);
 202
 203             DirectoryTraveller aSubTraveller (
 204                 m_aFile, m_aPath, aName, m_nOptions, m_nLevel + 1);
 205             aSubDir.travel (aSubTraveller);
 206         }
 207     }
 208     return sal_True;
 209 }
 210
 211 /*========================================================================
 212  *
 213  * main.
 214  *
 215  *======================================================================*/
 216 int SAL_CALL main (int argc, char **argv)
 217 {
 218 #if (defined(WNT) && defined(PROFILE))
 219     StartCAP();
 220 #else
 221     OTime aMainStartTime (OTime::getSystemTime());
 222 #endif /* PROFILE */
 223
 224     store::OStoreFile aFile;
 225     storeError eErrCode = store_E_None;
 226
 227     sal_uInt32 nOptions = 0;
 228     for (int i = 1; i < argc; i++)
 229     {
 230         const char *opt = argv[i];
 231         if (opt[0] == '-')
 232         {
 233             switch (ascii_toLowerCase(sal_Char(opt[1])))
 234             {
 235                 case 'f':
 236                     nOptions |= OPTION_FILE;
 237                     break;
 238
 239                 case 'd':
 240                     nOptions |= OPTION_DUMP;
 241                     break;
 242                 case 'i':
 243                     nOptions |= OPTION_ITER;
 244                     break;
 245                 case 'l':
 246                     nOptions |= OPTION_LINK;
 247                     break;
 248
 249                 case 'r':
 250                     nOptions |= OPTION_READ;
 251                     break;
 252                 case 'w':
 253                     nOptions |= OPTION_WRITE;
 254                     break;
 255                 case 'c':
 256                     nOptions |= OPTION_CREAT;
 257                     break;
 258                 case 't':
 259                     nOptions |= OPTION_TRUNC;
 260                     break;
 261
 262                 case 'p':
 263                     nOptions |= OPTION_PAUSE;
 264                     break;
 265
 266                 case 'h':
 267                 default:
 268                     nOptions |= OPTION_HELP;
 269                     break;
 270             }
 271         }
 272         else
 273         {
 274             if (nOptions & OPTION_FILE)
 275             {
 276                 OUString aName (
 277                     argv[i], rtl_str_getLength(argv[i]),
 278                     osl_getThreadTextEncoding());
 279                 if ((nOptions & OPTION_CREAT) && (nOptions & OPTION_TRUNC))
 280                     eErrCode = aFile.create (aName, store_AccessCreate);
 281                 else if (nOptions & OPTION_CREAT)
 282                     eErrCode = aFile.create (aName, store_AccessReadCreate);
 283                 else if (nOptions & OPTION_WRITE)
 284                     eErrCode = aFile.create (aName, store_AccessReadWrite);
 285                 else
 286                     eErrCode = aFile.create (aName, store_AccessReadOnly);
 287                 if (eErrCode != store_E_None)
 288                 {
 289                     printf ("Error: can't open file: %s\n", argv[i]);
 290                     exit (0);
 291                 }
 292             }
 293         }
 294     }
 295
 296     if ((nOptions == 0) || (nOptions & OPTION_HELP))
 297     {
 298         printf ("Usage:\tt_store "
 299                 "[[-c] [-t] [-r] [-w]] [[-i] [-d] [-h]] "
 300                 "[-f filename]\n");
 301
 302         printf ("\nOptions:\n");
 303         printf ("-c\tcreate\n");
 304         printf ("-t\ttruncate\n");
 305         printf ("-r\tread\n");
 306         printf ("-w\twrite\n");
 307         printf ("-i\titerate\n");
 308         printf ("-d\tdump\n");
 309         printf ("-h\thelp\n");
 310         printf ("-f\tfilename\n");
 311
 312         printf ("\nExamples:");
 313         printf ("\nt_store -c -w -f t_store.rdb\n");
 314         printf ("\tCreate file 't_store.rdb',\n"
 315                 "\twrite fixed number (1000) of streams.\n");
 316         printf ("\nt_store -c -i -d -f t_store.rdb\n");
 317         printf ("\tOpen file 't_store.rdb', "
 318                 "create '/' directory,\n"
 319                 "\titerate directory tree, "
 320                 "dump directory info.\n");
 321
 322         exit (0);
 323     }
 324
 325     if (!aFile.isValid())
 326     {
 327         eErrCode = aFile.createInMemory();
 328         if (eErrCode != store_E_None)
 329         {
 330             printf ("Error: can't create memory file\n");
 331             exit (0);
 332         }
 333     }
 334
 335     // Stream Read/Write.
 336     OUString aPath ("/");
 337     if ((nOptions & OPTION_READ) || (nOptions & OPTION_WRITE))
 338     {
 339         // Mode.
 340         storeAccessMode eMode = store_AccessReadOnly;
 341         if (nOptions & OPTION_WRITE)
 342             eMode = store_AccessReadWrite;
 343         if (nOptions & OPTION_CREAT)
 344             eMode = store_AccessCreate;
 345
 346         // Buffer.
 347         char pBuffer[_DEMOSTOR_BUFSIZ] = "Hello World";
 348         pBuffer[_DEMOSTOR_BUFSIZ - 2] = 'B';
 349         pBuffer[_DEMOSTOR_BUFSIZ - 1] = '\0';
 350
 351         // Load/Save.
 352 #ifndef PROFILE
 353         OTime aStartTime (OTime::getSystemTime());
 354 #endif /* PROFILE */
 355
 356         for (int i = 0; i < _DEMOSTOR_LOOPS; i++)
 357         {
 358             OUString aName ("demostor-");
 359             aName += OUString::valueOf ((sal_Int32)(i + 1), 10);
 360             aName += ".dat";
 361
 362 #if (_DEMOSTOR_REMOVE == 1)
 363             eErrCode = aFile.remove (aPath, aName);
 364             if ((eErrCode != store_E_None     ) &&
 365                 (eErrCode != store_E_NotExists)    )
 366                 break;
 367 #endif /* _REMOVE */
 368
 369             store::OStoreStream aStream;
 370             eErrCode = aStream.create (aFile, aPath, aName, eMode);
 371             if (eErrCode != store_E_None)
 372             {
 373                 OSL_TRACE("OStoreStream(%d)::create(): error: %d", i, eErrCode);
 374                 break;
 375             }
 376
 377             if (nOptions & OPTION_TRUNC)
 378             {
 379                 eErrCode = aStream.setSize(0);
 380                 if (eErrCode != store_E_None)
 381                 {
 382                     OSL_TRACE("OStoreStream(%d)::setSize(0): error: %d", i, eErrCode);
 383                     break;
 384                 }
 385             }
 386
 387             sal_uInt32 nDone = 0;
 388             if (nOptions & OPTION_WRITE)
 389             {
 390                 eErrCode = aStream.writeAt (
 391                     0, pBuffer, sizeof(pBuffer), nDone);
 392                 if (eErrCode != store_E_None)
 393                 {
 394                     OSL_TRACE("OStoreStream(%d)::writeAt(): error: %d", i, eErrCode);
 395                     break;
 396                 }
 397             }
 398
 399             if (nOptions & OPTION_READ)
 400             {
 401                 sal_uInt32 nOffset = 0;
 402                 for (;;)
 403                 {
 404                     eErrCode = aStream.readAt (
 405                         nOffset, pBuffer, sizeof(pBuffer), nDone);
 406                     if (eErrCode != store_E_None)
 407                     {
 408                         OSL_TRACE("OStoreStream(%d)::readAt(): error: %d", i, eErrCode);
 409                         break;
 410                     }
 411                     if (nDone == 0)
 412                         break;
 413                     nOffset += nDone;
 414                 }
 415             }
 416
 417             aStream.close();
 418
 419 #ifndef PROFILE
 420             if (((i + 1) % (_DEMOSTOR_LOOPS/10)) == 0)
 421             {
 422                 OTime aDelta (OTime::getSystemTime() - aStartTime);
 423
 424                 sal_uInt32 nDelta = aDelta.Seconds * 1000000;
 425                 nDelta += (aDelta.Nanosec / 1000);
 426
 427                 printf ("%d: %12.4g[usec]\n", (i+1),
 428                         (double)(nDelta)/(double)(i+1));
 429             }
 430 #endif /* PROFILE */
 431         }
 432
 433 #ifndef PROFILE
 434         OTime aDelta (OTime::getSystemTime() - aStartTime);
 435
 436         sal_uInt32 nDelta = aDelta.Seconds * 1000000;
 437         nDelta += (aDelta.Nanosec / 1000);
 438
 439         printf ("Total(rd,wr): %d[usec]\n", (unsigned int)(nDelta));
 440 #endif /* PROFILE */
 441     }
 442
 443     // Link/Rename.
 444     if (nOptions & OPTION_LINK)
 445     {
 446         // Create symlink to (root) directory.
 447         eErrCode = aFile.symlink (
 448             aPath,      OUString("000000/"),
 449             OUString(), aPath);
 450         OSL_POSTCOND(
 451             ((eErrCode == store_E_None         ) ||
 452              (eErrCode == store_E_AlreadyExists)    ),
 453             "t_store::main(): store_symlink() failed");
 454
 455         // Create symlink to file.
 456         OUString aLinkName ("demostor-1.lnk");
 457
 458         eErrCode = aFile.symlink (
 459             aPath, aLinkName,
 460             aPath, OUString("demostor-1.dat"));
 461         OSL_POSTCOND(
 462             ((eErrCode == store_E_None         ) ||
 463              (eErrCode == store_E_AlreadyExists)    ),
 464             "t_store::main(): store_symlink() failed");
 465         if ((eErrCode == store_E_None         ) ||
 466             (eErrCode == store_E_AlreadyExists)    )
 467         {
 468             OUString aShortcut (
 469                 "Shortcut to demostor-1.dat");
 470             eErrCode = aFile.rename (
 471                 aPath, aLinkName,
 472                 aPath, aShortcut);
 473             OSL_POSTCOND(
 474                 ((eErrCode == store_E_None         ) ||
 475                  (eErrCode == store_E_AlreadyExists)    ),
 476                 "t_store::main(): store_rename() failed");
 477         }
 478
 479         // Create directory.
 480         OUString aDirName ("demostor-1.dir");
 481         store::OStoreDirectory aDir;
 482
 483         eErrCode = aDir.create (
 484             aFile, aPath, aDirName, store_AccessReadCreate);
 485         OSL_POSTCOND(
 486             (eErrCode == store_E_None),
 487             "t_store::main(): store_createDirectory() failed");
 488         if (eErrCode == store_E_None)
 489         {
 490         }
 491     }
 492
 493     // Directory iteration.
 494     if (nOptions & OPTION_ITER)
 495     {
 496 #ifndef PROFILE
 497         OTime aStartTime (OTime::getSystemTime());
 498 #endif /* PROFILE */
 499         OUString aEmpty;
 500
 501         // Root directory.
 502         store::OStoreDirectory aRootDir;
 503         if (nOptions & OPTION_LINK)
 504         {
 505             // Open symlink entry.
 506             eErrCode = aRootDir.create (
 507                 aFile, aPath, OUString("000000"),
 508                 store_AccessReadOnly);
 509         }
 510         else
 511         {
 512             // Open direct entry.
 513             if (nOptions & OPTION_CREAT)
 514                 eErrCode = aRootDir.create (
 515                     aFile, aEmpty, aEmpty, store_AccessReadCreate);
 516             else if (nOptions & OPTION_WRITE)
 517                 eErrCode = aRootDir.create (
 518                     aFile, aEmpty, aEmpty, store_AccessReadWrite);
 519             else
 520                 eErrCode = aRootDir.create (
 521                     aFile, aEmpty, aEmpty, store_AccessReadOnly);
 522         }
 523
 524         if (eErrCode == store_E_None)
 525         {
 526             // Traverse directory tree.
 527             DirectoryTraveller aTraveller (
 528                 aFile, aEmpty, aEmpty, nOptions, 0);
 529             aRootDir.travel (aTraveller);
 530         }
 531         else
 532         {
 533             // Failure.
 534             printf ("Error: can't open directory: \"/\"\n");
 535         }
 536
 537 #ifndef PROFILE
 538         OTime aDelta (OTime::getSystemTime() - aStartTime);
 539
 540         sal_uInt32 nDelta = aDelta.Seconds * 1000000;
 541         nDelta += (aDelta.Nanosec / 1000);
 542
 543         printf ("Total(iter): %d[usec]\n", (unsigned int)(nDelta));
 544 #endif /* PROFILE */
 545     }
 546
 547     if (nOptions & OPTION_PAUSE)
 548     {
 549         TimeValue tv;
 550         tv.Seconds = 300;
 551         tv.Nanosec = 0;
 552         osl_waitThread (&tv);
 553     }
 554
 555     // Size.
 556     sal_uInt32 nSize = 0;
 557     aFile.getSize (nSize);
 558
 559     // Done.
 560     aFile.close();
 561
 562 #if (defined(WNT) && defined(PROFILE))
 563     StopCAP();
 564     DumpCAP();
 565 #endif /* PROFILE */
 566 #ifndef PROFILE
 567     OTime aDelta (OTime::getSystemTime() - aMainStartTime);
 568
 569     sal_uInt32 nDelta = aDelta.Seconds * 1000000;
 570     nDelta += (aDelta.Nanosec / 1000);
 571
 572     printf ("Total: %d[usec]\n", (unsigned int)(nDelta));
 573 #endif /* PROFILE */
 574
 575     return 0;
 576 }
 577
 578 /* vim:set shiftwidth=4 softtabstop=4 expandtab: */