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_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_path, storage_type) values (%Q, %Q, %d);",
  76                                                 MEDIA_SVC_DB_TABLE_STORAGE, storage_id, 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_invalid_storage(sqlite3 *handle, media_svc_storage_type_e storage_type, uid_t uid)
 124 {
 125         int ret = MS_MEDIA_ERR_NONE;
 126         char *sql = NULL;
 127         char *storage_id = NULL;
 128         char *thumb_path = NULL;
 129         sqlite3_stmt *sql_stmt = NULL;
 130         GArray *storage_list = NULL;
 131
 132         /* Get invalid storage_uuid (with storage_type)*/
 133         sql = sqlite3_mprintf("SELECT storage_uuid FROM '%q' WHERE validity = 0 AND storage_type=%d;", MEDIA_SVC_DB_TABLE_STORAGE, storage_type);
 134         ret = _media_svc_sql_prepare_to_step_simple(handle, sql, &sql_stmt);
 135         media_svc_retv_if(ret != MS_MEDIA_ERR_NONE, ret);
 136
 137         storage_list = g_array_new(false, false, sizeof(char *));
 138
 139         while (sqlite3_step(sql_stmt) == SQLITE_ROW) {
 140                 storage_id = g_strdup((const char *)sqlite3_column_text(sql_stmt, 0));
 141                 if (storage_id != NULL)
 142                         g_array_append_val(storage_list, storage_id);
 143
 144                 storage_id = NULL;
 145         }
 146
 147         SQLITE3_FINALIZE(sql_stmt);
 148
 149         while (storage_list->len > 0) {
 150                 storage_id = g_array_index(storage_list, char *, 0);
 151                 storage_list = g_array_remove_index(storage_list, 0);
 152
 153                 ret = ms_user_get_thumb_store_path(uid, storage_type, storage_id, &thumb_path);
 154                 if (ret != MS_MEDIA_ERR_NONE) {
 155                         media_svc_error("Fail to get thumbnail directory");
 156                         goto ERROR;
 157                 }
 158
 159                 ret = _media_svc_remove_dir(storage_type, thumb_path);
 160                 if (ret != MS_MEDIA_ERR_NONE) {
 161                         media_svc_error("Fail to remove thumbnail directory[%s]", thumb_path);
 162                         goto ERROR;
 163                 }
 164
 165                 /* remove media before drop table (for clear playlist, and tag table)*/
 166                 sql = sqlite3_mprintf("DELETE FROM '%q';DROP TABLE '%q';", storage_id);
 167                 ret = _media_svc_sql_query(sql, uid);
 168                 SQLITE3_SAFE_FREE(sql);
 169                 if (ret != MS_MEDIA_ERR_NONE) {
 170                         media_svc_error("Fail to drop table[%s]", storage_id);
 171                         goto ERROR;
 172                 }
 173
 174                 SAFE_FREE(storage_id);
 175         }
 176
 177         /* Update storage, folder table */
 178         sql = sqlite3_mprintf("DELETE FROM '%q' WHERE validity=0 AND storage_type=%d;DELETE FROM '%q' WHERE validity=0 AND storage_type=%d;",
 179                         MEDIA_SVC_DB_TABLE_STORAGE, storage_type, MEDIA_SVC_DB_TABLE_FOLDER, storage_type);
 180
 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 update storage table");
 185                 goto ERROR;
 186         }
 187
 188 ERROR:
 189         SAFE_FREE(storage_id);
 190         SAFE_FREE(thumb_path);
 191
 192         while (storage_list->len > 0) {
 193                 storage_id = g_array_index(storage_list, char *, 0);
 194                 storage_list = g_array_remove_index(storage_list, 0);
 195                 SAFE_FREE(storage_id);
 196         }
 197
 198         g_array_free(storage_list, false);
 199         storage_list = NULL;
 200
 201         return ret;
 202 }
 203
 204 int _media_svc_update_storage_validity(const char *storage_id, int validity, uid_t uid)
 205 {
 206         int ret = MS_MEDIA_ERR_NONE;
 207         char *sql = NULL;
 208
 209         if (storage_id == NULL)
 210                 sql = sqlite3_mprintf("UPDATE '%q' SET validity=%d WHERE storage_uuid!='media';", MEDIA_SVC_DB_TABLE_STORAGE, validity);
 211         else
 212                 sql = sqlite3_mprintf("UPDATE '%q' SET validity=%d WHERE storage_uuid=%Q;", MEDIA_SVC_DB_TABLE_STORAGE, validity, storage_id);
 213
 214         ret = _media_svc_sql_query(sql, uid);
 215         SQLITE3_SAFE_FREE(sql);
 216
 217         return ret;
 218 }
 219
 220 int _media_svc_get_storage_uuid(sqlite3 *handle, const char *path, char *storage_id, uid_t uid)
 221 {
 222         int ret = MS_MEDIA_ERR_NONE;
 223         sqlite3_stmt *sql_stmt = NULL;
 224         char *sql = NULL;
 225         char *storage_path = NULL;
 226         char *remain_path = NULL;
 227         int remain_len = 0;
 228         char *internal_path = NULL;
 229
 230         ret = ms_user_get_internal_root_path(uid, &internal_path);
 231         media_svc_retvm_if(ret != MS_MEDIA_ERR_NONE, ret, "Fail to get root path");
 232
 233         if (STRING_VALID(internal_path) && strncmp(path, internal_path, strlen(internal_path)) == 0) {
 234                 SAFE_STRLCPY(storage_id, MEDIA_SVC_DB_TABLE_MEDIA, MEDIA_SVC_UUID_SIZE+1);
 235                 SAFE_FREE(internal_path);
 236                 return MS_MEDIA_ERR_NONE;
 237         }
 238
 239         SAFE_FREE(internal_path);
 240
 241         remain_path = strstr(path + (STRING_VALID(MEDIA_ROOT_PATH_USB) ? strlen(MEDIA_ROOT_PATH_USB) : 0) + 1, "/");
 242         if (remain_path != NULL)
 243                 remain_len = strlen(remain_path);
 244
 245         storage_path = strndup(path, strlen(path) - remain_len);
 246
 247         sql = sqlite3_mprintf("SELECT storage_uuid FROM '%q' WHERE validity=1 AND storage_path = '%q'", MEDIA_SVC_DB_TABLE_STORAGE, storage_path);
 248
 249         ret = _media_svc_sql_prepare_to_step(handle, sql, &sql_stmt);
 250         SAFE_FREE(storage_path);
 251         media_svc_retv_if(ret != MS_MEDIA_ERR_NONE, ret);
 252
 253         if (STRING_VALID((const char *)sqlite3_column_text(sql_stmt, 0)))
 254                 SAFE_STRLCPY(storage_id, (const char *)sqlite3_column_text(sql_stmt, 0), MEDIA_SVC_UUID_SIZE+1);
 255
 256         SQLITE3_FINALIZE(sql_stmt);
 257
 258         if (!STRING_VALID(storage_id)) {
 259                 media_svc_error("Not found valid storage id [%s]", path);
 260                 ret = MS_MEDIA_ERR_INVALID_PARAMETER;
 261         }
 262
 263         return ret;
 264 }
 265
 266 int _media_svc_get_storage_path(sqlite3 *handle, const char *storage_id, char **storage_path)
 267 {
 268         int ret = MS_MEDIA_ERR_NONE;
 269         sqlite3_stmt *sql_stmt = NULL;
 270         char *sql = NULL;
 271
 272         media_svc_retvm_if(!STRING_VALID(storage_id), MS_MEDIA_ERR_INVALID_PARAMETER, "Invalid storage_id");
 273
 274         sql = sqlite3_mprintf("SELECT storage_path FROM '%q' WHERE (storage_uuid=%Q AND validity=1)", MEDIA_SVC_DB_TABLE_STORAGE, storage_id);
 275
 276         ret = _media_svc_sql_prepare_to_step(handle, sql, &sql_stmt);
 277
 278         if (ret != MS_MEDIA_ERR_NONE) {
 279                 if (ret == MS_MEDIA_ERR_DB_NO_RECORD)
 280                         media_svc_debug("there is no storage.");
 281                 else
 282                         media_svc_error("error when _media_svc_get_storage_type. err = [%d]", ret);
 283
 284                 return ret;
 285         }
 286
 287         *storage_path = g_strdup((char *)sqlite3_column_text(sql_stmt, 0));
 288
 289         SQLITE3_FINALIZE(sql_stmt);
 290
 291         return ret;
 292 }
 293
 294 static int __media_svc_count_all_storage(sqlite3 *handle, int *count)
 295 {
 296         int ret = MS_MEDIA_ERR_NONE;
 297         sqlite3_stmt *sql_stmt = NULL;
 298         char *sql = sqlite3_mprintf("SELECT count(*) FROM '%q' WHERE validity = 1", MEDIA_SVC_DB_TABLE_STORAGE);
 299
 300         ret = _media_svc_sql_prepare_to_step(handle, sql, &sql_stmt);
 301         if (ret != MS_MEDIA_ERR_NONE) {
 302                 media_svc_error("error when _media_svc_sql_prepare_to_step. err = [%d]", ret);
 303                 return ret;
 304         }
 305
 306         *count = sqlite3_column_int(sql_stmt, 0);
 307
 308         SQLITE3_FINALIZE(sql_stmt);
 309
 310         return MS_MEDIA_ERR_NONE;
 311 }
 312
 313 int _media_svc_get_all_storage(sqlite3 *handle, char ***storage_list, char ***storage_id_list, int *count)
 314 {
 315         int ret = MS_MEDIA_ERR_NONE;
 316         int idx = 0;
 317         sqlite3_stmt *sql_stmt = NULL;
 318         char *sql = NULL;
 319         int cnt = 0;
 320
 321         ret = __media_svc_count_all_storage(handle, &cnt);
 322         if (ret != MS_MEDIA_ERR_NONE) {
 323                 media_svc_error("error when __media_svc_count_all_folders. err = [%d]", ret);
 324                 return ret;
 325         }
 326
 327         if (cnt > 0) {
 328                 sql = sqlite3_mprintf("SELECT storage_path, storage_uuid FROM '%q' WHERE validity = 1", MEDIA_SVC_DB_TABLE_STORAGE);
 329         } else {
 330                 *storage_list = NULL;
 331                 return MS_MEDIA_ERR_NONE;
 332         }
 333
 334         *storage_list = malloc(sizeof(char *) * cnt);
 335         *storage_id_list = malloc(sizeof(char *) * cnt);
 336         if (*storage_list == NULL || *storage_id_list == NULL) {
 337                 media_svc_error("Allocation failed");
 338                 SAFE_FREE(*storage_list);
 339                 SAFE_FREE(*storage_id_list);
 340                 SQLITE3_SAFE_FREE(sql);
 341                 return MS_MEDIA_ERR_OUT_OF_MEMORY;
 342         }
 343
 344         ret = _media_svc_sql_prepare_to_step(handle, sql, &sql_stmt);
 345         if (ret != MS_MEDIA_ERR_NONE) {
 346                 media_svc_error("prepare error [%s]", sqlite3_errmsg(handle));
 347                 SAFE_FREE(*storage_list);
 348                 SAFE_FREE(*storage_id_list);
 349                 return ret;
 350         }
 351
 352         media_svc_debug("QEURY OK");
 353
 354         while (1) {
 355                 (*storage_list)[idx] = g_strdup((char *)sqlite3_column_text(sql_stmt, 0));
 356                 (*storage_id_list)[idx] = g_strdup((char *)sqlite3_column_text(sql_stmt, 1));
 357                 if (sqlite3_step(sql_stmt) != SQLITE_ROW)
 358                         break;
 359                 idx++;
 360         }
 361
 362         if (cnt == idx + 1) {
 363                 *count = cnt;
 364                 media_svc_debug("OK");
 365         } else {
 366                 /* free all data */
 367                 int i = 0;
 368                 for (i = 0; i < idx; i++) {
 369                         SAFE_FREE((*storage_list)[i]);
 370                         SAFE_FREE((*storage_id_list)[i]);
 371                 }
 372                 SAFE_FREE(*storage_list);
 373                 SAFE_FREE(*storage_id_list);
 374                 *count = 0;
 375                 ret = MS_MEDIA_ERR_INTERNAL;
 376         }
 377
 378         SQLITE3_FINALIZE(sql_stmt);
 379
 380         return ret;
 381 }