/*
 * Copyright (c) XMEDIA. All rights reserved.
 */
#include <linux/workqueue.h>
#include <linux/slab.h>
#include <linux/module.h>
#include "osal.h"

OSAL_LIST_HEAD(wq_list);
struct wq_node {
    struct osal_work_struct *osal_work;
    struct work_struct *work;
    struct osal_list_head node;
};

static struct osal_work_struct *osal_find_work(struct work_struct *work)
{
    struct osal_list_head *this = NULL;
    if (osal_list_empty(&wq_list)) {
        osal_printk("find work failed! wq_list is empty!\n");
        return NULL;
    }
    osal_list_for_each(this, &wq_list) {
        struct wq_node *ws = osal_list_entry(this, struct wq_node, node);
        if (ws->work == work) {
            return ws->osal_work;
        }
    }
    osal_printk("find work failed!\n");
    return NULL;
}

static int osal_del_work(struct work_struct *work)
{
    struct osal_list_head *this = NULL;
    struct osal_list_head *next = NULL;
    if (osal_list_empty(&wq_list)) {
        osal_printk("find work failed! wq_list is empty!\n");
        return -1;
    }
    osal_list_for_each_safe(this, next, &wq_list) {
        struct wq_node *ws = osal_list_entry(this, struct wq_node, node);
        if (ws->work == work) {
            osal_list_del(this);
            kfree(ws);
            return 0;
        }
    }
    osal_printk("del work failed!\n");
    return -1;
}

static void osal_work_func(struct work_struct *work)
{
    struct osal_work_struct *ow = osal_find_work(work);
    if ((ow != NULL) && (ow->func != NULL)) {
        ow->func(ow);
    }
}

int osal_init_work(struct osal_work_struct *work, osal_work_func_t func)
{
    struct work_struct *w = NULL;
    struct wq_node *w_node = NULL;
    w = kmalloc(sizeof(struct work_struct), GFP_ATOMIC);
    if (w == NULL) {
        osal_printk("osal_init_work kmalloc failed!\n");
        return -1;
    }

    w_node = kmalloc(sizeof(struct wq_node), GFP_ATOMIC);
    if (w_node == NULL) {
        osal_printk("osal_init_work kmalloc failed!\n");
        kfree(w);
        return -1;
    }
    INIT_WORK(w, osal_work_func);
    work->work = w;
    work->func = func;
    w_node->osal_work = work;
    w_node->work = w;
    osal_list_add(&(w_node->node), &wq_list);
    return 0;
}
EXPORT_SYMBOL(osal_init_work);

int osal_schedule_work(struct osal_work_struct *work)
{
    if ((work != NULL) && (work->work != NULL)) {
        return (int)schedule_work(work->work);
    } else {
        return (int)false;
    }
}
EXPORT_SYMBOL(osal_schedule_work);

void osal_destroy_work(struct osal_work_struct *work)
{
    if ((work != NULL) && (work->work != NULL)) {
        osal_del_work(work->work);
        kfree((struct work_struct *)work->work);
        work->work = NULL;
    }
}
EXPORT_SYMBOL(osal_destroy_work);