audin_oss.c

 
#include <freerdp/config.h>
 
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
 
#include <winpr/crt.h>
#include <winpr/synch.h>
#include <winpr/string.h>
#include <winpr/thread.h>
#include <winpr/cmdline.h>
 
#include <err.h>
#include <errno.h>
#include <fcntl.h>
#include <libgen.h>
#include <limits.h>
#include <unistd.h>
#include <oss-includes.h>
#include <sys/ioctl.h>
 
#include <freerdp/freerdp.h>
#include <freerdp/addin.h>
#include <freerdp/channels/rdpsnd.h>
 
#include "audin_main.h"
 
typedef struct
{
  IAudinDevice iface;
 
  HANDLE thread;
  HANDLE stopEvent;
 
  AUDIO_FORMAT format;
  UINT32 FramesPerPacket;
  int dev_unit;
 
  AudinReceive receive;
  void* user_data;
 
  rdpContext* rdpcontext;
} AudinOSSDevice;
 
static void OSS_LOG_ERR(const char* _text, int _error)
{
  if ((_error) != 0)
  {
    char buffer[256] = { 0 };
    WLog_ERR(TAG, "%s: %i - %s\n", (_text), (_error),
             winpr_strerror((_error), buffer, sizeof(buffer)));
  }
}
 
static UINT32 audin_oss_get_format(const AUDIO_FORMAT* format)
{
  switch (format->wFormatTag)
  {
    case WAVE_FORMAT_PCM:
      switch (format->wBitsPerSample)
      {
        case 8:
          return AFMT_S8;
 
        case 16:
          return AFMT_S16_LE;
 
        default:
          break;
      }
 
      break;
    default:
      break;
  }
 
  return 0;
}
 
static BOOL audin_oss_format_supported(IAudinDevice* device, const AUDIO_FORMAT* format)
{
  if (device == NULL || format == NULL)
    return FALSE;
 
  switch (format->wFormatTag)
  {
    case WAVE_FORMAT_PCM:
      if (format->cbSize != 0 || format->nSamplesPerSec > 48000 ||
          (format->wBitsPerSample != 8 && format->wBitsPerSample != 16) ||
          (format->nChannels != 1 && format->nChannels != 2))
        return FALSE;
 
      break;
 
    default:
      return FALSE;
  }
 
  return TRUE;
}
 
static UINT audin_oss_set_format(IAudinDevice* device, const AUDIO_FORMAT* format,
                                 UINT32 FramesPerPacket)
{
  AudinOSSDevice* oss = (AudinOSSDevice*)device;
 
  if (device == NULL || format == NULL)
    return ERROR_INVALID_PARAMETER;
 
  oss->FramesPerPacket = FramesPerPacket;
  oss->format = *format;
  return CHANNEL_RC_OK;
}
 
static DWORD WINAPI audin_oss_thread_func(LPVOID arg)
{
  char dev_name[PATH_MAX] = "/dev/dsp";
  int pcm_handle = -1;
  BYTE* buffer = NULL;
  unsigned long tmp = 0;
  size_t buffer_size = 0;
  AudinOSSDevice* oss = (AudinOSSDevice*)arg;
  UINT error = 0;
  DWORD status = 0;
 
  if (oss == NULL)
  {
    error = ERROR_INVALID_PARAMETER;
    goto err_out;
  }
 
  if (oss->dev_unit != -1)
    (void)sprintf_s(dev_name, (PATH_MAX - 1), "/dev/dsp%i", oss->dev_unit);
 
  WLog_INFO(TAG, "open: %s", dev_name);
 
  if ((pcm_handle = open(dev_name, O_RDONLY)) < 0)
  {
    OSS_LOG_ERR("sound dev open failed", errno);
    error = ERROR_INTERNAL_ERROR;
    goto err_out;
  }
 
  /* Set format. */
  tmp = audin_oss_get_format(&oss->format);
 
  if (ioctl(pcm_handle, SNDCTL_DSP_SETFMT, &tmp) == -1)
    OSS_LOG_ERR("SNDCTL_DSP_SETFMT failed", errno);
 
  tmp = oss->format.nChannels;
 
  if (ioctl(pcm_handle, SNDCTL_DSP_CHANNELS, &tmp) == -1)
    OSS_LOG_ERR("SNDCTL_DSP_CHANNELS failed", errno);
 
  tmp = oss->format.nSamplesPerSec;
 
  if (ioctl(pcm_handle, SNDCTL_DSP_SPEED, &tmp) == -1)
    OSS_LOG_ERR("SNDCTL_DSP_SPEED failed", errno);
 
  tmp = oss->format.nBlockAlign;
 
  if (ioctl(pcm_handle, SNDCTL_DSP_SETFRAGMENT, &tmp) == -1)
    OSS_LOG_ERR("SNDCTL_DSP_SETFRAGMENT failed", errno);
 
  buffer_size =
      (1ull * oss->FramesPerPacket * oss->format.nChannels * (oss->format.wBitsPerSample / 8ull));
  buffer = (BYTE*)calloc((buffer_size + sizeof(void*)), sizeof(BYTE));
 
  if (NULL == buffer)
  {
    OSS_LOG_ERR("malloc() fail", errno);
    error = ERROR_NOT_ENOUGH_MEMORY;
    goto err_out;
  }
 
  while (1)
  {
    SSIZE_T stmp = -1;
    status = WaitForSingleObject(oss->stopEvent, 0);
 
    if (status == WAIT_FAILED)
    {
      error = GetLastError();
      WLog_ERR(TAG, "WaitForSingleObject failed with error %" PRIu32 "", error);
      goto err_out;
    }
 
    if (status == WAIT_OBJECT_0)
      break;
 
    stmp = read(pcm_handle, buffer, buffer_size);
 
    /* Error happen. */
    if (stmp < 0)
    {
      OSS_LOG_ERR("read() error", errno);
      continue;
    }
 
    if ((size_t)stmp < buffer_size) /* Not enough data. */
      continue;
 
    if ((error = oss->receive(&oss->format, buffer, buffer_size, oss->user_data)))
    {
      WLog_ERR(TAG, "oss->receive failed with error %" PRIu32 "", error);
      break;
    }
  }
 
err_out:
 
  if (error && oss && oss->rdpcontext)
    setChannelError(oss->rdpcontext, error, "audin_oss_thread_func reported an error");
 
  if (pcm_handle != -1)
  {
    WLog_INFO(TAG, "close: %s", dev_name);
    close(pcm_handle);
  }
 
  free(buffer);
  ExitThread(error);
  return error;
}
 
static UINT audin_oss_open(IAudinDevice* device, AudinReceive receive, void* user_data)
{
  AudinOSSDevice* oss = (AudinOSSDevice*)device;
  oss->receive = receive;
  oss->user_data = user_data;
 
  if (!(oss->stopEvent = CreateEvent(NULL, TRUE, FALSE, NULL)))
  {
    WLog_ERR(TAG, "CreateEvent failed!");
    return ERROR_INTERNAL_ERROR;
  }
 
  if (!(oss->thread = CreateThread(NULL, 0, audin_oss_thread_func, oss, 0, NULL)))
  {
    WLog_ERR(TAG, "CreateThread failed!");
    (void)CloseHandle(oss->stopEvent);
    oss->stopEvent = NULL;
    return ERROR_INTERNAL_ERROR;
  }
 
  return CHANNEL_RC_OK;
}
 
static UINT audin_oss_close(IAudinDevice* device)
{
  UINT error = 0;
  AudinOSSDevice* oss = (AudinOSSDevice*)device;
 
  if (device == NULL)
    return ERROR_INVALID_PARAMETER;
 
  if (oss->stopEvent != NULL)
  {
    (void)SetEvent(oss->stopEvent);
 
    if (WaitForSingleObject(oss->thread, INFINITE) == WAIT_FAILED)
    {
      error = GetLastError();
      WLog_ERR(TAG, "WaitForSingleObject failed with error %" PRIu32 "", error);
      return error;
    }
 
    (void)CloseHandle(oss->stopEvent);
    oss->stopEvent = NULL;
    (void)CloseHandle(oss->thread);
    oss->thread = NULL;
  }
 
  oss->receive = NULL;
  oss->user_data = NULL;
  return CHANNEL_RC_OK;
}
 
static UINT audin_oss_free(IAudinDevice* device)
{
  AudinOSSDevice* oss = (AudinOSSDevice*)device;
  UINT error = 0;
 
  if (device == NULL)
    return ERROR_INVALID_PARAMETER;
 
  if ((error = audin_oss_close(device)))
  {
    WLog_ERR(TAG, "audin_oss_close failed with error code %" PRIu32 "!", error);
  }
 
  free(oss);
  return CHANNEL_RC_OK;
}
 
static UINT audin_oss_parse_addin_args(AudinOSSDevice* device, const ADDIN_ARGV* args)
{
  int status = 0;
  char* str_num = NULL;
  char* eptr = NULL;
  DWORD flags = 0;
  const COMMAND_LINE_ARGUMENT_A* arg = NULL;
  AudinOSSDevice* oss = device;
  COMMAND_LINE_ARGUMENT_A audin_oss_args[] = { { "dev", COMMAND_LINE_VALUE_REQUIRED, "<device>",
                                               NULL, NULL, -1, NULL, "audio device name" },
                                             { NULL, 0, NULL, NULL, NULL, -1, NULL, NULL } };
 
  flags =
      COMMAND_LINE_SIGIL_NONE | COMMAND_LINE_SEPARATOR_COLON | COMMAND_LINE_IGN_UNKNOWN_KEYWORD;
  status =
      CommandLineParseArgumentsA(args->argc, args->argv, audin_oss_args, flags, oss, NULL, NULL);
 
  if (status < 0)
    return ERROR_INVALID_PARAMETER;
 
  arg = audin_oss_args;
  errno = 0;
 
  do
  {
    if (!(arg->Flags & COMMAND_LINE_VALUE_PRESENT))
      continue;
 
    CommandLineSwitchStart(arg) CommandLineSwitchCase(arg, "dev")
    {
      str_num = _strdup(arg->Value);
 
      if (!str_num)
      {
        WLog_ERR(TAG, "_strdup failed!");
        return CHANNEL_RC_NO_MEMORY;
      }
 
      {
        long val = strtol(str_num, &eptr, 10);
 
        if ((errno != 0) || (val < INT32_MIN) || (val > INT32_MAX))
        {
          free(str_num);
          return CHANNEL_RC_NULL_DATA;
        }
 
        oss->dev_unit = (INT32)val;
      }
 
      if (oss->dev_unit < 0 || *eptr != '\0')
        oss->dev_unit = -1;
 
      free(str_num);
    }
    CommandLineSwitchEnd(arg)
  } while ((arg = CommandLineFindNextArgumentA(arg)) != NULL);
 
  return CHANNEL_RC_OK;
}
 
FREERDP_ENTRY_POINT(UINT VCAPITYPE oss_freerdp_audin_client_subsystem_entry(
    PFREERDP_AUDIN_DEVICE_ENTRY_POINTS pEntryPoints))
{
  const ADDIN_ARGV* args = NULL;
  AudinOSSDevice* oss = NULL;
  UINT error = 0;
  oss = (AudinOSSDevice*)calloc(1, sizeof(AudinOSSDevice));
 
  if (!oss)
  {
    WLog_ERR(TAG, "calloc failed!");
    return CHANNEL_RC_NO_MEMORY;
  }
 
  oss->iface.Open = audin_oss_open;
  oss->iface.FormatSupported = audin_oss_format_supported;
  oss->iface.SetFormat = audin_oss_set_format;
  oss->iface.Close = audin_oss_close;
  oss->iface.Free = audin_oss_free;
  oss->rdpcontext = pEntryPoints->rdpcontext;
  oss->dev_unit = -1;
  args = pEntryPoints->args;
 
  if ((error = audin_oss_parse_addin_args(oss, args)))
  {
    WLog_ERR(TAG, "audin_oss_parse_addin_args failed with errorcode %" PRIu32 "!", error);
    goto error_out;
  }
 
  if ((error = pEntryPoints->pRegisterAudinDevice(pEntryPoints->plugin, (IAudinDevice*)oss)))
  {
    WLog_ERR(TAG, "RegisterAudinDevice failed with error %" PRIu32 "!", error);
    goto error_out;
  }
 
  return CHANNEL_RC_OK;
error_out:
  free(oss);
  return error;
}