1 /*
2  * Copyright 2016 The Android Open Source Project
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 #define LOG_TAG "bluetooth-a2dp"
18 #define ATRACE_TAG ATRACE_TAG_AUDIO
19 
20 #include "a2dp_vendor_ldac_encoder.h"
21 
22 #include <stdio.h>
23 
24 #include <cstdint>
25 #include <string>
26 
27 #include "a2dp_codec_api.h"
28 #include "a2dp_vendor_ldac_constants.h"
29 #include "avdt_api.h"
30 #include "hardware/bt_av.h"
31 #include "ldacBT.h"
32 
33 #ifdef __ANDROID__
34 #include <cutils/trace.h>
35 #endif
36 #include <bluetooth/log.h>
37 #include <inttypes.h>
38 #include <ldacBT_abr.h>
39 #include <string.h>
40 
41 #include "a2dp_vendor_ldac.h"
42 #include "common/time_util.h"
43 #include "internal_include/bt_target.h"
44 #include "osi/include/allocator.h"
45 #include "osi/include/properties.h"
46 #include "stack/include/bt_hdr.h"
47 
48 //
49 // Encoder for LDAC Source Codec
50 //
51 
52 // Initial EQMID for ABR mode.
53 #define LDAC_ABR_MODE_EQMID LDACBT_EQMID_SQ
54 
55 // A2DP LDAC encoder interval in milliseconds
56 #define A2DP_LDAC_ENCODER_INTERVAL_MS 20
57 #define A2DP_LDAC_MEDIA_BYTES_PER_FRAME 128
58 
59 // offset
60 #define A2DP_LDAC_OFFSET (AVDT_MEDIA_OFFSET + A2DP_LDAC_MPL_HDR_LEN)
61 
62 using namespace bluetooth;
63 
64 namespace std {
65 template <>
66 struct formatter<LDACBT_SMPL_FMT_T> : enum_formatter<LDACBT_SMPL_FMT_T> {};
67 }  // namespace std
68 
69 typedef struct {
70   uint32_t sample_rate;
71   uint8_t channel_mode;
72   uint8_t bits_per_sample;
73   int quality_mode_index;
74   int pcm_wlength;
75   LDACBT_SMPL_FMT_T pcm_fmt;
76 } tA2DP_LDAC_ENCODER_PARAMS;
77 
78 typedef struct {
79   float counter;
80   uint32_t bytes_per_tick; /* pcm bytes read each media task tick */
81   uint64_t last_frame_us;
82 } tA2DP_LDAC_FEEDING_STATE;
83 
84 typedef struct {
85   uint64_t session_start_us;
86 
87   size_t media_read_total_expected_packets;
88   size_t media_read_total_expected_reads_count;
89   size_t media_read_total_expected_read_bytes;
90 
91   size_t media_read_total_dropped_packets;
92   size_t media_read_total_actual_reads_count;
93   size_t media_read_total_actual_read_bytes;
94 } a2dp_ldac_encoder_stats_t;
95 
96 typedef struct {
97   a2dp_source_read_callback_t read_callback;
98   a2dp_source_enqueue_callback_t enqueue_callback;
99   uint16_t TxAaMtuSize;
100   size_t TxQueueLength;
101 
102   bool use_SCMS_T;
103   tA2DP_ENCODER_INIT_PEER_PARAMS peer_params;
104   uint32_t timestamp;  // Timestamp for the A2DP frames
105 
106   HANDLE_LDAC_BT ldac_handle;
107   bool has_ldac_handle;  // True if ldac_handle is valid
108 
109   HANDLE_LDAC_ABR ldac_abr_handle;
110   bool has_ldac_abr_handle;
111   int last_ldac_abr_eqmid;
112   size_t ldac_abr_adjustments;
113 
114   tA2DP_FEEDING_PARAMS feeding_params;
115   tA2DP_LDAC_ENCODER_PARAMS ldac_encoder_params;
116   tA2DP_LDAC_FEEDING_STATE ldac_feeding_state;
117 
118   a2dp_ldac_encoder_stats_t stats;
119 } tA2DP_LDAC_ENCODER_CB;
120 
121 static bool ldac_abr_loaded = true;  // the library is statically linked
122 
123 static tA2DP_LDAC_ENCODER_CB a2dp_ldac_encoder_cb;
124 
125 static void a2dp_vendor_ldac_encoder_update(A2dpCodecConfig* a2dp_codec_config,
126                                             bool* p_restart_input, bool* p_restart_output,
127                                             bool* p_config_updated);
128 static void a2dp_ldac_get_num_frame_iteration(uint8_t* num_of_iterations, uint8_t* num_of_frames,
129                                               uint64_t timestamp_us);
130 static void a2dp_ldac_encode_frames(uint8_t nb_frame);
131 static bool a2dp_ldac_read_feeding(uint8_t* read_buffer, uint32_t* bytes_read);
132 static uint16_t adjust_effective_mtu(const tA2DP_ENCODER_INIT_PEER_PARAMS& peer_params);
133 static std::string quality_mode_index_to_name(int quality_mode_index);
134 
A2DP_VendorLoadEncoderLdac(void)135 bool A2DP_VendorLoadEncoderLdac(void) {
136   // Nothing to do - the library is statically linked
137   return true;
138 }
139 
A2DP_VendorUnloadEncoderLdac(void)140 void A2DP_VendorUnloadEncoderLdac(void) {
141   // Cleanup any LDAC-related state
142   a2dp_vendor_ldac_encoder_cleanup();
143 }
144 
a2dp_vendor_ldac_encoder_init(const tA2DP_ENCODER_INIT_PEER_PARAMS * p_peer_params,A2dpCodecConfig * a2dp_codec_config,a2dp_source_read_callback_t read_callback,a2dp_source_enqueue_callback_t enqueue_callback)145 void a2dp_vendor_ldac_encoder_init(const tA2DP_ENCODER_INIT_PEER_PARAMS* p_peer_params,
146                                    A2dpCodecConfig* a2dp_codec_config,
147                                    a2dp_source_read_callback_t read_callback,
148                                    a2dp_source_enqueue_callback_t enqueue_callback) {
149   a2dp_vendor_ldac_encoder_cleanup();
150 
151   a2dp_ldac_encoder_cb.stats.session_start_us = bluetooth::common::time_get_os_boottime_us();
152 
153   a2dp_ldac_encoder_cb.read_callback = read_callback;
154   a2dp_ldac_encoder_cb.enqueue_callback = enqueue_callback;
155   a2dp_ldac_encoder_cb.peer_params = *p_peer_params;
156   a2dp_ldac_encoder_cb.timestamp = 0;
157   a2dp_ldac_encoder_cb.ldac_abr_handle = NULL;
158   a2dp_ldac_encoder_cb.has_ldac_abr_handle = false;
159   a2dp_ldac_encoder_cb.last_ldac_abr_eqmid = -1;
160   a2dp_ldac_encoder_cb.ldac_abr_adjustments = 0;
161 
162   a2dp_ldac_encoder_cb.use_SCMS_T = false;
163 
164   // NOTE: Ignore the restart_input / restart_output flags - this initization
165   // happens when the audio session is (re)started.
166   bool restart_input = false;
167   bool restart_output = false;
168   bool config_updated = false;
169   a2dp_vendor_ldac_encoder_update(a2dp_codec_config, &restart_input, &restart_output,
170                                   &config_updated);
171 }
172 
173 // Update the A2DP LDAC encoder.
174 // |a2dp_codec_config| is the A2DP codec to use for the update.
a2dp_vendor_ldac_encoder_update(A2dpCodecConfig * a2dp_codec_config,bool * p_restart_input,bool * p_restart_output,bool * p_config_updated)175 static void a2dp_vendor_ldac_encoder_update(A2dpCodecConfig* a2dp_codec_config,
176                                             bool* p_restart_input, bool* p_restart_output,
177                                             bool* p_config_updated) {
178   tA2DP_LDAC_ENCODER_PARAMS* p_encoder_params = &a2dp_ldac_encoder_cb.ldac_encoder_params;
179   uint8_t codec_info[AVDT_CODEC_SIZE];
180 
181   *p_restart_input = false;
182   *p_restart_output = false;
183   *p_config_updated = false;
184 
185   if (!a2dp_ldac_encoder_cb.has_ldac_handle) {
186     a2dp_ldac_encoder_cb.ldac_handle = ldacBT_get_handle();
187     if (a2dp_ldac_encoder_cb.ldac_handle == NULL) {
188       log::error("Cannot get LDAC encoder handle");
189       return;  // TODO: Return an error?
190     }
191     a2dp_ldac_encoder_cb.has_ldac_handle = true;
192   }
193   log::assert_that(a2dp_ldac_encoder_cb.ldac_handle != nullptr,
194                    "assert failed: a2dp_ldac_encoder_cb.ldac_handle != nullptr");
195 
196   if (!a2dp_codec_config->copyOutOtaCodecConfig(codec_info)) {
197     log::error("Cannot update the codec encoder for {}: invalid codec config",
198                a2dp_codec_config->name());
199     return;
200   }
201   const uint8_t* p_codec_info = codec_info;
202   btav_a2dp_codec_config_t codec_config = a2dp_codec_config->getCodecConfig();
203 
204   // The feeding parameters
205   tA2DP_FEEDING_PARAMS* p_feeding_params = &a2dp_ldac_encoder_cb.feeding_params;
206   p_feeding_params->sample_rate = A2DP_VendorGetTrackSampleRateLdac(p_codec_info);
207   p_feeding_params->bits_per_sample = a2dp_codec_config->getAudioBitsPerSample();
208   p_feeding_params->channel_count = A2DP_VendorGetTrackChannelCountLdac(p_codec_info);
209   log::info("sample_rate={} bits_per_sample={} channel_count={}", p_feeding_params->sample_rate,
210             p_feeding_params->bits_per_sample, p_feeding_params->channel_count);
211   a2dp_vendor_ldac_feeding_reset();
212 
213   // The codec parameters
214   p_encoder_params->sample_rate = a2dp_ldac_encoder_cb.feeding_params.sample_rate;
215   p_encoder_params->channel_mode = A2DP_VendorGetChannelModeCodeLdac(p_codec_info);
216 
217   // Set the quality mode index
218   int old_quality_mode_index = p_encoder_params->quality_mode_index;
219   if (codec_config.codec_specific_1 != 0) {
220     p_encoder_params->quality_mode_index = codec_config.codec_specific_1 % 10;
221     log::info("setting quality mode to {}",
222               quality_mode_index_to_name(p_encoder_params->quality_mode_index));
223   } else {
224     p_encoder_params->quality_mode_index = osi_property_get_int32(
225             "persist.bluetooth.a2dp_ldac.default_quality_mode", A2DP_LDAC_QUALITY_ABR);
226     log::info("setting quality mode to default {}",
227               quality_mode_index_to_name(p_encoder_params->quality_mode_index));
228   }
229 
230   int ldac_eqmid = LDAC_ABR_MODE_EQMID;
231   if (p_encoder_params->quality_mode_index == A2DP_LDAC_QUALITY_ABR) {
232     if (!ldac_abr_loaded) {
233       p_encoder_params->quality_mode_index = A2DP_LDAC_QUALITY_MID;
234       log::warn("LDAC ABR library is not loaded, resetting quality mode to {}",
235                 quality_mode_index_to_name(p_encoder_params->quality_mode_index));
236     } else {
237       log::info("changing mode from {} to {}", quality_mode_index_to_name(old_quality_mode_index),
238                 quality_mode_index_to_name(p_encoder_params->quality_mode_index));
239       if (a2dp_ldac_encoder_cb.ldac_abr_handle != NULL) {
240         log::info("already in LDAC ABR mode, do nothing.");
241       } else {
242         log::info("get and init LDAC ABR handle.");
243         a2dp_ldac_encoder_cb.ldac_abr_handle = ldac_ABR_get_handle();
244         if (a2dp_ldac_encoder_cb.ldac_abr_handle != NULL) {
245           a2dp_ldac_encoder_cb.has_ldac_abr_handle = true;
246           a2dp_ldac_encoder_cb.last_ldac_abr_eqmid = -1;
247           a2dp_ldac_encoder_cb.ldac_abr_adjustments = 0;
248           ldac_ABR_Init(a2dp_ldac_encoder_cb.ldac_abr_handle, A2DP_LDAC_ENCODER_INTERVAL_MS);
249         } else {
250           p_encoder_params->quality_mode_index = A2DP_LDAC_QUALITY_MID;
251           log::info("get LDAC ABR handle failed, resetting quality mode to {}.",
252                     quality_mode_index_to_name(p_encoder_params->quality_mode_index));
253         }
254       }
255     }
256   } else {
257     ldac_eqmid = p_encoder_params->quality_mode_index;
258     log::info("in {} mode, free LDAC ABR handle.", quality_mode_index_to_name(ldac_eqmid));
259     if (a2dp_ldac_encoder_cb.has_ldac_abr_handle) {
260       ldac_ABR_free_handle(a2dp_ldac_encoder_cb.ldac_abr_handle);
261       a2dp_ldac_encoder_cb.ldac_abr_handle = NULL;
262       a2dp_ldac_encoder_cb.has_ldac_abr_handle = false;
263       a2dp_ldac_encoder_cb.last_ldac_abr_eqmid = -1;
264       a2dp_ldac_encoder_cb.ldac_abr_adjustments = 0;
265     }
266   }
267 
268   if (p_encoder_params->quality_mode_index != old_quality_mode_index) {
269     *p_config_updated = true;
270   }
271 
272   p_encoder_params->pcm_wlength = a2dp_ldac_encoder_cb.feeding_params.bits_per_sample >> 3;
273   // Set the Audio format from pcm_wlength
274   p_encoder_params->pcm_fmt = LDACBT_SMPL_FMT_S16;
275   if (p_encoder_params->pcm_wlength == 2) {
276     p_encoder_params->pcm_fmt = LDACBT_SMPL_FMT_S16;
277   } else if (p_encoder_params->pcm_wlength == 3) {
278     p_encoder_params->pcm_fmt = LDACBT_SMPL_FMT_S24;
279   } else if (p_encoder_params->pcm_wlength == 4) {
280     p_encoder_params->pcm_fmt = LDACBT_SMPL_FMT_S32;
281   }
282 
283   const tA2DP_ENCODER_INIT_PEER_PARAMS& peer_params = a2dp_ldac_encoder_cb.peer_params;
284   a2dp_ldac_encoder_cb.TxAaMtuSize = adjust_effective_mtu(peer_params);
285   log::info("MTU={}, peer_mtu={}", a2dp_ldac_encoder_cb.TxAaMtuSize, peer_params.peer_mtu);
286   log::info(
287           "sample_rate: {} channel_mode: {} quality_mode_index: {} pcm_wlength: {} "
288           "pcm_fmt: {}",
289           p_encoder_params->sample_rate, p_encoder_params->channel_mode,
290           p_encoder_params->quality_mode_index, p_encoder_params->pcm_wlength,
291           p_encoder_params->pcm_fmt);
292 
293   // Initialize the encoder.
294   // NOTE: MTU in the initialization must include the AVDT media header size.
295   int result = ldacBT_init_handle_encode(a2dp_ldac_encoder_cb.ldac_handle,
296                                          a2dp_ldac_encoder_cb.TxAaMtuSize + AVDT_MEDIA_HDR_SIZE,
297                                          ldac_eqmid, p_encoder_params->channel_mode,
298                                          p_encoder_params->pcm_fmt, p_encoder_params->sample_rate);
299   if (result != 0) {
300     int err_code = ldacBT_get_error_code(a2dp_ldac_encoder_cb.ldac_handle);
301     log::error(
302             "error initializing the LDAC encoder: {} api_error = {} handle_error = "
303             "{} block_error = {} error_code = 0x{:x}",
304             result, LDACBT_API_ERR(err_code), LDACBT_HANDLE_ERR(err_code),
305             LDACBT_BLOCK_ERR(err_code), err_code);
306   }
307 }
308 
a2dp_vendor_ldac_encoder_cleanup(void)309 void a2dp_vendor_ldac_encoder_cleanup(void) {
310   if (a2dp_ldac_encoder_cb.has_ldac_abr_handle) {
311     ldac_ABR_free_handle(a2dp_ldac_encoder_cb.ldac_abr_handle);
312   }
313   if (a2dp_ldac_encoder_cb.has_ldac_handle) {
314     ldacBT_free_handle(a2dp_ldac_encoder_cb.ldac_handle);
315   }
316   memset(&a2dp_ldac_encoder_cb, 0, sizeof(a2dp_ldac_encoder_cb));
317 }
318 
a2dp_vendor_ldac_feeding_reset(void)319 void a2dp_vendor_ldac_feeding_reset(void) {
320   /* By default, just clear the entire state */
321   memset(&a2dp_ldac_encoder_cb.ldac_feeding_state, 0,
322          sizeof(a2dp_ldac_encoder_cb.ldac_feeding_state));
323 
324   a2dp_ldac_encoder_cb.ldac_feeding_state.bytes_per_tick =
325           (a2dp_ldac_encoder_cb.feeding_params.sample_rate *
326            a2dp_ldac_encoder_cb.feeding_params.bits_per_sample / 8 *
327            a2dp_ldac_encoder_cb.feeding_params.channel_count * A2DP_LDAC_ENCODER_INTERVAL_MS) /
328           1000;
329 
330   log::info("PCM bytes per tick {}", a2dp_ldac_encoder_cb.ldac_feeding_state.bytes_per_tick);
331 }
332 
a2dp_vendor_ldac_feeding_flush(void)333 void a2dp_vendor_ldac_feeding_flush(void) {
334   a2dp_ldac_encoder_cb.ldac_feeding_state.counter = 0.0f;
335 }
336 
a2dp_vendor_ldac_get_encoder_interval_ms(void)337 uint64_t a2dp_vendor_ldac_get_encoder_interval_ms(void) { return A2DP_LDAC_ENCODER_INTERVAL_MS; }
338 
a2dp_vendor_ldac_get_effective_frame_size()339 int a2dp_vendor_ldac_get_effective_frame_size() { return a2dp_ldac_encoder_cb.TxAaMtuSize; }
340 
a2dp_vendor_ldac_send_frames(uint64_t timestamp_us)341 void a2dp_vendor_ldac_send_frames(uint64_t timestamp_us) {
342   uint8_t nb_frame = 0;
343   uint8_t nb_iterations = 0;
344 
345   a2dp_ldac_get_num_frame_iteration(&nb_iterations, &nb_frame, timestamp_us);
346   log::verbose("Sending {} frames per iteration, {} iterations", nb_frame, nb_iterations);
347   if (nb_frame == 0) {
348     return;
349   }
350 
351   for (uint8_t counter = 0; counter < nb_iterations; counter++) {
352     if (a2dp_ldac_encoder_cb.has_ldac_abr_handle) {
353       int flag_enable = 1;
354       int prev_eqmid = a2dp_ldac_encoder_cb.last_ldac_abr_eqmid;
355       a2dp_ldac_encoder_cb.last_ldac_abr_eqmid =
356               ldac_ABR_Proc(a2dp_ldac_encoder_cb.ldac_handle, a2dp_ldac_encoder_cb.ldac_abr_handle,
357                             a2dp_ldac_encoder_cb.TxQueueLength, flag_enable);
358       if (prev_eqmid != a2dp_ldac_encoder_cb.last_ldac_abr_eqmid) {
359         a2dp_ldac_encoder_cb.ldac_abr_adjustments++;
360       }
361 #ifdef __ANDROID__
362       ATRACE_INT("LDAC ABR level", a2dp_ldac_encoder_cb.last_ldac_abr_eqmid);
363 #endif
364     }
365     // Transcode frame and enqueue
366     a2dp_ldac_encode_frames(nb_frame);
367   }
368 }
369 
370 // Obtains the number of frames to send and number of iterations
371 // to be used. |num_of_iterations| and |num_of_frames| parameters
372 // are used as output param for returning the respective values.
a2dp_ldac_get_num_frame_iteration(uint8_t * num_of_iterations,uint8_t * num_of_frames,uint64_t timestamp_us)373 static void a2dp_ldac_get_num_frame_iteration(uint8_t* num_of_iterations, uint8_t* num_of_frames,
374                                               uint64_t timestamp_us) {
375   uint32_t result = 0;
376   uint8_t nof = 0;
377   uint8_t noi = 1;
378 
379   uint32_t pcm_bytes_per_frame = A2DP_LDAC_MEDIA_BYTES_PER_FRAME *
380                                  a2dp_ldac_encoder_cb.feeding_params.channel_count *
381                                  a2dp_ldac_encoder_cb.feeding_params.bits_per_sample / 8;
382   log::verbose("pcm_bytes_per_frame {}", pcm_bytes_per_frame);
383 
384   uint32_t us_this_tick = A2DP_LDAC_ENCODER_INTERVAL_MS * 1000;
385   uint64_t now_us = timestamp_us;
386   if (a2dp_ldac_encoder_cb.ldac_feeding_state.last_frame_us != 0) {
387     us_this_tick = (now_us - a2dp_ldac_encoder_cb.ldac_feeding_state.last_frame_us);
388   }
389   a2dp_ldac_encoder_cb.ldac_feeding_state.last_frame_us = now_us;
390 
391   a2dp_ldac_encoder_cb.ldac_feeding_state.counter +=
392           (float)a2dp_ldac_encoder_cb.ldac_feeding_state.bytes_per_tick * us_this_tick /
393           (A2DP_LDAC_ENCODER_INTERVAL_MS * 1000);
394 
395   result = a2dp_ldac_encoder_cb.ldac_feeding_state.counter / pcm_bytes_per_frame;
396   a2dp_ldac_encoder_cb.ldac_feeding_state.counter -= result * pcm_bytes_per_frame;
397   nof = result;
398 
399   log::verbose("effective num of frames {}, iterations {}", nof, noi);
400 
401   *num_of_frames = nof;
402   *num_of_iterations = noi;
403 }
404 
a2dp_ldac_encode_frames(uint8_t nb_frame)405 static void a2dp_ldac_encode_frames(uint8_t nb_frame) {
406   tA2DP_LDAC_ENCODER_PARAMS* p_encoder_params = &a2dp_ldac_encoder_cb.ldac_encoder_params;
407   uint8_t remain_nb_frame = nb_frame;
408   uint16_t ldac_frame_size;
409   uint8_t read_buffer[LDACBT_MAX_LSU * 4 /* byte/sample */ * 2 /* ch */];
410 
411   switch (p_encoder_params->sample_rate) {
412     case 176400:
413     case 192000:
414       ldac_frame_size = 512;  // sample/ch
415       break;
416     case 88200:
417     case 96000:
418       ldac_frame_size = 256;  // sample/ch
419       break;
420     case 44100:
421     case 48000:
422     default:
423       ldac_frame_size = 128;  // sample/ch
424       break;
425   }
426 
427   uint32_t count;
428   int32_t encode_count = 0;
429   int32_t out_frames = 0;
430   int written = 0;
431 
432   uint32_t bytes_read = 0;
433   while (nb_frame) {
434     BT_HDR* p_buf = (BT_HDR*)osi_malloc(BT_DEFAULT_BUFFER_SIZE);
435     p_buf->offset = A2DP_LDAC_OFFSET;
436     p_buf->len = 0;
437     p_buf->layer_specific = 0;
438     a2dp_ldac_encoder_cb.stats.media_read_total_expected_packets++;
439 
440     count = 0;
441     do {
442       //
443       // Read the PCM data and encode it
444       //
445       uint32_t temp_bytes_read = 0;
446       if (a2dp_ldac_read_feeding(read_buffer, &temp_bytes_read)) {
447         bytes_read += temp_bytes_read;
448         uint8_t* packet = (uint8_t*)(p_buf + 1) + p_buf->offset + p_buf->len;
449         if (a2dp_ldac_encoder_cb.ldac_handle == NULL) {
450           log::error("invalid LDAC handle");
451           a2dp_ldac_encoder_cb.stats.media_read_total_dropped_packets++;
452           osi_free(p_buf);
453           return;
454         }
455         int result =
456                 ldacBT_encode(a2dp_ldac_encoder_cb.ldac_handle, read_buffer, (int*)&encode_count,
457                               packet + count, (int*)&written, (int*)&out_frames);
458         if (result != 0) {
459           int err_code = ldacBT_get_error_code(a2dp_ldac_encoder_cb.ldac_handle);
460           log::error(
461                   "LDAC encoding error: {} api_error = {} handle_error = {} "
462                   "block_error = {} error_code = 0x{:x}",
463                   result, LDACBT_API_ERR(err_code), LDACBT_HANDLE_ERR(err_code),
464                   LDACBT_BLOCK_ERR(err_code), err_code);
465           a2dp_ldac_encoder_cb.stats.media_read_total_dropped_packets++;
466           osi_free(p_buf);
467           return;
468         }
469         count += written;
470         p_buf->len += written;
471         nb_frame--;
472         p_buf->layer_specific += out_frames;  // added a frame to the buffer
473       } else {
474         log::warn("underflow {}", nb_frame);
475         a2dp_ldac_encoder_cb.ldac_feeding_state.counter +=
476                 nb_frame * LDACBT_ENC_LSU * a2dp_ldac_encoder_cb.feeding_params.channel_count *
477                 a2dp_ldac_encoder_cb.feeding_params.bits_per_sample / 8;
478 
479         // no more pcm to read
480         nb_frame = 0;
481       }
482     } while ((written == 0) && nb_frame);
483 
484     if (p_buf->len) {
485       /*
486        * Timestamp of the media packet header represent the TS of the
487        * first frame, i.e the timestamp before including this frame.
488        */
489       *((uint32_t*)(p_buf + 1)) = a2dp_ldac_encoder_cb.timestamp;
490 
491       // Timestamp will wrap over to 0 if stream continues on long enough
492       // (>25H @ 48KHz). The parameters are promoted to 64bit to ensure that
493       // no unsigned overflow is triggered as ubsan is always enabled.
494       a2dp_ldac_encoder_cb.timestamp = ((uint64_t)a2dp_ldac_encoder_cb.timestamp +
495                                         (p_buf->layer_specific * ldac_frame_size)) &
496                                        UINT32_MAX;
497 
498       uint8_t done_nb_frame = remain_nb_frame - nb_frame;
499       remain_nb_frame = nb_frame;
500       if (!a2dp_ldac_encoder_cb.enqueue_callback(p_buf, done_nb_frame, bytes_read)) {
501         return;
502       }
503     } else {
504       // NOTE: Unlike the execution path for other codecs, it is normal for
505       // LDAC to NOT write encoded data to the last buffer if there wasn't
506       // enough data to write to. That data is accumulated internally by
507       // the codec and included in the next iteration. Therefore, here we
508       // don't increment the "media_read_total_dropped_packets" counter.
509       osi_free(p_buf);
510     }
511   }
512 }
513 
a2dp_ldac_read_feeding(uint8_t * read_buffer,uint32_t * bytes_read)514 static bool a2dp_ldac_read_feeding(uint8_t* read_buffer, uint32_t* bytes_read) {
515   uint32_t read_size = LDACBT_ENC_LSU * a2dp_ldac_encoder_cb.feeding_params.channel_count *
516                        a2dp_ldac_encoder_cb.feeding_params.bits_per_sample / 8;
517 
518   a2dp_ldac_encoder_cb.stats.media_read_total_expected_reads_count++;
519   a2dp_ldac_encoder_cb.stats.media_read_total_expected_read_bytes += read_size;
520 
521   /* Read Data from UIPC channel */
522   uint32_t nb_byte_read = a2dp_ldac_encoder_cb.read_callback(read_buffer, read_size);
523   a2dp_ldac_encoder_cb.stats.media_read_total_actual_read_bytes += nb_byte_read;
524 
525   if (nb_byte_read < read_size) {
526     if (nb_byte_read == 0) {
527       return false;
528     }
529 
530     /* Fill the unfilled part of the read buffer with silence (0) */
531     memset(((uint8_t*)read_buffer) + nb_byte_read, 0, read_size - nb_byte_read);
532     nb_byte_read = read_size;
533   }
534   a2dp_ldac_encoder_cb.stats.media_read_total_actual_reads_count++;
535 
536   *bytes_read = nb_byte_read;
537   return true;
538 }
539 
adjust_effective_mtu(const tA2DP_ENCODER_INIT_PEER_PARAMS & peer_params)540 static uint16_t adjust_effective_mtu(const tA2DP_ENCODER_INIT_PEER_PARAMS& peer_params) {
541   uint16_t mtu_size = BT_DEFAULT_BUFFER_SIZE - A2DP_LDAC_OFFSET - sizeof(BT_HDR);
542   if (mtu_size > peer_params.peer_mtu) {
543     mtu_size = peer_params.peer_mtu;
544   }
545   log::verbose("original AVDTP MTU size: {}", mtu_size);
546   return mtu_size;
547 }
548 
quality_mode_index_to_name(int quality_mode_index)549 static std::string quality_mode_index_to_name(int quality_mode_index) {
550   switch (quality_mode_index) {
551     case A2DP_LDAC_QUALITY_HIGH:
552       return "HIGH";
553     case A2DP_LDAC_QUALITY_MID:
554       return "MID";
555     case A2DP_LDAC_QUALITY_LOW:
556       return "LOW";
557     case A2DP_LDAC_QUALITY_ABR:
558       return "ABR";
559     default:
560       return "Unknown";
561   }
562 }
563 
a2dp_vendor_ldac_set_transmit_queue_length(size_t transmit_queue_length)564 void a2dp_vendor_ldac_set_transmit_queue_length(size_t transmit_queue_length) {
565   a2dp_ldac_encoder_cb.TxQueueLength = transmit_queue_length;
566 }
567 
debug_codec_dump(int fd)568 void A2dpCodecConfigLdacSource::debug_codec_dump(int fd) {
569   a2dp_ldac_encoder_stats_t* stats = &a2dp_ldac_encoder_cb.stats;
570   tA2DP_LDAC_ENCODER_PARAMS* p_encoder_params = &a2dp_ldac_encoder_cb.ldac_encoder_params;
571 
572   A2dpCodecConfig::debug_codec_dump(fd);
573 
574   dprintf(fd, "  LDAC quality mode                                       : %s\n",
575           quality_mode_index_to_name(p_encoder_params->quality_mode_index).c_str());
576 
577   dprintf(fd, "  LDAC transmission bitrate (Kbps)                        : %d\n",
578           ldacBT_get_bitrate(a2dp_ldac_encoder_cb.ldac_handle));
579 
580   dprintf(fd, "  LDAC saved transmit queue length                        : %zu\n",
581           a2dp_ldac_encoder_cb.TxQueueLength);
582   if (a2dp_ldac_encoder_cb.has_ldac_abr_handle) {
583     dprintf(fd, "  LDAC adaptive bit rate encode quality mode index        : %d\n",
584             a2dp_ldac_encoder_cb.last_ldac_abr_eqmid);
585     dprintf(fd, "  LDAC adaptive bit rate adjustments                      : %zu\n",
586             a2dp_ldac_encoder_cb.ldac_abr_adjustments);
587   }
588   dprintf(fd, "  Encoder interval (ms): %" PRIu64 "\n", a2dp_vendor_ldac_get_encoder_interval_ms());
589   dprintf(fd, "  Effective MTU: %d\n", a2dp_vendor_ldac_get_effective_frame_size());
590   dprintf(fd,
591           "  Packet counts (expected/dropped)                        : %zu / "
592           "%zu\n",
593           stats->media_read_total_expected_packets, stats->media_read_total_dropped_packets);
594 
595   dprintf(fd,
596           "  PCM read counts (expected/actual)                       : %zu / "
597           "%zu\n",
598           stats->media_read_total_expected_reads_count, stats->media_read_total_actual_reads_count);
599 
600   dprintf(fd,
601           "  PCM read bytes (expected/actual)                        : %zu / "
602           "%zu\n",
603           stats->media_read_total_expected_read_bytes, stats->media_read_total_actual_read_bytes);
604 }
605