Bump copyright year
[pazpar2-moved-to-github.git] / src / client.c
1 /* This file is part of Pazpar2.
2    Copyright (C) 2006-2010 Index Data
3
4 Pazpar2 is free software; you can redistribute it and/or modify it under
5 the terms of the GNU General Public License as published by the Free
6 Software Foundation; either version 2, or (at your option) any later
7 version.
8
9 Pazpar2 is distributed in the hope that it will be useful, but WITHOUT ANY
10 WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
12 for more details.
13
14 You should have received a copy of the GNU General Public License
15 along with this program; if not, write to the Free Software
16 Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
17
18 */
19
20 /** \file client.c
21     \brief Z39.50 client 
22 */
23
24 #if HAVE_CONFIG_H
25 #include <config.h>
26 #endif
27
28 #include <stdlib.h>
29 #include <stdio.h>
30 #include <string.h>
31 #if HAVE_SYS_TIME_H
32 #include <sys/time.h>
33 #endif
34 #if HAVE_UNISTD_H
35 #include <unistd.h>
36 #endif
37 #include <signal.h>
38 #include <assert.h>
39
40 #include <yaz/marcdisp.h>
41 #include <yaz/comstack.h>
42 #include <yaz/tcpip.h>
43 #include <yaz/proto.h>
44 #include <yaz/readconf.h>
45 #include <yaz/pquery.h>
46 #include <yaz/otherinfo.h>
47 #include <yaz/yaz-util.h>
48 #include <yaz/nmem.h>
49 #include <yaz/query-charset.h>
50 #include <yaz/querytowrbuf.h>
51 #include <yaz/oid_db.h>
52 #include <yaz/diagbib1.h>
53 #include <yaz/snprintf.h>
54 #include <yaz/rpn2cql.h>
55
56 #define USE_TIMING 0
57 #if USE_TIMING
58 #include <yaz/timing.h>
59 #endif
60
61 #include "pazpar2.h"
62 #include "parameters.h"
63 #include "client.h"
64 #include "connection.h"
65 #include "settings.h"
66 #include "relevance.h"
67
68 /** \brief Represents client state for a connection to one search target */
69 struct client {
70     struct session_database *database;
71     struct connection *connection;
72     struct session *session;
73     char *pquery; // Current search
74     char *cqlquery; // used for SRU targets only
75     Odr_int hits;
76     int record_offset;
77     int maxrecs;
78     int startrecs;
79     int diagnostic;
80     enum client_state state;
81     struct show_raw *show_raw;
82     struct client *next;     // next client in session or next in free list
83     ZOOM_resultset resultset;
84 };
85
86 struct show_raw {
87     int active; // whether this request has been sent to the server
88     int position;
89     int binary;
90     char *syntax;
91     char *esn;
92     void (*error_handler)(void *data, const char *addinfo);
93     void (*record_handler)(void *data, const char *buf, size_t sz);
94     void *data;
95     struct show_raw *next;
96 };
97
98 static const char *client_states[] = {
99     "Client_Connecting",
100     "Client_Idle",
101     "Client_Working",
102     "Client_Error",
103     "Client_Failed",
104     "Client_Disconnected"
105 };
106
107 static struct client *client_freelist = 0; /* thread pr */
108
109 const char *client_get_state_str(struct client *cl)
110 {
111     return client_states[cl->state];
112 }
113
114 enum client_state client_get_state(struct client *cl)
115 {
116     return cl->state;
117 }
118
119 void client_set_state(struct client *cl, enum client_state st)
120 {
121     cl->state = st;
122     if (cl->session)
123     {
124         int no_active = session_active_clients(cl->session);
125         if (no_active == 0)
126             session_alert_watch(cl->session, SESSION_WATCH_SHOW);
127     }
128 }
129
130 static void client_show_raw_error(struct client *cl, const char *addinfo);
131
132 // Close connection and set state to error
133 void client_fatal(struct client *cl)
134 {
135     yaz_log(YLOG_WARN, "Fatal error from %s", client_get_url(cl));
136     connection_destroy(cl->connection);
137     client_set_state(cl, Client_Error);
138 }
139
140 struct connection *client_get_connection(struct client *cl)
141 {
142     return cl->connection;
143 }
144
145 struct session_database *client_get_database(struct client *cl)
146 {
147     return cl->database;
148 }
149
150 struct session *client_get_session(struct client *cl)
151 {
152     return cl->session;
153 }
154
155 const char *client_get_pquery(struct client *cl)
156 {
157     return cl->pquery;
158 }
159
160 static void client_send_raw_present(struct client *cl);
161 static int nativesyntax_to_type(struct session_database *sdb, char *type,
162                                 ZOOM_record rec);
163
164 static void client_show_immediate(
165     ZOOM_resultset resultset, struct session_database *sdb, int position,
166     void *data,
167     void (*error_handler)(void *data, const char *addinfo),
168     void (*record_handler)(void *data, const char *buf, size_t sz),
169     int binary)
170 {
171     ZOOM_record rec = 0;
172     char type[80];
173     const char *buf;
174     int len;
175
176     if (!resultset)
177     {
178         error_handler(data, "no resultset");
179         return;
180     }
181     rec = ZOOM_resultset_record(resultset, position-1);
182     if (!rec)
183     {
184         error_handler(data, "no record");
185         return;
186     }
187     if (binary)
188         strcpy(type, "raw");
189     else
190         nativesyntax_to_type(sdb, type, rec);
191     buf = ZOOM_record_get(rec, type, &len);
192     if (!buf)
193     {
194         error_handler(data, "no record");
195         return;
196     }
197     record_handler(data, buf, len);
198 }
199
200
201 int client_show_raw_begin(struct client *cl, int position,
202                           const char *syntax, const char *esn,
203                           void *data,
204                           void (*error_handler)(void *data, const char *addinfo),
205                           void (*record_handler)(void *data, const char *buf,
206                                                  size_t sz),
207                           int binary)
208 {
209     if (syntax == 0 && esn == 0)
210         client_show_immediate(cl->resultset, client_get_database(cl),
211                               position, data,
212                               error_handler, record_handler,
213                               binary);
214     else
215     {
216         struct show_raw *rr, **rrp;
217
218         if (!cl->connection)
219             return -1;
220     
221
222         rr = xmalloc(sizeof(*rr));
223         rr->position = position;
224         rr->active = 0;
225         rr->data = data;
226         rr->error_handler = error_handler;
227         rr->record_handler = record_handler;
228         rr->binary = binary;
229         if (syntax)
230             rr->syntax = xstrdup(syntax);
231         else
232             rr->syntax = 0;
233         if (esn)
234             rr->esn = xstrdup(esn);
235         else
236             rr->esn = 0;
237         rr->next = 0;
238         
239         for (rrp = &cl->show_raw; *rrp; rrp = &(*rrp)->next)
240             ;
241         *rrp = rr;
242         
243         if (cl->state == Client_Failed)
244         {
245             client_show_raw_error(cl, "client failed");
246         }
247         else if (cl->state == Client_Disconnected)
248         {
249             client_show_raw_error(cl, "client disconnected");
250         }
251         else
252         {
253             client_send_raw_present(cl);
254         }
255     }
256     return 0;
257 }
258
259 static void client_show_raw_delete(struct show_raw *r)
260 {
261     xfree(r->syntax);
262     xfree(r->esn);
263     xfree(r);
264 }
265
266 void client_show_raw_remove(struct client *cl, void *data)
267 {
268     struct show_raw *rr = data;
269     struct show_raw **rrp = &cl->show_raw;
270     while (*rrp != rr)
271         rrp = &(*rrp)->next;
272     if (*rrp)
273     {
274         *rrp = rr->next;
275         client_show_raw_delete(rr);
276     }
277 }
278
279 void client_show_raw_dequeue(struct client *cl)
280 {
281     struct show_raw *rr = cl->show_raw;
282
283     cl->show_raw = rr->next;
284     client_show_raw_delete(rr);
285 }
286
287 static void client_show_raw_error(struct client *cl, const char *addinfo)
288 {
289     while (cl->show_raw)
290     {
291         cl->show_raw->error_handler(cl->show_raw->data, addinfo);
292         client_show_raw_dequeue(cl);
293     }
294 }
295
296 static void client_send_raw_present(struct client *cl)
297 {
298     struct session_database *sdb = client_get_database(cl);
299     struct connection *co = client_get_connection(cl);
300     ZOOM_resultset set = cl->resultset;
301
302     int offset = cl->show_raw->position;
303     const char *syntax = 0;
304     const char *elements = 0;
305
306     assert(cl->show_raw);
307     assert(set);
308
309     yaz_log(YLOG_DEBUG, "%s: trying to present %d record(s) from %d",
310             client_get_url(cl), 1, offset);
311
312     if (cl->show_raw->syntax)
313         syntax = cl->show_raw->syntax;
314     else
315         syntax = session_setting_oneval(sdb, PZ_REQUESTSYNTAX);
316     ZOOM_resultset_option_set(set, "preferredRecordSyntax", syntax);
317
318     if (cl->show_raw->esn)
319         elements = cl->show_raw->esn;
320     else
321         elements = session_setting_oneval(sdb, PZ_ELEMENTS);
322     if (elements && *elements)
323         ZOOM_resultset_option_set(set, "elementSetName", elements);
324
325     ZOOM_resultset_records(set, 0, offset-1, 1);
326     cl->show_raw->active = 1;
327
328     connection_continue(co);
329 }
330
331 static int nativesyntax_to_type(struct session_database *sdb, char *type,
332                                 ZOOM_record rec)
333 {
334     const char *s = session_setting_oneval(sdb, PZ_NATIVESYNTAX);
335
336     if (s && *s)
337     {
338         if (!strncmp(s, "iso2709", 7))
339         {
340             const char *cp = strchr(s, ';');
341             yaz_snprintf(type, 80, "xml; charset=%s", cp ? cp+1 : "marc-8s");
342         }
343         else if (!strncmp(s, "xml", 3))
344         {
345             strcpy(type, "xml");
346         }
347         else
348             return -1;
349         return 0;
350     }
351     else  /* attempt to deduce structure */
352     {
353         const char *syntax = ZOOM_record_get(rec, "syntax", NULL);
354         if (syntax)
355         {
356             if (!strcmp(syntax, "XML"))
357             {
358                 strcpy(type, "xml");
359                 return 0;
360             }
361             else if (!strcmp(syntax, "USmarc") || !strcmp(syntax, "MARC21"))
362             {
363                 strcpy(type, "xml; charset=marc8-s");
364                 return 0;
365             }
366             else return -1;
367         }
368         else return -1;
369     }
370 }
371
372 static void ingest_raw_record(struct client *cl, ZOOM_record rec)
373 {
374     const char *buf;
375     int len;
376     char type[80];
377
378     if (cl->show_raw->binary)
379         strcpy(type, "raw");
380     else
381     {
382         struct session_database *sdb = client_get_database(cl);
383         nativesyntax_to_type(sdb, type, rec);
384     }
385
386     buf = ZOOM_record_get(rec, type, &len);
387     cl->show_raw->record_handler(cl->show_raw->data,  buf, len);
388     client_show_raw_dequeue(cl);
389 }
390
391 void client_search_response(struct client *cl)
392 {
393     struct connection *co = cl->connection;
394     struct session *se = cl->session;
395     ZOOM_connection link = connection_get_link(co);
396     ZOOM_resultset resultset = cl->resultset;
397     const char *error, *addinfo;
398
399     if (ZOOM_connection_error(link, &error, &addinfo))
400     {
401         cl->hits = 0;
402         client_set_state(cl, Client_Error);
403         yaz_log(YLOG_WARN, "Search error %s (%s): %s",
404             error, addinfo, client_get_url(cl));
405     }
406     else
407     {
408         cl->record_offset = cl->startrecs;
409         cl->hits = ZOOM_resultset_size(resultset);
410         se->total_hits += cl->hits;
411     }
412 }
413
414
415 void client_record_response(struct client *cl)
416 {
417     struct connection *co = cl->connection;
418     ZOOM_connection link = connection_get_link(co);
419     ZOOM_resultset resultset = cl->resultset;
420     const char *error, *addinfo;
421
422     if (ZOOM_connection_error(link, &error, &addinfo))
423     {
424         client_set_state(cl, Client_Error);
425         yaz_log(YLOG_WARN, "Search error %s (%s): %s",
426             error, addinfo, client_get_url(cl));
427     }
428     else
429     {
430         ZOOM_record rec = 0;
431         const char *msg, *addinfo;
432         
433         if (cl->show_raw && cl->show_raw->active)
434         {
435             if ((rec = ZOOM_resultset_record(resultset,
436                                              cl->show_raw->position-1)))
437             {
438                 cl->show_raw->active = 0;
439                 ingest_raw_record(cl, rec);
440             }
441             else
442             {
443                 yaz_log(YLOG_WARN, "Expected record, but got NULL, offset=%d",
444                         cl->show_raw->position-1);
445             }
446         }
447         else
448         {
449             int offset = cl->record_offset;
450             if ((rec = ZOOM_resultset_record(resultset, offset)))
451             {
452                 cl->record_offset++;
453                 if (ZOOM_record_error(rec, &msg, &addinfo, 0))
454                     yaz_log(YLOG_WARN, "Record error %s (%s): %s (rec #%d)",
455                             error, addinfo, client_get_url(cl),
456                             cl->record_offset);
457                 else
458                 {
459                     struct session_database *sdb = client_get_database(cl);
460                     const char *xmlrec;
461                     char type[80];
462                     if (nativesyntax_to_type(sdb, type, rec))
463                         yaz_log(YLOG_WARN, "Failed to determine record type");
464                     if ((xmlrec = ZOOM_record_get(rec, type, NULL)))
465                     {
466                         if (ingest_record(cl, xmlrec, cl->record_offset))
467                         {
468                             session_alert_watch(cl->session, SESSION_WATCH_SHOW);
469                             session_alert_watch(cl->session, SESSION_WATCH_RECORD);
470                         }
471                         else
472                             yaz_log(YLOG_WARN, "Failed to ingest");
473                     }
474                     else
475                         yaz_log(YLOG_WARN, "Failed to extract ZOOM record");
476                 }
477
478             }
479             else
480             {
481                 yaz_log(YLOG_WARN, "Expected record, but got NULL, offset=%d",
482                         offset);
483             }
484         }
485     }
486 }
487
488 void client_start_search(struct client *cl)
489 {
490     struct session_database *sdb = client_get_database(cl);
491     struct connection *co = client_get_connection(cl);
492     ZOOM_connection link = connection_get_link(co);
493     ZOOM_resultset rs;
494     char *databaseName = sdb->database->databases[0];
495     const char *opt_piggyback = session_setting_oneval(sdb, PZ_PIGGYBACK);
496     const char *opt_queryenc = session_setting_oneval(sdb, PZ_QUERYENCODING);
497     const char *opt_elements = session_setting_oneval(sdb, PZ_ELEMENTS);
498     const char *opt_requestsyn = session_setting_oneval(sdb, PZ_REQUESTSYNTAX);
499     const char *opt_maxrecs = session_setting_oneval(sdb, PZ_MAXRECS);
500     const char *opt_sru = session_setting_oneval(sdb, PZ_SRU);
501     const char *opt_sort = session_setting_oneval(sdb, PZ_SORT);
502     char maxrecs_str[24], startrecs_str[24];
503
504     assert(link);
505
506     cl->hits = -1;
507     cl->record_offset = 0;
508     cl->diagnostic = 0;
509     client_set_state(cl, Client_Working);
510
511     if (*opt_piggyback)
512         ZOOM_connection_option_set(link, "piggyback", opt_piggyback);
513     else
514         ZOOM_connection_option_set(link, "piggyback", "1");
515     if (*opt_queryenc)
516         ZOOM_connection_option_set(link, "rpnCharset", opt_queryenc);
517     if (*opt_sru && *opt_elements)
518         ZOOM_connection_option_set(link, "schema", opt_elements);
519     else if (*opt_elements)
520         ZOOM_connection_option_set(link, "elementSetName", opt_elements);
521     if (*opt_requestsyn)
522         ZOOM_connection_option_set(link, "preferredRecordSyntax", opt_requestsyn);
523
524     if (!*opt_maxrecs)
525     {
526         sprintf(maxrecs_str, "%d", cl->maxrecs);
527         opt_maxrecs = maxrecs_str;
528     }
529     ZOOM_connection_option_set(link, "count", opt_maxrecs);
530
531
532     if (atoi(opt_maxrecs) > 20)
533         ZOOM_connection_option_set(link, "presentChunk", "20");
534     else
535         ZOOM_connection_option_set(link, "presentChunk", opt_maxrecs);
536
537     sprintf(startrecs_str, "%d", cl->startrecs);
538     ZOOM_connection_option_set(link, "start", startrecs_str);
539
540     if (databaseName)
541         ZOOM_connection_option_set(link, "databaseName", databaseName);
542
543     if (cl->cqlquery)
544     {
545         ZOOM_query q = ZOOM_query_create();
546         yaz_log(YLOG_LOG, "Search %s CQL: %s", sdb->database->url, cl->cqlquery);
547         ZOOM_query_cql(q, cl->cqlquery);
548         if (*opt_sort)
549             ZOOM_query_sortby(q, opt_sort);
550         rs = ZOOM_connection_search(link, q);
551         ZOOM_query_destroy(q);
552     }
553     else
554     {
555         yaz_log(YLOG_LOG, "Search %s PQF: %s", sdb->database->url, cl->pquery);
556         rs = ZOOM_connection_search_pqf(link, cl->pquery);
557     }
558     ZOOM_resultset_destroy(cl->resultset);
559     cl->resultset = rs;
560     connection_continue(co);
561 }
562
563 struct client *client_create(void)
564 {
565     struct client *r;
566     if (client_freelist)
567     {
568         r = client_freelist;
569         client_freelist = client_freelist->next;
570     }
571     else
572         r = xmalloc(sizeof(struct client));
573     r->maxrecs = 100;
574     r->startrecs = 0;
575     r->pquery = 0;
576     r->cqlquery = 0;
577     r->database = 0;
578     r->connection = 0;
579     r->session = 0;
580     r->hits = 0;
581     r->record_offset = 0;
582     r->diagnostic = 0;
583     r->state = Client_Disconnected;
584     r->show_raw = 0;
585     r->resultset = 0;
586     r->next = 0;
587     return r;
588 }
589
590 void client_destroy(struct client *c)
591 {
592     struct session *se = c->session;
593     if (c == se->clients)
594         se->clients = c->next;
595     else
596     {
597         struct client *cc;
598         for (cc = se->clients; cc && cc->next != c; cc = cc->next)
599             ;
600         if (cc)
601             cc->next = c->next;
602     }
603     xfree(c->pquery);
604     xfree(c->cqlquery);
605
606     if (c->connection)
607         connection_release(c->connection);
608
609     ZOOM_resultset_destroy(c->resultset);
610     c->resultset = 0;
611     c->next = client_freelist;
612     client_freelist = c;
613 }
614
615 void client_set_connection(struct client *cl, struct connection *con)
616 {
617     cl->connection = con;
618 }
619
620 void client_disconnect(struct client *cl)
621 {
622     if (cl->state != Client_Idle)
623         client_set_state(cl, Client_Disconnected);
624     client_set_connection(cl, 0);
625 }
626
627 // Extract terms from query into null-terminated termlist
628 static void extract_terms(NMEM nmem, struct ccl_rpn_node *query, char **termlist)
629 {
630     int num = 0;
631
632     pull_terms(nmem, query, termlist, &num);
633     termlist[num] = 0;
634 }
635
636 // Initialize CCL map for a target
637 static CCL_bibset prepare_cclmap(struct client *cl)
638 {
639     struct session_database *sdb = client_get_database(cl);
640     struct setting *s;
641     CCL_bibset res;
642
643     if (!sdb->settings)
644         return 0;
645     res = ccl_qual_mk();
646     for (s = sdb->settings[PZ_CCLMAP]; s; s = s->next)
647     {
648         char *p = strchr(s->name + 3, ':');
649         if (!p)
650         {
651             yaz_log(YLOG_WARN, "Malformed cclmap name: %s", s->name);
652             ccl_qual_rm(&res);
653             return 0;
654         }
655         p++;
656         ccl_qual_fitem(res, s->value, p);
657     }
658     return res;
659 }
660
661 // returns a xmalloced CQL query corresponding to the pquery in client
662 static char *make_cqlquery(struct client *cl)
663 {
664     cql_transform_t cqlt = cql_transform_create();
665     Z_RPNQuery *zquery;
666     char *r;
667     WRBUF wrb = wrbuf_alloc();
668     int status;
669     ODR odr_out = odr_createmem(ODR_ENCODE);
670
671     zquery = p_query_rpn(odr_out, cl->pquery);
672     yaz_log(YLOG_LOG, "PQF: %s", cl->pquery);
673     if ((status = cql_transform_rpn2cql_wrbuf(cqlt, wrb, zquery)))
674     {
675         yaz_log(YLOG_WARN, "Failed to generate CQL query, code=%d", status);
676         r = 0;
677     }
678     else
679     {
680         r = xstrdup(wrbuf_cstr(wrb));
681     }     
682     wrbuf_destroy(wrb);
683     odr_destroy(odr_out);
684     cql_transform_close(cqlt);
685     return r;
686 }
687
688 // Parse the query given the settings specific to this client
689 int client_parse_query(struct client *cl, const char *query)
690 {
691     struct session *se = client_get_session(cl);
692     struct session_database *sdb = client_get_database(cl);
693     struct ccl_rpn_node *cn;
694     int cerror, cpos;
695     CCL_bibset ccl_map = prepare_cclmap(cl);
696     const char *sru = session_setting_oneval(sdb, PZ_SRU);
697     const char *pqf_prefix = session_setting_oneval(sdb, PZ_PQF_PREFIX);
698     const char *pqf_strftime = session_setting_oneval(sdb, PZ_PQF_STRFTIME);
699
700     if (!ccl_map)
701         return -1;
702
703     cn = ccl_find_str(ccl_map, query, &cerror, &cpos);
704     ccl_qual_rm(&ccl_map);
705     if (!cn)
706     {
707         client_set_state(cl, Client_Error);
708         yaz_log(YLOG_WARN, "Failed to parse CCL query %s for %s",
709                 query,
710                 client_get_database(cl)->database->url);
711         return -1;
712     }
713     wrbuf_rewind(se->wrbuf);
714     if (*pqf_prefix)
715     {
716         wrbuf_puts(se->wrbuf, pqf_prefix);
717         wrbuf_puts(se->wrbuf, " ");
718     }
719     if (!pqf_strftime || !*pqf_strftime)
720         ccl_pquery(se->wrbuf, cn);
721     else
722     {
723         time_t cur_time = time(0);
724         struct tm *tm =  localtime(&cur_time);
725         char tmp_str[300];
726         const char *cp = tmp_str;
727
728         /* see man strftime(3) for things .. In particular %% gets converted
729          to %.. And That's our original query .. */
730         strftime(tmp_str, sizeof(tmp_str)-1, pqf_strftime, tm);
731         for (; *cp; cp++)
732         {
733             if (cp[0] == '%')
734                 ccl_pquery(se->wrbuf, cn);
735             else
736                 wrbuf_putc(se->wrbuf, cp[0]);
737         }
738     }
739     xfree(cl->pquery);
740     cl->pquery = xstrdup(wrbuf_cstr(se->wrbuf));
741
742     xfree(cl->cqlquery);
743     if (*sru)
744     {
745         if (!(cl->cqlquery = make_cqlquery(cl)))
746             return -1;
747     }
748     else
749         cl->cqlquery = 0;
750
751     if (!se->relevance)
752     {
753         // Initialize relevance structure with query terms
754         char *p[512];
755         extract_terms(se->nmem, cn, p);
756         se->relevance = relevance_create(
757             se->service->relevance_pct,
758             se->nmem, (const char **) p);
759     }
760
761     ccl_rpn_delete(cn);
762     return 0;
763 }
764
765 void client_set_session(struct client *cl, struct session *se)
766 {
767     cl->session = se;
768     cl->next = se->clients;
769     se->clients = cl;
770 }
771
772 int client_is_active(struct client *cl)
773 {
774     if (cl->connection && (cl->state == Client_Connecting ||
775                            cl->state == Client_Working))
776         return 1;
777     return 0;
778 }
779
780 struct client *client_next_in_session(struct client *cl)
781 {
782     if (cl)
783         return cl->next;
784     return 0;
785
786 }
787
788 Odr_int client_get_hits(struct client *cl)
789 {
790     return cl->hits;
791 }
792
793 int client_get_num_records(struct client *cl)
794 {
795     return cl->record_offset;
796 }
797
798 void client_set_diagnostic(struct client *cl, int diagnostic)
799 {
800     cl->diagnostic = diagnostic;
801 }
802
803 int client_get_diagnostic(struct client *cl)
804 {
805     return cl->diagnostic;
806 }
807
808 void client_set_database(struct client *cl, struct session_database *db)
809 {
810     cl->database = db;
811 }
812
813 struct host *client_get_host(struct client *cl)
814 {
815     return client_get_database(cl)->database->host;
816 }
817
818 const char *client_get_url(struct client *cl)
819 {
820     return client_get_database(cl)->database->url;
821 }
822
823 void client_set_maxrecs(struct client *cl, int v)
824 {
825     cl->maxrecs = v;
826 }
827
828 void client_set_startrecs(struct client *cl, int v)
829 {
830     cl->startrecs = v;
831 }
832
833 /*
834  * Local variables:
835  * c-basic-offset: 4
836  * c-file-style: "Stroustrup"
837  * indent-tabs-mode: nil
838  * End:
839  * vim: shiftwidth=4 tabstop=8 expandtab
840  */
841