Инициализируйте AVFormatContext из данных буфера - PullRequest
Новая
       25

Инициализируйте AVFormatContext из данных буфера

0 голосов
/ 23 апреля 2019

Я транслирую аудио AAC по сети, и мне нужно использовать ffmpeg для декодирования потока. Я пробовал в локальной сети, и все работает нормально, но по сети я не уверен, как инициализировать мой AVFormatContext.

Я посмотрел на функции av_probe_input_buffer* и av_probe_input_format*, но не похоже, что эти функции подходят для того, что я хочу сделать. Мой AVFormatContext всегда неполный, и я не могу найти аудиопоток, который мешает мне получить контекст кодека и инициализировать мой декодер.

Проблемный фрагмент кода выглядит примерно так: 

AVFormatContext *pFormatCtx = avformat_alloc_context();

AVFrame   *pFrame = av_frame_alloc();
AVPacket  *packet = (AVPacket *)av_malloc(sizeof(AVPacket));
av_init_packet(packet);
packet->buf = NULL;
packet->data = NULL;

pFormatCtx->flags |= AVFMT_FLAG_CUSTOM_IO;

// Read 10 packets to give ffmpeg some hint about the data
for (int i = 0; i < 10; i++) {
  uint32_t packet_size;
  fread(&packet_size, 1, sizeof(packet_size), f);

  uint8_t *pdata = (uint8_t*)malloc(packet_size);
  int len = fread(pdata, 1, packet_size, f);

  AVProbeData probeData;
  probeData.buf = pdata;
  probeData.buf_size = packet_size - 1;
  probeData.filename = "";
  pFormatCtx->iformat = av_probe_input_format(&probeData, 1);
}

// This is working, no error here
if (avformat_find_stream_info(pFormatCtx, NULL) < 0){
  printf("Error finding stream info!");
}

int audioStream = -1;
for (int i = 0; i < pFormatCtx->nb_streams; i++)
  if(pFormatCtx->streams[i]->codec->codec_type == AVMEDIA_TYPE_AUDIO){
    audioStream = i;
    break;
  }

// I get this error, so actually no audio stream is detected
if (audioStream == -1){
  printf("Didn't find a audio stream.\n");
  return -1;
}
printf("Audio stream found at index %d\n", audioStream);

// I do not get here, because an audio stream is not detected.
AVCodecContext *pCodecCtx = pFormatCtx->streams[audioStream]->codec; 

// This is where I want to be!
AVCodec *pCodec = avcodec_find_decoder(pCodecCtx->codec_id);
if (pCodec == NULL) {
  printf("Codec not found.\n");
  return -1;
}

// Open codec
if (avcodec_open2(pCodecCtx, pCodec, NULL) < 0) {
  printf("Could not open codec.\n");
  return -1;
}
...