2 * Copyright (c) 1995-2002, Index Data.
3 * See the file LICENSE for details.
5 * $Id: pquery.c,v 1.14 2002-05-01 10:22:52 adam Exp $
12 #include <yaz/proto.h>
14 #include <yaz/pquery.h>
16 static oid_value p_query_dfset = VAL_NONE;
19 const char *query_buf;
29 static Z_RPNStructure *rpn_structure (struct lex_info *li, ODR o, oid_proto,
30 int num_attr, int max_attr,
31 int *attr_list, char **attr_clist,
34 static enum oid_value query_oid_getvalbyname (struct lex_info *li)
41 memcpy (buf, li->lex_buf, li->lex_len);
42 buf[li->lex_len] = '\0';
43 value = oid_getvalbyname (buf);
47 static int compare_term (struct lex_info *li, const char *src, size_t off)
49 size_t len=strlen(src);
51 if (li->lex_len == len+off && !memcmp (li->lex_buf+off, src, len-off))
56 static int query_token (struct lex_info *li)
59 const char *sep_match;
60 const char **qptr = &li->query_buf;
67 if ((sep_match = strchr (li->left_sep, **qptr)))
69 sep_char = li->right_sep[sep_match - li->left_sep];
74 if (**qptr == li->escape_char && isdigit ((*qptr)[1]))
80 while (**qptr && **qptr != sep_char)
92 if (sep_char == ' ' &&
93 li->lex_len >= 1 && li->lex_buf[0] == li->escape_char)
95 if (compare_term (li, "and", 1))
97 if (compare_term (li, "or", 1))
99 if (compare_term (li, "not", 1))
101 if (compare_term (li, "attr", 1))
103 if (compare_term (li, "set", 1))
105 if (compare_term (li, "attrset", 1))
107 if (compare_term (li, "prox", 1))
109 if (compare_term (li, "term", 1))
115 static int lex (struct lex_info *li)
117 return li->query_look = query_token (li);
120 static int escape_string(char *out_buf, const char *in, int len)
125 if (*in == '\\' && len > 0)
151 sscanf (s, "%x", &n);
168 sscanf (s, "%o", &n);
180 return out - out_buf;
183 static int p_query_parse_attr(struct lex_info *li, ODR o,
184 int num_attr, int *attr_list,
185 char **attr_clist, oid_value *attr_set)
188 if (!(cp = strchr (li->lex_buf, '=')) ||
189 (size_t) (cp-li->lex_buf) > li->lex_len)
191 attr_set[num_attr] = query_oid_getvalbyname (li);
192 if (attr_set[num_attr] == VAL_NONE)
196 if (!(cp = strchr (li->lex_buf, '=')))
202 attr_set[num_attr] = attr_set[num_attr-1];
204 attr_set[num_attr] = VAL_NONE;
206 attr_list[2*num_attr] = atoi(li->lex_buf);
208 if (*cp >= '0' && *cp <= '9')
210 attr_list[2*num_attr+1] = atoi (cp);
211 attr_clist[num_attr] = 0;
215 int len = li->lex_len - (cp - li->lex_buf);
216 attr_list[2*num_attr+1] = 0;
217 attr_clist[num_attr] = (char *) odr_malloc (o, len+1);
218 len = escape_string(attr_clist[num_attr], cp, len);
219 attr_clist[num_attr][len] = '\0';
224 static Z_AttributesPlusTerm *rpn_term (struct lex_info *li, ODR o,
226 int num_attr, int *attr_list,
227 char **attr_clist, oid_value *attr_set)
229 Z_AttributesPlusTerm *zapt;
232 Z_AttributeElement **elements;
234 zapt = (Z_AttributesPlusTerm *)odr_malloc (o, sizeof(*zapt));
235 term_octet = (Odr_oct *)odr_malloc (o, sizeof(*term_octet));
236 term = (Z_Term *)odr_malloc (o, sizeof(*term));
239 elements = (Z_AttributeElement**)odr_nullval();
245 elements = (Z_AttributeElement**)
246 odr_malloc (o, num_attr * sizeof(*elements));
248 attr_tmp = (int *)odr_malloc (o, num_attr * 2 * sizeof(int));
249 memcpy (attr_tmp, attr_list, num_attr * 2 * sizeof(int));
250 for (i = num_attr; --i >= 0; )
253 for (j = i+1; j<num_attr; j++)
254 if (attr_tmp[2*j] == attr_tmp[2*i])
259 (Z_AttributeElement*)odr_malloc (o,sizeof(**elements));
260 elements[k]->attributeType = &attr_tmp[2*i];
261 elements[k]->attributeSet =
262 yaz_oidval_to_z3950oid(o, CLASS_ATTSET, attr_set[i]);
266 elements[k]->which = Z_AttributeValue_complex;
267 elements[k]->value.complex = (Z_ComplexAttribute *)
268 odr_malloc (o, sizeof(Z_ComplexAttribute));
269 elements[k]->value.complex->num_list = 1;
270 elements[k]->value.complex->list =
271 (Z_StringOrNumeric **)
272 odr_malloc (o, 1 * sizeof(Z_StringOrNumeric *));
273 elements[k]->value.complex->list[0] =
274 (Z_StringOrNumeric *)
275 odr_malloc (o, sizeof(Z_StringOrNumeric));
276 elements[k]->value.complex->list[0]->which =
277 Z_StringOrNumeric_string;
278 elements[k]->value.complex->list[0]->u.string =
280 elements[k]->value.complex->semanticAction = (int **)
282 elements[k]->value.complex->num_semanticAction = 0;
286 elements[k]->which = Z_AttributeValue_numeric;
287 elements[k]->value.numeric = &attr_tmp[2*i+1];
293 zapt->attributes = (Z_AttributeList *)
294 odr_malloc (o, sizeof(*zapt->attributes));
295 zapt->attributes->num_attributes = num_attr;
296 zapt->attributes->attributes = elements;
299 term->which = Z_Term_general;
300 term->u.general = term_octet;
301 term_octet->buf = (unsigned char *)odr_malloc (o, li->lex_len);
302 term_octet->size = term_octet->len =
303 escape_string ((char *) (term_octet->buf), li->lex_buf, li->lex_len);
307 static Z_Operand *rpn_simple (struct lex_info *li, ODR o, oid_proto proto,
308 int num_attr, int *attr_list, char **attr_clist,
313 zo = (Z_Operand *)odr_malloc (o, sizeof(*zo));
314 switch (li->query_look)
317 zo->which = Z_Operand_APT;
318 if (!(zo->u.attributesPlusTerm =
319 rpn_term (li, o, proto, num_attr, attr_list, attr_clist,
328 zo->which = Z_Operand_resultSetId;
329 zo->u.resultSetId = (char *)odr_malloc (o, li->lex_len+1);
330 memcpy (zo->u.resultSetId, li->lex_buf, li->lex_len);
331 zo->u.resultSetId[li->lex_len] = '\0';
340 static Z_ProximityOperator *rpn_proximity (struct lex_info *li, ODR o)
342 Z_ProximityOperator *p = (Z_ProximityOperator *)odr_malloc (o, sizeof(*p));
346 if (*li->lex_buf == '1')
348 p->exclusion = (int *)odr_malloc (o, sizeof(*p->exclusion));
351 else if (*li->lex_buf == '0')
353 p->exclusion = (int *)odr_malloc (o, sizeof(*p->exclusion));
361 p->distance = (int *)odr_malloc (o, sizeof(*p->distance));
362 *p->distance = atoi (li->lex_buf);
366 p->ordered = (int *)odr_malloc (o, sizeof(*p->ordered));
367 *p->ordered = atoi (li->lex_buf);
371 p->relationType = (int *)odr_malloc (o, sizeof(*p->relationType));
372 *p->relationType = atoi (li->lex_buf);
376 if (*li->lex_buf == 'k')
378 else if (*li->lex_buf == 'p')
381 p->which = atoi (li->lex_buf);
385 p->which = Z_ProximityOperator_known;
386 p->u.known = (int *)odr_malloc (o, sizeof(*p->u.known));
387 *p->u.known = atoi (li->lex_buf);
391 static Z_Complex *rpn_complex (struct lex_info *li, ODR o, oid_proto proto,
392 int num_attr, int max_attr,
393 int *attr_list, char **attr_clist,
399 zc = (Z_Complex *)odr_malloc (o, sizeof(*zc));
400 zo = (Z_Operator *)odr_malloc (o, sizeof(*zo));
402 switch (li->query_look)
405 zo->which = Z_Operator_and;
406 zo->u.and_not = odr_nullval();
409 zo->which = Z_Operator_or;
410 zo->u.and_not = odr_nullval();
413 zo->which = Z_Operator_and_not;
414 zo->u.and_not = odr_nullval();
417 zo->which = Z_Operator_prox;
418 zo->u.prox = rpn_proximity (li, o);
427 rpn_structure (li, o, proto, num_attr, max_attr, attr_list,
428 attr_clist, attr_set)))
431 rpn_structure (li, o, proto, num_attr, max_attr, attr_list,
432 attr_clist, attr_set)))
437 static Z_RPNStructure *rpn_structure (struct lex_info *li, ODR o,
439 int num_attr, int max_attr,
446 sz = (Z_RPNStructure *)odr_malloc (o, sizeof(*sz));
447 switch (li->query_look)
453 sz->which = Z_RPNStructure_complex;
454 if (!(sz->u.complex =
455 rpn_complex (li, o, proto, num_attr, max_attr, attr_list,
456 attr_clist, attr_set)))
461 sz->which = Z_RPNStructure_simple;
463 rpn_simple (li, o, proto, num_attr, attr_list,
464 attr_clist, attr_set)))
471 if (num_attr >= max_attr)
473 if (!p_query_parse_attr(li, o, num_attr, attr_list,
474 attr_clist, attr_set))
479 rpn_structure (li, o, proto, num_attr, max_attr, attr_list,
480 attr_clist, attr_set);
485 if (compare_term (li, "general", 0))
486 li->term_type = Z_Term_general;
487 else if (compare_term (li, "numeric", 0))
488 li->term_type = Z_Term_numeric;
489 else if (compare_term (li, "string", 0))
490 li->term_type = Z_Term_characterString;
491 else if (compare_term (li, "oid", 0))
492 li->term_type = Z_Term_oid;
493 else if (compare_term (li, "datetime", 0))
494 li->term_type = Z_Term_dateTime;
495 else if (compare_term (li, "null", 0))
496 li->term_type = Z_Term_null;
499 rpn_structure (li, o, proto, num_attr, max_attr, attr_list,
500 attr_clist, attr_set);
501 case 0: /* operator/operand expected! */
507 Z_RPNQuery *p_query_rpn_mk (ODR o, struct lex_info *li, oid_proto proto,
511 int attr_array[1024];
512 char *attr_clist[512];
513 oid_value attr_set[512];
514 oid_value topSet = VAL_NONE;
516 zq = (Z_RPNQuery *)odr_malloc (o, sizeof(*zq));
518 if (li->query_look == 'r')
521 topSet = query_oid_getvalbyname (li);
522 if (topSet == VAL_NONE)
527 if (topSet == VAL_NONE)
528 topSet = p_query_dfset;
529 if (topSet == VAL_NONE)
532 zq->attributeSetId = yaz_oidval_to_z3950oid(o, CLASS_ATTSET, topSet);
534 if (!zq->attributeSetId)
537 if (!(zq->RPNStructure = rpn_structure (li, o, proto, 0, 512,
538 attr_array, attr_clist, attr_set)))
543 Z_RPNQuery *p_query_rpn (ODR o, oid_proto proto,
549 li.right_sep = "}\"";
550 li.escape_char = '@';
551 li.term_type = Z_Term_general;
553 return p_query_rpn_mk (o, &li, proto, qbuf);
557 Z_AttributesPlusTerm *p_query_scan_mk (struct lex_info *li,
558 ODR o, oid_proto proto,
559 Odr_oid **attributeSetP,
563 char *attr_clist[512];
564 oid_value attr_set[512];
567 oid_value topSet = VAL_NONE;
570 if (li->query_look == 'r')
573 topSet = query_oid_getvalbyname (li);
577 if (topSet == VAL_NONE)
578 topSet = p_query_dfset;
579 if (topSet == VAL_NONE)
582 *attributeSetP = yaz_oidval_to_z3950oid (o, CLASS_ATTSET, topSet);
584 while (li->query_look == 'l')
589 if (num_attr >= max_attr)
591 if (!p_query_parse_attr(li, o, num_attr, attr_list,
592 attr_clist, attr_set))
599 return rpn_term (li, o, proto, num_attr, attr_list, attr_clist, attr_set);
602 Z_AttributesPlusTerm *p_query_scan (ODR o, oid_proto proto,
603 Odr_oid **attributeSetP,
609 li.right_sep = "}\"";
610 li.escape_char = '@';
611 li.term_type = Z_Term_general;
614 return p_query_scan_mk (&li, o, proto, attributeSetP, qbuf);
617 int p_query_attset (const char *arg)
619 p_query_dfset = oid_getvalbyname (arg);
620 return (p_query_dfset == VAL_NONE) ? -1 : 0;