Add systemd script

src/conf.c

@@ -0,0 +1,172 @@
+#include "conf.h"
+#include "tlog.h"
+#include "util.h"
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+
+#define MAX_LINE_LEN 1024
+#define MAX_KEY_LEN 64
+
+#define DEFAULT_DNS_CACHE_SIZE 512
+
+char dns_conf_server_ip[DNS_MAX_IPLEN];
+int dns_conf_cachesize = DEFAULT_DNS_CACHE_SIZE;
+struct dns_servers dns_conf_servers[DNS_MAX_SERVERS];
+int dns_conf_server_num;
+int dns_conf_loglevel = TLOG_ERROR;
+
+int config_bind(char *value)
+{
+	/* server bind address */
+	strncpy(dns_conf_server_ip, value, DNS_MAX_IPLEN);
+
+	return 0;
+}
+
+int config_server(char *value, dns_conf_server_type_t type)
+{
+	int index = dns_conf_server_num;
+	struct dns_servers *server;
+	int port = -1;
+
+	if (index >= DNS_MAX_SERVERS) {
+		tlog(TLOG_ERROR, "exceeds max server number");
+		return -1;
+	}
+
+	server = &dns_conf_servers[index];
+	/* parse ip, port from value */
+    if (parse_ip(value, server->server, &port) != 0) {
+		return -1;
+    }
+
+	/* if port is not defined, set port to default 53 */
+	if (port == PORT_NOT_DEFINED) {
+		port= DEFAULT_DNS_PORT;
+	} 
+	
+	server->type = type;
+	server->port = port;
+	dns_conf_server_num++;
+
+	return 0;
+}
+
+int config_server_udp(char *value)
+{
+	return config_server(value, DNS_CONF_TYPE_UDP);
+}
+
+int config_server_tcp(char *value)
+{
+	return config_server(value, DNS_CONF_TYPE_TCP);
+}
+
+int config_server_http(char *value)
+{
+	return config_server(value, DNS_CONF_TYPE_HTTP);
+}
+
+int config_cache_size(char *value)
+{
+	/* read dns cache size */
+	int cache_size = atoi(value);
+	if (cache_size < 0) {
+		return -1;
+	}
+
+	dns_conf_cachesize = cache_size;
+
+	return 0;
+}
+
+int config_log_level(char *value)
+{
+	/* read log level and set */
+	if (strncmp("debug", value, MAX_LINE_LEN) == 0) {
+		dns_conf_loglevel = TLOG_DEBUG;
+	} else if (strncmp("info", value, MAX_LINE_LEN) == 0) {
+		dns_conf_loglevel = TLOG_INFO;
+	} else if (strncmp("warn", value, MAX_LINE_LEN) == 0) {
+		dns_conf_loglevel = TLOG_WARN;
+	} else if (strncmp("error", value, MAX_LINE_LEN) == 0) {
+		dns_conf_loglevel = TLOG_ERROR;
+	}
+
+	return 0;
+}
+
+struct config_item {
+	const char *item;
+	int (*item_func)(char *value);
+};
+
+struct config_item config_item[] = {
+	{"bind", config_bind},
+	{"server", config_server_udp},
+    {"server-tcp", config_server_tcp},
+    {"server-http", config_server_http},
+	{"cache-size", config_cache_size},
+	{"loglevel", config_log_level},
+};
+int config_item_num = sizeof(config_item) / sizeof(struct config_item);
+
+int load_conf(const char *file)
+{
+	FILE *fp = NULL;
+	char line[MAX_LINE_LEN];
+	char key[MAX_KEY_LEN];
+	char value[MAX_LINE_LEN];
+	int filed_num = 0;
+	int line_num = 0;
+	int i;
+
+	fp = fopen(file, "r");
+	if (fp == NULL) {
+		tlog(TLOG_ERROR, "config file %s not exist.", file);
+		return -1;
+	}
+
+	while (fgets(line, MAX_LINE_LEN, fp)) {
+		line_num++;
+		filed_num = sscanf(line, "%63s %1023[^\r\n]s", key, value);
+		if (filed_num <= 0) {
+			continue;
+		}
+
+		/* comment, skip */
+		if (key[0] == '#') {
+			continue;
+		}
+
+		/* if field format is not key = value, error */
+		if (filed_num != 2) {
+			goto errout;
+		}
+
+		for (i = 0; i < config_item_num; i++) {
+			if (strncmp(config_item[i].item, key, MAX_KEY_LEN) != 0) {
+				continue;
+			}
+
+			/* call item function */
+			if (config_item[i].item_func(value) != 0) {
+				goto errout;
+			}
+
+			break;
+		}
+	}
+
+	fclose(fp);
+
+	return 0;
+errout:
+	printf("invalid config at line %d: %s", line_num, line);
+	if (fp) {
+		fclose(fp);
+	}
+	return -1;
+}