System V 信号量2

/**********************************
*Date:Sun Apr 13 09:29:06 EDT 2014
*Test for System V semaphore
**********************************/
#include <iostream>
extern "C"{
    #include <string.h>
    #include <errno.h>
    #include <sys/types.h>
    #include <sys/ipc.h>
    #include <sys/sem.h>
    #include <pthread.h>
    #include <unistd.h>
}
using namespace std;
#define SEM_R 0400
#define SEM_A 0200
#define SEM_KEY 0x1000
union semun{
    int               val;
    struct semid_ds*  buf;
    unsigned short*   array;
};
//global valiable
int iData = 0;

//sem_init
int user_sem_init(key_t& key,int val)
{
    int semfd = semget(key,1,IPC_CREAT|IPC_EXCL|SEM_R|SEM_A);
    if(semfd == -1){
        if(errno == EEXIST){
            semfd = semget(key,1,IPC_CREAT|SEM_R|SEM_A);
            if(semfd == -1){
                cerr << "failed to create sem,error:" << strerror(errno) << endl;
                return -1;
            }
        }
        else{
            return -1;
        }
    }
    union semun arg;
    arg.val = val;
    semctl(semfd,0,SETVAL,arg);
    return semfd;
}
//sem_wait
void usr_sem_wait(int& semfd)
{
    struct sembuf buf;
    buf.sem_num = 0;
    buf.sem_op = -1;
    buf.sem_flg = 0;
    semop(semfd,&buf,1);
}
//sem_post
void usr_sem_post(int& semfd)
{
    struct sembuf buf;
    buf.sem_num = 0;
    buf.sem_op = 1;
    buf.sem_flg = 0;
    semop(semfd,&buf,1);
}
//sem_getvalue
int usr_sem_getvalue(int& semfd)
{
    //union semun arg;//for GETALL the 4th parameter
    int res = semctl(semfd,0,GETVAL);
    return res;
}
//sem_destroy
void usr_sem_des(int& semfd)
{
    semctl(semfd,0,IPC_RMID);
}
//produce thread
void* produce(void* arg)
{
    int semfd = *(int*)arg;
    for(int i = 0;i < 10;i++){
        usleep(1);//just for test
        #ifdef _DEBUG
        cout << "produce:" << usr_sem_getvalue(semfd) << endl;
        #endif
        usr_sem_wait(semfd);
        #ifdef _DEBUG
        cout << "produce:" << usr_sem_getvalue(semfd) << endl;
        #endif
        iData++;
        usr_sem_post(semfd);
    }
}
//consume thread
void* consume(void* arg)
{
    int semfd = *(int*)arg;
    for(int i = 0;i < 10;i++){
        usleep(1);//just for test
        #ifdef _DEBUG
        cout << "consume:" << usr_sem_getvalue(semfd) << endl;
        #endif
        usr_sem_wait(semfd);
        #ifdef _DEBUG
        cout << "consume:" << usr_sem_getvalue(semfd) << endl;
        #endif
        cout << iData << endl;
        usr_sem_post(semfd);
    }
}
int main(int argc,char* argv[])
{
    pthread_t t1,t2;
    key_t key = (key_t)SEM_KEY;
    int semfd = user_sem_init(key,1);
    if(semfd == -1){
        return -1;
    }
    pthread_create(&t1,NULL,produce,(void*)&semfd);
    pthread_create(&t2,NULL,consume,(void*)&semfd);
    pthread_join(t1,NULL);
    pthread_join(t2,NULL);
    usr_sem_des(semfd);
    return 0;
}