| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809 |
- /*************************************************************************
- *
- * Copyright (C) 2018-2023 Ruilin Peng (Nick) <[email protected]>.
- *
- * smartdns is free software: you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation, either version 3 of the License, or
- * (at your option) any later version.
- *
- * smartdns is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- * GNU General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program. If not, see <http://www.gnu.org/licenses/>.
- */
- #include "dns_cache.h"
- #include "stringutil.h"
- #include "timer.h"
- #include "tlog.h"
- #include "util.h"
- #include <errno.h>
- #include <fcntl.h>
- #include <pthread.h>
- #include <stdio.h>
- #include <string.h>
- #include <sys/types.h>
- #define DNS_CACHE_MAX_HITNUM 6000
- #define DNS_CACHE_HITNUM_STEP 3
- #define DNS_CACHE_HITNUM_STEP_MAX 6
- #define DNS_CACHE_READ_TIMEOUT 60
- struct dns_cache_head {
- struct hash_table cache_hash;
- struct list_head cache_list;
- atomic_t num;
- int size;
- pthread_mutex_t lock;
- dns_cache_callback timeout_callback;
- };
- typedef int (*dns_cache_read_callback)(struct dns_cache_record *cache_record, struct dns_cache_data *cache_data);
- static int is_cache_init;
- static struct dns_cache_head dns_cache_head;
- int dns_cache_init(int size, dns_cache_callback timeout_callback)
- {
- int bits = 0;
- if (is_cache_init == 1) {
- return -1;
- }
- INIT_LIST_HEAD(&dns_cache_head.cache_list);
- bits = ilog2(size) - 1;
- if (bits >= 20) {
- bits = 20;
- } else if (bits < 12) {
- bits = 12;
- }
- hash_table_init(dns_cache_head.cache_hash, bits, malloc);
- atomic_set(&dns_cache_head.num, 0);
- dns_cache_head.size = size;
- dns_cache_head.timeout_callback = timeout_callback;
- pthread_mutex_init(&dns_cache_head.lock, NULL);
- is_cache_init = 1;
- return 0;
- }
- static struct dns_cache *_dns_cache_first(void)
- {
- return list_first_entry_or_null(&dns_cache_head.cache_list, struct dns_cache, list);
- }
- static void _dns_cache_delete(struct dns_cache *dns_cache)
- {
- hash_del(&dns_cache->node);
- list_del_init(&dns_cache->list);
- atomic_dec(&dns_cache_head.num);
- dns_cache_data_put(dns_cache->cache_data);
- dns_cache->cache_data = NULL;
- free(dns_cache);
- }
- void dns_cache_get(struct dns_cache *dns_cache)
- {
- if (atomic_inc_return(&dns_cache->ref) == 1) {
- tlog(TLOG_ERROR, "BUG: dns_cache is invalid.");
- return;
- }
- }
- void dns_cache_release(struct dns_cache *dns_cache)
- {
- if (dns_cache == NULL) {
- return;
- }
- if (!atomic_dec_and_test(&dns_cache->ref)) {
- return;
- }
- _dns_cache_delete(dns_cache);
- }
- static void _dns_cache_remove(struct dns_cache *dns_cache)
- {
- hash_del(&dns_cache->node);
- list_del_init(&dns_cache->list);
- dns_timer_del(&dns_cache->timer);
- dns_cache_release(dns_cache);
- }
- uint32_t dns_cache_get_query_flag(struct dns_cache *dns_cache)
- {
- return dns_cache->info.query_flag;
- }
- const char *dns_cache_get_dns_group_name(struct dns_cache *dns_cache)
- {
- return dns_cache->info.dns_group_name;
- }
- struct dns_cache_data *dns_cache_new_data_packet(void *packet, size_t packet_len)
- {
- struct dns_cache_packet *cache_packet = NULL;
- size_t data_size = 0;
- if (packet == NULL || packet_len <= 0) {
- return NULL;
- }
- data_size = sizeof(*cache_packet) + packet_len;
- cache_packet = malloc(data_size);
- if (cache_packet == NULL) {
- return NULL;
- }
- memcpy(cache_packet->data, packet, packet_len);
- memset(&cache_packet->head, 0, sizeof(cache_packet->head));
- cache_packet->head.size = packet_len;
- cache_packet->head.magic = MAGIC_CACHE_DATA;
- atomic_set(&cache_packet->head.ref, 1);
- return (struct dns_cache_data *)cache_packet;
- }
- static void dns_cache_timer_release(struct tw_timer_list *timer, void *data)
- {
- struct dns_cache *dns_cache = data;
- dns_cache_release(dns_cache);
- }
- static void dns_cache_expired(struct tw_timer_list *timer, void *data, unsigned long timestamp)
- {
- struct dns_cache *dns_cache = data;
- if (dns_cache->del_pending == 1) {
- dns_cache_release(dns_cache);
- return;
- }
- if (dns_cache_head.timeout_callback) {
- if (dns_cache_head.timeout_callback(dns_cache) != 0) {
- dns_cache_release(dns_cache);
- return;
- }
- }
- dns_cache->del_pending = 1;
- dns_timer_mod(&dns_cache->timer, 5);
- }
- static int _dns_cache_replace(struct dns_cache_key *cache_key, int rcode, int ttl, int speed, int timeout,
- int update_time, struct dns_cache_data *cache_data)
- {
- struct dns_cache *dns_cache = NULL;
- struct dns_cache_data *old_cache_data = NULL;
- if (dns_cache_head.size <= 0) {
- return 0;
- }
- /* lookup existing cache */
- dns_cache = dns_cache_lookup(cache_key);
- if (dns_cache == NULL) {
- return -1;
- }
- if (ttl < DNS_CACHE_TTL_MIN) {
- ttl = DNS_CACHE_TTL_MIN;
- }
- /* update cache data */
- pthread_mutex_lock(&dns_cache_head.lock);
- dns_cache->del_pending = 0;
- dns_cache->info.qtype = cache_key->qtype;
- dns_cache->info.query_flag = cache_key->query_flag;
- dns_cache->info.ttl = ttl;
- dns_cache->info.speed = speed;
- dns_cache->info.timeout = timeout;
- dns_cache->info.is_visited = 1;
- if (cache_data) {
- old_cache_data = dns_cache->cache_data;
- dns_cache->cache_data = cache_data;
- }
- if (update_time) {
- time(&dns_cache->info.insert_time);
- }
- time(&dns_cache->info.replace_time);
- list_del(&dns_cache->list);
- list_add_tail(&dns_cache->list, &dns_cache_head.cache_list);
- dns_timer_mod(&dns_cache->timer, timeout);
- pthread_mutex_unlock(&dns_cache_head.lock);
- if (old_cache_data) {
- dns_cache_data_put(old_cache_data);
- }
- dns_cache_release(dns_cache);
- return 0;
- }
- int dns_cache_replace(struct dns_cache_key *cache_key, int rcode, int ttl, int speed, int timeout, int update_time,
- struct dns_cache_data *cache_data)
- {
- return _dns_cache_replace(cache_key, rcode, ttl, speed, timeout, update_time, cache_data);
- }
- static void _dns_cache_remove_by_domain(struct dns_cache_key *cache_key)
- {
- uint32_t key = 0;
- struct dns_cache *dns_cache = NULL;
- key = hash_string(cache_key->domain);
- key = jhash(&cache_key->qtype, sizeof(cache_key->qtype), key);
- key = hash_string_initval(cache_key->dns_group_name, key);
- key = jhash(&cache_key->query_flag, sizeof(cache_key->query_flag), key);
- pthread_mutex_lock(&dns_cache_head.lock);
- hash_table_for_each_possible(dns_cache_head.cache_hash, dns_cache, node, key)
- {
- if (dns_cache->info.qtype != cache_key->qtype) {
- continue;
- }
- if (dns_cache->info.query_flag != cache_key->query_flag) {
- continue;
- }
- if (strncmp(cache_key->domain, dns_cache->info.domain, DNS_MAX_CNAME_LEN) != 0) {
- continue;
- }
- if (strncmp(dns_cache->info.dns_group_name, cache_key->dns_group_name, DNS_GROUP_NAME_LEN) != 0) {
- continue;
- }
- _dns_cache_remove(dns_cache);
- break;
- }
- pthread_mutex_unlock(&dns_cache_head.lock);
- }
- static int _dns_cache_insert(struct dns_cache_info *info, struct dns_cache_data *cache_data, struct list_head *head)
- {
- uint32_t key = 0;
- struct dns_cache *dns_cache = NULL;
- /* if cache already exists, free */
- struct dns_cache_key cache_key;
- cache_key.qtype = info->qtype;
- cache_key.query_flag = info->query_flag;
- cache_key.domain = info->domain;
- cache_key.dns_group_name = info->dns_group_name;
- _dns_cache_remove_by_domain(&cache_key);
- dns_cache = malloc(sizeof(*dns_cache));
- if (dns_cache == NULL) {
- goto errout;
- }
- memset(dns_cache, 0, sizeof(*dns_cache));
- key = hash_string(info->domain);
- key = jhash(&info->qtype, sizeof(info->qtype), key);
- key = hash_string_initval(info->dns_group_name, key);
- key = jhash(&info->query_flag, sizeof(info->query_flag), key);
- atomic_set(&dns_cache->ref, 1);
- memcpy(&dns_cache->info, info, sizeof(*info));
- dns_cache->del_pending = 0;
- dns_cache->cache_data = cache_data;
- dns_cache->timer.function = dns_cache_expired;
- dns_cache->timer.del_function = dns_cache_timer_release;
- dns_cache->timer.expires = info->timeout;
- dns_cache->timer.data = dns_cache;
- pthread_mutex_lock(&dns_cache_head.lock);
- hash_table_add(dns_cache_head.cache_hash, &dns_cache->node, key);
- list_add_tail(&dns_cache->list, head);
- INIT_LIST_HEAD(&dns_cache->check_list);
- /* Release extra cache, remove oldest cache record */
- if (atomic_inc_return(&dns_cache_head.num) > dns_cache_head.size) {
- struct dns_cache *del_cache = NULL;
- del_cache = _dns_cache_first();
- if (del_cache) {
- _dns_cache_remove(del_cache);
- }
- }
- dns_cache_get(dns_cache);
- dns_timer_add(&dns_cache->timer);
- pthread_mutex_unlock(&dns_cache_head.lock);
- return 0;
- errout:
- if (dns_cache) {
- dns_cache_release(dns_cache);
- }
- return -1;
- }
- int dns_cache_insert(struct dns_cache_key *cache_key, int rcode, int ttl, int speed, int timeout,
- struct dns_cache_data *cache_data)
- {
- struct dns_cache_info info;
- if (cache_data == NULL || cache_key == NULL || cache_key->dns_group_name == NULL || cache_key->domain == NULL) {
- return -1;
- }
- if (dns_cache_head.size <= 0) {
- dns_cache_data_put(cache_data);
- return 0;
- }
- if (ttl < DNS_CACHE_TTL_MIN) {
- ttl = DNS_CACHE_TTL_MIN;
- }
- memset(&info, 0, sizeof(info));
- info.hitnum = 3;
- safe_strncpy(info.domain, cache_key->domain, DNS_MAX_CNAME_LEN);
- info.qtype = cache_key->qtype;
- safe_strncpy(info.dns_group_name, cache_key->dns_group_name, DNS_GROUP_NAME_LEN);
- info.query_flag = cache_key->query_flag;
- info.ttl = ttl;
- info.hitnum_update_add = DNS_CACHE_HITNUM_STEP;
- info.speed = speed;
- info.timeout = timeout;
- info.is_visited = 1;
- info.rcode = rcode;
- time(&info.insert_time);
- time(&info.replace_time);
- return _dns_cache_insert(&info, cache_data, &dns_cache_head.cache_list);
- }
- struct dns_cache *dns_cache_lookup(struct dns_cache_key *cache_key)
- {
- uint32_t key = 0;
- struct dns_cache *dns_cache = NULL;
- struct dns_cache *dns_cache_ret = NULL;
- time_t now = 0;
- if (dns_cache_head.size <= 0) {
- return NULL;
- }
- key = hash_string(cache_key->domain);
- key = jhash(&cache_key->qtype, sizeof(cache_key->qtype), key);
- key = hash_string_initval(cache_key->dns_group_name, key);
- key = jhash(&cache_key->query_flag, sizeof(cache_key->query_flag), key);
- time(&now);
- /* find cache */
- pthread_mutex_lock(&dns_cache_head.lock);
- hash_table_for_each_possible(dns_cache_head.cache_hash, dns_cache, node, key)
- {
- if (dns_cache->info.qtype != cache_key->qtype) {
- continue;
- }
- if (strncmp(cache_key->domain, dns_cache->info.domain, DNS_MAX_CNAME_LEN) != 0) {
- continue;
- }
- if (strncmp(cache_key->dns_group_name, dns_cache->info.dns_group_name, DNS_GROUP_NAME_LEN) != 0) {
- continue;
- }
- if (cache_key->query_flag != dns_cache->info.query_flag) {
- continue;
- }
- dns_cache_ret = dns_cache;
- break;
- }
- if (dns_cache_ret) {
- dns_cache_get(dns_cache_ret);
- }
- pthread_mutex_unlock(&dns_cache_head.lock);
- return dns_cache_ret;
- }
- int dns_cache_get_ttl(struct dns_cache *dns_cache)
- {
- time_t now = 0;
- int ttl = 0;
- time(&now);
- ttl = dns_cache->info.insert_time + dns_cache->info.ttl - now;
- if (ttl < 0) {
- return 0;
- }
- return ttl;
- }
- struct dns_cache_data *dns_cache_get_data(struct dns_cache *dns_cache)
- {
- struct dns_cache_data *cache_data;
- pthread_mutex_lock(&dns_cache_head.lock);
- dns_cache_data_get(dns_cache->cache_data);
- cache_data = dns_cache->cache_data;
- pthread_mutex_unlock(&dns_cache_head.lock);
- return cache_data;
- }
- void dns_cache_data_get(struct dns_cache_data *cache_data)
- {
- if (atomic_inc_return(&cache_data->head.ref) == 1) {
- tlog(TLOG_ERROR, "BUG: dns_cache data is invalid.");
- return;
- }
- return;
- }
- void dns_cache_data_put(struct dns_cache_data *cache_data)
- {
- if (cache_data == NULL) {
- return;
- }
- if (!atomic_dec_and_test(&cache_data->head.ref)) {
- return;
- }
- free(cache_data);
- }
- int dns_cache_is_visited(struct dns_cache *dns_cache)
- {
- return dns_cache->info.is_visited;
- }
- void dns_cache_delete(struct dns_cache *dns_cache)
- {
- pthread_mutex_lock(&dns_cache_head.lock);
- _dns_cache_remove(dns_cache);
- pthread_mutex_unlock(&dns_cache_head.lock);
- }
- int dns_cache_hitnum_dec_get(struct dns_cache *dns_cache)
- {
- pthread_mutex_lock(&dns_cache_head.lock);
- dns_cache->info.hitnum--;
- if (dns_cache->info.hitnum_update_add > DNS_CACHE_HITNUM_STEP) {
- dns_cache->info.hitnum_update_add--;
- }
- pthread_mutex_unlock(&dns_cache_head.lock);
- return dns_cache->info.hitnum;
- }
- void dns_cache_update(struct dns_cache *dns_cache)
- {
- pthread_mutex_lock(&dns_cache_head.lock);
- if (!list_empty(&dns_cache->list)) {
- list_del_init(&dns_cache->list);
- list_add_tail(&dns_cache->list, &dns_cache_head.cache_list);
- dns_cache->info.hitnum += dns_cache->info.hitnum_update_add;
- if (dns_cache->info.hitnum > DNS_CACHE_MAX_HITNUM) {
- dns_cache->info.hitnum = DNS_CACHE_MAX_HITNUM;
- }
- if (dns_cache->info.hitnum_update_add < DNS_CACHE_HITNUM_STEP_MAX) {
- dns_cache->info.hitnum_update_add++;
- }
- dns_cache->info.is_visited = 1;
- }
- pthread_mutex_unlock(&dns_cache_head.lock);
- }
- static int _dns_cache_read_to_cache(struct dns_cache_record *cache_record, struct dns_cache_data *cache_data)
- {
- struct list_head *head = NULL;
- head = &dns_cache_head.cache_list;
- struct dns_cache_info *info = &cache_record->info;
- time_t now = time(NULL);
- unsigned int seed_tmp = now;
- int passed_time = now - info->replace_time;
- int timeout = info->timeout - passed_time;
- if (timeout < DNS_CACHE_READ_TIMEOUT * 2) {
- timeout = DNS_CACHE_READ_TIMEOUT + (rand_r(&seed_tmp) % DNS_CACHE_READ_TIMEOUT);
- }
- if (timeout > dns_conf_serve_expired_ttl && dns_conf_serve_expired_ttl >= 0) {
- timeout = dns_conf_serve_expired_ttl;
- }
- info->timeout = timeout;
- if (_dns_cache_insert(&cache_record->info, cache_data, head) != 0) {
- tlog(TLOG_ERROR, "insert cache data failed.");
- cache_data = NULL;
- goto errout;
- }
- dns_cache_data_get(cache_data);
- daemon_keepalive();
- return 0;
- errout:
- return -1;
- }
- static int _dns_cache_read_record(int fd, uint32_t cache_number, dns_cache_read_callback callback)
- {
- unsigned int i = 0;
- ssize_t ret = 0;
- struct dns_cache_record cache_record;
- struct dns_cache_data_head data_head;
- struct dns_cache_data *cache_data = NULL;
- for (i = 0; i < cache_number; i++) {
- ret = read(fd, &cache_record, sizeof(cache_record));
- if (ret != sizeof(cache_record)) {
- tlog(TLOG_ERROR, "read cache failed, %s", strerror(errno));
- goto errout;
- }
- if (cache_record.magic != MAGIC_RECORD) {
- tlog(TLOG_ERROR, "magic is invalid.");
- goto errout;
- }
- ret = read(fd, &data_head, sizeof(data_head));
- if (ret != sizeof(data_head)) {
- tlog(TLOG_ERROR, "read data head failed, %s", strerror(errno));
- goto errout;
- }
- if (data_head.magic != MAGIC_CACHE_DATA) {
- tlog(TLOG_ERROR, "data magic is invalid.");
- goto errout;
- }
- if (data_head.size > 1024 * 8) {
- tlog(TLOG_ERROR, "data may invalid, skip load cache.");
- goto errout;
- }
- cache_data = malloc(data_head.size + sizeof(data_head));
- if (cache_data == NULL) {
- tlog(TLOG_ERROR, "malloc cache data failed %s", strerror(errno));
- goto errout;
- }
- memcpy(&cache_data->head, &data_head, sizeof(data_head));
- atomic_set(&cache_data->head.ref, 1);
- ret = read(fd, cache_data->data, data_head.size);
- if (ret != data_head.size) {
- tlog(TLOG_ERROR, "read cache data failed, %s", strerror(errno));
- goto errout;
- }
- /* set cache unvisited, so that when refreshing ipset/nftset, reload ipset list by restarting smartdns */
- cache_record.info.is_visited = 0;
- cache_record.info.domain[DNS_MAX_CNAME_LEN - 1] = '\0';
- cache_record.info.dns_group_name[DNS_GROUP_NAME_LEN - 1] = '\0';
- ret = callback(&cache_record, cache_data);
- dns_cache_data_put(cache_data);
- cache_data = NULL;
- if (ret != 0) {
- goto errout;
- }
- }
- return 0;
- errout:
- if (cache_data) {
- dns_cache_data_put(cache_data);
- }
- return -1;
- }
- static int _dns_cache_file_read(const char *file, dns_cache_read_callback callback)
- {
- int fd = -1;
- ssize_t ret = 0;
- off_t filesize = 0;
- fd = open(file, O_RDONLY);
- if (fd < 0) {
- return 0;
- }
- filesize = lseek(fd, 0, SEEK_END);
- lseek(fd, 0, SEEK_SET);
- posix_fadvise(fd, 0, filesize, POSIX_FADV_WILLNEED | POSIX_FADV_SEQUENTIAL);
- struct dns_cache_file cache_file;
- ret = read(fd, &cache_file, sizeof(cache_file));
- if (ret != sizeof(cache_file)) {
- tlog(TLOG_ERROR, "read cache head failed.");
- goto errout;
- }
- if (cache_file.magic != MAGIC_NUMBER) {
- tlog(TLOG_ERROR, "cache file is invalid.");
- goto errout;
- }
- if (strncmp(cache_file.version, dns_cache_file_version(), DNS_CACHE_VERSION_LEN) != 0) {
- tlog(TLOG_WARN, "cache version is different, skip load cache.");
- goto errout;
- }
- tlog(TLOG_INFO, "load cache file %s, total %d records", file, cache_file.cache_number);
- if (_dns_cache_read_record(fd, cache_file.cache_number, callback) != 0) {
- goto errout;
- }
- close(fd);
- return 0;
- errout:
- if (fd > 0) {
- close(fd);
- }
- return -1;
- }
- int dns_cache_load(const char *file)
- {
- return _dns_cache_file_read(file, _dns_cache_read_to_cache);
- }
- static int _dns_cache_write_record(int fd, uint32_t *cache_number, struct list_head *head)
- {
- struct dns_cache *dns_cache = NULL;
- struct dns_cache *tmp = NULL;
- struct dns_cache_record cache_record;
- memset(&cache_record, 0, sizeof(cache_record));
- pthread_mutex_lock(&dns_cache_head.lock);
- list_for_each_entry_safe(dns_cache, tmp, head, list)
- {
- cache_record.magic = MAGIC_RECORD;
- memcpy(&cache_record.info, &dns_cache->info, sizeof(struct dns_cache_info));
- ssize_t ret = write(fd, &cache_record, sizeof(cache_record));
- if (ret != sizeof(cache_record)) {
- tlog(TLOG_ERROR, "write cache failed, %s", strerror(errno));
- goto errout;
- }
- struct dns_cache_data *cache_data = dns_cache->cache_data;
- ret = write(fd, cache_data, sizeof(*cache_data) + cache_data->head.size);
- if (ret != (int)sizeof(*cache_data) + cache_data->head.size) {
- tlog(TLOG_ERROR, "write cache data failed, %s", strerror(errno));
- goto errout;
- }
- (*cache_number)++;
- }
- pthread_mutex_unlock(&dns_cache_head.lock);
- return 0;
- errout:
- pthread_mutex_unlock(&dns_cache_head.lock);
- return -1;
- }
- static int _dns_cache_write_records(int fd, uint32_t *cache_number)
- {
- if (_dns_cache_write_record(fd, cache_number, &dns_cache_head.cache_list) != 0) {
- return -1;
- }
- return 0;
- }
- int dns_cache_save(const char *file, int check_lock)
- {
- int fd = -1;
- uint32_t cache_number = 0;
- tlog(TLOG_DEBUG, "write cache file %s", file);
- /* check lock */
- if (check_lock == 1) {
- if (pthread_mutex_trylock(&dns_cache_head.lock) != 0) {
- return -1;
- }
- pthread_mutex_unlock(&dns_cache_head.lock);
- }
- fd = open(file, O_TRUNC | O_CREAT | O_WRONLY, 0640);
- if (fd < 0) {
- tlog(TLOG_ERROR, "create file %s failed, %s", file, strerror(errno));
- goto errout;
- }
- struct dns_cache_file cache_file;
- memset(&cache_file, 0, sizeof(cache_file));
- cache_file.magic = MAGIC_NUMBER;
- safe_strncpy(cache_file.version, dns_cache_file_version(), DNS_CACHE_VERSION_LEN);
- cache_file.cache_number = 0;
- if (lseek(fd, sizeof(cache_file), SEEK_SET) < 0) {
- tlog(TLOG_ERROR, "seek file %s failed, %s", file, strerror(errno));
- goto errout;
- }
- if (_dns_cache_write_records(fd, &cache_number) != 0) {
- tlog(TLOG_ERROR, "write record to file %s failed.", file);
- goto errout;
- }
- if (lseek(fd, 0, SEEK_SET) < 0) {
- tlog(TLOG_ERROR, "seek file %s failed, %s", file, strerror(errno));
- goto errout;
- }
- cache_file.cache_number = cache_number;
- if (write(fd, &cache_file, sizeof(cache_file)) != sizeof(cache_file)) {
- tlog(TLOG_ERROR, "write file head %s failed, %s, %d", file, strerror(errno), fd);
- goto errout;
- }
- tlog(TLOG_DEBUG, "wrote total %d records.", cache_number);
- close(fd);
- return 0;
- errout:
- if (fd > 0) {
- close(fd);
- }
- return -1;
- }
- static int _dns_cache_print(struct dns_cache_record *cache_record, struct dns_cache_data *cache_data)
- {
- printf("domain: %s, qtype: %d, ttl: %d, speed: %.1fms\n", cache_record->info.domain, cache_record->info.qtype,
- cache_record->info.ttl, (float)cache_record->info.speed / 10);
- return 0;
- }
- int dns_cache_print(const char *file)
- {
- if (access(file, F_OK) != 0) {
- tlog(TLOG_ERROR, "cache file %s not exist.", file);
- return -1;
- }
- return _dns_cache_file_read(file, _dns_cache_print);
- }
- void dns_cache_destroy(void)
- {
- struct dns_cache *dns_cache = NULL;
- struct dns_cache *tmp = NULL;
- if (is_cache_init == 0) {
- return;
- }
- pthread_mutex_lock(&dns_cache_head.lock);
- list_for_each_entry_safe(dns_cache, tmp, &dns_cache_head.cache_list, list)
- {
- _dns_cache_remove(dns_cache);
- }
- pthread_mutex_unlock(&dns_cache_head.lock);
- pthread_mutex_destroy(&dns_cache_head.lock);
- hash_table_free(dns_cache_head.cache_hash, free);
- is_cache_init = 0;
- }
- const char *dns_cache_file_version(void)
- {
- const char *version = "cache ver 1.3";
- return version;
- }
|