jabberd2  2.2.16
sm/mod_vacation.c
Go to the documentation of this file.
00001 /*
00002  * jabberd - Jabber Open Source Server
00003  * Copyright (c) 2002-2003 Jeremie Miller, Thomas Muldowney,
00004  *                         Ryan Eatmon, Robert Norris
00005  *
00006  * This program is free software; you can redistribute it and/or modify
00007  * it under the terms of the GNU General Public License as published by
00008  * the Free Software Foundation; either version 2 of the License, or
00009  * (at your option) any later version.
00010  *
00011  * This program is distributed in the hope that it will be useful,
00012  * but WITHOUT ANY WARRANTY; without even the implied warranty of
00013  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.See the
00014  * GNU General Public License for more details.
00015  *
00016  * You should have received a copy of the GNU General Public License
00017  * along with this program; if not, write to the Free Software
00018  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA02111-1307USA
00019  */
00020 
00021 #include "sm.h"
00022 
00030 #define uri_VACATION    "http://jabber.org/protocol/vacation"
00031 static int ns_VACATION = 0;
00032 
00033 typedef struct _vacation_st {
00034     time_t  start;
00035     time_t  end;
00036     char    *msg;
00037 } *vacation_t;
00038 
00039 static mod_ret_t _vacation_in_sess(mod_instance_t mi, sess_t sess, pkt_t pkt) {
00040     module_t mod = mi->mod;
00041     vacation_t v = sess->user->module_data[mod->index];
00042     int ns, start, end, msg;
00043     char dt[30];
00044     pkt_t res;
00045     os_t os;
00046     os_object_t o;
00047 
00048     /* we only want to play with vacation iq packets */
00049     if((pkt->type != pkt_IQ && pkt->type != pkt_IQ_SET) || pkt->ns != ns_VACATION)
00050         return mod_PASS;
00051 
00052     /* if it has a to, throw it out */
00053     if(pkt->to != NULL)
00054         return -stanza_err_BAD_REQUEST;
00055 
00056     /* get */
00057     if(pkt->type == pkt_IQ) {
00058         if(v->msg == NULL) {
00059             res = pkt_create(mod->mm->sm, "iq", "result", NULL, NULL);
00060             pkt_id(pkt, res);
00061             pkt_free(pkt);
00062 
00063             pkt_sess(res, sess);
00064 
00065             return mod_HANDLED;
00066         }
00067 
00068         ns = nad_find_scoped_namespace(pkt->nad, uri_VACATION, NULL);
00069 
00070         if(v->start != 0) {
00071             datetime_out(v->start, dt_DATETIME, dt, 30);
00072             nad_insert_elem(pkt->nad, 2, ns, "start", dt);
00073         } else
00074             nad_insert_elem(pkt->nad, 2, ns, "start", NULL);
00075 
00076         if(v->end != 0) {
00077             datetime_out(v->end, dt_DATETIME, dt, 30);
00078             nad_insert_elem(pkt->nad, 2, ns, "end", dt);
00079         } else
00080             nad_insert_elem(pkt->nad, 2, ns, "end", NULL);
00081 
00082         nad_insert_elem(pkt->nad, 2, ns, "message", v->msg);
00083 
00084         pkt_tofrom(pkt);
00085         nad_set_attr(pkt->nad, 1, -1, "type", "result", 6);
00086 
00087         pkt_sess(pkt, sess);
00088 
00089         return mod_HANDLED;
00090     }
00091 
00092     /* set */
00093     ns = nad_find_scoped_namespace(pkt->nad, uri_VACATION, NULL);
00094 
00095     start = nad_find_elem(pkt->nad, 2, ns, "start", 1);
00096     end = nad_find_elem(pkt->nad, 2, ns, "end", 1);
00097     msg = nad_find_elem(pkt->nad, 2, ns, "message", 1);
00098 
00099     if(start < 0 || end < 0 || msg < 0) {
00100         /* forget */
00101         if(v->msg != NULL) {
00102             free(v->msg);
00103             v->msg = NULL;
00104         }
00105         v->start = 0;
00106         v->end = 0;
00107 
00108         storage_delete(mi->sm->st, "vacation-settings", jid_user(sess->jid), NULL);
00109 
00110         res = pkt_create(mod->mm->sm, "iq", "result", NULL, NULL);
00111         pkt_id(pkt, res);
00112         pkt_free(pkt);
00113 
00114         pkt_sess(res, sess);
00115 
00116         return mod_HANDLED;
00117     }
00118 
00119     if(NAD_CDATA_L(pkt->nad, start) > 0) {
00120         strncpy(dt, NAD_CDATA(pkt->nad, start), (30 < NAD_CDATA_L(pkt->nad, start) ? 30 : NAD_CDATA_L(pkt->nad, start)));
00121         v->start = datetime_in(dt);
00122     } else
00123         v->start = 0;
00124 
00125     if(NAD_CDATA_L(pkt->nad, end) > 0) {
00126         strncpy(dt, NAD_CDATA(pkt->nad, end), (30 < NAD_CDATA_L(pkt->nad, end) ? 30 : NAD_CDATA_L(pkt->nad, end)));
00127         v->end = datetime_in(dt);
00128     } else
00129         v->end = 0;
00130 
00131     v->msg = (char *) malloc(sizeof(char) * (NAD_CDATA_L(pkt->nad, msg) + 1));
00132     strncpy(v->msg, NAD_CDATA(pkt->nad, msg), NAD_CDATA_L(pkt->nad, msg));
00133     v->msg[NAD_CDATA_L(pkt->nad, msg)] = '\0';
00134 
00135     os = os_new();
00136     o = os_object_new(os);
00137     os_object_put(o, "start", &v->start, os_type_INTEGER);
00138     os_object_put(o, "end", &v->end, os_type_INTEGER);
00139     os_object_put(o, "message", v->msg, os_type_STRING);
00140 
00141     if(storage_replace(mod->mm->sm->st, "vacation-settings", jid_user(sess->user->jid), NULL, os) != st_SUCCESS) {
00142         free(v->msg);
00143         v->msg = NULL;
00144         v->start = 0;
00145         v->end = 0;
00146         return -stanza_err_INTERNAL_SERVER_ERROR;
00147     }
00148     
00149     res = pkt_create(mod->mm->sm, "iq", "result", NULL, NULL);
00150     pkt_id(pkt, res);
00151     pkt_free(pkt);
00152 
00153     pkt_sess(res, sess);
00154 
00155     return mod_HANDLED;
00156 }
00157 
00158 static mod_ret_t _vacation_pkt_user(mod_instance_t mi, user_t user, pkt_t pkt) {
00159     module_t mod = mi->mod;
00160     vacation_t v = user->module_data[mod->index];
00161     time_t t;
00162     pkt_t res;
00163 
00164     if(v->msg == NULL)
00165         return mod_PASS;
00166 
00167     /* only want messages, and only if they're offline */
00168     if(!(pkt->type & pkt_MESSAGE) || user->top != NULL)
00169         return mod_PASS;
00170 
00171     /* reply only to real, human users - they always have full JIDs in 'from' */
00172     jid_expand(pkt->from);
00173     if(pkt->from->node[0] == '\0' || pkt->from->resource[0] == '\0') {
00174         pkt_free(pkt);
00175         return mod_HANDLED;
00176     }
00177 
00178     t = time(NULL);
00179 
00180     if(v->start < t && (t < v->end || v->end == 0)) {
00181         res = pkt_create(mod->mm->sm, "message", NULL, jid_full(pkt->from), user->jid->domain);
00182         nad_insert_elem(res->nad, 1, NAD_ENS(res->nad, 1), "subject", "Automated reply");
00183         nad_insert_elem(res->nad, 1, NAD_ENS(res->nad, 1), "body", v->msg);
00184         pkt_router(res);
00185 
00186         /* !!! remember that we sent this */
00187     }
00188 
00189     return mod_PASS;
00190 }
00191 
00192 static void _vacation_user_free(vacation_t v) {
00193     if(v->msg != NULL)
00194         free(v->msg);
00195     free(v);
00196 }
00197 
00198 static int _vacation_user_load(mod_instance_t mi, user_t user) {
00199     module_t mod = mi->mod;
00200     vacation_t v;
00201     os_t os;
00202     os_object_t o;
00203 
00204     v = (vacation_t) calloc(1, sizeof(struct _vacation_st));
00205     user->module_data[mod->index] = v;
00206 
00207     if(storage_get(mod->mm->sm->st, "vacation-settings", jid_user(user->jid), NULL, &os) == st_SUCCESS) {
00208         if(os_iter_first(os)) {
00209             o = os_iter_object(os);
00210 
00211             if(os_object_get_time(os, o, "start", &v->start) &&
00212                os_object_get_time(os, o, "end", &v->end) &&
00213                os_object_get_str(os, o, "message", &v->msg))
00214                 v->msg = strdup(v->msg);
00215             else {
00216                 v->start = 0;
00217                 v->end = 0;
00218                 v->msg = NULL;
00219             }
00220         }
00221 
00222         os_free(os);
00223     }
00224 
00225     pool_cleanup(user->p, (void (*))(void *) _vacation_user_free, v);
00226 
00227     return 0;
00228 }
00229 
00230 static void _vacation_user_delete(mod_instance_t mi, jid_t jid) {
00231     log_debug(ZONE, "deleting vacations settings for %s", jid_user(jid));
00232 
00233     storage_delete(mi->sm->st, "vacation-settings", jid_user(jid), NULL);
00234 }
00235 
00236 static void _vacation_free(module_t mod) {
00237     sm_unregister_ns(mod->mm->sm, uri_VACATION);
00238     feature_unregister(mod->mm->sm, uri_VACATION);
00239 }
00240 
00241 DLLEXPORT int module_init(mod_instance_t mi, char *arg) {
00242     module_t mod = mi->mod;
00243 
00244     if(mod->init) return 0;
00245 
00246     mod->in_sess = _vacation_in_sess;
00247     mod->pkt_user = _vacation_pkt_user;
00248     mod->user_load = _vacation_user_load;
00249     mod->user_delete = _vacation_user_delete;
00250     mod->free = _vacation_free; /* mmm good! :) */
00251 
00252     ns_VACATION = sm_register_ns(mod->mm->sm, uri_VACATION);
00253     feature_register(mod->mm->sm, uri_VACATION);
00254 
00255     return 0;
00256 }