src/common/media-svc-storage.c

   1 /*
   2  * libmedia-service
   3  *
   4  * Copyright (c) 2000 - 2011 Samsung Electronics Co., Ltd. All rights reserved.
   5  *
   6  * Contact: Hyunjun Ko <zzoon.ko@samsung.com>, Haejeong Kim <backto.kim@samsung.com>
   7  *
   8  * Licensed under the Apache License, Version 2.0 (the "License");
   9  * you may not use this file except in compliance with the License.
  10  * You may obtain a copy of the License at
  11  *
  12  * http://www.apache.org/licenses/LICENSE-2.0
  13  *
  14  * Unless required by applicable law or agreed to in writing, software
  15  * distributed under the License is distributed on an "AS IS" BASIS,
  16  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  17  * See the License for the specific language governing permissions and
  18  * limitations under the License.
  19  *
  20  */
  21
  22 #include "media-util-err.h"
  23 #include "media-svc-debug.h"
  24 #include "media-svc-env.h"
  25 #include "media-svc-db-utils.h"
  26 #include "media-svc-util.h"
  27 #include "media-svc-storage.h"
  28
  29 int _media_svc_check_storage(sqlite3 *handle, const char *storage_id, char **storage_path, int *validity, uid_t uid)
  30 {
  31         int ret = MS_MEDIA_ERR_NONE;
  32         sqlite3_stmt *sql_stmt = NULL;
  33         char *sql = NULL;
  34
  35         *storage_path = NULL;
  36         *validity = 0;
  37
  38         sql = sqlite3_mprintf("SELECT storage_path, validity FROM '%q' WHERE storage_uuid=%Q", MEDIA_SVC_DB_TABLE_STORAGE, storage_id);
  39         ret = _media_svc_sql_prepare_to_step(handle, sql, &sql_stmt);
  40         media_svc_retv_if(ret != MS_MEDIA_ERR_NONE, ret);
  41
  42         *storage_path = g_strdup((const char *)sqlite3_column_text(sql_stmt, 0));
  43         *validity = sqlite3_column_int(sql_stmt, 1);
  44
  45         SQLITE3_FINALIZE(sql_stmt);
  46
  47         /*check storage media table*/
  48         if (STRING_VALID(storage_id)) {
  49                 int table_cnt = 0;
  50
  51                 /*Select list of storage*/
  52                 sql = sqlite3_mprintf("SELECT COUNT(*) FROM SQLITE_MASTER WHERE type='table' and name='%q'", storage_id);
  53                 ret = _media_svc_sql_prepare_to_step(handle, sql, &sql_stmt);
  54                 media_svc_retv_if(ret != MS_MEDIA_ERR_NONE, ret);
  55
  56                 table_cnt = sqlite3_column_int(sql_stmt, 0);
  57                 SQLITE3_FINALIZE(sql_stmt);
  58
  59                 if (table_cnt > 0) {
  60                         /*DO NOT THING*/
  61                 } else {
  62                         media_svc_error("media table not exist for storage [%s]", storage_id);
  63                         /*make storage media table*/
  64                         ret = _media_svc_create_media_table_with_id(storage_id, uid);
  65                         media_svc_retvm_if(ret != MS_MEDIA_ERR_NONE, ret, "create media table failed : %d", ret);
  66                 }
  67         }
  68
  69         return MS_MEDIA_ERR_NONE;
  70 }
  71
  72 int _media_svc_append_storage(const char *storage_id, const char *storage_name, const char *storage_path, media_svc_storage_type_e storage_type, uid_t uid)
  73 {
  74         int ret = MS_MEDIA_ERR_NONE;
  75         char *sql = sqlite3_mprintf("INSERT INTO %q (storage_uuid, storage_name, storage_path, storage_type) values (%Q, %Q, %Q, %d); ",
  76                                                 MEDIA_SVC_DB_TABLE_STORAGE, storage_id, storage_name, storage_path, storage_type);
  77
  78         ret = _media_svc_sql_query(sql, uid);
  79         SQLITE3_SAFE_FREE(sql);
  80
  81         return ret;
  82 }
  83
  84 int _media_svc_update_storage_path(sqlite3 *handle, const char *storage_id, const char *path, uid_t uid)
  85 {
  86         int ret = MS_MEDIA_ERR_NONE;
  87         char *sql = NULL;
  88         char *old_storage_path = NULL;
  89         int validity = 0;
  90
  91         /*Get old path*/
  92         ret = _media_svc_check_storage(handle, storage_id, &old_storage_path, &validity, uid);
  93         media_svc_retv_if(ret != MS_MEDIA_ERR_NONE, ret);
  94
  95         /*Storage table update*/
  96         sql = sqlite3_mprintf("UPDATE '%q' SET storage_path=%Q WHERE storage_uuid=%Q;", MEDIA_SVC_DB_TABLE_STORAGE, path, storage_id);
  97         ret = _media_svc_sql_query(sql, uid);
  98         SQLITE3_SAFE_FREE(sql);
  99         if (ret != MS_MEDIA_ERR_NONE) {
 100                 G_SAFE_FREE(old_storage_path);
 101                 return ret;
 102         }
 103
 104         /*Folder table update*/
 105         sql = sqlite3_mprintf("UPDATE '%q' SET path=REPLACE(path, %Q, %Q) WHERE storage_uuid=%Q;", MEDIA_SVC_DB_TABLE_FOLDER, old_storage_path, path, storage_id);
 106         ret = _media_svc_sql_query(sql, uid);
 107         SQLITE3_SAFE_FREE(sql);
 108         if (ret != MS_MEDIA_ERR_NONE) {
 109                 G_SAFE_FREE(old_storage_path);
 110                 return ret;
 111         }
 112
 113         /*Media table update*/
 114         sql = sqlite3_mprintf("UPDATE '%q' SET path=REPLACE(path, %Q, %Q);", storage_id, old_storage_path, path);
 115         ret = _media_svc_sql_query(sql, uid);
 116         SQLITE3_SAFE_FREE(sql);
 117         G_SAFE_FREE(old_storage_path);
 118         media_svc_retv_if(ret != MS_MEDIA_ERR_NONE, ret);
 119
 120         return ret;
 121 }
 122
 123 int _media_svc_delete_storage(const char *storage_id, uid_t uid)
 124 {
 125         int ret = MS_MEDIA_ERR_NONE;
 126         char *sql = NULL;
 127
 128         if (storage_id != NULL)
 129                 sql = sqlite3_mprintf("DELETE FROM '%q' WHERE storage_uuid=%Q;", MEDIA_SVC_DB_TABLE_STORAGE, storage_id);
 130
 131         ret = _media_svc_sql_query(sql, uid);
 132         SQLITE3_SAFE_FREE(sql);
 133
 134         return ret;
 135 }
 136
 137 int _media_svc_delete_invalid_storage(sqlite3 *handle, media_svc_storage_type_e storage_type, uid_t uid)
 138 {
 139         int ret = MS_MEDIA_ERR_NONE;
 140         char *sql = NULL;
 141         char *storage_id = NULL;
 142         char *thumb_path = NULL;
 143         sqlite3_stmt *sql_stmt = NULL;
 144         GArray *storage_list = NULL;
 145
 146         /* Get invalid storage_uuid (with storage_type)*/
 147         sql = sqlite3_mprintf("SELECT storage_uuid FROM '%q' WHERE validity = 0 AND storage_type=%d;", MEDIA_SVC_DB_TABLE_STORAGE, storage_type);
 148         ret = _media_svc_sql_prepare_to_step_simple(handle, sql, &sql_stmt);
 149         media_svc_retv_if(ret != MS_MEDIA_ERR_NONE, ret);
 150
 151         storage_list = g_array_new(false, false, sizeof(char *));
 152
 153         while (sqlite3_step(sql_stmt) == SQLITE_ROW) {
 154                 storage_id = g_strdup((const char *)sqlite3_column_text(sql_stmt, 0));
 155                 if (storage_id != NULL)
 156                         g_array_append_val(storage_list, storage_id);
 157
 158                 storage_id = NULL;
 159         }
 160
 161         SQLITE3_FINALIZE(sql_stmt);
 162
 163         while (storage_list->len > 0) {
 164                 storage_id = g_array_index(storage_list, char *, 0);
 165                 storage_list = g_array_remove_index(storage_list, 0);
 166
 167                 ret = ms_user_get_thumb_store_path(uid, storage_type, storage_id, &thumb_path);
 168                 if (ret != MS_MEDIA_ERR_NONE) {
 169                         media_svc_error("Fail to get thumbnail directory");
 170                         goto ERROR;
 171                 }
 172
 173                 ret = _media_svc_remove_dir(storage_type, thumb_path);
 174                 if (ret != MS_MEDIA_ERR_NONE) {
 175                         media_svc_error("Fail to remove thumbnail directory[%s]", thumb_path);
 176                         goto ERROR;
 177                 }
 178
 179                 /* remove media before drop table (for clear playlist, and tag table)*/
 180                 sql = sqlite3_mprintf("DELETE FROM '%q';DROP TABLE '%q';", storage_id);
 181                 ret = _media_svc_sql_query(sql, uid);
 182                 SQLITE3_SAFE_FREE(sql);
 183                 if (ret != MS_MEDIA_ERR_NONE) {
 184                         media_svc_error("Fail to drop table[%s]", storage_id);
 185                         goto ERROR;
 186                 }
 187
 188                 SAFE_FREE(storage_id);
 189         }
 190
 191         /* Update storage, folder table */
 192         sql = sqlite3_mprintf("DELETE FROM '%q' WHERE validity=0 AND storage_type=%d;DELETE FROM '%q' WHERE validity=0 AND storage_type=%d;",
 193                         MEDIA_SVC_DB_TABLE_STORAGE, storage_type, MEDIA_SVC_DB_TABLE_FOLDER, storage_type);
 194
 195         ret = _media_svc_sql_query(sql, uid);
 196         SQLITE3_SAFE_FREE(sql);
 197         if (ret != MS_MEDIA_ERR_NONE) {
 198                 media_svc_error("Fail to update storage table");
 199                 goto ERROR;
 200         }
 201
 202 ERROR:
 203         SAFE_FREE(storage_id);
 204         SAFE_FREE(thumb_path);
 205
 206         while (storage_list->len > 0) {
 207                 storage_id = g_array_index(storage_list, char *, 0);
 208                 storage_list = g_array_remove_index(storage_list, 0);
 209                 SAFE_FREE(storage_id);
 210         }
 211
 212         g_array_free(storage_list, false);
 213         storage_list = NULL;
 214
 215         return ret;
 216 }
 217
 218 int _media_svc_update_storage_validity(const char *storage_id, int validity, uid_t uid)
 219 {
 220         int ret = MS_MEDIA_ERR_NONE;
 221         char *sql = NULL;
 222
 223         if (storage_id == NULL)
 224                 sql = sqlite3_mprintf("UPDATE '%q' SET validity=%d WHERE storage_uuid!='media';", MEDIA_SVC_DB_TABLE_STORAGE, validity);
 225         else
 226                 sql = sqlite3_mprintf("UPDATE '%q' SET validity=%d WHERE storage_uuid=%Q;", MEDIA_SVC_DB_TABLE_STORAGE, validity, storage_id);
 227
 228         ret = _media_svc_sql_query(sql, uid);
 229         SQLITE3_SAFE_FREE(sql);
 230
 231         return ret;
 232 }
 233
 234 int _media_svc_get_storage_uuid(sqlite3 *handle, const char *path, char *storage_id, uid_t uid)
 235 {
 236         int ret = MS_MEDIA_ERR_NONE;
 237         sqlite3_stmt *sql_stmt = NULL;
 238         char *sql = NULL;
 239         char *storage_path = NULL;
 240         char *remain_path = NULL;
 241         int remain_len = 0;
 242         char *internal_path = NULL;
 243
 244         ret = ms_user_get_internal_root_path(uid, &internal_path);
 245         media_svc_retvm_if(ret != MS_MEDIA_ERR_NONE, ret, "Fail to get root path");
 246
 247         if (STRING_VALID(internal_path) && strncmp(path, internal_path, strlen(internal_path)) == 0) {
 248                 SAFE_STRLCPY(storage_id, MEDIA_SVC_DB_TABLE_MEDIA, MEDIA_SVC_UUID_SIZE+1);
 249                 SAFE_FREE(internal_path);
 250                 return MS_MEDIA_ERR_NONE;
 251         }
 252
 253         SAFE_FREE(internal_path);
 254
 255         remain_path = strstr(path + (STRING_VALID(MEDIA_ROOT_PATH_USB) ? strlen(MEDIA_ROOT_PATH_USB) : 0) + 1, "/");
 256         if (remain_path != NULL)
 257                 remain_len = strlen(remain_path);
 258
 259         storage_path = strndup(path, strlen(path) - remain_len);
 260
 261         sql = sqlite3_mprintf("SELECT storage_uuid FROM '%q' WHERE validity=1 AND storage_path = '%q'", MEDIA_SVC_DB_TABLE_STORAGE, storage_path);
 262
 263         ret = _media_svc_sql_prepare_to_step(handle, sql, &sql_stmt);
 264         SAFE_FREE(storage_path);
 265         media_svc_retv_if(ret != MS_MEDIA_ERR_NONE, ret);
 266
 267         if (STRING_VALID((const char *)sqlite3_column_text(sql_stmt, 0)))
 268                 SAFE_STRLCPY(storage_id, (const char *)sqlite3_column_text(sql_stmt, 0), MEDIA_SVC_UUID_SIZE+1);
 269
 270         SQLITE3_FINALIZE(sql_stmt);
 271
 272         if (!STRING_VALID(storage_id)) {
 273                 media_svc_error("Not found valid storage id [%s]", path);
 274                 ret = MS_MEDIA_ERR_INVALID_PARAMETER;
 275         }
 276
 277         return ret;
 278 }
 279
 280 int _media_svc_get_storage_type(sqlite3 *handle, const char *storage_id, media_svc_storage_type_e *storage_type)
 281 {
 282         int ret = MS_MEDIA_ERR_NONE;
 283         sqlite3_stmt *sql_stmt = NULL;
 284         char *sql = NULL;
 285
 286         media_svc_retvm_if(!STRING_VALID(storage_id), MS_MEDIA_ERR_INVALID_PARAMETER, "Invalid storage_id");
 287
 288         sql = sqlite3_mprintf("SELECT storage_type FROM '%q' WHERE storage_uuid=%Q", MEDIA_SVC_DB_TABLE_STORAGE, storage_id);
 289
 290         ret = _media_svc_sql_prepare_to_step(handle, sql, &sql_stmt);
 291
 292         if (ret != MS_MEDIA_ERR_NONE) {
 293                 if (ret == MS_MEDIA_ERR_DB_NO_RECORD)
 294                         media_svc_debug("there is no storage.");
 295                 else
 296                         media_svc_error("error when _media_svc_get_storage_type. err = [%d]", ret);
 297
 298                 return ret;
 299         }
 300
 301         *storage_type = sqlite3_column_int(sql_stmt, 0);
 302
 303         SQLITE3_FINALIZE(sql_stmt);
 304
 305         return ret;
 306 }
 307
 308 int _media_svc_get_storage_path(sqlite3 *handle, const char *storage_id, char **storage_path)
 309 {
 310         int ret = MS_MEDIA_ERR_NONE;
 311         sqlite3_stmt *sql_stmt = NULL;
 312         char *sql = NULL;
 313
 314         media_svc_retvm_if(!STRING_VALID(storage_id), MS_MEDIA_ERR_INVALID_PARAMETER, "Invalid storage_id");
 315
 316         sql = sqlite3_mprintf("SELECT storage_path FROM '%q' WHERE (storage_uuid=%Q AND validity=1)", MEDIA_SVC_DB_TABLE_STORAGE, storage_id);
 317
 318         ret = _media_svc_sql_prepare_to_step(handle, sql, &sql_stmt);
 319
 320         if (ret != MS_MEDIA_ERR_NONE) {
 321                 if (ret == MS_MEDIA_ERR_DB_NO_RECORD)
 322                         media_svc_debug("there is no storage.");
 323                 else
 324                         media_svc_error("error when _media_svc_get_storage_type. err = [%d]", ret);
 325
 326                 return ret;
 327         }
 328
 329         *storage_path = g_strdup((char *)sqlite3_column_text(sql_stmt, 0));
 330
 331         SQLITE3_FINALIZE(sql_stmt);
 332
 333         return ret;
 334 }
 335
 336 int _media_svc_get_storage_scan_status(sqlite3 *handle, const char *storage_id, media_svc_scan_status_type_e *scan_status)
 337 {
 338         int ret = MS_MEDIA_ERR_NONE;
 339         sqlite3_stmt *sql_stmt = NULL;
 340         char *sql = NULL;
 341
 342         if (!STRING_VALID(storage_id)) {
 343                 media_svc_error("Invalid storage_id");
 344                 return MS_MEDIA_ERR_INVALID_PARAMETER;
 345         }
 346
 347         sql = sqlite3_mprintf("SELECT scan_status FROM '%q' WHERE (storage_uuid=%Q AND validity=1)", MEDIA_SVC_DB_TABLE_STORAGE, storage_id);
 348
 349         ret = _media_svc_sql_prepare_to_step(handle, sql, &sql_stmt);
 350         if (ret != MS_MEDIA_ERR_NONE) {
 351                 if (ret == MS_MEDIA_ERR_DB_NO_RECORD)
 352                         media_svc_debug("there is no storage.");
 353                 else
 354                         media_svc_error("error when _media_svc_get_storage_scan_status. err = [%d]", ret);
 355
 356                 return ret;
 357         }
 358
 359         *scan_status = sqlite3_column_int(sql_stmt, 0);
 360
 361         SQLITE3_FINALIZE(sql_stmt);
 362
 363         return ret;
 364 }
 365
 366 int _media_svc_set_storage_scan_status(const char *storage_id, media_svc_scan_status_type_e scan_status, uid_t uid)
 367 {
 368         int ret = MS_MEDIA_ERR_NONE;
 369         char *sql = NULL;
 370
 371         if (storage_id == NULL)
 372                 sql = sqlite3_mprintf("UPDATE '%q' SET scan_status=%d WHERE storage_uuid != 'media';", MEDIA_SVC_DB_TABLE_STORAGE, scan_status);
 373         else
 374                 sql = sqlite3_mprintf("UPDATE '%q' SET scan_status=%d WHERE storage_uuid=%Q;", MEDIA_SVC_DB_TABLE_STORAGE, scan_status, storage_id);
 375
 376         ret = _media_svc_sql_query(sql, uid);
 377         SQLITE3_SAFE_FREE(sql);
 378
 379         return ret;
 380 }
 381
 382 static int __media_svc_count_all_storage(sqlite3 *handle, int *count)
 383 {
 384         int ret = MS_MEDIA_ERR_NONE;
 385         sqlite3_stmt *sql_stmt = NULL;
 386         char *sql = sqlite3_mprintf("SELECT count(*) FROM '%q' WHERE validity = 1", MEDIA_SVC_DB_TABLE_STORAGE);
 387
 388         ret = _media_svc_sql_prepare_to_step(handle, sql, &sql_stmt);
 389         if (ret != MS_MEDIA_ERR_NONE) {
 390                 media_svc_error("error when _media_svc_sql_prepare_to_step. err = [%d]", ret);
 391                 return ret;
 392         }
 393
 394         *count = sqlite3_column_int(sql_stmt, 0);
 395
 396         SQLITE3_FINALIZE(sql_stmt);
 397
 398         return MS_MEDIA_ERR_NONE;
 399 }
 400
 401 int _media_svc_get_all_storage(sqlite3 *handle, char ***storage_list, char ***storage_id_list, int **scan_status_list, int *count)
 402 {
 403         int ret = MS_MEDIA_ERR_NONE;
 404         int idx = 0;
 405         sqlite3_stmt *sql_stmt = NULL;
 406         char *sql = NULL;
 407         int cnt = 0;
 408
 409         ret = __media_svc_count_all_storage(handle, &cnt);
 410         if (ret != MS_MEDIA_ERR_NONE) {
 411                 media_svc_error("error when __media_svc_count_all_folders. err = [%d]", ret);
 412                 return ret;
 413         }
 414
 415         if (cnt > 0) {
 416                 sql = sqlite3_mprintf("SELECT storage_path, storage_uuid, scan_status FROM '%q' WHERE validity = 1", MEDIA_SVC_DB_TABLE_STORAGE);
 417         } else {
 418                 *storage_list = NULL;
 419                 *scan_status_list = NULL;
 420                 return MS_MEDIA_ERR_NONE;
 421         }
 422
 423         *storage_list = malloc(sizeof(char *) * cnt);
 424         *storage_id_list = malloc(sizeof(char *) * cnt);
 425         *scan_status_list = malloc(sizeof(int) * cnt);
 426         if (*storage_list == NULL || *storage_id_list == NULL || *scan_status_list == NULL) {
 427                 media_svc_error("Allocation failed");
 428                 SAFE_FREE(*storage_list);
 429                 SAFE_FREE(*storage_id_list);
 430                 SAFE_FREE(*scan_status_list);
 431                 SQLITE3_SAFE_FREE(sql);
 432                 return MS_MEDIA_ERR_OUT_OF_MEMORY;
 433         }
 434
 435         ret = _media_svc_sql_prepare_to_step(handle, sql, &sql_stmt);
 436         if (ret != MS_MEDIA_ERR_NONE) {
 437                 media_svc_error("prepare error [%s]", sqlite3_errmsg(handle));
 438                 SAFE_FREE(*storage_list);
 439                 SAFE_FREE(*storage_id_list);
 440                 SAFE_FREE(*scan_status_list);
 441                 return ret;
 442         }
 443
 444         media_svc_debug("QEURY OK");
 445
 446         while (1) {
 447                 (*storage_list)[idx] = g_strdup((char *)sqlite3_column_text(sql_stmt, 0));
 448                 (*storage_id_list)[idx] = g_strdup((char *)sqlite3_column_text(sql_stmt, 1));
 449                 (*scan_status_list)[idx] = (int)sqlite3_column_int(sql_stmt, 2);
 450                 if (sqlite3_step(sql_stmt) != SQLITE_ROW)
 451                         break;
 452                 idx++;
 453         }
 454
 455         if (cnt == idx + 1) {
 456                 *count = cnt;
 457                 media_svc_debug("OK");
 458         } else {
 459                 /* free all data */
 460                 int i = 0;
 461                 for (i = 0; i < idx; i++) {
 462                         SAFE_FREE((*storage_list)[i]);
 463                         SAFE_FREE((*storage_id_list)[i]);
 464                 }
 465                 SAFE_FREE(*storage_list);
 466                 SAFE_FREE(*storage_id_list);
 467                 SAFE_FREE(*scan_status_list);
 468                 *count = 0;
 469                 ret = MS_MEDIA_ERR_INTERNAL;
 470         }
 471
 472         SQLITE3_FINALIZE(sql_stmt);
 473
 474         return ret;
 475 }