Superkkt Blog

#include <sys/shm.h>
#include <sys/sem.h>
#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>

static int shm_id;
static int sem_id;
static int *num;
static pid_t main_pid;


/* 종료함수 */
void
terminator(void)
{
    /* main process에서만 작동 */
    if(getpid() == main_pid) {
        /* shared memory 제거 */
        shmctl(shm_id, IPC_RMID, NULL);
        shmdt((void *) num);

        /* semaphore 제거 */
        semctl(sem_id, 0, IPC_RMID);
    }
}


int
main(void)
{
    int a;
    pid_t pid;
    struct sembuf in = {0, -1, SEM_UNDO};
    struct sembuf out = {0, 1, SEM_UNDO};

    /* semaphores 초기값을 줄때 사용하는 공용체 */
    union semun {
        int val;
        struct semid_ds *buf;
        ushort *array;
    } argument;

    atexit(terminator);
    main_pid = getpid();

    /* shared memory 생성 */
    if((shm_id = shmget(IPC_PRIVATE, sizeof(int), 0666)) < 0) {
        perror("shmget error");
        exit(1);
    }
    if((num = (int *) shmat(shm_id, 0, 0)) == (void *) -1) {
        perror("shmat error");
        exit(1);
    }
    *num = 0;

    /* semaphores 생성 */
    if((sem_id = semget(IPC_PRIVATE, 1, 0666)) < 0) {
        perror("semget error");
        exit(1);
    }
    argument.val = 1;
    if(semctl(sem_id, 0, SETVAL, argument) == -1) {
        perror("semctl error");
        exit(1);
    }

    pid = fork();

    /* child */
    if(pid == 0) {
        a = 0;
        while(1) {
            fprintf(stderr, "trying to lock in child\n");
            if(semop(sem_id, &in, 1) == -1) {
                perror("semop error");
                exit(1);
            }

            fprintf(stderr, "locked in child\n");
            sleep(2);

            /* lock인 상태에서 프로세스가 종료했을때
             * 커널이 SEM_UNDO를 해서 자동으로 lock
             * 을 풀어주는지 테스트하기 위한 코드
             */
            if(a == 5) {
                fprintf(stderr, "child terminated\n");
                exit(0);
            }

            if(semop(sem_id, &out, 1) == -1) {
                perror("semop error");
                exit(1);
            }
            fprintf(stderr, "unlocked in child\n");
            a++;
            sleep(1);
        }
    }
    else if(pid > 0) {
        a = 0;
        while(1) {
            fprintf(stderr, "trying to lock in parent\n");
            if(semop(sem_id, &in, 1) == -1) {
                perror("semop error");
                exit(1);
            }

            fprintf(stderr, "locked in parent\n");
            sleep(2);

            if(semop(sem_id, &out, 1) == -1) {
                perror("semop error");
                exit(1);
            }
            fprintf(stderr, "unlocked in parent\n");
            sleep(1);
            if(a == 8)
                exit(0);
            a++;
        }
    }

    return 0;
}