2 * collectd - src/utils_fbhash.c
3 * Copyright (C) 2009 Florian octo Forster
5 * Permission is hereby granted, free of charge, to any person obtaining a
6 * copy of this software and associated documentation files (the "Software"),
7 * to deal in the Software without restriction, including without limitation
8 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
9 * and/or sell copies of the Software, and to permit persons to whom the
10 * Software is furnished to do so, subject to the following conditions:
12 * The above copyright notice and this permission notice shall be included in
13 * all copies or substantial portions of the Software.
15 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
20 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
21 * DEALINGS IN THE SOFTWARE.
24 * Florian octo Forster <octo at collectd.org>
31 #include "utils_fbhash.h"
32 #include "utils_avltree.h"
46 static void fbh_free_tree (c_avl_tree_t *tree) /* {{{ */
58 status = c_avl_pick (tree, (void *) &key, (void *) &value);
67 } /* }}} void fbh_free_tree */
69 static int fbh_read_file (fbhash_t *h) /* {{{ */
73 struct flock fl = { 0 };
77 fh = fopen (h->filename, "r");
82 fl.l_whence = SEEK_SET;
83 /* TODO: Lock file? -> fcntl */
85 status = fcntl (fileno (fh), F_SETLK, &fl);
92 tree = c_avl_create ((int (*) (const void *, const void *)) strcmp);
99 /* Read `fh' into `tree' */
100 while (fgets (buffer, sizeof (buffer), fh) != NULL) /* {{{ */
109 buffer[sizeof (buffer) - 1] = 0;
110 len = strlen (buffer);
112 /* Remove trailing newline characters. */
114 && ((buffer[len - 1] == '\n') || (buffer[len - 1] == '\r')))
120 /* Seek first non-space character */
122 while ((*key != 0) && isspace ((int) *key))
125 /* Skip empty lines and comments */
126 if ((key[0] == 0) || (key[0] == '#'))
129 /* Seek first colon */
130 value = strchr (key, ':');
134 /* Null-terminate `key'. */
138 /* Skip leading whitespace */
139 while ((*value != 0) && isspace ((int) *value))
142 /* Skip lines without value */
146 key_copy = strdup (key);
147 value_copy = strdup (value);
149 if ((key_copy == NULL) || (value_copy == NULL))
156 status = c_avl_insert (tree, key_copy, value_copy);
164 DEBUG ("utils_fbhash: fbh_read_file: key = %s; value = %s;",
166 } /* }}} while (fgets) */
170 fbh_free_tree (h->tree);
174 } /* }}} int fbh_read_file */
176 static int fbh_check_file (fbhash_t *h) /* {{{ */
178 struct stat statbuf = { 0 };
181 status = stat (h->filename, &statbuf);
185 if (h->mtime >= statbuf.st_mtime)
188 status = fbh_read_file (h);
190 h->mtime = statbuf.st_mtime;
193 } /* }}} int fbh_check_file */
198 fbhash_t *fbh_create (const char *file) /* {{{ */
206 h = calloc (1, sizeof (*h));
210 h->filename = strdup (file);
211 if (h->filename == NULL)
218 pthread_mutex_init (&h->lock, /* attr = */ NULL);
220 status = fbh_check_file (h);
229 } /* }}} fbhash_t *fbh_create */
231 void fbh_destroy (fbhash_t *h) /* {{{ */
236 pthread_mutex_destroy (&h->lock);
238 fbh_free_tree (h->tree);
239 } /* }}} void fbh_destroy */
241 char *fbh_get (fbhash_t *h, const char *key) /* {{{ */
247 if ((h == NULL) || (key == NULL))
253 pthread_mutex_lock (&h->lock);
255 /* TODO: Checking this every time may be a bit much..? */
258 status = c_avl_get (h->tree, key, (void *) &value);
261 assert (value != NULL);
262 value_copy = strdup (value);
265 pthread_mutex_unlock (&h->lock);
268 } /* }}} char *fbh_get */
270 /* vim: set sw=2 sts=2 et fdm=marker : */