/**

 * 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.

 */

#ifdef HAVE_CONFIG_H

#include "config.h"

#endif

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

#if defined(__OpenBSD__)

#include <soundcard.h>

#else

#include <sys/soundcard.h>

#endif

#include <sys/ioctl.h>

#include <freerdp/addin.h>

#include <freerdp/channels/rdpsnd.h>

#include "audin_main.h"

typedef struct _AudinOSSDevice

{

	IAudinDevice iface;

	HANDLE thread;

	HANDLE stopEvent;

	AUDIO_FORMAT format;

	UINT32 FramesPerPacket;

	int dev_unit;

	AudinReceive receive;

	void* user_data;

	rdpContext* rdpcontext;

} AudinOSSDevice;

#define OSS_LOG_ERR(_text, _error) \

	if (_error != 0)               \

		WLog_ERR(TAG, "%s: %i - %s\n", _text, _error, strerror(_error));

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;

			}

			break;

		case WAVE_FORMAT_ALAW:

			return AFMT_A_LAW;

		case WAVE_FORMAT_MULAW:

			return AFMT_MU_LAW;

	}

	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;

		case WAVE_FORMAT_ALAW:

		case WAVE_FORMAT_MULAW:

			return TRUE;

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

	char mixer_name[PATH_MAX] = "/dev/mixer";

	int pcm_handle = -1, mixer_handle;

	BYTE* buffer = NULL;

	unsigned long tmp;

	size_t buffer_size;

	AudinOSSDevice* oss = (AudinOSSDevice*)arg;

	UINT error = 0;

	DWORD status;

	if (oss == NULL)

	{

		error = ERROR_INVALID_PARAMETER;

		goto err_out;

	}

	if (oss->dev_unit != -1)

	{

		sprintf_s(dev_name, (PATH_MAX - 1), "/dev/dsp%i", oss->dev_unit);

		sprintf_s(mixer_name, PATH_MAX - 1, "/dev/mixer%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 rec volume to 100%. */

	if ((mixer_handle = open(mixer_name, O_RDWR)) < 0)

	{

		OSS_LOG_ERR("mixer open failed, not critical", errno);

	}

	else

	{

		tmp = (100 | (100 << 8));

		if (ioctl(mixer_handle, MIXER_WRITE(SOUND_MIXER_MIC), &tmp) == -1)

			OSS_LOG_ERR("WRITE_MIXER - SOUND_MIXER_MIC, not critical", errno);

		tmp = (100 | (100 << 8));

		if (ioctl(mixer_handle, MIXER_WRITE(SOUND_MIXER_RECLEV), &tmp) == -1)

			OSS_LOG_ERR("WRITE_MIXER - SOUND_MIXER_RECLEV, not critical", errno);

		close(mixer_handle);

	}

#if 0 /* FreeBSD OSS implementation at this moment (2015.03) does not set PCM_CAP_INPUT flag. */

	tmp = 0;

	if (ioctl(pcm_handle, SNDCTL_DSP_GETCAPS, &tmp) == -1)

	{

		OSS_LOG_ERR("SNDCTL_DSP_GETCAPS failed, try ignory", errno);

	}

	else if ((tmp & PCM_CAP_INPUT) == 0)

	{

		OSS_LOG_ERR("Device does not supports playback", EOPNOTSUPP);

		goto err_out;

	}

#endif

	/* 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 = (oss->FramesPerPacket * oss->format.nChannels * (oss->format.wBitsPerSample / 8));

	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;

		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 enouth 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!");

		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;

	AudinOSSDevice* oss = (AudinOSSDevice*)device;

	if (device == NULL)

		return ERROR_INVALID_PARAMETER;

	if (oss->stopEvent != NULL)

	{

		SetEvent(oss->stopEvent);

		if (WaitForSingleObject(oss->thread, INFINITE) == WAIT_FAILED)

		{

			error = GetLastError();

			WLog_ERR(TAG, "WaitForSingleObject failed with error %" PRIu32 "", error);

			return error;

		}

		CloseHandle(oss->stopEvent);

		oss->stopEvent = NULL;

		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;

	if (device == NULL)

		return ERROR_INVALID_PARAMETER;

	if ((error = audin_oss_close(device)))

	{

		WLog_ERR(TAG, "audin_oss_close failed with error code %d!", 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, ADDIN_ARGV* args)

{

	int status;

	char *str_num, *eptr;

	DWORD flags;

	COMMAND_LINE_ARGUMENT_A* arg;

	AudinOSSDevice* oss = (AudinOSSDevice*)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;

}

#ifdef BUILTIN_CHANNELS

#define freerdp_audin_client_subsystem_entry oss_freerdp_audin_client_subsystem_entry

#else

#define freerdp_audin_client_subsystem_entry FREERDP_API freerdp_audin_client_subsystem_entry

#endif

/**

 * Function description

 *

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

 */

UINT freerdp_audin_client_subsystem_entry(PFREERDP_AUDIN_DEVICE_ENTRY_POINTS pEntryPoints)

{

	ADDIN_ARGV* args;

	AudinOSSDevice* oss;

	UINT error;

	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;

}