/**

 * FreeRDP: A Remote Desktop Protocol Implementation

 * Audio Input Redirection Virtual Channel - OSS implementation

 *

 * Copyright (c) 2015 Rozhuk Ivan <rozhuk.im@gmail.com>

 * Copyright 2015 Thincast Technologies GmbH

 * Copyright 2015 DI (FH) Martin Haimberger <martin.haimberger@thincast.com>

 *

 * Licensed under the Apache License, Version 2.0 (the "License");

 * you may not use this file except in compliance with the License.

 * You may obtain a copy of the License at

 *

 *     http://www.apache.org/licenses/LICENSE-2.0

 *

 * Unless required by applicable law or agreed to in writing, software

 * distributed under the License is distributed on an "AS IS" BASIS,

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

 * See the License for the specific language governing permissions and

 * limitations under the License.

 */

#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;

}

/**

 * Function description

 *

 * @return 0 on success, otherwise a Win32 error code

 */

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;

}

/**

 * Function description

 *

 * @return 0 on success, otherwise a Win32 error code

 */

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;

}

/**

 * Function description

 *

 * @return 0 on success, otherwise a Win32 error code

 */

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;

}

/**

 * Function description

 *

 * @return 0 on success, otherwise a Win32 error code

 */

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;

}

/**

 * Function description

 *

 * @return 0 on success, otherwise a Win32 error code

 */

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;

}

/**

 * Function description

 *

 * @return 0 on success, otherwise a Win32 error code

 */

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;

}