/* * A true RPC. It return a NULL pointer * in case of failure, otherwise a valid * (ETERM *) pointer containing the reply */ int ei_rpc(ei_cnode* ec, int fd, char *mod, char *fun, const char* inbuf, int inbuflen, ei_x_buff* x) { int i, index; ei_term t; erlang_msg msg; char rex[MAXATOMLEN]; if (ei_rpc_to(ec, fd, mod, fun, inbuf, inbuflen) < 0) { return -1; } /* FIXME are we not to reply to the tick? */ while ((i = ei_rpc_from(ec, fd, ERL_NO_TIMEOUT, &msg, x)) == ERL_TICK) ; if (i == ERL_ERROR) return -1; /*ep = 'erl'_element(2,emsg.msg);*/ /* {RPC_Tag, RPC_Reply} */ index = 0; if (ei_decode_version(x->buff, &index, &i) < 0 || ei_decode_ei_term(x->buff, &index, &t) < 0) return -1; /* FIXME ei_decode_version don't set erl_errno as before */ /* FIXME this is strange, we don't check correct "rex" atom and we let it pass if not ERL_SMALL_TUPLE_EXT and arity == 2 */ if (t.ei_type == ERL_SMALL_TUPLE_EXT && t.arity == 2) if (ei_decode_atom(x->buff, &index, rex) < 0) return -1; /* remove header */ x->index -= index; memmove(x->buff, &x->buff[index], x->index); return 0; }
/* * Reads an Erlang term. * * Only accepts 't' (term) or 'e' (end of test), * exits program on error * returns 1 on 'e', 0 on 't' */ int get_bin_term(ei_x_buff* x, ei_term* term) { int len, version; ei_x_free(x); x->buff = read_packet(&len); x->buffsz = len; x->index = 0; switch (x->buff[x->index++]) { case 'e': return 1; case 't': if (ei_decode_version(x->buff, &x->index, &version) < 0 || ei_decode_ei_term(x->buff, &x->index, term) < 0) { fail("Failed to decode term"); exit(0); } return 0; default: fprintf(stderr, "Garbage received: "); dump(x->buff, len, 16); putc('\n', stderr); fail("C program received garbage"); exit(1); } }
int main(void) #endif { ErlConnect conp; Erl_IpAddr thisipaddr = (Erl_IpAddr)0; FILE *fp = (FILE *)0; char* charp = "foo"; double *doublep = NULL; double doublex = 0.0; ei_cnode xec; ei_reg *ei_regp = NULL; ei_term eterm; ei_x_buff eix; erlang_big *bigp = NULL; erlang_fun efun; erlang_msg *msgp = NULL; erlang_msg emsg; erlang_pid *pidp = NULL; erlang_pid epid; erlang_port eport; erlang_ref eref; erlang_trace etrace; int *intp = NULL; int intx = 0; long *longp = NULL; long longx = 0; short creation = 0; struct ei_reg_stat *ei_reg_statp = NULL; struct ei_reg_tabstat *ei_reg_tabstatp = NULL; struct hostent *hostp = NULL; unsigned char * ucharp = (unsigned char *)"foo"; unsigned long *ulongp = NULL; unsigned long ulongx = 0; void *voidp = NULL; #ifndef VXWORKS EI_LONGLONG *longlongp = (EI_LONGLONG*)NULL; EI_LONGLONG longlongx = 0; EI_ULONGLONG *ulonglongp = (EI_ULONGLONG*)NULL; EI_ULONGLONG ulonglongx = 0; #endif enum erlang_char_encoding enc; intx = erl_errno; ei_connect_init(&xec, charp, charp, creation); ei_connect_xinit (&xec, charp, charp, charp, thisipaddr, charp, creation); ei_connect(&xec, charp); ei_xconnect (&xec, thisipaddr, charp); ei_receive(intx, ucharp, intx); ei_receive_msg(intx, &emsg, &eix); ei_xreceive_msg(intx, &emsg, &eix); ei_send(intx, &epid, charp, intx); ei_reg_send(&xec, intx, charp, charp, intx); ei_rpc(&xec, intx, charp, charp, charp, intx, &eix); ei_rpc_to(&xec, intx, charp, charp, charp, intx); ei_rpc_from(&xec, intx, intx, &emsg, &eix); ei_publish(&xec, intx); ei_accept(&xec, intx, &conp); ei_unpublish(&xec); ei_thisnodename(&xec); ei_thishostname(&xec); ei_thisalivename(&xec); ei_self(&xec); ei_gethostbyname(charp); ei_gethostbyaddr(charp, intx, intx); ei_gethostbyname_r(charp, hostp, charp, intx, intp); ei_gethostbyaddr_r(charp, intx, intx, hostp, charp, intx, intp); ei_encode_version(charp, intp); ei_x_encode_version(&eix); ei_encode_long(charp, intp, longx); ei_x_encode_long(&eix, longx); ei_encode_ulong(charp, intp, ulongx); ei_x_encode_ulong(&eix, ulongx); ei_encode_double(charp, intp, doublex); ei_x_encode_double(&eix, doublex); ei_encode_boolean(charp, intp, intx); ei_x_encode_boolean(&eix, intx); ei_encode_char(charp, intp, 'a'); ei_x_encode_char(&eix, 'a'); ei_encode_string(charp, intp, charp); ei_encode_string_len(charp, intp, charp, intx); ei_x_encode_string(&eix, charp); ei_x_encode_string_len(&eix, charp, intx); ei_encode_atom(charp, intp, charp); ei_encode_atom_as(charp, intp, charp, ERLANG_LATIN1, ERLANG_UTF8); ei_encode_atom_len(charp, intp, charp, intx); ei_encode_atom_len_as(charp, intp, charp, intx, ERLANG_ASCII, ERLANG_LATIN1); ei_x_encode_atom(&eix, charp); ei_x_encode_atom_as(&eix, charp, ERLANG_LATIN1, ERLANG_UTF8); ei_x_encode_atom_len(&eix, charp, intx); ei_x_encode_atom_len_as(&eix, charp, intx, ERLANG_LATIN1, ERLANG_UTF8); ei_encode_binary(charp, intp, (void *)0, longx); ei_x_encode_binary(&eix, (void*)0, intx); ei_encode_pid(charp, intp, &epid); ei_x_encode_pid(&eix, &epid); ei_encode_fun(charp, intp, &efun); ei_x_encode_fun(&eix, &efun); ei_encode_port(charp, intp, &eport); ei_x_encode_port(&eix, &eport); ei_encode_ref(charp, intp, &eref); ei_x_encode_ref(&eix, &eref); ei_encode_trace(charp, intp, &etrace); ei_x_encode_trace(&eix, &etrace); ei_encode_tuple_header(charp, intp, intx); ei_x_encode_tuple_header(&eix, longx); ei_encode_list_header(charp, intp, intx); ei_x_encode_list_header(&eix, longx); /* #define ei_encode_empty_list(buf,i) ei_encode_list_header(buf,i,0) */ ei_x_encode_empty_list(&eix); ei_get_type(charp, intp, intp, intp); ei_get_type_internal(charp, intp, intp, intp); ei_decode_version(charp, intp, intp); ei_decode_long(charp, intp, longp); ei_decode_ulong(charp, intp, ulongp); ei_decode_double(charp, intp, doublep); ei_decode_boolean(charp, intp, intp); ei_decode_char(charp, intp, charp); ei_decode_string(charp, intp, charp); ei_decode_atom(charp, intp, charp); ei_decode_atom_as(charp, intp, charp, MAXATOMLEN_UTF8, ERLANG_WHATEVER, &enc, &enc); ei_decode_binary(charp, intp, (void *)0, longp); ei_decode_fun(charp, intp, &efun); free_fun(&efun); ei_decode_pid(charp, intp, &epid); ei_decode_port(charp, intp, &eport); ei_decode_ref(charp, intp, &eref); ei_decode_trace(charp, intp, &etrace); ei_decode_tuple_header(charp, intp, intp); ei_decode_list_header(charp, intp, intp); ei_decode_ei_term(charp, intp, &eterm); ei_print_term(fp, charp, intp); ei_s_print_term(&charp, charp, intp); ei_x_format(&eix, charp); ei_x_format_wo_ver(&eix, charp); ei_x_new(&eix); ei_x_new_with_version(&eix); ei_x_free(&eix); ei_x_append(&eix, &eix); ei_x_append_buf(&eix, charp, intx); ei_skip_term(charp, intp); ei_reg_open(intx); ei_reg_resize(ei_regp, intx); ei_reg_close(ei_regp); ei_reg_setival(ei_regp, charp, longx); ei_reg_setfval(ei_regp, charp, doublex); ei_reg_setsval(ei_regp, charp, charp); ei_reg_setpval(ei_regp, charp, voidp, intx); ei_reg_setval(ei_regp, charp, intx); ei_reg_getival(ei_regp, charp); ei_reg_getfval(ei_regp, charp); ei_reg_getsval(ei_regp, charp); ei_reg_getpval(ei_regp, charp, intp); ei_reg_getval(ei_regp, charp, intx); ei_reg_markdirty(ei_regp, charp); ei_reg_delete(ei_regp, charp); ei_reg_stat(ei_regp, charp, ei_reg_statp); ei_reg_tabstat(ei_regp, ei_reg_tabstatp); ei_reg_dump(intx, ei_regp, charp, intx); ei_reg_restore(intx, ei_regp, charp); ei_reg_purge(ei_regp); #if defined(HAVE_GMP_H) && defined(HAVE_LIBGMP) { mpz_t obj; ei_decode_bignum(charp, intp, obj); ei_encode_bignum(charp, intp, obj); ei_x_encode_bignum(&eix, obj); } #endif /* HAVE_GMP_H && HAVE_LIBGMP */ #ifndef VXWORKS ei_decode_longlong(charp, intp, longlongp); ei_decode_ulonglong(charp, intp, ulonglongp); ei_encode_longlong(charp, intp, longlongx); ei_encode_ulonglong(charp, intp, ulonglongx); ei_x_encode_longlong(&eix, longlongx); ei_x_encode_ulonglong(&eix, ulonglongx); #endif #ifdef USE_EI_UNDOCUMENTED ei_decode_intlist(charp, intp, longp, intp); ei_receive_encoded(intx, &charp, intp, msgp, intp); ei_send_encoded(intx, pidp, charp, intx); ei_send_reg_encoded(intx, pidp, charp, charp, intx); ei_decode_big(charp, intp, bigp); ei_big_comp(bigp, bigp); ei_big_to_double(bigp, doublep); ei_small_to_big(intx, bigp); ei_alloc_big(intx); ei_free_big(bigp); #endif /* USE_EI_UNDOCUMENTED */ return BUFSIZ + EAGAIN + EHOSTUNREACH + EIO + EI_BIN + EI_DELET + EI_DIRTY + EI_FLT + EI_FORCE + EI_INT + EI_NOPURGE + EI_STR + EMSGSIZE + ENOMEM + ERL_ERROR + ERL_EXIT + ERL_LINK + ERL_MSG + ERL_NO_TIMEOUT + ERL_REG_SEND + ERL_SEND + ERL_TICK + ERL_TIMEOUT + ERL_UNLINK + ETIMEDOUT + MAXATOMLEN; }
static ErlDrvSSizeT control(ErlDrvData drv_data, unsigned int command, char *buf, ErlDrvSizeT, char **rbuf, ErlDrvSizeT) { driver_data_t *data = (driver_data_t *)drv_data; ei_x_buff x; ei_x_new_with_version(&x); switch ((TPCE::eCommand)command) { case TPCE::CMD_ALL_INIT: { int index = 0, version, arity; char atom[MAXATOMLEN+1]; bool found = true; ei_decode_version(buf, &index, &version); ei_decode_list_header(buf, &index, &arity); for (int size = arity, i = 0; found && i < size; i++) { ei_decode_tuple_header(buf, &index, &arity); if (2 == arity) { ei_term term; ei_decode_atom(buf, &index, atom); ei_decode_ei_term(buf, &index, &term); if (0 == strcmp(atom, "id")) { data->id = term.value.i_val; } else if (0 == strcmp(atom, "path") && ERL_STRING_EXT == term.ei_type) { char v[term.size+1]; ei_decode_string(buf, &index, v); strncpy(data->path, v, sizeof(data->path)); } else if (0 == strcmp(atom, "path") && ERL_BINARY_EXT == term.ei_type) { char v[term.size+1]; long l = term.size; ei_decode_binary(buf, &index, v, &l); v[l] = '\0'; strncpy(data->path, v, sizeof(data->path)); } else if (0 == strcmp(atom, "configured_customer_count")) { data->configured_customer_count = term.value.i_val; } else if (0 == strcmp(atom, "active_customer_count")) { data->active_customer_count = term.value.i_val; } else if (0 == strcmp(atom, "scale_factor")) { data->scale_factor = term.value.i_val; } else if (0 == strcmp(atom, "days_of_initial_trades")) { data->days_of_initial_trades = term.value.i_val; } else { found = false; } } } if (found) { ei_x_encode_atom(&x, "ok"); } else { ergen_drv_encode_error1(&x, "badmatch", atom); } } break; case TPCE::CMD_ALL_CONFIG: { ei_x_encode_atom(&x, "ok"); } break; case TPCE::CMD_ALL_ALLOC: { void *ptr = ergen_driver_new(data); if (NULL != ptr) { data->driver = ptr; ei_x_encode_atom(&x, "ok"); } else { ergen_drv_encode_error0(&x, "nomem"); } } break; case TPCE::CMD_ALL_FREE: { void *ptr = data->driver; if (NULL != ptr) { ergen_driver_delete(ptr); data->driver = NULL; } ei_x_encode_atom(&x, "ok"); } break; default: { ergen_drv_encode_error0(&x, "badarg"); } break; } ErlDrvSizeT result = x.index; *rbuf = (char *)ergen_drv_bindup(x.buff, result); ei_x_free(&x); return result; }
static ErlDrvSSizeT call(ErlDrvData edd, unsigned int cmd, char *buf, ErlDrvSizeT len, char **rbuf, ErlDrvSizeT rlen, unsigned int *flags) { dnssd_drv_t* dd = (dnssd_drv_t*) edd; int version, out_len, index, rindex, local_only; DNSServiceErrorType err; char* out_atom_text; ei_term arg, name, type, domain, txt, host, hostport; char *name_tmp, *type_tmp, *domain_tmp, *txt_tmp, *host_tmp; /* don't allow reuse */ if (dd->sd_ref) return -1; index = 0; dd->sd_ref = NULL; ei_decode_version(buf, &index, &version); ei_decode_ei_term(buf, &index, &arg); if (cmd == DNSSD_CMD_ENUM) { if (arg.ei_type == ERL_ATOM_EXT) { if (strncmp(arg.value.atom_name, "browse", 6) == 0) { // init for enum browse err = DNSServiceEnumerateDomains(&dd->sd_ref, kDNSServiceFlagsBrowseDomains, kDNSServiceInterfaceIndexAny, (DNSServiceDomainEnumReply) EnumReply, dd); } else if (strncmp(arg.value.atom_name,"reg", 3) == 0) { // init for enum reg err = DNSServiceEnumerateDomains(&dd->sd_ref, kDNSServiceFlagsRegistrationDomains, kDNSServiceInterfaceIndexAny, (DNSServiceDomainEnumReply) EnumReply, dd); } else { goto badarg; } } else { goto badarg; } } else if (cmd == DNSSD_CMD_BROWSE) { if (!arg.ei_type == ERL_TUPLE || arg.arity != 2) goto badarg; /* decode type */ ei_decode_ei_term(buf, &index, &type); if (type.ei_type != ERL_BINARY_EXT) goto badarg; index += 5; // skip tag + 4 byte size type_tmp = (char*)driver_alloc(type.size + 1); memset(type_tmp, 0, type.size + 1); memcpy(type_tmp, buf + index, type.size); index += type.size; /* decode domain */ ei_decode_ei_term(buf, &index, &domain); if (domain.ei_type != ERL_BINARY_EXT) { driver_free(type_tmp); goto badarg; } index += 5; // skip tag + 4 byte size domain_tmp = (char *) driver_alloc(domain.size + 1); memset(domain_tmp, 0, domain.size + 1); memcpy(domain_tmp, buf + index, domain.size); err = DNSServiceBrowse(&dd->sd_ref, 0, // Flags kDNSServiceInterfaceIndexAny, type_tmp, domain_tmp, (DNSServiceBrowseReply) BrowseReply, dd); driver_free(type_tmp); driver_free(domain_tmp); } else if (cmd == DNSSD_CMD_RESOLVE) { if (!arg.ei_type == ERL_TUPLE || arg.arity != 3) goto badarg; /* decode name */ ei_decode_ei_term(buf, &index, &name); if (name.ei_type != ERL_BINARY_EXT) goto badarg; index += 5; // skip tag + 4 byte size name_tmp = (char *) driver_alloc(name.size + 1); memset(name_tmp, 0, name.size + 1); memcpy(name_tmp, buf + index, name.size); index += name.size; /* decode type */ ei_decode_ei_term(buf, &index, &type); if (type.ei_type != ERL_BINARY_EXT) { driver_free(name_tmp); goto badarg; } index += 5; // skip tag + 4 byte size type_tmp = (char *) driver_alloc(type.size + 1); memset(type_tmp, 0, type.size + 1); memcpy(type_tmp, buf + index, type.size); index += type.size; /* decode domain */ ei_decode_ei_term(buf, &index, &domain); if (domain.ei_type != ERL_BINARY_EXT) { driver_free(name_tmp); driver_free(type_tmp); goto badarg; } index += 5; // skip tag + 4 byte size domain_tmp = (char *) driver_alloc(domain.size + 1); memset(domain_tmp, 0, domain.size + 1); memcpy(domain_tmp, buf + index, domain.size); /* start op */ err = DNSServiceResolve(&dd->sd_ref, 0, // Flags kDNSServiceInterfaceIndexAny, name_tmp, type_tmp, domain_tmp, (DNSServiceResolveReply) ResolveReply, dd); driver_free(name_tmp); driver_free(type_tmp); driver_free(domain_tmp); } else if (cmd == DNSSD_CMD_REGISTER) { if (!arg.ei_type == ERL_TUPLE || arg.arity != 6) goto badarg; /* decode name */ ei_decode_ei_term(buf, &index, &name); if (name.ei_type != ERL_BINARY_EXT) goto badarg; index += 5; // skip tag + 4 byte size name_tmp = (char *) driver_alloc(name.size + 1); memset(name_tmp, 0, name.size + 1); memcpy(name_tmp, buf + index, name.size); index += name.size; /* decode type */ ei_decode_ei_term(buf, &index, &type); if (type.ei_type != ERL_BINARY_EXT) { driver_free(name_tmp); goto badarg; } index += 5; // skip tag + 4 byte size type_tmp = (char *) driver_alloc(type.size + 1); memset(type_tmp, 0, type.size + 1); memcpy(type_tmp, buf + index, type.size); index += type.size; /* decode domain */ ei_decode_ei_term(buf, &index, &domain); if (domain.ei_type != ERL_BINARY_EXT) { driver_free(name_tmp); driver_free(type_tmp); goto badarg; } index += 5; // skip tag + 4 byte size domain_tmp = (char *) driver_alloc(domain.size + 1); memset(domain_tmp, 0, domain.size + 1); memcpy(domain_tmp, buf + index, domain.size); index += domain.size; /* decode host */ ei_decode_ei_term(buf, &index, &host); if (host.ei_type != ERL_BINARY_EXT) { driver_free(name_tmp); driver_free(type_tmp); driver_free(domain_tmp); goto badarg; } index += 5; // skip tag + 4 byte size host_tmp = (char *) driver_alloc(host.size + 1); memset(host_tmp, 0, host.size + 1); memcpy(host_tmp, buf + index, host.size); index += host.size; /* decode port */ ei_decode_ei_term(buf, &index, &hostport); if (hostport.ei_type != ERL_INTEGER_EXT && hostport.ei_type != ERL_SMALL_INTEGER_EXT) { driver_free(name_tmp); driver_free(type_tmp); driver_free(domain_tmp); driver_free(host_tmp); goto badarg; } /* decode txt */ ei_decode_ei_term(buf, &index, &txt); if (txt.ei_type != ERL_BINARY_EXT) { driver_free(name_tmp); driver_free(type_tmp); driver_free(domain_tmp); driver_free(host_tmp); goto badarg; } index += 5; // skip tag + 4 byte size txt_tmp = (char *) driver_alloc(txt.size + 1); memset(txt_tmp, 0, txt.size + 1); memcpy(txt_tmp, buf + index, txt.size); local_only = (0 == strcmp("localhost", host_tmp)); err = DNSServiceRegister(&dd->sd_ref, 0, // Flags local_only ? -1 : 0, // Interface: local / any name_tmp, type_tmp, local_only ? "local" : domain_tmp, host_tmp, htons(hostport.value.i_val), txt.size, txt_tmp, (DNSServiceRegisterReply) RegisterReply, dd); driver_free(name_tmp); driver_free(type_tmp); driver_free(domain_tmp); driver_free(host_tmp); driver_free(txt_tmp); } else { goto badarg; } rindex = 0; out_len = 0; ei_encode_version(NULL, &out_len); if (err == kDNSServiceErr_NoError) { #ifdef __WIN32__ dd->event = WSACreateEvent(); WSAEventSelect(DNSServiceRefSockFD(dd->sd_ref), dd->event, FD_READ); driver_select(dd->erl_port, dd->event, ERL_DRV_READ, 1); #else driver_select(dd->erl_port, (ErlDrvEvent)(size_t) DNSServiceRefSockFD(dd->sd_ref), ERL_DRV_READ, 1); #endif out_atom_text = "ok"; ei_encode_atom(NULL, &out_len, out_atom_text); if(rlen < out_len) { *rbuf = driver_alloc(out_len); rlen = out_len; } ei_encode_version(*rbuf, &rindex); ei_encode_atom(*rbuf, &rindex, out_atom_text); return out_len; } else { out_atom_text = "error"; ei_encode_tuple_header(NULL, &out_len, 2); ei_encode_atom(NULL, &out_len, out_atom_text); ei_encode_long(NULL, &out_len, 1337); if(rlen < out_len) { *rbuf = driver_alloc(out_len); rlen = out_len; } ei_encode_version(*rbuf, &rindex); ei_encode_tuple_header(*rbuf, &rindex, 2); ei_encode_atom(*rbuf, &rindex, out_atom_text); ei_encode_long(*rbuf, &rindex, (long) err); return out_len; } badarg: return -1; }
/** * Query a table for specified rows. * \param _h structure representing database connection * \param _k key names * \param _op operators *\param _v values of the keys that must match * \param _c column names to return * \param _n number of key=values pairs to compare * \param _nc number of columns to return * \param _o order by the specified column * \param _r pointer to a structure representing the result * \return zero on success, negative value on failure */ int erlang_srdb1_query(const db1_con_t* _h, const db_key_t* _k, const db_op_t* _op, const db_val_t* _v, const db_key_t* _c, const int _n, const int _nc, const db_key_t _o, db1_res_t** _r) { ei_x_buff argbuf,retbuf; int retcode,i,j,x; int n_cols,n_rows,len; db1_res_t *res; db_row_t *rows = NULL, *row; db_val_t *val; char atom[MAXATOMLEN], *p; ei_term term; int ei_type,size; str *sname; if (!_h || !_r) { LM_ERR("invalid parameter value\n"); return -1; } *_r=NULL; LM_DBG("erlang_srdb1_query table %.*s\n",CON_TABLE(_h)->len, CON_TABLE(_h)->s); ei_x_new(&argbuf); //encode tuple {db_op, table, [cols], [params]} ei_x_encode_tuple_header(&argbuf, 5); ei_x_encode_atom(&argbuf,"select"); ei_x_encode_atom_len(&argbuf,CON_TABLE(_h)->s,CON_TABLE(_h)->len); srdb1_encode_c(_c, _nc, &argbuf); srdb1_encode_k(_k, _op, _v, _n, &argbuf); // ei_x_encode_atom_len(&argbuf,_o->s,_o->len); ei_x_encode_list_header(&argbuf, 0); retcode=erl_bind.do_erlang_call(&(CON_ERLANG(_h)->con),&(CON_ERLANG(_h)->regname), &argbuf, &retbuf); ei_x_free(&argbuf); if (retcode<0) { if(retbuf.buff) shm_free(retbuf.buff); return retcode; } // we have a tuple there: ei_decode_tuple_header(retbuf.buff, &(retbuf.index), &i); x=retbuf.index; ei_skip_term(retbuf.buff, &x); LM_DBG("erlang_srdb1_query: position of end of field list should be %d\n",x); //first is list of 5-element tuples containing name and type of field ei_decode_list_header(retbuf.buff, &(retbuf.index), &n_cols); LM_DBG("erlang_srdb1_query: length -f field_list is %d\n",n_cols); res=db_new_result(); if (db_allocate_columns(res, n_cols) != 0) { LM_ERR("erlang_srdb1_query: db_allocate_columns failed\n"); goto error; } RES_COL_N(res) = n_cols; for(i=0; i < n_cols; i++) { x=retbuf.index; ei_skip_term(retbuf.buff, &x); LM_DBG("erlang_srdb1_query: position of end of this field should be %d\n",x); ei_decode_tuple_header(retbuf.buff, &(retbuf.index), &j); if( j!=5) LM_ERR("erlang_srdb1_query name&type list element tuple is not 5\n"); ei_decode_atom(retbuf.buff, &(retbuf.index), atom); //1 name len=strlen(atom); sname = (str*)pkg_malloc(sizeof(str)+len+1); if (!sname) { LM_ERR("no private memory left\n"); goto error; } sname->len = len; sname->s = (char*)sname + sizeof(str); memcpy(sname->s, atom, len); sname->s[len] = '\0'; RES_NAMES(res)[i] = sname; LM_DBG("decoded header %d, fieled 1: %s\n",i,atom); ei_decode_atom(retbuf.buff, &(retbuf.index), atom); //2 type atom if(strcmp("int",atom)==0) { RES_TYPES(res)[i]=DB1_INT; } if(strcmp("string",atom)==0) { RES_TYPES(res)[i]=DB1_STRING; } if(strcmp("float",atom)==0) { RES_TYPES(res)[i]=DB1_DOUBLE; } if(strcmp("datetime",atom)==0) { RES_TYPES(res)[i]=DB1_DATETIME; } // if(strcmp("string",atom)==0) { RES_TYPES(res)[i]=DB1_BLOB; } ei_skip_term(retbuf.buff, &(retbuf.index)); //3 size (ignored) ei_skip_term(retbuf.buff, &(retbuf.index)); //4 default value (ignored) ei_skip_term(retbuf.buff, &(retbuf.index)); //3 null status (ignored) LM_DBG("end of %d record: %d\n",i,retbuf.index); } ei_decode_ei_term(retbuf.buff, &(retbuf.index), &term); // List tail, LM_DBG("erlang_srdb1_query: position after scanning is %d\n",retbuf.index); //now rows, list of tuples ei_decode_list_header(retbuf.buff, &(retbuf.index), &n_rows); LM_DBG("erlang_srdb1_query values list size is %d\n",n_rows); if (n_rows<=0) { LM_DBG("erlang_srdb1_query no rows returned\n"); RES_ROWS(res) = NULL; RES_NUM_ROWS(res)=0; *_r=res; return 0; } RES_NUM_ROWS(res)=n_rows; rows = pkg_realloc(rows, sizeof(db_row_t) * n_rows); if (rows == NULL) { LM_ERR("erlang_srdb1_query: pkg_realloc rows failed\n"); goto error; } RES_ROWS(res) = rows; for(i=0; i < n_rows; i++) { RES_ROW_N(res)=i+1; row = &RES_ROWS(res)[i]; if (db_allocate_row(res, row) != 0) { LM_ERR("erlang_srdb1_query: db_allocate_row failed for row %d\n",i); goto error; } ei_decode_tuple_header(retbuf.buff, &(retbuf.index), &j); if(j!=n_cols) { LM_ERR("erlang_srdb1_query: mismatch:values list element tuple size is %d n_cols from header was %d\n",j, n_cols); } for (j = 0, val = ROW_VALUES(row); j < RES_COL_N(res); j++, val++) { VAL_TYPE(val) = RES_TYPES(res)[j]; VAL_NULL(val) = 0; VAL_FREE(val) = 0; retcode=ei_get_type_internal(retbuf.buff, &(retbuf.index), &ei_type, &size); if (retcode < 0) { LM_ERR("erlang_srdb1_query: error getting type for element %d %d\n",i,j); goto error; } LM_DBG("erlang_srdb1_query: element %d %d ei_type=%d size=%d\n",i,j,ei_type, size); switch(ei_type) { case ERL_SMALL_INTEGER_EXT: case ERL_INTEGER_EXT: retcode=ei_decode_long(retbuf.buff, &(retbuf.index), &VAL_INT(val)); if(retcode < 0) goto error; LM_DBG("decoded interger %d\n",VAL_INT(val)); break; case ERL_FLOAT_EXT: case NEW_FLOAT_EXT: retcode=ei_decode_double(retbuf.buff, &(retbuf.index), &VAL_DOUBLE(val)); if(retcode < 0) goto error; LM_DBG("decoded float %f\n",VAL_DOUBLE(val)); break; case ERL_ATOM_EXT: case ERL_SMALL_ATOM_EXT: case ERL_ATOM_UTF8_EXT: case ERL_SMALL_ATOM_UTF8_EXT: p=pkg_malloc(size+1); if(!p) { LM_ERR("erlang_srdb1_query: no memory\n"); goto error; } retcode=ei_decode_atom(retbuf.buff, &(retbuf.index), p); if(retcode < 0) { pkg_free(p); goto error; } LM_DBG("decoded small_atom_utf %s\n",p); VAL_STRING(val)=p; VAL_FREE(val)=1; break; case ERL_STRING_EXT: p=pkg_malloc(size+1); if(!p) { LM_ERR("erlang_srdb1_query: no memory\n"); goto error; } retcode=ei_decode_string(retbuf.buff, &(retbuf.index), p); if(retcode < 0) { pkg_free(p); goto error; } LM_DBG("decoded string %s\n",p); VAL_STRING(val)=p; VAL_FREE(val)=1; break; case ERL_SMALL_TUPLE_EXT: case ERL_LARGE_TUPLE_EXT: LM_DBG("got tuple)\n"); if (VAL_TYPE(val)==DB1_DATETIME) { struct tm tm; LM_DBG("and col type is datetime\n"); retcode=ei_decode_tuple_header(retbuf.buff, &(retbuf.index), &x); if(retcode < 0) goto error; retcode=ei_decode_tuple_header(retbuf.buff, &(retbuf.index), &x); if(retcode < 0) goto error; retcode=ei_decode_long(retbuf.buff, &(retbuf.index), (long int *)&tm.tm_year);tm.tm_year -=1900; if(retcode < 0) goto error; retcode=ei_decode_long(retbuf.buff, &(retbuf.index), (long int *)&tm.tm_mon); tm.tm_mon -=1; if(retcode < 0) goto error; retcode=ei_decode_long(retbuf.buff, &(retbuf.index), (long int *)&tm.tm_mday); if(retcode < 0) goto error; retcode=ei_decode_tuple_header(retbuf.buff, &(retbuf.index), &x); if(retcode < 0) goto error; retcode=ei_decode_long(retbuf.buff, &(retbuf.index), (long int *)&tm.tm_hour); if(retcode < 0) goto error; retcode=ei_decode_long(retbuf.buff, &(retbuf.index), (long int *)&tm.tm_min); if(retcode < 0) goto error; retcode=ei_decode_long(retbuf.buff, &(retbuf.index), (long int *)&tm.tm_sec); if(retcode < 0) goto error; VAL_TIME(val)=mktime(&tm); break; } LM_ERR("erlang_srdb1_query: got tuple but valtype is not datetime element %d in row %d in response\n",j,i); break; case ERL_REFERENCE_EXT: case ERL_NEW_REFERENCE_EXT: case ERL_PORT_EXT: case ERL_PID_EXT: case ERL_NIL_EXT: case ERL_LIST_EXT: case ERL_BINARY_EXT: case ERL_SMALL_BIG_EXT: case ERL_LARGE_BIG_EXT: case ERL_NEW_FUN_EXT: case ERL_FUN_EXT: default: LM_ERR("erlang_srdb1_query: don't know how to handle element %d in row %d in response\n",j,i); } } } ei_decode_ei_term(retbuf.buff, &(retbuf.index), &term); // List tail, *_r=res; return 0; error: if (res) db_free_result(res); LM_ERR("erlang_srdb1_query: Failed\n"); return -1; }