static void on_error(struct CThostFtdcRspInfoField *rspinfo, int rid, int islast) { if (rspinfo && rspinfo->ErrorID != 0) xcb_log(XCB_LOG_ERROR, "Error occurred: errorid=%d", rspinfo->ErrorID); }
static void on_subscribe_market_data(struct ErrorRtnField *rspinfo) { if (rspinfo && rspinfo->ErrorID != 0) xcb_log(XCB_LOG_ERROR, "Error occurred: errorid=%d", rspinfo->ErrorID); }
static void on_subscribe_market_data(struct CThostFtdcSpecificInstrumentField *instrument, struct CThostFtdcRspInfoField *rspinfo, int rid, int islast) { if (rspinfo && rspinfo->ErrorID != 0) xcb_log(XCB_LOG_ERROR, "Error occurred: errorid=%d", rspinfo->ErrorID); }
static void on_front_disconnected(int reason) { xcb_log(XCB_LOG_NOTICE, "Front disconnected: reason=%d", reason); }
static int mm_impvbaw_exec(void *data, void *data2) { RAII_VAR(struct msg *, msg, (struct msg *)data, msg_decr); dstr *fields = NULL; int nfield = 0; time_t t; dstr spotname; double vol; NOT_USED(data2); fields = dstr_split_len(msg->data, strlen(msg->data), ",", 1, &nfield); /* FIXME */ if (nfield != 19) { xcb_log(XCB_LOG_WARNING, "Message '%s' garbled", msg->data); goto end; } t = (time_t)atoi(fields[1]); spotname = dstr_new(fields[11]); vol = atof(fields[5]); if (!isnan(vol)) { struct mm *mm; int flag = 0; table_lock(contracts); if ((mm = table_get_value(contracts, spotname)) == NULL) { if (NEW(mm)) { mm->min = mm->max = vol; flag = 1; table_insert(contracts, spotname, mm); } else xcb_log(XCB_LOG_WARNING, "Error allocating memory for mm"); } else { if (mm->min > vol) { mm->min = vol; flag = 1; } else if (mm->max < vol) { mm->max = vol; flag = 1; } dstr_free(spotname); } if (mm && flag) { struct tm lt; char datestr[64], res[512]; localtime_r(&t, <); lt.tm_hour = lt.tm_min = lt.tm_sec = 0; strftime(datestr, sizeof datestr, "%F %T", <); snprintf(res, sizeof res, "MM_IMPVBAW,%s.000,%s|%f,%f", datestr, fields[11], mm->min, mm->max); out2rmp(res); } table_unlock(contracts); } end: dstr_free_tokens(fields, nfield); return 0; }
static int impvbaw_exec(void *data, void *data2) { RAII_VAR(struct msg *, msg, (struct msg *)data, msg_decr); Quote *quote = (Quote *)msg->data; struct msgs *out = (struct msgs *)data2; dstr contract; float last; char *p; table_node_t node; contract = dstr_new(quote->thyquote.m_cHYDM); if (!strcmp(contract, "") || (fabs(quote->thyquote.m_dZXJ) <= 0.000001 && fabs(quote->thyquote.m_dMRJG1) <= 0.000001 && fabs(quote->thyquote.m_dMCJG1) <= 0.000001)) { xcb_log(XCB_LOG_WARNING, "Invalid quote: '%d,%d,%s,%.2f,%.2f,%.2f'", quote->thyquote.m_nTime, quote->m_nMSec, contract, quote->thyquote.m_dZXJ, quote->thyquote.m_dMRJG1, quote->thyquote.m_dMCJG1); goto end; } /* FIXME */ if (!strncasecmp(contract, "IO", 2) || !strncasecmp(contract, "HO", 2) || !strncasecmp(contract, "00", 2) || !strncasecmp(contract, "60", 2)) goto end; last = quote->thyquote.m_dZXJ; if ((p = strrchr(contract, 'C')) == NULL) p = strrchr(contract, 'P'); /* FIXME: option quote */ if (p && p != contract && p != contract + dstr_length(contract) - 1 && ((*(p - 1) == '-' && *(p + 1) == '-') || (isdigit(*(p - 1)) && isdigit(*(p + 1))))) { dstr spotname, type, strike; float spot; struct prices *prices; double expiry, vol, vol2, vol3; struct tm lt; char *res; spotname = *(p - 1) == '-' ? dstr_new_len(contract, p - contract - 1) : dstr_new_len(contract, p - contract); type = dstr_new_len(p, 1); strike = *(p + 1) == '-' ? dstr_new_len(p + 2, contract + dstr_length(contract) - p - 2) : dstr_new_len(p + 1, contract + dstr_length(contract) - p - 1); table_rwlock_rdlock(spots); if ((node = table_find(spots, spotname)) == NULL) { table_rwlock_unlock(spots); dstr_free(strike); dstr_free(type); dstr_free(spotname); goto end; } spot = table_node_float(node); table_rwlock_unlock(spots); if (fabs(spot) <= 0.000001) { xcb_log(XCB_LOG_WARNING, "The price of spot '%s' be zero", spotname); dstr_free(strike); dstr_free(type); dstr_free(spotname); goto end; } table_lock(optns); if ((node = table_find(optns, contract)) == NULL) { if (NEW(prices)) { prices->prelast = last; prices->prebid1 = quote->thyquote.m_dMRJG1; prices->preask1 = quote->thyquote.m_dMCJG1; prices->prespot = spot; table_insert(optns, dstr_new(contract), prices); } else xcb_log(XCB_LOG_WARNING, "Error allocating memory for prices"); } else { prices = (struct prices *)table_node_value(node); if (fabs(prices->prelast - last) <= 0.000001 && fabs(prices->prebid1 - quote->thyquote.m_dMRJG1) <= 0.000001 && fabs(prices->preask1 - quote->thyquote.m_dMCJG1) <= 0.000001 && fabs(prices->prespot - spot) <= 0.000001) { table_unlock(optns); dstr_free(strike); dstr_free(type); dstr_free(spotname); goto end; } else { prices->prelast = last; prices->prebid1 = quote->thyquote.m_dMRJG1; prices->preask1 = quote->thyquote.m_dMCJG1; prices->prespot = spot; } } table_unlock(optns); if ((node = table_find(expiries, contract))) expiry = table_node_double(node); else { char *month; struct timeval tv; int diff; xcb_log(XCB_LOG_WARNING, "No exact expiry date for '%s'", contract); month = spotname + dstr_length(spotname) - 2; gettimeofday(&tv, NULL); if ((diff = atoi(month) - localtime_r(&tv.tv_sec, <)->tm_mon - 1) < 0) diff += 12; if (diff == 0) diff = 1; expiry = diff / 12.0; } /* FIXME: last price */ if (fabs(last) <= 0.000001) vol = NAN; else vol = impv_baw(spot, atof(strike), r, r, expiry, last, !strcasecmp(type, "C") ? AMER_CALL : AMER_PUT); /* FIXME: bid price 1 */ if (fabs(quote->thyquote.m_dMRJG1) <= 0.000001) vol2 = NAN; else if (fabs(quote->thyquote.m_dMRJG1 - last) <= 0.000001) vol2 = vol; else vol2 = impv_baw(spot, atof(strike), r, r, expiry, quote->thyquote.m_dMRJG1, !strcasecmp(type, "C") ? AMER_CALL : AMER_PUT); /* FIXME: ask price 1 */ if (fabs(quote->thyquote.m_dMCJG1) <= 0.000001) vol3 = NAN; else if (fabs(quote->thyquote.m_dMCJG1 - last) <= 0.000001) vol3 = vol; else vol3 = impv_baw(spot, atof(strike), r, r, expiry, quote->thyquote.m_dMCJG1, !strcasecmp(type, "C") ? AMER_CALL : AMER_PUT); if ((res = ALLOC(512))) { time_t t = (time_t)quote->thyquote.m_nTime; char datestr[64]; strftime(datestr, sizeof datestr, "%F %T", localtime_r(&t, <)); snprintf(res, 512, "IMPVBAW,%s.%03d,%s|%.2f,%f,%.2f,%f,%.2f,%f,%.2f", datestr, quote->m_nMSec, contract, last, vol, quote->thyquote.m_dMRJG1, vol2, quote->thyquote.m_dMCJG1, vol3, spot); out2rmp(res); snprintf(res, 512, "IMPVBAW,%d,%d,%s,%.2f,%f,%.2f,%f,%.2f,%f,%.2f,%s,%s,%s,%f,%f,%d,0,0", quote->thyquote.m_nTime, quote->m_nMSec, contract, last, vol, quote->thyquote.m_dMRJG1, vol2, quote->thyquote.m_dMCJG1, vol3, spot, spotname, type, strike, r, expiry, quote->thyquote.m_nCJSL); if (out2msgs(res, out) == -1) FREE(res); } else xcb_log(XCB_LOG_WARNING, "Error allocating memory for result"); dstr_free(strike); dstr_free(type); dstr_free(spotname); /* future quote */ } else { table_rwlock_wrlock(spots); if ((node = table_find(spots, contract)) == NULL) { if ((node = table_insert_raw(spots, contract))) table_set_float(node, last); } else { table_set_float(node, last); dstr_free(contract); } table_rwlock_unlock(spots); return 0; } end: dstr_free(contract); return 0; }
static inline void load_config(void) { /* FIXME */ if ((cfg = config_load("/etc/xcb/vxo.conf"))) { char *cat = category_browse(cfg, NULL); struct variable *var; while (cat) { if (!strcasecmp(cat, "general")) { struct variable *var = variable_browse(cfg, cat); while (var) { if (!strcasecmp(var->name, "inmsg")) { if (strcmp(var->value, "")) inmsg = var->value; } else xcb_log(XCB_LOG_WARNING, "Unknown variable '%s' in " "category '%s' of vxo.conf", var->name, cat); var = var->next; } } else if (!strcasecmp(cat, "expiries")) { var = variable_browse(cfg, cat); while (var) { char *p; if ((p = strrchr(var->name, 'C')) == NULL) p = strrchr(var->name, 'P'); if (p && p != var->name && p != var->name + strlen(var->name) - 1 && ((*(p - 1) == '-' && *(p + 1) == '-') || (isdigit(*(p - 1)) && isdigit(*(p + 1))))) { dstr spotname, strike; struct pd *pd; struct scp *scp; spotname = *(p - 1) == '-' ? dstr_new_len(var->name, p - var->name - 1) : dstr_new_len(var->name, p - var->name); strike = *(p + 1) == '-' ? dstr_new_len(p + 2, var->name + strlen(var->name) - p - 2) : dstr_new_len(p + 1, var->name + strlen(var->name) - p - 1); if ((pd = table_get_value(spots, spotname)) == NULL) { if (NEW(scp)) { scp->strike = atof(strike); scp->cvol = scp->pvol = NAN; scp->cvol2 = scp->pvol2 = NAN; scp->cvol3 = scp->pvol3 = NAN; if (NEW(pd)) { pd->prevxo = pd->prevxo2 = pd->prevxo3 = NAN; pd->sep = *(p - 1) == '-' ? "-" : ""; pd->dlist = dlist_new(NULL, scpfree); dlist_insert_tail(pd->dlist, scp); table_insert(spots, spotname, pd); } else FREE(scp); } else dstr_free(spotname); } else { dlist_iter_t iter = dlist_iter_new(pd->dlist, DLIST_START_HEAD); dlist_node_t node; while ((node = dlist_next(iter))) { scp = (struct scp *)dlist_node_value(node); if (scp->strike > atof(strike) || fabs(scp->strike - atof(strike)) <= 0.000001) break; } dlist_iter_free(&iter); if (node == NULL || scp->strike > atof(strike)) { if (NEW(scp)) { scp->strike = atof(strike); scp->cvol = scp->pvol = NAN; scp->cvol2 = scp->pvol2 = NAN; scp->cvol3 = scp->pvol3 = NAN; } if (node == NULL) dlist_insert_tail(pd->dlist, scp); else dlist_insert(pd->dlist, node, scp, 0); } dstr_free(spotname); } dstr_free(strike); } var = var->next; } } cat = category_browse(cfg, cat); } } }
static int vxo_exec(void *data, void *data2) { RAII_VAR(struct msg *, msg, (struct msg *)data, msg_decr); struct msgs *out = (struct msgs *)data2; dstr *fields = NULL; int nfield = 0; double vol, vol2, vol3, spot, strike, r, expiry; int sec, msec; dstr spotname; char *type; struct pd *pd; struct scp *scp; fields = dstr_split_len(msg->data, strlen(msg->data), ",", 1, &nfield); /* FIXME */ if (nfield != 19) { xcb_log(XCB_LOG_WARNING, "Message '%s' garbled", msg->data); goto end; } vol = !strcasecmp(fields[5], "nan") ? NAN : atof(fields[5]); vol2 = !strcasecmp(fields[7], "nan") ? NAN : atof(fields[7]); vol3 = !strcasecmp(fields[9], "nan") ? NAN : atof(fields[9]); if (isnan(vol) && isnan(vol2) && isnan(vol3)) goto end; sec = atoi(fields[1]); msec = atoi(fields[2]); spot = atof(fields[10]); spotname = dstr_new(fields[11]); type = fields[12]; strike = atof(fields[13]); r = atof(fields[14]); expiry = atof(fields[15]); table_lock(spots); if ((pd = table_get_value(spots, spotname)) == NULL) { /* can't happen */ if (NEW(scp) == NULL) { xcb_log(XCB_LOG_WARNING, "Error allocating memory for scp"); table_unlock(spots); goto end; } scp->strike = strike; if (!strcasecmp(type, "C")) { scp->cvol = vol; scp->pvol = NAN; scp->cvol2 = vol2; scp->pvol2 = NAN; scp->cvol3 = vol3; scp->pvol3 = NAN; } else { scp->cvol = NAN; scp->pvol = vol; scp->cvol2 = NAN; scp->pvol2 = vol2; scp->cvol3 = NAN; scp->pvol3 = vol3; } if (NEW(pd) == NULL) { xcb_log(XCB_LOG_WARNING, "Error allocating memory for pd"); FREE(scp); table_unlock(spots); goto end; } pd->prevxo = pd->prevxo2 = pd->prevxo3 = NAN; pd->sep = strchr(fields[3], '-') ? "-" : ""; pd->dlist = dlist_new(NULL, scpfree); dlist_insert_tail(pd->dlist, scp); table_insert(spots, spotname, pd); } else { dlist_iter_t iter = dlist_iter_new(pd->dlist, DLIST_START_HEAD); dlist_node_t node; while ((node = dlist_next(iter))) { scp = (struct scp *)dlist_node_value(node); if (scp->strike > strike || fabs(scp->strike - strike) <= 0.000001) break; } if (node && fabs(scp->strike - strike) <= 0.000001) { if (!strcasecmp(type, "C")) { scp->cvol = vol; scp->cvol2 = vol2; scp->cvol3 = vol3; } else { scp->pvol = vol; scp->pvol2 = vol2; scp->pvol3 = vol3; } } else { /* can't happen */ if (NEW(scp) == NULL) { xcb_log(XCB_LOG_WARNING, "Error allocating memory for scp"); table_unlock(spots); goto end; } scp->strike = strike; if (!strcasecmp(type, "C")) { scp->cvol = vol; scp->pvol = NAN; scp->cvol2 = vol2; scp->pvol2 = NAN; scp->cvol3 = vol3; scp->pvol3 = NAN; } else { scp->cvol = NAN; scp->pvol = vol; scp->cvol2 = NAN; scp->pvol2 = vol2; scp->cvol3 = NAN; scp->pvol3 = vol3; } if (node == NULL) dlist_insert_tail(pd->dlist, scp); else dlist_insert(pd->dlist, node, scp, 0); } dlist_iter_rewind_head(iter, pd->dlist); while ((node = dlist_next(iter))) { scp = (struct scp *)dlist_node_value(node); if (scp->strike > spot || fabs(scp->strike - spot) <= 0.000001) break; } dlist_iter_free(&iter); if (node && dlist_node_prev(node)) { struct scp *scp1 = (struct scp *)dlist_node_value(node); struct scp *scp2 = (struct scp *)dlist_node_value(dlist_node_prev(node)); double vxo = NAN, vxo2 = NAN, vxo3 = NAN; int flag1, flag2, flag3; if (!isnan(scp1->cvol) && !isnan(scp1->pvol) && !isnan(scp2->cvol) && !isnan(scp2->pvol)) vxo = ((scp1->cvol + scp1->pvol) / 2) * (spot - scp2->strike) / (scp1->strike - scp2->strike) + ((scp2->cvol + scp2->pvol) / 2) * (scp1->strike - spot) / (scp1->strike - scp2->strike); if (!isnan(scp1->cvol2) && !isnan(scp1->pvol2) && !isnan(scp2->cvol2) && !isnan(scp2->pvol2)) vxo2 = ((scp1->cvol2 + scp1->pvol2) / 2) * (spot - scp2->strike) / (scp1->strike - scp2->strike) + ((scp2->cvol2 + scp2->pvol2) / 2) * (scp1->strike - spot) / (scp1->strike - scp2->strike); if (!isnan(scp1->cvol3) && !isnan(scp1->pvol3) && !isnan(scp2->cvol3) && !isnan(scp2->pvol3)) vxo3 = ((scp1->cvol3 + scp1->pvol3) / 2) * (spot - scp2->strike) / (scp1->strike - scp2->strike) + ((scp2->cvol3 + scp2->pvol3) / 2) * (scp1->strike - spot) / (scp1->strike - scp2->strike); if ((flag1 = (isnan(pd->prevxo) && !isnan(vxo)) || (!isnan(pd->prevxo) && isnan(vxo)) || (!isnan(pd->prevxo) && !isnan(vxo) && fabs(pd->prevxo - vxo) > 0.000001) ? 1 : 0)) pd->prevxo = vxo; if ((flag2 = (isnan(pd->prevxo2) && !isnan(vxo2)) || (!isnan(pd->prevxo2) && isnan(vxo2)) || (!isnan(pd->prevxo2) && !isnan(vxo2) && fabs(pd->prevxo2 - vxo2) > 0.000001) ? 1 : 0)) pd->prevxo2 = vxo2; if ((flag3 = (isnan(pd->prevxo3) && !isnan(vxo3)) || (!isnan(pd->prevxo3) && isnan(vxo3)) || (!isnan(pd->prevxo3) && !isnan(vxo3) && fabs(pd->prevxo3 - vxo3) > 0.000001) ? 1 : 0)) pd->prevxo3 = vxo3; if (flag1 || flag2 || flag3) { char *res; if ((res = ALLOC(4096))) { time_t t = (time_t)sec; struct tm lt; char datestr[64]; int off; strftime(datestr, sizeof datestr, "%F %T", localtime_r(&t, <)); snprintf(res, 4096, "VXO,%s.%03d,%s|%f,%f,%f", datestr, msec, spotname, vxo, vxo2, vxo3); out2rmp(res); off = snprintf(res, 4096, "VXO,%d,%d,%s,", sec, msec, spotname); iter = dlist_iter_new(pd->dlist, DLIST_START_HEAD); while ((node = dlist_next(iter))) { scp = (struct scp *)dlist_node_value(node); off += snprintf(res + off, 4096 - off, "%.f,%f,%f,%f,", scp->strike, vxo, vxo2, vxo3); } dlist_iter_free(&iter); snprintf(res + off, 4096 - off, "%.2f,%f,%f,%s", spot, r, expiry, pd->sep); if (out2msgs(res, out) == -1) FREE(res); } } } dstr_free(spotname); } table_unlock(spots); end: dstr_free_tokens(fields, nfield); return 0; }
static int greeks2_exec(void *data, void *data2) { RAII_VAR(struct msg *, msg, (struct msg *)data, msg_decr); dstr *fields = NULL; int nfield = 0; time_t t; char *contract, *type; double spot, strike, r, vol, vol2, vol3, expiry; int steps; NOT_USED(data2); fields = dstr_split_len(msg->data, strlen(msg->data), ",", 1, &nfield); /* FIXME */ if (nfield != 19) { xcb_log(XCB_LOG_WARNING, "Message '%s' garbled", msg->data); goto end; } t = (time_t)atoi(fields[1]); contract = fields[3]; type = fields[12]; spot = atof(fields[10]); strike = atof(fields[13]); r = atof(fields[14]); vol = atof(fields[5]); vol2 = atof(fields[7]); vol3 = atof(fields[9]); expiry = atof(fields[15]); steps = atoi(fields[18]); if (!isnan(vol) || !isnan(vol2) || !isnan(vol3)) { double delta, gamma, theta, vega, rho; double delta2, gamma2, theta2, vega2, rho2; double delta3, gamma3, theta3, vega3, rho3; struct tm lt; char datestr[64], res[512]; if (isnan(vol)) delta = gamma = theta = vega = rho = NAN; else { if (!strcasecmp(type, "C")) bi_amer_call_greeks(spot, strike, r, r, vol, expiry, steps, &delta, &gamma, &theta, &vega, &rho); else bi_amer_put_greeks (spot, strike, r, r, vol, expiry, steps, &delta, &gamma, &theta, &vega, &rho); } if (isnan(vol2)) delta2 = gamma2 = theta2 = vega2 = rho2 = NAN; else if (fabs(vol2 - vol) <= 0.000001) { delta2 = delta; gamma2 = gamma; theta2 = theta; vega2 = vega; rho2 = rho; } else { if (!strcasecmp(type, "C")) bi_amer_call_greeks(spot, strike, r, r, vol2, expiry, steps, &delta2, &gamma2, &theta2, &vega2, &rho2); else bi_amer_put_greeks (spot, strike, r, r, vol2, expiry, steps, &delta2, &gamma2, &theta2, &vega2, &rho2); } if (isnan(vol3)) delta3 = gamma3 = theta3 = vega3 = rho3 = NAN; else if (fabs(vol3 - vol) <= 0.000001) { delta3 = delta; gamma3 = gamma; theta3 = theta; vega3 = vega; rho3 = rho; } else { if (!strcasecmp(type, "C")) bi_amer_call_greeks(spot, strike, r, r, vol3, expiry, steps, &delta3, &gamma3, &theta3, &vega3, &rho3); else bi_amer_put_greeks (spot, strike, r, r, vol3, expiry, steps, &delta3, &gamma3, &theta3, &vega3, &rho3); } strftime(datestr, sizeof datestr, "%F %T", localtime_r(&t, <)); snprintf(res, sizeof res, "GREEKS2,%s.%03d,%s|%.2f,%f,%f,%f,%f,%f,%.2f,%f,%f,%f,%f,%f,%.2f,%f,%f,%f,%f,%f", datestr, atoi(fields[2]), contract, atof(fields[4]), delta, gamma, theta, vega, rho, atof(fields[6]), delta2, gamma2, theta2, vega2, rho2, atof(fields[8]), delta3, gamma3, theta3, vega3, rho3); out2rmp(res); } end: dstr_free_tokens(fields, nfield); return 0; }
/* FIXME */ static void *q_thread(void *data) { struct crss *crss = (struct crss *)data; char buf[64]; dstr rid, res = NULL; db_iterator_t *it; const char *key, *value; size_t klen, vlen; snprintf(buf, sizeof buf, "%p,", crss->c); rid = dstr_new(buf); rid = dstr_cat(rid, crss->rid); it = db_iterator_create(db, db_ro); db_iterator_seek(it, crss->start, dstr_length(crss->start)); /* seek failed */ if (!db_iterator_valid(it)) { res = dstr_new(crss->rid); res = dstr_cat(res, ",1\r\n\r\n"); pthread_spin_lock(&crss->c->lock); if (net_try_write(crss->c->fd, res, dstr_length(res), 10, NET_NONBLOCK) == -1) xcb_log(XCB_LOG_WARNING, "Writing to client '%p': %s", crss->c, strerror(errno)); pthread_spin_unlock(&crss->c->lock); goto end; } key = db_iterator_key(it, &klen); while (memcmp(key, crss->stop, dstr_length(crss->stop)) <= 0) { if (crss->cancel) break; if (!strcmp(crss->match, "") || strstr(key, crss->match)) { value = db_iterator_value(it, &vlen); res = dstr_new(crss->rid); res = dstr_cat(res, ",0,"); res = dstr_cat_len(res, key, klen); res = dstr_cat(res, ","); res = dstr_cat_len(res, value, vlen); res = dstr_cat(res, "\r\n"); pthread_spin_lock(&crss->c->lock); if (net_try_write(crss->c->fd, res, dstr_length(res), 10, NET_NONBLOCK) == -1) { xcb_log(XCB_LOG_WARNING, "Writing to client '%p': %s", crss->c, strerror(errno)); pthread_spin_unlock(&crss->c->lock); goto end; } pthread_spin_unlock(&crss->c->lock); dstr_free(res); } db_iterator_next(it); if (!db_iterator_valid(it) || crss->cancel) break; key = db_iterator_key(it, &klen); } res = dstr_new(crss->rid); res = dstr_cat(res, ",1\r\n\r\n"); pthread_spin_lock(&crss->c->lock); if (net_try_write(crss->c->fd, res, dstr_length(res), 10, NET_NONBLOCK) == -1) xcb_log(XCB_LOG_WARNING, "Writing to client '%p': %s", crss->c, strerror(errno)); pthread_spin_unlock(&crss->c->lock); end: db_iterator_destroy(&it); dstr_free(res); table_lock(rids); table_remove(rids, rid); table_unlock(rids); dstr_free(rid); client_decr(crss->c); dstr_free(crss->rid); dstr_free(crss->match); dstr_free(crss->start); dstr_free(crss->stop); FREE(crss); return NULL; }
/* FIXME */ void s_command(client c) { dstr pkey, skey; int i; dlist_t dlist; struct kvd *kvd; if (dstr_length(c->argv[0]) == 1) { add_reply_error(c, "index can't be empty\r\n"); return; } pkey = dstr_new(c->argv[0] + 1); skey = dstr_new(pkey); if (c->argc > 1) for (i = 1; i < c->argc; ++i) { skey = dstr_cat(skey, ","); skey = dstr_cat(skey, c->argv[i]); } table_lock(cache); if ((dlist = table_get_value(cache, pkey))) { dlist_iter_t iter = dlist_iter_new(dlist, DLIST_START_HEAD); dlist_node_t node; /* FIXME: still has room to improve */ while ((node = dlist_next(iter))) { kvd = (struct kvd *)dlist_node_value(node); if (dstr_length(kvd->key) >= dstr_length(skey) && !memcmp(kvd->key, skey, dstr_length(skey))) { dstr *fields = NULL; int nfield = 0; dstr res, contracts = NULL; fields = dstr_split_len(kvd->key, dstr_length(kvd->key), ",", 1, &nfield); res = dstr_new(fields[0]); if (nfield > 1) { contracts = dstr_new(fields[1]); for (i = 2; i < nfield; ++i) { contracts = dstr_cat(contracts, ","); contracts = dstr_cat(contracts, fields[i]); } } dstr_free_tokens(fields, nfield); fields = NULL, nfield = 0; fields = dstr_split_len(kvd->u.value, dstr_length(kvd->u.value), ",", 1, &nfield); res = dstr_cat(res, ","); res = dstr_cat(res, fields[0]); if (contracts) { res = dstr_cat(res, ","); res = dstr_cat(res, contracts); } for (i = 1; i < nfield; ++i) { res = dstr_cat(res, ","); res = dstr_cat(res, fields[i]); } res = dstr_cat(res, "\r\n"); pthread_spin_lock(&c->lock); if (!(c->flags & CLIENT_CLOSE_ASAP)) { if (net_try_write(c->fd, res, dstr_length(res), 10, NET_NONBLOCK) == -1) { xcb_log(XCB_LOG_WARNING, "Writing to client '%p': %s", c, strerror(errno)); if (++c->eagcount >= 3) { client_free_async(c); pthread_spin_unlock(&c->lock); dstr_free(contracts); dstr_free(res); dstr_free_tokens(fields, nfield); table_unlock(cache); dstr_free(skey); dstr_free(pkey); return; } } else if (c->eagcount) c->eagcount = 0; } pthread_spin_unlock(&c->lock); dstr_free(contracts); dstr_free(res); dstr_free_tokens(fields, nfield); } } dlist_iter_free(&iter); } table_unlock(cache); table_rwlock_wrlock(subscribers); if ((dlist = table_get_value(subscribers, pkey)) == NULL) { if (NEW(kvd)) { kvd->key = skey; kvd->u.dlist = dlist_new(NULL, NULL); dlist_insert_tail(kvd->u.dlist, c); dlist = dlist_new(cmpkvd, kdfree); dlist_insert_sort(dlist, kvd); table_insert(subscribers, pkey, dlist); } else { add_reply_error(c, "error allocating memory for kvd\r\n"); dstr_free(skey); dstr_free(pkey); } } else { if (NEW(kvd)) { dlist_node_t node; kvd->key = skey; kvd->u.dlist = dlist_new(NULL, NULL); if ((node = dlist_find(dlist, kvd)) == NULL) { dlist_insert_tail(kvd->u.dlist, c); dlist_insert_sort(dlist, kvd); } else { kdfree(kvd); kvd = (struct kvd *)dlist_node_value(node); if (dlist_find(kvd->u.dlist, c) == NULL) dlist_insert_tail(kvd->u.dlist, c); } } else { add_reply_error(c, "error allocating memory for kvd\r\n"); dstr_free(skey); } dstr_free(pkey); } table_rwlock_unlock(subscribers); }