webdiss/conf.c

#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <ctype.h>
#include <arpa/inet.h>

#include <jansson.h>
#include "conf.h"

static struct disabled_command *
conf_disable_commands(json_t *jtab);

struct conf *
conf_read(const char *filename) {

	json_t *j;
	json_error_t error;
	struct conf *conf;
	void *kv;

	/* defaults */
	conf = calloc(1, sizeof(struct conf));
	conf->redis_host = strdup("127.0.0.1");
	conf->redis_port = 6379;
	conf->http_host = strdup("0.0.0.0");
	conf->http_port = 7379;

	j = json_load_file(filename, 0, &error);
	if(!j) {
		fprintf(stderr, "Error: %s (line %d)\n", error.text, error.line);
		return conf;
	}

	for(kv = json_object_iter(j); kv; kv = json_object_iter_next(j, kv)) {
		json_t *jtmp = json_object_iter_value(kv);

		if(strcmp(json_object_iter_key(kv), "redis_host") == 0 && json_typeof(jtmp) == JSON_STRING) {
			free(conf->redis_host);
			conf->redis_host = strdup(json_string_value(jtmp));
		} else if(strcmp(json_object_iter_key(kv), "redis_port") == 0 && json_typeof(jtmp) == JSON_INTEGER) {
			conf->redis_port = (short)json_integer_value(jtmp);
		} else if(strcmp(json_object_iter_key(kv), "redis_auth") == 0 && json_typeof(jtmp) == JSON_STRING) {
			conf->redis_auth = strdup(json_string_value(jtmp));
		} else if(strcmp(json_object_iter_key(kv), "http_host") == 0 && json_typeof(jtmp) == JSON_STRING) {
			free(conf->http_host);
			conf->http_host = strdup(json_string_value(jtmp));
		} else if(strcmp(json_object_iter_key(kv), "http_port") == 0 && json_typeof(jtmp) == JSON_INTEGER) {
			conf->http_port = (short)json_integer_value(jtmp);
		} else if(strcmp(json_object_iter_key(kv), "disable") == 0 && json_typeof(jtmp) == JSON_OBJECT) {
			conf->disabled = conf_disable_commands(jtmp);
		}
	}

	json_decref(j);

	return conf;
}


struct disabled_command *
conf_disable_commands(json_t *jtab) {

	struct disabled_command *root = NULL;

	void *kv;
	for(kv = json_object_iter(jtab); kv; kv = json_object_iter_next(jtab, kv)) {

		unsigned int i, cur, n;
		char *p, *ip;
		const char *s;
		in_addr_t mask, subnet;
		unsigned short mask_bits = 0;

		struct disabled_command *dc;
		json_t *val = json_object_iter_value(kv);

		if(json_typeof(val) != JSON_ARRAY) {
			continue; /* TODO: report error? */
		}

		/* parse key in format "ip/mask" */
		s = json_object_iter_key(kv);
		p = strchr(s, '/');
		if(!p) {
			ip = strdup(s);
		} else {
			ip = calloc((size_t)(p - s + 1), 1);
			memcpy(ip, s, (size_t)(p - s));
			mask_bits = (unsigned short)atoi(p+1);
		}
		mask = (mask_bits == 0 ? 0 : (0xffffffff << (32 - mask_bits)));
		subnet = ntohl(inet_addr(ip)) & mask;

		/* count strings in the array */
		n = 0;
		for(i = 0; i < json_array_size(val); ++i) {
			json_t *jelem = json_array_get(val, (size_t)i);
			if(json_typeof(jelem) == JSON_STRING) {
				n++;
			}
		}

		/* allocate block */
		dc = calloc(1, sizeof(struct disabled_command));
		dc->commands = calloc((size_t)n, sizeof(char*));
		dc->subnet = subnet;
		dc->mask = mask;
		dc->count = n;
		dc->next = root;
		root = dc;

		/* add all disabled commands */
		for(i = 0, cur = 0; i < json_array_size(val); ++i) {
			json_t *jelem = json_array_get(val, i);
			if(json_typeof(jelem) == JSON_STRING) {
				size_t sz;
				s = json_string_value(jelem);
				sz = strlen(s);

				dc->commands[cur] = calloc(1 + sz, 1);
				memcpy(dc->commands[cur], s, sz);
				cur++;
			}
		}
	}

	return root;
}

void
conf_free(struct conf *conf) {

	free(conf->redis_host);
	free(conf->redis_auth);

	free(conf->http_host);

	free(conf);
}