Subversion Repositories freemyipod

//
//
//    Copyright 2010 TheSeven
//
//
//    This file is part of emCORE.
//
//    emCORE is free software: you can redistribute it and/or
//    modify it under the terms of the GNU General Public License as
//    published by the Free Software Foundation, either version 2 of the
//    License, or (at your option) any later version.
//
//    emCORE is distributed in the hope that it will be useful,
//    but WITHOUT ANY WARRANTY; without even the implied warranty of
//    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
//    See the GNU General Public License for more details.
//
//    You should have received a copy of the GNU General Public License along
//    with emCORE.  If not, see <http://www.gnu.org/licenses/>.
//
//


#include "global.h"
#include "library.h"
#include "malloc.h"
#include "execimage.h"
#include "thread.h"
#ifdef HAVE_STORAGE
#include "file.h"
#include "dir.h"
#endif
#ifdef HAVE_BOOTFLASH
#include "bootflash.h"
#endif
#ifdef HAVE_BUTTON
#include "button.h"
#endif


static struct library_handle* library_list_head;
static struct mutex library_mutex;


struct library_handle* library_register(void* image, struct emcorelib_header* header,
                                        int argc, const char** argv)
{
    mutex_lock(&library_mutex, TIMEOUT_BLOCK);
    struct library_handle* h;
    for (h = library_list_head; h; h = h->next)
        if (h->lib->identifier == header->identifier && h->lib->version == header->version)
        {
            mutex_unlock(&library_mutex);
            return NULL;
        }
    if ((header->setupfunc && header->setupfunc() < 0)
     || (header->initfunc && header->initfunc(argc, argv) < 0))
    {
        mutex_unlock(&library_mutex);
        return NULL;
    }
    struct library_handle* handle = (struct library_handle*)malloc(sizeof(struct library_handle));
    memset(handle, 0, sizeof(struct library_handle));
    reownalloc(handle, OWNER_TYPE(OWNER_LIBRARY, handle));
    reownalloc(image, OWNER_TYPE(OWNER_LIBRARY, handle));
    handle->next = library_list_head;
    handle->lib = header;
    handle->alloc = image;
    library_list_head = handle;
    mutex_unlock(&library_mutex);
    return handle;
}

int library_unload(struct emcorelib_header* lib)
{
    mutex_lock(&library_mutex, TIMEOUT_BLOCK);
    int i;
    bool found = false;
    struct library_handle* h;
    struct library_handle* prev;
    for (h = library_list_head; h; h = h->next)
        if (h->lib == lib)
        {
            found = true;
            break;
        }
        else prev = h;
    if (!found)
    {
        mutex_unlock(&library_mutex);
        return -1;
    }
    for (i = 0; i < ARRAYLEN(h->users); i++)
        if (h->users[i])
        {
            mutex_unlock(&library_mutex);
            return -2;
        }
    if (h->moreusers)
        for (i = 0; i < h->moreusers_size / 4; i++)
            if (h->moreusers[i])
            {
                mutex_unlock(&library_mutex);
                return -2;
            }
    if (lib->shutdownfunc && lib->shutdownfunc() < 0)
    {
        mutex_unlock(&library_mutex);
        return -3;
    }
    if (library_list_head == h) library_list_head = h->next;
    else prev->next = h->next;
    library_release_all_of_thread(OWNER_TYPE(OWNER_LIBRARY, h));
#ifdef HAVE_STORAGE
    close_all_of_process(OWNER_TYPE(OWNER_LIBRARY, h));
    closedir_all_of_process(OWNER_TYPE(OWNER_LIBRARY, h));
#endif
#ifdef HAVE_BUTTON
    button_unregister_all_of_thread(OWNER_TYPE(OWNER_LIBRARY, h));
#endif
    free_all_of_thread(OWNER_TYPE(OWNER_LIBRARY, h));
    mutex_unlock(&library_mutex);
    return 0;
}

struct emcorelib_header* get_library_ext(uint32_t identifier, uint32_t version,
                                         enum library_sourcetype sourcetype, void* source,
                                         struct scheduler_thread* owner)
{
    int i;
    struct library_handle* h;
    struct library_handle* lib = NULL;
    mutex_lock(&library_mutex, TIMEOUT_BLOCK);
    for (h = library_list_head; h; h = h->next)
        if (h->lib->identifier == identifier &&
            h->lib->minversion <= version && h->lib->version >= version)
        {
            lib = h;
            break;
        }
    if (!lib)
    {
        switch (sourcetype)
        {
        case LIBSOURCE_RAM_ALLOCED:
        {
            lib = (struct library_handle*)execimage(source, false, 0, NULL);
            break;
        }

        case LIBSOURCE_RAM_NEEDCOPY:
        {
            lib = (struct library_handle*)execimage(source, true, 0, NULL);
            break;
        }

#ifdef HAVE_BOOTFLASH
        case LIBSOURCE_BOOTFLASH:
        {
            int size = bootflash_filesize((char*)source);
            if (size <= 0) break;
            void* buffer = memalign(0x10, size);
            if (!buffer) break;
            if (bootflash_read((char*)source, buffer, 0, size) != size)
            {
                free(buffer);
                break;
            }
            lib = (struct library_handle*)execimage(buffer, false, 0, NULL);
            break;
        }
#endif

#ifdef HAVE_STORAGE
        case LIBSOURCE_FILESYSTEM:
        {
            int fd = file_open((char*)source, O_RDONLY);
            if (fd <= 0) break;
            int size = filesize(fd);
            if (size <= 0)
            {
                close(fd);
                break;
            }
            void* buffer = memalign(0x10, size);
            if (!buffer)
            {
                close(fd);
                break;
            }
            if (read(fd, buffer, size) != size)
            {
                free(buffer);
                close(fd);
                break;
            }
            close(fd);
            lib = (struct library_handle*)execimage(buffer, false, 0, NULL);
            break;
        }
#endif
        }
        if (lib && (lib->lib->identifier != identifier
                 || lib->lib->minversion > version && lib->lib->version < version))
            lib = NULL;
        if (!lib)
        {
            mutex_unlock(&library_mutex);
            return NULL;
        }
    }
    for (i = 0; i < ARRAYLEN(lib->users); i++)
        if (lib->users[i] == NULL)
        {
            lib->users[i] = owner;
            mutex_unlock(&library_mutex);
            return lib->lib;
        }
    if (lib->moreusers)
        for (i = 0; i < lib->moreusers_size / 4; i++)
            if (h->moreusers[i] == NULL)
                {
                    h->moreusers[i] = owner;
                    mutex_unlock(&library_mutex);
                    return lib->lib;
                }
    void* newalloc = realloc(lib->moreusers, lib->moreusers_size + 64);
    if (!newalloc)
    {
        mutex_unlock(&library_mutex);
        return NULL;
    }
    reownalloc(newalloc, OWNER_TYPE(OWNER_LIBRARY, lib));
    lib->moreusers = (void**)newalloc;
    lib->moreusers[lib->moreusers_size / 4] = owner;
    lib->moreusers_size += 64;
    mutex_unlock(&library_mutex);
    return lib->lib;
}

struct emcorelib_header* get_library(uint32_t identifier, uint32_t version,
                                     enum library_sourcetype sourcetype, void* source)
{
    return get_library_ext(identifier, version, sourcetype, source, current_thread);
}

int release_library_ext(struct emcorelib_header* lib, struct scheduler_thread* owner)
{
    int i;
    int rc = -2;
    struct library_handle* h;
    mutex_lock(&library_mutex, TIMEOUT_BLOCK);
    for (h = library_list_head; h; h = h->next)
        if (h->lib == lib)
        {
            rc = -1;
            for (i = 0; i < ARRAYLEN(h->users); i++)
                if (h->users[i] == owner)
                {
                    h->users[i] = NULL;
                    rc = 0;
                    break;
                }
            if (rc && h->moreusers)
                for (i = 0; i < h->moreusers_size / 4; i++)
                    if (h->moreusers[i] == owner)
                        {
                            h->moreusers[i] = NULL;
                            rc = 0;
                            break;
                        }
            break;
        }
    mutex_unlock(&library_mutex);
    return rc;
}

int release_library(struct emcorelib_header* lib)
{
    return release_library_ext(lib, current_thread);
}

int library_release_all_of_thread(struct scheduler_thread* thread)
{
    mutex_lock(&library_mutex, TIMEOUT_BLOCK);
    int i;
    int released = 0;
    struct library_handle* h;
    for (h = library_list_head; h; h = h->next)
    {
        for (i = 0; i < ARRAYLEN(h->users); i++)
            if (h->users[i] == thread)
            {
                h->users[i] = NULL;
                released++;
            }
        if (h->moreusers)
            for (i = 0; i < h->moreusers_size / 4; i++)
                if (h->moreusers[i] == thread)
                    {
                        h->moreusers[i] = NULL;
                        released++;
                    }
    }
    mutex_unlock(&library_mutex);
    return released;
}