1 files changed, 103 insertions, 0 deletions
diff --git a/claim/claim.c b/claim/claim.c
new file mode 100644
index 0000000000..75f0a437d6
--- /dev/null
+++ b/claim/claim.c
@@ -0,0 +1,103 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+#include "claim.h"
+#include "../registry/registry_internals.h"
+
+char *claiming_pending_arguments = NULL;
+
+static char *claiming_errors[] = {
+        "Agent claimed successfully",                   // 0
+        "Unknown argument",                             // 1
+        "Problems with claiming working directory",     // 2
+        "Missing dependencies",                         // 3
+        "Failure to connect to endpoint",               // 4
+        "Unknown HTTP error message",                   // 5
+        "invalid agent id",                             // 6
+        "invalid public key",                           // 7
+        "token has expired",                            // 8
+        "invalid token",                                // 9
+        "duplicate agent id",                           // 10
+        "claimed in another workspace",                 // 11
+        "internal server error"                         // 12
+};
+
+#define AGENT_UNCLAIMED 0
+#define AGENT_CLAIMED 1
+static uint8_t claiming_status = AGENT_UNCLAIMED;
+
+uint8_t is_agent_claimed(void)
+{
+    return (AGENT_CLAIMED == claiming_status);
+}
+
+#define CLAIMING_COMMAND_LENGTH 16384
+
+extern struct registry registry;
+
+/* rrd_init() must have been called before this function */
+void claim_agent(char *claiming_arguments)
+{
+    info("The claiming feature is under development and still subject to change before the next release");
+    return;
+
+    int exit_code;
+    pid_t command_pid;
+    char command_buffer[CLAIMING_COMMAND_LENGTH + 1];
+    FILE *fp;
+
+    snprintfz(command_buffer,
+              CLAIMING_COMMAND_LENGTH,
+              "exec netdata-claim.sh -hostname=%s -id=%s -url=%s %s",
+              netdata_configured_hostname,
+              localhost->machine_guid,
+              registry.cloud_base_url,
+              claiming_arguments);
+
+    info("Executing agent claiming command 'netdata-claim.sh'");
+    fp = mypopen(command_buffer, &command_pid);
+    if(!fp) {
+        error("Cannot popen(\"%s\").", command_buffer);
+        return;
+    }
+    info("Waiting for claiming command to finish.");
+    while (fgets(command_buffer, CLAIMING_COMMAND_LENGTH, fp) != NULL) {;}
+    exit_code = mypclose(fp, command_pid);
+    info("Agent claiming command returned with code %d", exit_code);
+    if (0 == exit_code) {
+        claiming_status = AGENT_CLAIMED;
+        info("Agent successfully claimed.");
+        return;
+    }
+    if (exit_code < 0) {
+        error("Agent claiming command failed to complete its run.");
+        return;
+    }
+    errno = 0;
+    unsigned maximum_known_exit_code = sizeof(claiming_errors) / sizeof(claiming_errors[0]);
+
+    if ((unsigned)exit_code > maximum_known_exit_code) {
+        error("Agent failed to be claimed with an unknown error.");
+        return;
+    }
+    error("Agent failed to be claimed with the following error message:");
+    error("\"%s\"", claiming_errors[exit_code]);
+}
+
+void load_claiming_state(void)
+{
+    info("The claiming feature is under development and still subject to change before the next release");
+    return;
+
+    char filename[FILENAME_MAX + 1];
+    struct stat statbuf;
+
+    snprintfz(filename, FILENAME_MAX, "%s/claim.d/is_claimed", netdata_configured_user_config_dir);
+    // check if the file exists
+    if (lstat(filename, &statbuf) != 0) {
+        info("File '%s' was not found. Setting state to AGENT_UNCLAIMED.", filename);
+        claiming_status = AGENT_UNCLAIMED;
+   } else {
+        info("File '%s' was found. Setting state to AGENT_CLAIMED.", filename);
+        claiming_status = AGENT_CLAIMED;
+    }
+}