125 lines
2.9 KiB
C
125 lines
2.9 KiB
C
|
|
||
|
|
/*
|
||
|
|
* Copyright (c) 2002, Intel Corporation. All rights reserved.
|
||
|
|
* Created by: crystal.xiong REMOVE-THIS AT intel DOT com
|
||
|
|
* This file is licensed under the GPL license. For the full content
|
||
|
|
* of this license, see the COPYING file at the top level of this
|
||
|
|
* source tree.
|
||
|
|
*
|
||
|
|
* This is a test about producer and consumer. Producer sends data
|
||
|
|
* to a buffer. Consumer keeps reading data from the buffer.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include <stdlib.h>
|
||
|
|
#include <string.h>
|
||
|
|
#include <errno.h>
|
||
|
|
#include "FreeRTOS_POSIX.h"
|
||
|
|
#include "FreeRTOS_POSIX/fcntl.h"
|
||
|
|
#include "FreeRTOS_POSIX/pthread.h"
|
||
|
|
#include "FreeRTOS_POSIX/semaphore.h"
|
||
|
|
|
||
|
|
#define BUF_SIZE 5
|
||
|
|
#define Max_Num 10
|
||
|
|
|
||
|
|
typedef struct {
|
||
|
|
int buffer[BUF_SIZE];
|
||
|
|
sem_t occupied;
|
||
|
|
sem_t empty;
|
||
|
|
sem_t lock;
|
||
|
|
} buf_t;
|
||
|
|
|
||
|
|
static int in, out;
|
||
|
|
|
||
|
|
static void producer(buf_t * buf)
|
||
|
|
{
|
||
|
|
int data;
|
||
|
|
int i;
|
||
|
|
|
||
|
|
for (i = 0; i < Max_Num; i++) {
|
||
|
|
if (-1 == sem_wait(&buf->occupied)) {
|
||
|
|
printf("sem_wait didn't return success \r\n");
|
||
|
|
pthread_exit((void *)1);
|
||
|
|
}
|
||
|
|
if (-1 == sem_wait(&buf->lock)) {
|
||
|
|
printf("sem_wait didn't return success \r\n");
|
||
|
|
pthread_exit((void *)1);
|
||
|
|
}
|
||
|
|
data = 100 * i;
|
||
|
|
buf->buffer[in] = data;
|
||
|
|
printf("producer has added %d to the buffer[%d] \r\n", data, in);
|
||
|
|
in = (in + 1) % BUF_SIZE;
|
||
|
|
if (-1 == sem_post(&buf->lock)) {
|
||
|
|
printf("sem_wait didn't return success \r\n");
|
||
|
|
pthread_exit((void *)1);
|
||
|
|
}
|
||
|
|
if (-1 == sem_post(&buf->empty)) {
|
||
|
|
printf("sem_wait didn't return success \r\n");
|
||
|
|
pthread_exit((void *)1);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
pthread_exit(NULL);
|
||
|
|
}
|
||
|
|
|
||
|
|
static void consumer(buf_t * buf)
|
||
|
|
{
|
||
|
|
int data;
|
||
|
|
int i;
|
||
|
|
|
||
|
|
for (i = 0; i < Max_Num; i++) {
|
||
|
|
if (-1 == sem_wait(&buf->empty)) {
|
||
|
|
printf("sem_wait didn't return success \r\n");
|
||
|
|
pthread_exit((void *)1);
|
||
|
|
}
|
||
|
|
if (-1 == sem_wait(&buf->lock)) {
|
||
|
|
printf("sem_wait didn't return success \r\n");
|
||
|
|
pthread_exit((void *)1);
|
||
|
|
}
|
||
|
|
data = buf->buffer[out];
|
||
|
|
printf("consumer has taken %d from buffer[%d] \r\n", data, out);
|
||
|
|
out = (out + 1) % BUF_SIZE;
|
||
|
|
if (-1 == sem_post(&buf->lock)) {
|
||
|
|
printf("sem_wait didn't return success \r\n");
|
||
|
|
pthread_exit((void *)1);
|
||
|
|
}
|
||
|
|
if (-1 == sem_post(&buf->occupied)) {
|
||
|
|
printf("sem_wait didn't return success \r\n");
|
||
|
|
pthread_exit((void *)1);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
pthread_exit(0);
|
||
|
|
}
|
||
|
|
|
||
|
|
int semaphores_test(void)
|
||
|
|
{
|
||
|
|
int shared = 1;
|
||
|
|
int occupied_value = BUF_SIZE;
|
||
|
|
int empty_value = 0;
|
||
|
|
int lock_value = 1;
|
||
|
|
buf_t buf;
|
||
|
|
pthread_t con, pro;
|
||
|
|
|
||
|
|
if (-1 == sem_init(&buf.occupied, shared, occupied_value)) {
|
||
|
|
printf("sem_init didn't return success \r\n");
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
if (-1 == sem_init(&buf.empty, shared, empty_value)) {
|
||
|
|
printf("sem_init didn't return success \r\n");
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
if (-1 == sem_init(&buf.lock, shared, lock_value)) {
|
||
|
|
printf("sem_init didn't return success \r\n");
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
in = out = 0;
|
||
|
|
|
||
|
|
pthread_create(&con, NULL, (void *)consumer, &buf);
|
||
|
|
pthread_create(&pro, NULL, (void *)producer, &buf);
|
||
|
|
pthread_join(con, NULL);
|
||
|
|
pthread_join(pro, NULL);
|
||
|
|
sem_destroy(&buf.occupied);
|
||
|
|
sem_destroy(&buf.empty);
|
||
|
|
|
||
|
|
printf("sema PASS\r\r\n");
|
||
|
|
return 0;
|
||
|
|
}
|