#include "util.h"

#include "sd-bus-protocol.h"

#include "bus-introspect.h"

#include "bus-signature.h"

#include "bus-internal.h"

int introspect_begin(struct introspect *i) {

assert(i);

zero(*i);

i->f = open_memstream(&i->introspection, &i->size);

if (!i->f)

return -ENOMEM;

fputs(SD_BUS_INTROSPECT_DOCTYPE

"<node>\n", i->f);

return 0;

}

int introspect_write_default_interfaces(struct introspect *i, bool object_manager) {

assert(i);

fputs(SD_BUS_INTROSPECT_INTERFACE_PEER

SD_BUS_INTROSPECT_INTERFACE_INTROSPECTABLE

SD_BUS_INTROSPECT_INTERFACE_PROPERTIES, i->f);

if (object_manager)

fputs(SD_BUS_INTROSPECT_INTERFACE_OBJECT_MANAGER, i->f);

return 0;

}

int introspect_write_child_nodes(struct introspect *i, Set *s, const char *prefix) {

char *node;

assert(i);

assert(prefix);

while ((node = set_steal_first(s))) {

const char *e;

e = object_path_startswith(node, prefix);

if (e && e[0])

fprintf(i->f, " <node name=\"%s\"/>\n", e);

free(node);

}

return 0;

}

static void introspect_write_flags(struct introspect *i, int type, int flags) {

if (flags & SD_BUS_VTABLE_DEPRECATED)

fputs(" <annotation name=\"org.freedesktop.DBus.Deprecated\" value=\"true\"/>\n", i->f);

if (type == _SD_BUS_VTABLE_METHOD && flags & SD_BUS_VTABLE_METHOD_NO_REPLY)

fputs(" <annotation name=\"org.freedesktop.DBus.Method.NoReply\" value=\"true\"/>\n", i->f);

if (type == _SD_BUS_VTABLE_PROPERTY || type == _SD_BUS_VTABLE_WRITABLE_PROPERTY) {

if (!(flags & SD_BUS_VTABLE_PROPERTY_EMITS_CHANGE))

fputs(" <annotation name=\"org.freedesktop.DBus.Property.EmitsChangedSignal\" value=\"false\"/>\n", i->f);

else if (flags & SD_BUS_VTABLE_PROPERTY_INVALIDATE_ONLY)

fputs(" <annotation name=\"org.freedesktop.DBus.Property.EmitsChangedSignal\" value=\"invalidates\"/>\n", i->f);

}

}

static int introspect_write_arguments(struct introspect *i, const char *signature, const char *direction) {

int r;

for (;;) {

size_t l;

if (!*signature)

return 0;

r = signature_element_length(signature, &l);

if (r < 0)

return r;

fprintf(i->f, " <arg type=\"%.*s\"", (int) l, signature);

if (direction)

fprintf(i->f, " direction=\"%s\"/>\n", direction);

else

fputs("/>\n", i->f);

signature += l;

}

}

int introspect_write_interface(struct introspect *i, const sd_bus_vtable *v) {

assert(i);

assert(v);

for (; v->type != _SD_BUS_VTABLE_END; v++) {

switch (v->type) {

case _SD_BUS_VTABLE_START:

if (v->flags & SD_BUS_VTABLE_DEPRECATED)

fputs(" <annotation name=\"org.freedesktop.DBus.Deprecated\" value=\"true\"/>\n", i->f);

break;

case _SD_BUS_VTABLE_METHOD:

fprintf(i->f, " <method name=\"%s\">\n", v->x.method.member);

introspect_write_arguments(i, strempty(v->x.method.signature), "in");

introspect_write_arguments(i, strempty(v->x.method.result), "out");

introspect_write_flags(i, v->type, v->flags);

fputs(" </method>\n", i->f);

break;

case _SD_BUS_VTABLE_PROPERTY:

case _SD_BUS_VTABLE_WRITABLE_PROPERTY:

fprintf(i->f, " <property name=\"%s\" type=\"%s\" access=\"%s\">\n",

v->x.property.member,

v->x.property.signature,

v->type == _SD_BUS_VTABLE_WRITABLE_PROPERTY ? "readwrite" : "read");

introspect_write_flags(i, v->type, v->flags);

fputs(" </property>\n", i->f);

break;

case _SD_BUS_VTABLE_SIGNAL:

fprintf(i->f, " <signal name=\"%s\">\n", v->x.signal.member);

introspect_write_arguments(i, strempty(v->x.signal.signature), NULL);

introspect_write_flags(i, v->type, v->flags);

fputs(" </signal>\n", i->f);

break;

}

}

return 0;

}

int introspect_finish(struct introspect *i, sd_bus *bus, sd_bus_message *m, sd_bus_message **reply) {

sd_bus_message *q;

int r;

assert(i);

assert(m);

assert(reply);

fputs("</node>\n", i->f);

fflush(i->f);

if (ferror(i->f))

return -ENOMEM;

r = sd_bus_message_new_method_return(bus, m, &q);

if (r < 0)

return r;

r = sd_bus_message_append(q, "s", i->introspection);

if (r < 0) {

sd_bus_message_unref(q);

return r;

}

*reply = q;

return 0;

}

void introspect_free(struct introspect *i) {

assert(i);

if (i->f)

fclose(i->f);

if (i->introspection)

free(i->introspection);

zero(*i);

}