path: root/src/lrtp.cc

/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set et sw=2 ts=2: */
/***************************************************************************
 *            lrtp.cc
 *
 *  Wed Aug 28 09:50:55 CEST 2013
 *  Copyright 2013 Bent Bisballe Nyeng
 *  deva@aasimon.org
 ****************************************************************************/

/*
 *  This file is part of lrtp.
 *
 *  lrtp is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  lrtp is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with lrtp; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA.
 */
#include "lrtp.h"

#include <string>
#include <string.h>
#include <map>
#include <stdio.h>

#include "rtp_profile.h"
#include "rtp_profile_amrwb.h"
#include "rtp_profile_opus.h"
#include "rtp_profile_raw.h"
#include "rtp_profile_l16.h"
#include "rtp_profile_jpeg.h"

#include "srtp.h"

#define SKIP_SRTP

#ifdef __cplusplus
extern "C" {
#endif

typedef std::map<csrc_t, lrtp_profile_t *> profile_map_t;

struct lrtp_t {
  SRTP *srtp;
  csrc_t ssrc;
  unsigned short seq;
  profile_map_t profiles;

  csrc_t next_csrc; // for frame iterator (see get_next_frame)

  std::list<outputframe_t *> framelist;
};

typedef struct {
  lrtp_profile_id_t id;
  struct lrtp_profile_t *(*create)(struct lrtp_t *lrtp, unsigned int csrc,
                                   va_list ap);
} profile_class_t;

static const profile_class_t registered_profiles[] = {
  { PROFILE_AMRWB, profile_amrwb_create },
  { PROFILE_OPUS, profile_opus_create },
  { PROFILE_RAW, profile_raw_create },
  { PROFILE_L16, profile_raw_create },
  { PROFILE_JPEG, profile_jpeg_create },
  { }
};

EXPORT
struct lrtp_t *lrtp_init(const char *key, unsigned int ssrc)
{
  struct lrtp_t *lrtp = new struct lrtp_t;

  lrtp->srtp = new SRTP(key, ssrc);
  lrtp->ssrc = ssrc;
  lrtp->seq = 0;

  return lrtp;
}

EXPORT
void lrtp_close(struct lrtp_t *lrtp)
{
  delete lrtp->srtp;
  delete lrtp;
}

EXPORT
int lrtp_create_profile(struct lrtp_t *lrtp,
                        lrtp_profile_id_t profile_id,
                        unsigned int csrc, ...)
{
  struct lrtp_profile_t *profile = NULL;

  if(lrtp->profiles.find(csrc) != lrtp->profiles.end()) {
    // TODO: CSRC already active
    printf("ERROR: CSRC already active\n");
    return 1;
  }

  va_list ap;
  va_start(ap, csrc);

  const profile_class_t *p = registered_profiles;
  while(p->create) {
    if(p->id == profile_id) {
      profile = p->create(lrtp, csrc, ap);
      profile->process_finished = NULL; // TODO: Figure out a way to set non-profile specific options.
      break;
    }
    p++;
  }

  va_end(ap);

  if(!profile) {
    printf("ERROR: Could not find profile [%d]\n", profile_id);
    return 1;
  }

  profile->id = profile_id;
  profile->lrtp = lrtp;
  profile->csrc = csrc;

  lrtp->profiles[csrc] = profile;
  
  return 0;
}

EXPORT
void lrtp_destroy_profile(struct lrtp_t *lrtp, unsigned int csrc)
{
  if(lrtp->profiles.find(csrc) == lrtp->profiles.end()) {
    // TODO: CSRC not found
    printf("ERROR: CSRC not found\n");
    return;
  }

  struct lrtp_profile_t *profile = lrtp->profiles[csrc];
  profile->destroy(profile);
  lrtp->profiles.erase(csrc);
}

EXPORT
int lrtp_enqueue_frame(struct lrtp_t *lrtp, unsigned int csrc,
                       const char *data, size_t size,
                       unsigned long int timestamp)
{
  if(lrtp->profiles.find(csrc) == lrtp->profiles.end()) {
    // TODO: CSRC not found
    printf("ERROR: CSRC not found\n");
    return 1;
  }

  struct lrtp_profile_t *profile = lrtp->profiles[csrc];

  inputframe_t *frame = new inputframe_t();
  frame->data = data;
  frame->size = size;
  frame->offset = 0;
  frame->timestamp = timestamp;

  //printf("lrtp_enqueue_frame: frame->size: %d\n", frame->size);

  profile->framelist.push_back(frame);

  return 0;
}

// Assume we have at least one csrc in the list
static csrc_t get_next_csrc(struct lrtp_t *lrtp)
{
  profile_map_t::iterator i = lrtp->profiles.find(lrtp->next_csrc);

  if(i == lrtp->profiles.end()) {
    // If we haven't got a valid csrc start from the beginning.
    i = lrtp->profiles.begin();
  } else {
    // Otherwise increase the iterator, potentially wrapping.
    i++;
    if(i == lrtp->profiles.end()) i = lrtp->profiles.begin();
  }

  lrtp->next_csrc = i->second->csrc;

  return lrtp->next_csrc;
}

static lrtp_profile_t *get_next_profile(struct lrtp_t *lrtp)
{
  if(lrtp->profiles.size() == 0) return NULL; // No profiles

  csrc_t start = get_next_csrc(lrtp);
  csrc_t cur = start;
  do {
    if(lrtp->profiles.find(cur) == lrtp->profiles.end()) {
      printf("ERROR: This shouldn't happen."
             " Selected profile is not in the list\n");
      break;
    }

    struct lrtp_profile_t *profile = lrtp->profiles.find(cur)->second;
    if(profile->framelist.size()) return profile;

    cur = get_next_csrc(lrtp);

  } while(cur != start);

  return NULL;
}

EXPORT
int lrtp_pack(struct lrtp_t *lrtp, char *packet, size_t maxsize)
{
  //if(!profile) return PACK_MISSING_PROFILE;

  // TODO: Check profile magic word

  struct lrtp_profile_t *profile = get_next_profile(lrtp);

  if(profile == NULL || profile->framelist.size() == 0) return 0;

  inputframe_t *frame = profile->framelist.front();

  RTP rtp;
  rtp.setSSrc(lrtp->ssrc);
  rtp.setTimestamp(frame->timestamp);
  rtp.setSeq(lrtp->seq);
  rtp.addCSrc(profile->csrc);

  int ret = profile->pack(profile,
                          frame->data + frame->offset,
                          frame->size - frame->offset,
                          rtp);

  if(ret < 0) return ret;

  frame->offset += ret;

  if(frame->size == frame->offset) {
    if(profile->process_finished) {
      profile->process_finished(profile, frame->data,
                                profile->process_finished_ptr);
    }
    profile->framelist.pop_front();
    delete frame;
  }

  if(rtp.isValid()) {
    size_t size = rtp.packet(packet, maxsize);
#ifndef SKIP_SRTP
    size = lrtp->srtp->encrypt(packet, size);
#endif/*SKIP_SRTP*/
    lrtp->seq++;
    return size;
  }

  return 0;
}

EXPORT
int lrtp_dequeue_frame(struct lrtp_t *lrtp,
                       char *frame, size_t maxsize,
                       unsigned int *csrc, unsigned int *ts)
{
  if(lrtp->framelist.size() == 0) return 0;

  outputframe_t *f = lrtp->framelist.front();

  if(f->size > maxsize) {
    printf("Buffer is too small\n");
    return -1; // Buffer too small.
  }

  size_t framesize = f->size;
  memcpy(frame, f->data, f->size);

  *csrc = f->csrc;
  *ts = f->ts;

  lrtp->framelist.pop_front();

  //printf("%d frames left in framelist.\n", lrtp->framelist.size());

  free(f->data);
  delete f;

  return framesize;
}

EXPORT
int lrtp_unpack(struct lrtp_t *lrtp, const char *packet, size_t size)
{
  if(!lrtp) return -UNPACK_MISSING_HANDLE;

  // TODO: Check lrtp magic word

  char *pkg = (char*)malloc(size);
  memcpy(pkg, packet, size);

  int ret = 0;

#ifndef SKIP_SRTP
  ret = lrtp->srtp->decrypt(pkg, size);

  if(ret < 0) {
    printf("lrtp_unpack::decrypt error: %d\n", ret);
    free(pkg);
    return -1;
  }

  size = ret;
#endif/*SKIP_SRTP*/

  RTP rtp;
  rtp.fromPacket(pkg, size);
  std::list<csrc_t> csrcs = rtp.getCSrcs();
  if(csrcs.size() == 0) {
    free(pkg);
    return -UNPACK_MISSING_CSRC;
  }
  if(csrcs.size() > 1) {
    free(pkg);
    return -UNPACK_TOO_MANY_CSRCS;
  }

  csrc_t csrc = *csrcs.begin();

  struct lrtp_profile_t *profile = NULL;
  if(lrtp->profiles.find(csrc) == lrtp->profiles.end()) {
    free(pkg);
    return -UNPACK_MISSING_PROFILE;
  }

  profile = lrtp->profiles[csrc];

  std::list<outputframe_t*> framelist;
  ret = profile->unpack(profile, rtp, framelist);

  if(ret < 0) {
    printf("lrtp_unpack::Profile unpack failed: %d\n", ret);
    free(pkg);
    return -1;
  }

  // Make sure all frames in the list have the correct csrc.
  std::list<outputframe_t*>::iterator fi = framelist.begin();
  while(fi != framelist.end()) {
    (*fi)->csrc = csrc;
    fi++;
  }

  lrtp->framelist.splice(lrtp->framelist.end(), framelist);

  free(pkg);

  return ret;
}

#ifdef __cplusplus
}
#endif