]> git.draconx.ca Git - upkg.git/blobdiff - src/uobject/uobject.c
uobject: Include package names in package fullname.
[upkg.git] / src / uobject / uobject.c
index 91740054514799f0f2c477f4622e99814eea4169..b8c24796dafd601c7669482c0d78338f6ffd317f 100644 (file)
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
+#include <stdarg.h>
 #include <stdbool.h>
+#include <inttypes.h>
 #include <assert.h>
 #include <glib-object.h>
 
 #include <uobject/uobject.h>
 #include <uobject/module.h>
+#include <uobject/package.h>
 #include "upkg.h"
 #include "pack.h"
 
@@ -172,7 +175,7 @@ decode_prop_header(struct upkg *upkg, struct prop_head *head,
        }
 
        rc = decode_tag_size(&head->size, tag_size, buf+len, n-len);
-       if (head->size == 0)
+       if (rc == 0 && head->size == 0)
                return 0;
        len += rc;
 
@@ -215,7 +218,7 @@ static unsigned long deserialize_property(UObject *uo, struct prop_head *head)
        unsigned long rc, len = 0;
        GValue val = {0};
 
-       rc = decode_prop_header(uo->pkg, head, priv->buf, priv->nbuf);
+       rc = decode_prop_header(uo->pkg_file->pkg, head, priv->buf, priv->nbuf);
        if (rc == 0)
                return 0;
        len += rc;
@@ -237,6 +240,11 @@ static unsigned long deserialize_property(UObject *uo, struct prop_head *head)
                g_value_set_ulong(&val, unpack_32_le(priv->buf+len));
                g_object_set_property(G_OBJECT(uo), head->prop_name, &val);
                break;
+       case PROPERTY_BOOLEAN:
+               g_value_init(&val, G_TYPE_BOOLEAN);
+               g_value_set_boolean(&val, head->tag_msb);
+               g_object_set_property(G_OBJECT(uo), head->prop_name, &val);
+               break;
        case PROPERTY_OBJECT:
                rc = decode_object_property(uo, &val, len);
                if (rc != 0)
@@ -304,22 +312,73 @@ static int deserialize(UObject *uo)
        return 0;
 }
 
-int u_object_deserialize(GObject *obj, struct upkg *pkg, unsigned long idx)
+/*
+ * Get the full hierarchical object name for an export, used for diagnostics.
+ * The package name is passed in pname.
+ *
+ * Returns a buffer allocated by malloc on success, or NULL on failure.
+ */
+static char *get_obj_fullname(const char *pname, const struct upkg_export *e)
+{
+       size_t total_len = strlen(pname) + 1, len;
+       char *fullname;
+
+       assert(e != NULL);
+
+       for (const struct upkg_export *c = e; c; c = c->parent) {
+               len = strlen(c->name) + 1;
+               if (total_len > SIZE_MAX - len)
+                       return NULL;
+               total_len += len;
+       }
+
+       fullname = malloc(total_len);
+       if (!fullname)
+               return NULL;
+
+       for (const struct upkg_export *c = e; c; c = c->parent) {
+               len = strlen(c->name);
+               assert(total_len > len);
+
+               total_len -= len + 1;
+               memcpy(fullname + total_len, c->name, len);
+               fullname[total_len + len] = c == e ? '\0' : '.';
+       }
+
+       assert(total_len == strlen(pname)+1);
+       memcpy(fullname, pname, total_len-1);
+       fullname[total_len-1] = '.';
+
+       return fullname;
+}
+
+int u_object_deserialize(GObject *obj, GTypeModule *pkg, unsigned long idx)
 {
        g_return_val_if_fail(IS_U_OBJECT(obj), -1);
+       g_return_val_if_fail(IS_U_PKG(pkg), -1);
+
        UObject *uo = U_OBJECT(obj);
+       struct upkg *upkg = U_PKG(pkg)->pkg;
+       const struct upkg_export *e;
        struct upkg_file *f;
        int rc;
 
        g_return_val_if_fail(uo->pkg_file == NULL, -1);
-       f = upkg_export_open(pkg, idx);
+       f = upkg_export_open(upkg, idx);
        if (!f) {
                return -1;
        }
 
-       uo->pkg      = pkg;
+       uo->pkg = pkg;
        uo->pkg_file = f;
 
+       e = upkg_get_export(upkg, idx);
+       if (!e)
+               return -1;
+       uo->pkg_name = get_obj_fullname(pkg->name, e);
+       if (!uo->pkg_name)
+               return -1;
+
        rc = U_OBJECT_GET_CLASS(obj)->deserialize(uo);
        if (rc != 0) {
                upkg_export_close(f);
@@ -329,19 +388,21 @@ int u_object_deserialize(GObject *obj, struct upkg *pkg, unsigned long idx)
        return rc;
 }
 
-GObject *u_object_new_from_package(struct upkg *upkg, unsigned long idx)
+GObject *u_object_new_from_package(GTypeModule *pkg, unsigned long idx)
 {
+       g_return_val_if_fail(IS_U_PKG(pkg), NULL);
+
        const struct upkg_export *export;
        const char *class, *package;
        GObject *obj = NULL;
        GType type;
 
-       class = upkg_export_class(upkg, idx, &package);
+       class = upkg_export_class(U_PKG(pkg)->pkg, idx, &package);
 
        type = u_object_module_get_class(package, class);
        if (type) {
                obj = g_object_new(type, NULL);
-               if (u_object_deserialize(obj, upkg, idx) != 0) {
+               if (u_object_deserialize(obj, pkg, idx) != 0) {
                        g_object_unref(obj);
                        return NULL;
                }
@@ -364,6 +425,8 @@ static void u_object_finalize(GObject *o)
                upkg_export_close(uo->pkg_file);
        }
 
+       free(uo->pkg_name);
+
        G_OBJECT_CLASS(u_object_parent_class)->finalize(o);
 }
 
@@ -375,3 +438,70 @@ static void u_object_class_init(UObjectClass *class)
        class->deserialize = deserialize;
        go->finalize       = u_object_finalize;
 }
+
+/*
+ * Prepend a prefix to a printf-style format string.  Unfortunately, there's
+ * no way to construct va_list values on the fly in C, so we cannot simply
+ * prepend %s and pass the prefix as an argument.  Any % characters in the
+ * prefix will be escaped.
+ *
+ * Returns a pointer to a newly allocated string, which should be freed by the
+ * caller, or NULL otherwise.
+ */
+static char *prepend_fmt(const char *prefix, const char *fmt)
+{
+       size_t prefix_len = strlen(prefix), fmt_len = strlen(fmt);
+       char *new_fmt;
+
+       if (prefix_len > SIZE_MAX/2 - sizeof ": ")
+               return NULL;
+       if (2*prefix_len > SIZE_MAX - fmt_len)
+               return NULL;
+
+       new_fmt = malloc(2*prefix_len + fmt_len + 1);
+       if (new_fmt) {
+               size_t offset = 0;
+
+               for (size_t i = 0; i < prefix_len; i++) {
+                       if (prefix[i] == '%')
+                               new_fmt[offset++] = '%';
+                       new_fmt[offset++] = prefix[i];
+               }
+
+               new_fmt[offset++] = ':';
+               new_fmt[offset++] = ' ';
+               strcpy(new_fmt+offset, fmt);
+       }
+
+       return new_fmt;
+}
+
+/* Logging helpers that automatically prepend the UObject class information. */
+void u_vlog_full(GObject *o, GLogLevelFlags level, const char *fmt, va_list ap)
+{
+       g_return_if_fail(IS_U_OBJECT(o));
+       UObject *uo = U_OBJECT(o);
+       char *new_fmt = NULL;
+
+       if (uo->pkg_name) {
+               new_fmt = prepend_fmt(uo->pkg_name, fmt);
+               if (!new_fmt) {
+                       g_log(G_OBJECT_TYPE_NAME(o), level, "%s",
+                             uo->pkg_file->name);
+               } else {
+                       fmt = new_fmt;
+               }
+       }
+
+       g_logv(G_OBJECT_TYPE_NAME(o), level, fmt, ap);
+       free(new_fmt);
+}
+
+void u_log_full(GObject *o, GLogLevelFlags level, const char *fmt, ...)
+{
+       va_list ap;
+
+       va_start(ap, fmt);
+       u_vlog_full(o, level, fmt, ap);
+       va_end(ap);
+}