2 * media-thumbnail-server
4 * Copyright (c) 2000 - 2011 Samsung Electronics Co., Ltd. All rights reserved.
6 * Contact: Hyunjun Ko <zzoon.ko@samsung.com>
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
12 * http://www.apache.org/licenses/LICENSE-2.0
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.
22 #include "thumb-server-internal.h"
34 #define LOG_TAG "MEDIA_THUMBNAIL_SERVER"
36 static __thread char **arr_path;
37 static __thread uid_t *arr_uid;
38 static __thread int g_idx = 0;
39 static __thread int g_cur_idx = 0;
41 GMainLoop *g_thumb_server_mainloop; // defined in thumb-server.c as extern
43 gboolean _thumb_server_send_msg_to_agent(int msg_type);
44 void _thumb_daemon_stop_job();
46 gboolean _thumb_daemon_start_jobs(gpointer data)
50 _thumb_server_send_msg_to_agent(MS_MSG_THUMB_SERVER_READY);
55 void _thumb_daemon_finish_jobs(void)
57 sqlite3 *sqlite_db_handle = _media_thumb_db_get_handle();
59 if (sqlite_db_handle != NULL) {
60 _media_thumb_db_disconnect();
61 thumb_dbg("sqlite3 handle is alive. So disconnect to sqlite3");
64 g_main_loop_quit(g_thumb_server_mainloop);
69 int _thumb_daemon_mmc_status(void)
74 err = vconf_get_int(VCONFKEY_SYSMAN_MMC_STATUS, &status);
77 } else if (err == -1) {
78 thumb_err("vconf_get_int failed : %d", err);
80 thumb_err("vconf_get_int Unexpected error code: %d", err);
86 void _thumb_daemon_mmc_eject_vconf_cb(void *data)
91 thumb_warn("_thumb_daemon_vconf_cb called");
93 err = vconf_get_int(VCONFKEY_SYSMAN_MMC_STATUS, &status);
95 if (status == VCONFKEY_SYSMAN_MMC_REMOVED || status == VCONFKEY_SYSMAN_MMC_INSERTED_NOT_MOUNTED) {
96 thumb_warn("SD card is ejected or not mounted. So media-thumbnail-server stops jobs to extract all thumbnails");
98 _thumb_daemon_stop_job();
100 } else if (err == -1) {
101 thumb_err("vconf_get_int failed : %d", err);
103 thumb_err("vconf_get_int Unexpected error code: %d", err);
109 void _thumb_daemon_vconf_cb(void *data)
114 thumb_warn("_thumb_daemon_vconf_cb called");
116 err = vconf_get_int(VCONFKEY_SYSMAN_MMC_FORMAT, &status);
118 if (status == VCONFKEY_SYSMAN_MMC_FORMAT_COMPLETED) {
119 thumb_warn("SD card format is completed. So media-thumbnail-server stops jobs to extract all thumbnails");
121 _thumb_daemon_stop_job();
123 thumb_dbg("not completed");
125 } else if (err == -1) {
126 thumb_err("vconf_get_int failed : %d", err);
128 thumb_err("vconf_get_int Unexpected error code: %d", err);
134 void _thumb_daemon_stop_job()
139 thumb_warn("There are %d jobs in the queue. But all jobs will be stopped", g_idx - g_cur_idx);
141 for (i = g_cur_idx; i < g_idx; i++) {
142 path = arr_path[g_cur_idx++];
149 int _thumb_daemon_process_job(thumbMsg *req_msg, thumbMsg *res_msg, uid_t uid)
151 int err = MS_MEDIA_ERR_NONE;
153 err = _media_thumb_process(req_msg, res_msg, uid);
154 if (err != MS_MEDIA_ERR_NONE) {
155 if (req_msg->msg_type == THUMB_REQUEST_SAVE_FILE) {
156 thumb_err("_media_thumb_process is failed: %d", err);
157 res_msg->status = THUMB_FAIL;
159 thumb_warn("_media_thumb_process is failed: %d, So use default thumb", err);
160 res_msg->status = THUMB_SUCCESS;
163 res_msg->status = THUMB_SUCCESS;
169 static int __thumb_daemon_process_job_raw(thumbMsg *req_msg, thumbMsg *res_msg, thumbRawAddMsg *res_raw_msg, uid_t uid)
171 int err = MS_MEDIA_ERR_NONE;
173 err = _media_thumb_process_raw(req_msg, res_msg, res_raw_msg, uid);
174 if (err != MS_MEDIA_ERR_NONE) {
175 if (err != MS_MEDIA_ERR_FILE_NOT_EXIST) {
176 thumb_warn("_media_thumb_process is failed: %d, So use default thumb", err);
177 res_msg->status = THUMB_SUCCESS;
179 thumb_warn("_media_thumb_process is failed: %d, (file not exist) ", err);
180 res_msg->status = THUMB_FAIL;
183 res_msg->status = THUMB_SUCCESS;
189 int _thumb_daemon_all_extract(uid_t uid)
191 int err = MS_MEDIA_ERR_NONE;
193 char query_string[MAX_PATH_SIZE + 1] = { 0, };
194 char path[MAX_PATH_SIZE + 1] = { 0, };
195 sqlite3 *sqlite_db_handle = NULL;
196 sqlite3_stmt *sqlite_stmt = NULL;
198 err = _media_thumb_db_connect(uid);
199 if (err != MS_MEDIA_ERR_NONE) {
200 thumb_err("_media_thumb_db_connect failed: %d", err);
204 sqlite_db_handle = _media_thumb_db_get_handle();
205 if (sqlite_db_handle == NULL) {
206 thumb_err("sqlite handle is NULL");
207 return MS_MEDIA_ERR_INTERNAL;
210 if (_thumb_daemon_mmc_status() == VCONFKEY_SYSMAN_MMC_MOUNTED) {
211 snprintf(query_string, sizeof(query_string), SELECT_PATH_FROM_UNEXTRACTED_THUMB_MEDIA);
213 snprintf(query_string, sizeof(query_string), SELECT_PATH_FROM_UNEXTRACTED_THUMB_INTERNAL_MEDIA);
216 thumb_warn("Query: %s", query_string);
218 err = sqlite3_prepare_v2(sqlite_db_handle, query_string, strlen(query_string), &sqlite_stmt, NULL);
219 if (SQLITE_OK != err) {
220 thumb_err("prepare error [%s]", sqlite3_errmsg(sqlite_db_handle));
221 _media_thumb_db_disconnect();
222 return MS_MEDIA_ERR_INTERNAL;
226 err = sqlite3_step(sqlite_stmt);
227 if (err != SQLITE_ROW) {
228 thumb_dbg("end of row [%s]", sqlite3_errmsg(sqlite_db_handle));
232 strncpy(path, (const char *)sqlite3_column_text(sqlite_stmt, 0), sizeof(path));
233 path[sizeof(path) - 1] = '\0';
234 count = sqlite3_column_int(sqlite_stmt, 1);
236 thumb_dbg("Path : %s", path);
239 arr_path = (char**)malloc(sizeof(char*));
240 arr_uid = (uid_t*)malloc(sizeof(uid_t));
242 arr_path = (char**)realloc(arr_path, (g_idx + 1) * sizeof(char*));
243 arr_uid = (uid_t*)realloc(arr_uid, (g_idx + 1) * sizeof(uid_t));
245 arr_uid[g_idx] = uid;
246 arr_path[g_idx++] = strdup(path);
249 sqlite3_finalize(sqlite_stmt);
250 _media_thumb_db_disconnect();
252 return MS_MEDIA_ERR_NONE;
255 int _thumb_daemon_process_queue_jobs(gpointer data)
257 int err = MS_MEDIA_ERR_NONE;
261 if (g_cur_idx < g_idx) {
262 thumb_warn("There are %d jobs in the queue", g_idx - g_cur_idx);
263 thumb_dbg("Current idx : [%d]", g_cur_idx);
264 uid = arr_uid[g_cur_idx];
265 path = arr_path[g_cur_idx++];
267 thumbMsg recv_msg, res_msg;
268 memset(&recv_msg, 0x00, sizeof(thumbMsg));
269 memset(&res_msg, 0x00, sizeof(thumbMsg));
271 recv_msg.msg_type = THUMB_REQUEST_DB_INSERT;
272 recv_msg.thumb_type = MEDIA_THUMB_LARGE;
273 strncpy(recv_msg.org_path, path, sizeof(recv_msg.org_path));
274 recv_msg.org_path[sizeof(recv_msg.org_path) - 1] = '\0';
276 _thumb_daemon_process_job(&recv_msg, &res_msg,uid );
278 if (res_msg.status == THUMB_SUCCESS) {
280 err = _media_thumb_db_connect(uid);
281 if (err != MS_MEDIA_ERR_NONE) {
282 thumb_err("_media_thumb_mb_svc_connect failed: %d", err);
286 /* Need to update DB once generating thumb is done */
287 err = _media_thumb_update_db(recv_msg.org_path,
289 res_msg.origin_width,
290 res_msg.origin_height,
293 thumb_err("_media_thumb_update_db failed : %d", err);
296 _media_thumb_db_disconnect();
303 thumb_warn("Deleting array");
306 //_media_thumb_db_disconnect();
308 _thumb_server_send_msg_to_agent(MS_MSG_THUMB_EXTRACT_ALL_DONE); // MS_MSG_THUMB_EXTRACT_ALL_DONE
316 gboolean _thumb_server_read_socket(GIOChannel *src,
317 GIOCondition condition,
320 struct sockaddr_un client_addr;
321 unsigned int client_addr_len;
325 thumbRawAddMsg res_raw_msg;
330 memset((void *)&recv_msg, 0, sizeof(recv_msg));
331 memset((void *)&res_msg, 0, sizeof(res_msg));
332 memset((void *)&res_raw_msg, 0, sizeof(res_raw_msg));
334 sock = g_io_channel_unix_get_fd(src);
336 thumb_err("sock fd is invalid!");
340 header_size = sizeof(thumbMsg) - MAX_PATH_SIZE * 2 - sizeof(unsigned char *);
342 if (_media_thumb_recv_udp_msg(sock, header_size, &recv_msg, &client_addr, &client_addr_len) < 0) {
343 thumb_err("_media_thumb_recv_udp_msg failed");
347 thumb_warn("Received [%d] %s(%d) from PID(%d) \n", recv_msg.msg_type, recv_msg.org_path, strlen(recv_msg.org_path), recv_msg.pid);
349 if (recv_msg.msg_type == THUMB_REQUEST_ALL_MEDIA) {
350 thumb_dbg("All thumbnails are being extracted now");
351 _thumb_daemon_all_extract(recv_msg.uid);
352 g_idle_add(_thumb_daemon_process_queue_jobs, NULL);
353 } else if(recv_msg.msg_type == THUMB_REQUEST_RAW_DATA) {
354 __thumb_daemon_process_job_raw(&recv_msg, &res_msg, &res_raw_msg, recv_msg.uid);
355 } else if(recv_msg.msg_type == THUMB_REQUEST_KILL_SERVER) {
356 thumb_warn("received KILL msg from thumbnail agent.");
358 long start = thumb_get_debug_time();
360 _thumb_daemon_process_job(&recv_msg, &res_msg,recv_msg.uid);
362 long end = thumb_get_debug_time();
363 thumb_dbg("Time : %f (%s)", ((double)(end - start) / (double)CLOCKS_PER_SEC), recv_msg.org_path);
366 if(res_msg.msg_type == 0)
367 res_msg.msg_type = recv_msg.msg_type;
368 res_msg.request_id = recv_msg.request_id;
369 strncpy(res_msg.org_path, recv_msg.org_path, recv_msg.origin_path_size);
370 res_msg.origin_path_size = recv_msg.origin_path_size;
371 res_msg.thumb_data = (unsigned char *)"\0";
372 if(res_msg.msg_type != THUMB_RESPONSE_RAW_DATA) {
373 res_msg.dest_path_size = strlen(res_msg.dst_path) + 1;
374 res_msg.thumb_size = 1;
376 res_msg.dest_path_size = 1;
377 res_msg.dst_path[0] = '\0';
383 unsigned char *buf = NULL;
384 _media_thumb_set_buffer_for_response(&res_msg, &buf, &buf_size);
386 if (sendto(sock, buf, buf_size, 0, (struct sockaddr *)&client_addr, sizeof(client_addr)) != buf_size) {
387 thumb_stderror("sendto failed");
392 thumb_warn("Sent %s(%d)", res_msg.dst_path, strlen(res_msg.dst_path));
396 //Add sendto_raw_data
397 if(recv_msg.msg_type == THUMB_REQUEST_RAW_DATA) {
398 _media_thumb_set_add_raw_data_buffer(&res_raw_msg, &buf, &buf_size);
400 while(buf_size > 0) {
401 if (buf_size < 512) {
402 block_size = buf_size;
404 if (sendto(sock, buf + sent_size, block_size, 0, (struct sockaddr *)&client_addr, sizeof(client_addr)) != block_size) {
405 thumb_err("sendto failed: %s", strerror(errno));
409 sent_size += block_size;
410 buf_size -= block_size;
412 thumb_warn_slog("Sent [%s] additional data[%d]", res_msg.org_path, sent_size);
417 if(recv_msg.msg_type == THUMB_REQUEST_KILL_SERVER) {
418 thumb_warn("Shutting down...");
419 g_main_loop_quit(g_thumb_server_mainloop);
425 gboolean _thumb_server_send_msg_to_agent(int msg_type)
428 const char *serv_ip = "127.0.0.1";
429 struct sockaddr_un serv_addr;
430 ms_thumb_server_msg send_msg;
432 if (ms_ipc_create_client_socket(MS_PROTOCOL_UDP, MS_TIMEOUT_SEC_10, &sock, MS_THUMB_COMM_PORT) < 0) {
433 thumb_err("ms_ipc_create_server_socket failed");
437 memset(&serv_addr, 0, sizeof(serv_addr));
439 serv_addr.sun_family = AF_UNIX;
440 strcpy(serv_addr.sun_path, "/var/run/media-server/media_ipc_thumbcomm.socket");
442 send_msg.msg_type = msg_type;
445 if (connect(sock, &serv_addr, sizeof(serv_addr)) < 0) {
446 thumb_err("connect failed [%s]",strerror(errno));
451 if (send(sock, &send_msg, sizeof(ms_thumb_server_msg), 0) != sizeof(ms_thumb_server_msg)) {
452 thumb_err("sendto failed: %s\n", strerror(errno));
457 thumb_dbg("Sending msg to thumbnail agent[%d] is successful", send_msg.msg_type);
463 gboolean _thumb_server_prepare_socket(int *sock_fd)
466 unsigned short serv_port;
471 memset((void *)&recv_msg, 0, sizeof(recv_msg));
472 memset((void *)&res_msg, 0, sizeof(res_msg));
473 serv_port = MS_THUMB_DAEMON_PORT;
475 if (ms_ipc_create_server_socket(MS_PROTOCOL_UDP, serv_port, &sock) < 0) {
476 thumb_err("ms_ipc_create_server_socket failed");