//What the hell does this function do??
//it's called every time a key matching a listener changes
static void
notify(const char *k, const char *v, pdict_reason_t r, const char *pde_oldval,
void *a)
{
notify_arg_t *na = a;
id_pde_map_t *oldp;
id_pde_map_t *ipm;
id_pde_map_t cm;
const char *oldVal;
int res;
na->na_pdss->pdss_pd_lock(na->na_pdss->pdss_pd_lock_arg);
cm.ipm_id = na->na_listenid;
cm.ipm_key = k;
if (ptree_contains((void *)&cm, na->na_pdss->pdss_pending, ipmcmp, (void **)&ipm))
{
oldVal = ipm->ipm_val;
ipm->ipm_val = strdup(v);
free((void *)oldVal);
ipm->ipm_reason = r;
na->na_pdss->pdss_pd_unlock(na->na_pdss->pdss_pd_lock_arg);
return;
}
if (!(ipm = malloc(sizeof (*ipm)))) {
pu_log(PUL_WARN, na->na_pdss->pdss_id, "some notifications dropped due to low-memory condition");
na->na_pdss->pdss_pd_unlock(na->na_pdss->pdss_pd_lock_arg);
return;
}
ipm->ipm_id = na->na_listenid;
ipm->ipm_key = strdup(k);
ipm->ipm_val = strdup(v);
ipm->ipm_reason = r;
if ((res = ptree_replace(ipm, &na->na_pdss->pdss_pending, ipmcmp, (void **)&oldp)) != 0 && oldp) {
pu_log(PUL_WARN, na->na_pdss->pdss_id, "some notifications dropped due to low-memory condition");
free((void *)oldp->ipm_key); oldp->ipm_key = NULL;
free((void *)oldp->ipm_val); oldp->ipm_val = NULL;
free((void *)oldp); oldp = NULL;
}
if (!res) {
free((void *)ipm->ipm_key); ipm->ipm_key = NULL;
free((void *)ipm->ipm_val); ipm->ipm_val = NULL;
free(ipm); ipm = NULL;
pu_log(PUL_WARN, na->na_pdss->pdss_id, "some notifications dropped due to low-memory condition");
}
na->na_pdss->pdss_pd_unlock(na->na_pdss->pdss_pd_lock_arg);
}
void
pds_process_line(pds_session_t *pdss, char *line)
{
regmatch_t pmatch[9];
char *cmdtag = NULL;
char *cmdstr = NULL;
char *key = NULL;
char *val = NULL;
char *buf = NULL;
char *listenid = NULL;
int reportperiod;
int forsession;
int res;
//DPRINT(line);
if ((res = regexec(&setex, line, 7, pmatch, 0)) == 0) {
getmatchsub(line, &cmdtag, pmatch, 1);
if (cmdtag && strcmp(cmdtag, "unknown") == 0) {
result(pdss, cmdtag,
"304 unknown is a reserved command tag");
free(cmdtag); cmdtag = NULL;
return;
}
if (!getmatchsub(line, &key, pmatch, 2) || !key)
goto fail;
if (!getmatchsub(line, &val, pmatch, 5) &&
!getmatchsub(line, &val, pmatch, 4))
goto fail;
if (!val)
goto fail;
if ((forsession = getmatchsub(line, NULL, pmatch, 6)) > 0) {
/* mark for expiration, but make sure it's only added once! */
if(plist_walk(pdss->pdss_expire, is_key_in_list, key))
{
char *e;
if (!(e = strdup(key)))
goto fail;
plist_add(e, NULL, &pdss->pdss_expire);
}
}
pdss->pdss_pd_lock(pdss->pdss_pd_lock_arg);
res = pdict_add(pdss->pdss_pd, key, val, NULL);
pdss->pdss_pd_unlock(pdss->pdss_pd_lock_arg);
if (!res)
result(pdss, cmdtag, "304 set failed");
else
result(pdss, cmdtag, "200 set successful");
if (cmdtag) {
free(cmdtag); cmdtag = NULL;
}
free(key); key = NULL;
free(val); val = NULL;
return;
}
if ((res = regexec(&listenex, line, 6, pmatch, 0)) == 0) {
notify_arg_t *nap;
notify_arg_t na;
getmatchsub(line, &cmdtag, pmatch, 1);
if (cmdtag && strcmp(cmdtag, "unknown") == 0) {
result(pdss, cmdtag,
"304 unknown is a reserved command tag");
free(cmdtag); cmdtag = NULL;
return;
}
if ((!getmatchsub(line, &buf, pmatch, 4) && !getmatchsub(line, &buf, pmatch, 3)) || !buf)
goto fail;
getmatchsub(line, &listenid, pmatch, 5);
na.na_pdss = pdss;
na.na_listenid = listenid;
pthread_mutex_lock(&pdss->pdss_lock);
pdss->pdss_pd_lock(pdss->pdss_pd_lock_arg);
if (!ptree_contains(&na, pdss->pdss_notify_args, nacmp, (void **)&nap)) {
if (!(nap = malloc(sizeof (*nap))))
{
pu_log(PUL_WARN, pdss->pdss_id, "insufficient memory");
pdss->pdss_pd_unlock(pdss->pdss_pd_lock_arg);
pthread_mutex_unlock(&pdss->pdss_lock);
goto fail;
}
nap->na_pdss = pdss;
nap->na_listenid = listenid;
nap->na_pd_ids = NULL;
if (!ptree_replace(nap, &pdss->pdss_notify_args, nacmp, NULL)) {
free(nap); nap = NULL;
pu_log(PUL_WARN, pdss->pdss_id, "insufficient memory");
pdss->pdss_pd_unlock(pdss->pdss_pd_lock_arg);
pthread_mutex_unlock(&pdss->pdss_lock);
free(listenid); listenid = NULL;
free(buf); buf = NULL;
goto fail;
}
}
if (!(res = pdict_add_persistent_change_listener(pdss->pdss_pd, buf, notify, (void *)nap))) {
if (!nap->na_pd_ids) {
ptree_remove(nap, &pdss->pdss_notify_args, nacmp, NULL);
free((void *)nap->na_listenid); nap->na_listenid = NULL;
free(nap); nap = NULL;
}
result(pdss, cmdtag, "303 listen not established--bad pattern?");
} else {
if (!plist_add((void *)res, NULL, &nap->na_pd_ids)) {
pdict_remove_persistent_change_listener(pdss->pdss_pd, res);
if (!nap->na_pd_ids) {
ptree_remove(nap, &pdss->pdss_notify_args, nacmp, NULL);
free((void *)nap->na_listenid); nap->na_listenid = NULL;
free(nap); nap = NULL;
}
free(buf); buf = NULL;
if (cmdtag) {
free(cmdtag); cmdtag = NULL;
}
pdss->pdss_pd_unlock(pdss->pdss_pd_lock_arg);
pthread_mutex_unlock(&pdss->pdss_lock);
goto fail;
} else {
result(pdss, cmdtag,"200 listening, id %s",
listenid);
}
}
pdss->pdss_pd_unlock(pdss->pdss_pd_lock_arg);
pthread_mutex_unlock(&pdss->pdss_lock);
free(buf); buf = NULL;
if (cmdtag) {
free(cmdtag); cmdtag = NULL;
}
return;
}
if ((res = regexec(&reportex, line, 5, pmatch, 0)) == 0) {
getmatchsub(line, &cmdtag, pmatch, 1);
if (cmdtag && strcmp(cmdtag, "unknown") == 0) {
result(pdss, cmdtag,
"304 unknown is a reserved command tag");
free(cmdtag); cmdtag = NULL;
return;
}
if (!getmatchsub(line, &buf, pmatch, 2) || !buf)
goto fail;
if ((reportperiod = atoi(buf)) < 0 || reportperiod > 10000) {
result(pdss, cmdtag, "301 invalid report/wait period"
"--specify milliseconds or 0 for off (max 10s)");
if (cmdtag) {
free(cmdtag); cmdtag = NULL;
}
free(buf); buf = NULL;
return;
}
free(buf); buf = NULL;
getmatchsub(line, &buf, pmatch, 3);
set_report_period(pdss, cmdtag, buf, reportperiod);
if (cmdtag) {
free(cmdtag); cmdtag = NULL;
}
if (buf) {
free(buf); buf = NULL;
}
return;
}
if ((res = regexec(&waitex, line, 5, pmatch, 0)) == 0) {
getmatchsub(line, &cmdtag, pmatch, 1);
if (cmdtag && strcmp(cmdtag, "unknown") == 0) {
result(pdss, cmdtag,
"304 unknown is a reserved command tag");
free(cmdtag); cmdtag = NULL;
return;
}
if (!getmatchsub(line, &buf, pmatch, 2) || !buf)
goto fail;
if (atoi(buf) < 0 || atoi(buf) > 10000) {
result(pdss, cmdtag, "301 invalid wait period"
"--specify milliseconds (max 10s)");
if (cmdtag) {
free(cmdtag); cmdtag = NULL;
}
return;
}
usleep(atoi(buf) * 1000);
result(pdss, cmdtag, "200 nothin' doin'");
if (cmdtag) {
free(cmdtag); cmdtag = NULL;
}
return;
}
if ((res = regexec(&flushex, line, 5, pmatch, 0)) == 0) {
getmatchsub(line, &cmdtag, pmatch, 1);
if (cmdtag && strcmp(cmdtag, "unknown") == 0) {
result(pdss, cmdtag,
"304 unknown is a reserved command tag");
free(cmdtag); cmdtag = NULL;
return;
}
pthread_mutex_lock(&pdss->pdss_lock);
pdss->pdss_pd_lock(pdss->pdss_pd_lock_arg);
_flush(pdss);
pdss->pdss_pd_unlock(pdss->pdss_pd_lock_arg);
pthread_mutex_unlock(&pdss->pdss_lock);
result(pdss, cmdtag, "200 glug glug");
if (cmdtag) {
free(cmdtag); cmdtag = NULL;
}
return;
}
if ((res = regexec(&walkex, line, 5, pmatch, 0)) == 0) {
wa_t wa;
getmatchsub(line, &cmdtag, pmatch, 1);
if (cmdtag && strcmp(cmdtag, "unknown") == 0) {
result(pdss, cmdtag,
"304 unknown is a reserved command tag");
free(cmdtag); cmdtag = NULL;
return;
}
if (!getmatchsub(line, &cmdstr, pmatch, 2) || !cmdstr)
goto fail;
if (!getmatchsub(line, &buf, pmatch, 3) || !buf) {
free(cmdstr); cmdstr = NULL;
goto fail;
}
if ((regcomp(&wa.wa_regex, buf, REG_EXTENDED)) != 0) {
free(cmdstr); cmdstr = NULL;
free(buf); buf = NULL;
result(pdss, cmdtag, "305 expression error");
if (cmdtag) {
free(cmdtag); cmdtag = NULL;
}
return;
}
wa.wa_pdss = pdss;
wa.wa_cmdtag = cmdtag;
wa.wa_l = NULL;
pdss->pdss_pd_lock(pdss->pdss_pd_lock_arg);
if (!pdict_walk(pdss->pdss_pd, add_to_wa_list, &wa)) {
int e = errno;
pu_log(PUL_WARN, pdss->pdss_id, "temporary failure: %s", strerror(e));
result(pdss, cmdtag, "300 temporary failure: %s",
strerror(e));
} else {
if (strcmp(cmdstr, "remove") == 0)
plist_walk(wa.wa_l, remove_wa_list, &wa);
else
plist_walk(wa.wa_l, print_wa_list, &wa);
result(pdss, cmdtag, "200 done");
}
plist_clear(&wa.wa_l);
pdss->pdss_pd_unlock(pdss->pdss_pd_lock_arg);
regfree(&wa.wa_regex);
if (cmdtag) {
free(cmdtag); cmdtag = NULL;
}
free(cmdstr); cmdstr = NULL;
free(buf); buf = NULL;
return;
}
if ((res = regexec(&ignoreex, line, 5, pmatch, 0)) == 0) {
notify_arg_t *nap;
notify_arg_t na;
void *arg[2];
int n;
getmatchsub(line, &cmdtag, pmatch, 1);
if (cmdtag && strcmp(cmdtag, "unknown") == 0) {
result(pdss, cmdtag,
"304 unknown is a reserved command tag");
if (cmdtag) {
free(cmdtag); cmdtag = NULL;
}
return;
}
if (!getmatchsub(line, &listenid, pmatch, 2) || !listenid)
goto fail;
pthread_mutex_lock(&pdss->pdss_lock);
pdss->pdss_pd_lock(pdss->pdss_pd_lock_arg);
na.na_pdss = pdss;
na.na_listenid = listenid;
if (!ptree_remove(&na, &pdss->pdss_notify_args, nacmp,
(void **)&nap)) {
pdss->pdss_pd_unlock(pdss->pdss_pd_lock_arg);
pthread_mutex_unlock(&pdss->pdss_lock);
result(pdss, cmdtag, "306 nonexistent key/id");
if (cmdtag) {
free(cmdtag); cmdtag = NULL;
}
return;
}
n = 0;
plist_walk(nap->na_pd_ids, _count, &n);
plist_walk(nap->na_pd_ids, remove_persistent_change_listener_cb,
pdss->pdss_pd);
arg[0] = (void *)nap->na_listenid;
arg[1] = &pdss->pdss_pending;
ptree_walk(pdss->pdss_pending, PTREE_POSTORDER, remove_pending_id, ipmcmp, arg);
free((void *)nap->na_listenid); nap->na_listenid = NULL;
plist_clear(&nap->na_pd_ids);
free(nap); nap = NULL;
assert(!ptree_contains(&na, pdss->pdss_notify_args, nacmp, NULL));
pdss->pdss_pd_unlock(pdss->pdss_pd_lock_arg);
pthread_mutex_unlock(&pdss->pdss_lock);
result(pdss, cmdtag, "200 %d listener%s ignored", n, n > 1 ? "s"
: "");
if (cmdtag) {
free(cmdtag); cmdtag = NULL;
}
free(listenid);
return;
}
if ((res = regexec(&quitex, line, 5, pmatch, 0)) == 0) {
getmatchsub(line, &cmdtag, pmatch, 1);
result(pdss, cmdtag, "200 goodbye");
pdss->pdss_should_close = 1;
pdss->pdss_close(pdss->pdss_wfd, NULL, 0);
if (cmdtag) {
free(cmdtag); cmdtag = NULL;
}
return;
}
if ((res = regexec(&getidex, line, 5, pmatch, 0)) == 0) {
getmatchsub(line, &cmdtag, pmatch, 1);
result(pdss, cmdtag, "200 %d", pdss->pdss_id);
if (cmdtag) {
free(cmdtag); cmdtag = NULL;
}
return;
}
if ((res = regexec(&okex, line, 5, pmatch, 0)) == 0) {
if (cmdtag) {
free(cmdtag); cmdtag = NULL;
}
return;
}
result(pdss, NULL, "400 input unrecognized: %s", line);
return;
fail:
result(pdss, cmdtag, "300 command failed: %s", strerror(errno));
if (cmdtag)
free (cmdtag);
}
/*
* Sets or resets an entry to the dictionary, returning 0 on failure.
*/
int
pdict_add(pdict_t *pd, const char *k, const char *v, const char **ovp)
{
pdict_ent_t *n;
pdict_ent_t n_copy;
const char *ov;
if (!(k = strdup(k)))
return 0;
if (!(v = strdup(v))) {
free((void *)k); k = NULL;
return 0;
}
memset(&n_copy, 0, sizeof(pdict_ent_t));
if (ptree_contains((void *)&k, pd->pd_ents, pdecmp, (void *)&n)) {
free((void *)k); k = NULL;
ov = n->pde_val;
n->pde_val = v;
if (ovp)
*ovp = ov;
else {
free((void *)ov); ov = NULL;
}
//We copy n so that it's safe if it gets removed during a callback
if(n->pde_listeners)
{
n_copy.pde_key = strdup(n->pde_key);
n_copy.pde_val = strdup(n->pde_val);
_pdict_ent_listeners_copy(n, &n_copy);
_pdict_ent_notify(&n_copy, PDR_VALUE_CHANGED, ov);
pdict_ent_remove_change_listeners(&n_copy);
free((char *)n_copy.pde_key);
free((char *)n_copy.pde_val);
}
return 1;
}
if (!(n = malloc(sizeof (*n)))) {
free((void *)k); k = NULL;
free((void *)v); v = NULL;
return (0);
}
memset(n, 0, sizeof (*n));
n->pde_key = k;
n->pde_val = v;
//Add dict listeners to the dict entry object
if (!_pdict_ent_add_persistent_change_listeners(pd, n)) {
free((void *)k); k = NULL;
free((void *)v); v = NULL;
free(n); n = NULL;
return (0);
}
//Add the dict entry to the dict (replacing if there is a matching key).
if (!ptree_replace(n, &pd->pd_ents, pdecmp, NULL)) {
pdict_ent_remove_change_listeners(n);
free((void *)k); k = NULL;
free((void *)v); v = NULL;
free(n); n = NULL;
return (0);
}
//notify the listeners
if(n->pde_listeners)
{
n_copy.pde_key = strdup(n->pde_key);
n_copy.pde_val = strdup(n->pde_val);
_pdict_ent_listeners_copy(n, &n_copy);
_pdict_ent_notify(&n_copy, PDR_ENTRY_ADDED, n_copy.pde_val);
pdict_ent_remove_change_listeners(&n_copy);
free((char *)n_copy.pde_key);
free((char *)n_copy.pde_val);
}
if (ovp)
*ovp = NULL;
return 1;
}
/*
* Return whether a given key is in the dictionary. If the given
* entry pointer is non-NULL, set it to the entry.
*/
static int
_pdict_ent_lookup(pdict_t *pd, const char *k, pdict_ent_t **e)
{
return ptree_contains((void *)&k, pd->pd_ents, pdecmp, (void **)e);
}