Subversion Repositories HelenOS

/*

 * Copyright (c) 2007 Jan Hudecek

 * Copyright (c) 2005 Jakub Jermar

 * All rights reserved.

 *

 * Redistribution and use in source and binary forms, with or without

 * modification, are permitted provided that the following conditions

 * are met:

 *

 * - Redistributions of source code must retain the above copyright

 *   notice, this list of conditions and the following disclaimer.

 * - Redistributions in binary form must reproduce the above copyright

 *   notice, this list of conditions and the following disclaimer in the

 *   documentation and/or other materials provided with the distribution.

 * - The name of the author may not be used to endorse or promote products

 *   derived from this software without specific prior written permission.

 *

 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR

 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES

 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.

 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,

 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT

 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,

 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY

 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT

 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF

 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

 */

#include <print.h>

#include <arch.h>

#include <test.h>

#include <proc/tasklet.h>

#include <synch/waitq.h>

#include <cpu.h>

#include <proc/thread.h>

#include <arch/types.h>

#include <config.h>

bool gquiet;

static void func(void *data)

{

    if (!gquiet)

        printf("cpu%d: %s",CPU->id, data);

}

#ifdef CONFIG_SMP

static void running_tasklet(void * data)

{

    waitq_t wq;

    waitq_initialize(&wq);

    tasklet_descriptor_t *tasklet_desc;

    //before we start we need to register a tasklet

    if (!gquiet)

        printf("cpu:%d, Registering tasklet...", CPU->id);

    if (!gquiet)

        tasklet_desc=tasklet_register(&func, "\nTasklet called and received data\n");

    else

        tasklet_desc=tasklet_register(&func, "");

    if (!gquiet)

        printf("Done!\n");

    //first we'll try disabling the tasklet

    if (!gquiet)

        printf("cpu:%d, Disabling tasklet...", CPU->id);

    tasklet_disable(tasklet_desc);

    if (!gquiet)

        printf("Done!\n");

    //we'll schedule the disabled tasklet

    if (!gquiet)

        printf("cpu:%d, Scheduling tasklet...", CPU->id);

    tasklet_schedule(tasklet_desc);

    if (!gquiet)

        printf("Done!\n");

    //and we'll wait if it gets called. It shouldn't however, because it's disabled

    if (!gquiet)

        printf("cpu:%d, Waiting 1s...\n", CPU->id);

    waitq_sleep_timeout(&wq,(uint32_t) 1000000,0);

    if (!gquiet)

        printf("cpu:%d, Done!\n", CPU->id);

    //then we'll try to enable it

    if (!gquiet)

        printf("cpu:%d, Enabling tasklet...", CPU->id);

    tasklet_enable(tasklet_desc);

    if (!gquiet)

        printf("Done!\n");

    //and wait if it gets called this time. It should because it's enabled

    if (!gquiet)

        printf("cpu:%d, Waiting 1s...\n", CPU->id);

    waitq_sleep_timeout(&wq,(uint32_t) 1000000,0);

    if (!gquiet)

        printf("cpu:%d, Done!\n", CPU->id);

    //finally we'll free the tasklet structure

    if (!gquiet)

        printf("cpu:%d, Freeing...", CPU->id);

    tasklet_free(tasklet_desc);

    if (!gquiet)

        printf("Done!\n");

}

#endif

char * test_tasklet1(bool quiet)

{

    gquiet = quiet;

    waitq_t wq;

    waitq_initialize(&wq);

    tasklet_descriptor_t *tasklet_desc;

#ifdef CONFIG_SMP

    thread_t* second_thread = NULL;

    if (!quiet)

        printf("cpus:%d\n", config.cpu_active);

    if (config.cpu_active >1) {

        second_thread = thread_create(&running_tasklet, NULL, TASK, THREAD_FLAG_WIRED,"running tasklet", false);

        if (CPU->id == 0)

            second_thread->cpu = &cpus[1];

        else

            second_thread->cpu = &cpus[0];

        thread_ready(second_thread);

    }

#endif

    //before we start we need to register a tasklet

    if (!quiet)

        printf("cpu:%d, Registering tasklet...", CPU->id);

    if (!quiet)

        tasklet_desc=tasklet_register(&func, "\nTasklet called and received data\n");

    else

        tasklet_desc=tasklet_register(&func, "");

    if (!quiet)

        printf("Done!\n");

    //first we'll try disabling the tasklet

    if (!quiet)

        printf("cpu:%d, Disabling tasklet...", CPU->id);

    tasklet_disable(tasklet_desc);

    if (!quiet)

        printf("Done!\n");

    //we'll schedule the disabled tasklet

    if (!quiet)

        printf("cpu:%d, Scheduling tasklet...", CPU->id);

    tasklet_schedule(tasklet_desc);

    if (!quiet)

        printf("Done!\n");

    //and we'll wait if it gets called. It shouldn't however, because it's disabled

    if (!quiet)

        printf("cpu:%d, Waiting 1s...\n", CPU->id);

    waitq_sleep_timeout(&wq,(uint32_t) 1000000,0);

    if (!quiet)

        printf("cpu:%d, Done!\n", CPU->id);

    //then we'll try to enable it

    if (!quiet)

        printf("cpu:%d, Enabling tasklet...", CPU->id);

    tasklet_enable(tasklet_desc);

    if (!quiet)

        printf("Done!\n");

    //and wait if it gets called this time. It should because it's enabled

    if (!quiet)

        printf("cpu:%d, Waiting 1s...\n", CPU->id);

    waitq_sleep_timeout(&wq,(uint32_t) 1000000,0);

    if (!quiet)

        printf("cpu:%d, Done!\n", CPU->id);

#ifdef CONFIG_SMP

        if (config.cpu_active >1) {

            printf("Joining with the second thread...");

            thread_join(second_thread);

            printf("Done\n");

        }

#endif

    //finally we'll free the tasklet structure

    if (!quiet)

        printf("cpu:%d, Freeing...", CPU->id);

    tasklet_free(tasklet_desc);

    if (!quiet)

        printf("Done!\n");

    return NULL;

}