winlator/android_alsa/module_pcm_android_aserver.c
2024-12-31 11:37:47 -03:00

467 lines
15 KiB
C

#include <stdio.h>
#include <stdbool.h>
#include <stdint.h>
#include <alsa/asoundlib.h>
#include <alsa/pcm_external.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/un.h>
#include <sys/mman.h>
#define MIN_REQUEST_LENGTH 5
#define BUFFER_OFFSET 4
#define REQUEST_CODE_CLOSE 0
#define REQUEST_CODE_START 1
#define REQUEST_CODE_STOP 2
#define REQUEST_CODE_PAUSE 3
#define REQUEST_CODE_PREPARE 4
#define REQUEST_CODE_WRITE 5
#define REQUEST_CODE_DRAIN 6
#define REQUEST_CODE_POINTER 7
#define REQUEST_CODE_MIN_BUFFER_SIZE 8
#define DATA_TYPE_U8 0
#define DATA_TYPE_S16LE 1
#define DATA_TYPE_S16BE 2
#define DATA_TYPE_FLOATLE 3
#define DATA_TYPE_FLOATBE 4
#define ARRAY_SIZE(arr) (sizeof(arr) / sizeof(arr[0]))
typedef struct snd_pcm_android_aserver {
snd_pcm_ioplug_t io;
int fd;
int frame_bytes;
int shm_size;
void* shm_ptr;
bool use_shm;
} snd_pcm_android_aserver_t;
static int android_aserver_recv_fd(int fd) {
char zero = 0;
struct iovec iovmsg = {.iov_base = &zero, .iov_len = 1};
struct {
struct cmsghdr align;
int fds[1];
} ctrlmsg;
struct msghdr msg = {
.msg_name = NULL,
.msg_namelen = 0,
.msg_iov = &iovmsg,
.msg_iovlen = 1,
.msg_flags = 0,
.msg_control = &ctrlmsg,
.msg_controllen = sizeof(struct cmsghdr) + sizeof(int)
};
struct cmsghdr *cmsg = CMSG_FIRSTHDR(&msg);
cmsg->cmsg_level = SOL_SOCKET;
cmsg->cmsg_type = SCM_RIGHTS;
cmsg->cmsg_len = msg.msg_controllen;
((int*)CMSG_DATA(cmsg))[0] = -1;
recvmsg(fd, &msg, 0);
return ((int*)CMSG_DATA(cmsg))[0];
}
static char parse_data_type(snd_pcm_format_t format) {
char data_type;
switch (format) {
case SND_PCM_FORMAT_U8:
data_type = DATA_TYPE_U8;
break;
case SND_PCM_FORMAT_S16_LE:
data_type = DATA_TYPE_S16LE;
break;
case SND_PCM_FORMAT_S16_BE:
data_type = DATA_TYPE_S16BE;
break;
case SND_PCM_FORMAT_FLOAT_LE:
data_type = DATA_TYPE_FLOATLE;
break;
case SND_PCM_FORMAT_FLOAT_BE:
data_type = DATA_TYPE_FLOATBE;
break;
default:
data_type = -EINVAL;
}
return data_type;
}
static int android_aserver_min_buffer_size(snd_pcm_ioplug_t* io, char channels, snd_pcm_format_t format, int rate) {
snd_pcm_android_aserver_t* android_aserver = io->private_data;
int request_length = 6;
char request_data[request_length + MIN_REQUEST_LENGTH];
request_data[0] = REQUEST_CODE_MIN_BUFFER_SIZE;
*(int*)(request_data + 1) = request_length;
request_data[5] = channels;
request_data[6] = parse_data_type(format);
*(int*)(request_data + 7) = rate;
int res = write(android_aserver->fd, &request_data, request_length + MIN_REQUEST_LENGTH);
if (res < 0) return 0;
int min_buffer_size;
res = read(android_aserver->fd, &min_buffer_size, 4);
if (res != 4) return 0;
return min_buffer_size;
}
static int android_aserver_close(snd_pcm_ioplug_t* io) {
snd_pcm_android_aserver_t* android_aserver = io->private_data;
if (!android_aserver) return 0;
if (android_aserver->fd >= 0) {
int request_length = 0;
char request_data[MIN_REQUEST_LENGTH];
request_data[0] = REQUEST_CODE_CLOSE;
*(int*)(request_data + 1) = request_length;
int res = write(android_aserver->fd, &request_data, MIN_REQUEST_LENGTH);
if (res > 0) close(android_aserver->fd);
}
if (android_aserver->shm_ptr) {
munmap(android_aserver->shm_ptr, android_aserver->shm_size);
android_aserver->shm_ptr = NULL;
android_aserver->shm_size = 0;
}
free(android_aserver);
return 0;
}
static int android_aserver_start(snd_pcm_ioplug_t* io) {
snd_pcm_android_aserver_t* android_aserver = io->private_data;
int request_length = 0;
char request_data[MIN_REQUEST_LENGTH];
request_data[0] = REQUEST_CODE_START;
*(int*)(request_data + 1) = request_length;
int res = write(android_aserver->fd, &request_data, MIN_REQUEST_LENGTH);
if (res < 0) return -EINVAL;
return 0;
}
static int android_aserver_stop(snd_pcm_ioplug_t* io) {
snd_pcm_android_aserver_t* android_aserver = io->private_data;
int request_length = 0;
char request_data[MIN_REQUEST_LENGTH];
request_data[0] = REQUEST_CODE_STOP;
*(int*)(request_data + 1) = request_length;
int res = write(android_aserver->fd, &request_data, MIN_REQUEST_LENGTH);
if (res < 0) return -EINVAL;
return 0;
}
static int android_aserver_pause(snd_pcm_ioplug_t* io, int enable) {
snd_pcm_android_aserver_t* android_aserver = io->private_data;
int request_length = 0;
char request_data[MIN_REQUEST_LENGTH];
request_data[0] = REQUEST_CODE_PAUSE;
*(int*)(request_data + 1) = request_length;
int res = write(android_aserver->fd, &request_data, MIN_REQUEST_LENGTH);
if (res < 0) return -EINVAL;
return 0;
}
static int android_aserver_prepare(snd_pcm_ioplug_t* io) {
snd_pcm_android_aserver_t* android_aserver = io->private_data;
android_aserver->frame_bytes = (snd_pcm_format_physical_width(io->format) * io->channels) / 8;
int request_length = 10;
char request_data[request_length + MIN_REQUEST_LENGTH];
request_data[0] = REQUEST_CODE_PREPARE;
*(int*)(request_data + 1) = request_length;
request_data[5] = (char)io->channels;
request_data[6] = parse_data_type(io->format);
*(int*)(request_data + 7) = io->rate;
*(int*)(request_data + 11) = io->buffer_size;
int res = write(android_aserver->fd, &request_data, request_length + MIN_REQUEST_LENGTH);
if (res < 0) return -EINVAL;
if (android_aserver->use_shm) {
if (android_aserver->shm_ptr) {
munmap(android_aserver->shm_ptr, android_aserver->shm_size);
android_aserver->shm_ptr = NULL;
android_aserver->shm_size = 0;
}
int fd = android_aserver_recv_fd(android_aserver->fd);
if (fd >= 0) {
int shm_size = io->buffer_size * android_aserver->frame_bytes + BUFFER_OFFSET;
void* shm_ptr = mmap(NULL, shm_size, PROT_WRITE | PROT_READ, MAP_SHARED, fd, 0);
if (shm_ptr != MAP_FAILED) {
memset(shm_ptr, 0, shm_size);
android_aserver->shm_ptr = shm_ptr;
android_aserver->shm_size = shm_size;
}
else android_aserver->use_shm = false;
close(fd);
}
}
return 0;
}
static snd_pcm_sframes_t android_aserver_pointer(snd_pcm_ioplug_t* io) {
snd_pcm_android_aserver_t* android_aserver = io->private_data;
uint32_t position;
if (android_aserver->use_shm) {
position = *(uint32_t*)(android_aserver->shm_ptr);
}
else {
int request_length = 0;
char request_data[MIN_REQUEST_LENGTH];
request_data[0] = REQUEST_CODE_POINTER;
*(int*)(request_data + 1) = request_length;
int res = write(android_aserver->fd, &request_data, MIN_REQUEST_LENGTH);
if (res < 0) return 0;
res = read(android_aserver->fd, &position, 4);
if (res != 4) return 0;
}
return position;
}
static snd_pcm_sframes_t android_aserver_transfer(snd_pcm_ioplug_t* io, const snd_pcm_channel_area_t* areas, snd_pcm_uframes_t offset, snd_pcm_uframes_t size) {
snd_pcm_android_aserver_t* android_aserver = io->private_data;
char* data = (char*)areas->addr + (areas->first + areas->step * offset) / 8;
int request_length = size * android_aserver->frame_bytes;
char request_data[MIN_REQUEST_LENGTH];
request_data[0] = REQUEST_CODE_WRITE;
*(int*)(request_data + 1) = request_length;
if (android_aserver->use_shm) memcpy(android_aserver->shm_ptr + BUFFER_OFFSET, data, request_length);
int res = write(android_aserver->fd, &request_data, MIN_REQUEST_LENGTH);
if (res < 0) return 0;
if (android_aserver->use_shm) {
char success = 0;
res = read(android_aserver->fd, &success, 1);
if (res != 1 || !success) return 0;
}
else {
res = write(android_aserver->fd, data, request_length);
if (res < 0) return 0;
}
return size;
}
static int android_aserver_drain(snd_pcm_ioplug_t* io) {
snd_pcm_android_aserver_t* android_aserver = io->private_data;
int request_length = 0;
char request_data[MIN_REQUEST_LENGTH];
request_data[0] = REQUEST_CODE_DRAIN;
*(int*)(request_data + 1) = request_length;
int res = write(android_aserver->fd, &request_data, MIN_REQUEST_LENGTH);
if (res < 0) return -EINVAL;
return 0;
}
static int android_aserver_hw_params(snd_pcm_ioplug_t* io, snd_pcm_hw_params_t* params) {
int err;
snd_pcm_format_t format;
err = snd_pcm_hw_params_get_format(params, &format);
if (err < 0) return err;
unsigned int channels;
err = snd_pcm_hw_params_get_channels(params, &channels);
if (err < 0) return err;
unsigned int rate;
err = snd_pcm_hw_params_get_rate(params, &rate, 0);
if (err < 0) return err;
int min_buffer_size = android_aserver_min_buffer_size(io, channels, format, rate);
if (min_buffer_size == 0) return 0;
int frame_bytes = (snd_pcm_format_physical_width(format) * channels) / 8;
snd_pcm_uframes_t buffer_size = min_buffer_size / frame_bytes;
snd_pcm_uframes_t period_size = buffer_size / frame_bytes;
snd_pcm_hw_params_t* refined_params;
snd_pcm_hw_params_alloca(&refined_params);
err = snd_pcm_hw_params_any(io->pcm, refined_params);
if (err < 0) return err;
snd_pcm_access_mask_t* access = alloca(snd_pcm_access_mask_sizeof());
err = snd_pcm_hw_params_get_access_mask(params, access);
if (err < 0) return err;
err = snd_pcm_hw_params_set_access_mask(io->pcm, refined_params, access);
if (err < 0) return err;
err = snd_pcm_hw_params_set_format(io->pcm, refined_params, format);
if (err < 0) return err;
err = snd_pcm_hw_params_set_channels(io->pcm, refined_params, channels);
if (err < 0) return err;
err = snd_pcm_hw_params_set_rate(io->pcm, refined_params, rate, 0);
if (err < 0) return err;
err = snd_pcm_hw_params_set_period_size(io->pcm, refined_params, period_size, 0);
if (err < 0) return err;
err = snd_pcm_hw_params_set_periods_integer(io->pcm, refined_params);
if (err < 0) return err;
err = snd_pcm_hw_params_set_buffer_size(io->pcm, refined_params, buffer_size);
if (err < 0) return err;
snd_pcm_hw_params_copy(params, refined_params);
return 0;
}
static int android_aserver_set_hw_constraint(snd_pcm_ioplug_t* io) {
static const unsigned int access_list[] = {SND_PCM_ACCESS_RW_INTERLEAVED};
static const unsigned int format_list[] = {SND_PCM_FORMAT_U8, SND_PCM_FORMAT_S16_LE, SND_PCM_FORMAT_S16_BE, SND_PCM_FORMAT_FLOAT_LE, SND_PCM_FORMAT_FLOAT_BE};
int err;
err = snd_pcm_ioplug_set_param_list(io, SND_PCM_IOPLUG_HW_ACCESS, ARRAY_SIZE(access_list), access_list);
if (err < 0) return err;
err = snd_pcm_ioplug_set_param_list(io, SND_PCM_IOPLUG_HW_FORMAT, ARRAY_SIZE(format_list), format_list);
if (err < 0) return err;
err = snd_pcm_ioplug_set_param_minmax(io, SND_PCM_IOPLUG_HW_CHANNELS, 1, 2);
if (err < 0) return err;
err = snd_pcm_ioplug_set_param_minmax(io, SND_PCM_IOPLUG_HW_RATE, 8000, 48000);
if (err < 0) return err;
err = snd_pcm_ioplug_set_param_minmax(io, SND_PCM_IOPLUG_HW_PERIOD_BYTES, 64, 64 * 1024);
if (err < 0) return err;
err = snd_pcm_ioplug_set_param_minmax(io, SND_PCM_IOPLUG_HW_PERIODS, 2, 64);
if (err < 0) return err;
return 0;
}
static snd_pcm_ioplug_callback_t android_aserver_callback = {
.close = android_aserver_close,
.start = android_aserver_start,
.stop = android_aserver_stop,
.pause = android_aserver_pause,
.resume = android_aserver_start,
.prepare = android_aserver_prepare,
.transfer = android_aserver_transfer,
.drain = android_aserver_drain,
.pointer = android_aserver_pointer,
.hw_params = android_aserver_hw_params,
};
static int android_aserver_connect() {
char* path = getenv("ANDROID_ALSA_SERVER");
int fd = socket(AF_UNIX, SOCK_STREAM, 0);
if (fd < 0) return -1;
struct sockaddr_un server_addr;
memset(&server_addr, 0, sizeof(server_addr));
server_addr.sun_family = AF_LOCAL;
strncpy(server_addr.sun_path, path, sizeof(server_addr.sun_path) - 1);
int res;
do {
res = 0;
if (connect(fd, (struct sockaddr*)&server_addr, sizeof(struct sockaddr_un)) < 0) res = -errno;
}
while (res == -EINTR);
if (res < 0) {
close(fd);
return -1;
}
return fd;
}
static int android_aserver_create(snd_pcm_t** pcmp, const char* name, snd_pcm_stream_t stream, int mode) {
snd_pcm_android_aserver_t* android_aserver;
if (stream != SND_PCM_STREAM_PLAYBACK) return -ENOTSUP;
android_aserver = calloc(1, sizeof(snd_pcm_android_aserver_t));
if (!android_aserver) return -ENOMEM;
android_aserver->io.version = SND_PCM_IOPLUG_VERSION;
android_aserver->io.name = "ALSA <-> Android AServer PCM Plugin";
android_aserver->io.callback = &android_aserver_callback;
android_aserver->io.mmap_rw = 0;
android_aserver->io.private_data = android_aserver;
int res = -EINVAL;
android_aserver->fd = android_aserver_connect();
if (android_aserver->fd < 0) goto error;
char* use_shm_value = getenv("ANDROID_ASERVER_USE_SHM");
android_aserver->use_shm = use_shm_value && (strcmp(use_shm_value, "true") == 0 || strcmp(use_shm_value, "1") == 0);
res = snd_pcm_ioplug_create(&android_aserver->io, name, stream, mode);
if (res < 0) goto error;
res = android_aserver_set_hw_constraint(&android_aserver->io);
if (res < 0) {
snd_pcm_ioplug_delete(&android_aserver->io);
goto error;
}
*pcmp = android_aserver->io.pcm;
return 0;
error:
if (android_aserver->fd >= 0) close(android_aserver->fd);
free(android_aserver);
return res;
}
SND_PCM_PLUGIN_DEFINE_FUNC(android_aserver) {
snd_config_iterator_t i, next;
snd_config_for_each(i, next, conf) {
snd_config_t* n = snd_config_iterator_entry(i);
const char* id;
if (snd_config_get_id(n, &id) < 0) continue;
if (strcmp(id, "type") == 0 || strcmp(id, "hint") == 0) continue;
return -EINVAL;
}
int err = android_aserver_create(pcmp, name, stream, mode);
return err;
}
SND_PCM_PLUGIN_SYMBOL(android_aserver);