RIOT/tests/sys/pthread_cooperation/main.c

/*
 * Copyright (C) 2014 Hamburg University of Applied Sciences
 *
 * This file is subject to the terms and conditions of the GNU Lesser
 * General Public License v2.1. See the file LICENSE in the top level
 * directory for more details.
 */

/**
 * @ingroup tests
 * @{
 *
 * @file
 * @brief pthread test application
 *
 * @author Raphael Hiesgen <raphael.hiesgen@haw-hamburg.de>
 *
 * @}
 */

#include <stdio.h>
#include <pthread.h>

#define NUM_THREADS 12

#define FACTORIAL_EXPECTED (479001600UL)

pthread_t ths[NUM_THREADS];

pthread_mutex_t mtx;
volatile uint32_t storage = 1;

void *run(void *parameter)
{
    int arg = (int) parameter;
    printf("My arg: %d\n", arg);

    int err = pthread_mutex_lock(&mtx);

    if (err != 0) {
        printf("[!!!] pthread_mutex_lock failed with %d\n", err);
        return NULL;
    }

    storage *= arg;
    printf("val = %"PRIu32"\n", storage);
    pthread_mutex_unlock(&mtx);

    return NULL;
}

int main(void)
{
    puts("START");
    pthread_attr_t th_attr;
    pthread_attr_init(&th_attr);
    pthread_mutex_init(&mtx, NULL);

    for (int i = 0; i < NUM_THREADS; ++i) {
        printf("Creating thread with arg %d\n", (i + 1));
        pthread_create(&ths[i], &th_attr, run, (void *)(i + 1));
    }

    for (int i = 0; i < NUM_THREADS; ++i) {
        printf("join thread %d\n", (i + 1));
        pthread_join(ths[i], NULL);
    }

    printf("Factorial: %"PRIu32"\n", storage);

    pthread_mutex_destroy(&mtx);
    pthread_attr_destroy(&th_attr);

    if (storage == FACTORIAL_EXPECTED) {
        puts("SUCCESS");
    }
    else {
        puts("FAILURE: Error, expected: 12!= 479001600.");
    }

    return 0;
}