SPO-3-NIX RC2

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/sem.h>
#include <sys/ipc.h>
#include <sys/types.h>
#include <sys/shm.h>
#include <string.h>
#include <errno.h>
#include <iostream>
#include <string>

#define SHMSZ 10

int main(int argc,char *argv[])
{

    key_t key1 = 12345;
    int semid;
    unsigned short semval;
    int cnt = 5;

    struct sembuf wait0, wait1, signal0, signal1;

    wait0.sem_num = 0;
    wait0.sem_op = -1;
    wait0.sem_flg = SEM_UNDO;

    wait1.sem_num = 1;
    wait1.sem_op = -1;
    wait1.sem_flg = SEM_UNDO;

    signal0.sem_num = 0;
    signal0.sem_op = 1;
    signal0.sem_flg = SEM_UNDO;

    signal1.sem_num = 1;
    signal1.sem_op = 1;
    signal1.sem_flg = SEM_UNDO;

    semid = semget(key1, 2 ,IPC_CREAT);

    semval = 1;
    semctl(semid,0, SETVAL, semval);
    semctl(semid,1, SETVAL, semval);

    int shmid;  // shared memory segment ID
    key_t key = 5678; // key for shared memory
    char *shm, *s; // pointers to shared memory

    if ((shmid = shmget(key, SHMSZ, IPC_CREAT | 0666)) < 0) {
        perror("shmget");
        exit(1);
    }
    if ((shm = (char*)shmat(shmid, NULL, 0)) == (char *) -1) {
        perror("shmat");
        exit(1);
    }

    int childpid = fork();
    int messageSentFully = 1;
    int messageReceivedFully = 1;
    if(childpid == 0)
    {
        std::string fullMessage;
        while ( 1 )
        {
            semop(semid,&wait1,1);
            messageReceivedFully = 0;
            while( messageReceivedFully == 0)
            {
                semop(semid,&wait0,1);
                // if ( messageReceivedFully == 1)
                // {
                //  puts("\nReceived message:");
                // }
                puts("1");
                char msg[SHMSZ] = {0};
                s = shm;
                for (int i = 0; i < SHMSZ; ++i)
                {
                    msg[i] = s[i];
                }
                fullMessage += msg;
                if ( msg[SHMSZ-1] == 0 )
                {
                    messageReceivedFully = 1;
                }
                else
                {
                    messageReceivedFully = 0;
                }
                semop(semid,&signal0,1);
            }
            puts("\nReceived message:");
            puts( fullMessage.c_str() );
            if (fullMessage == "quit\n")
            {
                break;
            }
            fullMessage = "";
            semop(semid,&signal1,1);
        }

        semctl(semid,0,IPC_RMID);
        printf("Semaphore removed from the System = %s\n", strerror(errno));
    }
    else
    {
        while(1)
        {
            semop(semid,&wait1,1);
            std::string message;
            puts("\nEnter message:");
            while( messageSentFully == 0)
            {
                semop(semid, &wait0, 1);
                char msg[SHMSZ] = {0};
                int bytesRead = read( STDIN_FILENO, msg, SHMSZ );
                if (bytesRead < SHMSZ)
                {
                    messageSentFully = 1;
                }
                else
                {
                    messageSentFully = 0;
                }
                s = shm;
                for (int i = 0; i < SHMSZ; ++i)
                {
                    s[i] = msg[i];
                }
                message += msg;
                semop(semid, &signal0, 1);
            }
            message = "";
            if ( message == "quit\n")
            {
                break;
            }
            semop(semid,&signal1,1);
        }
    }
    shmdt((void*)shm);
    return 0;
}