共享内存

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>
#include <sys/stat.h>
#include <unistd.h>
#include <sys/types.h>
#include <errno.h>
#include <signal.h>
#include <sys/msg.h>
#include <sys/shm.h>
#include <sys/sem.h>
#include <sys/ipc.h>

#define ERROR(flag,msg)            if(flag)                        {                                    printf("%d: ",__LINE__);        fflush(stdout);                    perror(msg);                    exit(errno);                }

#define PATH_NAME "/home/"
#define PROJ_ID 100
#define SHMBUF_SIZE 1000

static int shmid;
static char *shmaddr;
static int semid;


void create_shm()
{
    key_t key = ftok(PATH_NAME,PROJ_ID);
    ERROR(key == -1,"ftok()");

    shmid = shmget(key,SHMBUF_SIZE,IPC_CREAT | 0600);
    ERROR(shmid == -1,"shmget()");

    shmaddr = shmat(shmid,NULL,0);
    ERROR(shmaddr == (void *)-1,"shmat()");
}

void delete_shm()
{
    int ret = shmdt(shmaddr);
    ERROR(ret == -1,"shmdt");


    ret = shmctl(shmid,IPC_RMID,NULL);
}

void create_sem()
{
    key_t key = ftok(PATH_NAME,PROJ_ID);
    ERROR(key == -1,"ftok()");

    semid = semget(key,2,IPC_CREAT | 0600);
    ERROR(semid == -1,"semget()");

#if 0
    int ret = semctl(semid,0,SETVAL,1);
    ERROR(ret == -1,"semctl()");
#endif
}

void delete_sem()
{
    semctl(semid,0,IPC_RMID);
}

void lock()
{
    struct sembuf sops;

    sops.sem_num = 0;
    sops.sem_op = -1;
    sops.sem_flg = 0;

    semop(semid,&sops,1);
}

void unlock()
{
    struct sembuf sops;

    sops.sem_num = 0;
    sops.sem_op = 1;
    sops.sem_flg = 0;

    semop(semid,&sops,1);
}

void handler(int signum)
{
    delete_sem();
    delete_shm();
    exit(0);
}

int main(int argc, char *argv[])
{
    signal(SIGINT,handler);

    create_shm();
    create_sem();

    lock();
    bzero(shmaddr,SHMBUF_SIZE);
    memcpy(shmaddr,argv[2],SHMBUF_SIZE);
    unlock();

    while(1)
    {
        lock();

        puts(shmaddr);
        bzero(shmaddr,SHMBUF_SIZE);
        memcpy(shmaddr,argv[2],SHMBUF_SIZE);
        sleep(atoi(argv[1]));

        unlock();
    }

    delete_sem();
    delete_shm();
    return 0;
}

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>
#include <sys/stat.h>
#include <unistd.h>
#include <sys/types.h>
#include <errno.h>
#include <signal.h>
#include <sys/msg.h>
#include <sys/shm.h>
#include <sys/sem.h>
#include <sys/ipc.h>

#define ERROR(flag,msg)            if(flag)                        {                                    printf("%d: ",__LINE__);        fflush(stdout);                    perror(msg);                    exit(errno);                }

#define PATH_NAME "/home/"
#define PROJ_ID 100
#define SHMBUF_SIZE 1000

static int shmid;
static char *shmaddr;
static int semid;


void create_shm()
{
    key_t key = ftok(PATH_NAME,PROJ_ID);
    ERROR(key == -1,"ftok()");

    shmid = shmget(key,SHMBUF_SIZE,IPC_CREAT | 0600);
    ERROR(shmid == -1,"shmget()");

    shmaddr = shmat(shmid,NULL,0);
    ERROR(shmaddr == (void *)-1,"shmat()");
}

void delete_shm()
{
    int ret = shmdt(shmaddr);
    ERROR(ret == -1,"shmdt");


    ret = shmctl(shmid,IPC_RMID,NULL);
}

void create_sem()
{
    key_t key = ftok(PATH_NAME,PROJ_ID);
    ERROR(key == -1,"ftok()");

    semid = semget(key,2,IPC_CREAT | 0600);
    ERROR(semid == -1,"semget()");

    int ret = semctl(semid,0,SETVAL,1);
    ERROR(ret == -1,"semctl()");

}

void delete_sem()
{
    semctl(semid,0,IPC_RMID);
}

void lock()
{
    struct sembuf sops;

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

    semop(semid,&sops,1);
}

void unlock()
{
    struct sembuf sops;

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

    semop(semid,&sops,1);
}

void handler(int signum)
{
    delete_sem();
    delete_shm();
    exit(0);
}

int main(int argc, char *argv[])
{
    signal(SIGINT,handler);

    create_shm();
    create_sem();

    lock();
    bzero(shmaddr,SHMBUF_SIZE);
    memcpy(shmaddr,argv[2],SHMBUF_SIZE);
    unlock();
    while(1)
    {
        lock();

        puts(shmaddr);
        bzero(shmaddr,SHMBUF_SIZE);
        memcpy(shmaddr,argv[2],SHMBUF_SIZE);
        sleep(atoi(argv[1]));

        unlock();
    }

    delete_sem();
    delete_shm();
    return 0;
}