X-Git-Url: http://git.indexdata.com/?a=blobdiff_plain;f=src%2Fhttp_command.c;h=8ec3264deda56f1d06c1ed1387bc9eb90be7f364;hb=4dbb25eef6b8fc15f058729ab84c6d492aefb42e;hp=54a16549f8703e47ac0d7052531f629282d70471;hpb=dde4a5914bbb5511a91e73d8ab45210b48637596;p=pazpar2-moved-to-github.git diff --git a/src/http_command.c b/src/http_command.c index 54a1654..8ec3264 100644 --- a/src/http_command.c +++ b/src/http_command.c @@ -1,38 +1,59 @@ -/* - * $Id: http_command.c,v 1.3 2007-01-03 06:23:44 quinn Exp $ - */ +/* This file is part of Pazpar2. + Copyright (C) 2006-2009 Index Data + +Pazpar2 is free software; you can redistribute it and/or modify it under +the terms of the GNU General Public License as published by the Free +Software Foundation; either version 2, or (at your option) any later +version. + +Pazpar2 is distributed in the hope that it will be useful, but WITHOUT ANY +WARRANTY; without even the implied warranty of MERCHANTABILITY or +FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License +for more details. + +You should have received a copy of the GNU General Public License +along with this program; if not, write to the Free Software +Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA + +*/ +#if HAVE_CONFIG_H +#include +#endif #include #include -#include +#if HAVE_UNISTD_H #include +#endif #include -#include -#include +#include +#if HAVE_SYS_TIME_H #include - +#endif +#include #include -#include "command.h" #include "util.h" #include "eventl.h" #include "pazpar2.h" #include "http.h" #include "http_command.h" +#include "settings.h" +#include "client.h" -extern struct parameters global_parameters; -extern IOCHAN channel_list; +// Update this when the protocol changes +#define PAZPAR2_PROTOCOL_VERSION "1" struct http_session { IOCHAN timeout_iochan; // NOTE: This is NOT associated with a socket struct session *psession; unsigned int session_id; int timestamp; + NMEM nmem; struct http_session *next; }; static struct http_session *session_list = 0; - void http_session_destroy(struct http_session *s); static void session_timeout(IOCHAN i, int event) @@ -41,19 +62,22 @@ static void session_timeout(IOCHAN i, int event) http_session_destroy(s); } -struct http_session *http_session_create() +struct http_session *http_session_create(void) { - struct http_session *r = xmalloc(sizeof(*r)); - r->psession = new_session(); + NMEM nmem = nmem_create(); + struct http_session *r = nmem_malloc(nmem, sizeof(*r)); + + r->psession = new_session(nmem); r->session_id = 0; r->timestamp = 0; + r->nmem = nmem; r->next = session_list; session_list = r; r->timeout_iochan = iochan_create(-1, session_timeout, 0); iochan_setdata(r->timeout_iochan, r); iochan_settimeout(r->timeout_iochan, global_parameters.session_timeout); - r->timeout_iochan->next = channel_list; - channel_list = r->timeout_iochan; + + pazpar2_add_channel(r->timeout_iochan); return r; } @@ -67,36 +91,94 @@ void http_session_destroy(struct http_session *s) *p = (*p)->next; break; } + yaz_log(YLOG_LOG, "Destroying session %u", s->session_id); iochan_destroy(s->timeout_iochan); destroy_session(s->psession); - xfree(s); + nmem_destroy(s->nmem); } -static void error(struct http_response *rs, char *code, char *msg, char *txt) +static const char *get_msg(enum pazpar2_error_code code) { - struct http_channel *c = rs->channel; - char tmp[1024]; + struct pazpar2_error_msg { + enum pazpar2_error_code code; + const char *msg; + }; + static const struct pazpar2_error_msg ar[] = { + { PAZPAR2_NO_SESSION, "Session does not exist or it has expired"}, + { PAZPAR2_MISSING_PARAMETER, "Missing parameter"}, + { PAZPAR2_MALFORMED_PARAMETER_VALUE, "Malformed parameter value"}, + { PAZPAR2_MALFORMED_PARAMETER_ENCODING, "Malformed parameter encoding"}, + { PAZPAR2_MALFORMED_SETTING, "Malformed setting argument"}, + { PAZPAR2_HITCOUNTS_FAILED, "Failed to retrieve hitcounts"}, + { PAZPAR2_RECORD_MISSING, "Record missing"}, + { PAZPAR2_NO_TARGETS, "No targets"}, + { PAZPAR2_CONFIG_TARGET, "Target cannot be configured"}, + { PAZPAR2_RECORD_FAIL, "Record command failed"}, + { PAZPAR2_NOT_IMPLEMENTED, "Not implemented"}, + { PAZPAR2_LAST_ERROR, "Last error"}, + { 0, 0 } + }; + int i = 0; + while (ar[i].msg) + { + if (code == ar[i].code) + return ar[i].msg; + i++; + } + return "No error"; +} - if (!txt) - txt = msg; +static void error(struct http_response *rs, + enum pazpar2_error_code code, + const char *addinfo) +{ + struct http_channel *c = rs->channel; + WRBUF text = wrbuf_alloc(); + const char *http_status = "417"; + const char *msg = get_msg(code); + rs->msg = nmem_strdup(c->nmem, msg); - strcpy(rs->code, code); - sprintf(tmp, "%s", txt); - rs->payload = nmem_strdup(c->nmem, tmp); + strcpy(rs->code, http_status); + + wrbuf_printf(text, "", (int) code, + msg); + if (addinfo) + wrbuf_xmlputs(text, addinfo); + wrbuf_puts(text, ""); + + yaz_log(YLOG_WARN, "HTTP %s %s%s%s", http_status, + msg, addinfo ? ": " : "" , addinfo ? addinfo : ""); + rs->payload = nmem_strdup(c->nmem, wrbuf_cstr(text)); + wrbuf_destroy(text); http_send_response(c); } -unsigned int make_sessionid() +unsigned int make_sessionid(void) { - struct timeval t; - unsigned int res; static int seq = 0; + unsigned int res; seq++; - if (gettimeofday(&t, 0) < 0) - abort(); - res = t.tv_sec; - res = ((res << 8) | (seq & 0xff)) & ((unsigned int) (1 << 31) - 1); + if (global_parameters.debug_mode) + res = seq; + else + { +#ifdef WIN32 + res = seq; +#else + struct timeval t; + + if (gettimeofday(&t, 0) < 0) + { + yaz_log(YLOG_WARN|YLOG_ERRNO, "gettimeofday"); + exit(1); + } + /* at most 256 sessions per second .. + (long long would be more appropriate)*/ + res = t.tv_sec; + res = ((res << 8) | (seq & 0xff)) & ((1U << 31) - 1); +#endif + } return res; } @@ -108,7 +190,7 @@ static struct http_session *locate_session(struct http_request *rq, struct http_ if (!session) { - error(rs, "417", "Must supply session", 0); + error(rs, PAZPAR2_MISSING_PARAMETER, "session"); return 0; } id = atoi(session); @@ -118,32 +200,131 @@ static struct http_session *locate_session(struct http_request *rq, struct http_ iochan_activity(p->timeout_iochan); return p; } - error(rs, "417", "Session does not exist, or it has expired", 0); + error(rs, PAZPAR2_NO_SESSION, session); + return 0; +} + +// Decode settings parameters and apply to session +// Syntax: setting[target]=value +static int process_settings(struct session *se, struct http_request *rq, + struct http_response *rs) +{ + struct http_argument *a; + + for (a = rq->arguments; a; a = a->next) + if (strchr(a->name, '[')) + { + char **res; + int num; + char *dbname; + char *setting; + + // Nmem_strsplit *rules*!!! + nmem_strsplit(se->session_nmem, "[]", a->name, &res, &num); + if (num != 2) + { + error(rs, PAZPAR2_MALFORMED_SETTING, a->name); + return -1; + } + setting = res[0]; + dbname = res[1]; + session_apply_setting(se, dbname, setting, + nmem_strdup(se->session_nmem, a->value)); + } return 0; } static void cmd_exit(struct http_channel *c) { yaz_log(YLOG_WARN, "exit"); - exit(0); + http_close_server(); } - static void cmd_init(struct http_channel *c) { unsigned int sesid; char buf[1024]; + const char *clear = http_argbyname(c->request, "clear"); struct http_session *s = http_session_create(); struct http_response *rs = c->response; yaz_log(YLOG_DEBUG, "HTTP Session init"); + if (!clear || *clear == '0') + session_init_databases(s->psession); + else + yaz_log(YLOG_LOG, "No databases preloaded"); sesid = make_sessionid(); s->session_id = sesid; - sprintf(buf, "OK%u", sesid); + if (process_settings(s->psession, c->request, c->response) < 0) + return; + sprintf(buf, "OK%u" + "" PAZPAR2_PROTOCOL_VERSION "", sesid); rs->payload = nmem_strdup(c->nmem, buf); http_send_response(c); } +static void cmd_settings(struct http_channel *c) +{ + struct http_response *rs = c->response; + struct http_request *rq = c->request; + struct http_session *s = locate_session(rq, rs); + + if (!s) + return; + + if (process_settings(s->psession, rq, rs) < 0) + return; + rs->payload = "OK"; + http_send_response(c); +} + +// Compares two hitsbytarget nodes by hitcount +static int cmp_ht(const void *p1, const void *p2) +{ + const struct hitsbytarget *h1 = p1; + const struct hitsbytarget *h2 = p2; + return h2->hits - h1->hits; +} + +// This implements functionality somewhat similar to 'bytarget', but in a termlist form +static void targets_termlist(WRBUF wrbuf, struct session *se, int num, + NMEM nmem) +{ + struct hitsbytarget *ht; + int count, i; + + ht = hitsbytarget(se, &count, nmem); + qsort(ht, count, sizeof(struct hitsbytarget), cmp_ht); + for (i = 0; i < count && i < num && ht[i].hits > 0; i++) + { + + // do only print terms which have display names + + wrbuf_puts(wrbuf, "\n"); + + wrbuf_puts(wrbuf, ""); + wrbuf_xmlputs(wrbuf, ht[i].id); + wrbuf_puts(wrbuf, "\n"); + + wrbuf_puts(wrbuf, ""); + if (!ht[i].name || !ht[i].name[0]) + wrbuf_xmlputs(wrbuf, "NO TARGET NAME"); + else + wrbuf_xmlputs(wrbuf, ht[i].name); + wrbuf_puts(wrbuf, "\n"); + + wrbuf_printf(wrbuf, "%d\n", ht[i].hits); + + wrbuf_puts(wrbuf, ""); + wrbuf_xmlputs(wrbuf, ht[i].state); + wrbuf_puts(wrbuf, "\n"); + + wrbuf_printf(wrbuf, "%d\n", + ht[i].diagnostic); + wrbuf_puts(wrbuf, "\n"); + } +} + static void cmd_termlist(struct http_channel *c) { struct http_response *rs = c->response; @@ -153,18 +334,26 @@ static void cmd_termlist(struct http_channel *c) int len; int i; char *name = http_argbyname(rq, "name"); + char *nums = http_argbyname(rq, "num"); + int num = 15; + int status; if (!s) return; + status = session_active_clients(s->psession); + if (!name) name = "subject"; if (strlen(name) > 255) return; + if (nums) + num = atoi(nums); wrbuf_rewind(c->wrbuf); - wrbuf_puts(c->wrbuf, ""); + wrbuf_puts(c->wrbuf, "\n"); + wrbuf_printf(c->wrbuf, "%d\n", status); while (*name) { char tname[256]; @@ -175,23 +364,38 @@ static void cmd_termlist(struct http_channel *c) strncpy(tname, name, tp - name); tname[tp - name] = '\0'; - p = termlist(s->psession, tname, &len); - wrbuf_printf(c->wrbuf, "\n\n", tname); - if (p) - for (i = 0; i < len; i++) - { - wrbuf_puts(c->wrbuf, "\n"); - wrbuf_printf(c->wrbuf, "%s", p[i]->term); - wrbuf_printf(c->wrbuf, "%d", p[i]->frequency); - wrbuf_puts(c->wrbuf, ""); - } - wrbuf_puts(c->wrbuf, "\n"); + wrbuf_puts(c->wrbuf, "wrbuf, tname); + wrbuf_puts(c->wrbuf, "\">\n"); + if (!strcmp(tname, "xtargets")) + targets_termlist(c->wrbuf, s->psession, num, c->nmem); + else + { + p = termlist(s->psession, tname, &len); + if (p) + for (i = 0; i < len && i < num; i++){ + // prevnt sending empty term elements + if (!p[i]->term || !p[i]->term[0]) + continue; + + wrbuf_puts(c->wrbuf, ""); + wrbuf_puts(c->wrbuf, ""); + wrbuf_xmlputs(c->wrbuf, p[i]->term); + wrbuf_puts(c->wrbuf, ""); + + wrbuf_printf(c->wrbuf, + "%d", + p[i]->frequency); + wrbuf_puts(c->wrbuf, "\n"); + } + } + wrbuf_puts(c->wrbuf, "\n"); name = tp; if (*name == ',') name++; } - wrbuf_puts(c->wrbuf, ""); - rs->payload = nmem_strdup(rq->channel->nmem, wrbuf_buf(c->wrbuf)); + wrbuf_puts(c->wrbuf, "\n"); + rs->payload = nmem_strdup(rq->channel->nmem, wrbuf_cstr(c->wrbuf)); http_send_response(c); } @@ -206,39 +410,243 @@ static void cmd_bytarget(struct http_channel *c) if (!s) return; - if (!(ht = hitsbytarget(s->psession, &count))) - { - error(rs, "500", "Failed to retrieve hitcounts", 0); - return; - } + ht = hitsbytarget(s->psession, &count, c->nmem); wrbuf_rewind(c->wrbuf); wrbuf_puts(c->wrbuf, "OK"); for (i = 0; i < count; i++) { wrbuf_puts(c->wrbuf, "\n"); - wrbuf_printf(c->wrbuf, "%s\n", ht[i].id); + + wrbuf_puts(c->wrbuf, ""); + wrbuf_xmlputs(c->wrbuf, ht[i].id); + wrbuf_puts(c->wrbuf, "\n"); + + if (ht[i].name && ht[i].name[0]) + { + wrbuf_puts(c->wrbuf, ""); + wrbuf_xmlputs(c->wrbuf, ht[i].name); + wrbuf_puts(c->wrbuf, "\n"); + } + wrbuf_printf(c->wrbuf, "%d\n", ht[i].hits); wrbuf_printf(c->wrbuf, "%d\n", ht[i].diagnostic); wrbuf_printf(c->wrbuf, "%d\n", ht[i].records); - wrbuf_printf(c->wrbuf, "%s\n", ht[i].state); + + wrbuf_puts(c->wrbuf, ""); + wrbuf_xmlputs(c->wrbuf, ht[i].state); + wrbuf_puts(c->wrbuf, "\n"); + wrbuf_puts(c->wrbuf, ""); } wrbuf_puts(c->wrbuf, ""); - rs->payload = nmem_strdup(c->nmem, wrbuf_buf(c->wrbuf)); + rs->payload = nmem_strdup(c->nmem, wrbuf_cstr(c->wrbuf)); http_send_response(c); } -static void show_records(struct http_channel *c) +static void write_metadata(WRBUF w, struct conf_service *service, + struct record_metadata **ml, int full) +{ + int imeta; + + for (imeta = 0; imeta < service->num_metadata; imeta++) + { + struct conf_metadata *cmd = &service->metadata[imeta]; + struct record_metadata *md; + if (!cmd->brief && !full) + continue; + for (md = ml[imeta]; md; md = md->next) + { + wrbuf_printf(w, "\n", cmd->name); + + switch (cmd->type) + { + case Metadata_type_generic: + wrbuf_xmlputs(w, md->data.text.disp); + break; + case Metadata_type_year: + wrbuf_printf(w, "%d", md->data.number.min); + if (md->data.number.min != md->data.number.max) + wrbuf_printf(w, "-%d", md->data.number.max); + break; + default: + wrbuf_puts(w, "[can't represent]"); + } + wrbuf_printf(w, "", cmd->name); + } + } +} + +static void write_subrecord(struct record *r, WRBUF w, + struct conf_service *service, int show_details) +{ + const char *name = session_setting_oneval( + client_get_database(r->client), PZ_NAME); + + wrbuf_puts(w, "client)->database->url); + wrbuf_puts(w, "\" "); + + wrbuf_puts(w, "name=\""); + wrbuf_xmlputs(w, *name ? name : "Unknown"); + wrbuf_puts(w, "\">"); + + write_metadata(w, service, r->metadata, show_details); + wrbuf_puts(w, "\n"); +} + +static void show_raw_record_error(void *data, const char *addinfo) +{ + http_channel_observer_t obs = data; + struct http_channel *c = http_channel_observer_chan(obs); + struct http_response *rs = c->response; + + http_remove_observer(obs); + + error(rs, PAZPAR2_RECORD_FAIL, addinfo); +} + +static void show_raw_record_ok(void *data, const char *buf, size_t sz) +{ + http_channel_observer_t obs = data; + struct http_channel *c = http_channel_observer_chan(obs); + struct http_response *rs = c->response; + + http_remove_observer(obs); + + wrbuf_write(c->wrbuf, buf, sz); + rs->payload = nmem_strdup(c->nmem, wrbuf_cstr(c->wrbuf)); + http_send_response(c); +} + + +static void show_raw_record_ok_binary(void *data, const char *buf, size_t sz) +{ + http_channel_observer_t obs = data; + struct http_channel *c = http_channel_observer_chan(obs); + struct http_response *rs = c->response; + + http_remove_observer(obs); + + wrbuf_write(c->wrbuf, buf, sz); + rs->payload = nmem_strdup(c->nmem, wrbuf_cstr(c->wrbuf)); + + rs->content_type = "application/octet-stream"; + http_send_response(c); +} + + +void show_raw_reset(void *data, struct http_channel *c, void *data2) +{ + //struct client *client = data; + //client_show_raw_remove(client, data2); +} + +static void cmd_record_ready(void *data); + +static void cmd_record(struct http_channel *c) +{ + struct http_response *rs = c->response; + struct http_request *rq = c->request; + struct http_session *s = locate_session(rq, rs); + struct record_cluster *rec; + struct record *r; + struct conf_service *service = global_parameters.server->service; + const char *idstr = http_argbyname(rq, "id"); + const char *offsetstr = http_argbyname(rq, "offset"); + const char *binarystr = http_argbyname(rq, "binary"); + + if (!s) + return; + if (!idstr) + { + error(rs, PAZPAR2_MISSING_PARAMETER, "id"); + return; + } + wrbuf_rewind(c->wrbuf); + if (!(rec = show_single(s->psession, idstr))) + { + if (session_set_watch(s->psession, SESSION_WATCH_RECORD, + cmd_record_ready, c, c) != 0) + { + error(rs, PAZPAR2_RECORD_MISSING, idstr); + } + return; + } + if (offsetstr) + { + int offset = atoi(offsetstr); + const char *syntax = http_argbyname(rq, "syntax"); + const char *esn = http_argbyname(rq, "esn"); + int i; + struct record*r = rec->records; + int binary = 0; + + if (binarystr && *binarystr != '0') + binary = 1; + + for (i = 0; i < offset && r; r = r->next, i++) + ; + if (!r) + { + error(rs, PAZPAR2_RECORD_FAIL, "no record at offset given"); + return; + } + else + { + void *data2; + http_channel_observer_t obs = + http_add_observer(c, r->client, show_raw_reset); + int ret = + client_show_raw_begin(r->client, r->position, syntax, esn, + obs /* data */, + show_raw_record_error, + (binary ? + show_raw_record_ok_binary : + show_raw_record_ok), + &data2, + (binary ? 1 : 0)); + if (ret == -1) + { + http_remove_observer(obs); + error(rs, PAZPAR2_NO_SESSION, 0); + return; + } + } + } + else + { + wrbuf_puts(c->wrbuf, "\n"); + wrbuf_puts(c->wrbuf, ""); + wrbuf_xmlputs(c->wrbuf, rec->recid); + wrbuf_puts(c->wrbuf, "\n"); + write_metadata(c->wrbuf, service, rec->metadata, 1); + for (r = rec->records; r; r = r->next) + write_subrecord(r, c->wrbuf, service, 1); + wrbuf_puts(c->wrbuf, "\n"); + rs->payload = nmem_strdup(c->nmem, wrbuf_cstr(c->wrbuf)); + http_send_response(c); + } +} + +static void cmd_record_ready(void *data) +{ + struct http_channel *c = (struct http_channel *) data; + + cmd_record(c); +} + +static void show_records(struct http_channel *c, int active) { struct http_request *rq = c->request; struct http_response *rs = c->response; struct http_session *s = locate_session(rq, rs); - struct record **rl; - NMEM nmem_show; + struct record_cluster **rl; + struct reclist_sortparms *sp; char *start = http_argbyname(rq, "start"); char *num = http_argbyname(rq, "num"); + char *sort = http_argbyname(rq, "sort"); int startn = 0; int numn = 20; int total; @@ -248,16 +656,27 @@ static void show_records(struct http_channel *c) if (!s) return; + // We haven't counted clients yet if we're called on a block release + if (active < 0) + active = session_active_clients(s->psession); + if (start) startn = atoi(start); if (num) numn = atoi(num); + if (!sort) + sort = "relevance"; + if (!(sp = reclist_parse_sortparms(c->nmem, sort))) + { + error(rs, PAZPAR2_MALFORMED_PARAMETER_VALUE, "sort"); + return; + } - nmem_show = nmem_create(); - rl = show(s->psession, startn, &numn, &total, &total_hits, nmem_show); + rl = show(s->psession, sp, startn, &numn, &total, &total_hits, c->nmem); wrbuf_rewind(c->wrbuf); wrbuf_puts(c->wrbuf, "\nOK\n"); + wrbuf_printf(c->wrbuf, "%d\n", active); wrbuf_printf(c->wrbuf, "%d\n", total); wrbuf_printf(c->wrbuf, "%d\n", total_hits); wrbuf_printf(c->wrbuf, "%d\n", startn); @@ -267,27 +686,31 @@ static void show_records(struct http_channel *c) { int ccount; struct record *p; + struct record_cluster *rec = rl[i]; + struct conf_service *service = global_parameters.server->service; wrbuf_puts(c->wrbuf, "\n"); - wrbuf_printf(c->wrbuf, "%s\n", rl[i]->title); - for (ccount = 1, p = rl[i]->next_cluster; p; p = p->next_cluster, ccount++) - ; + write_metadata(c->wrbuf, service, rec->metadata, 0); + for (ccount = 0, p = rl[i]->records; p; p = p->next, ccount++) + write_subrecord(p, c->wrbuf, service, 0); // subrecs w/o details if (ccount > 1) wrbuf_printf(c->wrbuf, "%d\n", ccount); + wrbuf_puts(c->wrbuf, ""); + wrbuf_xmlputs(c->wrbuf, rec->recid); + wrbuf_puts(c->wrbuf, "\n"); wrbuf_puts(c->wrbuf, "\n"); } wrbuf_puts(c->wrbuf, "\n"); - rs->payload = nmem_strdup(c->nmem, wrbuf_buf(c->wrbuf)); + rs->payload = nmem_strdup(c->nmem, wrbuf_cstr(c->wrbuf)); http_send_response(c); - nmem_destroy(nmem_show); } static void show_records_ready(void *data) { struct http_channel *c = (struct http_channel *) data; - show_records(c); + show_records(c, -1); } static void cmd_show(struct http_channel *c) @@ -296,21 +719,28 @@ static void cmd_show(struct http_channel *c) struct http_response *rs = c->response; struct http_session *s = locate_session(rq, rs); char *block = http_argbyname(rq, "block"); + int status; if (!s) return; + status = session_active_clients(s->psession); + if (block) { - if (!s->psession->reclist || !s->psession->reclist->num_records) + if (status && (!s->psession->reclist || !s->psession->reclist->num_records)) { - session_set_watch(s->psession, SESSION_WATCH_RECORDS, show_records_ready, c); - yaz_log(YLOG_DEBUG, "Blocking on cmd_show"); + // if there is already a watch/block. we do not block this one + if (session_set_watch(s->psession, SESSION_WATCH_SHOW, + show_records_ready, c, c) != 0) + { + yaz_log(YLOG_DEBUG, "Blocking on cmd_show"); + } return; } } - show_records(c); + show_records(c, status); } static void cmd_ping(struct http_channel *c) @@ -324,25 +754,57 @@ static void cmd_ping(struct http_channel *c) http_send_response(c); } +static int utf_8_valid(const char *str) +{ + yaz_iconv_t cd = yaz_iconv_open("utf-8", "utf-8"); + if (cd) + { + /* check that query is UTF-8 encoded */ + char *inbuf = (char *) str; /* we know iconv does not alter this */ + size_t inbytesleft = strlen(inbuf); + + size_t outbytesleft = strlen(inbuf) + 10; + char *out = xmalloc(outbytesleft); + char *outbuf = out; + size_t r = yaz_iconv(cd, &inbuf, &inbytesleft, &outbuf, &outbytesleft); + + /* if OK, try flushing the rest */ + if (r != (size_t) (-1)) + r = yaz_iconv(cd, 0, 0, &outbuf, &outbytesleft); + yaz_iconv_close(cd); + xfree(out); + if (r == (size_t) (-1)) + return 0; + } + return 1; +} + static void cmd_search(struct http_channel *c) { struct http_request *rq = c->request; struct http_response *rs = c->response; struct http_session *s = locate_session(rq, rs); char *query = http_argbyname(rq, "query"); - char *res; + char *filter = http_argbyname(rq, "filter"); + enum pazpar2_error_code code; + const char *addinfo = 0; if (!s) return; if (!query) { - error(rs, "417", "Must supply query", 0); + error(rs, PAZPAR2_MISSING_PARAMETER, "query"); return; } - res = search(s->psession, query); - if (res) + if (!utf_8_valid(query)) { - error(rs, "417", res, res); + error(rs, PAZPAR2_MALFORMED_PARAMETER_ENCODING, "query"); + return; + } + code = search(s->psession, query, filter, &addinfo); + if (code) + { + error(rs, code, addinfo); return; } rs->payload = "OK"; @@ -356,36 +818,71 @@ static void cmd_stat(struct http_channel *c) struct http_response *rs = c->response; struct http_session *s = locate_session(rq, rs); struct statistics stat; + int clients; + + float progress = 0; if (!s) return; + clients = session_active_clients(s->psession); statistics(s->psession, &stat); + if (stat.num_clients > 0) { + progress = (stat.num_clients - clients) / (float)stat.num_clients; + } + wrbuf_rewind(c->wrbuf); wrbuf_puts(c->wrbuf, ""); + wrbuf_printf(c->wrbuf, "%d\n", clients); wrbuf_printf(c->wrbuf, "%d\n", stat.num_hits); wrbuf_printf(c->wrbuf, "%d\n", stat.num_records); wrbuf_printf(c->wrbuf, "%d\n", stat.num_clients); wrbuf_printf(c->wrbuf, "%d\n", stat.num_no_connection); wrbuf_printf(c->wrbuf, "%d\n", stat.num_connecting); - wrbuf_printf(c->wrbuf, "%d\n", stat.num_initializing); - wrbuf_printf(c->wrbuf, "%d\n", stat.num_searching); - wrbuf_printf(c->wrbuf, "%d\n", stat.num_presenting); + wrbuf_printf(c->wrbuf, "%d\n", stat.num_working); wrbuf_printf(c->wrbuf, "%d\n", stat.num_idle); wrbuf_printf(c->wrbuf, "%d\n", stat.num_failed); wrbuf_printf(c->wrbuf, "%d\n", stat.num_error); + wrbuf_printf(c->wrbuf, "%.2f\n", progress); wrbuf_puts(c->wrbuf, ""); - rs->payload = nmem_strdup(c->nmem, wrbuf_buf(c->wrbuf)); + rs->payload = nmem_strdup(c->nmem, wrbuf_cstr(c->wrbuf)); http_send_response(c); } +static void cmd_info(struct http_channel *c) +{ + char yaz_version_str[20]; + struct http_response *rs = c->response; + + wrbuf_rewind(c->wrbuf); + wrbuf_puts(c->wrbuf, "\n"); + wrbuf_puts(c->wrbuf, " \n"); + wrbuf_puts(c->wrbuf, ""); + wrbuf_xmlputs(c->wrbuf, VERSION); + wrbuf_puts(c->wrbuf, ""); + + + yaz_version(yaz_version_str, 0); + wrbuf_puts(c->wrbuf, " wrbuf, YAZ_VERSION); + wrbuf_puts(c->wrbuf, "\">"); + wrbuf_xmlputs(c->wrbuf, yaz_version_str); + wrbuf_puts(c->wrbuf, "\n"); + + wrbuf_puts(c->wrbuf, " \n"); + + wrbuf_puts(c->wrbuf, ""); + rs->payload = nmem_strdup(c->nmem, wrbuf_cstr(c->wrbuf)); + http_send_response(c); +} struct { char *name; void (*fun)(struct http_channel *c); } commands[] = { { "init", cmd_init }, + { "settings", cmd_settings }, { "stat", cmd_stat }, { "bytarget", cmd_bytarget }, { "show", cmd_show }, @@ -393,6 +890,8 @@ struct { { "termlist", cmd_termlist }, { "exit", cmd_exit }, { "ping", cmd_ping }, + { "record", cmd_record }, + { "info", cmd_info }, {0,0} }; @@ -403,9 +902,13 @@ void http_command(struct http_channel *c) int i; c->response = rs; + + http_addheader(rs, "Expires", "Thu, 19 Nov 1981 08:52:00 GMT"); + http_addheader(rs, "Cache-Control", "no-store, no-cache, must-revalidate, post-check=0, pre-check=0"); + if (!command) { - error(rs, "417", "Must supply command", 0); + error(rs, PAZPAR2_MISSING_PARAMETER, "command"); return; } for (i = 0; commands[i].name; i++) @@ -415,7 +918,7 @@ void http_command(struct http_channel *c) break; } if (!commands[i].name) - error(rs, "417", "Unknown command", 0); + error(rs, PAZPAR2_MALFORMED_PARAMETER_VALUE, "command"); return; } @@ -423,7 +926,9 @@ void http_command(struct http_channel *c) /* * Local variables: * c-basic-offset: 4 + * c-file-style: "Stroustrup" * indent-tabs-mode: nil * End: * vim: shiftwidth=4 tabstop=8 expandtab */ +