cui/main.c

   1 /* -*- Mode: C ; c-basic-offset: 2 -*- */
   2 /*****************************************************************************
   3  *
   4  * DESCRIPTION:
   5  *  Portable frontend to MediaDatabase
   6  *
   7  * COMPILATION:
   8  *  gcc main.c -lmysqlclient -o mdb_cui [-DOPENBSD]
   9  *
  10  * AUTHOR:
  11  *  Nedko Arnaudov <nedko@users.sourceforge.net>
  12  *
  13  * LICENSE:
  14  *  GNU GENERAL PUBLIC LICENSE version 2
  15  *
  16  *****************************************************************************/
  17
  18 #include <stdio.h>
  19 #include <stdlib.h>
  20 #include <string.h>
  21 #include <sys/types.h>
  22 #include <dirent.h>
  23 #include <sys/stat.h>
  24 #include <errno.h>
  25 #include <time.h>
  26 #include <sys/wait.h>
  27 #include <pwd.h>
  28 #include <unistd.h>
  29 #include <cfl.h>
  30 #include <assert.h>
  31 extern char **environ;
  32
  33 #include "../libdb/libdb.h"
  34 #include "../libdb/memory.h"
  35
  36 #define MB_T_AUDIO      1
  37 #define MB_T_DATA       2
  38 #define MB_T_EMPTY      3
  39
  40 #define MAX_TITLE_SIZE 128      /* in db */
  41 #define MAX_PASS_SIZE 1024
  42 #define DEFAULT_MYSQL_USER      "mediadatabase"
  43 #define DEFAULT_MYSQL_PASS      ""
  44 #define DEFAULT_MYSQL_HOST      "localhost"
  45 #define DEFAULT_MYSQL_DB        "mediadatabase"
  46 #if defined(OPENBSD)
  47 #define DEFAULT_MOUNT_CMD       "sudo /sbin/mount /cdrom"
  48 #define DEFAULT_UNMOUNT_CMD     "sudo /sbin/umount /cdrom"
  49 #else
  50 #define DEFAULT_MOUNT_CMD       "/bin/mount /cdrom"
  51 #define DEFAULT_UNMOUNT_CMD     "/bin/umount /cdrom"
  52 #endif
  53 #define DEFAULT_MOUNTDIR        "/cdrom"
  54 #define MAX_PATH_DEPTH          65535
  55
  56 #define CONF_FILE            ".mediadatabase"
  57 #define CFL_SECTION_GLOBAL   NULL
  58 #define CFL_SECTION_CUI      "cui"
  59 #define CFL_SECTION_MYSQL    "MySQL"
  60 #define CFL_SECTION_SQLITE   "SQLite"
  61 #define CFL_VALUE_MYSQL_HOST "host"
  62 #define CFL_VALUE_MYSQL_USER "user"
  63 #define CFL_VALUE_MYSQL_PASS "pass"
  64 #define CFL_VALUE_MYSQL_DB   "db"
  65 #define CFL_VALUE_MOUNT      "mount"
  66 #define CFL_VALUE_UNMOUNT    "unmount"
  67 #define CFL_VALUE_MOUNTDIR   "mountdir"
  68
  69 //#define PRINT_FILENAMES
  70
  71 char strTitle[MAX_TITLE_SIZE+2]; /* + newline + null */
  72 char strPass[MAX_PASS_SIZE];
  73 char *pszPathBuffer = NULL;
  74 size_t nPathBufferSize = 0;
  75 size_t nMountPrefixLength;
  76 unsigned long nTotalFiles;
  77 unsigned long long nTotalSize;
  78 mediadb_uint nMediaID = 0ULL;
  79 mediadb hDB = MEDIADB_INVALID_VALUE;
  80 int blnMediaMounted = 0;
  81 int blnExiting = 0;
  82 char *pszHost = NULL;
  83 char *pszUser = NULL;
  84 char *pszPass = NULL;
  85 char *pszDB = NULL;
  86 char *pszMountMediaCommand = NULL;
  87 char *pszUnmountMediaCommand = NULL;
  88 char *pszMountDir = NULL;
  89 char *pszTitle = NULL;
  90
  91 #define MODE_LIST_MEDIA     0
  92 #define MODE_ADD_MEDIA      1
  93 #define MODE_UPDATE_MEDIA   2
  94 unsigned int nMode = MODE_LIST_MEDIA;
  95
  96 void
  97 ExecuteCommand(const char *pszCommand);
  98
  99 void
 100 MountMedia()
 101 {
 102   printf("--> Mounting media ...\n");
 103   ExecuteCommand(pszMountMediaCommand);
 104   printf("--> Media mounted.\n");
 105   blnMediaMounted = 1;
 106 }
 107
 108 void
 109 UnmountMedia()
 110 {
 111   printf("--> Unmounting media ...\n");
 112   ExecuteCommand(pszUnmountMediaCommand);
 113   printf("--> Media unmounted.\n");
 114   blnMediaMounted = 0;
 115 }
 116
 117 void
 118 Exit(int nRet)
 119 {
 120   mediadb_result r;
 121
 122   if (blnExiting)
 123     return;
 124
 125   blnExiting = 1;
 126
 127   if (hDB != MEDIADB_INVALID_VALUE)
 128   {
 129     r = mediadb_close(hDB);
 130   }
 131
 132   if (blnMediaMounted)
 133     UnmountMedia();
 134
 135   if (pszHost != NULL)
 136     free(pszHost);
 137
 138   if (pszUser != NULL)
 139     free(pszUser);
 140
 141   if (pszPass != NULL)
 142     free(pszPass);
 143
 144   if (pszDB != NULL)
 145     free(pszDB);
 146
 147   if (pszMountMediaCommand != NULL)
 148     free(pszMountMediaCommand);
 149
 150   if (pszUnmountMediaCommand != NULL)
 151     free(pszUnmountMediaCommand);
 152
 153   if (pszMountDir != NULL)
 154     free(pszMountDir);
 155
 156   if (pszTitle != NULL)
 157     free(pszTitle);
 158
 159   exit(nRet);
 160 }
 161
 162 void
 163 ExecuteCommand(const char *pszCommand)
 164 {
 165   int nRet;
 166
 167   nRet = system(pszCommand);
 168   if (nRet == -1)
 169   {
 170     fprintf(stderr,
 171             "Cannot execute command \"%s\". "
 172             "system() returned -1, error is %d (%s)\n",
 173             pszCommand,
 174             errno,
 175             strerror(errno));
 176     Exit(1);
 177   }
 178
 179   if (WIFEXITED(nRet))
 180   {
 181     if (WEXITSTATUS(nRet) != 0)
 182     {
 183       fprintf(stderr,
 184               "Cannot execute command \"%s\". "
 185               "Command returned %d\n",
 186               pszCommand,
 187               (int)WEXITSTATUS(nRet));
 188       Exit(1);
 189     }
 190   }
 191   else if (WIFSIGNALED(nRet))
 192   {
 193     fprintf(stderr,
 194             "Cannot execute command \"%s\". "
 195             "Cammand was terminated because of signal %d\n",
 196             pszCommand,
 197             (int)WTERMSIG(nRet));
 198     Exit(1);
 199   }
 200   else
 201   {
 202     fprintf(stderr,
 203             "Cannot execute command \"%s\". "
 204             "system() returned %d\n",
 205             pszCommand,
 206             nRet);
 207     Exit(1);
 208   }
 209 }
 210
 211 void
 212 ReadTitleExitOnEmpty()
 213 {
 214   char *pch;
 215
 216   printf("Title: ");
 217   if (fgets(strTitle, sizeof(strTitle), stdin) == NULL)
 218   {
 219     printf("\n");
 220
 221     if (feof(stdin))
 222     {
 223       Exit(0);
 224     }
 225
 226     fprintf(stderr, "Error occured while reading standard input.\n");
 227     Exit(1);
 228   }
 229   else if (strTitle[0] == '\n')
 230   {
 231     Exit(0);
 232   }
 233
 234   if ((pch = strchr(strTitle, '\n')) == NULL)
 235   {
 236     fprintf(stderr, "input line too long.\n");
 237     Exit(1);
 238   }
 239
 240   *pch = '\0';
 241 }
 242
 243 void
 244 ReadMediaID()
 245 {
 246   char *pch;
 247   char strMediaID[100];
 248
 249   printf("Media ID: ");
 250   if (fgets(strMediaID, sizeof(strMediaID), stdin) == NULL)
 251   {
 252     printf("\n");
 253
 254     if (feof(stdin))
 255     {
 256       Exit(0);
 257     }
 258
 259     fprintf(stderr, "Error occured while reading standard input.\n");
 260     Exit(1);
 261   }
 262   else if (strMediaID[0] == '\n')
 263   {
 264     Exit(0);
 265   }
 266
 267   if ((pch = strchr(strMediaID, '\n')) == NULL)
 268   {
 269     fprintf(stderr, "input line too long.\n");
 270     Exit(1);
 271   }
 272
 273   *pch = '\0';
 274
 275   nMediaID = strtoull(strMediaID, NULL, 10);
 276   if (nMediaID == 0)
 277   {
 278     Exit(1);
 279   }
 280 }
 281
 282 void
 283 ReadTitleIgnoreOnEmpty()
 284 {
 285   char *pch;
 286
 287   printf("Title: ");
 288   if (fgets(strTitle, sizeof(strTitle), stdin) == NULL)
 289   {
 290     printf("\n");
 291
 292     if (feof(stdin))
 293     {
 294       strcpy(strTitle, pszTitle);
 295       return;
 296     }
 297
 298     fprintf(stderr, "Error occured while reading standard input.\n");
 299     Exit(1);
 300   }
 301   else if (strTitle[0] == '\n')
 302   {
 303     strcpy(strTitle, pszTitle);
 304     return;
 305   }
 306
 307   if ((pch = strchr(strTitle, '\n')) == NULL)
 308   {
 309     fprintf(stderr, "input line too long.\n");
 310     Exit(1);
 311   }
 312
 313   *pch = '\0';
 314 }
 315
 316 void
 317 AskForPassword()
 318 {
 319   char *pszPass;
 320
 321   pszPass = getpass("MySQL password: ");
 322
 323   strcpy(strPass, pszPass);
 324 }
 325
 326 void
 327 MaybeEnlargeBuffer(char ** ppBuffer,
 328                    size_t *pnBufferSize,
 329                    size_t nSizeRequired)
 330 {
 331   mediadb_result r;
 332
 333   r = maybe_enlarge_buffer(
 334     ppBuffer,
 335     pnBufferSize,
 336     nSizeRequired);
 337   if (MEDIADB_IS_ERROR(r))
 338   {
 339       fprintf(stderr,
 340               "Cannot enlarge buffer.\n");
 341       Exit(0);
 342   }
 343 }
 344
 345 void
 346 ScanDir(unsigned long nDepth)
 347 {
 348   mediadb_result r;
 349   DIR *pDir;
 350   struct dirent *pDE;
 351   size_t nCurrentPathLength;
 352   struct stat st;
 353   size_t sizeName;
 354
 355   if (nDepth >= MAX_PATH_DEPTH)
 356   {
 357     fprintf(stderr,
 358             "Max path depth of reached \"%u\"\n",
 359             (unsigned int)MAX_PATH_DEPTH);
 360     Exit(1);
 361   }
 362
 363   nCurrentPathLength = strlen(pszPathBuffer);
 364
 365   pDir = opendir(pszPathBuffer);
 366   if (pDir == NULL)
 367   {
 368     fprintf(stderr, "Cannot open \"%s\"\n", pszPathBuffer);
 369     Exit(1);
 370   }
 371
 372   if (pDir)
 373   {
 374     while ((pDE = readdir(pDir)) != NULL)
 375     {
 376       if (strcmp(pDE->d_name, ".") == 0)
 377         continue;
 378
 379       if (strcmp(pDE->d_name, "..") == 0)
 380         continue;
 381
 382 #if defined(OPENBSD)
 383       sizeName = pDE->d_namlen;
 384 #else
 385       sizeName = strlen(pDE->d_name);
 386 #endif
 387
 388       MaybeEnlargeBuffer(&pszPathBuffer,
 389                          &nPathBufferSize,
 390                          nCurrentPathLength + sizeName + 1);
 391
 392       memcpy(pszPathBuffer + nCurrentPathLength, pDE->d_name, sizeName);
 393       pszPathBuffer[nCurrentPathLength + sizeName] = 0;
 394
 395       if (lstat(pszPathBuffer, &st) != 0)
 396       {
 397         fprintf(stderr,
 398                 "Cannot stat() %s. Error is %d (%s)\n",
 399                 pszPathBuffer,
 400                 errno,
 401                 strerror(errno));
 402       }
 403
 404       pszPathBuffer[nCurrentPathLength] = 0;
 405
 406       r = mediadb_file_add_new(
 407         hDB,
 408         nMediaID,
 409         (S_ISDIR(st.st_mode))?MEDIADB_FILETYPE_DIR:MEDIADB_FILETYPE_FILE,
 410         pszPathBuffer + nMountPrefixLength,
 411         pDE->d_name,
 412         ((S_ISDIR(st.st_mode))?0:st.st_size),
 413         st.st_ctime);
 414       if (MEDIADB_IS_ERROR(r))
 415       {
 416         fprintf(stderr,
 417                 "Failed to add new file to database. Error is %d (%s)\n",
 418                 (int)r,
 419                 mediadb_get_error_message(hDB));
 420       }
 421
 422       memcpy(pszPathBuffer + nCurrentPathLength, pDE->d_name, sizeName);
 423       pszPathBuffer[nCurrentPathLength + sizeName] = 0;
 424
 425 #ifdef PRINT_FILENAMES
 426       printf("%s", pszPathBuffer + nMountPrefixLength);
 427 #endif
 428
 429       nTotalFiles++;
 430
 431       if (S_ISDIR(st.st_mode))
 432       {
 433 #ifdef PRINT_FILENAMES
 434         printf("/\n");
 435 #else
 436         printf("/");
 437 #endif
 438         fflush(stdout);
 439         memcpy(pszPathBuffer + nCurrentPathLength + sizeName, "/", 2);
 440         ScanDir(nDepth + 1);
 441       }
 442       else
 443       {
 444         nTotalSize += st.st_size;
 445 #ifdef PRINT_FILENAMES
 446         printf("\n");
 447 #else
 448         printf(".");
 449 #endif
 450         fflush(stdout);
 451       }
 452     }
 453
 454     closedir(pDir);
 455   }
 456 }
 457
 458 void
 459 SetDefaults()
 460 {
 461   if (pszHost == NULL)
 462   {
 463     pszHost = strdup(DEFAULT_MYSQL_HOST);
 464     if (pszHost == NULL)
 465     {
 466       fprintf(stderr,
 467               "Out of memory.\n");
 468       Exit(1);
 469     }
 470   }
 471
 472   if (pszUser == NULL)
 473   {
 474     pszUser = strdup(DEFAULT_MYSQL_USER);
 475     if (pszUser == NULL)
 476     {
 477       fprintf(stderr,
 478               "Out of memory.\n");
 479       Exit(1);
 480     }
 481   }
 482
 483   if (pszPass == NULL)
 484   {
 485     pszPass = strdup(DEFAULT_MYSQL_PASS);
 486     if (pszPass == NULL)
 487     {
 488       fprintf(stderr,
 489               "Out of memory.\n");
 490       Exit(1);
 491     }
 492   }
 493
 494   if (pszDB == NULL)
 495   {
 496     pszDB = strdup(DEFAULT_MYSQL_DB);
 497     if (pszDB == NULL)
 498     {
 499       fprintf(stderr,
 500               "Out of memory.\n");
 501       Exit(1);
 502     }
 503   }
 504
 505   if (pszMountMediaCommand == NULL)
 506   {
 507     pszMountMediaCommand = strdup(DEFAULT_MOUNT_CMD);
 508     if (pszMountMediaCommand == NULL)
 509     {
 510       fprintf(stderr,
 511               "Out of memory.\n");
 512       Exit(1);
 513     }
 514   }
 515
 516   if (pszUnmountMediaCommand == NULL)
 517   {
 518     pszUnmountMediaCommand = strdup(DEFAULT_UNMOUNT_CMD);
 519     if (pszUnmountMediaCommand == NULL)
 520     {
 521       fprintf(stderr,
 522               "Out of memory.\n");
 523       Exit(1);
 524     }
 525   }
 526
 527   if (pszMountDir == NULL)
 528   {
 529     pszMountDir = strdup(DEFAULT_MOUNTDIR);
 530     if (pszMountDir == NULL)
 531     {
 532       fprintf(stderr,
 533               "Out of memory.\n");
 534       Exit(1);
 535     }
 536   }
 537 }
 538
 539 void
 540 Help(char *arg0)
 541 {
 542   char *pszExecutable;
 543
 544   pszExecutable = strrchr(arg0, '/');
 545
 546   if (pszExecutable == NULL)
 547   {
 548     pszExecutable = arg0;
 549   }
 550   else
 551   {
 552     pszExecutable++;
 553   }
 554
 555   SetDefaults();
 556
 557   printf("==============================\n");
 558   printf("Offline media content database\n");
 559   printf("==============================\n");
 560   printf("Usage:\n");
 561   printf("[<path_to>]%s [options]\n", pszExecutable);
 562   printf("  Options can be:\n");
 563   printf("  -? - This help\n");
 564   printf("  -L - List media mode\n");
 565   printf("  -A - Add media mode\n");
 566   printf("  -U - Update media mode\n");
 567   printf("  -m <mount_cmd> - command to mount media, default is \"%s\"\n", pszMountMediaCommand);
 568   printf("  -n <unmount_cmd> - command to unmount, default is \"%s\"\n", pszUnmountMediaCommand);
 569   printf("  -d <mount_dir> - where media is mounted, default is \"%s\"\n", pszMountDir);
 570   printf("  -h <mysql_host> - MySQL server host, default is \"%s\"\n", pszHost);
 571   printf("  -u <mysql_user> - MySQL user, default is \"%s\"\n", pszUser);
 572   printf("  -p [mysql_pass] - MySQL password, default is \"%s\"\n", pszPass);
 573   printf("  -b <mysql_database> - MySQL database, default is \"%s\"\n", pszDB);
 574   Exit(0);
 575 }
 576
 577 void
 578 AddMedia()
 579 {
 580   mediadb_result r;
 581   time_t timeAdded;
 582
 583   printf("NOTE: Currently audio media is not supported.\n");
 584
 585 Loop:
 586   memcpy(pszPathBuffer + nMountPrefixLength, "/", 2);
 587
 588   printf("--> Please insert media and enter title.\n");
 589
 590   ReadTitleExitOnEmpty();
 591
 592   MountMedia();
 593
 594   printf("--> Processing media ...\n");
 595
 596   r = mediadb_media_add_new(
 597     hDB,
 598     strTitle,
 599     "",
 600     MEDIADB_MT_DATA,
 601     &nMediaID);
 602
 603   if (MEDIADB_IS_ERROR(r))
 604   {
 605     fprintf(stderr,
 606             "Failed to add new media to database. Error is %d (%s)\n",
 607             (int)r,
 608             mediadb_get_error_message(hDB));
 609     Exit(1);
 610   }
 611
 612   ScanDir(0);
 613
 614   printf("\n");
 615
 616   timeAdded = time(NULL);
 617
 618   r = mediadb_media_update_properties(
 619     hDB,
 620     nMediaID,
 621     timeAdded,
 622     nTotalFiles,
 623     nTotalSize);
 624
 625   if (MEDIADB_IS_ERROR(r))
 626   {
 627     fprintf(stderr,
 628             "Failed to update new media to database. Error is %d (%s)\n",
 629             (int)r,
 630             mediadb_get_error_message(hDB));
 631     Exit(1);
 632   }
 633
 634   UnmountMedia();
 635
 636   printf("--> SUCCESS - Media added.\n");
 637   printf("Media title: %s\n", strTitle);
 638   printf("Media ID: %u\n", (unsigned int)nMediaID);
 639   printf("Added: %s", ctime(&timeAdded));
 640   printf("Total files: %u\n", (unsigned int)nTotalFiles);
 641   printf("Total size: %u\n", (unsigned int)nTotalSize);
 642
 643   goto Loop;
 644 }
 645
 646 void
 647 UpdateMedia()
 648 {
 649   mediadb_result r;
 650   time_t timeAdded;
 651   mediadb_uint nTimeAdded;
 652   mediadb_uint nTotalFiles;
 653   mediadb_uint nTotalSize;
 654
 655   printf("NOTE: Currently audio media is not supported.\n");
 656
 657 Loop:
 658   printf("--> Please enter Media ID.\n");
 659
 660   ReadMediaID();
 661
 662   memcpy(pszPathBuffer + nMountPrefixLength, "/", 2);
 663
 664   r = mediadb_media_get_properties(
 665     hDB,
 666     nMediaID,
 667     &nTimeAdded,
 668     &nTotalFiles,
 669     &nTotalSize,
 670     &pszTitle);
 671   if (MEDIADB_IS_ERROR(r))
 672   {
 673     fprintf(stderr,
 674             "Failed to get media properties. Error is %d (%s)\n",
 675             (int)r,
 676             mediadb_get_error_message(hDB));
 677     Exit(1);
 678   }
 679
 680   if (strlen(pszTitle) + 1 > MAX_TITLE_SIZE)
 681   {
 682     fprintf(stderr,
 683             "Title too big\n");
 684     Exit(1);
 685   }
 686
 687   timeAdded = (time_t)nTimeAdded;
 688
 689   printf("Media title: %s\n", pszTitle);
 690   printf("Added: %s", ctime(&timeAdded));
 691   printf("Total files: %u\n", (unsigned int)nTotalFiles);
 692   printf("Total size: %u\n", (unsigned int)nTotalSize);
 693
 694   printf("--> Please insert media and enter title (Just press return to keep current title).\n");
 695
 696   ReadTitleIgnoreOnEmpty();
 697
 698   MountMedia();
 699
 700   printf("--> Processing media ...\n");
 701
 702   if (strcmp(pszTitle, strTitle) != 0)
 703   {
 704     r = mediadb_media_update_name(
 705       hDB,
 706       nMediaID,
 707       strTitle);
 708     if (MEDIADB_IS_ERROR(r))
 709     {
 710       fprintf(stderr,
 711               "Cannot update media title. Error is %d (%s)\n",
 712               (int)r,
 713               mediadb_get_error_message(hDB));
 714       Exit(1);
 715     }
 716   }
 717
 718   r = mediadb_delete_media_files(
 719     hDB,
 720     nMediaID);
 721
 722   if (MEDIADB_IS_ERROR(r))
 723   {
 724     fprintf(stderr,
 725             "Failed to delete media files in database. Error is %d (%s)\n",
 726             (int)r,
 727             mediadb_get_error_message(hDB));
 728     Exit(1);
 729   }
 730
 731   ScanDir(0);
 732
 733   printf("\n");
 734
 735   //timeAdded = time(NULL);
 736
 737   r = mediadb_media_update_properties(
 738     hDB,
 739     nMediaID,
 740     timeAdded,
 741     nTotalFiles,
 742     nTotalSize);
 743
 744   if (MEDIADB_IS_ERROR(r))
 745   {
 746     fprintf(stderr,
 747             "Failed to update new media to database. Error is %d (%s)\n",
 748             (int)r,
 749             mediadb_get_error_message(hDB));
 750     Exit(1);
 751   }
 752
 753   UnmountMedia();
 754
 755   printf("--> SUCCESS - Media added.\n");
 756   printf("Media title: %s\n", strTitle);
 757   printf("Media ID: %u\n", (unsigned int)nMediaID);
 758   printf("Added: %s", ctime(&timeAdded));
 759   printf("Total files: %u\n", (unsigned int)nTotalFiles);
 760   printf("Total size: %u\n", (unsigned int)nTotalSize);
 761
 762   goto Loop;
 763 }
 764
 765 void
 766 ListMedia()
 767 {
 768   printf("List media mode not implemented yet.\n");
 769 }
 770
 771 int
 772 main(
 773   int argc,
 774   char ** argv
 775   )
 776 {
 777   int i;
 778   mediadb_result r;
 779   int nRet;
 780   cflh_t hConf;
 781   FILE *hFile;
 782   const char *pszHOME;
 783   char **ppsz;
 784   char *pszConf;
 785   size_t s;
 786   char *pszCFLValue;
 787   cfl_value_type_t CFLValueType;
 788   char *pszExecutable;
 789
 790   pszExecutable = strrchr(argv[0], '/');
 791
 792   if (pszExecutable == NULL)
 793   {
 794     pszExecutable = argv[0];
 795   }
 796   else
 797   {
 798     pszExecutable++;
 799   }
 800
 801   if (strcmp(pszExecutable, "mdb_add") == 0)
 802   {
 803     nMode = MODE_ADD_MEDIA;
 804   }
 805   else if (strcmp(pszExecutable, "mdb_update") == 0)
 806   {
 807     nMode = MODE_UPDATE_MEDIA;
 808   }
 809   else
 810   {
 811     nMode = MODE_LIST_MEDIA;
 812   }
 813
 814   /* Find HOME environment variable */
 815   pszHOME = NULL;
 816   ppsz = environ;
 817   while (*ppsz)
 818   {
 819     if (strncasecmp(*ppsz, "HOME=", 5) == 0)
 820     {
 821       pszHOME = *ppsz + 5;
 822       break;
 823     }
 824     ppsz++;
 825   }
 826
 827   if (pszHOME == NULL)
 828   {
 829     fprintf(stderr,
 830             "Please set the HOME encironment variable.\n");
 831     Exit(1);
 832   }
 833
 834   /* Look into configuration file */
 835
 836   s = strlen(pszHOME) + 1 + strlen(CONF_FILE) + 1;
 837   pszConf = (char *)malloc(s);
 838   sprintf(pszConf, "%s/%s", pszHOME, CONF_FILE);
 839
 840   hFile = fopen(pszConf, "r");
 841   if (hFile != NULL)
 842   {
 843     hConf = cfl_alloc();
 844     if (hConf == NULL)
 845     {
 846       fprintf(stderr,
 847               "Out of memory.\n");
 848       Exit(1);
 849     }
 850
 851     nRet = cfl_init_from_file(
 852       hConf,
 853       hFile,
 854       CFL_FILE_TYPE_DEFAULT);
 855     if (nRet != 0)
 856     {
 857       printf("cfl_init_from_file() failed. (%s)\n", cfl_errcode_to_message(nRet));
 858     }
 859
 860     pszCFLValue = cfl_value_get_by_name(
 861       hConf,
 862       CFL_SECTION_GLOBAL,
 863       CFL_VALUE_MOUNT,
 864       &CFLValueType);
 865
 866     if (pszCFLValue != NULL &&
 867         CFLValueType == CFL_TYPE_STRING)
 868     {
 869       pszMountMediaCommand = strdup(pszCFLValue);
 870       if (pszMountMediaCommand == NULL)
 871       {
 872         fprintf(stderr,
 873                 "Out of memory.\n");
 874         Exit(1);
 875       }
 876     }
 877
 878     pszCFLValue = cfl_value_get_by_name(
 879       hConf,
 880       CFL_SECTION_GLOBAL,
 881       CFL_VALUE_UNMOUNT,
 882       &CFLValueType);
 883
 884     if (pszCFLValue != NULL &&
 885         CFLValueType == CFL_TYPE_STRING)
 886     {
 887       pszUnmountMediaCommand = strdup(pszCFLValue);
 888       if (pszUnmountMediaCommand == NULL)
 889       {
 890         fprintf(stderr,
 891                 "Out of memory.\n");
 892         Exit(1);
 893       }
 894     }
 895
 896     pszCFLValue = cfl_value_get_by_name(
 897       hConf,
 898       CFL_SECTION_GLOBAL,
 899       CFL_VALUE_MOUNTDIR,
 900       &CFLValueType);
 901
 902     if (pszCFLValue != NULL &&
 903         CFLValueType == CFL_TYPE_STRING)
 904     {
 905       pszMountDir = strdup(pszCFLValue);
 906       if (pszMountDir == NULL)
 907       {
 908         fprintf(stderr,
 909                 "Out of memory.\n");
 910         Exit(1);
 911       }
 912     }
 913
 914     pszCFLValue = cfl_value_get_by_name(
 915       hConf,
 916       CFL_SECTION_MYSQL,
 917       CFL_VALUE_MYSQL_HOST,
 918       &CFLValueType);
 919
 920     if (pszCFLValue != NULL &&
 921         CFLValueType == CFL_TYPE_STRING)
 922     {
 923       pszHost = strdup(pszCFLValue);
 924       if (pszHost == NULL)
 925       {
 926         fprintf(stderr,
 927                 "Out of memory.\n");
 928         Exit(1);
 929       }
 930     }
 931
 932     pszCFLValue = cfl_value_get_by_name(
 933       hConf,
 934       CFL_SECTION_MYSQL,
 935       CFL_VALUE_MYSQL_DB,
 936       &CFLValueType);
 937
 938     if (pszCFLValue != NULL &&
 939         CFLValueType == CFL_TYPE_STRING)
 940     {
 941       pszDB = strdup(pszCFLValue);
 942       if (pszDB == NULL)
 943       {
 944         fprintf(stderr,
 945                 "Out of memory.\n");
 946         Exit(1);
 947       }
 948     }
 949
 950     pszCFLValue = cfl_value_get_by_name(
 951       hConf,
 952       CFL_SECTION_MYSQL,
 953       CFL_VALUE_MYSQL_USER,
 954       &CFLValueType);
 955
 956     if (pszCFLValue != NULL &&
 957         CFLValueType == CFL_TYPE_STRING)
 958     {
 959       pszUser = strdup(pszCFLValue);
 960       if (pszUser == NULL)
 961       {
 962         fprintf(stderr,
 963                 "Out of memory.\n");
 964         Exit(1);
 965       }
 966     }
 967
 968     pszCFLValue = cfl_value_get_by_name(
 969       hConf,
 970       CFL_SECTION_MYSQL,
 971       CFL_VALUE_MYSQL_PASS,
 972       &CFLValueType);
 973
 974     if (pszCFLValue != NULL &&
 975         CFLValueType == CFL_TYPE_STRING)
 976     {
 977       pszPass = strdup(pszCFLValue);
 978       if (pszPass == NULL)
 979       {
 980         fprintf(stderr,
 981                 "Out of memory.\n");
 982         Exit(1);
 983       }
 984     }
 985
 986     cfl_free(hConf);
 987
 988     fclose(hFile);
 989   }
 990
 991   /* Process options from command line */
 992
 993   for (i = 1; i < argc ; i++)
 994   {
 995     if (strcmp(argv[i],"-L") == 0)
 996     {
 997       nMode = MODE_LIST_MEDIA;
 998     }
 999     else if (strcmp(argv[i],"-A") == 0)
1000     {
1001       nMode = MODE_ADD_MEDIA;
1002     }
1003     else if (strcmp(argv[i],"-U") == 0)
1004     {
1005       nMode = MODE_UPDATE_MEDIA;
1006     }
1007     else if (strcmp(argv[i],"-h") == 0)
1008     {
1009       i++;
1010       if (i < argc)
1011       {
1012         if (pszHost != NULL)
1013           free(pszHost);
1014         pszHost = strdup(argv[i]);
1015         if (pszHost == NULL)
1016         {
1017           fprintf(stderr,
1018                   "Out of memory.\n");
1019           Exit(1);
1020         }
1021       }
1022       else
1023       {
1024         fprintf(stderr,
1025                 "Missing parameter of -h option\n");
1026         Help(argv[0]);
1027       }
1028     }
1029     else if (strcmp(argv[i],"-u") == 0)
1030     {
1031       i++;
1032       if (i < argc)
1033       {
1034         if (pszUser != NULL)
1035           free(pszUser);
1036         pszUser = strdup(argv[i]);
1037         if (pszUser == NULL)
1038         {
1039           fprintf(stderr,
1040                   "Out of memory.\n");
1041           Exit(1);
1042         }
1043       }
1044       else
1045       {
1046         fprintf(stderr,
1047                 "Missing parameter of -u option\n");
1048         Help(argv[0]);
1049       }
1050     }
1051     else if (strcmp(argv[i],"-p") == 0)
1052     {
1053       if (i+1 < argc && argv[i+1][0] != '-')
1054       {
1055         i++;
1056         if (pszPass != NULL)
1057           free(pszPass);
1058         pszPass = strdup(argv[i]);
1059         if (pszPass == NULL)
1060         {
1061           fprintf(stderr,
1062                   "Out of memory.\n");
1063           Exit(1);
1064         }
1065       }
1066       else
1067       {
1068         AskForPassword();
1069         if (pszPass != NULL)
1070           free(pszPass);
1071         pszPass = strdup(strPass);
1072         if (pszPass == NULL)
1073         {
1074           fprintf(stderr,
1075                   "Out of memory.\n");
1076           Exit(1);
1077         }
1078       }
1079     }
1080     else if (strcmp(argv[i],"-b") == 0)
1081     {
1082       i++;
1083       if (i < argc)
1084       {
1085         if (pszDB != NULL)
1086           free(pszDB);
1087         pszDB = strdup(argv[i]);
1088         if (pszDB == NULL)
1089         {
1090           fprintf(stderr,
1091                   "Out of memory.\n");
1092           Exit(1);
1093         }
1094       }
1095       else
1096       {
1097         fprintf(stderr,
1098                 "Missing parameter of -b option\n");
1099         Help(argv[0]);
1100       }
1101     }
1102     else if (strcmp(argv[i],"-m") == 0)
1103     {
1104       i++;
1105       if (i < argc)
1106       {
1107         if (pszMountMediaCommand != NULL)
1108           free(pszMountMediaCommand);
1109         pszMountMediaCommand = strdup(argv[i]);
1110         if (pszMountMediaCommand == NULL)
1111         {
1112           fprintf(stderr,
1113                   "Out of memory.\n");
1114           Exit(1);
1115         }
1116       }
1117       else
1118       {
1119         fprintf(stderr,
1120                 "Missing parameter of -m option\n");
1121         Help(argv[0]);
1122       }
1123     }
1124     else if (strcmp(argv[i],"-n") == 0)
1125     {
1126       i++;
1127       if (i < argc)
1128       {
1129         if (pszUnmountMediaCommand != NULL)
1130           free(pszUnmountMediaCommand);
1131         pszUnmountMediaCommand = strdup(argv[i]);
1132         if (pszUnmountMediaCommand == NULL)
1133         {
1134           fprintf(stderr,
1135                   "Out of memory.\n");
1136           Exit(1);
1137         }
1138       }
1139       else
1140       {
1141         fprintf(stderr,
1142                 "Missing parameter of -n option\n");
1143         Help(argv[0]);
1144       }
1145     }
1146     else if (strcmp(argv[i],"-d") == 0)
1147     {
1148       i++;
1149       if (i < argc)
1150       {
1151         if (pszMountDir != NULL)
1152           free(pszMountDir);
1153         pszMountDir = strdup(argv[i]);
1154         if (pszMountDir == NULL)
1155         {
1156           fprintf(stderr,
1157                   "Out of memory.\n");
1158           Exit(1);
1159         }
1160       }
1161       else
1162       {
1163         fprintf(stderr,
1164                 "Missing parameter of -d option\n");
1165         Help(argv[0]);
1166       }
1167     }
1168     else if (strcmp(argv[i],"-?") == 0)
1169     {
1170       Help(argv[0]);
1171     }
1172     else
1173     {
1174       fprintf(stderr,
1175               "Unknown option \"%s\"\n",
1176               argv[i]);
1177       Help(argv[0]);
1178     }
1179   }
1180
1181   SetDefaults();
1182
1183   r = mediadb_open(
1184     MEDIADB_DBTYPE_MYSQL,
1185     pszHost,
1186     pszUser,
1187     pszPass,
1188     pszDB,
1189     &hDB);
1190   if (MEDIADB_IS_ERROR(r))
1191   {
1192     fprintf(stderr, "Failed to connect to database.\n");
1193     Exit(1);
1194   }
1195
1196   nMountPrefixLength = strlen(pszMountDir);
1197
1198   MaybeEnlargeBuffer(&pszPathBuffer,
1199                      &nPathBufferSize,
1200                      nMountPrefixLength + 2);
1201
1202   memcpy(pszPathBuffer, pszMountDir, nMountPrefixLength);
1203
1204   if (nMode == MODE_ADD_MEDIA)
1205   {
1206     printf("=========================\n");
1207     printf("Add media mode\n");
1208     printf("=========================\n");
1209     AddMedia();
1210   }
1211   else if (nMode == MODE_UPDATE_MEDIA)
1212   {
1213     printf("=========================\n");
1214     printf("Update media mode\n");
1215     printf("=========================\n");
1216     UpdateMedia();
1217   }
1218   else if (nMode == MODE_LIST_MEDIA)
1219   {
1220     printf("=========================\n");
1221     printf("List media mode\n");
1222     printf("=========================\n");
1223     ListMedia();
1224   }
1225   else
1226   {
1227     assert(0);
1228     return 1;
1229   }
1230
1231   Exit(0);
1232   return 1;
1233 }