teshsuite/mc/mcmini/producer_consumer_deadlock.c

   1 #include <stdlib.h>
   2 #include <stdio.h>
   3 #include <pthread.h>
   4 #include <semaphore.h>
   5
   6 int MaxItems; // Maximum items a producer can produce or a consumer can consume
   7 int BufferSize; // Size of the buffer
   8
   9 sem_t empty;
  10 sem_t full;
  11 int in = 0;
  12 int out = 0;
  13 int *buffer;
  14 pthread_mutex_t mutex;
  15
  16 int DEBUG = 0; // Debug flag
  17
  18 static void *producer(void *pno)
  19 {
  20     int item;
  21     for(int i = 0; i < MaxItems; i++) {
  22         item = rand(); // Produce a random item
  23         pthread_mutex_lock(&mutex);
  24         sem_wait(&empty);
  25         buffer[in] = item;
  26         if(DEBUG) printf("Producer %d: Insert Item %d at %d\n", *((int *)pno),buffer[in],in);
  27         in = (in+1)%BufferSize;
  28         pthread_mutex_unlock(&mutex);
  29         sem_post(&full);
  30     }
  31     return NULL;
  32 }
  33
  34 static void *consumer(void *cno)
  35 {
  36     for(int i = 0; i < MaxItems; i++) {
  37         pthread_mutex_lock(&mutex);
  38         sem_wait(&full);
  39         int item = buffer[out];
  40         if(DEBUG) printf("Consumer %d: Remove Item %d from %d\n",*((int *)cno),item, out);
  41         out = (out+1)%BufferSize;
  42         pthread_mutex_unlock(&mutex);
  43         sem_post(&empty);
  44     }
  45     return NULL;
  46 }
  47
  48 int main(int argc, char* argv[]) {
  49     if(argc != 4){
  50         printf("Usage: %s MAX_ITEMS BUFFER_SIZE DEBUG\n", argv[0]);
  51         return 1;
  52     }
  53
  54     MaxItems = atoi(argv[1]);
  55     BufferSize = atoi(argv[2]);
  56     DEBUG = atoi(argv[3]);
  57
  58     buffer = (int*) malloc(BufferSize * sizeof(int));
  59
  60     pthread_t pro[5],con[5];
  61     pthread_mutex_init(&mutex, NULL);
  62     sem_init(&empty,0,BufferSize);
  63     sem_init(&full,0,0);
  64
  65     int a[5] = {1,2,3,4,5}; //Just used for numbering the producer and consumer
  66
  67     for(int i = 0; i < 5; i++) {
  68         pthread_create(&pro[i], NULL, producer, (void *)&a[i]);
  69     }
  70     for(int i = 0; i < 5; i++) {
  71         pthread_create(&con[i], NULL, consumer, (void *)&a[i]);
  72     }
  73
  74     for(int i = 0; i < 5; i++) {
  75         pthread_join(pro[i], NULL);
  76     }
  77     for(int i = 0; i < 5; i++) {
  78         pthread_join(con[i], NULL);
  79     }
  80
  81     free(buffer);
  82     return 0;
  83 }