From cf758158efa2832b38a49770f3489f408cb93b41 Mon Sep 17 00:00:00 2001

From: Matthew Almond <malmond@meta.com>

Date: Fri, 6 Dec 2024 08:00:31 +0100

Subject: [PATCH] Measure plugin

---

 macros.in              |   1 +

 plugins/CMakeLists.txt |   1 +

 plugins/measure.c      | 232 +++++++++++++++++++++++++++++++++++++++++

 3 files changed, 234 insertions(+)

 create mode 100644 plugins/measure.c

diff --git a/macros.in b/macros.in

index df7defe..1af187a 100644

--- a/macros.in

+++ b/macros.in

@@ -1189,6 +1189,7 @@ Supplements:   (%{name} = %{version}-%{release} and langpacks-%{1})\

 # Transaction plugin macros

 %__plugindir		%{_libdir}/rpm-plugins

 %__transaction_reflink		%{__plugindir}/reflink.so

+%__transaction_measure		%{__plugindir}/measure.so

 %__transaction_systemd_inhibit	%{__plugindir}/systemd_inhibit.so

 %__transaction_selinux		%{__plugindir}/selinux.so

 %__transaction_syslog		%{__plugindir}/syslog.so

diff --git a/plugins/CMakeLists.txt b/plugins/CMakeLists.txt

index ed63a43..cda8f96 100644

--- a/plugins/CMakeLists.txt

+++ b/plugins/CMakeLists.txt

@@ -1,6 +1,7 @@

 add_library(prioreset MODULE prioreset.c)

 add_library(syslog MODULE syslog.c)

 add_library(reflink MODULE reflink.c)

+add_library(measure MODULE measure.c)

 if(WITH_SELINUX)

 	add_library(selinux MODULE selinux.c)

diff --git a/plugins/measure.c b/plugins/measure.c

new file mode 100644

index 0000000..593e463

--- /dev/null

+++ b/plugins/measure.c

@@ -0,0 +1,232 @@

+#include "system.h"

+#include "time.h"

+

+#include <rpm/rpmlog.h>

+#include <rpm/rpmmacro.h>

+#include <rpm/rpmts.h>

+#include <rpm/rpmlib.h>

+#include <rpm/rpmstring.h>

+

+#include "rpmplugin.h"

+

+struct measurestat {

+    /* We're counting psm not packages because packages often run psm_pre/post

+       more than once and we want to accumulate the time

+    */

+    unsigned int psm_count;

+    unsigned int scriptlet_count;

+    struct timespec plugin_start;

+    struct timespec psm_start;

+    struct timespec scriptlet_start;

+};

+

+static rpmRC push(const char *format, const char *value, const char *formatted)

+{

+    char *key = NULL;

+    rpmRC rc = RPMRC_FAIL;

+    if (formatted == NULL) {

+        /* yes we're okay with discarding const here */

+        key = (char *)format;

+    } else {

+        if (rasprintf(&key, format, formatted) == -1) {

+            rpmlog(

+                RPMLOG_ERR,

+                _("measure: Failed to allocate formatted key %s, %s\n"),

+                format,

+                formatted

+            );

+            goto exit;

+        }

+    }

+    if (rpmPushMacro(NULL, key, NULL, value, RMIL_GLOBAL)) {

+        rpmlog(RPMLOG_ERR, _("measure: Failed to set %s\n"), key);

+        goto exit;

+    }

+    rc = RPMRC_OK;

+exit:

+    if (formatted != NULL) {

+        free(key);

+    }

+    return rc;

+}

+

+static rpmRC diff_ms(char **ms, struct timespec *start, struct timespec *end)

+{

+    if (rasprintf(ms, "%ld", (

+        (end->tv_sec - start->tv_sec) * 1000 +

+        (end->tv_nsec - start->tv_nsec) / 1000000

+    )) == -1) {

+        rpmlog(RPMLOG_ERR, _("measure: Failed to allocate formatted ms\n"));

+        return RPMRC_FAIL;

+    }

+    return RPMRC_OK;

+}

+

+static rpmRC measure_init(rpmPlugin plugin, rpmts ts)

+{

+    rpmRC rc = RPMRC_FAIL;

+    struct measurestat *state = rcalloc(1, sizeof(*state));

+    if (clock_gettime(CLOCK_MONOTONIC, &state->plugin_start)) {

+        rpmlog(RPMLOG_ERR, _("measure: Failed to get plugin_start time\n"));

+        goto exit;

+    }

+    state->psm_count = 0;

+    state->scriptlet_count = 0;

+    rpmPluginSetData(plugin, state);

+    rc = RPMRC_OK;

+exit:

+    return rc;

+}

+

+static void measure_cleanup(rpmPlugin plugin)

+{

+    struct measurestat *state = rpmPluginGetData(plugin);

+    free(state);

+}

+

+static rpmRC measure_tsm_post(rpmPlugin plugin, rpmts ts, int res)

+{

+    struct measurestat *state = rpmPluginGetData(plugin);

+    char *psm_count = NULL, *scriptlet_count = NULL;

+    rpmRC rc = RPMRC_FAIL;

+

+    if (rasprintf(&psm_count, "%d", state->psm_count) == -1) {

+        rpmlog(RPMLOG_ERR, _("measure: Failed to allocate formatted psm_count\n"));

+        goto exit;

+    }

+    if (rasprintf(&scriptlet_count, "%d", state->scriptlet_count) == -1) {

+        rpmlog(RPMLOG_ERR, _("measure: Failed to allocate formatted scriptlet_count\n"));

+        goto exit;

+    }

+    if (push("_measure_plugin_psm_count", psm_count, NULL) != RPMRC_OK) {

+        goto exit;

+    }

+    if (push("_measure_plugin_scriptlet_count", scriptlet_count, NULL) != RPMRC_OK) {

+        goto exit;

+    }

+    rc = RPMRC_OK;

+exit:

+    free(psm_count);

+    free(scriptlet_count);

+    return rc;

+}

+

+static rpmRC measure_psm_pre(rpmPlugin plugin, rpmte te)

+{

+    struct measurestat *state = rpmPluginGetData(plugin);

+    rpmRC rc = RPMRC_FAIL;

+

+    if (clock_gettime(CLOCK_MONOTONIC, &state->psm_start)) {

+        rpmlog(RPMLOG_ERR, _("measure: Failed to get psm_start time\n"));

+        goto exit;

+    }

+    rc = RPMRC_OK;

+exit:

+    return rc;

+}

+

+static rpmRC measure_psm_post(rpmPlugin plugin, rpmte te, int res)

+{

+    struct measurestat *state = rpmPluginGetData(plugin);

+    struct timespec end;

+    char *offset = NULL, *duration = NULL, *prefix = NULL;

+    Header h = rpmteHeader(te);

+    rpmRC rc = RPMRC_FAIL;

+

+    if (clock_gettime(CLOCK_MONOTONIC, &end)) {

+        rpmlog(RPMLOG_ERR, _("measure: Failed to get psm end time\n"));

+        goto exit;

+    }

+    if (rasprintf(&prefix, "_measure_plugin_package_%u", state->psm_count) == -1) {

+        rpmlog(RPMLOG_ERR, _("measure: Failed to allocate prefix\n"));

+        goto exit;

+    }

+    if (diff_ms(&offset, &state->plugin_start, &state->psm_start) != RPMRC_OK) {

+        goto exit;

+    }

+    if (diff_ms(&duration, &state->psm_start, &end) != RPMRC_OK) {

+        goto exit;

+    }

+    if (push("%s_nevra", rpmteNEVRA(te), prefix) != RPMRC_OK) {

+        goto exit;

+    }

+    if (push("%s_compressor", headerGetString(h, RPMTAG_PAYLOADCOMPRESSOR), prefix) != RPMRC_OK) {

+        goto exit;

+    }

+    if (push("%s_offset", offset, prefix) != RPMRC_OK) {

+        goto exit;

+    }

+    if (push("%s_ms", duration, prefix) != RPMRC_OK) {

+        goto exit;

+    }

+    state->psm_count += 1;

+    rc = RPMRC_OK;

+exit:

+    headerFree(h);

+    free(prefix);

+    free(duration);

+    free(offset);

+    return rc;

+}

+

+static rpmRC measure_scriptlet_pre(rpmPlugin plugin,

+					const char *s_name, int type)

+{

+    struct measurestat *state = rpmPluginGetData(plugin);

+    if (clock_gettime(CLOCK_MONOTONIC, &state->scriptlet_start)) {

+        rpmlog(RPMLOG_ERR, _("measure: Failed to get scriptlet_start time\n"));

+        return RPMRC_FAIL;

+    }

+    return RPMRC_OK;

+}

+

+static rpmRC measure_scriptlet_post(rpmPlugin plugin,

+					const char *s_name, int type, int res)

+{

+    struct measurestat *state = rpmPluginGetData(plugin);

+    struct timespec end;

+    char *offset = NULL, *duration = NULL, *prefix = NULL;

+    rpmRC rc = RPMRC_FAIL;

+

+    if (clock_gettime(CLOCK_MONOTONIC, &end)) {

+        rpmlog(RPMLOG_ERR, _("measure: Failed to get end time\n"));

+        goto exit;

+    }

+

+    if (rasprintf(&prefix, "_measure_plugin_scriptlet_%d", state->scriptlet_count) == -1) {

+        rpmlog(RPMLOG_ERR, _("measure: Failed to allocate formatted prefix\n"));

+        goto exit;

+    }

+    if (diff_ms(&offset, &state->plugin_start, &state->scriptlet_start) != RPMRC_OK) {

+        goto exit;

+    }

+    if (diff_ms(&duration, &state->scriptlet_start, &end) != RPMRC_OK) {

+        goto exit;

+    }

+    if (push("%s_name", s_name, prefix) != RPMRC_OK) {

+        goto exit;

+    }

+    if (push("%s_offset", offset, prefix) != RPMRC_OK) {

+        goto exit;

+    }

+    if (push("%s_ms", duration, prefix) != RPMRC_OK) {

+        goto exit;

+    }

+    state->scriptlet_count += 1;

+    rc = RPMRC_OK;

+exit:

+    free(prefix);

+    free(duration);

+    free(offset);

+    return rc;

+}

+

+struct rpmPluginHooks_s measure_hooks = {

+    .init = measure_init,

+    .cleanup = measure_cleanup,

+    .tsm_post = measure_tsm_post,

+    .psm_pre = measure_psm_pre,

+    .psm_post = measure_psm_post,

+    .scriptlet_pre = measure_scriptlet_pre,

+    .scriptlet_post = measure_scriptlet_post,

+};

-- 

2.47.1