jabberd2
2.2.16
|
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 }