1 /*
2  * jabberd - Jabber Open Source Server
3  * Copyright (c) 2002-2003 Jeremie Miller, Thomas Muldowney,
4  *                         Ryan Eatmon, Robert Norris
5  *
6  * This program is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation; either version 2 of the License, or
9  * (at your option) any later version.
10  *
11  * This program is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program; if not, write to the Free Software
18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA02111-1307USA
19  */
20 
21 #include "sm.h"
22 
23 /** @file sm/mod_vacation.c
24   * @brief vacation messages
25   * @author Robert Norris
26   * $Date: 2005/08/17 07:48:28 $
27   * $Revision: 1.13 $
28   */
29 
30 #define uri_VACATION    "http://jabber.org/protocol/vacation"
31 static int ns_VACATION = 0;
32 
33 typedef struct _vacation_st {
34     time_t  start;
35     time_t  end;
36     char    *msg;
37 } *vacation_t;
38 
_vacation_in_sess(mod_instance_t mi,sess_t sess,pkt_t pkt)39 static mod_ret_t _vacation_in_sess(mod_instance_t mi, sess_t sess, pkt_t pkt) {
40     module_t mod = mi->mod;
41     vacation_t v = sess->user->module_data[mod->index];
42     int ns, start, end, msg;
43     char dt[30];
44     pkt_t res;
45     os_t os;
46     os_object_t o;
47 
48     /* we only want to play with vacation iq packets */
49     if((pkt->type != pkt_IQ && pkt->type != pkt_IQ_SET) || pkt->ns != ns_VACATION)
50         return mod_PASS;
51 
52     /* if it has a to, throw it out */
53     if(pkt->to != NULL)
54         return -stanza_err_BAD_REQUEST;
55 
56     /* get */
57     if(pkt->type == pkt_IQ) {
58         if(v->msg == NULL) {
59             res = pkt_create(mod->mm->sm, "iq", "result", NULL, NULL);
60             pkt_id(pkt, res);
61             pkt_free(pkt);
62 
63             pkt_sess(res, sess);
64 
65             return mod_HANDLED;
66         }
67 
68         ns = nad_find_scoped_namespace(pkt->nad, uri_VACATION, NULL);
69 
70         if(v->start != 0) {
71             datetime_out(v->start, dt_DATETIME, dt, 30);
72             nad_insert_elem(pkt->nad, 2, ns, "start", dt);
73         } else
74             nad_insert_elem(pkt->nad, 2, ns, "start", NULL);
75 
76         if(v->end != 0) {
77             datetime_out(v->end, dt_DATETIME, dt, 30);
78             nad_insert_elem(pkt->nad, 2, ns, "end", dt);
79         } else
80             nad_insert_elem(pkt->nad, 2, ns, "end", NULL);
81 
82         nad_insert_elem(pkt->nad, 2, ns, "message", v->msg);
83 
84         pkt_tofrom(pkt);
85         nad_set_attr(pkt->nad, 1, -1, "type", "result", 6);
86 
87         pkt_sess(pkt, sess);
88 
89         return mod_HANDLED;
90     }
91 
92     /* set */
93     ns = nad_find_scoped_namespace(pkt->nad, uri_VACATION, NULL);
94 
95     start = nad_find_elem(pkt->nad, 2, ns, "start", 1);
96     end = nad_find_elem(pkt->nad, 2, ns, "end", 1);
97     msg = nad_find_elem(pkt->nad, 2, ns, "message", 1);
98 
99     if(start < 0 || end < 0 || msg < 0) {
100         /* forget */
101         if(v->msg != NULL) {
102             free(v->msg);
103             v->msg = NULL;
104         }
105         v->start = 0;
106         v->end = 0;
107 
108         storage_delete(mi->sm->st, "vacation-settings", jid_user(sess->jid), NULL);
109 
110         res = pkt_create(mod->mm->sm, "iq", "result", NULL, NULL);
111         pkt_id(pkt, res);
112         pkt_free(pkt);
113 
114         pkt_sess(res, sess);
115 
116         return mod_HANDLED;
117     }
118 
119     if(NAD_CDATA_L(pkt->nad, start) > 0) {
120         strncpy(dt, NAD_CDATA(pkt->nad, start), (30 < NAD_CDATA_L(pkt->nad, start) ? 30 : NAD_CDATA_L(pkt->nad, start)));
121         v->start = datetime_in(dt);
122     } else
123         v->start = 0;
124 
125     if(NAD_CDATA_L(pkt->nad, end) > 0) {
126         strncpy(dt, NAD_CDATA(pkt->nad, end), (30 < NAD_CDATA_L(pkt->nad, end) ? 30 : NAD_CDATA_L(pkt->nad, end)));
127         v->end = datetime_in(dt);
128     } else
129         v->end = 0;
130 
131     v->msg = (char *) malloc(sizeof(char) * (NAD_CDATA_L(pkt->nad, msg) + 1));
132     strncpy(v->msg, NAD_CDATA(pkt->nad, msg), NAD_CDATA_L(pkt->nad, msg));
133     v->msg[NAD_CDATA_L(pkt->nad, msg)] = '\0';
134 
135     os = os_new();
136     o = os_object_new(os);
137     os_object_put(o, "start", &v->start, os_type_INTEGER);
138     os_object_put(o, "end", &v->end, os_type_INTEGER);
139     os_object_put(o, "message", v->msg, os_type_STRING);
140 
141     if(storage_replace(mod->mm->sm->st, "vacation-settings", jid_user(sess->user->jid), NULL, os) != st_SUCCESS) {
142         free(v->msg);
143         v->msg = NULL;
144         v->start = 0;
145         v->end = 0;
146         return -stanza_err_INTERNAL_SERVER_ERROR;
147     }
148 
149     res = pkt_create(mod->mm->sm, "iq", "result", NULL, NULL);
150     pkt_id(pkt, res);
151     pkt_free(pkt);
152 
153     pkt_sess(res, sess);
154 
155     return mod_HANDLED;
156 }
157 
_vacation_pkt_user(mod_instance_t mi,user_t user,pkt_t pkt)158 static mod_ret_t _vacation_pkt_user(mod_instance_t mi, user_t user, pkt_t pkt) {
159     module_t mod = mi->mod;
160     vacation_t v = user->module_data[mod->index];
161     time_t t;
162     pkt_t res;
163 
164     if(v->msg == NULL)
165         return mod_PASS;
166 
167     /* only want messages, and only if they're offline */
168     if(!(pkt->type & pkt_MESSAGE) || user->top != NULL)
169         return mod_PASS;
170 
171     /* reply only to real, human users - they always have full JIDs in 'from' */
172     jid_expand(pkt->from);
173     if(pkt->from->node[0] == '\0' || pkt->from->resource[0] == '\0') {
174         pkt_free(pkt);
175         return mod_HANDLED;
176     }
177 
178     t = time(NULL);
179 
180     if(v->start < t && (t < v->end || v->end == 0)) {
181         res = pkt_create(mod->mm->sm, "message", NULL, jid_full(pkt->from), user->jid->domain);
182         nad_insert_elem(res->nad, 1, NAD_ENS(res->nad, 1), "subject", "Automated reply");
183         nad_insert_elem(res->nad, 1, NAD_ENS(res->nad, 1), "body", v->msg);
184         pkt_router(res);
185 
186         /* !!! remember that we sent this */
187     }
188 
189     return mod_PASS;
190 }
191 
_vacation_user_free(vacation_t v)192 static void _vacation_user_free(vacation_t v) {
193     if(v->msg != NULL)
194         free(v->msg);
195     free(v);
196 }
197 
_vacation_user_load(mod_instance_t mi,user_t user)198 static int _vacation_user_load(mod_instance_t mi, user_t user) {
199     module_t mod = mi->mod;
200     vacation_t v;
201     os_t os;
202     os_object_t o;
203 
204     v = (vacation_t) calloc(1, sizeof(struct _vacation_st));
205     user->module_data[mod->index] = v;
206 
207     if(storage_get(mod->mm->sm->st, "vacation-settings", jid_user(user->jid), NULL, &os) == st_SUCCESS) {
208         if(os_iter_first(os)) {
209             o = os_iter_object(os);
210 
211             if(os_object_get_time(os, o, "start", &v->start) &&
212                os_object_get_time(os, o, "end", &v->end) &&
213                os_object_get_str(os, o, "message", &v->msg))
214                 v->msg = strdup(v->msg);
215             else {
216                 v->start = 0;
217                 v->end = 0;
218                 v->msg = NULL;
219             }
220         }
221 
222         os_free(os);
223     }
224 
225     pool_cleanup(user->p, (void (*))(void *) _vacation_user_free, v);
226 
227     return 0;
228 }
229 
_vacation_user_delete(mod_instance_t mi,jid_t jid)230 static void _vacation_user_delete(mod_instance_t mi, jid_t jid) {
231     log_debug(ZONE, "deleting vacations settings for %s", jid_user(jid));
232 
233     storage_delete(mi->sm->st, "vacation-settings", jid_user(jid), NULL);
234 }
235 
_vacation_free(module_t mod)236 static void _vacation_free(module_t mod) {
237     sm_unregister_ns(mod->mm->sm, uri_VACATION);
238     feature_unregister(mod->mm->sm, uri_VACATION);
239 }
240 
module_init(mod_instance_t mi,const char * arg)241 DLLEXPORT int module_init(mod_instance_t mi, const char *arg) {
242     module_t mod = mi->mod;
243 
244     if(mod->init) return 0;
245 
246     mod->in_sess = _vacation_in_sess;
247     mod->pkt_user = _vacation_pkt_user;
248     mod->user_load = _vacation_user_load;
249     mod->user_delete = _vacation_user_delete;
250     mod->free = _vacation_free; /* mmm good! :) */
251 
252     ns_VACATION = sm_register_ns(mod->mm->sm, uri_VACATION);
253     feature_register(mod->mm->sm, uri_VACATION);
254 
255     return 0;
256 }
257