1 /**
2 * collection4 - utils_search.c
3 * Copyright (C) 2010 Florian octo Forster
4 *
5 * This program is free software; you can redistribute it and/or
6 * modify it under the terms of the GNU Lesser General Public
7 * License as published by the Free Software Foundation; either
8 * version 2.1 of the License, or (at your option) any later version.
9 *
10 * This program is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 * Lesser General Public License for more details.
14 *
15 * You should have received a copy of the GNU Lesser General Public
16 * License along with this program; if not, write to the Free Software
17 * Foundation, Inc., 51 Franklin Street, Fifth Floor,
18 * Boston, MA 02110-1301 USA
19 *
20 * Authors:
21 * Florian octo Forster <ff at octo.it>
22 **/
24 #include <stdlib.h>
25 #include <stdio.h>
26 #include <string.h>
27 #include <ctype.h>
28 #include <errno.h>
30 #include "utils_search.h"
31 #include "graph_ident.h"
32 #include "graph_instance.h"
33 #include "utils_array.h"
35 #include <fcgiapp.h>
36 #include <fcgi_stdio.h>
38 struct search_info_s
39 {
40 char *host;
41 char *plugin;
42 char *plugin_instance;
43 char *type;
44 char *type_instance;
46 str_array_t *terms;
47 };
49 /*
50 * Private functions
51 */
52 static char *read_quoted_string (const char **buffer) /* {{{ */
53 {
54 const char *ptr = *buffer;
55 char *ret;
56 size_t ret_len;
58 if (ptr[0] != '"')
59 return (NULL);
60 ptr++;
62 ret_len = 0;
63 while ((*ptr != '"') && (*ptr != 0))
64 {
65 ret_len++;
67 if (*ptr == '\\')
68 ptr += 2;
69 else
70 ptr++;
71 }
73 if ((ret_len < 1) || (*ptr != '"'))
74 return (NULL);
76 ret = malloc (ret_len + 1);
77 if (ret == NULL)
78 return (NULL);
80 ptr = *buffer + 1;
81 ret_len = 0;
82 while ((*ptr != '"') && (*ptr != 0))
83 {
84 if (*ptr == '"')
85 break;
87 if (*ptr == '\\')
88 ptr++;
90 ret[ret_len] = *ptr;
92 ptr++;
93 ret_len++;
94 }
96 /* terminate string */
97 ret[ret_len] = 0;
99 /* "ptr" points to the '"' sign, so advance one more */
100 ptr++;
101 *buffer = ptr;
103 return (ret);
104 } /* }}} char *read_quoted_string */
106 static char *read_unquoted_word (const char **buffer) /* {{{ */
107 {
108 const char *ptr = *buffer;
109 char *ret;
110 size_t ret_len;
112 ret_len = 0;
113 while (!isspace ((int) ptr[ret_len]) && (ptr[ret_len] != 0))
114 ret_len++;
116 if (ret_len < 1)
117 return (NULL);
119 ret = malloc (ret_len + 1);
120 if (ret == NULL)
121 return (NULL);
123 memcpy (ret, ptr, ret_len);
124 ret[ret_len] = 0;
126 ptr += ret_len;
127 *buffer = ptr;
129 return (ret);
130 } /* }}} char *read_unquoted_word */
132 static char *next_token (const char **buffer) /* {{{ */
133 {
134 const char *ptr = *buffer;
135 char *ret;
137 while (isspace ((int) (*ptr)))
138 ptr++;
140 if (ptr[0] == 0)
141 return (NULL);
142 else if (ptr[0] == '"')
143 {
144 ret = read_quoted_string (&ptr);
145 if (ret != NULL)
146 {
147 *buffer = ptr;
148 return (ret);
149 }
150 }
152 ret = read_unquoted_word (&ptr);
153 if (ret != NULL)
154 *buffer = ptr;
156 return (ret);
157 } /* }}} char *next_token */
159 static int store_token_field (char **field, const char *token) /* {{{ */
160 {
161 char *copy;
163 if ((field == NULL) || (token == NULL))
164 return (EINVAL);
166 copy = strdup (token);
167 if (copy == NULL)
168 return (ENOMEM);
170 free (*field);
171 *field = copy;
173 return (0);
174 } /* }}} int store_token_field */
176 static int store_token (search_info_t *si, const char *token) /* {{{ */
177 {
178 if (strncmp ("host:", token, strlen ("host:")) == 0)
179 return (store_token_field (&si->host, token + strlen ("host:")));
180 else if (strncmp ("plugin:", token, strlen ("plugin:")) == 0)
181 return (store_token_field (&si->plugin, token + strlen ("plugin:")));
182 else if (strncmp ("plugin_instance:", token, strlen ("plugin_instance:")) == 0)
183 return (store_token_field (&si->plugin_instance, token + strlen ("plugin_instance:")));
184 else if (strncmp ("type:", token, strlen ("type:")) == 0)
185 return (store_token_field (&si->type, token + strlen ("type:")));
186 else if (strncmp ("type_instance:", token, strlen ("type_instance:")) == 0)
187 return (store_token_field (&si->type_instance, token + strlen ("type_instance:")));
189 return (array_append (si->terms, token));
190 } /* }}} int store_token */
192 /*
193 * Public functions
194 */
195 search_info_t *search_parse (const char *search) /* {{{ */
196 {
197 const char *ptr;
198 char *token;
199 search_info_t *si;
201 si = malloc (sizeof (*si));
202 if (si == NULL)
203 return (NULL);
204 memset (si, 0, sizeof (*si));
206 si->terms = array_create ();
207 if (si->terms == NULL)
208 {
209 free (si);
210 return (NULL);
211 }
213 ptr = search;
215 while ((token = next_token (&ptr)) != NULL)
216 {
217 store_token (si, token);
218 free (token);
219 }
221 return (si);
222 } /* }}} search_info_t *search_parse */
224 void search_destroy (search_info_t *si) /* {{{ */
225 {
226 if (si == NULL)
227 return;
229 free (si->host);
230 free (si->plugin);
231 free (si->plugin_instance);
232 free (si->type);
233 free (si->type_instance);
235 array_destroy (si->terms);
236 } /* }}} void search_destroy */
238 _Bool search_has_selector (search_info_t *si) /* {{{ */
239 {
240 if (si == NULL)
241 return (0);
243 if ((si->host != NULL)
244 || (si->plugin != NULL) || (si->plugin_instance != NULL)
245 || (si->type != NULL) || (si->type_instance != NULL))
246 return (1);
248 return (0);
249 } /* }}} _Bool search_has_selector */
251 graph_ident_t *search_to_ident (search_info_t *si) /* {{{ */
252 {
253 if (si == NULL)
254 return (NULL);
256 return (ident_create ((si->host == NULL) ? ANY_TOKEN : si->host,
257 (si->plugin == NULL) ? ANY_TOKEN : si->plugin,
258 (si->plugin_instance == NULL) ? ANY_TOKEN : si->plugin_instance,
259 (si->type == NULL) ? ANY_TOKEN : si->type,
260 (si->type_instance == NULL) ? ANY_TOKEN : si->type_instance));
261 } /* }}} graph_ident_t *search_to_ident */
263 search_info_t *search_from_ident (const graph_ident_t *ident) /* {{{ */
264 {
265 search_info_t *si;
267 if (ident == NULL)
268 return (NULL);
270 si = malloc (sizeof (*si));
271 if (si == NULL)
272 return (NULL);
273 memset (si, 0, sizeof (*si));
274 si->terms = NULL;
276 #define COPY_FIELD(f) do { \
277 const char *tmp = ident_get_##f (ident); \
278 if (tmp == NULL) \
279 si->f = NULL; \
280 else \
281 si->f = strdup (tmp); \
282 } while (0)
284 COPY_FIELD(host);
285 COPY_FIELD(plugin);
286 COPY_FIELD(plugin_instance);
287 COPY_FIELD(type);
288 COPY_FIELD(type_instance);
290 #undef COPY_FIELD
292 return (si);
293 } /* }}} search_info_t *search_from_ident */
295 _Bool search_graph_title_matches (search_info_t *si, /* {{{ */
296 const char *title)
297 {
298 char **argv;
299 int argc;
300 int i;
302 if ((si == NULL) || (title == NULL))
303 return (0);
305 if (si->terms == NULL)
306 return (1);
308 argc = array_argc (si->terms);
309 argv = array_argv (si->terms);
310 for (i = 0; i < argc; i++)
311 if (strstr (title, argv[i]) == NULL)
312 return (0);
314 return (1);
315 } /* }}} _Bool search_graph_title_matches */
317 _Bool search_graph_inst_matches (search_info_t *si, /* {{{ */
318 graph_config_t *cfg, graph_instance_t *inst,
319 const char *title)
320 {
321 char **argv;
322 int argc;
323 int i;
325 if ((si == NULL) || (cfg == NULL) || (inst == NULL))
326 return (0);
328 if ((si->host != NULL)
329 && !inst_matches_field (inst, GIF_HOST, si->host))
330 return (0);
331 else if ((si->plugin != NULL)
332 && !inst_matches_field (inst, GIF_PLUGIN, si->plugin))
333 return (0);
334 else if ((si->plugin_instance != NULL)
335 && !inst_matches_field (inst, GIF_PLUGIN_INSTANCE, si->plugin_instance))
336 return (0);
337 else if ((si->type != NULL)
338 && !inst_matches_field (inst, GIF_TYPE, si->type))
339 return (0);
340 else if ((si->type_instance != NULL)
341 && !inst_matches_field (inst, GIF_TYPE_INSTANCE, si->type_instance))
342 return (0);
344 if (si->terms == NULL)
345 return (1);
347 argc = array_argc (si->terms);
348 argv = array_argv (si->terms);
349 for (i = 0; i < argc; i++)
350 {
351 if (inst_matches_string (cfg, inst, argv[i]))
352 continue;
354 if ((title != NULL) && (strstr (title, argv[i]) != NULL))
355 continue;
357 return (0);
358 }
360 return (1);
361 } /* }}} _Bool search_graph_inst_matches */
363 /* vim: set sw=2 sts=2 et fdm=marker : */