Milestone 5: deliver embedded RDP sessions and lifecycle hardening

third_party/FreeRDP/libfreerdp/codec/dsp_fdk_aac.c

@@ -0,0 +1,154 @@
+/**
+ * FreeRDP: A Remote Desktop Protocol Implementation
+ * Digital Sound Processing
+ *
+ * Copyright 2022 Armin Novak <anovak@thincast.com>
+ * Copyright 2022 Thincast Technologies GmbH
+ *
+ * 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 "dsp_fdk_aac.h"
+#include "dsp_fdk_impl.h"
+
+#include <freerdp/log.h>
+#define TAG FREERDP_TAG("dsp.fdk")
+
+WINPR_ATTR_FORMAT_ARG(2, 3)
+static void write_log(unsigned log_level, WINPR_FORMAT_ARG const char* fmt, ...)
+{
+	wLog* log = WLog_Get(TAG);
+
+	if (WLog_IsLevelActive(log, log_level))
+	{
+		char buffer[1024] = WINPR_C_ARRAY_INIT;
+
+		va_list ap = WINPR_C_ARRAY_INIT;
+		va_start(ap, fmt);
+		(void)vsnprintf(buffer, sizeof(buffer), fmt, ap);
+		va_end(ap);
+
+		WLog_PrintTextMessage(log, log_level, __LINE__, __FILE__, __func__, "%s", buffer);
+	}
+}
+
+BOOL fdk_aac_dsp_encode(FREERDP_DSP_COMMON_CONTEXT* context, const AUDIO_FORMAT* srcFormat,
+                        const BYTE* data, size_t length, wStream* out)
+{
+	WINPR_ASSERT(context);
+	WINPR_ASSERT(srcFormat);
+
+	if (srcFormat->wFormatTag != WAVE_FORMAT_PCM)
+	{
+		WLog_WARN(TAG, "Feeding %s format data to encoder function, but require %s",
+		          audio_format_get_tag_string(srcFormat->wFormatTag),
+		          audio_format_get_tag_string(WAVE_FORMAT_PCM));
+		return FALSE;
+	}
+
+	if (!context->fdkSetup)
+	{
+		ssize_t rc = fdk_aac_dsp_impl_config(
+		    context->fdkAacInstance, &context->buffersize, context->encoder,
+		    context->format.nSamplesPerSec, context->format.nChannels,
+		    context->format.nAvgBytesPerSec, context->frames_per_packet, write_log);
+		if (rc < 0)
+			return FALSE;
+
+		context->fdkSetup = TRUE;
+	}
+
+	if (!Stream_EnsureRemainingCapacity(out, context->buffersize))
+		return FALSE;
+
+	{
+		const ssize_t encoded =
+		    fdk_aac_dsp_impl_encode(context->fdkAacInstance, data, length, Stream_Pointer(out),
+		                            Stream_GetRemainingCapacity(out), write_log);
+		if (encoded < 0)
+			return FALSE;
+		Stream_Seek(out, (size_t)encoded);
+		return TRUE;
+	}
+}
+
+BOOL fdk_aac_dsp_decode(FREERDP_DSP_COMMON_CONTEXT* context, const AUDIO_FORMAT* srcFormat,
+                        const BYTE* data, size_t length, wStream* out)
+{
+	WINPR_ASSERT(context);
+	WINPR_ASSERT(srcFormat);
+
+	if (srcFormat->wFormatTag != WAVE_FORMAT_AAC_MS)
+	{
+		WLog_WARN(TAG, "Feeding %s format data to encoder function, but require %s",
+		          audio_format_get_tag_string(srcFormat->wFormatTag),
+		          audio_format_get_tag_string(WAVE_FORMAT_AAC_MS));
+		return FALSE;
+	}
+
+	if (!context->fdkSetup)
+	{
+		ssize_t rc = fdk_aac_dsp_impl_config(
+		    context->fdkAacInstance, &context->buffersize, context->encoder,
+		    context->format.nSamplesPerSec, context->format.nChannels,
+		    context->format.nAvgBytesPerSec, context->frames_per_packet, write_log);
+		if (rc < 0)
+			return FALSE;
+
+		context->fdkSetup = TRUE;
+	}
+
+	ssize_t rest = 0;
+	do
+	{
+		rest = fdk_aac_dsp_impl_decode_fill(context->fdkAacInstance, data, length, write_log);
+		if (rest < 0)
+		{
+			WLog_WARN(TAG, "DecodeFill() failed");
+			return FALSE;
+		}
+
+		ssize_t ret = -1;
+		do
+		{
+			const size_t expect = context->buffersize;
+			if (!Stream_EnsureRemainingCapacity(out, expect))
+				return FALSE;
+
+			ret = fdk_aac_dsp_impl_decode_read(context->fdkAacInstance, Stream_Pointer(out), expect,
+			                                   write_log);
+			if (ret < 0)
+				return FALSE;
+
+			Stream_Seek(out, (size_t)ret);
+		} while (ret > 0);
+	} while (rest > 0);
+
+	return TRUE;
+}
+
+void fdk_aac_dsp_uninit(FREERDP_DSP_COMMON_CONTEXT* context)
+{
+	WINPR_ASSERT(context);
+
+	fdk_aac_dsp_impl_uninit(&context->fdkAacInstance, context->encoder, write_log);
+}
+
+BOOL fdk_aac_dsp_init(FREERDP_DSP_COMMON_CONTEXT* context, size_t frames_per_packet)
+{
+	WINPR_ASSERT(context);
+	context->fdkSetup = FALSE;
+	WINPR_ASSERT(frames_per_packet <= UINT_MAX);
+	context->frames_per_packet = (unsigned)frames_per_packet;
+	return fdk_aac_dsp_impl_init(&context->fdkAacInstance, context->encoder, write_log);
+}