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