src/dvr_record.c

#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include <pthread.h>
#include <string.h>
#include "dvr_types.h"
#include "dvr_record.h"
#include "dvr_crypto.h"
#include "record_device.h"
#include "segment.h"
#include <sys/time.h>

//#define DEBUG_PERFORMANCE
#define MAX_DVR_RECORD_SESSION_COUNT 2
#define RECORD_BLOCK_SIZE (256 * 1024)

/**\brief DVR index file type*/
typedef enum {
  DVR_INDEX_TYPE_PCR,                                                   /**< DVR index file use pcr*/
  DVR_INDEX_TYPE_LOCAL_CLOCK,                                           /**< DVR index file use local clock*/
  DVR_INDEX_TYPE_INVALID                                                 /**< DVR index file type invalid type*/
} DVR_IndexType_t;

/**\brief DVR VOD context*/
typedef struct {
  pthread_mutex_t                 mutex;                                /**< VOD mutex lock*/
  pthread_cond_t                  cond;                                 /**< VOD condition*/
  void                            *buffer;                              /**< VOD buffer*/
  uint32_t                        buf_len;                              /**< VOD buffer len*/
} DVR_VodContext_t;

/**\brief DVR record secure mode buffer*/
typedef struct {
  uint32_t addr;                                                        /**< Secure mode record buffer address*/
  uint32_t len;                                                         /**< Secure mode record buffer length*/
} DVR_SecureBuffer_t;

/**\brief DVR record new dmx secure mode buffer*/
typedef struct {
  uint32_t buf_start;                                                        /**< Secure mode record buffer address*/
  uint32_t buf_end;                                                         /**< Secure mode record buffer length*/
  uint32_t data_start;                                                        /**< Secure mode record buffer address*/
  uint32_t data_end;                                                         /**< Secure mode record buffer length*/
} DVR_NewDmxSecureBuffer_t;

/**\brief DVR record context*/
typedef struct {
  pthread_t                       thread;                               /**< DVR thread handle*/
  Record_DeviceHandle_t           dev_handle;                           /**< DVR device handle*/
  Segment_Handle_t                segment_handle;                       /**< DVR segment handle*/
  DVR_RecordState_t               state;                                /**< DVR record state*/
  char                            location[DVR_MAX_LOCATION_SIZE];      /**< DVR record file location*/
  DVR_RecordSegmentStartParams_t  segment_params;                       /**< DVR record start parameters*/
  DVR_RecordSegmentInfo_t         segment_info;                         /**< DVR record current segment info*/
  size_t                          notification_size;                    /**< DVR record nogification size*/
  DVR_RecordEventFunction_t       event_notify_fn;                      /**< DVR record event notify function*/
  void                            *event_userdata;                      /**< DVR record event userdata*/
  //DVR_VodContext_t                vod;                                  /**< DVR record vod context*/
  int                             is_vod;                               /**< Indicate current mode is VOD record mode*/
  DVR_CryptoFunction_t            enc_func;                             /**< Encrypt function*/
  void                            *enc_userdata;                        /**< Encrypt userdata*/
  int                             is_secure_mode;                       /**< Record session run in secure pipeline */
  size_t                          last_send_size;                       /**< Last send notify segment size */
  uint32_t                        block_size;                           /**< DVR record block size */
  DVR_Bool_t                      is_new_dmx;                           /**< DVR is used new dmx driver */
} DVR_RecordContext_t;

static DVR_RecordContext_t record_ctx[MAX_DVR_RECORD_SESSION_COUNT] = {
  {
    .state = DVR_RECORD_STATE_CLOSED
  },
  {
    .state = DVR_RECORD_STATE_CLOSED
  }
};

static int record_is_valid_pid(DVR_RecordContext_t *p_ctx, int pid)
{
  int i;

  for (i = 0; i < p_ctx->segment_info.nb_pids; i++) {
    if (pid == p_ctx->segment_info.pids[i].pid)
      return 1;
  }
  return 0;
}

static int record_save_pcr(DVR_RecordContext_t *p_ctx, uint8_t *buf, loff_t pos)
{
  uint8_t *p = buf;
  int len;
  uint8_t afc;
  uint64_t pcr = 0;
  int has_pcr = 0;
  int pid;
  int adp_field_len;

  pid = ((p[1] & 0x1f) << 8) | p[2];
  if (pid == 0x1fff || !record_is_valid_pid(p_ctx, pid))
    return has_pcr;

  //scramble = p[3] >> 6;
  //cc = p[3] & 0x0f;
  afc = (p[3] >> 4) & 0x03;

  p += 4;
  len = 184;

  if (afc & 2) {
    adp_field_len = p[0];
    /* Skip adaptation len */
    p++;
    len--;
    /* Parse pcr field, see 13818 spec table I-2-6,adaptation_field */
    if (p[0] & 0x10 && len >= 6 && adp_field_len >= 6) {
    /* get pcr value,pcr is 33bit value */
    pcr = (((uint64_t)(p[1])) << 25)
        | (((uint64_t)p[2]) << 17)
        | (((uint64_t)(p[3])) << 9)
        | (((uint64_t)p[4]) << 1)
        | ((((uint64_t)p[5]) & 0x80) >> 7);
      has_pcr = 1;
    }

    p += adp_field_len;
    len -= adp_field_len;

    if (len < 0) {
      DVR_DEBUG(1, "parser pcr: illegal adaptation field length");
      return 0;
    }
  }

  if (has_pcr) {
    segment_update_pts(p_ctx->segment_handle, pcr/90, pos);
  }
  return has_pcr;
}

static int record_do_pcr_index(DVR_RecordContext_t *p_ctx, uint8_t *buf, int len)
{
  uint8_t *p = buf;
  int left = len;
  loff_t pos;
  int has_pcr = 0;

  pos = segment_tell_position(p_ctx->segment_handle);
  while (left >= 188) {
    if (*p == 0x47) {
      has_pcr |= record_save_pcr(p_ctx, p, pos);
      p += 188;
      left -= 188;
      pos += 188;
    } else {
      p++;
      left --;
      pos++;
    }
  }
  return has_pcr;
}

static int get_diff_time(struct timeval start_tv, struct timeval end_tv)
{
  return end_tv.tv_sec * 1000 + end_tv.tv_usec / 1000 - start_tv.tv_sec * 1000 - start_tv.tv_usec / 1000;
}

void *record_thread(void *arg)
{
  DVR_RecordContext_t *p_ctx = (DVR_RecordContext_t *)arg;
  ssize_t len;
  uint8_t *buf, *buf_out;
  uint32_t block_size = p_ctx->block_size;
  loff_t pos = 0;
  int ret;
  struct timespec start_ts, end_ts;
  DVR_RecordStatus_t record_status;
  int has_pcr;
  int index_type = DVR_INDEX_TYPE_INVALID;
  time_t pre_time = 0;
  #define DVR_STORE_INFO_TIME (400)
  DVR_SecureBuffer_t secure_buf;
  DVR_NewDmxSecureBuffer_t new_dmx_secure_buf;

  buf = (uint8_t *)malloc(block_size);
  if (!buf) {
    DVR_DEBUG(1, "%s, malloc failed", __func__);
    return NULL;
  }

  buf_out = (uint8_t *)malloc(block_size + 188);
  if (!buf_out) {
    DVR_DEBUG(1, "%s, malloc failed", __func__);
    free(buf);
    return NULL;
  }

  memset(&record_status, 0, sizeof(record_status));
  record_status.state = DVR_RECORD_STATE_STARTED;
  if (p_ctx->event_notify_fn) {
    record_status.info.id = p_ctx->segment_info.id;
    p_ctx->event_notify_fn(DVR_RECORD_EVENT_STATUS, &record_status, p_ctx->event_userdata);
    DVR_DEBUG(1, "%s line %d notify record status, state:%d id=%lld",
          __func__,__LINE__, record_status.state, p_ctx->segment_info.id);
  }
  DVR_DEBUG(1, "%s, secure_mode:%d, block_size:%d", __func__, p_ctx->is_secure_mode, block_size);

  clock_gettime(CLOCK_MONOTONIC, &start_ts);

  struct timeval t1, t2, t3, t4, t5, t6, t7;
  while (p_ctx->state == DVR_RECORD_STATE_STARTED) {
    gettimeofday(&t1, NULL);

    /* data from dmx, normal dvr case */
    if (p_ctx->is_secure_mode) {
      if (p_ctx->is_new_dmx) {
          memset(&new_dmx_secure_buf, 0, sizeof(new_dmx_secure_buf));
          len = record_device_read(p_ctx->dev_handle, &new_dmx_secure_buf, sizeof(new_dmx_secure_buf), 1000);
      } else {
          memset(&secure_buf, 0, sizeof(secure_buf));
          len = record_device_read(p_ctx->dev_handle, &secure_buf, sizeof(secure_buf), 1000);
      }
      if (len != DVR_FAILURE) {
        DVR_DEBUG(1, "%s, secure_buf:%#x, size:%#x", __func__, secure_buf.addr, secure_buf.len);
      }
    } else {
      len = record_device_read(p_ctx->dev_handle, buf, block_size, 1000);
    }
    if (len == DVR_FAILURE) {
      //usleep(10*1000);
      DVR_DEBUG(1, "%s, start_read error", __func__);
      continue;
    }
    gettimeofday(&t2, NULL);

    /* Got data from device, record it */
    if (p_ctx->enc_func) {
      /* Encrypt record data */
      DVR_CryptoParams_t crypto_params;

      memset(&crypto_params, 0, sizeof(crypto_params));
      crypto_params.type = DVR_CRYPTO_TYPE_ENCRYPT;
      memcpy(crypto_params.location, p_ctx->location, sizeof(p_ctx->location));
      crypto_params.segment_id = p_ctx->segment_info.id;
      crypto_params.offset = p_ctx->segment_info.size;

      if (p_ctx->is_secure_mode) {
        crypto_params.input_buffer.type = DVR_BUFFER_TYPE_SECURE;
        if (p_ctx->is_new_dmx) {
          crypto_params.input_buffer.addr = new_dmx_secure_buf.data_start;
          crypto_params.input_buffer.size = new_dmx_secure_buf.data_end - new_dmx_secure_buf.data_start;
        } else {
          crypto_params.input_buffer.addr = secure_buf.addr;
          crypto_params.input_buffer.size = secure_buf.len;
        }
      } else {
        crypto_params.input_buffer.type = DVR_BUFFER_TYPE_NORMAL;
        crypto_params.input_buffer.addr = (size_t)buf;
        crypto_params.input_buffer.size = len;
      }

      crypto_params.output_buffer.type = DVR_BUFFER_TYPE_NORMAL;
      crypto_params.output_buffer.addr = (size_t)buf_out;
      crypto_params.output_buffer.size = block_size + 188;

      p_ctx->enc_func(&crypto_params, p_ctx->enc_userdata);
      gettimeofday(&t3, NULL);
      /* Out buffer length may not equal in buffer length */
      if (crypto_params.output_size > 0) {
        ret = segment_write(p_ctx->segment_handle, buf_out, crypto_params.output_size);
        len = crypto_params.output_size;
      }
    } else {
      gettimeofday(&t3, NULL);
      ret = segment_write(p_ctx->segment_handle, buf, len);
    }
    gettimeofday(&t4, NULL);
    //add DVR_RECORD_EVENT_WRITE_ERROR event if write error
    if (ret == -1 && len > 0 && p_ctx->event_notify_fn) {
      //send write event
       if (p_ctx->event_notify_fn) {
         memset(&record_status, 0, sizeof(record_status));
         DVR_DEBUG(1, "%s,send event write error", __func__,__LINE__);
         p_ctx->event_notify_fn(DVR_RECORD_EVENT_WRITE_ERROR, &record_status, p_ctx->event_userdata);
        }
        DVR_DEBUG(1, "%s,write error %d", __func__,__LINE__);
      goto end;
    }
    /* Do time index */
    uint8_t *index_buf = p_ctx->enc_func ? buf_out : buf;
    pos = segment_tell_position(p_ctx->segment_handle);
    has_pcr = record_do_pcr_index(p_ctx, index_buf, len);
    if (has_pcr == 0 && index_type == DVR_INDEX_TYPE_INVALID) {
      clock_gettime(CLOCK_MONOTONIC, &end_ts);
      if ((end_ts.tv_sec*1000 + end_ts.tv_nsec/1000000) -
          (start_ts.tv_sec*1000 + start_ts.tv_nsec/1000000) > 40) {
        /* PCR interval threshlod > 40 ms*/
        DVR_DEBUG(1, "%s use local clock time index", __func__);
        index_type = DVR_INDEX_TYPE_LOCAL_CLOCK;
      }
    } else if (has_pcr && index_type == DVR_INDEX_TYPE_INVALID){
      DVR_DEBUG(1, "%s use pcr time index", __func__);
      index_type = DVR_INDEX_TYPE_PCR;
    }
    gettimeofday(&t5, NULL);

    /* Update segment info */
    p_ctx->segment_info.size += len;
    /*Duration need use pcr to calculate, todo...*/
    if (index_type == DVR_INDEX_TYPE_PCR) {
      p_ctx->segment_info.duration = segment_tell_total_time(p_ctx->segment_handle);
      if (pre_time == 0)
       pre_time = p_ctx->segment_info.duration;
    } else if (index_type == DVR_INDEX_TYPE_LOCAL_CLOCK) {
      clock_gettime(CLOCK_MONOTONIC, &end_ts);
      p_ctx->segment_info.duration = (end_ts.tv_sec*1000 + end_ts.tv_nsec/1000000) -
        (start_ts.tv_sec*1000 + start_ts.tv_nsec/1000000);
            if (pre_time == 0)
       pre_time = p_ctx->segment_info.duration;
      segment_update_pts(p_ctx->segment_handle, p_ctx->segment_info.duration, pos);
    } else {
      DVR_DEBUG(1, "%s can NOT do time index", __func__);
    }
    p_ctx->segment_info.nb_packets = p_ctx->segment_info.size/188;

    if (p_ctx->segment_info.duration - pre_time > DVR_STORE_INFO_TIME) {
      pre_time = p_ctx->segment_info.duration + DVR_STORE_INFO_TIME;
      segment_store_info(p_ctx->segment_handle, &(p_ctx->segment_info));
    }
    gettimeofday(&t6, NULL);
     /*Event notification*/
    if (p_ctx->notification_size &&
        p_ctx->event_notify_fn &&
        /*!(p_ctx->segment_info.size % p_ctx->notification_size)*/
    (p_ctx->segment_info.size -p_ctx->last_send_size) >= p_ctx->notification_size&&
        p_ctx->segment_info.duration > 0) {
      memset(&record_status, 0, sizeof(record_status));
      //clock_gettime(CLOCK_MONOTONIC, &end_ts);
      p_ctx->last_send_size = p_ctx->segment_info.size;
      record_status.state = p_ctx->state;
      record_status.info.id = p_ctx->segment_info.id;
      record_status.info.duration = p_ctx->segment_info.duration;
      record_status.info.size = p_ctx->segment_info.size;
      record_status.info.nb_packets = p_ctx->segment_info.size/188;
      p_ctx->event_notify_fn(DVR_RECORD_EVENT_STATUS, &record_status, p_ctx->event_userdata);
      DVR_DEBUG(1, "%s notify record status, state:%d, id:%lld, duration:%ld ms, size:%zu",
          __func__, record_status.state, record_status.info.id, record_status.info.duration, record_status.info.size);
    }
    gettimeofday(&t7, NULL);
#ifdef DEBUG_PERFORMANCE
    DVR_DEBUG(1, "record count, read:%dms, encrypt:%dms, write:%dms, index:%dms, store:%dms, notify:%dms total:%dms read len:%zd ",
        get_diff_time(t1, t2), get_diff_time(t2, t3), get_diff_time(t3, t4), get_diff_time(t4, t5),
        get_diff_time(t5, t6), get_diff_time(t6, t7), get_diff_time(t1, t5), len);
#endif
  }
end:
  free((void *)buf);
  free((void *)buf_out);
  DVR_DEBUG(1, "exit %s", __func__);
  return NULL;
}

int dvr_record_open(DVR_RecordHandle_t *p_handle, DVR_RecordOpenParams_t *params)
{
  DVR_RecordContext_t *p_ctx;
  Record_DeviceOpenParams_t dev_open_params;
  int ret;
  uint32_t i;

  DVR_RETURN_IF_FALSE(p_handle);
  DVR_RETURN_IF_FALSE(params);

  for (i = 0; i < MAX_DVR_RECORD_SESSION_COUNT; i++) {
    if (record_ctx[i].state == DVR_RECORD_STATE_CLOSED) {
      break;
    }
  }
  DVR_RETURN_IF_FALSE(i < MAX_DVR_RECORD_SESSION_COUNT);
  DVR_RETURN_IF_FALSE(record_ctx[i].state == DVR_RECORD_STATE_CLOSED);
  p_ctx = &record_ctx[i];
  DVR_DEBUG(1, "%s , current state:%d, dmx_id:%d, notification_size:%zu ", __func__,
      p_ctx->state, params->dmx_dev_id, params->notification_size);

  /*Process event params*/
  p_ctx->notification_size = params->notification_size;
  p_ctx->event_notify_fn = params->event_fn;
  p_ctx->event_userdata = params->event_userdata;
  p_ctx->last_send_size = 0;
  //check is new driver
  p_ctx->is_new_dmx = dvr_check_dmx_isNew();
  /*Process crypto params, todo*/
  memset((void *)&dev_open_params, 0, sizeof(dev_open_params));
  if (params->data_from_memory) {
    /* data from memory, VOD case */
    p_ctx->is_vod = 1;
  } else {
    p_ctx->is_vod = 0;
    /* data from dmx, normal dvr case */
    dev_open_params.dmx_dev_id = params->dmx_dev_id;
    dev_open_params.buf_size = (params->flush_size > 0 ? params->flush_size : RECORD_BLOCK_SIZE);

    ret = record_device_open(&p_ctx->dev_handle, &dev_open_params);
    if (ret != DVR_SUCCESS) {
      DVR_DEBUG(1, "%s, open record devices failed", __func__);
      return DVR_FAILURE;
    }
  }

  p_ctx->block_size = (params->flush_size > 0 ? params->flush_size : RECORD_BLOCK_SIZE);
  p_ctx->enc_func = NULL;
  p_ctx->enc_userdata = NULL;
  p_ctx->is_secure_mode = 0;
  p_ctx->state = DVR_RECORD_STATE_OPENED;

  *p_handle = p_ctx;
  return DVR_SUCCESS;
}

int dvr_record_close(DVR_RecordHandle_t handle)
{
  DVR_RecordContext_t *p_ctx;
  int ret;
  uint32_t i;

  p_ctx = (DVR_RecordContext_t *)handle;
  for (i = 0; i < MAX_DVR_RECORD_SESSION_COUNT; i++) {
    if (p_ctx == &record_ctx[i])
      break;
  }
  DVR_RETURN_IF_FALSE(p_ctx == &record_ctx[i]);

  DVR_DEBUG(1, "%s , current state:%d", __func__, p_ctx->state);
  DVR_RETURN_IF_FALSE(p_ctx->state != DVR_RECORD_STATE_CLOSED);

  if (p_ctx->is_vod) {
    ret = DVR_SUCCESS;
  } else {
    ret = record_device_close(p_ctx->dev_handle);
    if (ret != DVR_SUCCESS) {
      DVR_DEBUG(1, "%s, failed", __func__);
    }
  }

  p_ctx->state = DVR_RECORD_STATE_CLOSED;
  return ret;
}

#if 0
int dvr_record_register_encryption(DVR_RecordHandle_t handle,
    DVR_CryptoFunction_t cb,
    DVR_CryptoParams_t params,
    void *userdata)
{
  return DVR_SUCCESS;
}
#endif

int dvr_record_start_segment(DVR_RecordHandle_t handle, DVR_RecordStartParams_t *params)
{
  DVR_RecordContext_t *p_ctx;
  Segment_OpenParams_t open_params;
  int ret;
  uint32_t i;

  p_ctx = (DVR_RecordContext_t *)handle;
  for (i = 0; i < MAX_DVR_RECORD_SESSION_COUNT; i++) {
    if (p_ctx == &record_ctx[i])
      break;
  }
  DVR_RETURN_IF_FALSE(p_ctx == &record_ctx[i]);

  DVR_DEBUG(1, "%s , current state:%d", __func__, p_ctx->state);
  DVR_RETURN_IF_FALSE(p_ctx->state != DVR_RECORD_STATE_STARTED);
  DVR_RETURN_IF_FALSE(p_ctx->state != DVR_RECORD_STATE_CLOSED);
  DVR_RETURN_IF_FALSE(params);

  DVR_RETURN_IF_FALSE(strlen((const char *)params->location) < DVR_MAX_LOCATION_SIZE);
  memset(&open_params, 0, sizeof(open_params));
  memcpy(open_params.location, params->location, sizeof(params->location));
  open_params.segment_id = params->segment.segment_id;
  open_params.mode = SEGMENT_MODE_WRITE;

  ret = segment_open(&open_params, &p_ctx->segment_handle);
  DVR_RETURN_IF_FALSE(ret == DVR_SUCCESS);

  /*process params*/
  {
    memcpy(p_ctx->location, params->location, sizeof(params->location));
    //need all params??
    memcpy(&p_ctx->segment_params, &params->segment, sizeof(params->segment));
    /*save current segment info*/
    memset(&p_ctx->segment_info, 0, sizeof(p_ctx->segment_info));
    p_ctx->segment_info.id = params->segment.segment_id;
    p_ctx->segment_info.nb_pids = params->segment.nb_pids;
    memcpy(p_ctx->segment_info.pids, params->segment.pids, params->segment.nb_pids*sizeof(DVR_StreamPid_t));
  }

  if (!p_ctx->is_vod) {
    /* normal dvr case */
    for (i = 0; i < params->segment.nb_pids; i++) {
      ret = record_device_add_pid(p_ctx->dev_handle, params->segment.pids[i].pid);
      DVR_RETURN_IF_FALSE(ret == DVR_SUCCESS);
    }
    ret = record_device_start(p_ctx->dev_handle);
    DVR_RETURN_IF_FALSE(ret == DVR_SUCCESS);
  }

  ret = segment_store_info(p_ctx->segment_handle, &p_ctx->segment_info);

  p_ctx->state = DVR_RECORD_STATE_STARTED;
  if (!p_ctx->is_vod)
    pthread_create(&p_ctx->thread, NULL, record_thread, p_ctx);

  return DVR_SUCCESS;
}

int dvr_record_next_segment(DVR_RecordHandle_t handle, DVR_RecordStartParams_t *params, DVR_RecordSegmentInfo_t *p_info)
{
  DVR_RecordContext_t *p_ctx;
  Segment_OpenParams_t open_params;
  int ret;
  uint32_t i;
  loff_t pos;

  p_ctx = (DVR_RecordContext_t *)handle;
  for (i = 0; i < MAX_DVR_RECORD_SESSION_COUNT; i++) {
    if (p_ctx == &record_ctx[i])
      break;
  }
  DVR_RETURN_IF_FALSE(p_ctx == &record_ctx[i]);

  DVR_DEBUG(1, "%s , current state:%d", __func__, p_ctx->state);
  DVR_RETURN_IF_FALSE(p_ctx->state == DVR_RECORD_STATE_STARTED);
  //DVR_RETURN_IF_FALSE(p_ctx->state != DVR_RECORD_STATE_CLOSED);
  DVR_RETURN_IF_FALSE(params);
  DVR_RETURN_IF_FALSE(p_info);
  DVR_RETURN_IF_FALSE(!p_ctx->is_vod);

  /*Stop the on going record segment*/
  //ret = record_device_stop(p_ctx->dev_handle);
  //DVR_RETURN_IF_FALSE(ret == DVR_SUCCESS);
  p_ctx->state = DVR_RECORD_STATE_STOPPED;
  pthread_join(p_ctx->thread, NULL);

  //add index file store
  pos = segment_tell_position(p_ctx->segment_handle);
  segment_update_pts_force(p_ctx->segment_handle, p_ctx->segment_info.duration, pos);

  p_ctx->segment_info.duration = segment_tell_total_time(p_ctx->segment_handle);
  /*Update segment info*/
  memcpy(p_info, &p_ctx->segment_info, sizeof(p_ctx->segment_info));

  ret = segment_store_info(p_ctx->segment_handle, p_info);
  DVR_RETURN_IF_FALSE(ret == DVR_SUCCESS);

  DVR_DEBUG(1, "%s dump segment info, id:%lld, nb_pids:%d, duration:%ld ms, size:%zu, nb_packets:%d params->segment.nb_pids:%d",
      __func__, p_info->id, p_info->nb_pids, p_info->duration, p_info->size, p_info->nb_packets, params->segment.nb_pids);

  /*Close current segment*/
  ret = segment_close(p_ctx->segment_handle);
  DVR_RETURN_IF_FALSE(ret == DVR_SUCCESS);
  /*Open the new record segment*/
  memset(&open_params, 0, sizeof(open_params));
  memcpy(open_params.location, p_ctx->location, sizeof(p_ctx->location));
  open_params.segment_id = params->segment.segment_id;
  open_params.mode = SEGMENT_MODE_WRITE;
  DVR_DEBUG(1, "%s: p_ctx->location:%s  params->location:%s", __func__, p_ctx->location,params->location);

  ret = segment_open(&open_params, &p_ctx->segment_handle);
  DVR_RETURN_IF_FALSE(ret == DVR_SUCCESS);
  /*process params*/
  {
    //need all params??
    memcpy(&p_ctx->segment_params, &params->segment, sizeof(params->segment));
    /*save current segment info*/
    memset(&p_ctx->segment_info, 0, sizeof(p_ctx->segment_info));
    p_ctx->segment_info.id = params->segment.segment_id;
    memcpy(p_ctx->segment_info.pids, params->segment.pids, params->segment.nb_pids*sizeof(DVR_StreamPid_t));
  }

  p_ctx->segment_info.nb_pids = 0;
  for (i = 0; i < params->segment.nb_pids; i++) {
    switch (params->segment.pid_action[i]) {
      case DVR_RECORD_PID_CREATE:
        DVR_DEBUG(1, "%s create pid:%d", __func__, params->segment.pids[i].pid);
        ret = record_device_add_pid(p_ctx->dev_handle, params->segment.pids[i].pid);
        p_ctx->segment_info.nb_pids++;
        DVR_RETURN_IF_FALSE(ret == DVR_SUCCESS);
        break;
      case DVR_RECORD_PID_KEEP:
        DVR_DEBUG(1, "%s keep pid:%d", __func__, params->segment.pids[i].pid);
        p_ctx->segment_info.nb_pids++;
        break;
      case DVR_RECORD_PID_CLOSE:
        DVR_DEBUG(1, "%s close pid:%d", __func__, params->segment.pids[i].pid);
        ret = record_device_remove_pid(p_ctx->dev_handle, params->segment.pids[i].pid);
        DVR_RETURN_IF_FALSE(ret == DVR_SUCCESS);
        break;
      default:
        DVR_DEBUG(1, "%s wrong action pid:%d", __func__, params->segment.pids[i].pid);
        return DVR_FAILURE;
    }
  }

  //ret = record_device_start(p_ctx->dev_handle);
  //DVR_RETURN_IF_FALSE(ret == DVR_SUCCESS);
  /*Update segment info*/
  ret = segment_store_info(p_ctx->segment_handle, &p_ctx->segment_info);

  p_ctx->state = DVR_RECORD_STATE_STARTED;
  pthread_create(&p_ctx->thread, NULL, record_thread, p_ctx);
  return DVR_SUCCESS;
}

int dvr_record_stop_segment(DVR_RecordHandle_t handle, DVR_RecordSegmentInfo_t *p_info)
{
  DVR_RecordContext_t *p_ctx;
  int ret;
  uint32_t i;
  loff_t pos;

  p_ctx = (DVR_RecordContext_t *)handle;
  for (i = 0; i < MAX_DVR_RECORD_SESSION_COUNT; i++) {
    if (p_ctx == &record_ctx[i])
      break;
  }
  DVR_RETURN_IF_FALSE(p_ctx == &record_ctx[i]);

  DVR_DEBUG(1, "%s , current state:%d", __func__, p_ctx->state);
  DVR_RETURN_IF_FALSE(p_ctx->state != DVR_RECORD_STATE_STOPPED);
  DVR_RETURN_IF_FALSE(p_ctx->state != DVR_RECORD_STATE_CLOSED);
  DVR_RETURN_IF_FALSE(p_info);/*should support NULL*/

  p_ctx->state = DVR_RECORD_STATE_STOPPED;
  if (p_ctx->is_vod) {
    p_ctx->segment_info.duration = segment_tell_total_time(p_ctx->segment_handle);
    p_ctx->segment_info.duration = 10*1000; //debug, should delete it
  } else {
    ret = record_device_stop(p_ctx->dev_handle);
    //DVR_RETURN_IF_FALSE(ret == DVR_SUCCESS);
    if (ret != DVR_SUCCESS)
      goto end;
    //p_ctx->state = DVR_RECORD_STATE_STOPPED;
    pthread_join(p_ctx->thread, NULL);
  }

  //add index file store
  pos = segment_tell_position(p_ctx->segment_handle);
  segment_update_pts_force(p_ctx->segment_handle, p_ctx->segment_info.duration, pos);
  p_ctx->segment_info.duration = segment_tell_total_time(p_ctx->segment_handle);

  /*Update segment info*/
  memcpy(p_info, &p_ctx->segment_info, sizeof(p_ctx->segment_info));

  ret = segment_store_info(p_ctx->segment_handle, p_info);
  //DVR_RETURN_IF_FALSE(ret == DVR_SUCCESS);
  if (ret != DVR_SUCCESS)
      goto end;

  DVR_DEBUG(1, "%s dump segment info, id:%lld, nb_pids:%d, duration:%ld ms, size:%zu, nb_packets:%d",
      __func__, p_info->id, p_info->nb_pids, p_info->duration, p_info->size, p_info->nb_packets);

end:
  ret = segment_close(p_ctx->segment_handle);
  DVR_RETURN_IF_FALSE(ret == DVR_SUCCESS);
  return DVR_SUCCESS;
}

int dvr_record_resume_segment(DVR_RecordHandle_t handle, DVR_RecordStartParams_t *params, uint64_t *p_resume_size)
{
  DVR_RecordContext_t *p_ctx;
  uint32_t i;
  int ret;

  p_ctx = (DVR_RecordContext_t *)handle;
  for (i = 0; i < MAX_DVR_RECORD_SESSION_COUNT; i++) {
    if (p_ctx == &record_ctx[i])
      break;
  }
  DVR_RETURN_IF_FALSE(p_ctx == &record_ctx[i]);
  DVR_RETURN_IF_FALSE(params);
  DVR_RETURN_IF_FALSE(p_resume_size);

  DVR_DEBUG(1, "%s , current state:%d, resume size:%lld", __func__, p_ctx->state, *p_resume_size);
  ret = dvr_record_start_segment(handle, params);
  DVR_RETURN_IF_FALSE(ret == DVR_SUCCESS);

  p_ctx->segment_info.size = *p_resume_size;

  return DVR_SUCCESS;
}

int dvr_record_get_status(DVR_RecordHandle_t handle, DVR_RecordStatus_t *p_status)
{
  DVR_RecordContext_t *p_ctx;
  int i;

  p_ctx = (DVR_RecordContext_t *)handle;
  for (i = 0; i < MAX_DVR_RECORD_SESSION_COUNT; i++) {
    if (p_ctx == &record_ctx[i])
      break;
  }
  DVR_RETURN_IF_FALSE(p_ctx == &record_ctx[i]);
  DVR_RETURN_IF_FALSE(p_status);

  //lock
  p_status->state = p_ctx->state;
  p_status->info.id = p_ctx->segment_info.id;
  p_status->info.duration = p_ctx->segment_info.duration;
  p_status->info.size = p_ctx->segment_info.size;
  p_status->info.nb_packets = p_ctx->segment_info.size/188;

  return DVR_SUCCESS;
}

int dvr_record_write(DVR_RecordHandle_t handle, void *buffer, uint32_t len)
{
  DVR_RecordContext_t *p_ctx;
  uint32_t i;
  off_t pos = 0;
  int ret;
  int has_pcr;

  p_ctx = (DVR_RecordContext_t *)handle;
  for (i = 0; i < MAX_DVR_RECORD_SESSION_COUNT; i++) {
    if (p_ctx == &record_ctx[i])
      break;
  }
  DVR_RETURN_IF_FALSE(p_ctx == &record_ctx[i]);
  DVR_RETURN_IF_FALSE(buffer);
  DVR_RETURN_IF_FALSE(len);

  pos = segment_tell_position(p_ctx->segment_handle);
  has_pcr = record_do_pcr_index(p_ctx, buffer, len);
  if (has_pcr == 0) {
    /* Pull VOD record shoud use PCR time index */
    DVR_DEBUG(1, "%s has no pcr, can NOT do time index", __func__);
  }
  ret = segment_write(p_ctx->segment_handle, buffer, len);
  if (ret != len) {
    DVR_DEBUG(1, "%s write error ret:%d len:%d", __func__, ret, len);
  }
  p_ctx->segment_info.size += len;
  p_ctx->segment_info.nb_packets = p_ctx->segment_info.size/188;

  return DVR_SUCCESS;
}

int dvr_record_set_encrypt_callback(DVR_RecordHandle_t handle, DVR_CryptoFunction_t func, void *userdata)
{
  DVR_RecordContext_t *p_ctx;
  uint32_t i;

  p_ctx = (DVR_RecordContext_t *)handle;
  for (i = 0; i < MAX_DVR_RECORD_SESSION_COUNT; i++) {
    if (p_ctx == &record_ctx[i])
      break;
  }
  DVR_RETURN_IF_FALSE(p_ctx == &record_ctx[i]);
  DVR_RETURN_IF_FALSE(func);

  DVR_DEBUG(1, "%s , current state:%d", __func__, p_ctx->state);
  DVR_RETURN_IF_FALSE(p_ctx->state != DVR_RECORD_STATE_STARTED);
  DVR_RETURN_IF_FALSE(p_ctx->state != DVR_RECORD_STATE_CLOSED);

  p_ctx->enc_func = func;
  p_ctx->enc_userdata = userdata;
  return DVR_SUCCESS;
}

int dvr_record_set_secure_buffer(DVR_RecordHandle_t handle, uint8_t *p_secure_buf, uint32_t len)
{
  DVR_RecordContext_t *p_ctx;
  uint32_t i;
  int ret;

  p_ctx = (DVR_RecordContext_t *)handle;
  for (i = 0; i < MAX_DVR_RECORD_SESSION_COUNT; i++) {
    if (p_ctx == &record_ctx[i])
      break;
  }
  DVR_RETURN_IF_FALSE(p_ctx == &record_ctx[i]);
  DVR_RETURN_IF_FALSE(p_secure_buf);
  DVR_RETURN_IF_FALSE(len);

  DVR_DEBUG(1, "%s , current state:%d", __func__, p_ctx->state);
  DVR_RETURN_IF_FALSE(p_ctx->state != DVR_RECORD_STATE_STARTED);
  DVR_RETURN_IF_FALSE(p_ctx->state != DVR_RECORD_STATE_CLOSED);

  ret = record_device_set_secure_buffer(p_ctx->dev_handle, p_secure_buf, len);
  DVR_RETURN_IF_FALSE(ret == DVR_SUCCESS);

  p_ctx->is_secure_mode = 1;
  return ret;
}