#include "bh_api.h"
#include "svsem.h"
#include "msg_mgr.h"

int svsem_get(key_t key, unsigned int value) {
  int semid, perms;

  perms = S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH | S_IWOTH;

  semid = semget(key, 2, IPC_CREAT | IPC_EXCL | perms);

  if (semid != -1) { /* Successfully created the semaphore */
    union semun arg;
    struct sembuf sop;

    arg.val = 0; /* So initialize it to 0 */
    if (semctl(semid, 0, SETVAL, arg) == -1)
      err_exit(errno, "semctl 1");

    arg.val = 1;
    if (semctl(semid, 1, SETVAL, arg) == -1)
      err_exit(errno, "semctl 2");
    
    /* Perform a "no-op" semaphore operation - changes sem_otime
       so other processes can see we've initialized the set. */

    sop.sem_num = 0; /* Operate on semaphore 0 */
    sop.sem_op = value;
    sop.sem_flg = SEM_UNDO;
    if (semop(semid, &sop, 1) == -1)
      err_exit(errno, "semop");

  } else { /* We didn't create the semaphore set */

    if (errno != EEXIST) { /* Unexpected error from semget() */
      err_exit(errno, "semget 1");

    } else { /* Someone else already created it */
      const int MAX_TRIES = 10;
      int j;
      union semun arg;
      struct semid_ds ds;

      semid = semget(key, 1, perms); /* So just get ID */
      if (semid == -1)
        err_exit(errno, "semget 2");

      /* Wait until another process has called semop() */

      arg.buf = &ds;
      for (j = 0; j < MAX_TRIES; j++) {
        if (semctl(semid, 0, IPC_STAT, arg) == -1)
          err_exit(errno, "semctl 2");

        if (ds.sem_otime != 0) /* Semop() performed? */
          break;               /* Yes, quit loop */
        sleep(1);              /* If not, wait and retry */
      }

      if (ds.sem_otime == 0) /* Loop ran to completion! */
        err_exit(errno, "Existing semaphore not initialized");
    }
  }
  return semid;
}

/* Reserve semaphore (blocking), return 0 on success, or -1 with 'errno'
   set to EINTR if operation was interrupted by a signal handler */

/* Reserve semaphore - decrement it by 1 */
int svsem_wait(int semid) {
  struct sembuf sops;

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

  while (semop(semid, &sops, 1) == -1)
    if (errno != EINTR) {
      err_msg(errno, "svsem_dec");
      
      return -1;
    }

#if defined(MSG_HANDLER)
  Msg_info msg_obj;

  msg_obj.key = inter_key_get();
  msg_obj.id = semid;
  msg_obj.act = SEM_GET;
  msg_distrib(SEM_TYPE_ID, &msg_obj);
#endif 

  return 0;
}

int svsem_trywait(int semid) {
  struct sembuf sops;

  sops.sem_num = 0;
  sops.sem_op = -1;
  sops.sem_flg = IPC_NOWAIT | SEM_UNDO;
 
#if defined(MSG_HANDLER)
  Msg_info msg_obj;

  msg_obj.key = inter_key_get();
  msg_obj.id = semid;
  msg_obj.act = SEM_GET;
  msg_distrib(SEM_TYPE_ID, &msg_obj);
#endif 

  return semop(semid, &sops, 1) ;
}

int svsem_timedwait(int semid, struct timespec *timeout) {
  struct sembuf sops;

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

  while (semtimedop(semid, &sops, 1, timeout) == -1)
    if (errno != EINTR) {
      // err_msg(errno, "svsem_psem_timedwait");
      return -1;
    }

#if defined(MSG_HANDLER)
  Msg_info msg_obj;

  msg_obj.key = inter_key_get();
  msg_obj.id = semid;
  msg_obj.act = SEM_GET;
  msg_distrib(SEM_TYPE_ID, &msg_obj);
#endif 

  return 0;
}

int svsem_uni_wait(int semid) {
  struct timespec res;

  res.tv_sec = SEM_WT_TIMEOUT;
  res.tv_nsec = 0;
  
  while(1) {
    if(svsem_timedwait(semid, &res) != 0) {

      if(svsem_post(semid) != 0) {
        err_msg(errno, "_inc");
      }

    } else {

      break;
    }
  }

  return 0;
}

/* Release semaphore - increment it by 1 */
int svsem_post(int semid) {
  struct sembuf sops;

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

  int rv = semop(semid, &sops, 1);
  if (rv == -1) {
    // err_msg(errno, "svsem_inc");
    return -1;
  }

#if defined(MSG_HANDLER)
  Msg_info msg_obj;

  msg_obj.key = inter_key_get();
  msg_obj.id = semid;
  msg_obj.act = SEM_POST;
  msg_distrib(SEM_TYPE_ID, &msg_obj);
#endif 

  return 0;
}

int svsem_cond_wait(int semid ){

  struct sembuf sops[2];
  union semun arg;

  arg.val = 1;
  if (semctl(semid, 1, SETVAL, arg) == -1) {
    err_msg(errno, "svsem_set");
    return -1;
  }   

  //释放mutex
  sops[0].sem_num = 0;
  sops[0].sem_op = 1;
  sops[0].sem_flg = SEM_UNDO;

  // 等待cond
  sops[1].sem_num = 1;
  sops[1].sem_op = 0;
  sops[1].sem_flg = SEM_UNDO;

  while (semop(semid, sops, 2) == -1)
    if (errno != EINTR) {
      // err_msg(errno, "Svsvsem_dec");
      return -1;
    }


   //重新获取mutex
  sops[0].sem_num = 0;
  sops[0].sem_op = -1;
  sops[0].sem_flg = SEM_UNDO;

  while (semop(semid, sops, 1) == -1)
    if (errno != EINTR) {
      // err_msg(errno, "Svsvsem_dec");
      return -1;
    }

#if defined(MSG_HANDLER)
  Msg_info msg_obj;

  msg_obj.key = inter_key_get();
  msg_obj.id = semid;
  msg_obj.act = SEM_GET;
  msg_distrib(SEM_TYPE_ID, &msg_obj);
#endif 

  return 0;
}


int svsem_cond_signal(int semid ){
  union semun arg;
  arg.val = 0;
  
  if (semctl(semid, 1, SETVAL, arg) == -1) {
    err_msg(errno, "svsem_set");
    return -1;
  }
  return 0;
}


/**
 * If sem_op equals 0, the value of the semaphore is checked to see whether it
 * currently equals 0. If it does, the operation completes immediately; otherwise,
 * semop() blocks until the semaphore value becomes 0.
 */
int svsem_zero(int semid) {
// logger.debug("%d: svsem_dec\n", semid);
  struct sembuf sops;

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

  while (semop(semid, &sops, 1) == -1)
    if (errno != EINTR) {
      // err_msg(errno, "svsem_zero");
      return -1;
    }

#if defined(MSG_HANDLER)
  Msg_info msg_obj;

  msg_obj.key = inter_key_get();
  msg_obj.id = semid;
  msg_obj.act = SEM_RESET;
  msg_distrib(SEM_TYPE_ID, &msg_obj);
#endif 

  return 0;
}


int svsem_zero_nowait(int semid) {
  struct sembuf sops;

  sops.sem_num = 0;
  sops.sem_op = 0;
  sops.sem_flg = IPC_NOWAIT | SEM_UNDO;

  while (semop(semid, &sops, 1) == -1)
    if (errno != EINTR) {
      // err_msg(errno, "svsem_zero_nowait");
      return -1;
    }

#if defined(MSG_HANDLER)
  Msg_info msg_obj;
  msg_obj.key = inter_key_get();
  msg_obj.id = semid;
  msg_obj.act = SEM_RESET;
  msg_distrib(SEM_TYPE_ID, &msg_obj);
#endif 

  return 0;
}

int svsem_zero_timeout(int semid, struct timespec *timeout) {
  struct sembuf sops;

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

  while (semtimedop(semid, &sops, 1, timeout) == -1)
    if (errno != EINTR) {
      // err_msg(errno, "svsem_zero_timeout");
      return -1;
    }

#if defined(MSG_HANDLER)
  Msg_info msg_obj;
  msg_obj.key = inter_key_get();
  msg_obj.id = semid;
  msg_obj.act = SEM_RESET;
  msg_distrib(SEM_TYPE_ID, &msg_obj);
#endif 

  return 0;
}

int svsem_set(int semid, int val) {
  union semun arg;
  arg.val = val;
  
  return semctl(semid, 0, SETVAL, arg);
}

void svsem_remove(int semid) {
  union semun dummy;

  if (semctl(semid, 0, IPC_RMID, dummy) == -1)
    err_msg(errno, "svsem_remove");

#if defined(MSG_HANDLER)
  Msg_info msg_obj;
  msg_obj.key = inter_key_get();
  msg_obj.id = semid;
  msg_obj.act = SEM_RM;
  msg_distrib(SEM_TYPE_ID, &msg_obj);
#endif 

}