/*
 * libmedia-thumbnail
 *
 * Copyright (c) 2000 - 2011 Samsung Electronics Co., Ltd. All rights reserved.
 *
 * Contact: Hyunjun Ko <zzoon.ko@samsung.com>
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 */

#include "media-thumbnail.h"
#include "media-thumb-ipc.h"
#include "media-thumb-util.h"
#include "media-thumb-db.h"
#include "media-thumb-debug.h"
#include <glib.h>
#include <fcntl.h>
#include <string.h>
#include <errno.h>
#include <grp.h>
#include <pwd.h>

#define THUMB_IPC_PATH tzplatform_mkpath(TZ_SYS_RUN, "media-server/media_ipc_thumbcreator.socket")

static GQueue *g_manage_queue = NULL;
static GQueue *g_manage_raw_queue = NULL;


typedef struct {
	GIOChannel *channel;
	int msg_type;
	unsigned int request_id;
	bool isCanceled;
	bool isRequested;
	int source_id;
	uid_t uid;
	char *path;
	thumbUserData *userData;
} thumbReq;

typedef struct {
	GIOChannel *channel;
	int msg_type;
	bool isCanceled;
	bool isRequested;
	int request_id;
	int source_id;
	unsigned int width;
	unsigned int height;
	uid_t uid;
	char *path;
	thumbRawUserData *userData;
} thumbRawReq;

static int __media_thumb_send_request(void);
static int __media_thumb_raw_data_send_request(void);

static int __media_thumb_cancel(unsigned int request_id)
{
	int len = 0, i;
	thumbReq *req = NULL;

	thumb_retv_if(!g_manage_queue, MS_MEDIA_ERR_NONE);

	len = g_queue_get_length(g_manage_queue);

	for (i = 0; i < len; i++) {
		req = (thumbReq *)g_queue_peek_nth(g_manage_queue, i);
		if (!req)
			continue;

		if (req->request_id == request_id) {
			if (req->isRequested) {
				req->isCanceled = true;
			} else {
				g_queue_pop_nth(g_manage_queue, i);

				SAFE_FREE(req->path);
				SAFE_FREE(req->userData);
				SAFE_FREE(req);
			}

			return MS_MEDIA_ERR_NONE;
		}
	}

	return MS_MEDIA_ERR_INTERNAL;
}

static int __media_thumb_cancel_raw_data(int request_id)
{
	int len = 0, i;
	thumbRawReq *req = NULL;

	thumb_retv_if(!g_manage_raw_queue, MS_MEDIA_ERR_NONE);

	len = g_queue_get_length(g_manage_raw_queue);

	for (i = 0; i < len; i++) {
		req = (thumbRawReq *)g_queue_peek_nth(g_manage_raw_queue, i);
		if (!req)
			continue;

		if (req->request_id == request_id) {
			if (req->isRequested) {
				req->isCanceled = true;
			} else {
				g_queue_pop_nth(g_manage_raw_queue, i);

				SAFE_FREE(req->path);
				SAFE_FREE(req->userData);
				SAFE_FREE(req);
			}

			return MS_MEDIA_ERR_NONE;
		}
	}

	return MS_MEDIA_ERR_INTERNAL;
}

static bool __media_thumb_is_canceled(void)
{
	thumbReq *req = (thumbReq *)g_queue_peek_head(g_manage_queue);

	if (!req)
		return true;

	return req->isCanceled;
}

static bool __media_thumb_is_canceled_for_raw(void)
{
	thumbRawReq *req = (thumbRawReq *)g_queue_peek_head(g_manage_raw_queue);

	if (!req)
		return true;

	return req->isCanceled;
}

static int __media_thumb_recv_msg(int sock, int header_size, thumbMsg *msg)
{
	int remain_size = 0;
	int recv_len = 0;
	int recv_pos = 0;
	unsigned char *buf = NULL;

	THUMB_MALLOC(buf, header_size);
	thumb_retv_if(!buf, MS_MEDIA_ERR_OUT_OF_MEMORY);

	while (header_size > 0) {
		if ((recv_len = recv(sock, buf + recv_pos, header_size, 0)) < 0) {
			thumb_stderror("recv failed");
			SAFE_FREE(buf);
			return MS_MEDIA_ERR_IPC;
		}
		header_size -= recv_len;
		recv_pos += recv_len;
	}

	header_size = recv_pos;
	recv_pos = 0;

	memcpy(msg, buf, header_size);
	SAFE_FREE(buf);

	thumb_retvm_if(strlen(msg->org_path) == 0 || strlen(msg->org_path) >= MAX_FILEPATH_LEN, MS_MEDIA_ERR_IPC, "Invalid org_path");
	thumb_retvm_if(strlen(msg->dst_path) >= MAX_FILEPATH_LEN, MS_MEDIA_ERR_IPC, "Invalid dst_path");
	thumb_retvm_if(msg->thumb_size < 0, MS_MEDIA_ERR_IPC, "Invalid thumb_size");
	thumb_retv_if(msg->thumb_size == 0, MS_MEDIA_ERR_NONE);

	remain_size = msg->thumb_size;
	THUMB_MALLOC(buf, remain_size);
	thumb_retv_if(!buf, MS_MEDIA_ERR_OUT_OF_MEMORY);

	while (remain_size > 0) {
		if ((recv_len = recv(sock, buf + recv_pos, remain_size, 0)) < 0) {
			thumb_stderror("recv failed");
			SAFE_FREE(buf);
			return MS_MEDIA_ERR_IPC;
		}

		fsync(sock);
		remain_size -= recv_len;
		recv_pos += recv_len;
	}

	SAFE_FREE(msg->thumb_data);
	THUMB_MALLOC(msg->thumb_data, msg->thumb_size);
	if (msg->thumb_data) {
		memcpy(msg->thumb_data, buf, msg->thumb_size);
	} else {
		SAFE_FREE(buf);
		return MS_MEDIA_ERR_OUT_OF_MEMORY;
	}

	SAFE_FREE(buf);

	return MS_MEDIA_ERR_NONE;
}

int _media_thumb_set_buffer(thumbMsg *req_msg, unsigned char **buf, int *buf_size)
{
	if (req_msg == NULL || buf == NULL)
		return MS_MEDIA_ERR_INVALID_PARAMETER;

	int thumb_data_len = 0;
	int size = 0;
	int header_size = 0;

	header_size = sizeof(thumbMsg) - sizeof(unsigned char *);
	thumb_data_len = req_msg->thumb_size;
	if (thumb_data_len < 0)
		return MS_MEDIA_ERR_INVALID_PARAMETER;

	size = header_size + thumb_data_len;
	THUMB_MALLOC(*buf, size);
	if (*buf == NULL) {
		*buf_size = 0;
		return 0;
	}
	memcpy(*buf, req_msg, header_size);
	if (thumb_data_len > 0)
		memcpy((*buf) + header_size, req_msg->thumb_data, thumb_data_len);

	*buf_size = size;

	return MS_MEDIA_ERR_NONE;
}

static void __media_thumb_pop(void)
{
	int len = 0;
	int sock = 0;
	thumbReq *req = NULL;
	GSource *source_id = NULL;

	if (!g_manage_queue)
		return;

	req = (thumbReq *)g_queue_pop_head(g_manage_queue);
	if (req) {
		source_id = g_main_context_find_source_by_id(g_main_context_get_thread_default(), req->source_id);
		sock = g_io_channel_unix_get_fd(req->channel);

		g_io_channel_shutdown(req->channel, TRUE, NULL);
		g_io_channel_unref(req->channel);
		close(sock);
		if (source_id)
			g_source_destroy(source_id);

		SAFE_FREE(req->path);
		SAFE_FREE(req->userData);
		SAFE_FREE(req);
	}

	/* Check manage queue */
	len = g_queue_get_length(g_manage_queue);
	if (len > 0)
		__media_thumb_send_request();
}

static void __media_thumb_pop_raw_data(void)
{
	int len = 0;
	int sock = 0;
	thumbRawReq *req = NULL;
	GSource *source_id = NULL;

	if (!g_manage_raw_queue)
		return;

	req = (thumbRawReq *)g_queue_pop_head(g_manage_raw_queue);
	if (req) {
		source_id = g_main_context_find_source_by_id(g_main_context_get_thread_default(), req->source_id);
		sock = g_io_channel_unix_get_fd(req->channel);

		g_io_channel_shutdown(req->channel, TRUE, NULL);
		g_io_channel_unref(req->channel);
		close(sock);
		if (source_id)
			g_source_destroy(source_id);

		SAFE_FREE(req->path);
		SAFE_FREE(req->userData);
		SAFE_FREE(req);
	}

	/* Check manage queue */
	len = g_queue_get_length(g_manage_raw_queue);
	if (len > 0)
		__media_thumb_raw_data_send_request();
}

static gboolean __media_thumb_write_socket(GIOChannel *src, GIOCondition condition, gpointer data)
{
	thumbMsg recv_msg;
	int header_size = 0;
	int sock = 0;
	int err = MS_MEDIA_ERR_NONE;
	thumbUserData *userdata = NULL;

	memset((void *)&recv_msg, 0, sizeof(thumbMsg));
	sock = g_io_channel_unix_get_fd(src);

	header_size = sizeof(thumbMsg) - sizeof(unsigned char *);

	if ((err = __media_thumb_recv_msg(sock, header_size, &recv_msg)) < 0) {
		thumb_err("_media_thumb_recv_msg failed ");
		goto NEXT;
	}

	if (recv_msg.status != MS_MEDIA_ERR_NONE) {
		err = recv_msg.status;
		thumb_err("Failed to make thumbnail (%d)", err);
	}

	if (__media_thumb_is_canceled() || !data)
		goto NEXT;

	userdata = (thumbUserData*)data;
	if (userdata->func)
		userdata->func(err, recv_msg.dst_path, userdata->user_data);

	thumb_dbg("Done");
NEXT:
	__media_thumb_pop();

	return G_SOURCE_REMOVE;
}

static gboolean __media_thumb_raw_data_write_socket(GIOChannel *src, GIOCondition condition, gpointer data)
{
	thumbMsg recv_msg;
	int header_size = 0;
	int sock = 0;
	int err = MS_MEDIA_ERR_NONE;
	thumbRawUserData *userdata = NULL;

	memset((void *)&recv_msg, 0, sizeof(thumbMsg));
	sock = g_io_channel_unix_get_fd(src);

	header_size = sizeof(thumbMsg) - sizeof(unsigned char *);

	thumb_err("_media_thumb_write_socket socket : %d", sock);

	if ((err = __media_thumb_recv_msg(sock, header_size, &recv_msg)) < 0) {
		thumb_err("_media_thumb_recv_msg failed ");
		goto NEXT;
	}

	if (recv_msg.status != MS_MEDIA_ERR_NONE) {
		err = recv_msg.status;
		thumb_err("Failed to make thumbnail (%d)", err);
	}

	if (__media_thumb_is_canceled_for_raw() || !data)
		goto NEXT;

	userdata = (thumbRawUserData*)data;
	if (userdata->func)
		userdata->func(err, recv_msg.request_id, recv_msg.org_path, recv_msg.thumb_width, recv_msg.thumb_height, recv_msg.thumb_data, recv_msg.thumb_size, userdata->user_data);

	thumb_dbg("Done");
NEXT:
	__media_thumb_pop_raw_data();

	return G_SOURCE_REMOVE;
}

static int __media_thumb_send_request(void)
{
	int err = MS_MEDIA_ERR_NONE;
	int sock = -1;
	struct sockaddr_un serv_addr;
	thumbReq *req_manager = NULL;
	int pid;

	err = ms_ipc_create_client_socket(MS_TIMEOUT_SEC_10, &sock);
	thumb_retvm_if(err != MS_MEDIA_ERR_NONE, err, "ms_ipc_create_client_socket failed");

	memset(&serv_addr, 0, sizeof(serv_addr));
	serv_addr.sun_family = AF_UNIX;
	SAFE_STRLCPY(serv_addr.sun_path, THUMB_IPC_PATH, sizeof(serv_addr.sun_path));

	GIOChannel *channel = NULL;
	channel = g_io_channel_unix_new(sock);
	int source_id = -1;

	/* Connecting to the thumbnail server */
	if (connect(sock, (struct sockaddr*)&serv_addr, sizeof(serv_addr)) < 0) {
		thumb_stderror("connect");
		if (errno == EACCES)
			err = MS_MEDIA_ERR_PERMISSION_DENIED;
		else
			err = MS_MEDIA_ERR_IPC;

		g_io_channel_shutdown(channel, TRUE, NULL);
		g_io_channel_unref(channel);
		close(sock);

		return err;
	}

	req_manager = (thumbReq *)g_queue_peek_head(g_manage_queue);
	if (!req_manager) {
		thumb_err("queue peek fail");
		g_io_channel_shutdown(channel, TRUE, NULL);
		g_io_channel_unref(channel);
		close(sock);
		return MS_MEDIA_ERR_INVALID_PARAMETER;
	}

	GSource *source = NULL;
	source = g_io_create_watch(channel, G_IO_IN);
	g_source_set_callback(source, (GSourceFunc)__media_thumb_write_socket, req_manager->userData, NULL);
	source_id = g_source_attach(source, g_main_context_get_thread_default());

	thumbMsg req_msg;
	memset((void *)&req_msg, 0, sizeof(thumbMsg));

	pid = getpid();
	req_msg.pid = pid;
	req_msg.msg_type = req_manager->msg_type;
	req_msg.request_id = 0;
	req_msg.uid = req_manager->uid;
	SAFE_STRLCPY(req_msg.org_path, req_manager->path, sizeof(req_msg.org_path));
	req_msg.dst_path[0] = '\0';
	req_msg.thumb_size = 0;

	unsigned char *buf = NULL;
	int buf_size = 0;
	_media_thumb_set_buffer(&req_msg, &buf, &buf_size);

	if (send(sock, buf, buf_size, 0) < 0) {
		thumb_err("send failed: %d", errno);
		SAFE_FREE(buf);
		g_source_destroy(g_main_context_find_source_by_id(g_main_context_get_thread_default(), source_id));
		g_io_channel_shutdown(channel, TRUE, NULL);
		g_io_channel_unref(channel);
		close(sock);
		return MS_MEDIA_ERR_IPC;
	}

	SAFE_FREE(buf);
	thumb_dbg("Sending msg to thumbnail daemon is successful");

	if (req_manager->msg_type == THUMB_REQUEST_DB_INSERT) {
		req_manager->channel = channel;
		req_manager->source_id = source_id;
		req_manager->isRequested = true;
	}

	return err;
}

static int __media_thumb_raw_data_send_request(void)
{
	int err = MS_MEDIA_ERR_NONE;
	int sock = -1;
	struct sockaddr_un serv_addr;
	thumbRawReq *req_manager = NULL;
	int pid;

	err = ms_ipc_create_client_socket(MS_TIMEOUT_SEC_10, &sock);
	thumb_retvm_if(err != MS_MEDIA_ERR_NONE, err, "ms_ipc_create_client_socket failed");

	memset(&serv_addr, 0, sizeof(serv_addr));
	serv_addr.sun_family = AF_UNIX;
	SAFE_STRLCPY(serv_addr.sun_path, THUMB_IPC_PATH, sizeof(serv_addr.sun_path));

	GIOChannel *channel = NULL;
	channel = g_io_channel_unix_new(sock);
	int source_id = -1;

	/* Connecting to the thumbnail server */
	if (connect(sock, (struct sockaddr*)&serv_addr, sizeof(serv_addr)) < 0) {
		thumb_stderror("connect error");
		if (errno == EACCES)
			err = MS_MEDIA_ERR_PERMISSION_DENIED;
		else
			err = MS_MEDIA_ERR_IPC;

		g_io_channel_shutdown(channel, TRUE, NULL);
		g_io_channel_unref(channel);
		close(sock);
		return err;
	}

	req_manager = (thumbRawReq *)g_queue_peek_head(g_manage_raw_queue);

	if (!req_manager) {
		thumb_err("queue peek fail");
		g_io_channel_shutdown(channel, TRUE, NULL);
		g_io_channel_unref(channel);
		close(sock);
		return MS_MEDIA_ERR_INVALID_PARAMETER;
	}

	GSource *source = NULL;
	source = g_io_create_watch(channel, G_IO_IN);
	g_source_set_callback(source, (GSourceFunc)__media_thumb_raw_data_write_socket, req_manager->userData, NULL);
	source_id = g_source_attach(source, g_main_context_get_thread_default());

	thumbMsg req_msg;
	memset((void *)&req_msg, 0, sizeof(thumbMsg));

	pid = getpid();
	req_msg.pid = pid;
	req_msg.msg_type = req_manager->msg_type;
	req_msg.request_id = req_manager->request_id;
	req_msg.thumb_width = req_manager->width;
	req_msg.thumb_height = req_manager->height;
	req_msg.uid = req_manager->uid;

	SAFE_STRLCPY(req_msg.org_path, req_manager->path, sizeof(req_msg.org_path));
	req_msg.dst_path[0] = '\0';
	req_msg.thumb_size = 0;

	unsigned char *buf = NULL;
	int buf_size = 0;
	_media_thumb_set_buffer(&req_msg, &buf, &buf_size);

	if (send(sock, buf, buf_size, 0) < 0) {
		thumb_err("send failed: %d", errno);
		SAFE_FREE(buf);
		g_source_destroy(g_main_context_find_source_by_id(g_main_context_get_thread_default(), source_id));
		g_io_channel_shutdown(channel, TRUE, NULL);
		g_io_channel_unref(channel);
		return MS_MEDIA_ERR_IPC;
	}

	SAFE_FREE(buf);

	if (req_manager->msg_type == THUMB_REQUEST_RAW_DATA) {
		req_manager->channel = channel;
		req_manager->source_id = source_id;
		req_manager->isRequested = true;
	}
	return MS_MEDIA_ERR_NONE;
}

int _media_thumb_request_async(int msg_type, unsigned int request_id, const char *origin_path, thumbUserData *userData, uid_t uid)
{
	int err = MS_MEDIA_ERR_NONE;
	int len = 0;

	if (msg_type == THUMB_REQUEST_CANCEL_MEDIA)
		return __media_thumb_cancel(request_id);

	if (g_manage_queue == NULL)
		g_manage_queue = g_queue_new();

	thumbReq *thumb_req = NULL;
	THUMB_MALLOC(thumb_req, sizeof(thumbReq));
	thumb_retvm_if(thumb_req == NULL, MS_MEDIA_ERR_INVALID_PARAMETER, "Failed to create request element");

	thumb_req->msg_type = msg_type;
	thumb_req->path = g_strdup(origin_path);
	thumb_req->userData = userData;
	thumb_req->isCanceled = false;
	thumb_req->isRequested = false;
	thumb_req->request_id = request_id;
	thumb_req->uid = uid;

	len = g_queue_get_length(g_manage_queue);
	g_queue_push_tail(g_manage_queue, (gpointer)thumb_req);

	if (len == 0)
		err = __media_thumb_send_request();

	return err;
}

int _media_thumb_request_raw_data_async(int msg_type, int request_id, const char *origin_path, int width, int height, thumbRawUserData *userData, uid_t uid)
{
	int err = MS_MEDIA_ERR_NONE;
	int len = 0;
	thumbRawReq *thumb_req = NULL;

	if (msg_type == THUMB_REQUEST_CANCEL_RAW_DATA)
		return __media_thumb_cancel_raw_data(request_id);

	if (g_manage_raw_queue == NULL)
		g_manage_raw_queue = g_queue_new();

	THUMB_MALLOC(thumb_req, sizeof(thumbRawReq));
	if (thumb_req == NULL) {
		thumb_err("Failed to create request element");
		return MS_MEDIA_ERR_INVALID_PARAMETER;
	}

	thumb_req->msg_type = msg_type;
	thumb_req->request_id = request_id;
	thumb_req->path = g_strdup(origin_path);
	thumb_req->width = width;
	thumb_req->height = height;
	thumb_req->userData = userData;
	thumb_req->isCanceled = false;
	thumb_req->isRequested = false;
	thumb_req->uid = uid;

	len = g_queue_get_length(g_manage_raw_queue);
	g_queue_push_tail(g_manage_raw_queue, (gpointer)thumb_req);

	if (len == 0)
		err = __media_thumb_raw_data_send_request();

	return err;
}