pipewire/spa/tools/spa-monitor.c

/* Simple Plugin API
 * Copyright (C) 2016 Wim Taymans <wim.taymans@gmail.com>
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Library General Public
 * License as published by the Free Software Foundation; either
 * version 2 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Library General Public License for more details.
 *
 * You should have received a copy of the GNU Library General Public
 * License along with this library; if not, write to the
 * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
 * Boston, MA 02110-1301, USA.
 */

#include <string.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <dlfcn.h>
#include <errno.h>
#include <poll.h>

#include <spa/log.h>
#include <spa/type-map.h>
#include <spa/monitor.h>
#include <spa/loop.h>
#include <lib/debug.h>
#include <lib/mapper.h>

typedef struct {
  SpaTypeMonitor monitor;
} Type;

typedef struct {
  Type type;

  SpaTypeMap *map;
  SpaLog *log;
  SpaLoop main_loop;

  SpaSupport support[3];
  uint32_t   n_support;

  unsigned int n_sources;
  SpaSource sources[16];

  bool rebuild_fds;
  struct pollfd fds[16];
  unsigned int n_fds;
} AppData;


static void
inspect_item (AppData *data, SpaMonitorItem *item)
{
  spa_debug_pod (&item->pod, data->map);
}

static void
on_monitor_event  (SpaMonitor      *monitor,
                   SpaEvent        *event,
                   void            *user_data)
{
  AppData *data = user_data;

  if (SPA_EVENT_TYPE (event) == data->type.monitor.Added) {
    fprintf (stderr, "added:\n");
    inspect_item (data, (SpaMonitorItem*)event);
  }
  else if (SPA_EVENT_TYPE (event) == data->type.monitor.Removed) {
    fprintf (stderr, "removed:\n");
    inspect_item (data, (SpaMonitorItem*)event);
  }
  else if (SPA_EVENT_TYPE (event) == data->type.monitor.Changed) {
    fprintf (stderr, "changed:\n");
    inspect_item (data, (SpaMonitorItem*)event);
  }
}

static SpaResult
do_add_source (SpaLoop   *loop,
               SpaSource *source)
{
  AppData *data = SPA_CONTAINER_OF (loop, AppData, main_loop);

  data->sources[data->n_sources] = *source;
  data->n_sources++;
  data->rebuild_fds = true;

  return SPA_RESULT_OK;
}

static SpaResult
do_update_source (SpaSource  *source)
{
  return SPA_RESULT_OK;
}

static void
do_remove_source (SpaSource  *source)
{
}

static void
handle_monitor (AppData *data, SpaMonitor *monitor)
{
  SpaResult res;
  uint32_t index;

  if (monitor->info)
    spa_debug_dict (monitor->info);

  for (index = 0; ; index++) {
    SpaMonitorItem *item;

    if ((res = spa_monitor_enum_items (monitor, &item, index)) < 0) {
      if (res != SPA_RESULT_ENUM_END)
        printf ("spa_monitor_enum_items: got error %d\n", res);
      break;
    }
    inspect_item (data, item);
  }

  spa_monitor_set_event_callback (monitor, on_monitor_event, &data);

  while (true) {
    int i, r;

    /* rebuild */
    if (data->rebuild_fds) {
      for (i = 0; i < data->n_sources; i++) {
        SpaSource *p = &data->sources[i];
        data->fds[i].fd = p->fd;
        data->fds[i].events = p->mask;
      }
      data->n_fds = data->n_sources;
      data->rebuild_fds = false;
    }

    r = poll ((struct pollfd *) data->fds, data->n_fds, -1);
    if (r < 0) {
      if (errno == EINTR)
        continue;
      break;
    }
    if (r == 0) {
      fprintf (stderr, "monitor %p: select timeout", monitor);
      break;
    }

    /* after */
    for (i = 0; i < data->n_sources; i++) {
      SpaSource *p = &data->sources[i];
      p->func (p);
    }
  }
}

int
main (int argc, char *argv[])
{
  AppData data = { 0 };
  SpaResult res;
  void *handle;
  SpaEnumHandleFactoryFunc enum_func;
  uint32_t fidx;

  data.map = spa_type_map_get_default ();
  data.log = NULL;
  data.main_loop.size = sizeof (SpaLoop);
  data.main_loop.add_source = do_add_source;
  data.main_loop.update_source = do_update_source;
  data.main_loop.remove_source = do_remove_source;

  data.support[0].type = SPA_TYPE__TypeMap;
  data.support[0].data = data.map;
  data.support[1].type = SPA_TYPE__Log;
  data.support[1].data = data.log;
  data.support[2].type = SPA_TYPE_LOOP__MainLoop;
  data.support[2].data = &data.main_loop;
  data.n_support = 3;

  spa_type_monitor_map (data.map, &data.type.monitor);

  if (argc < 2) {
    printf ("usage: %s <plugin.so>\n", argv[0]);
    return -1;
  }

  if ((handle = dlopen (argv[1], RTLD_NOW)) == NULL) {
    printf ("can't load %s\n", argv[1]);
    return -1;
  }
  if ((enum_func = dlsym (handle, "spa_enum_handle_factory")) == NULL) {
    printf ("can't find function\n");
    return -1;
  }

  for (fidx = 0;; fidx++) {
    const SpaHandleFactory *factory;
    uint32_t iidx;

    if ((res = enum_func (&factory, fidx)) < 0) {
      if (res != SPA_RESULT_ENUM_END)
        printf ("can't enumerate factories: %d\n", res);
      break;
    }

    for (iidx = 0;; iidx++) {
      const SpaInterfaceInfo *info;

      if ((res = spa_handle_factory_enum_interface_info (factory, &info, iidx)) < 0) {
        if (res != SPA_RESULT_ENUM_END)
          printf ("can't enumerate interfaces: %d\n", res);
        break;
      }

      if (!strcmp (info->type, SPA_TYPE__Monitor)) {
        SpaHandle *handle;
        void *interface;

        handle = calloc (1, factory->size);
        if ((res = spa_handle_factory_init (factory, handle, NULL, data.support, data.n_support)) < 0) {
          printf ("can't make factory instance: %d\n", res);
          continue;
        }

        if ((res = spa_handle_get_interface (handle, data.type.monitor.Monitor, &interface)) < 0) {
          printf ("can't get interface: %d\n", res);
          continue;
        }
        handle_monitor (&data, interface);
      }
    }
  }

  return 0;
}