75b551d0df88cdbb91d57be3774600690e9ee5f0
[rxpd] / src / rxpd_file.c
1 /*
2     rxpd_file.c - regex policy daemon
3
4   Copyright (C)
5     2007,               Christian Thaeter <ct@pipapo.org>
6
7   This program is free software; you can redistribute it and/or
8   modify it under the terms of the GNU General Public License as
9   published by the Free Software Foundation; either version 2 of the
10   License, or (at your option) any later version.
11
12   This program is distributed in the hope that it will be useful,
13   but WITHOUT ANY WARRANTY; without even the implied warranty of
14   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15   GNU General Public License for more details.
16
17   You should have received a copy of the GNU General Public License
18   along with this program; if not, write to the Free Software
19   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
20 */
21
22 #include "rxpd.h"
23
24 struct rxpd_file*
25 rxpd_file_new (struct rxpd_base* base, const char* filename)
26 {
27   char buf[4096];
28   struct rxpd_file* self = NULL;
29
30   if (!filename ||
31       strcspn(filename, RXPD_FILE_ILG_CHARS) != strlen (filename) ||
32       strlen (filename) + strlen (base->basedir) > 4095)
33     {
34       rxpd_log (base, LOG_ERR, "illegal filename: '%s'\n", filename?filename:"");
35       return NULL;
36     }
37
38   strcpy (buf, base->basedir);
39   strcat (buf, filename);
40   filename = rxpd_strdup (buf);
41
42   self = rxpd_malloc (sizeof (struct rxpd_file));
43   self->filename = filename;
44   self->base = base;
45   const char* basename = strrchr (filename, '/');
46   if (basename)
47     ++basename;
48   else
49     basename = filename;
50   psplay_init (&self->node, basename);
51   llist_init (&self->rules);
52
53   psplay_insert (&base->files, &self->node);
54
55   rxpd_log (base, LOG_INFO, "new file: '%s'\n", filename);
56   return self;
57 }
58
59 void
60 rxpd_file_delete (PSplay f)
61 {
62   if (f)
63     {
64       struct rxpd_file* file = (struct rxpd_file*)f;
65       LLIST_WHILE_HEAD (&file->rules, n)
66         {
67           struct rxpd_rule* node = (struct rxpd_rule*)n;
68           rxpd_rule_delete (node);
69         }
70       free ((void*)file->filename);
71       free (f);
72     }
73 }
74
75 int
76 rxpd_file_load (struct rxpd_file* self)
77 {
78   FILE* f = fopen (self->filename, "r");
79   // TODO error handling
80   if (f)
81     {
82       /* First purge old rules */
83       LLIST_WHILE_HEAD (&self->rules, n)
84         {
85           struct rxpd_rule* node = (struct rxpd_rule*)n;
86           rxpd_rule_delete (node);
87         }
88
89       // TODO test excess line length = error
90       char buf[4096];
91
92       rxpd_log (self->base, LOG_NOTICE, "loading '%s'\n", self->filename);
93
94       while (fgets (buf, 4096, f))
95         {
96           size_t last = strlen(buf);
97           if (buf[last-1] == '\n')
98             buf[last-1] = '\0';
99
100           struct rxpd_rule* rule;
101           rule = rxpd_rule_new (buf);
102           if (!rule)
103             abort();
104
105           rxpd_log (self->base, LOG_DEBUG, "new rule '%s'\n", rule->string);
106
107           llist_insert_tail (&self->rules, &rule->node);
108         }
109
110       fclose (f);
111       return 1;
112     }
113   else
114     {
115       rxpd_log (self->base, LOG_ERR, "failed loading '%s'\n", self->filename);
116       return 0;
117     }
118 }
119
120 int
121 rxpd_file_save (struct rxpd_file* self)
122 {
123   FILE* f = fopen (self->filename, "w");
124   // TODO error handling
125   if (f)
126     {
127       LLIST_FOREACH (&self->rules, n)
128         {
129           struct rxpd_rule* node = (struct rxpd_rule*)n;
130           if (node->atime != (time_t)-1)
131             fprintf (f, "%ld:%s\n", node->atime, node->string);
132           else if (*node->string != '#')
133             fprintf (f, ":%s\n", node->string);
134           else
135             fprintf (f, "%s\n", node->string);
136         }
137
138       fclose (f);
139       rxpd_log (self->base, LOG_NOTICE, "saved '%s'\n", self->filename);
140       return 1;
141     }
142   else
143     {
144       rxpd_log (self->base, LOG_ERR, "failed saving '%s'\n", self->filename);
145       return 0;
146     }
147 }
148
149 int
150 rxpd_file_cmp (const void* A, const void* B)
151 {
152   return strcmp (A, B);
153 }
154
155