src/claim/cloud-conf.c

Rama master · réplica de lectura del repositorio original.

// SPDX-License-Identifier: GPL-3.0-or-later

#include "claim.h"

struct config cloud_config = APPCONFIG_INITIALIZER;

const char *cloud_config_url_get(void) {
    return inicfg_get(&cloud_config, CONFIG_SECTION_GLOBAL, "url", DEFAULT_CLOUD_BASE_URL);
}

void cloud_config_url_set(const char *url)
{
    if (!url || !*url) return;

    const char *existing = cloud_config_url_get();
    if (strcmp(existing, url) != 0)
        inicfg_set(&cloud_config, CONFIG_SECTION_GLOBAL, "url", url);
}

static const char *cloud_config_proxy_source = NULL;
static bool cloud_config_proxy_explicitly_set = false;

const char *cloud_config_proxy_get(void) {
    // load cloud.conf or internal default
    const char *proxy = inicfg_get(&cloud_config, CONFIG_SECTION_GLOBAL, "proxy", "env");

    // backwards compatibility, from when proxy was in netdata.conf
    // netdata.conf has bigger priority
    if (inicfg_exists(&netdata_config, CONFIG_SECTION_CLOUD, "proxy")) {
        // get it from netdata.conf
        proxy = inicfg_get(&netdata_config, CONFIG_SECTION_CLOUD, "proxy", proxy);

        // update cloud.conf
        proxy = inicfg_set(&cloud_config, CONFIG_SECTION_GLOBAL, "proxy", proxy);
        cloud_config_proxy_source = "netdata.conf [cloud]";
        cloud_config_proxy_explicitly_set = true;
    }
    else {
        // set in netdata.conf the proxy of cloud.conf
        inicfg_set(&netdata_config, CONFIG_SECTION_CLOUD, "proxy", proxy);
        cloud_config_proxy_source = "cloud.conf";
    }

    return proxy;
}

const char *cloud_config_proxy_source_get(void) {
    return cloud_config_proxy_source;
}

bool cloud_config_proxy_is_explicitly_set(void) {
    return cloud_config_proxy_explicitly_set;
}

bool cloud_config_insecure_get(void) {
    // load it from cloud.conf or use internal default
    return inicfg_get_boolean(&cloud_config, CONFIG_SECTION_GLOBAL, "insecure", CONFIG_BOOLEAN_NO);
}

static void cloud_conf_load_defaults(void) {
    inicfg_get(&cloud_config, CONFIG_SECTION_GLOBAL, "url", DEFAULT_CLOUD_BASE_URL);
    inicfg_get(&cloud_config, CONFIG_SECTION_GLOBAL, "proxy", "env");
    inicfg_get(&cloud_config, CONFIG_SECTION_GLOBAL, "token", "");
    inicfg_get(&cloud_config, CONFIG_SECTION_GLOBAL, "rooms", "");
    inicfg_get_boolean(&cloud_config, CONFIG_SECTION_GLOBAL, "insecure", CONFIG_BOOLEAN_NO);
    inicfg_get(&cloud_config, CONFIG_SECTION_GLOBAL, "machine_guid", "");
    inicfg_get(&cloud_config, CONFIG_SECTION_GLOBAL, "claimed_id", "");
    inicfg_get(&cloud_config, CONFIG_SECTION_GLOBAL, "hostname", "");
}

void cloud_conf_load(int silent) {
    netdata_conf_section_directories();

    errno_clear();
    char *filename = filename_from_path_entry_strdupz(netdata_configured_cloud_dir, "cloud.conf");
    int ret = inicfg_load(&cloud_config, filename, 1, NULL);

    if(!ret && !silent)
        nd_log(NDLS_DAEMON, NDLP_ERR,
               "CLAIM: cannot load cloud config '%s'. Running with internal defaults.", filename);

    freez(filename);

    inicfg_move(&cloud_config,
                   CONFIG_SECTION_GLOBAL, "cloud base url",
                   CONFIG_SECTION_GLOBAL, "url");

    // check if proxy was explicitly set in cloud.conf before defaults overwrite it
    cloud_config_proxy_explicitly_set = inicfg_exists(&cloud_config, CONFIG_SECTION_GLOBAL, "proxy");

    cloud_conf_load_defaults();
}

void cloud_conf_init_after_registry(void) {
    const char *machine_guid = inicfg_get(&cloud_config, CONFIG_SECTION_GLOBAL, "machine_guid", "");
    const char *hostname = inicfg_get(&cloud_config, CONFIG_SECTION_GLOBAL, "hostname", "");

    // for machine guid and hostname we have to use inicfg_set() for that they will be saved uncommented
    if(!machine_guid || !*machine_guid)
        inicfg_set(&cloud_config, CONFIG_SECTION_GLOBAL, "machine_guid", machine_guid_get_txt());

    if(!hostname || !*hostname)
        inicfg_set(&cloud_config, CONFIG_SECTION_GLOBAL, "hostname", registry_get_this_machine_hostname());
}

bool cloud_conf_save(void) {
    char filename[FILENAME_MAX + 1];

    CLEAN_BUFFER *wb = buffer_create(0, NULL);
    inicfg_generate(&cloud_config, wb, false, false);
    snprintfz(filename, sizeof(filename), "%s/cloud.conf", netdata_configured_cloud_dir);
    FILE *fp = fopen(filename, "w");
    if(!fp) {
        nd_log(NDLS_DAEMON, NDLP_ERR, "Cannot open file '%s' for writing.", filename);
        return false;
    }

    int written = fprintf(fp, "%s", buffer_tostring(wb));
    int saved_errno = errno;
    if(fclose(fp) != 0 || written < 0) {
        if(written >= 0)
            saved_errno = errno;

        errno = saved_errno;
        nd_log(NDLS_DAEMON, NDLP_ERR, "Cannot save file '%s'.", filename);
        return false;
    }

    return true;
}

bool cloud_conf_regenerate(const char *claimed_id_str, const char *machine_guid, const char *hostname, const char *token, const char *rooms, const char *url, const char *proxy, bool insecure) {
    // for backwards compatibility (older agents), save the claimed_id to its file
    claimed_id_save_to_file(claimed_id_str);

    inicfg_set(&cloud_config, CONFIG_SECTION_GLOBAL, "url", url);
    inicfg_set(&cloud_config, CONFIG_SECTION_GLOBAL, "proxy", proxy ? proxy : "");
    inicfg_set(&cloud_config, CONFIG_SECTION_GLOBAL, "token", token ? token : "");
    inicfg_set(&cloud_config, CONFIG_SECTION_GLOBAL, "rooms", rooms ? rooms : "");
    inicfg_set_boolean(&cloud_config, CONFIG_SECTION_GLOBAL, "insecure", insecure);
    inicfg_set(&cloud_config, CONFIG_SECTION_GLOBAL, "machine_guid", machine_guid ? machine_guid : "");
    inicfg_set(&cloud_config, CONFIG_SECTION_GLOBAL, "claimed_id", claimed_id_str ? claimed_id_str : "");
    inicfg_set(&cloud_config, CONFIG_SECTION_GLOBAL, "hostname", hostname ? hostname : "");

    return cloud_conf_save();
}