some-usable-scripts/C_Language/ncurses_semaphores/consommateur.c

203 lines
4.1 KiB
C

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/ipc.h>
#include <assert.h>
#include <unistd.h>
#include <fcntl.h>
#include <stdarg.h>
#include <errno.h>
#include <sys/sem.h>
#include <sys/shm.h>
#include <signal.h>
#include <curses.h>
#include <ctype.h>
#include "consommateur.h"
#include "types.h"
#include "sema.h"
#include "constantes.h"
void quitter(int signal);
WINDOW *creation_fenetre(int n,int d,char *t);
typedef struct prod_s
{
int idp;
WINDOW w;
} PROD;
int shmid, shm_key, i = 0;
int mutex_data, mutex_tpa;
key_t sem_key_data = MUTEX_DATA;
key_t sem_key_tpa = MUTEX_TPA;
int main( int argc, char **argv)
{
if(argc < 2)
{
printf("Usage : %s nSHM \n", argv[0]);
exit(EXIT_FAILURE);
}
shm_key = (key_t) atoi(argv[1]);
MEMP * memoireP;
MEMP temp;
MSG msgtemp;
WINDOW *tWindow[MAX_PROD];
WINDOW *w ;
char c ;
signal(SIGHUP, quitter);
signal(SIGINT, quitter);
signal(SIGQUIT, quitter);
if((shmid = shmget(shm_key, sizeof(MEMP), IPC_CREAT|IPC_EXCL|0766)) == -1)
{
perror("shmget");
exit(EXIT_FAILURE);
}
if((memoireP = (MEMP *) shmat(shmid, 0 , 0766)) == (void *) -1)
{
perror("shmat");
exit(EXIT_FAILURE);
}
if((mutex_data = creat_sem( sem_key_data, 1)) == -1)
{
perror("creat_sem"); exit(EXIT_FAILURE);
}
if((mutex_tpa = creat_sem( sem_key_tpa, 1)) == -1)
{
perror("creat_sem");
exit(EXIT_FAILURE);
}
temp.tete = 0;
temp.queue = 0;
for( i = 0; i < MAX_PROD ; i++)
temp.tpa[i] = -1;
P(mutex_data);
*memoireP = temp;
V(mutex_data);
// nbDeProd => est-ce qu'il y a des producteur (si 0 on quitte)
int nbDeProd;
// numTete est un entier qui détermine la "tête" courante. Si elle a changé, c'est qu'on a ajouté un caractère
int numTete = 0;
int vartemp;
// Variable temporaire qui indique le premier lancement du consommateur.c
// Si = 0 alors aucun producteur ne s'est connecté, on ne quitte pas le programme
// Si != 0 et si nbDeProd = 0 alors on quitte
int premier_lancement = 0;
initscr() ; /* initialisation (obligatoire) de curses */
noecho() ; /* suppression de l'echo des caracteres tapes*/
cbreak() ; /* lecture non bufferisee */
for( i = 0; i < MAX_PROD; i++)
{
tWindow[i] = creation_fenetre(LINES / MAX_PROD, i*(LINES/MAX_PROD), "En attente");
}
while (1)
{
// On (re)met nbDeProd à 0
nbDeProd = 0;
// On vérifie qu'il y ait toujours des producteurs
P(mutex_tpa);
for(i = 0 ; i < MAX_PROD && nbDeProd == 0 ; i++ )
{
if(memoireP->tpa[i] != -1)
{
nbDeProd++;
premier_lancement++;
}
}
V(mutex_tpa);
P(mutex_data);
vartemp = (int) memoireP->tete;
msgtemp = (MSG) memoireP->f[( vartemp -1 + MAX_BUF) % MAX_BUF ];
V(mutex_data);
if(numTete != vartemp)
{
numTete = vartemp;
c = msgtemp.c;
w = tWindow[msgtemp.idp];
waddch(w,c) ;
wrefresh(w) ;
}
P(mutex_data);
memoireP->queue = (memoireP->tete -1 + MAX_BUF) % MAX_BUF;
V(mutex_data);
// S'il n'y a plus de producteurs, on quitte
if(nbDeProd == 0 && premier_lancement != 0)
{
quitter(PLUSDEPROD);
}
// Ralentissement volontaire du programme
// Pour cause d'utilisation excessive de CPU
usleep(2);
}
exit(EXIT_FAILURE);
}
WINDOW *creation_fenetre(int n,int d,char *t)
{
WINDOW *cadre ; /* la fenetre pour le cadre */
WINDOW *w ; /* la fenetre de dialogue */
/* creation du cadre */
cadre= newwin(n,COLS,d,0) ;
box(cadre,0,0) ;
mvwprintw(cadre,0,COLS/2-strlen(t)/2,t) ;
wrefresh(cadre) ;
/* creation de la fenetre de dialogue */
w= newwin(n-2,COLS-2,d+1,1) ;
idlok(w,TRUE) ;
scrollok(w,TRUE) ; /* mise en place du defilement (scrolling) */
wclear(w) ;
wrefresh(w) ;
return w ;
}
void quitter(int signal)
{
if(shmctl(shmid, IPC_RMID, 0) < 0)
{
perror("shmctl");
exit(EXIT_FAILURE);
}
if(mutex_data >= 0)
{
del_sem(sem_key_data);
}
if(mutex_tpa >= 0)
{
del_sem(sem_key_tpa);
}
endwin() ;
if(signal == PLUSDEPROD)
printf("Plus de producteurs.\n");
printf("FIN.\n");
exit(EXIT_SUCCESS);
}