Межпроцессное взаимодействие - C++ Linux - Ответ 7694758

#include <stdio.h>
#include <pthread.h>
#include <unistd.h>
#include <stdlib.h>
#include <errno.h>
#include <string.h>
#include <sys/types.h>
#include <sys/ipc.h>
#include <sys/shm.h>
#include <sys/sem.h>
 
 
#define MAX_LENGHT 100
key_t key = 234;
int sem_id; //ид группы семафоров
 
struct shared_info{
    char buffer[MAX_LENGHT];
};
int END; //символ выхода
#define CLOSE  4
union semun {
    int val;
    struct semid_ds *buf;
    unsigned short *array;
};
//инициализация группы семафоров
void sem_ini(void)
{
    key_t key=567;
   // key = ftok("/etc/fstab", getpid());
 
    //формируем группу из 1 семафоров
    if ((sem_id = semget(key, 1, IPC_CREAT|0666))==-1)
    {
        exit(0);
    }
}
 
//удаление группы семафоров
void del_semvalue(int sem_id)
{
    if (semctl(sem_id, 0, IPC_RMID, 0) == -1){
        exit(0);}
}
 
/**
 * @brief set_semvalue - устанавливаем значение семафору
 * @param sem_id - ид группы семафоров
 * @param num - номер семафора в группе
 * @param val - значение
 */
int set_semvalue(int sem_id, int num, int val) {
    union semun sem_union;
    sem_union.val = val;
 
    if (semctl(sem_id, num, SETVAL, sem_union) == -1){ return(0);}
 
    return(1);
}
 
/**
 * @brief sem_lock - блокировка семафора
 * @param sem_id - ид группы семафоров
 * @param num - номер семафора в группе
 * @param sem_op - занчение на которое изменяется семафор
 */
int sem_lock(int sem_id, int sem_num, int sem_op) {
 struct sembuf sem_b;
 sem_b.sem_num = sem_num;
 sem_b.sem_op = sem_op;
 sem_b.sem_flg = SEM_UNDO;
 if (semop(sem_id, &sem_b, 1) == -1) {
  return(0);
 }
 return(1);
}
 
 
/**
 * @brief sem_unlock - разблокировка семафора
 * @param sem_id - ид группы семафоров
 * @param num - номер семафора в группе
 * @param sem_op - занчение на которое изменяется семафор
 */
int sem_unlock(int sem_id, int sem_num, int sem_op) {
 struct sembuf sem_b;
 sem_b.sem_num = sem_num;
 sem_b.sem_op = sem_op;
 sem_b.sem_flg = SEM_UNDO;
 if (semop(sem_id, &sem_b, 1) == -1) {
  return(0);
 }
 return(1);
}
 
//поток потребитель
void consumer( void )
{
    printf("consune");
    //получаем указатель на разд. память - кольцевой буффер
    //указатель на разделяемую память
    //создаем разделяемую память
    struct shared_info *ptr_share;
    int shmid;
    if ((shmid = shmget(key, sizeof( *ptr_share ), 0)) < 0) {
    perror("shmget");
    exit(1);
    }
 
    if ((ptr_share = (struct shared_info*)shmat(shmid, NULL, 0)) == (int *) -1) {
    perror("shmat");
    exit(1);
    }
 
//    пока не нашли символ выхода
//    считываем буффер
    while (1){
        int i;
        int c;
        //цикл просмотра буффера
        for(i=0; i<MAX_LENGHT; i++)
        {
            printf("cons wait \n");
            //ждем мютекс
            if (!sem_lock(sem_id, 0, -1)) exit(0);
            printf("cons nowait \n");
 
            //считываем символ
            c =  ptr_share->buffer[i];
            printf("%c", c);
            sleep(2);
 
             //если прочитали символ завершения
             //отдаем мютекс, завершаем поток
            if(c == END) {
                printf("\nconsumer read last symbol\n");
                 if (!sem_unlock(sem_id, 0, -1)) exit(0);
                exit(0);}
 
            //отдаем мютекс
            if (!sem_unlock(sem_id, 0, -1)) exit(0);
 
            usleep(100000);
 
        }
    }
 
    if(shmdt(ptr_share) < 0){
            printf("Can't detach shared memory\n");
            exit(-1);
        }
 
        exit(0);
}
//закрытие файла разделяемой памяти
//void closeShare(int fd_share)
//{
//    close(fd_share);
//    //очищение разделяемой памяти
//    if ( fd_share == CLOSE ) {
//        shm_unlink(SHARED_MEMORY_OBJECT_NAME);
//    }
//}
int main(int argc, char* argv[])
{
 
    if(argc > 1)
    {
        consumer();
    }
    else{
            printf("Lab5. Task 6. \n\n");
            printf("Enter last symbol: ");
            //считывание данных с консоли
            //получаем символ завершения
            END = getchar();
 
            //создаем разделяемую память
            struct shared_info *ptr_share;
            int shmid;
            if ((shmid = shmget(key, sizeof( *ptr_share ), IPC_CREAT|0666)) < 0) {
            perror("shmget");
            exit(1);
            }
 
            if ((ptr_share = (struct shared_info*)shmat(shmid, NULL, 0)) == (int *) -1) {
            perror("shmat");
            exit(1);
            }
 
          // execl(  argv[0], argv[0], "1", NULL );
            int pid;
 
               pid=fork();
               printf("pid %d \n", pid);
 
               switch(pid) {
 
               case 0:
                   consumer();
               default:
                   //producer:
                   
                    printf("\nproducer \n");
                    if (!set_semvalue(sem_id, 0, 1)) {
                        exit(0);}
                    //pthread_mutex_lock(&ptr_share->mutex );
 
 
            //пока не прочитали символ выхода
            //записываем в буффер
             int S;
             while(1){
                 //цикл записи в буффер
                 int i;
                 for(i=0; i<MAX_LENGHT; i++)
                 {
                     printf("prod wait \n");
                     //ждем мютекс
                     if (!sem_lock(sem_id, 0, -1)) exit(0);
 
                    printf("prod nowait \n");
                     //считываем символ с консоли
                    fflush(stdin);
                     S = getchar();
                     printf("prod 1 %c \n", S);
                     putchar (S);
                     //записываем его в буффер
                     ptr_share->buffer[i] = S;
                     //если нашли символ завершения
                     //записываем его в буффер
                     //отдаем мютекс
                     // и выходим из цикла записи
                     printf("prod 2%d \n");
                     if(S == END)
                     {  printf("\nproducer get last symbol\n");
                        if (!sem_unlock(sem_id, 0, -1)) exit(0);
                        break;
                     }
                      printf("prod rel%d \n");
                     //отдаем мютекс
                     if (!sem_unlock(sem_id, 0, -1)) exit(0);
 
                 }
 
                 //выходим из бесконечного цикла
                 if(S == END)
                 {break;}
             }
 
 
             del_semvalue(sem_id);
 
             if(shmdt(ptr_share) < 0){
                     printf("Can't detach shared memory\n");
                     exit(-1);
                 }
 
 
//            // closeShare(fd);
 
            printf("\n--- END --- \n");
               }
          }
 
 
    return 0;
}