Improve parent path validation function
[platform/core/multimedia/libmedia-service.git] / src / common / media-svc-media-folder.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 <glib/gstdio.h>
23 #include <media-util-err.h>
24 #include "media-svc-media-folder.h"
25 #include "media-svc-debug.h"
26 #include "media-svc-env.h"
27 #include "media-svc-util.h"
28 #include "media-svc-db-utils.h"
29
30 int _media_svc_get_folder_id_by_foldername(sqlite3 *handle, const char *storage_id, const char *folder_name, char *folder_id, uid_t uid)
31 {
32         int ret = MS_MEDIA_ERR_NONE;
33         sqlite3_stmt *sql_stmt = NULL;
34         char *sql = NULL;
35
36         sql = sqlite3_mprintf("SELECT folder_id FROM %q WHERE storage_uuid=%Q AND folder_path=%Q", MEDIA_SVC_DB_TABLE_FOLDER, storage_id, folder_name);
37
38         ret = _media_svc_sql_prepare_to_step(handle, sql, &sql_stmt);
39         if (ret != MS_MEDIA_ERR_NONE) {
40                 if (ret == MS_MEDIA_ERR_DB_NO_RECORD)
41                         media_svc_debug("No record");
42                 else
43                         media_svc_error("error when _media_svc_get_folder_id_by_foldername. err = [%d]", ret);
44
45                 return ret;
46         }
47
48         SAFE_STRLCPY(folder_id, (const char *)sqlite3_column_text(sql_stmt, 0), MEDIA_SVC_UUID_SIZE + 1);
49
50         SQLITE3_FINALIZE(sql_stmt);
51
52         return ret;
53 }
54
55 static int __media_svc_append_folder(bool is_direct, const char *storage_id, ms_user_storage_type_e storage_type, const char *folder_id, const char *folder_path, uid_t uid)
56 {
57         int ret = MS_MEDIA_ERR_NONE;
58         char *folder_name = NULL;
59         int folder_modified_date = 0;
60
61         folder_name = g_path_get_basename(folder_path);
62         folder_modified_date = _media_svc_get_file_time(folder_path);
63
64         /*Update Pinyin If Support Pinyin*/
65         char *folder_name_pinyin = NULL;
66         if (_media_svc_check_pinyin_support())
67                 _media_svc_get_pinyin_str(folder_name, &folder_name_pinyin);
68         /* Sometime SQLITE3 returns NO_RECORD, so need to consider conflict case.. */
69         char *sql = sqlite3_mprintf("INSERT OR IGNORE INTO %q (folder_id, folder_path, folder_name, storage_uuid, folder_storage_type, folder_modified_time, folder_name_pinyin) values (%Q, %Q, %Q, %Q, '%d', '%d', %Q); ",
70                                 MEDIA_SVC_DB_TABLE_FOLDER, folder_id, folder_path, folder_name, storage_id, storage_type, folder_modified_date, folder_name_pinyin);
71
72         if (is_direct)
73                 ret = _media_svc_sql_query_direct(sql, uid);
74         else
75                 ret = _media_svc_sql_query(sql, uid);
76
77         SQLITE3_SAFE_FREE(sql);
78
79         SAFE_FREE(folder_name);
80         SAFE_FREE(folder_name_pinyin);
81
82         return ret;
83 }
84
85 int _media_svc_update_folder_modified_time_by_folder_uuid(const char *folder_uuid, const char *folder_path, uid_t uid)
86 {
87         int ret = MS_MEDIA_ERR_NONE;
88         int modified_time = 0;
89
90         modified_time = _media_svc_get_file_time(folder_path);
91
92         char *sql = sqlite3_mprintf("UPDATE %q SET folder_modified_time=%d WHERE folder_id=%Q;", MEDIA_SVC_DB_TABLE_FOLDER, modified_time, folder_uuid);
93
94         ret = _media_svc_sql_query(sql, uid);
95         SQLITE3_SAFE_FREE(sql);
96
97         return ret;
98 }
99
100 static int __media_svc_get_and_append_parent_folder(sqlite3 *handle, bool is_direct, const char *storage_id, const char *path, ms_user_storage_type_e storage_type, char *folder_id, uid_t uid)
101 {
102         int ret = MS_MEDIA_ERR_NONE;
103         unsigned int next_pos = 0;
104         char *next = NULL;
105         char *dir_path = NULL;
106         const char *token = "/";
107         char *folder_uuid = NULL;
108         char tmp_folder_uuid[MEDIA_SVC_UUID_SIZE + 1] = {0, };
109         bool folder_search_end = false;
110         char *internal_path = NULL;
111
112         memset(tmp_folder_uuid, 0, sizeof(tmp_folder_uuid));
113         ret = ms_user_get_internal_root_path(uid, &internal_path);
114         media_svc_retvm_if(ret != MS_MEDIA_ERR_NONE, ret, "Fail to get root path");
115
116         if (STRING_VALID(internal_path) && (strncmp(path, internal_path, strlen(internal_path)) == 0))
117                 next_pos = strlen(internal_path);
118         else if (STRING_VALID(MEDIA_ROOT_PATH_EXTERNAL) && strncmp(path, MEDIA_ROOT_PATH_EXTERNAL, strlen(MEDIA_ROOT_PATH_EXTERNAL)) == 0)
119                 next_pos = strlen(MEDIA_ROOT_PATH_EXTERNAL) + 1;
120         else {
121                 media_svc_error("Invalid Path");
122                 media_svc_sec_error("Invalid Path [%s], internal_path [%s]", path, internal_path);
123                 SAFE_FREE(internal_path);
124                 return MS_MEDIA_ERR_INTERNAL;
125         }
126
127         SAFE_FREE(internal_path);
128
129         while (!folder_search_end) {
130                 next = strstr(path + next_pos, token);
131                 if (next != NULL) {
132                         next_pos = (next - path);
133                         dir_path = strndup(path, next_pos);
134                         next_pos++;
135                 } else {
136                         dir_path = strndup(path, strlen(path));
137                         folder_search_end = true;
138                         media_svc_sec_debug("End Path[%s]", dir_path);
139                 }
140
141                 ret = _media_svc_get_folder_id_by_foldername(handle, storage_id, dir_path, tmp_folder_uuid, uid);
142                 if (ret == MS_MEDIA_ERR_DB_NO_RECORD) {
143                         folder_uuid = _media_info_generate_uuid();
144                         if (folder_uuid == NULL) {
145                                 media_svc_error("Invalid UUID");
146                                 SAFE_FREE(dir_path);
147                                 return MS_MEDIA_ERR_INTERNAL;
148                         }
149
150                         ret = __media_svc_append_folder(is_direct, storage_id, storage_type, folder_uuid, dir_path, uid);
151                         if (ret != MS_MEDIA_ERR_NONE)
152                                 media_svc_error("__media_svc_append_folder is failed");
153
154                         media_svc_sec_debug("Append new folder path[%s] uuid[%s]", dir_path, folder_uuid);
155                 }
156
157                 SAFE_FREE(dir_path);
158         }
159
160         if (STRING_VALID(folder_uuid)) {
161                 SAFE_STRLCPY(folder_id, folder_uuid, MEDIA_SVC_UUID_SIZE + 1);
162         } else {
163                 media_svc_error("Fail to get folder_uuid");
164                 return MS_MEDIA_ERR_INTERNAL;
165         }
166
167         return MS_MEDIA_ERR_NONE;
168 }
169
170 int _media_svc_get_and_append_folder(sqlite3 *handle, bool is_direct, const char *storage_id, const char *path, ms_user_storage_type_e storage_type, char *folder_id, uid_t uid)
171 {
172         int ret = MS_MEDIA_ERR_NONE;
173
174         ret = _media_svc_get_folder_id_by_foldername(handle, storage_id, path, folder_id, uid);
175         if (ret == MS_MEDIA_ERR_DB_NO_RECORD)
176                 ret = __media_svc_get_and_append_parent_folder(handle, is_direct, storage_id, path, storage_type, folder_id, uid);
177
178         return ret;
179 }
180
181 int _media_svc_get_and_append_folder_id_by_path(sqlite3 *handle, bool is_direct, const char *storage_id, const char *path, ms_user_storage_type_e storage_type, char *folder_id, uid_t uid)
182 {
183         char *dir_path = NULL;
184         int ret = MS_MEDIA_ERR_NONE;
185
186         dir_path = g_path_get_dirname(path);
187
188         ret = _media_svc_get_and_append_folder(handle, is_direct, storage_id, dir_path, storage_type, folder_id, uid);
189
190         SAFE_FREE(dir_path);
191
192         return ret;
193 }
194
195 int _media_svc_get_and_append_folder_id_by_folder_path(sqlite3 *handle, const char *storage_id, const char *path, ms_user_storage_type_e storage_type, uid_t uid)
196 {
197         char *path_name = NULL;
198         int ret = MS_MEDIA_ERR_NONE;
199         char *sql = NULL;
200         char folder_id[MEDIA_SVC_UUID_SIZE + 1] = {0, };
201
202         path_name = strdup(path);
203         if (path_name == NULL) {
204                 media_svc_error("out of memory");
205                 return MS_MEDIA_ERR_OUT_OF_MEMORY;
206         }
207
208         ret = _media_svc_get_folder_id_by_foldername(handle, storage_id, path_name, folder_id, uid);
209         if (ret == MS_MEDIA_ERR_DB_NO_RECORD) {
210                 ret = __media_svc_get_and_append_parent_folder(handle, true, storage_id, path_name, storage_type, folder_id, uid);
211         } else {
212                 sql = sqlite3_mprintf("UPDATE %q SET validity=1 WHERE storage_uuid = '%q' AND folder_path = '%q';", MEDIA_SVC_DB_TABLE_FOLDER, storage_id, path);
213                 ret = _media_svc_sql_query_direct(sql, uid);
214                 SQLITE3_SAFE_FREE(sql);
215         }
216
217         SAFE_FREE(path_name);
218
219         return ret;
220 }
221
222 int _media_svc_set_folder_validity(sqlite3 *handle, bool is_direct, const char *storage_id, const char *start_path, int validity, bool is_recursive, uid_t uid)
223 {
224         int ret = MS_MEDIA_ERR_NONE;
225         char *sql = NULL;
226
227         if (is_recursive) {
228                 sql = sqlite3_mprintf("UPDATE %q SET validity=%d WHERE storage_uuid='%q' AND (folder_path LIKE '%q/%%' OR folder_path='%q');",
229                                                 MEDIA_SVC_DB_TABLE_FOLDER, validity, storage_id, start_path, start_path);
230         } else {
231                 sql = sqlite3_mprintf("UPDATE %q SET validity=%d WHERE storage_uuid='%q' AND folder_path='%q';",
232                                                 MEDIA_SVC_DB_TABLE_FOLDER, validity, storage_id, start_path);
233         }
234
235         if (is_direct)
236                 ret = _media_svc_sql_query_direct(sql, uid);
237         else
238                 ret = _media_svc_sql_query(sql, uid);
239
240         SQLITE3_SAFE_FREE(sql);
241
242         return ret;
243 }
244
245 int _media_svc_count_folder_with_path(sqlite3 *handle, const char *storage_id, const char *path, int *count)
246 {
247         int ret = MS_MEDIA_ERR_NONE;
248         sqlite3_stmt *sql_stmt = NULL;
249         char *sql = NULL;
250
251         media_svc_retvm_if(!STRING_VALID(storage_id), MS_MEDIA_ERR_INVALID_PARAMETER, "storage_id is NULL");
252         media_svc_retvm_if(!STRING_VALID(path), MS_MEDIA_ERR_INVALID_PARAMETER, "path is NULL");
253
254         sql = sqlite3_mprintf("SELECT count(*) FROM %q WHERE (storage_uuid='%q' AND folder_path='%q')", MEDIA_SVC_DB_TABLE_FOLDER, storage_id, path);
255
256         ret = _media_svc_sql_prepare_to_step(handle, sql, &sql_stmt);
257
258         media_svc_retv_if(ret != MS_MEDIA_ERR_NONE, ret);
259
260         *count = sqlite3_column_int(sql_stmt, 0);
261
262         SQLITE3_FINALIZE(sql_stmt);
263
264         return MS_MEDIA_ERR_NONE;
265 }